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