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