]> sigrok.org Git - libsigrok.git/blob - bindings/cxx/classes.cpp
C++: Wrap resource access API
[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 static void datafeed_callback(const struct sr_dev_inst *sdi,
1029         const struct sr_datafeed_packet *pkt, void *cb_data)
1030 {
1031         auto callback = static_cast<DatafeedCallbackData *>(cb_data);
1032         callback->run(sdi, pkt);
1033 }
1034         
1035 void Session::add_datafeed_callback(DatafeedCallbackFunction callback)
1036 {
1037         auto cb_data = new DatafeedCallbackData(this, callback);
1038         check(sr_session_datafeed_callback_add(_structure,
1039                 datafeed_callback, cb_data));
1040         _datafeed_callbacks.push_back(cb_data);
1041 }
1042
1043 void Session::remove_datafeed_callbacks(void)
1044 {
1045         check(sr_session_datafeed_callback_remove_all(_structure));
1046         for (auto callback : _datafeed_callbacks)
1047                 delete callback;
1048         _datafeed_callbacks.clear();
1049 }
1050
1051 static int source_callback(int fd, int revents, void *cb_data)
1052 {
1053         (void) fd;
1054         auto callback = (SourceCallbackData *) cb_data;
1055         return callback->run(revents);
1056 }
1057
1058 void Session::add_source(shared_ptr<EventSource> source)
1059 {
1060         if (_source_callbacks.count(source) == 1)
1061                 throw Error(SR_ERR_ARG);
1062
1063         auto cb_data = new SourceCallbackData(source);
1064
1065         switch (source->_type)
1066         {
1067                 case EventSource::SOURCE_FD:
1068                         check(sr_session_source_add(_structure, source->_fd, source->_events,
1069                                 source->_timeout, source_callback, cb_data));
1070                         break;
1071                 case EventSource::SOURCE_POLLFD:
1072                         check(sr_session_source_add_pollfd(_structure,
1073                                 source->_pollfd.gobj(), source->_timeout, source_callback,
1074                                 cb_data));
1075                         break;
1076                 case EventSource::SOURCE_IOCHANNEL:
1077                         check(sr_session_source_add_channel(_structure,
1078                                 source->_channel->gobj(), source->_events, source->_timeout,
1079                                 source_callback, cb_data));
1080                         break;
1081         }
1082
1083         _source_callbacks[source] = cb_data;
1084 }
1085
1086 void Session::remove_source(shared_ptr<EventSource> source)
1087 {
1088         if (_source_callbacks.count(source) == 0)
1089                 throw Error(SR_ERR_ARG);
1090
1091         switch (source->_type)
1092         {
1093                 case EventSource::SOURCE_FD:
1094                         check(sr_session_source_remove(_structure, source->_fd));
1095                         break;
1096                 case EventSource::SOURCE_POLLFD:
1097                         check(sr_session_source_remove_pollfd(_structure,
1098                                 source->_pollfd.gobj()));
1099                         break;
1100                 case EventSource::SOURCE_IOCHANNEL:
1101                         check(sr_session_source_remove_channel(_structure,
1102                                 source->_channel->gobj()));
1103                         break;
1104         }
1105
1106         delete _source_callbacks[source];
1107
1108         _source_callbacks.erase(source);
1109 }
1110
1111 shared_ptr<Trigger> Session::trigger()
1112 {
1113         return _trigger;
1114 }
1115
1116 void Session::set_trigger(shared_ptr<Trigger> trigger)
1117 {
1118         if (!trigger)
1119                 // Set NULL trigger, i.e. remove any trigger from the session.
1120                 check(sr_session_trigger_set(_structure, NULL));
1121         else
1122                 check(sr_session_trigger_set(_structure, trigger->_structure));
1123         _trigger = trigger;
1124 }
1125
1126 string Session::filename()
1127 {
1128         return _filename;
1129 }
1130
1131 shared_ptr<Context> Session::context()
1132 {
1133         return _context;
1134 }
1135
1136 Packet::Packet(shared_ptr<Device> device,
1137         const struct sr_datafeed_packet *structure) :
1138         UserOwned(structure),
1139         _device(device)
1140 {
1141         switch (structure->type)
1142         {
1143                 case SR_DF_HEADER:
1144                         _payload = new Header(
1145                                 static_cast<const struct sr_datafeed_header *>(
1146                                         structure->payload));
1147                         break;
1148                 case SR_DF_META:
1149                         _payload = new Meta(
1150                                 static_cast<const struct sr_datafeed_meta *>(
1151                                         structure->payload));
1152                         break;
1153                 case SR_DF_LOGIC:
1154                         _payload = new Logic(
1155                                 static_cast<const struct sr_datafeed_logic *>(
1156                                         structure->payload));
1157                         break;
1158                 case SR_DF_ANALOG:
1159                         _payload = new Analog(
1160                                 static_cast<const struct sr_datafeed_analog *>(
1161                                         structure->payload));
1162                         break;
1163                 default:
1164                         _payload = nullptr;
1165                         break;
1166         }
1167 }
1168
1169 Packet::~Packet()
1170 {
1171         if (_payload)
1172                 delete _payload;
1173 }
1174
1175 const PacketType *Packet::type()
1176 {
1177         return PacketType::get(_structure->type);
1178 }
1179
1180 shared_ptr<PacketPayload> Packet::payload()
1181 {
1182         if (_payload)
1183                 return _payload->get_shared_pointer(this);
1184         else
1185                 throw Error(SR_ERR_NA);
1186 }
1187
1188 PacketPayload::PacketPayload()
1189 {
1190 }
1191
1192 PacketPayload::~PacketPayload()
1193 {
1194 }
1195
1196 Header::Header(const struct sr_datafeed_header *structure) :
1197         ParentOwned(structure),
1198         PacketPayload()
1199 {
1200 }
1201
1202 Header::~Header()
1203 {
1204 }
1205
1206 shared_ptr<PacketPayload> Header::get_shared_pointer(Packet *_parent)
1207 {
1208         return static_pointer_cast<PacketPayload>(
1209                 ParentOwned::get_shared_pointer(_parent));
1210 }
1211
1212 int Header::feed_version()
1213 {
1214         return _structure->feed_version;
1215 }
1216
1217 Glib::TimeVal Header::start_time()
1218 {
1219         return Glib::TimeVal(
1220                 _structure->starttime.tv_sec,
1221                 _structure->starttime.tv_usec);
1222 }
1223
1224 Meta::Meta(const struct sr_datafeed_meta *structure) :
1225         ParentOwned(structure),
1226         PacketPayload()
1227 {
1228 }
1229
1230 Meta::~Meta()
1231 {
1232 }
1233
1234 shared_ptr<PacketPayload> Meta::get_shared_pointer(Packet *_parent)
1235 {
1236         return static_pointer_cast<PacketPayload>(
1237                 ParentOwned::get_shared_pointer(_parent));
1238 }
1239
1240 map<const ConfigKey *, Glib::VariantBase> Meta::config()
1241 {
1242         map<const ConfigKey *, Glib::VariantBase> result;
1243         for (auto l = _structure->config; l; l = l->next)
1244         {
1245                 auto config = (struct sr_config *) l->data;
1246                 result[ConfigKey::get(config->key)] = Glib::VariantBase(config->data);
1247         }
1248         return result;
1249 }
1250
1251 Logic::Logic(const struct sr_datafeed_logic *structure) :
1252         ParentOwned(structure),
1253         PacketPayload()
1254 {
1255 }
1256
1257 Logic::~Logic()
1258 {
1259 }
1260
1261 shared_ptr<PacketPayload> Logic::get_shared_pointer(Packet *_parent)
1262 {
1263         return static_pointer_cast<PacketPayload>(
1264                 ParentOwned::get_shared_pointer(_parent));
1265 }
1266
1267 void *Logic::data_pointer()
1268 {
1269         return _structure->data;
1270 }
1271
1272 size_t Logic::data_length()
1273 {
1274         return _structure->length;
1275 }
1276
1277 unsigned int Logic::unit_size()
1278 {
1279         return _structure->unitsize;
1280 }
1281
1282 Analog::Analog(const struct sr_datafeed_analog *structure) :
1283         ParentOwned(structure),
1284         PacketPayload()
1285 {
1286 }
1287
1288 Analog::~Analog()
1289 {
1290 }
1291
1292 shared_ptr<PacketPayload> Analog::get_shared_pointer(Packet *_parent)
1293 {
1294         return static_pointer_cast<PacketPayload>(
1295                 ParentOwned::get_shared_pointer(_parent));
1296 }
1297
1298 float *Analog::data_pointer()
1299 {
1300         return _structure->data;
1301 }
1302
1303 unsigned int Analog::num_samples()
1304 {
1305         return _structure->num_samples;
1306 }
1307
1308 vector<shared_ptr<Channel>> Analog::channels()
1309 {
1310         vector<shared_ptr<Channel>> result;
1311         for (auto l = _structure->channels; l; l = l->next)
1312                 result.push_back(_parent->_device->get_channel(
1313                         (struct sr_channel *)l->data));
1314         return result;
1315 }
1316
1317 const Quantity *Analog::mq()
1318 {
1319         return Quantity::get(_structure->mq);
1320 }
1321
1322 const Unit *Analog::unit()
1323 {
1324         return Unit::get(_structure->unit);
1325 }
1326
1327 vector<const QuantityFlag *> Analog::mq_flags()
1328 {
1329         return QuantityFlag::flags_from_mask(_structure->mqflags);
1330 }
1331
1332 InputFormat::InputFormat(const struct sr_input_module *structure) :
1333         ParentOwned(structure)
1334 {
1335 }
1336
1337 InputFormat::~InputFormat()
1338 {
1339 }
1340
1341 string InputFormat::name()
1342 {
1343         return valid_string(sr_input_id_get(_structure));
1344 }
1345
1346 string InputFormat::description()
1347 {
1348         return valid_string(sr_input_description_get(_structure));
1349 }
1350
1351 vector<string> InputFormat::extensions()
1352 {
1353         vector<string> exts;
1354         for (const char *const *e = sr_input_extensions_get(_structure);
1355                 e && *e; e++)
1356                 exts.push_back(*e);
1357         return exts;
1358 }
1359
1360 map<string, shared_ptr<Option>> InputFormat::options()
1361 {
1362         const struct sr_option **options = sr_input_options_get(_structure);
1363         map<string, shared_ptr<Option>> result;
1364         if (options)
1365         {
1366                 auto option_array = shared_ptr<const struct sr_option *>(
1367                         options, sr_input_options_free);
1368                 for (int i = 0; options[i]; i++)
1369                         result[options[i]->id] = shared_ptr<Option>(
1370                                 new Option(options[i], option_array), Option::Deleter());
1371         }
1372         return result;
1373 }
1374
1375 shared_ptr<Input> InputFormat::create_input(
1376         map<string, Glib::VariantBase> options)
1377 {
1378         auto input = sr_input_new(_structure, map_to_hash_variant(options));
1379         if (!input)
1380                 throw Error(SR_ERR_ARG);
1381         return shared_ptr<Input>(
1382                 new Input(_parent->shared_from_this(), input), Input::Deleter());
1383 }
1384
1385 Input::Input(shared_ptr<Context> context, const struct sr_input *structure) :
1386         UserOwned(structure),
1387         _context(context),
1388         _device(nullptr)
1389 {
1390 }
1391
1392 shared_ptr<InputDevice> Input::device()
1393 {
1394         if (!_device)
1395         {
1396                 auto sdi = sr_input_dev_inst_get(_structure);
1397                 if (!sdi)
1398                         throw Error(SR_ERR_NA);
1399                 _device = new InputDevice(shared_from_this(), sdi);
1400         }
1401
1402         return _device->get_shared_pointer(shared_from_this());
1403 }
1404
1405 void Input::send(void *data, size_t length)
1406 {
1407         auto gstr = g_string_new_len((gchar *)data, length);
1408         auto ret = sr_input_send(_structure, gstr);
1409         g_string_free(gstr, false);
1410         check(ret);
1411 }
1412
1413 void Input::end()
1414 {
1415         check(sr_input_end(_structure));
1416 }
1417
1418 Input::~Input()
1419 {
1420         if (_device)
1421                 delete _device;
1422         sr_input_free(_structure);
1423 }
1424
1425 InputDevice::InputDevice(shared_ptr<Input> input,
1426                 struct sr_dev_inst *structure) :
1427         ParentOwned(structure),
1428         Device(structure),
1429         _input(input)
1430 {
1431 }
1432
1433 InputDevice::~InputDevice()
1434 {
1435 }
1436
1437 shared_ptr<Device> InputDevice::get_shared_from_this()
1438 {
1439         return static_pointer_cast<Device>(shared_from_this());
1440 }
1441
1442 Option::Option(const struct sr_option *structure,
1443                 shared_ptr<const struct sr_option *> structure_array) :
1444         UserOwned(structure),
1445         _structure_array(structure_array)
1446 {
1447 }
1448
1449 Option::~Option()
1450 {
1451 }
1452
1453 string Option::id()
1454 {
1455         return valid_string(_structure->id);
1456 }
1457
1458 string Option::name()
1459 {
1460         return valid_string(_structure->name);
1461 }
1462
1463 string Option::description()
1464 {
1465         return valid_string(_structure->desc);
1466 }
1467
1468 Glib::VariantBase Option::default_value()
1469 {
1470         return Glib::VariantBase(_structure->def, true);
1471 }
1472
1473 vector<Glib::VariantBase> Option::values()
1474 {
1475         vector<Glib::VariantBase> result;
1476         for (auto l = _structure->values; l; l = l->next)
1477                 result.push_back(Glib::VariantBase((GVariant *) l->data, true));
1478         return result;
1479 }
1480
1481 OutputFormat::OutputFormat(const struct sr_output_module *structure) :
1482         ParentOwned(structure)
1483 {
1484 }
1485
1486 OutputFormat::~OutputFormat()
1487 {
1488 }
1489
1490 string OutputFormat::name()
1491 {
1492         return valid_string(sr_output_id_get(_structure));
1493 }
1494
1495 string OutputFormat::description()
1496 {
1497         return valid_string(sr_output_description_get(_structure));
1498 }
1499
1500 vector<string> OutputFormat::extensions()
1501 {
1502         vector<string> exts;
1503         for (const char *const *e = sr_output_extensions_get(_structure);
1504                 e && *e; e++)
1505                 exts.push_back(*e);
1506         return exts;
1507 }
1508
1509 map<string, shared_ptr<Option>> OutputFormat::options()
1510 {
1511         const struct sr_option **options = sr_output_options_get(_structure);
1512         map<string, shared_ptr<Option>> result;
1513         if (options)
1514         {
1515                 auto option_array = shared_ptr<const struct sr_option *>(
1516                         options, sr_output_options_free);
1517                 for (int i = 0; options[i]; i++)
1518                         result[options[i]->id] = shared_ptr<Option>(
1519                                 new Option(options[i], option_array), Option::Deleter());
1520         }
1521         return result;
1522 }
1523
1524 shared_ptr<Output> OutputFormat::create_output(
1525         shared_ptr<Device> device, map<string, Glib::VariantBase> options)
1526 {
1527         return shared_ptr<Output>(
1528                 new Output(shared_from_this(), device, options),
1529                 Output::Deleter());
1530 }
1531
1532 shared_ptr<Output> OutputFormat::create_output(string filename,
1533         shared_ptr<Device> device, map<string, Glib::VariantBase> options)
1534 {
1535         return shared_ptr<Output>(
1536                 new Output(filename, shared_from_this(), device, options),
1537                 Output::Deleter());
1538 }
1539
1540 bool OutputFormat::test_flag(const OutputFlag *flag)
1541 {
1542         return sr_output_test_flag(_structure, flag->id());
1543 }
1544
1545 Output::Output(shared_ptr<OutputFormat> format,
1546                 shared_ptr<Device> device, map<string, Glib::VariantBase> options) :
1547         UserOwned(sr_output_new(format->_structure,
1548                 map_to_hash_variant(options), device->_structure, NULL)),
1549         _format(format),
1550         _device(device),
1551         _options(options)
1552 {
1553 }
1554
1555 Output::Output(string filename, shared_ptr<OutputFormat> format,
1556                 shared_ptr<Device> device, map<string, Glib::VariantBase> options) :
1557         UserOwned(sr_output_new(format->_structure,
1558                 map_to_hash_variant(options), device->_structure, filename.c_str())),
1559         _format(format),
1560         _device(device),
1561         _options(options)
1562 {
1563 }
1564
1565 Output::~Output()
1566 {
1567         check(sr_output_free(_structure));
1568 }
1569
1570 string Output::receive(shared_ptr<Packet> packet)
1571 {
1572         GString *out;
1573         check(sr_output_send(_structure, packet->_structure, &out));
1574         if (out)
1575         {
1576                 auto result = string(out->str, out->str + out->len);
1577                 g_string_free(out, true);
1578                 return result;
1579         }
1580         else
1581         {
1582                 return string();
1583         }
1584 }
1585
1586 #include <enums.cpp>
1587
1588 }