]> git.tdb.fi Git - r2c2.git/blobdiff - source/libr2c2/track.cpp
Give zones a preferred running direction
[r2c2.git] / source / libr2c2 / track.cpp
index c2b190f0551410b149a408e450ee16041151295c..aeccaa697e4bc29a3e8b341737401200bd5bd82e 100644 (file)
@@ -1,14 +1,31 @@
 #include <cmath>
+#include <msp/core/maputils.h>
+#include <msp/strings/format.h>
 #include "block.h"
 #include "catalogue.h"
 #include "driver.h"
 #include "layout.h"
 #include "track.h"
+#include "trackattachment.h"
 #include "tracktype.h"
 
 using namespace std;
 using namespace Msp;
 
+namespace {
+
+struct AttachmentCompare
+{
+       unsigned entry;
+
+       AttachmentCompare(unsigned e): entry(e) { }
+
+       bool operator()(const R2C2::TrackAttachment *a1, const R2C2::TrackAttachment *a2) const
+       { return a1->get_offset_from_endpoint(entry)<a2->get_offset_from_endpoint(entry); }
+};
+
+}
+
 namespace R2C2 {
 
 Track::Track(Layout &l, const TrackType &t):
@@ -17,19 +34,27 @@ Track::Track(Layout &l, const TrackType &t):
        block(0),
        slope(0),
        flex(false),
-       turnout_id(0),
-       sensor_id(0),
+       turnout_addr(0),
+       sensor_addr(0),
        links(type.get_endpoints().size()),
        active_path(0),
-       path_changing(false)
+       path_changing(false),
+       preferred_exit(-1)
 {
        if(type.is_turnout())
-               turnout_id = layout.allocate_turnout_id();
+       {
+               turnout_addr = layout.allocate_turnout_address();
 
-       layout.add_track(*this);
+               if(layout.has_driver())
+               {
+                       Driver &driver = layout.get_driver();
+                       turnout_id = driver.add_turnout(turnout_addr, type);
+                       driver.signal_turnout.connect(sigc::mem_fun(this, &Track::turnout_event));
+                       driver.signal_turnout_failed.connect(sigc::mem_fun(this, &Track::turnout_failed));
+               }
+       }
 
-       if(layout.has_driver())
-               layout.get_driver().signal_turnout.connect(sigc::mem_fun(this, &Track::turnout_event));
+       layout.add(*this);
 
        for(unsigned paths = type.get_paths(); !(paths&1); ++active_path, paths>>=1) ;
 }
@@ -37,7 +62,9 @@ Track::Track(Layout &l, const TrackType &t):
 Track::~Track()
 {
        break_links();
-       layout.remove_track(*this);
+       if(layout.has_driver() && turnout_id)
+               layout.get_driver().remove_turnout(turnout_id);
+       layout.remove(*this);
 }
 
 Track *Track::clone(Layout *to_layout) const
@@ -69,22 +96,25 @@ Block &Track::get_block() const
 void Track::set_position(const Vector &p)
 {
        position = p;
-       for(vector<Track *>::const_iterator i=links.begin(); i!=links.end(); ++i)
-               if(*i)
-                       (*i)->check_slope();
+       signal_moved.emit();
+       propagate_slope();
 }
 
 void Track::set_rotation(const Angle &r)
 {
        rotation = wrap_positive(r);
+       signal_moved.emit();
 }
 
-void Track::set_slope(float s)
+void Track::set_tilt(const Angle &t)
 {
        if(links.size()!=2)
                return;
 
-       slope = s;
+       tilt = t;
+       slope = tan(tilt)*type.get_path_length(0);
+       signal_moved.emit();
+       propagate_slope();
 }
 
 void Track::set_flex(bool f)
@@ -92,6 +122,13 @@ void Track::set_flex(bool f)
        flex = f;
 }
 
+void Track::propagate_slope()
+{
+       for(vector<Track *>::const_iterator i=links.begin(); i!=links.end(); ++i)
+               if(*i)
+                       (*i)->check_slope();
+}
+
 void Track::check_slope()
 {
        if(links.size()!=2)
@@ -103,6 +140,7 @@ void Track::check_slope()
                Vector epp1 = links[1]->get_snap_node(links[1]->get_link_slot(*this)).position;
                position.z = epp0.z;
                slope = epp1.z-position.z;
+               tilt = Geometry::atan(slope/type.get_path_length(0));
        }
        else
        {
@@ -117,68 +155,91 @@ void Track::check_slope()
                        position.z = epp.z-slope;
                }
        }
+
+       signal_moved.emit();
 }
 
