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