Full vehicle unification

[r2c2.git] / source / libmarklin / train.h

diff --git a/source/libmarklin/train.h b/source/libmarklin/train.h
index 24c9db9fc04acb238cf5e265ccee9e02d5f6b2b2..e8785f5e5994deb4ae275e670a0187676dd03534 100644 (file)
--- a/source/libmarklin/train.h
+++ b/source/libmarklin/train.h
@@ -1,3 +1,10 @@
+/* $Id$
+
+This file is part of the MSP Märklin suite
+Copyright © 2006-2010  Mikkosoft Productions, Mikko Rasa
+Distributed under the GPL
+*/
+
 #ifndef LIBMARKLIN_TRAIN_H_
 #define LIBMARKLIN_TRAIN_H_
 
@@ -8,48 +15,147 @@
 
 namespace Marklin {
 
-class Locomotive;
-class Sensor;
-class TrafficManager;
+class ControlModel;
+class Route;
+class Timetable;
+class Vehicle;
+class VehicleType;
 
 class Train: public sigc::trackable
 {
+public:
+       class Loader: public Msp::DataFile::BasicLoader<Train>
+       {
+       private:
+               Block *prev_block;
+               bool blocks_valid;
+
+       public:
+               Loader(Train &);
+       private:
+               void block(unsigned);
+               void block_hint(unsigned);
+               void name(const std::string &);
+               void real_speed(unsigned, float, float);
+               void route(const std::string &);
+               void timetable();
+               void vehicle(unsigned);
+       };
+
+       sigc::signal<void, const std::string &> signal_name_changed;
+       sigc::signal<void, const std::string &, float> signal_control_changed;
+       sigc::signal<void, unsigned, bool> signal_function_changed;
+       sigc::signal<void, const Route *> signal_route_changed;
+       sigc::signal<void> signal_arrived;
+       sigc::signal<void, const std::string &> signal_status_changed;
+
 private:
        struct BlockRef
        {
                Block *block;
                unsigned entry;
 
-               BlockRef(Block *s, unsigned e): block(s), entry(e) { }
+               BlockRef(Block *, unsigned);
+               BlockRef next() const;
        };
 
-       TrafficManager &trfc_mgr;
+       struct RealSpeed
+       {
+               float speed;
+               float weight;
+
+               RealSpeed();
+               void add(float, float);
+       };
+
+       Layout &layout;
+       const VehicleType &loco_type;
+       unsigned address;
        std::string name;
-       Locomotive &loco;
+       int priority;
+       std::vector<Vehicle *> vehicles;
        std::list<BlockRef> cur_blocks;
        std::list<BlockRef> rsv_blocks;
-       unsigned target_speed;
-       Msp::Time::TimeStamp try_reserve;
+       Block *pending_block;
+       ControlModel *control;
+       Timetable *timetable;
+       bool active;
+       unsigned current_speed;
+       bool speed_changing;
+       bool reverse;
+       Msp::Time::TimeStamp stop_timeout;
+       unsigned functions;
+       const Route *route;
+       const Route *next_route;
+       bool end_of_route;
        std::string status;
 
-public:
-       sigc::signal<void, const std::string &> signal_name_changed;
-       sigc::signal<void, const std::string &> signal_status_changed;
+       Msp::Time::TimeStamp last_entry_time;
+       float travel_dist;
+       bool pure_speed;
+       std::vector<RealSpeed> real_speed;
+       bool accurate_position;
+       float overshoot_dist;
 
-       Train(TrafficManager &, Locomotive &);
+public:
+       Train(Layout &, const VehicleType &, unsigned);
+       ~Train();
 
+       Layout &get_layout() const { return layout; }
+       const VehicleType &get_locomotive_type() const { return loco_type; }
+       unsigned get_address() const { return address; }
        void set_name(const std::string &);
-       void set_speed(unsigned);
        const std::string &get_name() const { return name; }
-       Locomotive &get_locomotive() const { return loco; }
+       void set_priority(int);
+       int get_priority() const { return priority; }
+       ControlModel &get_control() const { return *control; }
+
+       void add_vehicle(const VehicleType &);
+       void remove_vehicle(unsigned);
+       unsigned get_n_vehicles() const;
+       Vehicle &get_vehicle(unsigned);
+       const Vehicle &get_vehicle(unsigned) const;
+
+       void set_control(const std::string &, float);
+       void set_active(bool);
+       void set_function(unsigned, bool);
+       float get_control(const std::string &) const;
+       bool is_active() const { return active; }
+       bool get_function(unsigned) const;
+       unsigned get_functions() const { return functions; }
+
+       void set_timetable(Timetable *);
+       Timetable *get_timetable() { return timetable; }
+
+       void set_route(const Route *);
+       void go_to(const Track &);
+       const Route *get_route() const { return route; }
+       void place(Block &, unsigned);
+       bool is_placed() const { return !cur_blocks.empty(); }
+       bool free_block(Block &);
+       int get_entry_to_block(Block &) const;
+       float get_reserved_distance() const;
+
        const std::string &get_status() const { return status; }
-       void place(Block *, unsigned);
-       bool free_block(Block *);
-       void tick(const Msp::Time::TimeStamp &);
+
+       void tick(const Msp::Time::TimeStamp &, const Msp::Time::TimeDelta &);
+
+       void save(std::list<Msp::DataFile::Statement> &) const;
 private:
-       void sensor_event(bool, Sensor *);
-       bool reserve_more();
+       void loco_speed_event(unsigned, unsigned, bool);
+       void loco_func_event(unsigned, unsigned, bool);
+       void sensor_event(unsigned, bool);
+       void turnout_event(unsigned, bool);
+       void halt_event(bool);
+       void block_reserved(const Block &, const Train *);
+       unsigned reserve_more();
+       float get_real_speed(unsigned) const;
+       unsigned find_speed(float) const;
+       float get_travel_speed() const;
        void set_status(const std::string &);
+       void release_blocks(std::list<BlockRef> &);
+       void release_blocks(std::list<BlockRef> &, std::list<BlockRef>::iterator, std::list<BlockRef>::iterator);
+       void reverse_blocks(std::list<BlockRef> &) const;
 };
 
 } // namespace Marklin