]> git.tdb.fi Git - r2c2.git/blob - source/libr2c2/trainrouter.h
Add a distance metric to turn the routing into an A* search
[r2c2.git] / source / libr2c2 / trainrouter.h
1 #ifndef LIBR2C2_TRAINROUTER_H_
2 #define LIBR2C2_TRAINROUTER_H_
3
4 #include <list>
5 #include <msp/datafile/objectloader.h>
6 #include "trainai.h"
7
8 namespace R2C2 {
9
10 class Block;
11 class Layout;
12 class Track;
13 class TrackChain;
14 class TrainRouteMetric;
15
16 class TrainRouter: public TrainAI
17 {
18 public:
19         class Loader: public Msp::DataFile::ObjectLoader<TrainRouter>
20         {
21         public:
22                 Loader(TrainRouter &);
23         private:
24                 void route(const std::string &);
25         };
26
27         sigc::signal<void, const Route *> signal_route_changed;
28         sigc::signal<void, const TrackChain *> signal_arrived;
29         sigc::signal<void, const TrackChain *> signal_waypoint_reached;
30
31 private:
32         struct Wait
33         {
34                 Block *block;
35                 Train *train;
36
37                 Wait();
38         };
39
40         typedef std::list<const Route *> RouteList;
41
42         int priority;
43         RouteList routes;
44         unsigned arriving;
45         const TrackChain *destination;
46         std::vector<const TrackChain *> waypoints;
47         std::vector<TrainRouteMetric *> metrics;
48         std::list<Wait> waits;
49         Msp::Time::TimeDelta delay;
50
51         bool update_pending;
52
53 public:
54         TrainRouter(Train &);
55         ~TrainRouter();
56
57         void set_priority(int);
58         int get_priority() const { return priority; }
59
60         bool set_route(const Route *);
61         void add_wait(Block &, Train *);
62         const Route *get_route() const;
63
64         void set_destination(const TrackChain &);
65         const TrackChain *get_destination() const { return destination; }
66         bool is_destination(Track &) const;
67         void add_waypoint(const TrackChain &);
68         unsigned get_n_waypoints() const { return waypoints.size(); }
69         bool is_waypoint(unsigned, Track &) const;
70         const TrainRouteMetric &get_metric(int = -1) const;
71         void set_departure_delay(const Msp::Time::TimeDelta &);
72         const Msp::Time::TimeDelta &get_departure_delay() const { return delay; }
73
74         virtual void message(const Message &);
75         virtual void tick(const Msp::Time::TimeDelta &);
76
77         void save(std::list<Msp::DataFile::Statement> &) const;
78
79 private:
80         void block_reserved(Block &, Train *);
81         void train_advanced(Block &);
82         const Route *get_route_for_block(const Block &) const;
83
84         void create_metrics();
85         Route *create_lead_route(Route *, const Route *);
86         bool advance_route(RouteList::iterator &, const Block &);
87         bool is_on_route(const Block &);
88
89         static void create_plans(Layout &);
90 };
91
92 } // namespace R2C2
93
94 #endif