/* $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
*/
#include <cmath>
+#include "driver.h"
+#include "layout.h"
#include "track.h"
#include "tracktype.h"
namespace Marklin {
-Track::Track(const TrackType &t):
+Track::Track(Layout &l, const TrackType &t):
+ layout(l),
type(t),
rot(0),
slope(0),
flex(false),
turnout_id(0),
sensor_id(0),
- links(t.get_endpoints().size())
-{ }
+ links(t.get_endpoints().size()),
+ active_path(0)
+{
+ layout.add_track(*this);
+
+ if(layout.has_driver())
+ layout.get_driver().signal_turnout.connect(sigc::mem_fun(this, &Track::turnout_event));
+}
Track::~Track()
{
break_links();
+ layout.remove_track(*this);
}
void Track::set_position(const Point &p)
{
- pos=p;
+ pos = p;
}
void Track::set_rotation(float r)
{
- rot=r;
+ rot = r;
while(rot<0)
- rot+=M_PI*2;
+ rot += M_PI*2;
while(rot>M_PI*2)
- rot-=M_PI*2;
+ rot -= M_PI*2;
}
void Track::set_slope(float s)
{
if(links.size()!=2) return;
- slope=s;
+ slope = s;
}
void Track::set_flex(bool f)
{
- flex=f;
+ flex = f;
}
void Track::check_slope()
if(links[0] && links[1])
{
- Point epp0=links[0]->get_endpoint_position(links[0]->get_endpoint_by_link(*this));
- Point epp1=links[1]->get_endpoint_position(links[1]->get_endpoint_by_link(*this));
- pos.z=epp0.z;
- slope=epp1.z-pos.z;
+ Point epp0 = links[0]->get_endpoint_position(links[0]->get_endpoint_by_link(*this));
+ Point epp1 = links[1]->get_endpoint_position(links[1]->get_endpoint_by_link(*this));
+ pos.z = epp0.z;
+ slope = epp1.z-pos.z;
}
else
{
- slope=0;
+ slope = 0;
if(links[0])
{
- Point epp=links[0]->get_endpoint_position(links[0]->get_endpoint_by_link(*this));
- pos.z=epp.z;
+ Point epp = links[0]->get_endpoint_position(links[0]->get_endpoint_by_link(*this));
+ pos.z = epp.z;
}
else if(links[1])
{
- Point epp=links[1]->get_endpoint_position(links[1]->get_endpoint_by_link(*this));
- pos.z=epp.z;
+ Point epp = links[1]->get_endpoint_position(links[1]->get_endpoint_by_link(*this));
+ pos.z = epp.z;
}
}
}
void Track::set_turnout_id(unsigned i)
{
- turnout_id=i;
+ if(!type.is_turnout())
+ throw InvalidState("Not a turnout");
+
+ turnout_id = i;
+ layout.create_blocks(*this);
+ if(layout.has_driver() && turnout_id)
+ layout.get_driver().add_turnout(turnout_id);
}
void Track::set_sensor_id(unsigned i)
{
- sensor_id=i;
+ if(type.is_turnout())
+ throw InvalidState("Can't set sensor on a turnout");
+
+ sensor_id = i;
+ layout.create_blocks(*this);
+ if(layout.has_driver() && sensor_id)
+ layout.get_driver().add_sensor(sensor_id);
+}
+
+void Track::set_active_path(unsigned p)
+{
+ if(!turnout_id)
+ throw InvalidState("Not a turnout");
+ if(!(type.get_paths()&(1<<p)))
+ throw InvalidParameterValue("Invalid path");
+
+ layout.get_driver().set_turnout(turnout_id, p&1);
+ if(type.get_n_paths()>2)
+ layout.get_driver().set_turnout(turnout_id+1, p&2);
}
int Track::get_endpoint_by_link(const Track &other) const
Point Track::get_endpoint_position(unsigned epi) const
{
- const vector<Endpoint> &eps=type.get_endpoints();
+ const vector<Endpoint> &eps = type.get_endpoints();
if(epi>=eps.size())
throw InvalidParameterValue("Endpoint index out of range");
- const Endpoint &ep=eps[epi];
+ const Endpoint &ep = eps[epi];
- float c=cos(rot);
- float s=sin(rot);
+ float c = cos(rot);
+ float s = sin(rot);
Point p(pos.x+c*ep.pos.x-s*ep.pos.y, pos.y+s*ep.pos.x+c*ep.pos.y, pos.z);
if(eps.size()==2 && epi==1)
- p.z+=slope;
+ p.z += slope;
return p;
}
float Track::get_endpoint_direction(unsigned epi) const
{
- const vector<Endpoint> &eps=type.get_endpoints();
+ const vector<Endpoint> &eps = type.get_endpoints();
if(epi>=eps.size())
throw InvalidParameterValue("Endpoint index out of range");
- const Endpoint &ep=eps[epi];
+ const Endpoint &ep = eps[epi];
return rot+ep.dir;
}
bool Track::snap_to(Track &other, bool link)
{
- float limit=(link && !flex) ? 1e-6 : 1e-4;
- const vector<Endpoint> &eps=type.get_endpoints();
- const vector<Endpoint> &other_eps=other.get_type().get_endpoints();
+ float limit = (link && !flex) ? 1e-6 : 1e-4;
+ const vector<Endpoint> &eps = type.get_endpoints();
+ const vector<Endpoint> &other_eps = other.get_type().get_endpoints();
for(unsigned i=0; i<eps.size(); ++i)
{
- Point epp=get_endpoint_position(i);
+ Point epp = get_endpoint_position(i);
for(unsigned j=0; j<other_eps.size(); ++j)
{
if(other.get_link(j))
continue;
- Point epp2=other.get_endpoint_position(j);
- float dx=epp2.x-epp.x;
- float dy=epp2.y-epp.y;
+ Point epp2 = other.get_endpoint_position(j);
+ float dx = epp2.x-epp.x;
+ float dy = epp2.y-epp.y;
if(dx*dx+dy*dy<limit)
{
set_rotation(other.rot+other_eps[j].dir-eps[i].dir+M_PI);
- set_position(Point(epp2.x-(eps[i].pos.x*cos(rot)-eps[i].pos.y*sin(rot)), epp2.y-(eps[i].pos.y*cos(rot)+eps[i].pos.x*sin(rot)), epp2.z));
+ Point p(epp2.x-(eps[i].pos.x*cos(rot)-eps[i].pos.y*sin(rot)),
+ epp2.y-(eps[i].pos.y*cos(rot)+eps[i].pos.x*sin(rot)),
+ epp2.z);
+ if(eps.size()==2 && i==1)
+ p.z -= slope;
+ set_position(p);
if(link)
{
if(links[i])
break_link(*links[i]);
- links[i]=&other;
- other.links[j]=this;
+ links[i] = &other;
+ other.links[j] = this;
+ layout.create_blocks(*this);
}
return true;
bool Track::snap(Point &pt, float &d) const
{
- const vector<Endpoint> &eps=type.get_endpoints();
+ const vector<Endpoint> &eps = type.get_endpoints();
for(unsigned i=0; i<eps.size(); ++i)
{
- Point epp=get_endpoint_position(i);
- float dx=pt.x-epp.x;
- float dy=pt.y-epp.y;
+ Point epp = get_endpoint_position(i);
+ float dx = pt.x-epp.x;
+ float dy = pt.y-epp.y;
if(dx*dx+dy*dy<1e-4)
{
- pt=epp;
- d=rot+eps[i].dir;
+ pt = epp;
+ d = rot+eps[i].dir;
return true;
}
}
for(vector<Track *>::iterator i=links.begin(); i!=links.end(); ++i)
if(*i==&trk)
{
- *i=0;
+ *i = 0;
trk.break_link(*this);
+ // XXX Creates the blocks twice
+ layout.create_blocks(*this);
return;
}
}
for(vector<Track *>::iterator i=links.begin(); i!=links.end(); ++i)
if(Track *trk=*i)
{
- *i=0;
+ *i = 0;
trk->break_link(*this);
}
}
return links[i];
}
-int Track::traverse(unsigned i, unsigned route) const
+unsigned Track::traverse(unsigned i, unsigned path) const
{
- const vector<Endpoint> &eps=type.get_endpoints();
+ const vector<Endpoint> &eps = type.get_endpoints();
if(i>=eps.size())
throw InvalidParameterValue("Endpoint index out of range");
- const Endpoint &ep=eps[i];
+ const Endpoint &ep = eps[i];
- if(ep.routes&(1<<route))
+ if(ep.paths&(1<<path))
{
- // Find the other endpoint for this route
+ // Find the other endpoint for this path
for(unsigned j=0; j<eps.size(); ++j)
- if((eps[j].routes&(1<<route)) && j!=i)
+ if((eps[j].paths&(1<<path)) && j!=i)
return j;
}
else
{
- // Find an endpoint that's connected to this one and has the requested route
+ // Find an endpoint that's connected to this one and has the requested path
for(unsigned j=0; j<eps.size(); ++j)
- if((eps[j].routes&(1<<route)) && (eps[j].routes&ep.routes))
+ if((eps[j].paths&(1<<path)) && (eps[j].paths&ep.paths))
return j;
}
- return -1;
+ throw Exception("Track endpoint did not have a counterpart");
}
-Point Track::get_point(unsigned epi, unsigned route, float d) const
+Point Track::get_point(unsigned epi, unsigned path, float d) const
{
- const vector<Endpoint> &eps=type.get_endpoints();
+ const vector<Endpoint> &eps = type.get_endpoints();
if(epi>=eps.size())
throw InvalidParameterValue("Endpoint index out of range");
- float x=eps[epi].pos.x;
- float y=eps[epi].pos.y;
+ float x = eps[epi].pos.x;
+ float y = eps[epi].pos.y;
+ float z = 0;
+ float slope_norm = 0;
+ if(eps.size()==2)
+ {
+ slope_norm = slope/type.get_total_length();
+ if(epi==1)
+ {
+ z = slope;
+ slope_norm = -slope_norm;
+ }
+ }
- const vector<TrackPart> &parts=type.get_parts();
- const TrackPart *last_part=0;
+ const vector<TrackPart> &parts = type.get_parts();
+ const TrackPart *last_part = 0;
while(1)
{
for(vector<TrackPart>::const_iterator i=parts.begin(); i!=parts.end(); ++i)
{
- if((eps[epi].routes&(1<<route)) && i->route!=route)
+ if((eps[epi].paths&(1<<path)) && i->path!=path)
continue;
if(&*i==last_part)
continue;
i->collect_endpoints(part_eps);
for(unsigned j=0; j<part_eps.size(); ++j)
{
- float dx=part_eps[j].pos.x-x;
- float dy=part_eps[j].pos.y-y;
+ float dx = part_eps[j].pos.x-x;
+ float dy = part_eps[j].pos.y-y;
if(dx*dx+dy*dy<1e-6)
{
- float plen=i->length;
+ float plen = i->length;
if(i->radius)
- plen*=abs(i->radius);
+ plen *= abs(i->radius);
if(d<plen)
{
+ z += d*slope_norm;
if(j==1)
- d=plen-d;
- Point p=i->get_point(d);
- float c=cos(rot);
- float s=sin(rot);
- return Point(pos.x+c*p.x-s*p.y, pos.y+c*p.y+s*p.x);
+ d = plen-d;
+ Point p = i->get_point(d);
+ float c = cos(rot);
+ float s = sin(rot);
+ return Point(pos.x+c*p.x-s*p.y, pos.y+c*p.y+s*p.x, pos.z+z);
}
else if(part_eps.size()>1)
{
- d-=plen;
- x=part_eps[1-j].pos.x;
- y=part_eps[1-j].pos.y;
- last_part=&*i;
- i=parts.begin();
+ d -= plen;
+ x = part_eps[1-j].pos.x;
+ y = part_eps[1-j].pos.y;
+ z += plen*slope_norm;
+ last_part = &*i;
+ i = parts.begin();
break;
}
else
}
}
-Track *Track::copy() const
-{
- Track *trk=new Track(type);
- trk->set_position(pos);
- trk->set_rotation(rot);
- trk->set_slope(slope);
- trk->set_flex(flex);
-
- return trk;
-}
-
void Track::save(list<DataFile::Statement> &st) const
{
st.push_back((DataFile::Statement("position"), pos.x, pos.y, pos.z));
st.push_back((DataFile::Statement("flex"), true));
}
+void Track::turnout_event(unsigned addr, bool state)
+{
+ if(!turnout_id)
+ return;
+
+ if(addr==turnout_id)
+ active_path = (active_path&2) | (state ? 1 : 0);
+ else if(type.is_double_address() && addr==turnout_id+1)
+ active_path = (active_path&1) | (state ? 2 : 0);
+}
+
Track::Loader::Loader(Track &t):
DataFile::BasicLoader<Track>(t)
add("position", &Loader::position);
add("rotation", &Track::rot);
add("slope", &Track::slope);
- add("turnout_id", &Track::turnout_id);
- add("sensor_id", &Track::sensor_id);
+ add("turnout_id", &Loader::turnout_id);
+ add("sensor_id", &Loader::sensor_id);
add("flex", &Track::flex);
}
void Track::Loader::position(float x, float y, float z)
{
- obj.pos=Point(x, y, z);
+ obj.pos = Point(x, y, z);
+}
+
+void Track::Loader::sensor_id(unsigned id)
+{
+ obj.set_sensor_id(id);
+}
+
+void Track::Loader::turnout_id(unsigned id)
+{
+ obj.set_turnout_id(id);
}
} // namespace Marklin
projects / r2c2.git / blobdiff