X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;ds=sidebyside;f=source%2Flibmarklin%2Ftrackpart.cpp;h=37c5f97a90b0fc81cae5eb150bb2cbd8a654e8d5;hb=9b05c573a38639827697fe393d55b7c76f5bde45;hp=f2b61be1f37f2050105d94f020d4347aeb1dd2dc;hpb=6dc18b0e518407bd2a86602bae1e9bbae05da7c8;p=r2c2.git

diff --git a/source/libmarklin/trackpart.cpp b/source/libmarklin/trackpart.cpp
index f2b61be..37c5f97 100644
--- a/source/libmarklin/trackpart.cpp
+++ b/source/libmarklin/trackpart.cpp
@@ -1,7 +1,7 @@
 /* $Id$
 
 This file is part of the MSP Märklin suite
-Copyright © 2006-2009 Mikkosoft Productions, Mikko Rasa
+Copyright © 2006-2010  Mikkosoft Productions, Mikko Rasa
 Distributed under the GPL
 */
 
@@ -9,6 +9,7 @@ Distributed under the GPL
 #include "trackpart.h"
 
 using namespace std;
+using namespace Msp;
 
 namespace Marklin {
 
@@ -18,26 +19,23 @@ TrackPart::TrackPart():
 	radius(0),
 	path(0),
 	dead_end(false)
-{ }
-
-void TrackPart::collect_endpoints(vector<Endpoint> &eps) const
 {
-	eps.push_back(Endpoint(pos.x, pos.y, dir+M_PI, 1<<path));
+	links[0] = 0;
+	links[1] = 0;
+}
 
-	if(dead_end)
-		;
-	else if(radius)
-	{
-		float a = ((radius<0) ? -length : length);
-		Point p = get_point(length*abs(radius));
-		eps.push_back(Endpoint(p.x, p.y, dir+a, 1<<path));
-	}
+float TrackPart::get_length() const
+{
+	if(radius)
+		return abs(radius)*length;
 	else
-		eps.push_back(Endpoint(pos.x+cos(dir)*length, pos.y+sin(dir)*length, dir, 1<<path));
+		return length;
 }
 
-Point TrackPart::get_point(float d) const
+TrackPoint TrackPart::get_point(float d) const
 {
+	TrackPoint result;
+
 	if(radius)
 	{
 		float a = d/radius;
@@ -45,10 +43,53 @@ Point TrackPart::get_point(float d) const
 		float s = sin(a);
 		float rx = radius*sin(dir);
 		float ry = -radius*cos(dir);
-		return Point(pos.x+c*rx-s*ry-rx, pos.y+c*ry+s*rx-ry);
+		result.pos = Point(pos.x+c*rx-s*ry-rx, pos.y+c*ry+s*rx-ry);
+		result.dir = dir+a;
 	}
 	else
-		return Point(pos.x+cos(dir)*d, pos.y+sin(dir)*d);
+	{
+		result.pos = Point(pos.x+cos(dir)*d, pos.y+sin(dir)*d);
+		result.dir = dir;
+	}
+
+	return result;
+}
+
+void TrackPart::check_link(TrackPart &other)
+{
+	unsigned n_eps = (dead_end ? 1 : 2);
+	unsigned n_other_eps = (other.is_dead_end() ? 1 : 2);
+	for(unsigned i=0; i<n_eps; ++i)
+	{
+		TrackPoint p1 = get_point(i ? get_length() : 0);
+		for(unsigned j=0; j<n_other_eps; ++j)
+		{
+			TrackPoint p2 = other.get_point(j ? other.get_length() : 0);
+
+			float dx = p2.pos.x-p1.pos.x;
+			float dy = p2.pos.y-p1.pos.y;
+
+			float da = p2.dir-p1.dir+M_PI*((i+j+1)%2);
+			while(da>M_PI)
+				da -= M_PI*2;
+			while(da<-M_PI)
+				da += M_PI*2;
+
+			if(dx*dx+dy*dy<1e-6 && da>=-0.01 && da<=0.01)
+			{
+				links[i] = &other;
+				other.links[j] = this;
+				return;
+			}
+		}
+	}
+}
+
+TrackPart *TrackPart::get_link(unsigned i) const
+{
+	if(i>=2)
+		throw InvalidParameterValue("Index out of range");
+	return links[i];
 }