]> git.tdb.fi Git - r2c2.git/blob - source/libr2c2/layout.h
Make use of the unified storage class for other parts of Layout
[r2c2.git] / source / libr2c2 / layout.h
1 #ifndef LIBR2C2_LAYOUT_H_
2 #define LIBR2C2_LAYOUT_H_
3
4 #include <set>
5 #include <sigc++/sigc++.h>
6 #include <msp/datafile/objectloader.h>
7 #include <msp/time/timestamp.h>
8 #include "geometry.h"
9 #include "sensor.h"
10
11 namespace R2C2 {
12
13 class ArticleNumber;
14 class Block;
15 class Catalogue;
16 class Driver;
17 class Object;
18 class Route;
19 class Signal;
20 class Track;
21 class TrackChain;
22 class Train;
23 class Vehicle;
24 class Zone;
25
26 class Layout
27 {
28 public:
29         class Loader: public Msp::DataFile::ObjectLoader<Layout>
30         {
31         private:
32                 bool new_tracks;
33
34         public:
35                 Loader(Layout &);
36         private:
37                 void route();
38                 void route(const std::string &);
39                 void signal(ArticleNumber);
40                 void track(unsigned);
41                 void track(ArticleNumber);
42                 void train(unsigned, unsigned, const std::string &);
43                 void train(ArticleNumber, unsigned, const std::string &);
44                 void turnout(unsigned, unsigned);
45                 void zone();
46         };
47
48 private:
49         template<typename B>
50         class Storage
51         {
52         private:
53                 std::set<B *> base;
54                 mutable std::list<Msp::Variant> caches;
55
56         public:
57                 ~Storage();
58
59                 bool insert(B &);
60                 bool erase(B &);
61
62                 const std::set<B *> &get() const { return base; }
63
64                 template<typename T>
65                 const std::set<T *> &get() const;
66
67                 template<typename T>
68                 void del();
69         };
70
71 public:
72         typedef std::vector<Zone *> ZoneArray;
73
74         sigc::signal<void, Object &> signal_object_added;
75         sigc::signal<void, Object &> signal_object_removed;
76         sigc::signal<void, TrackChain &> signal_track_chain_added;
77         sigc::signal<void, TrackChain &> signal_track_chain_removed;
78         sigc::signal<void, Train &> signal_train_added;
79         sigc::signal<void, Train &> signal_train_removed;
80         sigc::signal<void, Block &, Train *> signal_block_reserved;
81         sigc::signal<void, Sensor &, Sensor::State> signal_sensor_state_changed;
82         sigc::signal<void, const std::string &> signal_emergency;
83
84 private:
85         Catalogue &catalogue;
86         Driver *driver;
87         std::string base;
88         Storage<Object> objects;
89         Storage<TrackChain> track_chains;
90         Storage<Sensor> sensors;
91         std::map<unsigned, Train *> trains;
92         Msp::Time::TimeStamp last_tick;
93         unsigned next_turnout_id;
94
95 public:
96         Layout(Catalogue &, Driver * = 0);
97         ~Layout();
98
99         Catalogue &get_catalogue() const { return catalogue; }
100         bool has_driver() const { return driver; }
101         Driver &get_driver() const;
102         const std::string &get_base() const { return base; }
103
104         void add(Object &);
105         void add(Track &);
106         void add(TrackChain &);
107         void add(Block &);
108         void add(Sensor &);
109
110         template<typename T>
111         const std::set<T *> &get_all() const;
112
113         template<typename T>
114         T *pick(const Ray &);
115
116         void remove(Object &);
117         void remove(Track &);
118         void remove(TrackChain &);
119         void remove(Sensor &);
120
121         unsigned allocate_turnout_id();
122
123         Block &get_block(unsigned) const;
124         void create_blocks();
125         void create_blocks(Track &);
126
127         Route &get_route(const std::string &) const;
128         void update_routes();
129
130         ZoneArray get_zones(const std::string &) const;
131         Zone &get_zone(const std::string &, unsigned) const;
132
133         void add_train(Train &);
134         Train &get_train(unsigned) const;
135         const std::map<unsigned, Train *> &get_trains() const { return trains; }
136         void remove_train(Train &);
137
138         void tick();
139         void emergency(const std::string &);
140
141         void save(const std::string &) const;
142         void save_dynamic(const std::string &) const;
143 private:
144         void sensor_state_changed(Sensor &, Sensor::State);
145 };
146
147 } // namespace R2C2
148
149 #endif
150