]> sigrok.org Git - libsigrok.git/blob - bindings/cxx/classes.cpp
C++: Add bindings for session stop notification
[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         for (auto channel : channels)
329                 analog->channels = g_slist_append(analog->channels, channel->_structure);
330         analog->num_samples = num_samples;
331         analog->mq = mq->id();
332         analog->unit = unit->id();
333         analog->mqflags = QuantityFlag::mask_from_flags(mqflags);
334         analog->data = data_pointer;
335         auto packet = g_new(struct sr_datafeed_packet, 1);
336         packet->type = SR_DF_ANALOG;
337         packet->payload = analog;
338         return shared_ptr<Packet>(new Packet(nullptr, packet), Packet::Deleter());
339 }
340
341 shared_ptr<Session> Context::load_session(string filename)
342 {
343         return shared_ptr<Session>(
344                 new Session(shared_from_this(), filename), Session::Deleter());
345 }
346
347 shared_ptr<Trigger> Context::create_trigger(string name)
348 {
349         return shared_ptr<Trigger>(
350                 new Trigger(shared_from_this(), name), Trigger::Deleter());
351 }
352
353 shared_ptr<Input> Context::open_file(string filename)
354 {
355         const struct sr_input *input;
356
357         check(sr_input_scan_file(filename.c_str(), &input));
358         return shared_ptr<Input>(
359                 new Input(shared_from_this(), input), Input::Deleter());
360 }
361
362 shared_ptr<Input> Context::open_stream(string header)
363 {
364         const struct sr_input *input;
365
366         auto gstr = g_string_new(header.c_str());
367         auto ret = sr_input_scan_buffer(gstr, &input);
368         g_string_free(gstr, true);
369         check(ret);
370         return shared_ptr<Input>(
371                 new Input(shared_from_this(), input), Input::Deleter());
372 }
373
374 map<string, string> Context::serials(shared_ptr<Driver> driver)
375 {
376         GSList *serial_list = sr_serial_list(driver ? driver->_structure : NULL);
377         map<string, string> serials;
378
379         for (GSList *serial = serial_list; serial; serial = serial->next) {
380                 struct sr_serial_port *port = (sr_serial_port *) serial->data;
381                 serials[string(port->name)] = string(port->description);
382         }
383
384         g_slist_free_full(serial_list, (GDestroyNotify)sr_serial_free);
385         return serials;
386 }
387
388 Driver::Driver(struct sr_dev_driver *structure) :
389         ParentOwned(structure),
390         Configurable(structure, NULL, NULL),
391         _initialized(false)
392 {
393 }
394
395 Driver::~Driver()
396 {
397 }
398
399 string Driver::name()
400 {
401         return valid_string(_structure->name);
402 }
403
404 string Driver::long_name()
405 {
406         return valid_string(_structure->longname);
407 }
408
409 vector<shared_ptr<HardwareDevice>> Driver::scan(
410         map<const ConfigKey *, Glib::VariantBase> options)
411 {
412         /* Initialise the driver if not yet done. */
413         if (!_initialized)
414         {
415                 check(sr_driver_init(_parent->_structure, _structure));
416                 _initialized = true;
417         }
418
419         /* Translate scan options to GSList of struct sr_config pointers. */
420         GSList *option_list = NULL;
421         for (auto entry : options)
422         {
423                 auto key = entry.first;
424                 auto value = entry.second;
425                 auto config = g_new(struct sr_config, 1);
426                 config->key = key->id();
427                 config->data = value.gobj();
428                 option_list = g_slist_append(option_list, config);
429         }
430
431         /* Run scan. */
432         GSList *device_list = sr_driver_scan(_structure, option_list);
433
434         /* Free option list. */
435         g_slist_free_full(option_list, g_free);
436
437
438         /* Create device objects. */
439         vector<shared_ptr<HardwareDevice>> result;
440         for (GSList *device = device_list; device; device = device->next)
441         {
442                 auto sdi = (struct sr_dev_inst *) device->data;
443                 result.push_back(shared_ptr<HardwareDevice>(
444                         new HardwareDevice(shared_from_this(), sdi),
445                         HardwareDevice::Deleter()));
446         }
447
448         /* Free GSList returned from scan. */
449         g_slist_free(device_list);
450
451         return result;
452 }
453
454 Configurable::Configurable(
455                 struct sr_dev_driver *driver,
456                 struct sr_dev_inst *sdi,
457                 struct sr_channel_group *cg) :
458         config_driver(driver),
459         config_sdi(sdi),
460         config_channel_group(cg)
461 {
462 }
463
464 Configurable::~Configurable()
465 {
466 }
467
468 Glib::VariantBase Configurable::config_get(const ConfigKey *key)
469 {
470         GVariant *data;
471         check(sr_config_get(
472                 config_driver, config_sdi, config_channel_group,
473                 key->id(), &data));
474         return Glib::VariantBase(data);
475 }
476
477 void Configurable::config_set(const ConfigKey *key, Glib::VariantBase value)
478 {
479         check(sr_config_set(
480                 config_sdi, config_channel_group,
481                 key->id(), value.gobj()));
482 }
483
484 Glib::VariantContainerBase Configurable::config_list(const ConfigKey *key)
485 {
486         GVariant *data;
487         check(sr_config_list(
488                 config_driver, config_sdi, config_channel_group,
489                 key->id(), &data));
490         return Glib::VariantContainerBase(data);
491 }
492
493 map<const ConfigKey *, set<Capability>> Configurable::config_keys(const ConfigKey *key)
494 {
495         GVariant *gvar_opts;
496         gsize num_opts;
497         const uint32_t *opts;
498         map<const ConfigKey *, set<Capability>> result;
499
500         check(sr_config_list(
501                 config_driver, config_sdi, config_channel_group,
502                 key->id(), &gvar_opts));
503
504         opts = (const uint32_t *) g_variant_get_fixed_array(
505                 gvar_opts, &num_opts, sizeof(uint32_t));
506
507         for (gsize i = 0; i < num_opts; i++)
508         {
509                 auto key = ConfigKey::get(opts[i] & SR_CONF_MASK);
510                 set<Capability> capabilities;
511                 if (opts[i] & SR_CONF_GET)
512                         capabilities.insert(GET);
513                 if (opts[i] & SR_CONF_SET)
514                         capabilities.insert(SET);
515                 if (opts[i] & SR_CONF_LIST)
516                         capabilities.insert(LIST);
517                 result[key] = capabilities;
518         }
519
520         g_variant_unref(gvar_opts);
521
522         return result;
523 }
524
525 bool Configurable::config_check(const ConfigKey *key,
526         const ConfigKey *index_key)
527 {
528         GVariant *gvar_opts;
529         gsize num_opts;
530         const uint32_t *opts;
531
532         if (sr_config_list(config_driver, config_sdi, config_channel_group,
533                         index_key->id(), &gvar_opts) != SR_OK)
534                 return false;
535
536         opts = (const uint32_t *) g_variant_get_fixed_array(
537                 gvar_opts, &num_opts, sizeof(uint32_t));
538
539         for (gsize i = 0; i < num_opts; i++)
540         {
541                 if ((opts[i] & SR_CONF_MASK) == (uint32_t) key->id())
542                 {
543                         g_variant_unref(gvar_opts);
544                         return true;
545                 }
546         }
547
548         g_variant_unref(gvar_opts);
549
550         return false;
551 }
552
553 Device::Device(struct sr_dev_inst *structure) :
554         Configurable(sr_dev_inst_driver_get(structure), structure, NULL),
555         _structure(structure)
556 {
557         for (GSList *entry = sr_dev_inst_channels_get(structure); entry; entry = entry->next)
558         {
559                 auto channel = (struct sr_channel *) entry->data;
560                 _channels[channel] = new Channel(channel);
561         }
562
563         for (GSList *entry = sr_dev_inst_channel_groups_get(structure); entry; entry = entry->next)
564         {
565                 auto group = (struct sr_channel_group *) entry->data;
566                 _channel_groups[group->name] = new ChannelGroup(this, group);
567         }
568 }
569
570 Device::~Device()
571 {
572         for (auto entry : _channels)
573                 delete entry.second;
574         for (auto entry : _channel_groups)
575                 delete entry.second;
576 }
577
578 string Device::vendor()
579 {
580         return valid_string(sr_dev_inst_vendor_get(_structure));
581 }
582
583 string Device::model()
584 {
585         return valid_string(sr_dev_inst_model_get(_structure));
586 }
587
588 string Device::version()
589 {
590         return valid_string(sr_dev_inst_version_get(_structure));
591 }
592
593 string Device::serial_number()
594 {
595         return valid_string(sr_dev_inst_sernum_get(_structure));
596 }
597
598 string Device::connection_id()
599 {
600         return valid_string(sr_dev_inst_connid_get(_structure));
601 }
602
603 vector<shared_ptr<Channel>> Device::channels()
604 {
605         vector<shared_ptr<Channel>> result;
606         for (auto channel = sr_dev_inst_channels_get(_structure); channel; channel = channel->next)
607                 result.push_back(
608                         _channels[(struct sr_channel *) channel->data]->get_shared_pointer(
609                                 get_shared_from_this()));
610         return result;
611 }
612
613 shared_ptr<Channel> Device::get_channel(struct sr_channel *ptr)
614 {
615         return _channels[ptr]->get_shared_pointer(get_shared_from_this());
616 }
617
618 map<string, shared_ptr<ChannelGroup>>
619 Device::channel_groups()
620 {
621         map<string, shared_ptr<ChannelGroup>> result;
622         for (auto entry: _channel_groups)
623         {
624                 auto name = entry.first;
625                 auto channel_group = entry.second;
626                 result[name] = channel_group->get_shared_pointer(get_shared_from_this());
627         }
628         return result;
629 }
630
631 void Device::open()
632 {
633         check(sr_dev_open(_structure));
634 }
635
636 void Device::close()
637 {
638         check(sr_dev_close(_structure));
639 }
640
641 HardwareDevice::HardwareDevice(shared_ptr<Driver> driver,
642                 struct sr_dev_inst *structure) :
643         UserOwned(structure),
644         Device(structure),
645         _driver(driver)
646 {
647 }
648
649 HardwareDevice::~HardwareDevice()
650 {
651 }
652
653 shared_ptr<Device> HardwareDevice::get_shared_from_this()
654 {
655         return static_pointer_cast<Device>(shared_from_this());
656 }
657
658 shared_ptr<Driver> HardwareDevice::driver()
659 {
660         return _driver;
661 }
662
663 UserDevice::UserDevice(string vendor, string model, string version) :
664         UserOwned(sr_dev_inst_user_new(
665                 vendor.c_str(), model.c_str(), version.c_str())),
666         Device(UserOwned::_structure)
667 {
668 }
669
670 UserDevice::~UserDevice()
671 {
672 }
673
674 shared_ptr<Device> UserDevice::get_shared_from_this()
675 {
676         return static_pointer_cast<Device>(shared_from_this());
677 }
678
679 shared_ptr<Channel> UserDevice::add_channel(unsigned int index,
680         const ChannelType *type, string name)
681 {
682         check(sr_dev_inst_channel_add(Device::_structure,
683                 index, type->id(), name.c_str()));
684         struct sr_channel *structure = (struct sr_channel *)
685                         g_slist_last(sr_dev_inst_channels_get(Device::_structure))->data;
686         Channel *channel = new Channel(structure);
687         _channels[structure] = channel;
688         return get_channel(structure);
689 }
690
691 Channel::Channel(struct sr_channel *structure) :
692         ParentOwned(structure),
693         _type(ChannelType::get(_structure->type))
694 {
695 }
696
697 Channel::~Channel()
698 {
699 }
700
701 string Channel::name()
702 {
703         return valid_string(_structure->name);
704 }
705
706 void Channel::set_name(string name)
707 {
708         check(sr_dev_channel_name_set(_structure, name.c_str()));
709 }
710
711 const ChannelType *Channel::type()
712 {
713         return ChannelType::get(_structure->type);
714 }
715
716 bool Channel::enabled()
717 {
718         return _structure->enabled;
719 }
720
721 void Channel::set_enabled(bool value)
722 {
723         check(sr_dev_channel_enable(_structure, value));
724 }
725
726 unsigned int Channel::index()
727 {
728         return _structure->index;
729 }
730
731 ChannelGroup::ChannelGroup(Device *device,
732                 struct sr_channel_group *structure) :
733         ParentOwned(structure),
734         Configurable(sr_dev_inst_driver_get(device->_structure), device->_structure, structure)
735 {
736         for (GSList *entry = structure->channels; entry; entry = entry->next)
737                 _channels.push_back(device->_channels[(struct sr_channel *)entry->data]);
738 }
739
740 ChannelGroup::~ChannelGroup()
741 {
742 }
743
744 string ChannelGroup::name()
745 {
746         return valid_string(_structure->name);
747 }
748
749 vector<shared_ptr<Channel>> ChannelGroup::channels()
750 {
751         vector<shared_ptr<Channel>> result;
752         for (auto channel : _channels)
753                 result.push_back(channel->get_shared_pointer(_parent));
754         return result;
755 }
756
757 Trigger::Trigger(shared_ptr<Context> context, string name) : 
758         UserOwned(sr_trigger_new(name.c_str())),
759         _context(context)
760 {
761         for (auto stage = _structure->stages; stage; stage = stage->next)
762                 _stages.push_back(
763                         new TriggerStage((struct sr_trigger_stage *) stage->data));
764 }
765
766 Trigger::~Trigger()
767 {
768         for (auto stage: _stages)
769                 delete stage;
770
771         sr_trigger_free(_structure);
772 }
773
774 string Trigger::name()
775 {
776         return _structure->name;
777 }
778
779 vector<shared_ptr<TriggerStage>> Trigger::stages()
780 {
781         vector<shared_ptr<TriggerStage>> result;
782         for (auto stage : _stages)
783                 result.push_back(stage->get_shared_pointer(this));
784         return result;
785 }
786
787 shared_ptr<TriggerStage> Trigger::add_stage()
788 {
789         auto stage = new TriggerStage(sr_trigger_stage_add(_structure));
790         _stages.push_back(stage);
791         return stage->get_shared_pointer(this);
792 }
793
794 TriggerStage::TriggerStage(struct sr_trigger_stage *structure) : 
795         ParentOwned(structure)
796 {
797 }
798
799 TriggerStage::~TriggerStage()
800 {
801         for (auto match : _matches)
802                 delete match;
803 }
804         
805 int TriggerStage::number()
806 {
807         return _structure->stage;
808 }
809
810 vector<shared_ptr<TriggerMatch>> TriggerStage::matches()
811 {
812         vector<shared_ptr<TriggerMatch>> result;
813         for (auto match : _matches)
814                 result.push_back(match->get_shared_pointer(this));
815         return result;
816 }
817
818 void TriggerStage::add_match(shared_ptr<Channel> channel,
819         const TriggerMatchType *type, float value)
820 {
821         check(sr_trigger_match_add(_structure,
822                 channel->_structure, type->id(), value));
823         _matches.push_back(new TriggerMatch(
824                 (struct sr_trigger_match *) g_slist_last(
825                         _structure->matches)->data, channel));
826 }
827
828 void TriggerStage::add_match(shared_ptr<Channel> channel,
829         const TriggerMatchType *type)
830 {
831         add_match(channel, type, NAN);
832 }
833
834 TriggerMatch::TriggerMatch(struct sr_trigger_match *structure,
835                 shared_ptr<Channel> channel) :
836         ParentOwned(structure),
837         _channel(channel)
838 {
839 }
840
841 TriggerMatch::~TriggerMatch()
842 {
843 }
844
845 shared_ptr<Channel> TriggerMatch::channel()
846 {
847         return _channel;
848 }
849
850 const TriggerMatchType *TriggerMatch::type()
851 {
852         return TriggerMatchType::get(_structure->match);
853 }
854
855 float TriggerMatch::value()
856 {
857         return _structure->value;
858 }
859
860 DatafeedCallbackData::DatafeedCallbackData(Session *session,
861                 DatafeedCallbackFunction callback) :
862         _callback(callback),
863         _session(session)
864 {
865 }
866
867 void DatafeedCallbackData::run(const struct sr_dev_inst *sdi,
868         const struct sr_datafeed_packet *pkt)
869 {
870         auto device = _session->get_device(sdi);
871         auto packet = shared_ptr<Packet>(new Packet(device, pkt), Packet::Deleter());
872         _callback(device, packet);
873 }
874
875 SourceCallbackData::SourceCallbackData(shared_ptr<EventSource> source) :
876         _source(source)
877 {
878 }
879
880 bool SourceCallbackData::run(int revents)
881 {
882         return _source->_callback((Glib::IOCondition) revents);
883 }
884
885 shared_ptr<EventSource> EventSource::create(int fd, Glib::IOCondition events,
886         int timeout, SourceCallbackFunction callback)
887 {
888         auto result = new EventSource(timeout, callback);
889         result->_type = EventSource::SOURCE_FD;
890         result->_fd = fd;
891         result->_events = events;
892         return shared_ptr<EventSource>(result, EventSource::Deleter());
893 }
894
895 shared_ptr<EventSource> EventSource::create(Glib::PollFD pollfd, int timeout,
896         SourceCallbackFunction callback)
897 {
898         auto result = new EventSource(timeout, callback);
899         result->_type = EventSource::SOURCE_POLLFD;
900         result->_pollfd = pollfd;
901         return shared_ptr<EventSource>(result, EventSource::Deleter());
902 }
903
904 shared_ptr<EventSource> EventSource::create(Glib::RefPtr<Glib::IOChannel> channel,
905         Glib::IOCondition events, int timeout, SourceCallbackFunction callback)
906 {
907         auto result = new EventSource(timeout, callback);
908         result->_type = EventSource::SOURCE_IOCHANNEL;
909         result->_channel = channel;
910         result->_events = events;
911         return shared_ptr<EventSource>(result, EventSource::Deleter());
912 }
913
914 EventSource::EventSource(int timeout, SourceCallbackFunction callback) :
915         _timeout(timeout),
916         _callback(callback)
917 {
918 }
919
920 EventSource::~EventSource()
921 {
922 }
923
924 SessionDevice::SessionDevice(struct sr_dev_inst *structure) :
925         ParentOwned(structure),
926         Device(structure)
927 {
928 }
929
930 SessionDevice::~SessionDevice()
931 {
932 }
933
934 shared_ptr<Device> SessionDevice::get_shared_from_this()
935 {
936         return static_pointer_cast<Device>(shared_from_this());
937 }
938
939 Session::Session(shared_ptr<Context> context) :
940         UserOwned(_structure),
941         _context(context)
942 {
943         check(sr_session_new(context->_structure, &_structure));
944         _context->_session = this;
945 }
946
947 Session::Session(shared_ptr<Context> context, string filename) :
948         UserOwned(_structure),
949         _context(context),
950         _filename(filename)
951 {
952         check(sr_session_load(context->_structure, filename.c_str(), &_structure));
953         GSList *dev_list;
954         check(sr_session_dev_list(_structure, &dev_list));
955         for (GSList *dev = dev_list; dev; dev = dev->next)
956         {
957                 auto sdi = (struct sr_dev_inst *) dev->data;
958                 _owned_devices[sdi] = new SessionDevice(sdi);
959         }
960         _context->_session = this;
961 }
962
963 Session::~Session()
964 {
965         check(sr_session_destroy(_structure));
966
967         for (auto callback : _datafeed_callbacks)
968                 delete callback;
969
970         for (auto entry : _source_callbacks)
971                 delete entry.second;
972
973         for (auto entry : _owned_devices)
974                 delete entry.second;
975 }
976
977 shared_ptr<Device> Session::get_device(const struct sr_dev_inst *sdi)
978 {
979         if (_owned_devices.count(sdi))
980                 return static_pointer_cast<Device>(
981                         _owned_devices[sdi]->get_shared_pointer(this));
982         else if (_other_devices.count(sdi))
983                 return _other_devices[sdi];
984         else
985                 throw Error(SR_ERR_BUG);
986 }
987
988 void Session::add_device(shared_ptr<Device> device)
989 {
990         check(sr_session_dev_add(_structure, device->_structure));
991         _other_devices[device->_structure] = device;
992 }
993
994 vector<shared_ptr<Device>> Session::devices()
995 {
996         GSList *dev_list;
997         check(sr_session_dev_list(_structure, &dev_list));
998         vector<shared_ptr<Device>> result;
999         for (GSList *dev = dev_list; dev; dev = dev->next)
1000         {
1001                 auto sdi = (struct sr_dev_inst *) dev->data;
1002                 result.push_back(get_device(sdi));
1003         }
1004         return result;
1005 }
1006
1007 void Session::remove_devices()
1008 {
1009         _other_devices.clear();
1010         check(sr_session_dev_remove_all(_structure));
1011 }
1012
1013 void Session::start()
1014 {
1015         check(sr_session_start(_structure));
1016 }
1017
1018 void Session::run()
1019 {
1020         check(sr_session_run(_structure));
1021 }
1022
1023 void Session::stop()
1024 {
1025         check(sr_session_stop(_structure));
1026 }
1027
1028 bool Session::is_running() const
1029 {
1030         const int ret = sr_session_is_running(_structure);
1031         if (ret < 0)
1032                 throw Error{ret};
1033         return (ret != 0);
1034 }
1035
1036 static void session_stopped_callback(void *data)
1037 {
1038         auto *const callback = static_cast<SessionStoppedCallback*>(data);
1039         (*callback)();
1040 }
1041
1042 void Session::set_stopped_callback(SessionStoppedCallback callback)
1043 {
1044         _stopped_callback = move(callback);
1045         if (_stopped_callback)
1046                 check(sr_session_stopped_callback_set(_structure,
1047                                 &session_stopped_callback, &_stopped_callback));
1048         else
1049                 check(sr_session_stopped_callback_set(_structure,
1050                                 nullptr, nullptr));
1051 }
1052
1053 static void datafeed_callback(const struct sr_dev_inst *sdi,
1054         const struct sr_datafeed_packet *pkt, void *cb_data)
1055 {
1056         auto callback = static_cast<DatafeedCallbackData *>(cb_data);
1057         callback->run(sdi, pkt);
1058 }
1059         
1060 void Session::add_datafeed_callback(DatafeedCallbackFunction callback)
1061 {
1062         auto cb_data = new DatafeedCallbackData(this, callback);
1063         check(sr_session_datafeed_callback_add(_structure,
1064                 datafeed_callback, cb_data));
1065         _datafeed_callbacks.push_back(cb_data);
1066 }
1067
1068 void Session::remove_datafeed_callbacks(void)
1069 {
1070         check(sr_session_datafeed_callback_remove_all(_structure));
1071         for (auto callback : _datafeed_callbacks)
1072                 delete callback;
1073         _datafeed_callbacks.clear();
1074 }
1075
1076 static int source_callback(int fd, int revents, void *cb_data)
1077 {
1078         (void) fd;
1079         auto callback = (SourceCallbackData *) cb_data;
1080         return callback->run(revents);
1081 }
1082
1083 void Session::add_source(shared_ptr<EventSource> source)
1084 {
1085         if (_source_callbacks.count(source) == 1)
1086                 throw Error(SR_ERR_ARG);
1087
1088         auto cb_data = new SourceCallbackData(source);
1089
1090         switch (source->_type)
1091         {
1092                 case EventSource::SOURCE_FD:
1093                         check(sr_session_source_add(_structure, source->_fd, source->_events,
1094                                 source->_timeout, source_callback, cb_data));
1095                         break;
1096                 case EventSource::SOURCE_POLLFD:
1097                         check(sr_session_source_add_pollfd(_structure,
1098                                 source->_pollfd.gobj(), source->_timeout, source_callback,
1099                                 cb_data));
1100                         break;
1101                 case EventSource::SOURCE_IOCHANNEL:
1102                         check(sr_session_source_add_channel(_structure,
1103                                 source->_channel->gobj(), source->_events, source->_timeout,
1104                                 source_callback, cb_data));
1105                         break;
1106         }
1107
1108         _source_callbacks[source] = cb_data;
1109 }
1110
1111 void Session::remove_source(shared_ptr<EventSource> source)
1112 {
1113         if (_source_callbacks.count(source) == 0)
1114                 throw Error(SR_ERR_ARG);
1115
1116         switch (source->_type)
1117         {
1118                 case EventSource::SOURCE_FD:
1119                         check(sr_session_source_remove(_structure, source->_fd));
1120                         break;
1121                 case EventSource::SOURCE_POLLFD:
1122                         check(sr_session_source_remove_pollfd(_structure,
1123                                 source->_pollfd.gobj()));
1124                         break;
1125                 case EventSource::SOURCE_IOCHANNEL:
1126                         check(sr_session_source_remove_channel(_structure,
1127                                 source->_channel->gobj()));
1128                         break;
1129         }
1130
1131         delete _source_callbacks[source];
1132
1133         _source_callbacks.erase(source);
1134 }
1135
1136 shared_ptr<Trigger> Session::trigger()
1137 {
1138         return _trigger;
1139 }
1140
1141 void Session::set_trigger(shared_ptr<Trigger> trigger)
1142 {
1143         if (!trigger)
1144                 // Set NULL trigger, i.e. remove any trigger from the session.
1145                 check(sr_session_trigger_set(_structure, NULL));
1146         else
1147                 check(sr_session_trigger_set(_structure, trigger->_structure));
1148         _trigger = trigger;
1149 }
1150
1151 string Session::filename()
1152 {
1153         return _filename;
1154 }
1155
1156 shared_ptr<Context> Session::context()
1157 {
1158         return _context;
1159 }
1160
1161 Packet::Packet(shared_ptr<Device> device,
1162         const struct sr_datafeed_packet *structure) :
1163         UserOwned(structure),
1164         _device(device)
1165 {
1166         switch (structure->type)
1167         {
1168                 case SR_DF_HEADER:
1169                         _payload = new Header(
1170                                 static_cast<const struct sr_datafeed_header *>(
1171                                         structure->payload));
1172                         break;
1173                 case SR_DF_META:
1174                         _payload = new Meta(
1175                                 static_cast<const struct sr_datafeed_meta *>(
1176                                         structure->payload));
1177                         break;
1178                 case SR_DF_LOGIC:
1179                         _payload = new Logic(
1180                                 static_cast<const struct sr_datafeed_logic *>(
1181                                         structure->payload));
1182                         break;
1183                 case SR_DF_ANALOG:
1184                         _payload = new Analog(
1185                                 static_cast<const struct sr_datafeed_analog *>(
1186                                         structure->payload));
1187                         break;
1188                 default:
1189                         _payload = nullptr;
1190                         break;
1191         }
1192 }
1193
1194 Packet::~Packet()
1195 {
1196         if (_payload)
1197                 delete _payload;
1198 }
1199
1200 const PacketType *Packet::type()
1201 {
1202         return PacketType::get(_structure->type);
1203 }
1204
1205 shared_ptr<PacketPayload> Packet::payload()
1206 {
1207         if (_payload)
1208                 return _payload->get_shared_pointer(this);
1209         else
1210                 throw Error(SR_ERR_NA);
1211 }
1212
1213 PacketPayload::PacketPayload()
1214 {
1215 }
1216
1217 PacketPayload::~PacketPayload()
1218 {
1219 }
1220
1221 Header::Header(const struct sr_datafeed_header *structure) :
1222         ParentOwned(structure),
1223         PacketPayload()
1224 {
1225 }
1226
1227 Header::~Header()
1228 {
1229 }
1230
1231 shared_ptr<PacketPayload> Header::get_shared_pointer(Packet *_parent)
1232 {
1233         return static_pointer_cast<PacketPayload>(
1234                 ParentOwned::get_shared_pointer(_parent));
1235 }
1236
1237 int Header::feed_version()
1238 {
1239         return _structure->feed_version;
1240 }
1241
1242 Glib::TimeVal Header::start_time()
1243 {
1244         return Glib::TimeVal(
1245                 _structure->starttime.tv_sec,
1246                 _structure->starttime.tv_usec);
1247 }
1248
1249 Meta::Meta(const struct sr_datafeed_meta *structure) :
1250         ParentOwned(structure),
1251         PacketPayload()
1252 {
1253 }
1254
1255 Meta::~Meta()
1256 {
1257 }
1258
1259 shared_ptr<PacketPayload> Meta::get_shared_pointer(Packet *_parent)
1260 {
1261         return static_pointer_cast<PacketPayload>(
1262                 ParentOwned::get_shared_pointer(_parent));
1263 }
1264
1265 map<const ConfigKey *, Glib::VariantBase> Meta::config()
1266 {
1267         map<const ConfigKey *, Glib::VariantBase> result;
1268         for (auto l = _structure->config; l; l = l->next)
1269         {
1270                 auto config = (struct sr_config *) l->data;
1271                 result[ConfigKey::get(config->key)] = Glib::VariantBase(config->data);
1272         }
1273         return result;
1274 }
1275
1276 Logic::Logic(const struct sr_datafeed_logic *structure) :
1277         ParentOwned(structure),
1278         PacketPayload()
1279 {
1280 }
1281
1282 Logic::~Logic()
1283 {
1284 }
1285
1286 shared_ptr<PacketPayload> Logic::get_shared_pointer(Packet *_parent)
1287 {
1288         return static_pointer_cast<PacketPayload>(
1289                 ParentOwned::get_shared_pointer(_parent));
1290 }
1291
1292 void *Logic::data_pointer()
1293 {
1294         return _structure->data;
1295 }
1296
1297 size_t Logic::data_length()
1298 {
1299         return _structure->length;
1300 }
1301
1302 unsigned int Logic::unit_size()
1303 {
1304         return _structure->unitsize;
1305 }
1306
1307 Analog::Analog(const struct sr_datafeed_analog *structure) :
1308         ParentOwned(structure),
1309         PacketPayload()
1310 {
1311 }
1312
1313 Analog::~Analog()
1314 {
1315 }
1316
1317 shared_ptr<PacketPayload> Analog::get_shared_pointer(Packet *_parent)
1318 {
1319         return static_pointer_cast<PacketPayload>(
1320                 ParentOwned::get_shared_pointer(_parent));
1321 }
1322
1323 float *Analog::data_pointer()
1324 {
1325         return _structure->data;
1326 }
1327
1328 unsigned int Analog::num_samples()
1329 {
1330         return _structure->num_samples;
1331 }
1332
1333 vector<shared_ptr<Channel>> Analog::channels()
1334 {
1335         vector<shared_ptr<Channel>> result;
1336         for (auto l = _structure->channels; l; l = l->next)
1337                 result.push_back(_parent->_device->get_channel(
1338                         (struct sr_channel *)l->data));
1339         return result;
1340 }
1341
1342 const Quantity *Analog::mq()
1343 {
1344         return Quantity::get(_structure->mq);
1345 }
1346
1347 const Unit *Analog::unit()
1348 {
1349         return Unit::get(_structure->unit);
1350 }
1351
1352 vector<const QuantityFlag *> Analog::mq_flags()
1353 {
1354         return QuantityFlag::flags_from_mask(_structure->mqflags);
1355 }
1356
1357 InputFormat::InputFormat(const struct sr_input_module *structure) :
1358         ParentOwned(structure)
1359 {
1360 }
1361
1362 InputFormat::~InputFormat()
1363 {
1364 }
1365
1366 string InputFormat::name()
1367 {
1368         return valid_string(sr_input_id_get(_structure));
1369 }
1370
1371 string InputFormat::description()
1372 {
1373         return valid_string(sr_input_description_get(_structure));
1374 }
1375
1376 vector<string> InputFormat::extensions()
1377 {
1378         vector<string> exts;
1379         for (const char *const *e = sr_input_extensions_get(_structure);
1380                 e && *e; e++)
1381                 exts.push_back(*e);
1382         return exts;
1383 }
1384
1385 map<string, shared_ptr<Option>> InputFormat::options()
1386 {
1387         const struct sr_option **options = sr_input_options_get(_structure);
1388         map<string, shared_ptr<Option>> result;
1389         if (options)
1390         {
1391                 auto option_array = shared_ptr<const struct sr_option *>(
1392                         options, sr_input_options_free);
1393                 for (int i = 0; options[i]; i++)
1394                         result[options[i]->id] = shared_ptr<Option>(
1395                                 new Option(options[i], option_array), Option::Deleter());
1396         }
1397         return result;
1398 }
1399
1400 shared_ptr<Input> InputFormat::create_input(
1401         map<string, Glib::VariantBase> options)
1402 {
1403         auto input = sr_input_new(_structure, map_to_hash_variant(options));
1404         if (!input)
1405                 throw Error(SR_ERR_ARG);
1406         return shared_ptr<Input>(
1407                 new Input(_parent->shared_from_this(), input), Input::Deleter());
1408 }
1409
1410 Input::Input(shared_ptr<Context> context, const struct sr_input *structure) :
1411         UserOwned(structure),
1412         _context(context),
1413         _device(nullptr)
1414 {
1415 }
1416
1417 shared_ptr<InputDevice> Input::device()
1418 {
1419         if (!_device)
1420         {
1421                 auto sdi = sr_input_dev_inst_get(_structure);
1422                 if (!sdi)
1423                         throw Error(SR_ERR_NA);
1424                 _device = new InputDevice(shared_from_this(), sdi);
1425         }
1426
1427         return _device->get_shared_pointer(shared_from_this());
1428 }
1429
1430 void Input::send(void *data, size_t length)
1431 {
1432         auto gstr = g_string_new_len((gchar *)data, length);
1433         auto ret = sr_input_send(_structure, gstr);
1434         g_string_free(gstr, false);
1435         check(ret);
1436 }
1437
1438 void Input::end()
1439 {
1440         check(sr_input_end(_structure));
1441 }
1442
1443 Input::~Input()
1444 {
1445         if (_device)
1446                 delete _device;
1447         sr_input_free(_structure);
1448 }
1449
1450 InputDevice::InputDevice(shared_ptr<Input> input,
1451                 struct sr_dev_inst *structure) :
1452         ParentOwned(structure),
1453         Device(structure),
1454         _input(input)
1455 {
1456 }
1457
1458 InputDevice::~InputDevice()
1459 {
1460 }
1461
1462 shared_ptr<Device> InputDevice::get_shared_from_this()
1463 {
1464         return static_pointer_cast<Device>(shared_from_this());
1465 }
1466
1467 Option::Option(const struct sr_option *structure,
1468                 shared_ptr<const struct sr_option *> structure_array) :
1469         UserOwned(structure),
1470         _structure_array(structure_array)
1471 {
1472 }
1473
1474 Option::~Option()
1475 {
1476 }
1477
1478 string Option::id()
1479 {
1480         return valid_string(_structure->id);
1481 }
1482
1483 string Option::name()
1484 {
1485         return valid_string(_structure->name);
1486 }
1487
1488 string Option::description()
1489 {
1490         return valid_string(_structure->desc);
1491 }
1492
1493 Glib::VariantBase Option::default_value()
1494 {
1495         return Glib::VariantBase(_structure->def, true);
1496 }
1497
1498 vector<Glib::VariantBase> Option::values()
1499 {
1500         vector<Glib::VariantBase> result;
1501         for (auto l = _structure->values; l; l = l->next)
1502                 result.push_back(Glib::VariantBase((GVariant *) l->data, true));
1503         return result;
1504 }
1505
1506 OutputFormat::OutputFormat(const struct sr_output_module *structure) :
1507         ParentOwned(structure)
1508 {
1509 }
1510
1511 OutputFormat::~OutputFormat()
1512 {
1513 }
1514
1515 string OutputFormat::name()
1516 {
1517         return valid_string(sr_output_id_get(_structure));
1518 }
1519
1520 string OutputFormat::description()
1521 {
1522         return valid_string(sr_output_description_get(_structure));
1523 }
1524
1525 vector<string> OutputFormat::extensions()
1526 {
1527         vector<string> exts;
1528         for (const char *const *e = sr_output_extensions_get(_structure);
1529                 e && *e; e++)
1530                 exts.push_back(*e);
1531         return exts;
1532 }
1533
1534 map<string, shared_ptr<Option>> OutputFormat::options()
1535 {
1536         const struct sr_option **options = sr_output_options_get(_structure);
1537         map<string, shared_ptr<Option>> result;
1538         if (options)
1539         {
1540                 auto option_array = shared_ptr<const struct sr_option *>(
1541                         options, sr_output_options_free);
1542                 for (int i = 0; options[i]; i++)
1543                         result[options[i]->id] = shared_ptr<Option>(
1544                                 new Option(options[i], option_array), Option::Deleter());
1545         }
1546         return result;
1547 }
1548
1549 shared_ptr<Output> OutputFormat::create_output(
1550         shared_ptr<Device> device, map<string, Glib::VariantBase> options)
1551 {
1552         return shared_ptr<Output>(
1553                 new Output(shared_from_this(), device, options),
1554                 Output::Deleter());
1555 }
1556
1557 shared_ptr<Output> OutputFormat::create_output(string filename,
1558         shared_ptr<Device> device, map<string, Glib::VariantBase> options)
1559 {
1560         return shared_ptr<Output>(
1561                 new Output(filename, shared_from_this(), device, options),
1562                 Output::Deleter());
1563 }
1564
1565 bool OutputFormat::test_flag(const OutputFlag *flag)
1566 {
1567         return sr_output_test_flag(_structure, flag->id());
1568 }
1569
1570 Output::Output(shared_ptr<OutputFormat> format,
1571                 shared_ptr<Device> device, map<string, Glib::VariantBase> options) :
1572         UserOwned(sr_output_new(format->_structure,
1573                 map_to_hash_variant(options), device->_structure, NULL)),
1574         _format(format),
1575         _device(device),
1576         _options(options)
1577 {
1578 }
1579
1580 Output::Output(string filename, shared_ptr<OutputFormat> format,
1581                 shared_ptr<Device> device, map<string, Glib::VariantBase> options) :
1582         UserOwned(sr_output_new(format->_structure,
1583                 map_to_hash_variant(options), device->_structure, filename.c_str())),
1584         _format(format),
1585         _device(device),
1586         _options(options)
1587 {
1588 }
1589
1590 Output::~Output()
1591 {
1592         check(sr_output_free(_structure));
1593 }
1594
1595 string Output::receive(shared_ptr<Packet> packet)
1596 {
1597         GString *out;
1598         check(sr_output_send(_structure, packet->_structure, &out));
1599         if (out)
1600         {
1601                 auto result = string(out->str, out->str + out->len);
1602                 g_string_free(out, true);
1603                 return result;
1604         }
1605         else
1606         {
1607                 return string();
1608         }
1609 }
1610
1611 #include <enums.cpp>
1612
1613 }