X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fdesigner%2Fdesigner.cpp;h=6ae9c0726c45701a5915adc9f852d9fdc1b75e66;hb=7ff47d1f59711542f4943aef02655d1bee51a8f6;hp=a59b3a12dd127839616a13190ceb51b7ee345b4f;hpb=b6af6301c76d3664d969a72d8d889020093ac848;p=r2c2.git

diff --git a/source/designer/designer.cpp b/source/designer/designer.cpp
index a59b3a1..6ae9c07 100644
--- a/source/designer/designer.cpp
+++ b/source/designer/designer.cpp
@@ -148,9 +148,9 @@ Designer::Designer(int argc, char **argv):
 	cat_view->view_all(true);
 	main_view->view_all();
 
-	const set<Track *> &tracks = layout->get_all<Track>();
-	for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
-		update_track_icon(layout_3d->get_3d(**i));
+	const set<Object *> &objects = layout->get_all<Object>();
+	for(set<Object *>::const_iterator i=objects.begin(); i!=objects.end(); ++i)
+		update_object_icon(**i);
 }
 
 Designer::~Designer()
@@ -406,7 +406,7 @@ void Designer::key_press(unsigned key)
 				if(*j!=*i)
 					(*i)->snap_to(**j, true);
 
-			update_track_icon(layout_3d->get_3d(**i));
+			update_object_icon(**i);
 		}
 	}
 	else if(key==Msp::Input::KEY_F)
@@ -509,25 +509,39 @@ Object *Designer::pick_object(const Vector &pointer)
 	return view.get_layout().get_layout().pick<Object>(Ray(cpos, Vector(cray)));
 }
 
-void Designer::update_track_icon(Track3D &track)
+void Designer::update_object_icon(Object &obj)
 {
-	overlay->clear(track);
+	Object3D &obj3d = layout_3d->get_3d(obj);
+	overlay->clear(obj3d);
 
-	if(track.get_track().get_flex())
-		overlay->add_graphic(track, "flex");
+	if(Track *track = dynamic_cast<Track *>(&obj))
+	{
+		if(track->get_flex())
+			overlay->add_graphic(obj3d, "flex");
 
-	if(unsigned saddr = track.get_track().get_sensor_address())
+		if(unsigned saddr = track->get_sensor_address())
+		{
+			overlay->add_graphic(obj3d, "trackcircuit");
+			overlay->set_label(obj3d, lexical_cast<string>(saddr));
+		}
+		else if(unsigned taddr = track->get_turnout_address())
+		{
+			if(taddr<0x800)
+			{
+				overlay->add_graphic(obj3d, "turnout");
+				overlay->set_label(obj3d, lexical_cast<string>(taddr));
+			}
+		}
+	}
+	else if(Signal *signal = dynamic_cast<Signal *>(&obj))
 	{
-		overlay->add_graphic(track, "sensor");
-		overlay->set_label(track, lexical_cast<string>(saddr));
+		overlay->add_graphic(obj3d, "signal");
+		overlay->set_label(obj3d, lexical_cast<string>(signal->get_address()));
 	}
-	else if(unsigned taddr = track.get_track().get_turnout_address())
+	else if(BeamGate *gate = dynamic_cast<BeamGate *>(&obj))
 	{
-		if(taddr<0x800)
-		{
-			overlay->add_graphic(track, "turnout");
-			overlay->set_label(track, lexical_cast<string>(taddr));
-		}
+		overlay->add_graphic(obj3d, "beamgate");
+		overlay->set_label(obj3d, lexical_cast<string>(gate->get_address()));
 	}
 }
 
@@ -538,9 +552,9 @@ void Designer::tool_status(const string &status)
 
 void Designer::object_properties_response(int)
 {
-	const set<Track *> &tracks = selection.get_objects<Track>();
-	for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
-		update_track_icon(layout_3d->get_3d(**i));
+	const set<Object *> &objects = selection.get_objects();
+	for(set<Object *>::const_iterator i=objects.begin(); i!=objects.end(); ++i)
+		update_object_icon(**i);
 }
 
 void Designer::route_name_accept(const string &text)