X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibmarklin%2Ftrain.h;h=60a0c0824ade3030e52497ee0ec6a19b4f79f679;hb=651698847d5293cfb15b6fb23a394701388c0151;hp=21f1382c8c7ba5704797cfae4cf9b248dd1afd62;hpb=52cbe8d99669f843f8f75c51128e2748584dd03a;p=r2c2.git

diff --git a/source/libmarklin/train.h b/source/libmarklin/train.h
index 21f1382..60a0c08 100644
--- a/source/libmarklin/train.h
+++ b/source/libmarklin/train.h
@@ -1,18 +1,49 @@
+/* $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_
 
 #include <sigc++/signal.h>
 #include <sigc++/trackable.h>
+#include <msp/time/timestamp.h>
 #include "block.h"
 
 namespace Marklin {
 
-class Locomotive;
-class Sensor;
-class TrafficManager;
+class LocoType;
+class Route;
 
 class Train: public sigc::trackable
 {
+public:
+	class Loader: public Msp::DataFile::BasicLoader<Train>
+	{
+	private:
+		Block *prev_block;
+
+	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 &);
+	};
+
+	sigc::signal<void, const std::string &> signal_name_changed;
+	sigc::signal<void, unsigned> signal_target_speed_changed;
+	sigc::signal<void, unsigned> signal_speed_changed;
+	sigc::signal<void, bool> signal_reverse_changed;
+	sigc::signal<void, unsigned, bool> signal_function_changed;
+	sigc::signal<void, const Route *> signal_route_changed;
+	sigc::signal<void, const std::string &> signal_status_changed;
+
 private:
 	struct BlockRef
 	{
@@ -22,29 +53,88 @@ private:
 		BlockRef(Block *s, unsigned e): block(s), entry(e) { }
 	};
 
-	TrafficManager &trfc_mgr;
+	struct RealSpeed
+	{
+		float speed;
+		float weight;
+
+		RealSpeed();
+		void add(float, float);
+	};
+
+	Layout &layout;
+	const LocoType &loco_type;
+	unsigned address;
 	std::string name;
-	Locomotive &loco;
 	std::list<BlockRef> cur_blocks;
 	std::list<BlockRef> rsv_blocks;
+	Block *pending_block;
 	unsigned target_speed;
+	unsigned current_speed;
+	bool reverse;
+	Msp::Time::TimeStamp stop_timeout;
+	unsigned functions;
+	const Route *route;
+	std::string status;
+
+	Msp::Time::TimeStamp last_entry_time;
+	float travel_dist;
+	unsigned travel_speed;
+	bool pure_speed;
+	std::vector<RealSpeed> real_speed;
+
+	Track *cur_track;
+	unsigned cur_track_ep;
+	float offset;
+	Point pos;
 
 public:
-	sigc::signal<void, const std::string &> signal_name_changed;
+	Train(Layout &, const LocoType &, unsigned);
+	~Train();
 
-	Train(TrafficManager &, Locomotive &);
+	const LocoType &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 place(Block *, unsigned);
-	bool free_block(Block *);
-	void tick();
+
+	void set_speed(unsigned);
+	void set_reverse(bool);
+	void set_function(unsigned, bool);
+	unsigned get_target_speed() const { return target_speed; }
+	unsigned get_speed() const { return current_speed; }
+	bool get_reverse() const { return reverse; }
+	bool get_function(unsigned) const;
+	unsigned get_functions() const { return functions; }
+
+	void set_route(const Route *);
+	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;
+
+	const std::string &get_status() const { return status; }
+	const Point &get_position() const { return pos; }
+
+	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 block_reserved(const Block &, const Train *);
+	unsigned reserve_more();
+	void update_speed();
+	float get_real_speed(unsigned) const;
+	unsigned find_speed(float) const;
+	void set_status(const std::string &);
+	void set_position(const Block::Endpoint &);
+	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;
 };
-typedef std::list<Train *> TrainSeq;
 
 } // namespace Marklin