X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fengineer%2Fengineer.h;h=f6e23bc28e18c1fc27d6b48f1e44c528563176ce;hb=447d621e93e71b710fed0920b473350122426b97;hp=1575c0c31e381ad1d18a0a9727b2727746fff3db;hpb=aead7f2da831199fbb8f6950b067656b2965c2fa;p=r2c2.git

diff --git a/source/engineer/engineer.h b/source/engineer/engineer.h
index 1575c0c..f6e23bc 100644
--- a/source/engineer/engineer.h
+++ b/source/engineer/engineer.h
@@ -8,6 +8,7 @@
 #include <msp/gltk/root.h>
 #include "libmarklin/catalogue.h"
 #include "libmarklin/control.h"
+#include "libmarklin/locotype.h"
 #include "libmarklin/trafficmanager.h"
 #include "libmarklin/train.h"
 #include "3d/layout.h"
@@ -18,17 +19,7 @@ class TrainProperties;
 
 class Engineer: public Msp::Application
 {
-public:
-	Engineer(int argc, char **argv);
-	~Engineer();
-
-	Marklin::Control &get_control()    { return control; }
-	Marklin::Train   *add_train(unsigned);
-	int      main();
-	void     quit() { exit(0); }
 private:
-	typedef std::list<TrainPanel *> TrainPanelSeq;
-
 	Msp::Graphics::Display   *dpy;
 	Msp::Graphics::Window    *wnd;
 	Msp::Graphics::GLContext *glc;
@@ -44,7 +35,7 @@ private:
 	float              cam_rot;
 	Msp::GLtk::Resources ui_res;
 	MainPanel          *main_panel;
-	TrainPanelSeq      train_panels;
+	std::list<TrainPanel *> train_panels;
 	bool               no_lighting;
 	Marklin::TrafficManager *trfc_mgr;
 	Marklin::Train     *placing_train;
@@ -54,13 +45,26 @@ private:
 	TrainProperties    *train_prop;
 	bool               train_prop_stale;
 
+public:
+	Engineer(int argc, char **argv);
+	~Engineer();
+
+	const Marklin::Catalogue &get_catalogue() const { return catalogue; }
+	Marklin::Control &get_control()    { return control; }
+	void             add_train();
+	Marklin::Train   *add_train(const Marklin::LocoType &, unsigned);
+	void     place_train(Marklin::Train &);
+	int      main();
+	void     quit() { exit(0); }
+private:
 	void tick();
 	void key_press(unsigned, unsigned, wchar_t);
 	void button_press(int, int, unsigned, unsigned);
 	void button_release(int, int, unsigned, unsigned);
 	void pointer_motion(int, int);
 	void view_all();
-	void set_block_color(const Marklin::Block &, const Marklin::Color &);
+	void set_block_color(const Marklin::Block &, const Msp::GL::Color &);
+	void reset_block_color(const Marklin::Block &);
 	void sensor_event(bool, Marklin::Sensor *);
 	void block_reserved(const Marklin::Block &, const Marklin::Train *);
 	void project_3d();