1 #ifndef LIBR2C2_ARDUCONTROL_H_
2 #define LIBR2C2_ARDUCONTROL_H_
4 #include <msp/core/mutex.h>
5 #include <msp/core/thread.h>
6 #include <msp/io/serial.h>
7 #include <msp/time/timedelta.h>
8 #include <msp/time/timestamp.h>
13 class ArduControl: public Driver
20 READ_POWER_STATE = 0x03,
21 READ_TRACK_CURRENT = 0x08,
22 SET_OVERCURRENT_LIMIT = 0x09,
23 READ_INPUT_VOLTAGE = 0x0A,
24 MOTOROLA_SPEED = 0x11,
25 MOTOROLA_REVERSE = 0x12,
26 MOTOROLA_SPEED_DIRECTION = 0x13,
27 MOTOROLA_SPEED_FUNCTION = 0x14,
28 MOTOROLA_SOLENOID = 0x15,
29 MFX_SET_STATION_ID = 0x21,
32 MFX_ASSIGN_ADDRESS = 0x24,
35 MFX_SPEED_FUNCS8 = 0x29,
36 MFX_SPEED_FUNCS16 = 0x2A,
39 RECEIVE_OVERRUN = 0x81,
41 INVALID_COMMAND = 0x83,
64 unsigned char command;
65 unsigned short serial;
70 operator bool() const { return type!=NONE; }
92 struct ControlledVariable
96 unsigned short serial;
98 ControlledVariable(): current(), pending(), serial(0) { }
99 ControlledVariable(T v): current(v), pending(v), serial(0) { }
101 bool set(T v) { if(v==pending) return false; pending = v; ++serial; return true; }
102 bool commit(unsigned short s) { if(s!=serial) return false; current = pending; return true; }
104 operator T() const { return current; }
119 ControlledVariable<unsigned> speed;
120 ControlledVariable<bool> reverse;
121 ControlledVariable<unsigned> funcs;
122 unsigned last_change_age;
124 Locomotive(Protocol, unsigned);
126 unsigned create_speed_dir_command(char *) const;
127 unsigned create_speed_func_command(unsigned, char *) const;
147 ControlledVariable<unsigned> state;
149 Msp::Time::TimeDelta active_time;
151 Accessory(Kind, unsigned, unsigned);
153 unsigned create_state_command(unsigned, bool, char *) const;
164 ControlledVariable<bool> state;
169 struct PendingCommand
173 unsigned char length;
174 unsigned repeat_count;
177 PendingCommand(GeneralCommand);
178 PendingCommand(Locomotive &, Locomotive::Command, unsigned = 0);
179 PendingCommand(Accessory &, Accessory::Command, unsigned = 0);
189 virtual bool get_work(PendingCommand &) = 0;
190 virtual void process_reply(const char *, unsigned) { }
193 class RefreshTask: public Task
196 ArduControl &control;
201 RefreshTask(ArduControl &);
203 virtual bool get_work(PendingCommand &);
206 class S88Task: public Task
209 ArduControl &control;
210 unsigned octets_remaining;
213 S88Task(ArduControl &);
215 virtual bool get_work(PendingCommand &);
216 virtual void process_reply(const char *, unsigned);
219 class MfxAnnounceTask: public Task
222 ArduControl &control;
223 Msp::Time::TimeStamp next;
226 MfxAnnounceTask(ArduControl &);
228 virtual bool get_work(PendingCommand &);
231 class MfxSearchTask: public Task
234 ArduControl &control;
235 Msp::Time::TimeStamp next;
241 MfxSearchTask(ArduControl &);
243 virtual bool get_work(PendingCommand &);
244 virtual void process_reply(const char *, unsigned);
247 class ControlThread: public Msp::Thread
250 ArduControl &control;
252 std::vector<Task *> tasks;
255 ControlThread(ArduControl &);
260 bool get_work(PendingCommand &);
261 unsigned do_command(const PendingCommand &);
262 unsigned process_reply(const char *, unsigned);
265 typedef std::map<unsigned, Locomotive> LocomotiveMap;
266 typedef std::list<Locomotive *> LocomotivePtrList;
267 typedef std::map<unsigned, Accessory> AccessoryMap;
268 typedef std::list<Accessory *> AccessoryPtrList;
269 typedef std::map<unsigned, Sensor> SensorMap;
271 Msp::IO::Serial serial;
274 ControlledVariable<bool> power;
276 LocomotiveMap locomotives;
277 LocomotivePtrList refresh_cycle;
278 LocomotivePtrList::iterator next_refresh;
279 unsigned refresh_counter;
280 AccessoryMap accessories;
281 AccessoryPtrList accessory_queue;
282 Accessory *active_accessory;
283 Msp::Time::TimeStamp off_timeout;
284 std::list<PendingCommand> command_queue;
285 std::list<Tag> completed_commands;
288 unsigned n_s88_octets;
290 unsigned mfx_announce_serial;
291 unsigned next_mfx_address;
294 ControlThread thread;
296 static ProtocolInfo protocol_info[2];
299 ArduControl(const std::string &);
302 virtual void set_power(bool);
303 virtual bool get_power() const { return power; }
304 virtual void halt(bool);
305 virtual bool is_halted() const { return false; }
307 virtual const char *enumerate_protocols(unsigned) const;
309 static Protocol map_protocol(const std::string &);
311 virtual unsigned get_protocol_speed_steps(const std::string &) const;
313 virtual unsigned add_loco(unsigned, const std::string &, const VehicleType &);
314 virtual void remove_loco(unsigned);
315 virtual void set_loco_speed(unsigned, unsigned);
316 virtual void set_loco_reverse(unsigned, bool);
317 virtual void set_loco_function(unsigned, unsigned, bool);
319 void add_loco_to_refresh(Locomotive &);
320 void remove_loco_from_refresh(Locomotive &);
321 Locomotive *get_loco_to_refresh();
322 void advance_next_refresh();
325 virtual unsigned add_turnout(unsigned, const TrackType &);
326 virtual void remove_turnout(unsigned);
327 virtual void set_turnout(unsigned, unsigned);
328 virtual unsigned get_turnout(unsigned) const;
330 virtual unsigned add_signal(unsigned, const SignalType &);
331 virtual void remove_signal(unsigned);
332 virtual void set_signal(unsigned, unsigned);
333 virtual unsigned get_signal(unsigned) const;
336 unsigned add_accessory(Accessory::Kind, unsigned, unsigned);
337 void remove_accessory(Accessory::Kind, unsigned);
338 void set_accessory(Accessory::Kind, unsigned, unsigned);
339 unsigned get_accessory(Accessory::Kind, unsigned) const;
342 virtual unsigned add_sensor(unsigned);
343 virtual void remove_sensor(unsigned);
344 virtual void set_sensor(unsigned, bool) { }
345 virtual bool get_sensor(unsigned) const;
348 virtual void flush();
351 void push_command(const PendingCommand &);
352 bool pop_command(PendingCommand &);
353 void push_completed_tag(const Tag &);
354 Tag pop_completed_tag();