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