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