]> sigrok.org Git - libsigrok.git/blob - bindings/cxx/classes.cpp
classes.cpp: Define _XOPEN_SOURCE for isascii()
[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         _saving(false)
886 {
887         check(sr_session_new(context->_structure, &_structure));
888         _context->_session = this;
889 }
890
891 Session::Session(shared_ptr<Context> context, string filename) :
892         UserOwned(_structure),
893         _context(context),
894         _filename(filename),
895         _saving(false)
896 {
897         check(sr_session_load(context->_structure, filename.c_str(), &_structure));
898         GSList *dev_list;
899         check(sr_session_dev_list(_structure, &dev_list));
900         for (GSList *dev = dev_list; dev; dev = dev->next)
901         {
902                 auto sdi = (struct sr_dev_inst *) dev->data;
903                 _owned_devices[sdi] = new SessionDevice(sdi);
904         }
905         _context->_session = this;
906 }
907
908 Session::~Session()
909 {
910         check(sr_session_destroy(_structure));
911
912         for (auto callback : _datafeed_callbacks)
913                 delete callback;
914
915         for (auto entry : _source_callbacks)
916                 delete entry.second;
917
918         for (auto entry : _owned_devices)
919                 delete entry.second;
920 }
921
922 shared_ptr<Device> Session::get_device(const struct sr_dev_inst *sdi)
923 {
924         if (_owned_devices.count(sdi))
925                 return static_pointer_cast<Device>(
926                         _owned_devices[sdi]->get_shared_pointer(this));
927         else if (_other_devices.count(sdi))
928                 return _other_devices[sdi];
929         else
930                 throw Error(SR_ERR_BUG);
931 }
932
933 void Session::add_device(shared_ptr<Device> device)
934 {
935         check(sr_session_dev_add(_structure, device->_structure));
936         _other_devices[device->_structure] = device;
937 }
938
939 vector<shared_ptr<Device>> Session::devices()
940 {
941         GSList *dev_list;
942         check(sr_session_dev_list(_structure, &dev_list));
943         vector<shared_ptr<Device>> result;
944         for (GSList *dev = dev_list; dev; dev = dev->next)
945         {
946                 auto sdi = (struct sr_dev_inst *) dev->data;
947                 result.push_back(get_device(sdi));
948         }
949         return result;
950 }
951
952 void Session::remove_devices()
953 {
954         _other_devices.clear();
955         check(sr_session_dev_remove_all(_structure));
956 }
957
958 void Session::start()
959 {
960         check(sr_session_start(_structure));
961 }
962
963 void Session::run()
964 {
965         check(sr_session_run(_structure));
966 }
967
968 void Session::stop()
969 {
970         check(sr_session_stop(_structure));
971 }
972
973 void Session::begin_save(string filename)
974 {
975         _saving = true;
976         _save_initialized = false;
977         _save_filename = filename;
978         _save_samplerate = 0;
979 }
980
981 void Session::append(shared_ptr<Packet> packet)
982 {
983         if (!_saving)
984                 throw Error(SR_ERR);
985
986         switch (packet->_structure->type)
987         {
988                 case SR_DF_META:
989                 {
990                         auto meta = (const struct sr_datafeed_meta *)
991                                 packet->_structure->payload;
992
993                         for (auto l = meta->config; l; l = l->next)
994                         {
995                                 auto config = (struct sr_config *) l->data;
996                                 if (config->key == SR_CONF_SAMPLERATE)
997                                         _save_samplerate = g_variant_get_uint64(config->data);
998                         }
999
1000                         break;
1001                 }
1002                 case SR_DF_LOGIC:
1003                 {
1004                         if (_save_samplerate == 0)
1005                         {
1006                                 GVariant *samplerate;
1007
1008                                 check(sr_config_get(sr_dev_inst_driver_get(packet->_device->_structure),
1009                                         packet->_device->_structure, NULL, SR_CONF_SAMPLERATE,
1010                                         &samplerate));
1011
1012                                 _save_samplerate = g_variant_get_uint64(samplerate);
1013
1014                                 g_variant_unref(samplerate);
1015                         }
1016
1017                         if (!_save_initialized)
1018                         {
1019                                 vector<shared_ptr<Channel>> save_channels;
1020
1021                                 for (auto channel : packet->_device->channels())
1022                                         if (channel->_structure->enabled &&
1023                                                         channel->_structure->type == SR_CHANNEL_LOGIC)
1024                                                 save_channels.push_back(channel);
1025
1026                                 auto channels = g_new(char *, save_channels.size());
1027
1028                                 int i = 0;
1029                                 for (auto channel : save_channels)
1030                                                 channels[i++] = channel->_structure->name;
1031                                 channels[i] = NULL;
1032
1033                                 int ret = sr_session_save_init(_structure, _save_filename.c_str(),
1034                                                 _save_samplerate, channels);
1035
1036                                 g_free(channels);
1037
1038                                 if (ret != SR_OK)
1039                                         throw Error(ret);
1040
1041                                 _save_initialized = true;
1042                         }
1043
1044                         auto logic = (const struct sr_datafeed_logic *)
1045                                 packet->_structure->payload;
1046
1047                         check(sr_session_append(_structure, _save_filename.c_str(),
1048                                 (uint8_t *) logic->data, logic->unitsize,
1049                                 logic->length / logic->unitsize));
1050                 }
1051         }
1052 }
1053
1054 void Session::append(void *data, size_t length, unsigned int unit_size)
1055 {
1056         check(sr_session_append(_structure, _save_filename.c_str(),
1057                 (uint8_t *) data, unit_size, length));
1058 }
1059
1060 static void datafeed_callback(const struct sr_dev_inst *sdi,
1061         const struct sr_datafeed_packet *pkt, void *cb_data)
1062 {
1063         auto callback = static_cast<DatafeedCallbackData *>(cb_data);
1064         callback->run(sdi, pkt);
1065 }
1066         
1067 void Session::add_datafeed_callback(DatafeedCallbackFunction callback)
1068 {
1069         auto cb_data = new DatafeedCallbackData(this, callback);
1070         check(sr_session_datafeed_callback_add(_structure,
1071                 datafeed_callback, cb_data));
1072         _datafeed_callbacks.push_back(cb_data);
1073 }
1074
1075 void Session::remove_datafeed_callbacks(void)
1076 {
1077         check(sr_session_datafeed_callback_remove_all(_structure));
1078         for (auto callback : _datafeed_callbacks)
1079                 delete callback;
1080         _datafeed_callbacks.clear();
1081 }
1082
1083 static int source_callback(int fd, int revents, void *cb_data)
1084 {
1085         (void) fd;
1086         auto callback = (SourceCallbackData *) cb_data;
1087         return callback->run(revents);
1088 }
1089
1090 void Session::add_source(shared_ptr<EventSource> source)
1091 {
1092         if (_source_callbacks.count(source) == 1)
1093                 throw Error(SR_ERR_ARG);
1094
1095         auto cb_data = new SourceCallbackData(source);
1096
1097         switch (source->_type)
1098         {
1099                 case EventSource::SOURCE_FD:
1100                         check(sr_session_source_add(_structure, source->_fd, source->_events,
1101                                 source->_timeout, source_callback, cb_data));
1102                         break;
1103                 case EventSource::SOURCE_POLLFD:
1104                         check(sr_session_source_add_pollfd(_structure,
1105                                 source->_pollfd.gobj(), source->_timeout, source_callback,
1106                                 cb_data));
1107                         break;
1108                 case EventSource::SOURCE_IOCHANNEL:
1109                         check(sr_session_source_add_channel(_structure,
1110                                 source->_channel->gobj(), source->_events, source->_timeout,
1111                                 source_callback, cb_data));
1112                         break;
1113         }
1114
1115         _source_callbacks[source] = cb_data;
1116 }
1117
1118 void Session::remove_source(shared_ptr<EventSource> source)
1119 {
1120         if (_source_callbacks.count(source) == 0)
1121                 throw Error(SR_ERR_ARG);
1122
1123         switch (source->_type)
1124         {
1125                 case EventSource::SOURCE_FD:
1126                         check(sr_session_source_remove(_structure, source->_fd));
1127                         break;
1128                 case EventSource::SOURCE_POLLFD:
1129                         check(sr_session_source_remove_pollfd(_structure,
1130                                 source->_pollfd.gobj()));
1131                         break;
1132                 case EventSource::SOURCE_IOCHANNEL:
1133                         check(sr_session_source_remove_channel(_structure,
1134                                 source->_channel->gobj()));
1135                         break;
1136         }
1137
1138         delete _source_callbacks[source];
1139
1140         _source_callbacks.erase(source);
1141 }
1142
1143 shared_ptr<Trigger> Session::trigger()
1144 {
1145         return _trigger;
1146 }
1147
1148 void Session::set_trigger(shared_ptr<Trigger> trigger)
1149 {
1150         if (!trigger)
1151                 // Set NULL trigger, i.e. remove any trigger from the session.
1152                 check(sr_session_trigger_set(_structure, NULL));
1153         else
1154                 check(sr_session_trigger_set(_structure, trigger->_structure));
1155         _trigger = trigger;
1156 }
1157
1158 string Session::filename()
1159 {
1160         return _filename;
1161 }
1162
1163 shared_ptr<Context> Session::context()
1164 {
1165         return _context;
1166 }
1167
1168 Packet::Packet(shared_ptr<Device> device,
1169         const struct sr_datafeed_packet *structure) :
1170         UserOwned(structure),
1171         _device(device)
1172 {
1173         switch (structure->type)
1174         {
1175                 case SR_DF_HEADER:
1176                         _payload = new Header(
1177                                 static_cast<const struct sr_datafeed_header *>(
1178                                         structure->payload));
1179                         break;
1180                 case SR_DF_META:
1181                         _payload = new Meta(
1182                                 static_cast<const struct sr_datafeed_meta *>(
1183                                         structure->payload));
1184                         break;
1185                 case SR_DF_LOGIC:
1186                         _payload = new Logic(
1187                                 static_cast<const struct sr_datafeed_logic *>(
1188                                         structure->payload));
1189                         break;
1190                 case SR_DF_ANALOG:
1191                         _payload = new Analog(
1192                                 static_cast<const struct sr_datafeed_analog *>(
1193                                         structure->payload));
1194                         break;
1195                 default:
1196                         _payload = nullptr;
1197                         break;
1198         }
1199 }
1200
1201 Packet::~Packet()
1202 {
1203         if (_payload)
1204                 delete _payload;
1205 }
1206
1207 const PacketType *Packet::type()
1208 {
1209         return PacketType::get(_structure->type);
1210 }
1211
1212 shared_ptr<PacketPayload> Packet::payload()
1213 {
1214         if (_payload)
1215                 return _payload->get_shared_pointer(this);
1216         else
1217                 throw Error(SR_ERR_NA);
1218 }
1219
1220 PacketPayload::PacketPayload()
1221 {
1222 }
1223
1224 PacketPayload::~PacketPayload()
1225 {
1226 }
1227
1228 Header::Header(const struct sr_datafeed_header *structure) :
1229         ParentOwned(structure),
1230         PacketPayload()
1231 {
1232 }
1233
1234 Header::~Header()
1235 {
1236 }
1237
1238 shared_ptr<PacketPayload> Header::get_shared_pointer(Packet *_parent)
1239 {
1240         return static_pointer_cast<PacketPayload>(
1241                 ParentOwned::get_shared_pointer(_parent));
1242 }
1243
1244 int Header::feed_version()
1245 {
1246         return _structure->feed_version;
1247 }
1248
1249 Glib::TimeVal Header::start_time()
1250 {
1251         return Glib::TimeVal(
1252                 _structure->starttime.tv_sec,
1253                 _structure->starttime.tv_usec);
1254 }
1255
1256 Meta::Meta(const struct sr_datafeed_meta *structure) :
1257         ParentOwned(structure),
1258         PacketPayload()
1259 {
1260 }
1261
1262 Meta::~Meta()
1263 {
1264 }
1265
1266 shared_ptr<PacketPayload> Meta::get_shared_pointer(Packet *_parent)
1267 {
1268         return static_pointer_cast<PacketPayload>(
1269                 ParentOwned::get_shared_pointer(_parent));
1270 }
1271
1272 map<const ConfigKey *, Glib::VariantBase> Meta::config()
1273 {
1274         map<const ConfigKey *, Glib::VariantBase> result;
1275         for (auto l = _structure->config; l; l = l->next)
1276         {
1277                 auto config = (struct sr_config *) l->data;
1278                 result[ConfigKey::get(config->key)] = Glib::VariantBase(config->data);
1279         }
1280         return result;
1281 }
1282
1283 Logic::Logic(const struct sr_datafeed_logic *structure) :
1284         ParentOwned(structure),
1285         PacketPayload()
1286 {
1287 }
1288
1289 Logic::~Logic()
1290 {
1291 }
1292
1293 shared_ptr<PacketPayload> Logic::get_shared_pointer(Packet *_parent)
1294 {
1295         return static_pointer_cast<PacketPayload>(
1296                 ParentOwned::get_shared_pointer(_parent));
1297 }
1298
1299 void *Logic::data_pointer()
1300 {
1301         return _structure->data;
1302 }
1303
1304 size_t Logic::data_length()
1305 {
1306         return _structure->length;
1307 }
1308
1309 unsigned int Logic::unit_size()
1310 {
1311         return _structure->unitsize;
1312 }
1313
1314 Analog::Analog(const struct sr_datafeed_analog *structure) :
1315         ParentOwned(structure),
1316         PacketPayload()
1317 {
1318 }
1319
1320 Analog::~Analog()
1321 {
1322 }
1323
1324 shared_ptr<PacketPayload> Analog::get_shared_pointer(Packet *_parent)
1325 {
1326         return static_pointer_cast<PacketPayload>(
1327                 ParentOwned::get_shared_pointer(_parent));
1328 }
1329
1330 float *Analog::data_pointer()
1331 {
1332         return _structure->data;
1333 }
1334
1335 unsigned int Analog::num_samples()
1336 {
1337         return _structure->num_samples;
1338 }
1339
1340 vector<shared_ptr<Channel>> Analog::channels()
1341 {
1342         vector<shared_ptr<Channel>> result;
1343         for (auto l = _structure->channels; l; l = l->next)
1344                 result.push_back(_parent->_device->get_channel(
1345                         (struct sr_channel *)l->data));
1346         return result;
1347 }
1348
1349 const Quantity *Analog::mq()
1350 {
1351         return Quantity::get(_structure->mq);
1352 }
1353
1354 const Unit *Analog::unit()
1355 {
1356         return Unit::get(_structure->unit);
1357 }
1358
1359 vector<const QuantityFlag *> Analog::mq_flags()
1360 {
1361         return QuantityFlag::flags_from_mask(_structure->mqflags);
1362 }
1363
1364 InputFormat::InputFormat(const struct sr_input_module *structure) :
1365         ParentOwned(structure)
1366 {
1367 }
1368
1369 InputFormat::~InputFormat()
1370 {
1371 }
1372
1373 string InputFormat::name()
1374 {
1375         return valid_string(sr_input_id_get(_structure));
1376 }
1377
1378 string InputFormat::description()
1379 {
1380         return valid_string(sr_input_description_get(_structure));
1381 }
1382
1383 vector<string> InputFormat::extensions()
1384 {
1385         vector<string> exts;
1386         for (const char *const *e = sr_input_extensions_get(_structure);
1387                 e && *e; e++)
1388                 exts.push_back(*e);
1389         return exts;
1390 }
1391
1392 map<string, shared_ptr<Option>> InputFormat::options()
1393 {
1394         const struct sr_option **options = sr_input_options_get(_structure);
1395         map<string, shared_ptr<Option>> result;
1396         if (options)
1397         {
1398                 auto option_array = shared_ptr<const struct sr_option *>(
1399                         options, sr_input_options_free);
1400                 for (int i = 0; options[i]; i++)
1401                         result[options[i]->id] = shared_ptr<Option>(
1402                                 new Option(options[i], option_array), Option::Deleter());
1403         }
1404         return result;
1405 }
1406
1407 shared_ptr<Input> InputFormat::create_input(
1408         map<string, Glib::VariantBase> options)
1409 {
1410         auto input = sr_input_new(_structure, map_to_hash_variant(options));
1411         if (!input)
1412                 throw Error(SR_ERR_ARG);
1413         return shared_ptr<Input>(
1414                 new Input(_parent->shared_from_this(), input), Input::Deleter());
1415 }
1416
1417 Input::Input(shared_ptr<Context> context, const struct sr_input *structure) :
1418         UserOwned(structure),
1419         _context(context),
1420         _device(nullptr)
1421 {
1422 }
1423
1424 shared_ptr<InputDevice> Input::device()
1425 {
1426         if (!_device)
1427         {
1428                 auto sdi = sr_input_dev_inst_get(_structure);
1429                 if (!sdi)
1430                         throw Error(SR_ERR_NA);
1431                 _device = new InputDevice(shared_from_this(), sdi);
1432         }
1433
1434         return _device->get_shared_pointer(shared_from_this());
1435 }
1436
1437 void Input::send(void *data, size_t length)
1438 {
1439         auto gstr = g_string_new_len((gchar *)data, length);
1440         auto ret = sr_input_send(_structure, gstr);
1441         g_string_free(gstr, false);
1442         check(ret);
1443 }
1444
1445 void Input::end()
1446 {
1447         check(sr_input_end(_structure));
1448 }
1449
1450 Input::~Input()
1451 {
1452         if (_device)
1453                 delete _device;
1454         sr_input_free(_structure);
1455 }
1456
1457 InputDevice::InputDevice(shared_ptr<Input> input,
1458                 struct sr_dev_inst *structure) :
1459         ParentOwned(structure),
1460         Device(structure),
1461         _input(input)
1462 {
1463 }
1464
1465 InputDevice::~InputDevice()
1466 {
1467 }
1468
1469 shared_ptr<Device> InputDevice::get_shared_from_this()
1470 {
1471         return static_pointer_cast<Device>(shared_from_this());
1472 }
1473
1474 Option::Option(const struct sr_option *structure,
1475                 shared_ptr<const struct sr_option *> structure_array) :
1476         UserOwned(structure),
1477         _structure_array(structure_array)
1478 {
1479 }
1480
1481 Option::~Option()
1482 {
1483 }
1484
1485 string Option::id()
1486 {
1487         return valid_string(_structure->id);
1488 }
1489
1490 string Option::name()
1491 {
1492         return valid_string(_structure->name);
1493 }
1494
1495 string Option::description()
1496 {
1497         return valid_string(_structure->desc);
1498 }
1499
1500 Glib::VariantBase Option::default_value()
1501 {
1502         return Glib::VariantBase(_structure->def, true);
1503 }
1504
1505 vector<Glib::VariantBase> Option::values()
1506 {
1507         vector<Glib::VariantBase> result;
1508         for (auto l = _structure->values; l; l = l->next)
1509                 result.push_back(Glib::VariantBase((GVariant *) l->data, true));
1510         return result;
1511 }
1512
1513 OutputFormat::OutputFormat(const struct sr_output_module *structure) :
1514         ParentOwned(structure)
1515 {
1516 }
1517
1518 OutputFormat::~OutputFormat()
1519 {
1520 }
1521
1522 string OutputFormat::name()
1523 {
1524         return valid_string(sr_output_id_get(_structure));
1525 }
1526
1527 string OutputFormat::description()
1528 {
1529         return valid_string(sr_output_description_get(_structure));
1530 }
1531
1532 vector<string> OutputFormat::extensions()
1533 {
1534         vector<string> exts;
1535         for (const char *const *e = sr_output_extensions_get(_structure);
1536                 e && *e; e++)
1537                 exts.push_back(*e);
1538         return exts;
1539 }
1540
1541 map<string, shared_ptr<Option>> OutputFormat::options()
1542 {
1543         const struct sr_option **options = sr_output_options_get(_structure);
1544         map<string, shared_ptr<Option>> result;
1545         if (options)
1546         {
1547                 auto option_array = shared_ptr<const struct sr_option *>(
1548                         options, sr_output_options_free);
1549                 for (int i = 0; options[i]; i++)
1550                         result[options[i]->id] = shared_ptr<Option>(
1551                                 new Option(options[i], option_array), Option::Deleter());
1552         }
1553         return result;
1554 }
1555
1556 shared_ptr<Output> OutputFormat::create_output(
1557         shared_ptr<Device> device, map<string, Glib::VariantBase> options)
1558 {
1559         return shared_ptr<Output>(
1560                 new Output(shared_from_this(), device, options),
1561                 Output::Deleter());
1562 }
1563
1564 shared_ptr<Output> OutputFormat::create_output(string filename,
1565         shared_ptr<Device> device, map<string, Glib::VariantBase> options)
1566 {
1567         return shared_ptr<Output>(
1568                 new Output(filename, shared_from_this(), device, options),
1569                 Output::Deleter());
1570 }
1571
1572 bool OutputFormat::test_flag(const OutputFlag *flag)
1573 {
1574         return sr_output_test_flag(_structure, flag->id());
1575 }
1576
1577 Output::Output(shared_ptr<OutputFormat> format,
1578                 shared_ptr<Device> device, map<string, Glib::VariantBase> options) :
1579         UserOwned(sr_output_new(format->_structure,
1580                 map_to_hash_variant(options), device->_structure, NULL)),
1581         _format(format),
1582         _device(device),
1583         _options(options)
1584 {
1585 }
1586
1587 Output::Output(string filename, shared_ptr<OutputFormat> format,
1588                 shared_ptr<Device> device, map<string, Glib::VariantBase> options) :
1589         UserOwned(sr_output_new(format->_structure,
1590                 map_to_hash_variant(options), device->_structure, filename.c_str())),
1591         _format(format),
1592         _device(device),
1593         _options(options)
1594 {
1595 }
1596
1597 Output::~Output()
1598 {
1599         check(sr_output_free(_structure));
1600 }
1601
1602 string Output::receive(shared_ptr<Packet> packet)
1603 {
1604         GString *out;
1605         check(sr_output_send(_structure, packet->_structure, &out));
1606         if (out)
1607         {
1608                 auto result = string(out->str, out->str + out->len);
1609                 g_string_free(out, true);
1610                 return result;
1611         }
1612         else
1613         {
1614                 return string();
1615         }
1616 }
1617
1618 #include <enums.cpp>
1619
1620 }