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