source/libr2c2/vehicle.h

   1 #ifndef LIBR2C2_VEHICLE_H_
   2 #define LIBR2C2_VEHICLE_H_
   3
   4 #include "geometry.h"
   5 #include "object.h"
   6 #include "trackoffsetiter.h"
   7 #include "vehicletype.h"
   8
   9 namespace R2C2 {
  10
  11 class Layout;
  12 class Train;
  13
  14 class attachment_error: public std::logic_error
  15 {
  16 public:
  17         attachment_error(const std::string &w): std::logic_error(w) { }
  18         virtual ~attachment_error() throw() { }
  19 };
  20
  21
  22 class Vehicle: public Object
  23 {
  24 public:
  25         enum PlaceMode
  26         {
  27                 CENTER,
  28                 FRONT_AXLE,
  29                 FRONT_BUFFER,
  30                 BACK_AXLE,
  31                 BACK_BUFFER
  32         };
  33
  34         struct Axle
  35         {
  36                 const VehicleType::Axle *type;
  37                 Angle angle;
  38
  39                 Axle(const VehicleType::Axle &);
  40         };
  41
  42         struct Bogie
  43         {
  44                 const VehicleType::Bogie *type;
  45                 Angle direction;
  46                 std::vector<Axle *> axles;
  47
  48                 Bogie(const VehicleType::Bogie &);
  49         };
  50
  51         struct Rod
  52         {
  53                 const VehicleType::Rod *type;
  54                 Vector position;
  55                 Angle angle;
  56
  57                 Rod(const VehicleType::Rod &);
  58         };
  59
  60 private:
  61         const VehicleType &type;
  62         Train *train;
  63         Vehicle *next;
  64         Vehicle *prev;
  65         TrackOffsetIter track;
  66         std::vector<Axle> axles;
  67         std::vector<Axle *> fixed_axles;
  68         std::vector<Bogie> bogies;
  69         std::vector<Rod> rods;
  70         unsigned front_sensor;
  71         unsigned back_sensor;
  72
  73 public:
  74         Vehicle(Layout &, const VehicleType &);
  75         ~Vehicle();
  76
  77         virtual Vehicle *clone(Layout * = 0) const;
  78         virtual const VehicleType &get_type() const { return type; }
  79
  80         void set_train(Train *);
  81         Train *get_train() const { return train; }
  82         void attach_back(Vehicle &);
  83         void attach_front(Vehicle &);
  84         void detach_back();
  85         void detach_front();
  86         Vehicle *get_next() const { return next; }
  87         Vehicle *get_previous() const { return prev; }
  88
  89         // TODO implement these - should call place() with suitable parameters
  90         virtual void set_position(const Vector &) { }
  91         virtual void set_rotation(const Angle &) { }
  92         virtual void set_tilt(const Angle &) { }
  93         void place(const TrackIter &, float, PlaceMode = CENTER);
  94         void unplace();
  95         void advance(float);
  96         const TrackOffsetIter &get_track_iter() const { return track; }
  97         Track *get_track() const { return track.track(); }
  98         unsigned get_entry() const { return track.entry(); }
  99         float get_offset() const { return track.offset(); }
 100         const Axle &get_axle(unsigned) const;
 101         const Axle &get_fixed_axle(unsigned) const;
 102         const Bogie &get_bogie(unsigned) const;
 103         const Axle &get_bogie_axle(unsigned, unsigned) const;
 104         const Rod &get_rod(unsigned) const;
 105 private:
 106         void update_position();
 107         void update_position_from(const Vehicle &);
 108         void propagate_position();
 109         void propagate_forward();
 110         void propagate_backward();
 111         void check_sensor(float, unsigned &);
 112         void turn_axles(float);
 113         void update_rods();
 114
 115         void adjust_for_distance(TrackOffsetIter &, TrackOffsetIter &, float, float = 0.5) const;
 116         OrientedPoint get_point(const Vector &, const Vector &, float = 0.5) const;
 117         OrientedPoint get_point(const TrackOffsetIter &, float, float = 0.5) const;
 118
 119 public:
 120         virtual unsigned get_n_link_slots() const;
 121         virtual Vehicle *get_link(unsigned) const;
 122         virtual int get_link_slot(const Object &) const;
 123 };
 124
 125 } // namespace R2C2
 126
 127 #endif