source/libr2c2/train.h

   1 #ifndef LIBR2C2_TRAIN_H_
   2 #define LIBR2C2_TRAIN_H_
   3
   4 #include <sigc++/signal.h>
   5 #include <sigc++/trackable.h>
   6 #include <msp/datafile/objectloader.h>
   7 #include <msp/time/timestamp.h>
   8 #include "block.h"
   9 #include "blockallocator.h"
  10 #include "controller.h"
  11 #include "trainai.h"
  12
  13 namespace R2C2 {
  14
  15 class ArticleNumber;
  16 class SpeedQuantizer;
  17 class Vehicle;
  18 class VehicleType;
  19
  20 class Train: public sigc::trackable
  21 {
  22 public:
  23         class Loader: public Msp::DataFile::ObjectLoader<Train>
  24         {
  25         private:
  26                 Block *prev_block;
  27                 bool blocks_valid;
  28
  29         public:
  30                 Loader(Train &);
  31         private:
  32                 virtual void finish();
  33                 void blocks();
  34                 void name(const std::string &);
  35                 void quantized_speed();
  36                 void router();
  37                 void timetable();
  38                 void vehicle(ArticleNumber);
  39         };
  40
  41         sigc::signal<void, const std::string &> signal_name_changed;
  42         sigc::signal<void, const std::string &, float> signal_control_changed;
  43         sigc::signal<void, unsigned, bool> signal_function_changed;
  44         sigc::signal<void, TrainAI &, const TrainAI::Message &> signal_ai_event;
  45         sigc::signal<void, Block &> signal_advanced;
  46
  47 private:
  48         Layout &layout;
  49         const VehicleType &loco_type;
  50         unsigned address;
  51         std::string protocol;
  52         std::string name;
  53         const Train *preceding_train;
  54         std::vector<Vehicle *> vehicles;
  55         BlockAllocator allocator;
  56         bool advancing;
  57         Controller *controller;
  58         std::list<TrainAI *> ais;
  59         bool active;
  60         unsigned current_speed_step;
  61         bool speed_changing;
  62         bool reverse;
  63         Msp::Time::TimeStamp stop_timeout;
  64         unsigned functions;
  65
  66         BlockIter last_entry_block;
  67         Msp::Time::TimeStamp last_entry_time;
  68         bool pure_speed;
  69         SpeedQuantizer *speed_quantizer;
  70         bool accurate_position;
  71         float overshoot_dist;
  72
  73 public:
  74         Train(Layout &, const VehicleType &, unsigned, const std::string &);
  75         ~Train();
  76
  77         Layout &get_layout() const { return layout; }
  78         const VehicleType &get_locomotive_type() const { return loco_type; }
  79         unsigned get_address() const { return address; }
  80         const std::string &get_protocol() const { return protocol; }
  81         void set_name(const std::string &);
  82         const std::string &get_name() const { return name; }
  83         const Train *get_preceding_train() const { return preceding_train; }
  84         Controller &get_controller() const { return *controller; }
  85
  86         void add_vehicle(const VehicleType &);
  87         void remove_vehicle(unsigned);
  88         unsigned get_n_vehicles() const;
  89         Vehicle &get_vehicle(unsigned);
  90         const Vehicle &get_vehicle(unsigned) const;
  91
  92         void set_control(const std::string &, float);
  93         void set_active(bool);
  94         void set_function(unsigned, bool);
  95         float get_control(const std::string &) const;
  96         float get_speed() const;
  97         float get_quantized_speed() const;
  98         unsigned get_speed_step() const { return current_speed_step; }
  99         bool is_active() const { return active; }
 100         bool get_function(unsigned) const;
 101         unsigned get_functions() const { return functions; }
 102
 103         void add_ai(TrainAI &);
 104         void remove_ai(TrainAI &);
 105         void ai_message(const TrainAI::Message &);
 106
 107         template<typename T>
 108         T *get_ai_of_type() const
 109         {
 110                 for(std::list<TrainAI *>::const_iterator i=ais.begin(); i!=ais.end(); ++i)
 111                         if(T *ai = dynamic_cast<T *>(*i))
 112                                 return ai;
 113                 return 0;
 114         }
 115
 116         void place(const BlockIter &);
 117         void unplace();
 118         bool is_placed() const { return !allocator.empty(); }
 119         void stop_at(Block *);
 120         bool free_block(Block &);
 121         void free_noncritical_blocks();
 122         void reserve_more();
 123         const BlockAllocator &get_block_allocator() const { return allocator; }
 124         float get_reserved_distance() const;
 125
 126         void tick(const Msp::Time::TimeStamp &, const Msp::Time::TimeDelta &);
 127
 128         void save(std::list<Msp::DataFile::Statement> &) const;
 129 private:
 130         void control_changed(const Controller::Control &);
 131         void loco_speed_event(unsigned, unsigned, bool);
 132         void loco_func_event(unsigned, unsigned, bool);
 133         void block_state_changed(Block &, Block::State);
 134         void halt_event(bool);
 135         void block_reserved(const Block &, const Train *);
 136         float get_reserved_distance_until(const Block *) const;
 137 };
 138
 139 } // namespace R2C2
 140
 141 #endif