]> git.tdb.fi Git - r2c2.git/blob - source/libmarklin/trackpart.cpp
Add framework for generating simple meshes for vehicles
[r2c2.git] / source / libmarklin / trackpart.cpp
1 /* $Id$
2
3 This file is part of the MSP Märklin suite
4 Copyright © 2006-2010  Mikkosoft Productions, Mikko Rasa
5 Distributed under the GPL
6 */
7
8 #include <cmath>
9 #include "trackpart.h"
10
11 using namespace std;
12 using namespace Msp;
13
14 namespace Marklin {
15
16 TrackPart::TrackPart():
17         dir(0),
18         length(0),
19         radius(0),
20         path(0),
21         dead_end(false)
22 {
23         links[0] = 0;
24         links[1] = 0;
25 }
26
27 float TrackPart::get_length() const
28 {
29         if(radius)
30                 return abs(radius)*length;
31         else
32                 return length;
33 }
34
35 TrackPoint TrackPart::get_point(float d) const
36 {
37         TrackPoint result;
38
39         if(radius)
40         {
41                 float a = d/radius;
42                 float c = cos(a);
43                 float s = sin(a);
44                 float rx = radius*sin(dir);
45                 float ry = -radius*cos(dir);
46                 result.pos = Point(pos.x+c*rx-s*ry-rx, pos.y+c*ry+s*rx-ry);
47                 result.dir = dir+a;
48         }
49         else
50         {
51                 result.pos = Point(pos.x+cos(dir)*d, pos.y+sin(dir)*d);
52                 result.dir = dir;
53         }
54
55         return result;
56 }
57
58 void TrackPart::check_link(TrackPart &other)
59 {
60         unsigned n_eps = (dead_end ? 1 : 2);
61         unsigned n_other_eps = (other.is_dead_end() ? 1 : 2);
62         for(unsigned i=0; i<n_eps; ++i)
63         {
64                 TrackPoint p1 = get_point(i ? get_length() : 0);
65                 for(unsigned j=0; j<n_other_eps; ++j)
66                 {
67                         TrackPoint p2 = other.get_point(j ? other.get_length() : 0);
68
69                         float dx = p2.pos.x-p1.pos.x;
70                         float dy = p2.pos.y-p1.pos.y;
71
72                         float da = p2.dir-p1.dir+M_PI*((i+j+1)%2);
73                         while(da>M_PI)
74                                 da -= M_PI*2;
75                         while(da<-M_PI)
76                                 da += M_PI*2;
77
78                         if(dx*dx+dy*dy<1e-6 && da>=-0.01 && da<=0.01)
79                         {
80                                 links[i] = &other;
81                                 other.links[j] = this;
82                                 return;
83                         }
84                 }
85         }
86 }
87
88 TrackPart *TrackPart::get_link(unsigned i) const
89 {
90         if(i>=2)
91                 throw InvalidParameterValue("Index out of range");
92         return links[i];
93 }
94
95
96 TrackPart::Loader::Loader(TrackPart &p):
97         Msp::DataFile::BasicLoader<TrackPart>(p)
98 {
99         add("start",    &Loader::start);
100         add("length",   &TrackPart::length);
101         add("radius",   &TrackPart::radius);
102         add("path",     &TrackPart::path);
103         add("dead_end", &TrackPart::dead_end);
104 }
105
106 void TrackPart::Loader::finish()
107 {
108         if(obj.radius)
109         {
110                 obj.length *= M_PI/180;
111                 obj.radius /= 1000;
112         }
113         else
114                 obj.length /= 1000;
115
116         obj.pos.x /= 1000;
117         obj.pos.y /= 1000;
118         obj.dir *= M_PI/180;
119 }
120
121 void TrackPart::Loader::start(float x, float y, float d)
122 {
123         obj.pos = Point(x, y);
124         obj.dir = d;
125 }
126
127 } // namespace Marklin