Make use of the mspmath library

[r2c2.git] / source / libr2c2 / layout.h

diff --git a/source/libr2c2/layout.h b/source/libr2c2/layout.h
index b89cacb90b1d11a5c1279d0ff2c11b5f9f9ef464..48a2796dcb7003affb134f99471e03ebcefb3324 100644 (file)
--- a/source/libr2c2/layout.h
+++ b/source/libr2c2/layout.h
@@ -6,6 +6,7 @@
 #include <msp/datafile/objectloader.h>
 #include <msp/time/timestamp.h>
 #include "block.h"
+#include "geometry.h"
 
 namespace R2C2 {
 
@@ -13,9 +14,9 @@ class ArticleNumber;
 class Catalogue;
 class Driver;
 class Route;
+class Signal;
 class Track;
 class Train;
-class Vector;
 class Vehicle;
 class Zone;
 
@@ -30,9 +31,9 @@ public:
        public:
                Loader(Layout &);
        private:
-               virtual void finish();
                void route();
                void route(const std::string &);
+               void signal(ArticleNumber);
                void track(unsigned);
                void track(ArticleNumber);
                void train(unsigned, unsigned, const std::string &);
@@ -46,6 +47,8 @@ public:
 
        sigc::signal<void, Track &> signal_track_added;
        sigc::signal<void, Track &> signal_track_removed;
+       sigc::signal<void, Signal &> signal_signal_added;
+       sigc::signal<void, Signal &> signal_signal_removed;
        sigc::signal<void, Route &> signal_route_added;
        sigc::signal<void, Route &> signal_route_removed;
        sigc::signal<void, Zone &> signal_zone_added;
@@ -63,6 +66,7 @@ private:
        Driver *driver;
        std::string base;
        std::set<Track *> tracks;
+       std::set<Signal *> signals;
        std::set<Route *> routes;
        ZoneSet zones;
        std::set<Block *> blocks;
@@ -86,6 +90,18 @@ public:
        void remove_track(Track &);
        unsigned allocate_turnout_id();
 
+       void add_signal(Signal &);
+       const std::set<Signal *> &get_signals() const { return signals; }
+       Signal *pick_signal(const Vector &, const Vector &);
+       void remove_signal(Signal &);
+
+       Object *pick_object(const Vector &, const Vector &);
+
+private:
+       template<typename T>
+       static T *pick(const std::set<T *> &, const Vector &, const Vector &);
+
+public:
        void add_block(Block &);
        Block &get_block(unsigned) const;
        const std::set<Block *> &get_blocks() const { return blocks; }