3 This file is part of the MSP Märklin suite
4 Copyright © 2006-2010 Mikkosoft Productions, Mikko Rasa
5 Distributed under the GPL
12 #include "tracktype.h"
19 Track::Track(Layout &l, const TrackType &t):
25 turnout_id(type.is_turnout() ? layout.allocate_turnout_id() : 0),
27 links(type.get_endpoints().size()),
30 layout.add_track(*this);
32 if(layout.has_driver())
33 layout.get_driver().signal_turnout.connect(sigc::mem_fun(this, &Track::turnout_event));
35 for(unsigned paths = type.get_paths(); !(paths&1); ++active_path, paths>>=1) ;
41 layout.remove_track(*this);
44 void Track::set_position(const Point &p)
49 void Track::set_rotation(float r)
58 void Track::set_slope(float s)
66 void Track::set_flex(bool f)
71 void Track::check_slope()
76 if(links[0] && links[1])
78 Point epp0 = links[0]->get_endpoint_position(links[0]->get_endpoint_by_link(*this));
79 Point epp1 = links[1]->get_endpoint_position(links[1]->get_endpoint_by_link(*this));
88 Point epp = links[0]->get_endpoint_position(links[0]->get_endpoint_by_link(*this));
93 Point epp = links[1]->get_endpoint_position(links[1]->get_endpoint_by_link(*this));
99 void Track::set_turnout_id(unsigned i)
101 if(!type.is_turnout())
102 throw InvalidState("Not a turnout");
105 layout.create_blocks(*this);
106 layout.update_routes();
107 if(layout.has_driver() && turnout_id)
109 layout.get_driver().add_turnout(turnout_id);
110 if(type.is_double_address())
111 layout.get_driver().add_turnout(turnout_id+1);
115 void Track::set_sensor_id(unsigned i)
117 if(type.is_turnout())
118 throw InvalidState("Can't set sensor on a turnout");
121 layout.create_blocks(*this);
122 if(layout.has_driver() && sensor_id)
123 layout.get_driver().add_sensor(sensor_id);
126 void Track::set_active_path(unsigned p)
129 throw InvalidState("Not a turnout");
130 if(!(type.get_paths()&(1<<p)))
131 throw InvalidParameterValue("Invalid path");
133 layout.get_driver().set_turnout(turnout_id, p&1);
134 if(type.is_double_address())
135 layout.get_driver().set_turnout(turnout_id+1, p&2);
136 else if(type.get_n_paths()>2)
137 active_path = (active_path&1) | (p&2);
140 int Track::get_endpoint_by_link(const Track &other) const
142 for(unsigned i=0; i<links.size(); ++i)
149 Point Track::get_endpoint_position(unsigned epi) const
151 const vector<Endpoint> &eps = type.get_endpoints();
153 throw InvalidParameterValue("Endpoint index out of range");
155 const Endpoint &ep = eps[epi];
160 Point p(pos.x+c*ep.pos.x-s*ep.pos.y, pos.y+s*ep.pos.x+c*ep.pos.y, pos.z);
161 if(eps.size()==2 && epi==1)
166 float Track::get_endpoint_direction(unsigned epi) const
168 const vector<Endpoint> &eps = type.get_endpoints();
170 throw InvalidParameterValue("Endpoint index out of range");
172 const Endpoint &ep = eps[epi];
177 bool Track::snap_to(Track &other, bool link)
179 float limit = (link && !flex && !other.get_flex()) ? 1e-6 : 1e-4;
180 const vector<Endpoint> &eps = type.get_endpoints();
181 const vector<Endpoint> &other_eps = other.get_type().get_endpoints();
183 for(unsigned i=0; i<eps.size(); ++i)
185 Point epp = get_endpoint_position(i);
187 for(unsigned j=0; j<other_eps.size(); ++j)
189 if(other.get_link(j))
192 Point epp2 = other.get_endpoint_position(j);
193 float dx = epp2.x-epp.x;
194 float dy = epp2.y-epp.y;
195 if(dx*dx+dy*dy<limit)
197 if(!link || (!flex && !other.get_flex()))
199 set_rotation(other.rot+other_eps[j].dir-eps[i].dir+M_PI);
200 Point p(epp2.x-(eps[i].pos.x*cos(rot)-eps[i].pos.y*sin(rot)),
201 epp2.y-(eps[i].pos.y*cos(rot)+eps[i].pos.x*sin(rot)),
203 if(eps.size()==2 && i==1)
211 break_link(*links[i]);
213 other.links[j] = this;
214 layout.create_blocks(*this);
225 bool Track::snap(Point &pt, float &d) const
227 const vector<Endpoint> &eps = type.get_endpoints();
229 for(unsigned i=0; i<eps.size(); ++i)
231 Point epp = get_endpoint_position(i);
232 float dx = pt.x-epp.x;
233 float dy = pt.y-epp.y;
245 void Track::break_link(Track &trk)
247 for(vector<Track *>::iterator i=links.begin(); i!=links.end(); ++i)
251 trk.break_link(*this);
252 // XXX Creates the blocks twice
253 layout.create_blocks(*this);
258 void Track::break_links()
260 for(vector<Track *>::iterator i=links.begin(); i!=links.end(); ++i)
264 trk->break_link(*this);
268 Track *Track::get_link(unsigned i) const
271 throw InvalidParameterValue("Link index out of range");
276 unsigned Track::traverse(unsigned i, unsigned path) const
278 const vector<Endpoint> &eps = type.get_endpoints();
280 throw InvalidParameterValue("Endpoint index out of range");
282 const Endpoint &ep = eps[i];
284 if(ep.paths&(1<<path))
286 // Find the other endpoint for this path
287 for(unsigned j=0; j<eps.size(); ++j)
288 if((eps[j].paths&(1<<path)) && j!=i)
293 // Find an endpoint that's connected to this one and has the requested path
294 for(unsigned j=0; j<eps.size(); ++j)
295 if((eps[j].paths&(1<<path)) && (eps[j].paths&ep.paths))
299 throw Exception("Track endpoint did not have a counterpart");
302 unsigned Track::traverse(unsigned i) const
304 return traverse(i, active_path);
307 TrackPoint Track::get_point(unsigned epi, unsigned path, float d) const
309 TrackPoint p = type.get_point(epi, path, d);
313 p.pos = Point(pos.x+c*p.pos.x-s*p.pos.y, pos.y+s*p.pos.x+c*p.pos.y, 0);
315 if(type.get_endpoints().size()==2)
317 float len = type.get_path_length(path);
318 float grade = slope/len;
321 p.pos.z = pos.z+grade*d;
326 p.pos.z = pos.z+slope-grade*d;
334 TrackPoint Track::get_point(unsigned epi, float d) const
336 return get_point(epi, active_path, d);
339 void Track::save(list<DataFile::Statement> &st) const
341 st.push_back((DataFile::Statement("position"), pos.x, pos.y, pos.z));
342 st.push_back((DataFile::Statement("rotation"), rot));
343 st.push_back((DataFile::Statement("slope"), slope));
345 st.push_back((DataFile::Statement("turnout_id"), turnout_id));
347 st.push_back((DataFile::Statement("sensor_id"), sensor_id));
349 st.push_back((DataFile::Statement("flex"), true));
352 void Track::turnout_event(unsigned addr, bool state)
358 active_path = (active_path&2) | (state ? 1 : 0);
359 else if(type.is_double_address() && addr==turnout_id+1)
360 active_path = (active_path&1) | (state ? 2 : 0);
364 Track::Loader::Loader(Track &t):
365 DataFile::BasicLoader<Track>(t)
367 add("position", &Loader::position);
368 add("rotation", &Track::rot);
369 add("slope", &Track::slope);
370 add("turnout_id", &Loader::turnout_id);
371 add("sensor_id", &Loader::sensor_id);
372 add("flex", &Track::flex);
375 void Track::Loader::position(float x, float y, float z)
377 obj.pos = Point(x, y, z);
380 void Track::Loader::sensor_id(unsigned id)
382 obj.set_sensor_id(id);
385 void Track::Loader::turnout_id(unsigned id)
387 obj.set_turnout_id(id);
390 } // namespace Marklin