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/datafile/objectloader.h>
7 #include <msp/fs/path.h>
8 #include <msp/io/serial.h>
9 #include <msp/time/timedelta.h>
10 #include <msp/time/timestamp.h>
15 class ArduControl: public Driver
18 class Loader: public Msp::DataFile::ObjectLoader<ArduControl>
21 Loader(ArduControl &);
24 void mfx_announce_serial(unsigned);
25 void mfx_locomotive(unsigned);
33 READ_POWER_STATE = 0x03,
34 READ_TRACK_CURRENT = 0x08,
35 SET_OVERCURRENT_LIMIT = 0x09,
36 READ_INPUT_VOLTAGE = 0x0A,
37 MOTOROLA_SPEED = 0x11,
38 MOTOROLA_REVERSE = 0x12,
39 MOTOROLA_SPEED_DIRECTION = 0x13,
40 MOTOROLA_SPEED_FUNCTION = 0x14,
41 MOTOROLA_SOLENOID = 0x15,
42 MFX_SET_STATION_ID = 0x21,
45 MFX_ASSIGN_ADDRESS = 0x24,
48 MFX_SPEED_FUNCS8 = 0x29,
49 MFX_SPEED_FUNCS16 = 0x2A,
53 RECEIVE_OVERRUN = 0x81,
55 INVALID_COMMAND = 0x83,
59 BAUD_CHANGE_FAILED = 0xA1,
64 MFX_SEARCH_FEEDBACK = 0xD1,
65 MFX_PING_FEEDBACK = 0xD2
80 unsigned char command;
81 unsigned short serial;
86 operator bool() const { return type!=NONE; }
103 unsigned max_address;
109 struct ControlledVariable
113 unsigned short serial;
115 ControlledVariable(): current(), pending(), serial(0) { }
116 ControlledVariable(T v): current(v), pending(v), serial(0) { }
118 bool set(T v) { if(v==pending) return false; pending = v; ++serial; return true; }
119 bool commit(unsigned short s) { if(s!=serial) return false; current = pending; return true; }
120 void rollback() { pending = current; ++serial; }
122 operator T() const { return current; }
137 ControlledVariable<unsigned> speed;
138 ControlledVariable<bool> reverse;
139 ControlledVariable<unsigned> funcs;
140 unsigned last_change_age;
142 Locomotive(Protocol, unsigned);
144 unsigned create_speed_dir_command(char *) const;
145 unsigned create_speed_func_command(unsigned, char *) const;
148 struct MfxInfo: public DetectedLocomotive
150 class Loader: public Msp::DataFile::ObjectLoader<MfxInfo>
176 unsigned valid_states;
177 ControlledVariable<unsigned> state;
180 Msp::Time::TimeDelta active_time;
182 Accessory(Kind, unsigned, unsigned, unsigned);
184 unsigned create_state_command(unsigned, bool, char *) const;
195 ControlledVariable<bool> state;
200 struct PendingCommand
204 unsigned char length;
205 unsigned repeat_count;
208 PendingCommand(GeneralCommand);
209 PendingCommand(Locomotive &, Locomotive::Command, unsigned = 0);
210 PendingCommand(Accessory &, Accessory::Command, unsigned = 0);
221 void push(const T &);
233 virtual bool get_work(PendingCommand &) = 0;
234 virtual void process_reply(const char *, unsigned) { }
237 class CommandQueueTask: public Task
240 Queue<PendingCommand> queue;
243 virtual bool get_work(PendingCommand &);
245 void push(const PendingCommand &);
246 bool empty() const { return queue.empty(); }
249 class RefreshTask: public Task
252 typedef std::list<Locomotive *> LocomotivePtrList;
254 LocomotivePtrList cycle;
255 LocomotivePtrList::iterator next;
264 virtual bool get_work(PendingCommand &);
266 void add_loco(Locomotive &);
267 void remove_loco(Locomotive &);
269 Locomotive *get_next_loco();
273 class S88Task: public Task
276 ArduControl &control;
278 unsigned octets_remaining;
282 S88Task(ArduControl &);
284 virtual bool get_work(PendingCommand &);
285 virtual void process_reply(const char *, unsigned);
287 void set_n_octets(unsigned);
288 void grow_n_octets(unsigned);
291 class MfxAnnounceTask: public Task
295 Msp::Time::TimeStamp next;
300 virtual bool get_work(PendingCommand &);
302 void set_serial(unsigned);
303 unsigned get_serial() const { return serial; }
306 class MfxSearchTask: public Task
309 ArduControl &control;
310 unsigned next_address;
311 Msp::Time::TimeStamp next;
315 Queue<MfxInfo> queue;
318 MfxSearchTask(ArduControl &);
320 virtual bool get_work(PendingCommand &);
321 virtual void process_reply(const char *, unsigned);
323 void set_next_address(unsigned);
324 bool pop_info(MfxInfo &);
327 class MonitorTask: public Task
334 Msp::Time::TimeStamp next_poll;
340 virtual bool get_work(PendingCommand &);
341 virtual void process_reply(const char *, unsigned);
343 float get_voltage() const { return voltage; }
344 float get_current() const { return current; }
346 float get_peak() const { return peak_level-base_level; }
349 class ControlThread: public Msp::Thread
352 ArduControl &control;
354 std::vector<Task *> tasks;
357 ControlThread(ArduControl &);
362 void init_baud_rate();
363 bool get_work(PendingCommand &);
364 unsigned do_command(const PendingCommand &, const Msp::Time::TimeDelta &);
365 unsigned process_reply(const char *, unsigned);
366 void set_power(bool);
369 typedef std::map<unsigned, Locomotive> LocomotiveMap;
370 typedef std::vector<MfxInfo> MfxInfoArray;
371 typedef std::map<unsigned, Accessory> AccessoryMap;
372 typedef std::list<Accessory *> AccessoryPtrList;
373 typedef std::map<unsigned, Sensor> SensorMap;
375 Msp::IO::Serial serial;
377 Msp::FS::Path state_file;
379 ControlledVariable<bool> power;
382 LocomotiveMap locomotives;
383 MfxInfoArray mfx_info;
384 AccessoryMap accessories;
385 AccessoryPtrList accessory_queue;
386 Accessory *active_accessory;
387 unsigned char active_index;
388 Msp::Time::TimeStamp off_timeout;
392 Msp::Time::TimeDelta command_timeout;
393 CommandQueueTask command_queue;
394 Queue<Tag> completed_commands;
397 MfxAnnounceTask mfx_announce;
398 MfxSearchTask mfx_search;
400 ControlThread thread;
402 static ProtocolInfo protocol_info[2];
405 ArduControl(const Options &);
408 virtual void set_power(bool);
409 virtual bool get_power() const { return power; }
410 virtual void halt(bool);
411 virtual bool is_halted() const { return halted; }
413 virtual const char *enumerate_protocols(unsigned) const;
415 static Protocol map_protocol(const std::string &);
417 virtual unsigned get_protocol_speed_steps(const std::string &) const;
419 virtual const DetectedLocomotive *enumerate_detected_locos(unsigned) const;
420 virtual unsigned add_loco(unsigned, const std::string &, const VehicleType &);
422 MfxInfoArray::iterator add_mfx_info(const MfxInfo &);
424 virtual void remove_loco(unsigned);
425 virtual void set_loco_speed(unsigned, unsigned);
426 virtual void set_loco_reverse(unsigned, bool);
427 virtual void set_loco_function(unsigned, unsigned, bool);
429 virtual unsigned add_turnout(unsigned, const TrackType &);
430 virtual void remove_turnout(unsigned);
431 virtual void set_turnout(unsigned, unsigned);
432 virtual unsigned get_turnout(unsigned) const;
434 virtual unsigned add_signal(unsigned, const SignalType &);
435 virtual void remove_signal(unsigned);
436 virtual void set_signal(unsigned, unsigned);
437 virtual unsigned get_signal(unsigned) const;
440 unsigned add_accessory(Accessory::Kind, unsigned, unsigned, unsigned);
441 void remove_accessory(Accessory::Kind, unsigned);
442 void set_accessory(Accessory::Kind, unsigned, unsigned);
443 unsigned get_accessory(Accessory::Kind, unsigned) const;
444 void activate_accessory_by_mask(Accessory &, unsigned);
447 virtual unsigned add_sensor(unsigned);
448 virtual void remove_sensor(unsigned);
449 virtual void set_sensor(unsigned, bool) { }
450 virtual bool get_sensor(unsigned) const;
453 virtual void flush();
455 void save_state() const;