]> sigrok.org Git - libsigrok.git/blob - bindings/cxx/classes.cpp
bindings: Use SR_DF_ANALOG, drop SR_DF_ANALOG_OLD support.
[libsigrok.git] / bindings / cxx / classes.cpp
1 /*
2  * This file is part of the libsigrok project.
3  *
4  * Copyright (C) 2013-2014 Martin Ling <martin-sigrok@earth.li>
5  *
6  * This program is free software: you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation, either version 3 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
18  */
19
20 /* Needed for isascii(), as used in the GNU libstdc++ headers */
21 #ifndef _XOPEN_SOURCE
22 #define _XOPEN_SOURCE 600
23 #endif
24
25 #include <config.h>
26 #include <libsigrokcxx/libsigrokcxx.hpp>
27
28 #include <sstream>
29 #include <cmath>
30
31 namespace sigrok
32 {
33
34 /** Helper function to translate C errors to C++ exceptions. */
35 static void check(int result)
36 {
37         if (result != SR_OK)
38                 throw Error(result);
39 }
40
41 /** Helper function to obtain valid strings from possibly null input. */
42 static const char *valid_string(const char *input)
43 {
44         if (input != NULL)
45                 return input;
46         else
47                 return "";
48 }
49
50 /** Helper function to convert between map<string, VariantBase> and GHashTable */
51 static GHashTable *map_to_hash_variant(map<string, Glib::VariantBase> input)
52 {
53         auto output = g_hash_table_new_full(
54                 g_str_hash, g_str_equal, g_free, (GDestroyNotify) g_variant_unref);
55         for (auto entry : input)
56                 g_hash_table_insert(output,
57                         g_strdup(entry.first.c_str()),
58                         entry.second.gobj_copy());
59     return output;
60 }
61
62 Error::Error(int result) : result(result)
63 {
64 }
65
66 const char *Error::what() const throw()
67 {
68         return sr_strerror(result);
69 }
70
71 Error::~Error() throw()
72 {
73 }
74
75 ResourceReader::~ResourceReader()
76 {
77 }
78
79 SR_PRIV int ResourceReader::open_callback(struct sr_resource *res,
80                 const char *name, void *cb_data)
81 {
82         try {
83                 auto *const reader = static_cast<ResourceReader*>(cb_data);
84                 reader->open(res, name);
85         } catch (const Error &err) {
86                 return err.result;
87         } catch (...) {
88                 return SR_ERR;
89         }
90         return SR_OK;
91 }
92
93 SR_PRIV int ResourceReader::close_callback(struct sr_resource *res, void *cb_data)
94 {
95         try {
96                 auto *const reader = static_cast<ResourceReader*>(cb_data);
97                 reader->close(res);
98         } catch (const Error &err) {
99                 return err.result;
100         } catch (...) {
101                 return SR_ERR;
102         }
103         return SR_OK;
104 }
105
106 SR_PRIV ssize_t ResourceReader::read_callback(const struct sr_resource *res,
107                 void *buf, size_t count, void *cb_data)
108 {
109         try {
110                 auto *const reader = static_cast<ResourceReader*>(cb_data);
111                 return reader->read(res, buf, count);
112         } catch (const Error &err) {
113                 return err.result;
114         } catch (...) {
115                 return SR_ERR;
116         }
117 }
118
119 shared_ptr<Context> Context::create()
120 {
121         return shared_ptr<Context>(new Context(), Context::Deleter());
122 }
123
124 Context::Context() :
125         UserOwned(_structure),
126         _session(NULL)
127 {
128         check(sr_init(&_structure));
129
130         struct sr_dev_driver **driver_list = sr_driver_list(_structure);
131         if (driver_list)
132                 for (int i = 0; driver_list[i]; i++)
133                         _drivers[driver_list[i]->name] =
134                                 new Driver(driver_list[i]);
135         const struct sr_input_module **input_list = sr_input_list();
136         if (input_list)
137                 for (int i = 0; input_list[i]; i++)
138                         _input_formats[sr_input_id_get(input_list[i])] =
139                                 new InputFormat(input_list[i]);
140         const struct sr_output_module **output_list = sr_output_list();
141         if (output_list)
142                 for (int i = 0; output_list[i]; i++)
143                         _output_formats[sr_output_id_get(output_list[i])] =
144                                 new OutputFormat(output_list[i]);
145 }
146
147 string Context::package_version()
148 {
149         return sr_package_version_string_get();
150 }
151
152 string Context::lib_version()
153 {
154         return sr_lib_version_string_get();
155 }
156
157 map<string, shared_ptr<Driver>> Context::drivers()
158 {
159         map<string, shared_ptr<Driver>> result;
160         for (auto entry: _drivers)
161         {
162                 auto name = entry.first;
163                 auto driver = entry.second;
164                 result[name] = driver->get_shared_pointer(this);
165         }
166         return result;
167 }
168
169 map<string, shared_ptr<InputFormat>> Context::input_formats()
170 {
171         map<string, shared_ptr<InputFormat>> result;
172         for (auto entry: _input_formats)
173         {
174                 auto name = entry.first;
175                 auto input_format = entry.second;
176                 result[name] = input_format->get_shared_pointer(this);
177         }
178         return result;
179 }
180
181 map<string, shared_ptr<OutputFormat>> Context::output_formats()
182 {
183         map<string, shared_ptr<OutputFormat>> result;
184         for (auto entry: _output_formats)
185         {
186                 auto name = entry.first;
187                 auto output_format = entry.second;
188                 result[name] = output_format->get_shared_pointer(this);
189         }
190         return result;
191 }
192
193 Context::~Context()
194 {
195         for (auto entry : _drivers)
196                 delete entry.second;
197         for (auto entry : _input_formats)
198                 delete entry.second;
199         for (auto entry : _output_formats)
200                 delete entry.second;
201         check(sr_exit(_structure));
202 }
203
204 const LogLevel *Context::log_level()
205 {
206         return LogLevel::get(sr_log_loglevel_get());
207 }
208
209 void Context::set_log_level(const LogLevel *level)
210 {
211         check(sr_log_loglevel_set(level->id()));
212 }
213
214 static int call_log_callback(void *cb_data, int loglevel, const char *format, va_list args)
215 {
216         va_list args_copy;
217         va_copy(args_copy, args);
218         int length = vsnprintf(NULL, 0, format, args_copy);
219         va_end(args_copy);
220         char *buf = (char *) g_malloc(length + 1);
221         vsprintf(buf, format, args);
222         string message(buf, length);
223         g_free(buf);
224
225         LogCallbackFunction callback = *((LogCallbackFunction *) cb_data);
226
227         try
228         {
229                 callback(LogLevel::get(loglevel), message);
230         }
231         catch (Error e)
232         {
233                 return e.result;
234         }
235
236         return SR_OK;
237 }
238
239 void Context::set_log_callback(LogCallbackFunction callback)
240 {
241         _log_callback = callback;
242         check(sr_log_callback_set(call_log_callback, &_log_callback));
243 }
244
245 void Context::set_log_callback_default()
246 {
247         check(sr_log_callback_set_default());
248         _log_callback = nullptr;
249 }
250
251 void Context::set_resource_reader(ResourceReader *reader)
252 {
253         if (reader) {
254                 check(sr_resource_set_hooks(_structure,
255                                 &ResourceReader::open_callback,
256                                 &ResourceReader::close_callback,
257                                 &ResourceReader::read_callback, reader));
258         } else {
259                 check(sr_resource_set_hooks(_structure,
260                                 nullptr, nullptr, nullptr, nullptr));
261         }
262 }
263
264 shared_ptr<Session> Context::create_session()
265 {
266         return shared_ptr<Session>(
267                 new Session(shared_from_this()), Session::Deleter());
268 }
269
270 shared_ptr<UserDevice> Context::create_user_device(
271                 string vendor, string model, string version)
272 {
273         return shared_ptr<UserDevice>(
274                 new UserDevice(vendor, model, version), UserDevice::Deleter());
275 }
276
277 shared_ptr<Packet> Context::create_header_packet(Glib::TimeVal start_time)
278 {
279         auto header = g_new(struct sr_datafeed_header, 1);
280         header->feed_version = 1;
281         header->starttime.tv_sec = start_time.tv_sec;
282         header->starttime.tv_usec = start_time.tv_usec;
283         auto packet = g_new(struct sr_datafeed_packet, 1);
284         packet->type = SR_DF_HEADER;
285         packet->payload = header;
286         return shared_ptr<Packet>(new Packet(nullptr, packet), Packet::Deleter());
287 }
288
289 shared_ptr<Packet> Context::create_meta_packet(
290         map<const ConfigKey *, Glib::VariantBase> config)
291 {
292         auto meta = g_new0(struct sr_datafeed_meta, 1);
293         for (auto input : config)
294         {
295                 auto key = input.first;
296                 auto value = input.second;
297                 auto output = g_new(struct sr_config, 1);
298                 output->key = key->id();
299                 output->data = value.gobj();
300                 g_variant_ref(output->data);
301                 meta->config = g_slist_append(meta->config, output);
302         }
303         auto packet = g_new(struct sr_datafeed_packet, 1);
304         packet->type = SR_DF_META;
305         packet->payload = meta;
306         return shared_ptr<Packet>(new Packet(nullptr, packet), Packet::Deleter());
307 }
308
309 shared_ptr<Packet> Context::create_logic_packet(
310         void *data_pointer, size_t data_length, unsigned int unit_size)
311 {
312         auto logic = g_new(struct sr_datafeed_logic, 1);
313         logic->length = data_length;
314         logic->unitsize = unit_size;
315         logic->data = data_pointer;
316         auto packet = g_new(struct sr_datafeed_packet, 1);
317         packet->type = SR_DF_LOGIC;
318         packet->payload = logic;
319         return shared_ptr<Packet>(new Packet(nullptr, packet), Packet::Deleter());
320 }
321
322 shared_ptr<Packet> Context::create_analog_packet(
323         vector<shared_ptr<Channel> > channels,
324         float *data_pointer, unsigned int num_samples, const Quantity *mq,
325         const Unit *unit, vector<const QuantityFlag *> mqflags)
326 {
327         auto analog = g_new0(struct sr_datafeed_analog, 1);
328         auto meaning = g_new0(struct sr_analog_meaning, 1);
329
330         analog->meaning = meaning;
331
332         for (auto channel : channels)
333                 meaning->channels = g_slist_append(meaning->channels, channel->_structure);
334         analog->num_samples = num_samples;
335         meaning->mq = (sr_mq)mq->id();
336         meaning->unit = (sr_unit)unit->id();
337         meaning->mqflags = (sr_mqflag)QuantityFlag::mask_from_flags(mqflags);
338         analog->data = data_pointer;
339         auto packet = g_new(struct sr_datafeed_packet, 1);
340         packet->type = SR_DF_ANALOG;
341         packet->payload = analog;
342         return shared_ptr<Packet>(new Packet(nullptr, packet), Packet::Deleter());
343 }
344
345 shared_ptr<Session> Context::load_session(string filename)
346 {
347         return shared_ptr<Session>(
348                 new Session(shared_from_this(), filename), Session::Deleter());
349 }
350
351 shared_ptr<Trigger> Context::create_trigger(string name)
352 {
353         return shared_ptr<Trigger>(
354                 new Trigger(shared_from_this(), name), Trigger::Deleter());
355 }
356
357 shared_ptr<Input> Context::open_file(string filename)
358 {
359         const struct sr_input *input;
360
361         check(sr_input_scan_file(filename.c_str(), &input));
362         return shared_ptr<Input>(
363                 new Input(shared_from_this(), input), Input::Deleter());
364 }
365
366 shared_ptr<Input> Context::open_stream(string header)
367 {
368         const struct sr_input *input;
369
370         auto gstr = g_string_new(header.c_str());
371         auto ret = sr_input_scan_buffer(gstr, &input);
372         g_string_free(gstr, true);
373         check(ret);
374         return shared_ptr<Input>(
375                 new Input(shared_from_this(), input), Input::Deleter());
376 }
377
378 map<string, string> Context::serials(shared_ptr<Driver> driver)
379 {
380         GSList *serial_list = sr_serial_list(driver ? driver->_structure : NULL);
381         map<string, string> serials;
382
383         for (GSList *serial = serial_list; serial; serial = serial->next) {
384                 struct sr_serial_port *port = (sr_serial_port *) serial->data;
385                 serials[string(port->name)] = string(port->description);
386         }
387
388         g_slist_free_full(serial_list, (GDestroyNotify)sr_serial_free);
389         return serials;
390 }
391
392 Driver::Driver(struct sr_dev_driver *structure) :
393         ParentOwned(structure),
394         Configurable(structure, NULL, NULL),
395         _initialized(false)
396 {
397 }
398
399 Driver::~Driver()
400 {
401 }
402
403 string Driver::name()
404 {
405         return valid_string(_structure->name);
406 }
407
408 string Driver::long_name()
409 {
410         return valid_string(_structure->longname);
411 }
412
413 vector<shared_ptr<HardwareDevice>> Driver::scan(
414         map<const ConfigKey *, Glib::VariantBase> options)
415 {
416         /* Initialise the driver if not yet done. */
417         if (!_initialized)
418         {
419                 check(sr_driver_init(_parent->_structure, _structure));
420                 _initialized = true;
421         }
422
423         /* Translate scan options to GSList of struct sr_config pointers. */
424         GSList *option_list = NULL;
425         for (auto entry : options)
426         {
427                 auto key = entry.first;
428                 auto value = entry.second;
429                 auto config = g_new(struct sr_config, 1);
430                 config->key = key->id();
431                 config->data = value.gobj();
432                 option_list = g_slist_append(option_list, config);
433         }
434
435         /* Run scan. */
436         GSList *device_list = sr_driver_scan(_structure, option_list);
437
438         /* Free option list. */
439         g_slist_free_full(option_list, g_free);
440
441
442         /* Create device objects. */
443         vector<shared_ptr<HardwareDevice>> result;
444         for (GSList *device = device_list; device; device = device->next)
445         {
446                 auto sdi = (struct sr_dev_inst *) device->data;
447                 result.push_back(shared_ptr<HardwareDevice>(
448                         new HardwareDevice(shared_from_this(), sdi),
449                         HardwareDevice::Deleter()));
450         }
451
452         /* Free GSList returned from scan. */
453         g_slist_free(device_list);
454
455         return result;
456 }
457
458 Configurable::Configurable(
459                 struct sr_dev_driver *driver,
460                 struct sr_dev_inst *sdi,
461                 struct sr_channel_group *cg) :
462         config_driver(driver),
463         config_sdi(sdi),
464         config_channel_group(cg)
465 {
466 }
467
468 Configurable::~Configurable()
469 {
470 }
471
472 Glib::VariantBase Configurable::config_get(const ConfigKey *key)
473 {
474         GVariant *data;
475         check(sr_config_get(
476                 config_driver, config_sdi, config_channel_group,
477                 key->id(), &data));
478         return Glib::VariantBase(data);
479 }
480
481 void Configurable::config_set(const ConfigKey *key, Glib::VariantBase value)
482 {
483         check(sr_config_set(
484                 config_sdi, config_channel_group,
485                 key->id(), value.gobj()));
486 }
487
488 Glib::VariantContainerBase Configurable::config_list(const ConfigKey *key)
489 {
490         GVariant *data;
491         check(sr_config_list(
492                 config_driver, config_sdi, config_channel_group,
493                 key->id(), &data));
494         return Glib::VariantContainerBase(data);
495 }
496
497 map<const ConfigKey *, set<Capability>> Configurable::config_keys(const ConfigKey *key)
498 {
499         GVariant *gvar_opts;
500         gsize num_opts;
501         const uint32_t *opts;
502         map<const ConfigKey *, set<Capability>> result;
503
504         check(sr_config_list(
505                 config_driver, config_sdi, config_channel_group,
506                 key->id(), &gvar_opts));
507
508         opts = (const uint32_t *) g_variant_get_fixed_array(
509                 gvar_opts, &num_opts, sizeof(uint32_t));
510
511         for (gsize i = 0; i < num_opts; i++)
512         {
513                 auto key = ConfigKey::get(opts[i] & SR_CONF_MASK);
514                 set<Capability> capabilities;
515                 if (opts[i] & SR_CONF_GET)
516                         capabilities.insert(GET);
517                 if (opts[i] & SR_CONF_SET)
518                         capabilities.insert(SET);
519                 if (opts[i] & SR_CONF_LIST)
520                         capabilities.insert(LIST);
521                 result[key] = capabilities;
522         }
523
524         g_variant_unref(gvar_opts);
525
526         return result;
527 }
528
529 bool Configurable::config_check(const ConfigKey *key,
530         const ConfigKey *index_key)
531 {
532         GVariant *gvar_opts;
533         gsize num_opts;
534         const uint32_t *opts;
535
536         if (sr_config_list(config_driver, config_sdi, config_channel_group,
537                         index_key->id(), &gvar_opts) != SR_OK)
538                 return false;
539
540         opts = (const uint32_t *) g_variant_get_fixed_array(
541                 gvar_opts, &num_opts, sizeof(uint32_t));
542
543         for (gsize i = 0; i < num_opts; i++)
544         {
545                 if ((opts[i] & SR_CONF_MASK) == (uint32_t) key->id())
546                 {
547                         g_variant_unref(gvar_opts);
548                         return true;
549                 }
550         }
551
552         g_variant_unref(gvar_opts);
553
554         return false;
555 }
556
557 Device::Device(struct sr_dev_inst *structure) :
558         Configurable(sr_dev_inst_driver_get(structure), structure, NULL),
559         _structure(structure)
560 {
561         for (GSList *entry = sr_dev_inst_channels_get(structure); entry; entry = entry->next)
562         {
563                 auto channel = (struct sr_channel *) entry->data;
564                 _channels[channel] = new Channel(channel);
565         }
566
567         for (GSList *entry = sr_dev_inst_channel_groups_get(structure); entry; entry = entry->next)
568         {
569                 auto group = (struct sr_channel_group *) entry->data;
570                 _channel_groups[group->name] = new ChannelGroup(this, group);
571         }
572 }
573
574 Device::~Device()
575 {
576         for (auto entry : _channels)
577                 delete entry.second;
578         for (auto entry : _channel_groups)
579                 delete entry.second;
580 }
581
582 string Device::vendor()
583 {
584         return valid_string(sr_dev_inst_vendor_get(_structure));
585 }
586
587 string Device::model()
588 {
589         return valid_string(sr_dev_inst_model_get(_structure));
590 }
591
592 string Device::version()
593 {
594         return valid_string(sr_dev_inst_version_get(_structure));
595 }
596
597 string Device::serial_number()
598 {
599         return valid_string(sr_dev_inst_sernum_get(_structure));
600 }
601
602 string Device::connection_id()
603 {
604         return valid_string(sr_dev_inst_connid_get(_structure));
605 }
606
607 vector<shared_ptr<Channel>> Device::channels()
608 {
609         vector<shared_ptr<Channel>> result;
610         for (auto channel = sr_dev_inst_channels_get(_structure); channel; channel = channel->next)
611                 result.push_back(
612                         _channels[(struct sr_channel *) channel->data]->get_shared_pointer(
613                                 get_shared_from_this()));
614         return result;
615 }
616
617 shared_ptr<Channel> Device::get_channel(struct sr_channel *ptr)
618 {
619         return _channels[ptr]->get_shared_pointer(get_shared_from_this());
620 }
621
622 map<string, shared_ptr<ChannelGroup>>
623 Device::channel_groups()
624 {
625         map<string, shared_ptr<ChannelGroup>> result;
626         for (auto entry: _channel_groups)
627         {
628                 auto name = entry.first;
629                 auto channel_group = entry.second;
630                 result[name] = channel_group->get_shared_pointer(get_shared_from_this());
631         }
632         return result;
633 }
634
635 void Device::open()
636 {
637         check(sr_dev_open(_structure));
638 }
639
640 void Device::close()
641 {
642         check(sr_dev_close(_structure));
643 }
644
645 HardwareDevice::HardwareDevice(shared_ptr<Driver> driver,
646                 struct sr_dev_inst *structure) :
647         UserOwned(structure),
648         Device(structure),
649         _driver(driver)
650 {
651 }
652
653 HardwareDevice::~HardwareDevice()
654 {
655 }
656
657 shared_ptr<Device> HardwareDevice::get_shared_from_this()
658 {
659         return static_pointer_cast<Device>(shared_from_this());
660 }
661
662 shared_ptr<Driver> HardwareDevice::driver()
663 {
664         return _driver;
665 }
666
667 UserDevice::UserDevice(string vendor, string model, string version) :
668         UserOwned(sr_dev_inst_user_new(
669                 vendor.c_str(), model.c_str(), version.c_str())),
670         Device(UserOwned::_structure)
671 {
672 }
673
674 UserDevice::~UserDevice()
675 {
676 }
677
678 shared_ptr<Device> UserDevice::get_shared_from_this()
679 {
680         return static_pointer_cast<Device>(shared_from_this());
681 }
682
683 shared_ptr<Channel> UserDevice::add_channel(unsigned int index,
684         const ChannelType *type, string name)
685 {
686         check(sr_dev_inst_channel_add(Device::_structure,
687                 index, type->id(), name.c_str()));
688         struct sr_channel *structure = (struct sr_channel *)
689                         g_slist_last(sr_dev_inst_channels_get(Device::_structure))->data;
690         Channel *channel = new Channel(structure);
691         _channels[structure] = channel;
692         return get_channel(structure);
693 }
694
695 Channel::Channel(struct sr_channel *structure) :
696         ParentOwned(structure),
697         _type(ChannelType::get(_structure->type))
698 {
699 }
700
701 Channel::~Channel()
702 {
703 }
704
705 string Channel::name()
706 {
707         return valid_string(_structure->name);
708 }
709
710 void Channel::set_name(string name)
711 {
712         check(sr_dev_channel_name_set(_structure, name.c_str()));
713 }
714
715 const ChannelType *Channel::type()
716 {
717         return ChannelType::get(_structure->type);
718 }
719
720 bool Channel::enabled()
721 {
722         return _structure->enabled;
723 }
724
725 void Channel::set_enabled(bool value)
726 {
727         check(sr_dev_channel_enable(_structure, value));
728 }
729
730 unsigned int Channel::index()
731 {
732         return _structure->index;
733 }
734
735 ChannelGroup::ChannelGroup(Device *device,
736                 struct sr_channel_group *structure) :
737         ParentOwned(structure),
738         Configurable(sr_dev_inst_driver_get(device->_structure), device->_structure, structure)
739 {
740         for (GSList *entry = structure->channels; entry; entry = entry->next)
741                 _channels.push_back(device->_channels[(struct sr_channel *)entry->data]);
742 }
743
744 ChannelGroup::~ChannelGroup()
745 {
746 }
747
748 string ChannelGroup::name()
749 {
750         return valid_string(_structure->name);
751 }
752
753 vector<shared_ptr<Channel>> ChannelGroup::channels()
754 {
755         vector<shared_ptr<Channel>> result;
756         for (auto channel : _channels)
757                 result.push_back(channel->get_shared_pointer(_parent));
758         return result;
759 }
760
761 Trigger::Trigger(shared_ptr<Context> context, string name) : 
762         UserOwned(sr_trigger_new(name.c_str())),
763         _context(context)
764 {
765         for (auto stage = _structure->stages; stage; stage = stage->next)
766                 _stages.push_back(
767                         new TriggerStage((struct sr_trigger_stage *) stage->data));
768 }
769
770 Trigger::~Trigger()
771 {
772         for (auto stage: _stages)
773                 delete stage;
774
775         sr_trigger_free(_structure);
776 }
777
778 string Trigger::name()
779 {
780         return _structure->name;
781 }
782
783 vector<shared_ptr<TriggerStage>> Trigger::stages()
784 {
785         vector<shared_ptr<TriggerStage>> result;
786         for (auto stage : _stages)
787                 result.push_back(stage->get_shared_pointer(this));
788         return result;
789 }
790
791 shared_ptr<TriggerStage> Trigger::add_stage()
792 {
793         auto stage = new TriggerStage(sr_trigger_stage_add(_structure));
794         _stages.push_back(stage);
795         return stage->get_shared_pointer(this);
796 }
797
798 TriggerStage::TriggerStage(struct sr_trigger_stage *structure) : 
799         ParentOwned(structure)
800 {
801 }
802
803 TriggerStage::~TriggerStage()
804 {
805         for (auto match : _matches)
806                 delete match;
807 }
808         
809 int TriggerStage::number()
810 {
811         return _structure->stage;
812 }
813
814 vector<shared_ptr<TriggerMatch>> TriggerStage::matches()
815 {
816         vector<shared_ptr<TriggerMatch>> result;
817         for (auto match : _matches)
818                 result.push_back(match->get_shared_pointer(this));
819         return result;
820 }
821
822 void TriggerStage::add_match(shared_ptr<Channel> channel,
823         const TriggerMatchType *type, float value)
824 {
825         check(sr_trigger_match_add(_structure,
826                 channel->_structure, type->id(), value));
827         _matches.push_back(new TriggerMatch(
828                 (struct sr_trigger_match *) g_slist_last(
829                         _structure->matches)->data, channel));
830 }
831
832 void TriggerStage::add_match(shared_ptr<Channel> channel,
833         const TriggerMatchType *type)
834 {
835         add_match(channel, type, NAN);
836 }
837
838 TriggerMatch::TriggerMatch(struct sr_trigger_match *structure,
839                 shared_ptr<Channel> channel) :
840         ParentOwned(structure),
841         _channel(channel)
842 {
843 }
844
845 TriggerMatch::~TriggerMatch()
846 {
847 }
848
849 shared_ptr<Channel> TriggerMatch::channel()
850 {
851         return _channel;
852 }
853
854 const TriggerMatchType *TriggerMatch::type()
855 {
856         return TriggerMatchType::get(_structure->match);
857 }
858
859 float TriggerMatch::value()
860 {
861         return _structure->value;
862 }
863
864 DatafeedCallbackData::DatafeedCallbackData(Session *session,
865                 DatafeedCallbackFunction callback) :
866         _callback(callback),
867         _session(session)
868 {
869 }
870
871 void DatafeedCallbackData::run(const struct sr_dev_inst *sdi,
872         const struct sr_datafeed_packet *pkt)
873 {
874         auto device = _session->get_device(sdi);
875         auto packet = shared_ptr<Packet>(new Packet(device, pkt), Packet::Deleter());
876         _callback(device, packet);
877 }
878
879 SessionDevice::SessionDevice(struct sr_dev_inst *structure) :
880         ParentOwned(structure),
881         Device(structure)
882 {
883 }
884
885 SessionDevice::~SessionDevice()
886 {
887 }
888
889 shared_ptr<Device> SessionDevice::get_shared_from_this()
890 {
891         return static_pointer_cast<Device>(shared_from_this());
892 }
893
894 Session::Session(shared_ptr<Context> context) :
895         UserOwned(_structure),
896         _context(context)
897 {
898         check(sr_session_new(context->_structure, &_structure));
899         _context->_session = this;
900 }
901
902 Session::Session(shared_ptr<Context> context, string filename) :
903         UserOwned(_structure),
904         _context(context),
905         _filename(filename)
906 {
907         check(sr_session_load(context->_structure, filename.c_str(), &_structure));
908         GSList *dev_list;
909         check(sr_session_dev_list(_structure, &dev_list));
910         for (GSList *dev = dev_list; dev; dev = dev->next)
911         {
912                 auto sdi = (struct sr_dev_inst *) dev->data;
913                 _owned_devices[sdi] = new SessionDevice(sdi);
914         }
915         _context->_session = this;
916 }
917
918 Session::~Session()
919 {
920         check(sr_session_destroy(_structure));
921
922         for (auto callback : _datafeed_callbacks)
923                 delete callback;
924
925         for (auto entry : _owned_devices)
926                 delete entry.second;
927 }
928
929 shared_ptr<Device> Session::get_device(const struct sr_dev_inst *sdi)
930 {
931         if (_owned_devices.count(sdi))
932                 return static_pointer_cast<Device>(
933                         _owned_devices[sdi]->get_shared_pointer(this));
934         else if (_other_devices.count(sdi))
935                 return _other_devices[sdi];
936         else
937                 throw Error(SR_ERR_BUG);
938 }
939
940 void Session::add_device(shared_ptr<Device> device)
941 {
942         check(sr_session_dev_add(_structure, device->_structure));
943         _other_devices[device->_structure] = device;
944 }
945
946 vector<shared_ptr<Device>> Session::devices()
947 {
948         GSList *dev_list;
949         check(sr_session_dev_list(_structure, &dev_list));
950         vector<shared_ptr<Device>> result;
951         for (GSList *dev = dev_list; dev; dev = dev->next)
952         {
953                 auto sdi = (struct sr_dev_inst *) dev->data;
954                 result.push_back(get_device(sdi));
955         }
956         return result;
957 }
958
959 void Session::remove_devices()
960 {
961         _other_devices.clear();
962         check(sr_session_dev_remove_all(_structure));
963 }
964
965 void Session::start()
966 {
967         check(sr_session_start(_structure));
968 }
969
970 void Session::run()
971 {
972         check(sr_session_run(_structure));
973 }
974
975 void Session::stop()
976 {
977         check(sr_session_stop(_structure));
978 }
979
980 bool Session::is_running() const
981 {
982         const int ret = sr_session_is_running(_structure);
983         if (ret < 0)
984                 throw Error{ret};
985         return (ret != 0);
986 }
987
988 static void session_stopped_callback(void *data)
989 {
990         auto *const callback = static_cast<SessionStoppedCallback*>(data);
991         (*callback)();
992 }
993
994 void Session::set_stopped_callback(SessionStoppedCallback callback)
995 {
996         _stopped_callback = move(callback);
997         if (_stopped_callback)
998                 check(sr_session_stopped_callback_set(_structure,
999                                 &session_stopped_callback, &_stopped_callback));
1000         else
1001                 check(sr_session_stopped_callback_set(_structure,
1002                                 nullptr, nullptr));
1003 }
1004
1005 static void datafeed_callback(const struct sr_dev_inst *sdi,
1006         const struct sr_datafeed_packet *pkt, void *cb_data)
1007 {
1008         auto callback = static_cast<DatafeedCallbackData *>(cb_data);
1009         callback->run(sdi, pkt);
1010 }
1011         
1012 void Session::add_datafeed_callback(DatafeedCallbackFunction callback)
1013 {
1014         auto cb_data = new DatafeedCallbackData(this, callback);
1015         check(sr_session_datafeed_callback_add(_structure,
1016                 datafeed_callback, cb_data));
1017         _datafeed_callbacks.push_back(cb_data);
1018 }
1019
1020 void Session::remove_datafeed_callbacks(void)
1021 {
1022         check(sr_session_datafeed_callback_remove_all(_structure));
1023         for (auto callback : _datafeed_callbacks)
1024                 delete callback;
1025         _datafeed_callbacks.clear();
1026 }
1027
1028 shared_ptr<Trigger> Session::trigger()
1029 {
1030         return _trigger;
1031 }
1032
1033 void Session::set_trigger(shared_ptr<Trigger> trigger)
1034 {
1035         if (!trigger)
1036                 // Set NULL trigger, i.e. remove any trigger from the session.
1037                 check(sr_session_trigger_set(_structure, NULL));
1038         else
1039                 check(sr_session_trigger_set(_structure, trigger->_structure));
1040         _trigger = trigger;
1041 }
1042
1043 string Session::filename()
1044 {
1045         return _filename;
1046 }
1047
1048 shared_ptr<Context> Session::context()
1049 {
1050         return _context;
1051 }
1052
1053 Packet::Packet(shared_ptr<Device> device,
1054         const struct sr_datafeed_packet *structure) :
1055         UserOwned(structure),
1056         _device(device)
1057 {
1058         switch (structure->type)
1059         {
1060                 case SR_DF_HEADER:
1061                         _payload = new Header(
1062                                 static_cast<const struct sr_datafeed_header *>(
1063                                         structure->payload));
1064                         break;
1065                 case SR_DF_META:
1066                         _payload = new Meta(
1067                                 static_cast<const struct sr_datafeed_meta *>(
1068                                         structure->payload));
1069                         break;
1070                 case SR_DF_LOGIC:
1071                         _payload = new Logic(
1072                                 static_cast<const struct sr_datafeed_logic *>(
1073                                         structure->payload));
1074                         break;
1075                 case SR_DF_ANALOG:
1076                         _payload = new Analog(
1077                                 static_cast<const struct sr_datafeed_analog *>(
1078                                         structure->payload));
1079                         break;
1080                 default:
1081                         _payload = nullptr;
1082                         break;
1083         }
1084 }
1085
1086 Packet::~Packet()
1087 {
1088         if (_payload)
1089                 delete _payload;
1090 }
1091
1092 const PacketType *Packet::type()
1093 {
1094         return PacketType::get(_structure->type);
1095 }
1096
1097 shared_ptr<PacketPayload> Packet::payload()
1098 {
1099         if (_payload)
1100                 return _payload->get_shared_pointer(this);
1101         else
1102                 throw Error(SR_ERR_NA);
1103 }
1104
1105 PacketPayload::PacketPayload()
1106 {
1107 }
1108
1109 PacketPayload::~PacketPayload()
1110 {
1111 }
1112
1113 Header::Header(const struct sr_datafeed_header *structure) :
1114         ParentOwned(structure),
1115         PacketPayload()
1116 {
1117 }
1118
1119 Header::~Header()
1120 {
1121 }
1122
1123 shared_ptr<PacketPayload> Header::get_shared_pointer(Packet *_parent)
1124 {
1125         return static_pointer_cast<PacketPayload>(
1126                 ParentOwned::get_shared_pointer(_parent));
1127 }
1128
1129 int Header::feed_version()
1130 {
1131         return _structure->feed_version;
1132 }
1133
1134 Glib::TimeVal Header::start_time()
1135 {
1136         return Glib::TimeVal(
1137                 _structure->starttime.tv_sec,
1138                 _structure->starttime.tv_usec);
1139 }
1140
1141 Meta::Meta(const struct sr_datafeed_meta *structure) :
1142         ParentOwned(structure),
1143         PacketPayload()
1144 {
1145 }
1146
1147 Meta::~Meta()
1148 {
1149 }
1150
1151 shared_ptr<PacketPayload> Meta::get_shared_pointer(Packet *_parent)
1152 {
1153         return static_pointer_cast<PacketPayload>(
1154                 ParentOwned::get_shared_pointer(_parent));
1155 }
1156
1157 map<const ConfigKey *, Glib::VariantBase> Meta::config()
1158 {
1159         map<const ConfigKey *, Glib::VariantBase> result;
1160         for (auto l = _structure->config; l; l = l->next)
1161         {
1162                 auto config = (struct sr_config *) l->data;
1163                 result[ConfigKey::get(config->key)] = Glib::VariantBase(config->data);
1164         }
1165         return result;
1166 }
1167
1168 Logic::Logic(const struct sr_datafeed_logic *structure) :
1169         ParentOwned(structure),
1170         PacketPayload()
1171 {
1172 }
1173
1174 Logic::~Logic()
1175 {
1176 }
1177
1178 shared_ptr<PacketPayload> Logic::get_shared_pointer(Packet *_parent)
1179 {
1180         return static_pointer_cast<PacketPayload>(
1181                 ParentOwned::get_shared_pointer(_parent));
1182 }
1183
1184 void *Logic::data_pointer()
1185 {
1186         return _structure->data;
1187 }
1188
1189 size_t Logic::data_length()
1190 {
1191         return _structure->length;
1192 }
1193
1194 unsigned int Logic::unit_size()
1195 {
1196         return _structure->unitsize;
1197 }
1198
1199 Analog::Analog(const struct sr_datafeed_analog *structure) :
1200         ParentOwned(structure),
1201         PacketPayload()
1202 {
1203 }
1204
1205 Analog::~Analog()
1206 {
1207 }
1208
1209 shared_ptr<PacketPayload> Analog::get_shared_pointer(Packet *_parent)
1210 {
1211         return static_pointer_cast<PacketPayload>(
1212                 ParentOwned::get_shared_pointer(_parent));
1213 }
1214
1215 void *Analog::data_pointer()
1216 {
1217         return _structure->data;
1218 }
1219
1220 unsigned int Analog::num_samples()
1221 {
1222         return _structure->num_samples;
1223 }
1224
1225 vector<shared_ptr<Channel>> Analog::channels()
1226 {
1227         vector<shared_ptr<Channel>> result;
1228         for (auto l = _structure->meaning->channels; l; l = l->next)
1229                 result.push_back(_parent->_device->get_channel(
1230                         (struct sr_channel *)l->data));
1231         return result;
1232 }
1233
1234 const Quantity *Analog::mq()
1235 {
1236         return Quantity::get(_structure->meaning->mq);
1237 }
1238
1239 const Unit *Analog::unit()
1240 {
1241         return Unit::get(_structure->meaning->unit);
1242 }
1243
1244 vector<const QuantityFlag *> Analog::mq_flags()
1245 {
1246         return QuantityFlag::flags_from_mask(_structure->meaning->mqflags);
1247 }
1248
1249 InputFormat::InputFormat(const struct sr_input_module *structure) :
1250         ParentOwned(structure)
1251 {
1252 }
1253
1254 InputFormat::~InputFormat()
1255 {
1256 }
1257
1258 string InputFormat::name()
1259 {
1260         return valid_string(sr_input_id_get(_structure));
1261 }
1262
1263 string InputFormat::description()
1264 {
1265         return valid_string(sr_input_description_get(_structure));
1266 }
1267
1268 vector<string> InputFormat::extensions()
1269 {
1270         vector<string> exts;
1271         for (const char *const *e = sr_input_extensions_get(_structure);
1272                 e && *e; e++)
1273                 exts.push_back(*e);
1274         return exts;
1275 }
1276
1277 map<string, shared_ptr<Option>> InputFormat::options()
1278 {
1279         const struct sr_option **options = sr_input_options_get(_structure);
1280         map<string, shared_ptr<Option>> result;
1281         if (options)
1282         {
1283                 auto option_array = shared_ptr<const struct sr_option *>(
1284                         options, sr_input_options_free);
1285                 for (int i = 0; options[i]; i++)
1286                         result[options[i]->id] = shared_ptr<Option>(
1287                                 new Option(options[i], option_array), Option::Deleter());
1288         }
1289         return result;
1290 }
1291
1292 shared_ptr<Input> InputFormat::create_input(
1293         map<string, Glib::VariantBase> options)
1294 {
1295         auto input = sr_input_new(_structure, map_to_hash_variant(options));
1296         if (!input)
1297                 throw Error(SR_ERR_ARG);
1298         return shared_ptr<Input>(
1299                 new Input(_parent->shared_from_this(), input), Input::Deleter());
1300 }
1301
1302 Input::Input(shared_ptr<Context> context, const struct sr_input *structure) :
1303         UserOwned(structure),
1304         _context(context),
1305         _device(nullptr)
1306 {
1307 }
1308
1309 shared_ptr<InputDevice> Input::device()
1310 {
1311         if (!_device)
1312         {
1313                 auto sdi = sr_input_dev_inst_get(_structure);
1314                 if (!sdi)
1315                         throw Error(SR_ERR_NA);
1316                 _device = new InputDevice(shared_from_this(), sdi);
1317         }
1318
1319         return _device->get_shared_pointer(shared_from_this());
1320 }
1321
1322 void Input::send(void *data, size_t length)
1323 {
1324         auto gstr = g_string_new_len((gchar *)data, length);
1325         auto ret = sr_input_send(_structure, gstr);
1326         g_string_free(gstr, false);
1327         check(ret);
1328 }
1329
1330 void Input::end()
1331 {
1332         check(sr_input_end(_structure));
1333 }
1334
1335 Input::~Input()
1336 {
1337         if (_device)
1338                 delete _device;
1339         sr_input_free(_structure);
1340 }
1341
1342 InputDevice::InputDevice(shared_ptr<Input> input,
1343                 struct sr_dev_inst *structure) :
1344         ParentOwned(structure),
1345         Device(structure),
1346         _input(input)
1347 {
1348 }
1349
1350 InputDevice::~InputDevice()
1351 {
1352 }
1353
1354 shared_ptr<Device> InputDevice::get_shared_from_this()
1355 {
1356         return static_pointer_cast<Device>(shared_from_this());
1357 }
1358
1359 Option::Option(const struct sr_option *structure,
1360                 shared_ptr<const struct sr_option *> structure_array) :
1361         UserOwned(structure),
1362         _structure_array(structure_array)
1363 {
1364 }
1365
1366 Option::~Option()
1367 {
1368 }
1369
1370 string Option::id()
1371 {
1372         return valid_string(_structure->id);
1373 }
1374
1375 string Option::name()
1376 {
1377         return valid_string(_structure->name);
1378 }
1379
1380 string Option::description()
1381 {
1382         return valid_string(_structure->desc);
1383 }
1384
1385 Glib::VariantBase Option::default_value()
1386 {
1387         return Glib::VariantBase(_structure->def, true);
1388 }
1389
1390 vector<Glib::VariantBase> Option::values()
1391 {
1392         vector<Glib::VariantBase> result;
1393         for (auto l = _structure->values; l; l = l->next)
1394                 result.push_back(Glib::VariantBase((GVariant *) l->data, true));
1395         return result;
1396 }
1397
1398 OutputFormat::OutputFormat(const struct sr_output_module *structure) :
1399         ParentOwned(structure)
1400 {
1401 }
1402
1403 OutputFormat::~OutputFormat()
1404 {
1405 }
1406
1407 string OutputFormat::name()
1408 {
1409         return valid_string(sr_output_id_get(_structure));
1410 }
1411
1412 string OutputFormat::description()
1413 {
1414         return valid_string(sr_output_description_get(_structure));
1415 }
1416
1417 vector<string> OutputFormat::extensions()
1418 {
1419         vector<string> exts;
1420         for (const char *const *e = sr_output_extensions_get(_structure);
1421                 e && *e; e++)
1422                 exts.push_back(*e);
1423         return exts;
1424 }
1425
1426 map<string, shared_ptr<Option>> OutputFormat::options()
1427 {
1428         const struct sr_option **options = sr_output_options_get(_structure);
1429         map<string, shared_ptr<Option>> result;
1430         if (options)
1431         {
1432                 auto option_array = shared_ptr<const struct sr_option *>(
1433                         options, sr_output_options_free);
1434                 for (int i = 0; options[i]; i++)
1435                         result[options[i]->id] = shared_ptr<Option>(
1436                                 new Option(options[i], option_array), Option::Deleter());
1437         }
1438         return result;
1439 }
1440
1441 shared_ptr<Output> OutputFormat::create_output(
1442         shared_ptr<Device> device, map<string, Glib::VariantBase> options)
1443 {
1444         return shared_ptr<Output>(
1445                 new Output(shared_from_this(), device, options),
1446                 Output::Deleter());
1447 }
1448
1449 shared_ptr<Output> OutputFormat::create_output(string filename,
1450         shared_ptr<Device> device, map<string, Glib::VariantBase> options)
1451 {
1452         return shared_ptr<Output>(
1453                 new Output(filename, shared_from_this(), device, options),
1454                 Output::Deleter());
1455 }
1456
1457 bool OutputFormat::test_flag(const OutputFlag *flag)
1458 {
1459         return sr_output_test_flag(_structure, flag->id());
1460 }
1461
1462 Output::Output(shared_ptr<OutputFormat> format,
1463                 shared_ptr<Device> device, map<string, Glib::VariantBase> options) :
1464         UserOwned(sr_output_new(format->_structure,
1465                 map_to_hash_variant(options), device->_structure, NULL)),
1466         _format(format),
1467         _device(device),
1468         _options(options)
1469 {
1470 }
1471
1472 Output::Output(string filename, shared_ptr<OutputFormat> format,
1473                 shared_ptr<Device> device, map<string, Glib::VariantBase> options) :
1474         UserOwned(sr_output_new(format->_structure,
1475                 map_to_hash_variant(options), device->_structure, filename.c_str())),
1476         _format(format),
1477         _device(device),
1478         _options(options)
1479 {
1480 }
1481
1482 Output::~Output()
1483 {
1484         check(sr_output_free(_structure));
1485 }
1486
1487 string Output::receive(shared_ptr<Packet> packet)
1488 {
1489         GString *out;
1490         check(sr_output_send(_structure, packet->_structure, &out));
1491         if (out)
1492         {
1493                 auto result = string(out->str, out->str + out->len);
1494                 g_string_free(out, true);
1495                 return result;
1496         }
1497         else
1498         {
1499                 return string();
1500         }
1501 }
1502
1503 #include <enums.cpp>
1504
1505 }