X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibr2c2%2Flayout.cpp;h=4679fd9e166674fec43a8ba01947b015bc8c294c;hb=b88d3630a0e3fdd763db018bbe5dbfe3180a95f2;hp=fefa3e9b3fb4045e2d6ecf84280c0319fa2431ed;hpb=3e3675f8d86b06cdea7df197dfe8866ffb285aec;p=r2c2.git

diff --git a/source/libr2c2/layout.cpp b/source/libr2c2/layout.cpp
index fefa3e9..4679fd9 100644
--- a/source/libr2c2/layout.cpp
+++ b/source/libr2c2/layout.cpp
@@ -72,6 +72,11 @@ void Layout::add_track(Track &t)
 	}
 }
 
+Track *Layout::pick_track(const Vector &start, const Vector &ray)
+{
+	return pick(tracks, start, ray);
+}
+
 void Layout::remove_track(Track &t)
 {
 	if(tracks.erase(&t))
@@ -81,40 +86,55 @@ void Layout::remove_track(Track &t)
 	}
 }
 
+unsigned Layout::allocate_turnout_id()
+{
+	set<unsigned> used_ids;
+	for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
+		if((*i)->get_turnout_id())
+			used_ids.insert((*i)->get_turnout_id());
+
+	unsigned result = next_turnout_id;
+	while(used_ids.count(result))
+		++result;
+	next_turnout_id = result+1;
+
+	return result;
+}
+
 void Layout::add_signal(Signal &s)
 {
 	if(signals.insert(&s).second)
 		signal_signal_added.emit(s);
 }
 
+Signal *Layout::pick_signal(const Vector &start, const Vector &ray)
+{
+	return pick(signals, start, ray);
+}
+
 void Layout::remove_signal(Signal &s)
 {
 	if(signals.erase(&s))
 		signal_signal_removed.emit(s);
 }
 
-Track *Layout::pick_track(const Vector &start, const Vector &ray)
+Object *Layout::pick_object(const Vector &start, const Vector &ray)
 {
-	for(set<Track *>::iterator i=tracks.begin(); i!=tracks.end(); ++i)
-		if((*i)->collide_ray(start, ray))
-			return *i;
-
+	if(Object *obj = pick_track(start, ray))
+		return obj;
+	else if((obj = pick_signal(start, ray)))
+		return obj;
 	return 0;
 }
 
-unsigned Layout::allocate_turnout_id()
+template<typename T>
+T *Layout::pick(const set<T *> &objs, const Vector &start, const Vector &ray)
 {
-	set<unsigned> used_ids;
-	for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
-		if((*i)->get_turnout_id())
-			used_ids.insert((*i)->get_turnout_id());
-
-	unsigned result = next_turnout_id;
-	while(used_ids.count(result))
-		++result;
-	next_turnout_id = result+1;
+	for(typename set<T *>::const_iterator i=objs.begin(); i!=objs.end(); ++i)
+		if((*i)->collide_ray(start, ray))
+			return *i;
 
-	return result;
+	return 0;
 }
 
 void Layout::add_block(Block &b)