X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fdesigner%2Fextendtool.cpp;h=c31c09f951149429182609d3ecaa7d9dd09aef27;hb=81ee4fbd16b472892b67b35ea85956423a2247ed;hp=9b6ec9ab606474bd5c1facfdd9d2b623fb86d80c;hpb=50462807446fd62dbacfba90a7afffc1ff70d3b1;p=r2c2.git

diff --git a/source/designer/extendtool.cpp b/source/designer/extendtool.cpp
index 9b6ec9a..c31c09f 100644
--- a/source/designer/extendtool.cpp
+++ b/source/designer/extendtool.cpp
@@ -26,19 +26,18 @@ ExtendTool::ExtendTool(Designer &d, Input::Keyboard &k, Input::Mouse &m, const s
 		set_done(false);
 	}
 
-	const Catalogue::ObjectMap &object_types = designer.get_catalogue().get_all();
-	for(Catalogue::ObjectMap::const_iterator i=object_types.begin(); i!=object_types.end(); ++i)
-		if(const TrackType *tt = dynamic_cast<const TrackType *>(i->second))
-		{
-			const vector<TrackPart> &parts = tt->get_parts();
-			if(parts.size()!=1)
-				continue;
-			if(parts.front().is_curved() || parts.front().is_dead_end())
-				continue;
+	list<TrackType *> track_types = designer.get_catalogue().get_list<TrackType>();
+	for(list<TrackType *>::iterator i=track_types.begin(); i!=track_types.end(); ++i)
+	{
+		const vector<TrackPart> &parts = (*i)->get_parts();
+		if(parts.size()!=1)
+			continue;
+		if(parts.front().is_curved() || parts.front().is_dead_end())
+			continue;
 
-			types_by_length[parts.front().get_length()] = tt;
-			max_preference = max(max_preference, tt->get_autofit_preference());
-		}
+		types_by_length[parts.front().get_length()] = *i;
+		max_preference = max(max_preference, (*i)->get_autofit_preference());
+	}
 }
 
 ExtendTool::~ExtendTool()
@@ -62,7 +61,7 @@ void ExtendTool::connect()
 	for(vector<TrackIter>::const_iterator i=unlinked_endpoints.begin(); i!=unlinked_endpoints.end(); ++i)
 	{
 		start_sn = (*i)->get_snap_node(i->entry());
-		limit = (*i)->get_type().get_appearance().get_gauge()/10;
+		limit = (*i)->get_type().get_gauge()/10;
 		
 		for(vector<TrackIter>::const_iterator j=i; ++j!=unlinked_endpoints.end(); )
 		{