-void Track::set_turnout_id(unsigned i)
+void Track::set_turnout_address(unsigned a)
 {
        if(!type.is_turnout())
                throw logic_error("not a turnout");
+       if(!a)
+               throw invalid_argument("Track::set_turnout_id");
+
+       Driver *driver = (layout.has_driver() ? &layout.get_driver() : 0);
 
-       turnout_id = i;
+       if(driver && turnout_id)
+               driver->remove_turnout(turnout_id);
+       turnout_addr = a;
        layout.create_blocks(*this);
        layout.update_routes();
-       if(layout.has_driver() && turnout_id)
-               layout.get_driver().add_turnout(turnout_id, type);
+       if(driver && turnout_addr)
+               turnout_id = driver->add_turnout(turnout_addr, type);
+       else
+               turnout_id = 0;
 }
 
-void Track::set_sensor_id(unsigned i)
+void Track::set_sensor_address(unsigned a)
 {
        if(type.is_turnout())
                throw logic_error("is a turnout");
 
-       sensor_id = i;
+       sensor_addr = a;
        layout.create_blocks(*this);
-       if(layout.has_driver() && sensor_id)
-               layout.get_driver().add_sensor(sensor_id);
+}
+
+void Track::set_preferred_exit(int e)
+{
+       preferred_exit = e;
 }
 
 void Track::set_active_path(unsigned p)
 {
-       if(!turnout_id)
+       if(!type.is_turnout())
                throw logic_error("not a turnout");
        if(!(type.get_paths()&(1<<p)))
                throw invalid_argument("Track::set_active_path");
 
+       if(active_path==p)
+               return;
+
+       signal_path_changing(p);
        path_changing = true;
        layout.get_driver().set_turnout(turnout_id, p);
 }
 
-TrackPoint Track::get_point(unsigned epi, unsigned path, float d) const
+float Track::get_path_length(int p) const
 {
-       TrackPoint p = type.get_point(epi, path, d);
+       if(p<0)
+               p = active_path;
+       return type.get_path_length(p);
+}
 
-       p.pos = position+rotated_vector(p.pos, rotation);
-       p.dir += rotation;
+OrientedPoint Track::get_point(unsigned epi, unsigned path, float d) const
+{
+       OrientedPoint p = type.get_point(epi, path, d);
+
+       p.position = position+rotated_vector(p.position, rotation);
+       p.rotation += rotation;
        if(type.get_endpoints().size()==2)
        {
-               float len = type.get_path_length(path);
-               float grade = slope/len;
+               float dz = tan(tilt)*d;
                if(epi==0)
                {
-                       p.pos.z += grade*d;
-                       p.grade = grade;
+                       p.position.z += dz;
+                       p.tilt = tilt;
                }
                else
                {
-                       p.pos.z += slope-grade*d;
-                       p.grade = -grade;
+                       p.position.z += slope-dz;
+                       p.tilt = -tilt;
                }
        }
 
        return p;
 }
 
-TrackPoint Track::get_point(unsigned epi, float d) const
+OrientedPoint Track::get_point(unsigned epi, float d) const
 {
        return get_point(epi, active_path, d);
 }
@@ -215,12 +276,12 @@ bool Track::snap(Snap &sn, float limit, SnapType what) const
        {
                Vector local = rotated_vector(sn.position-position, -rotation);
 
-               TrackPoint tp = type.get_nearest_point(local);
-               Vector span = local-tp.pos;
+               OrientedPoint np = type.get_nearest_point(local);
+               Vector span = local-np.position;
                if(dot(span, span)<=limit*limit)
                {
-                       sn.position = position+rotated_vector(tp.pos, rotation);
-                       sn.rotation = tp.dir+rotation;
+                       sn.position = position+rotated_vector(np.position, rotation);
+                       sn.rotation = np.rotation+rotation;
                        return true;
                }
        }
@@ -264,7 +325,11 @@ bool Track::link_to(Object &other)
        if(!otrack)
                return false;
 
-       float limit = layout.get_catalogue().get_gauge();
+       float gauge_ratio = otrack->get_type().get_gauge()/type.get_gauge();
+       if(gauge_ratio<0.99 || gauge_ratio>1.01)
+               return false;
+
+       float limit = type.get_gauge();
        if(!flex && !otrack->get_flex())
                limit /= 10;
        limit *= limit;
@@ -283,6 +348,7 @@ bool Track::link_to(Object &other)
                        if(dot(span, span)<limit && abs(da).radians()<0.01)
                        {
                                break_link(i);
+                               otrack->break_link(j);
                                links[i] = otrack;
                                otrack->links[j] = this;
                                check_slope();
@@ -308,33 +374,63 @@ bool Track::break_link(unsigned i)
                return false;
 
        links[i] = 0;
-       other->break_link(*this);
-       // XXX Creates the blocks twice, because the other track calls this too
-       layout.create_blocks(*this);
+       if(!other->break_link(*this))
+       {
+               /* If the call doesn't succeed, it means that the other track already
+               broke the link and is calling us right now.  Recreate blocks in the inner
+               call so it occurs before any signals are emitted. */
+               layout.create_blocks(*this);
+       }
+
        signal_link_changed.emit(i, 0);
 
        return true;
 }
 
