Major code refactoring:

[r2c2.git] / source / libmarklin / control.cpp

diff --git a/source/libmarklin/control.cpp b/source/libmarklin/control.cpp
index 0ce16f088740524490e2f891b3d62561f588a309..02b8f0f16949a2aab1c5f49013de37e32bf10415 100644 (file)
--- a/source/libmarklin/control.cpp
+++ b/source/libmarklin/control.cpp
@@ -2,7 +2,7 @@
 #include <termios.h>
 #include <sys/poll.h>
 #include <iostream>
-#include <msp/core/error.h>
+#include <msp/core/except.h>
 #include <msp/time/units.h>
 #include <msp/time/utils.h>
 #include "command.h"
@@ -30,32 +30,6 @@ void Control::set_power(bool p)
                command(string(1, CMD_POWER_OFF));
 }
 
-Turnout *Control::get_turnout(unsigned id) const
-{
-       TurnoutMap::const_iterator i=turnouts.find(id);
-       if(i!=turnouts.end())
-               return i->second;
-       
-       return 0;
-}
-
-Locomotive *Control::get_locomotive(unsigned id) const
-{
-       for(LocomotiveSeq::const_iterator i=locomotives.begin(); i!=locomotives.end(); ++i)
-               if((*i)->get_address()==id)
-                       return *i;
-       return 0;
-}
-
-Sensor *Control::get_sensor(unsigned id) const
-{
-       SensorMap::const_iterator i=sensors.find(id);
-       if(i!=sensors.end())
-               return i->second;
-       
-       return 0;
-}
-
 void Control::open(const string &dev)
 {
        serial_fd=::open(dev.c_str(), O_RDWR);
@@ -114,31 +88,56 @@ Command &Control::command(const string &cmd)
        return queue.back();
 }
 
-void Control::add_turnout(Turnout *t)
+void Control::add_turnout(Turnout &t)
 {
-       if(turnouts.count(t->get_address())==0)
-               turnouts.insert(TurnoutMap::value_type(t->get_address(), t));
+       turnouts[t.get_address()]=&t;
 }
 
-void Control::add_locomotive(Locomotive *l)
+Turnout &Control::get_turnout(unsigned id) const
 {
-       if(find(locomotives.begin(), locomotives.end(), l)==locomotives.end())
-               locomotives.push_back(l);
+       map<unsigned, Turnout *>::const_iterator i=turnouts.find(id);
+       if(i==turnouts.end())
+               throw KeyError("Unknown turnout");
+
+       return *i->second;
 }
 
-void Control::add_sensor(Sensor *s)
+void Control::add_locomotive(Locomotive &l)
 {
-       if(sensors.count(s->get_address())==0)
-       {
-               sensors.insert(SensorMap::value_type(s->get_address(), s));
-               poll_sensors=true;
-       }
+       locomotives[l.get_address()]=&l;
+}
+
+Locomotive &Control::get_locomotive(unsigned id) const
+{
+       map<unsigned, Locomotive *>::const_iterator i=locomotives.find(id);
+       if(i==locomotives.end())
+               throw KeyError("Unknown locomotive");
+
+       return *i->second;
+}
+
+void Control::add_sensor(Sensor &s)
+{
+       sensors[s.get_address()]=&s;
+       poll_sensors=true;
+}
+
+Sensor &Control::get_sensor(unsigned id) const
+{
+       map<unsigned, Sensor *>::const_iterator i=sensors.find(id);
+       if(i==sensors.end())
+               throw KeyError("Unknown sensor");
+
+       return *i->second;
 }
 
 void Control::tick()
 {
        const Time::TimeStamp t=Time::now();
 
+       for(map<unsigned, Sensor *>::const_iterator i=sensors.begin(); i!=sensors.end(); ++i)
+               i->second->tick();
+
        if(t>next_event_query)
        {
                next_event_query=t+300*Time::msec;
@@ -203,14 +202,19 @@ void Control::tick()
        }
 }
 
+Time::Timer::Slot &Control::set_timer(const Time::TimeDelta &dt)
+{
+       return timer.add(dt);
+}
+
 Control::~Control()
 {
-       for(SensorMap::iterator i=sensors.begin(); i!=sensors.end(); ++i)
+       for(map<unsigned, Sensor *>::iterator i=sensors.begin(); i!=sensors.end(); ++i)
+               delete i->second;
+       for(map<unsigned, Turnout *>::iterator i=turnouts.begin(); i!=turnouts.end(); ++i)
                delete i->second;
-       for(TurnoutMap::iterator i=turnouts.begin(); i!=turnouts.end(); ++i)
+       for(map<unsigned, Locomotive *>::iterator i=locomotives.begin(); i!=locomotives.end(); ++i)
                delete i->second;
-       for(LocomotiveSeq::iterator i=locomotives.begin(); i!=locomotives.end(); ++i)
-               delete *i;
        close(serial_fd);
 }