+void Track::add_attachment(TrackAttachment &a)
+{
+       if(find(attachments.begin(), attachments.end(), &a)!=attachments.end())
+               throw key_error(&a);
+       attachments.push_back(&a);
+}
+
+void Track::remove_attachment(TrackAttachment &a)
+{
+       AttachmentList::iterator i = find(attachments.begin(), attachments.end(), &a);
+       if(i==attachments.end())
+               throw key_error(&a);
+       attachments.erase(i);
+}
+
+Track::AttachmentList Track::get_attachments_ordered(unsigned epi) const
+{
+       AttachmentList result = attachments;
+       result.sort(AttachmentCompare(epi));
+       return result;
+}
+
 void Track::save(list<DataFile::Statement> &st) const
 {
        st.push_back((DataFile::Statement("position"), position.x, position.y, position.z));
        st.push_back((DataFile::Statement("rotation"), rotation.radians()));
-       st.push_back((DataFile::Statement("slope"), slope));
-       if(turnout_id)
-               st.push_back((DataFile::Statement("turnout_id"), turnout_id));
-       if(sensor_id)
-               st.push_back((DataFile::Statement("sensor_id"), sensor_id));
+       st.push_back((DataFile::Statement("tilt"), tilt.radians()));
+       if(turnout_addr)
+               st.push_back((DataFile::Statement("turnout_address"), turnout_addr));
+       if(sensor_addr)
+               st.push_back((DataFile::Statement("sensor_address"), sensor_addr));
        if(flex)
                st.push_back((DataFile::Statement("flex"), true));
 }
 
-void Track::turnout_event(unsigned addr, unsigned state)
+void Track::save_dynamic(list<DataFile::Statement> &st) const
 {
-       if(!turnout_id)
-               return;
+       if(turnout_addr)
+               st.push_back((DataFile::Statement("path"), active_path));
+}
 
-       if(addr==turnout_id)
+void Track::turnout_event(unsigned id, unsigned state)
+{
+       if(id==turnout_id)
        {
                active_path = state;
                path_changing = false;
@@ -342,36 +438,71 @@ void Track::turnout_event(unsigned addr, unsigned state)
        }
 }
 
+void Track::turnout_failed(unsigned id)
+{
+       if(id==turnout_id)
+       {
+               path_changing = false;
+               layout.emergency(block, "Turnout failed");
+       }
+}
+
 
 Track::Loader::Loader(Track &t):
        DataFile::ObjectLoader<Track>(t)
 {
+       add("path",       &Loader::path);
        add("position",   &Loader::position);
        add("rotation",   &Loader::rotation);
-       add("slope",      &Track::slope);
-       add("turnout_id", &Loader::turnout_id);
-       add("sensor_id",  &Loader::sensor_id);
+       add("tilt",       &Loader::tilt);
+       add("turnout_id", &Loader::turnout_address);
+       add("turnout_address", &Loader::turnout_address);
+       add("sensor_id",  &Loader::sensor_address);
+       add("sensor_address",  &Loader::sensor_address);
        add("flex",       &Track::flex);
+
+       // deprecated
+       add("slope",      &Loader::slope);
+}
+
+void Track::Loader::path(unsigned p)
+{
+       obj.set_active_path(p);
+       if(obj.path_changing)
+       {
+               obj.active_path = p;
+               obj.signal_path_changed.emit(p);
+       }
 }
 
 void Track::Loader::position(float x, float y, float z)
 {
-       obj.position = Vector(x, y, z);
+       obj.set_position(Vector(x, y, z));
 }
 
 void Track::Loader::rotation(float r)
 {
-       obj.rotation = Angle::from_radians(r);
+       obj.set_rotation(Angle::from_radians(r));
+}
+
+void Track::Loader::sensor_address(unsigned addr)
+{
+       obj.set_sensor_address(addr);
+}
+
+void Track::Loader::slope(float s)
+{
+       obj.set_tilt(Geometry::atan(s/obj.type.get_path_length(0)));
 }
 
-void Track::Loader::sensor_id(unsigned id)
+void Track::Loader::tilt(float t)
 {
-       obj.set_sensor_id(id);
+       obj.set_tilt(Angle::from_radians(t));
 }
 
-void Track::Loader::turnout_id(unsigned id)
+void Track::Loader::turnout_address(unsigned addr)
 {
-       obj.set_turnout_id(id);
+       obj.set_turnout_address(addr);
 }
 
 } // namespace R2C2