]> sigrok.org Git - libsigrok.git/blob - bindings/cxx/classes.cpp
C++: Rename StructureWrapper to ParentOwned.
[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 "libsigrok/libsigrok.hpp"
21
22 #include <sstream>
23
24 namespace sigrok
25 {
26
27 /** Helper function to translate C errors to C++ exceptions. */
28 static void check(int result)
29 {
30         if (result != SR_OK)
31                 throw Error(result);
32 }
33
34 /** Helper function to obtain valid strings from possibly null input. */
35 static const char *valid_string(const char *input)
36 {
37         if (input != NULL)
38                 return input;
39         else
40                 return "";
41 }
42
43 /** Helper function to convert between map<string, VariantBase> and GHashTable */
44 static GHashTable *map_to_hash_variant(map<string, Glib::VariantBase> input)
45 {
46         auto output = g_hash_table_new_full(
47                 g_str_hash, g_str_equal, g_free, (GDestroyNotify) g_variant_unref);
48         for (auto entry : input)
49                 g_hash_table_insert(output,
50                         g_strdup(entry.first.c_str()),
51                         entry.second.gobj_copy());
52     return output;
53 }
54
55 Error::Error(int result) : result(result)
56 {
57 }
58
59 const char *Error::what() const throw()
60 {
61         return sr_strerror(result);
62 }
63
64 Error::~Error() throw()
65 {
66 }
67
68 shared_ptr<Context> Context::create()
69 {
70         return shared_ptr<Context>(new Context(), Context::Deleter());
71 }
72
73 Context::Context() :
74         session(NULL)
75 {
76         check(sr_init(&structure));
77         struct sr_dev_driver **driver_list = sr_driver_list();
78         if (driver_list)
79                 for (int i = 0; driver_list[i]; i++)
80                         drivers[driver_list[i]->name] =
81                                 new Driver(driver_list[i]);
82         const struct sr_input_module **input_list = sr_input_list();
83         if (input_list)
84                 for (int i = 0; input_list[i]; i++)
85                         input_formats[sr_input_id_get(input_list[i])] =
86                                 new InputFormat(input_list[i]);
87         const struct sr_output_module **output_list = sr_output_list();
88         if (output_list)
89                 for (int i = 0; output_list[i]; i++)
90                         output_formats[sr_output_id_get(output_list[i])] =
91                                 new OutputFormat(output_list[i]);
92 }
93
94 string Context::get_package_version()
95 {
96         return sr_package_version_string_get();
97 }
98
99 string Context::get_lib_version()
100 {
101         return sr_lib_version_string_get();
102 }
103
104 map<string, shared_ptr<Driver>> Context::get_drivers()
105 {
106         map<string, shared_ptr<Driver>> result;
107         for (auto entry: drivers)
108         {
109                 auto name = entry.first;
110                 auto driver = entry.second;
111                 result[name] = driver->get_shared_pointer(this);
112         }
113         return result;
114 }
115
116 map<string, shared_ptr<InputFormat>> Context::get_input_formats()
117 {
118         map<string, shared_ptr<InputFormat>> result;
119         for (auto entry: input_formats)
120         {
121                 auto name = entry.first;
122                 auto input_format = entry.second;
123                 result[name] = input_format->get_shared_pointer(this);
124         }
125         return result;
126 }
127
128 map<string, shared_ptr<OutputFormat>> Context::get_output_formats()
129 {
130         map<string, shared_ptr<OutputFormat>> result;
131         for (auto entry: output_formats)
132         {
133                 auto name = entry.first;
134                 auto output_format = entry.second;
135                 result[name] = output_format->get_shared_pointer(this);
136         }
137         return result;
138 }
139
140 Context::~Context()
141 {
142         for (auto entry : drivers)
143                 delete entry.second;
144         for (auto entry : input_formats)
145                 delete entry.second;
146         for (auto entry : output_formats)
147                 delete entry.second;
148         check(sr_exit(structure));
149 }
150
151 const LogLevel *Context::get_log_level()
152 {
153         return LogLevel::get(sr_log_loglevel_get());
154 }
155
156 void Context::set_log_level(const LogLevel *level)
157 {
158         check(sr_log_loglevel_set(level->get_id()));
159 }
160
161 string Context::get_log_domain()
162 {
163         return valid_string(sr_log_logdomain_get());
164 }
165
166 void Context::set_log_domain(string value)
167 {
168         check(sr_log_logdomain_set(value.c_str()));
169 }
170
171 static int call_log_callback(void *cb_data, int loglevel, const char *format, va_list args)
172 {
173         va_list args_copy;
174         va_copy(args_copy, args);
175         int length = vsnprintf(NULL, 0, format, args_copy);
176         va_end(args_copy);
177         char *buf = (char *) g_malloc(length + 1);
178         vsprintf(buf, format, args);
179         string message(buf, length);
180         g_free(buf);
181
182         LogCallbackFunction callback = *((LogCallbackFunction *) cb_data);
183
184         try
185         {
186                 callback(LogLevel::get(loglevel), message);
187         }
188         catch (Error e)
189         {
190                 return e.result;
191         }
192
193         return SR_OK;
194 }
195
196 void Context::set_log_callback(LogCallbackFunction callback)
197 {
198         log_callback = callback;
199         check(sr_log_callback_set(call_log_callback, &log_callback));
200
201
202 void Context::set_log_callback_default()
203 {
204         check(sr_log_callback_set_default());
205         log_callback = nullptr;
206
207
208 shared_ptr<Session> Context::create_session()
209 {
210         return shared_ptr<Session>(
211                 new Session(shared_from_this()), Session::Deleter());
212 }
213
214 shared_ptr<Session> Context::load_session(string filename)
215 {
216         return shared_ptr<Session>(
217                 new Session(shared_from_this(), filename), Session::Deleter());
218 }
219
220 shared_ptr<Trigger> Context::create_trigger(string name)
221 {
222         return shared_ptr<Trigger>(
223                 new Trigger(shared_from_this(), name), Trigger::Deleter());
224 }
225
226 shared_ptr<Input> Context::open_file(string filename)
227 {
228         const struct sr_input *input;
229
230         check( sr_input_scan_file(filename.c_str(), &input));
231         return shared_ptr<Input>(
232                 new Input(shared_from_this(), input), Input::Deleter());
233 }
234
235 shared_ptr<Input> Context::open_stream(string header)
236 {
237         const struct sr_input *input;
238
239         auto gstr = g_string_new(header.c_str());
240         auto ret = sr_input_scan_buffer(gstr, &input);
241         g_string_free(gstr, true);
242         check(ret);
243         return shared_ptr<Input>(
244                 new Input(shared_from_this(), input), Input::Deleter());
245 }
246
247 Driver::Driver(struct sr_dev_driver *structure) :
248         ParentOwned(structure),
249         initialized(false)
250 {
251 }
252
253 Driver::~Driver()
254 {
255         for (auto device : devices)
256                 delete device;
257 }
258
259 string Driver::get_name()
260 {
261         return valid_string(structure->name);
262 }
263
264 string Driver::get_long_name()
265 {
266         return valid_string(structure->longname);
267 }
268
269 vector<shared_ptr<HardwareDevice>> Driver::scan(
270         map<const ConfigKey *, Glib::VariantBase> options)
271 {
272         /* Initialise the driver if not yet done. */
273         if (!initialized)
274         {
275                 check(sr_driver_init(parent->structure, structure));
276                 initialized = true;
277         }
278
279         /* Clear all existing instances. */
280         for (auto device : devices)
281                 delete device;
282         devices.clear();
283
284         /* Translate scan options to GSList of struct sr_config pointers. */
285         GSList *option_list = NULL;
286         for (auto entry : options)
287         {
288                 auto key = entry.first;
289                 auto value = entry.second;
290                 auto config = g_new(struct sr_config, 1);
291                 config->key = key->get_id();
292                 config->data = value.gobj();
293                 option_list = g_slist_append(option_list, config);
294         }
295
296         /* Run scan. */
297         GSList *device_list = sr_driver_scan(structure, option_list);
298
299         /* Free option list. */
300         g_slist_free_full(option_list, g_free);
301
302         /* Create device objects. */
303         for (GSList *device = device_list; device; device = device->next)
304         {
305                 auto sdi = (struct sr_dev_inst *) device->data;
306                 devices.push_back(new HardwareDevice(this, sdi));
307         }
308
309         /* Free GSList returned from scan. */
310         g_slist_free(device_list);
311
312         /* Create list of shared pointers to device instances for return. */
313         vector<shared_ptr<HardwareDevice>> result;
314         for (auto device : devices)
315                 result.push_back(device->get_shared_pointer(parent));
316         return result;
317 }
318
319 Configurable::Configurable(
320                 struct sr_dev_driver *driver,
321                 struct sr_dev_inst *sdi,
322                 struct sr_channel_group *cg) :
323         config_driver(driver),
324         config_sdi(sdi),
325         config_channel_group(cg)
326 {
327 }
328
329 Configurable::~Configurable()
330 {
331 }
332
333 Glib::VariantBase Configurable::config_get(const ConfigKey *key)
334 {
335         GVariant *data;
336         check(sr_config_get(
337                 config_driver, config_sdi, config_channel_group,
338                 key->get_id(), &data));
339         return Glib::VariantBase(data);
340 }
341
342 void Configurable::config_set(const ConfigKey *key, Glib::VariantBase value)
343 {
344         check(sr_config_set(
345                 config_sdi, config_channel_group,
346                 key->get_id(), value.gobj()));
347 }
348
349 Glib::VariantContainerBase Configurable::config_list(const ConfigKey *key)
350 {
351         GVariant *data;
352         check(sr_config_list(
353                 config_driver, config_sdi, config_channel_group,
354                 key->get_id(), &data));
355         return Glib::VariantContainerBase(data);
356 }
357
358 Device::Device(struct sr_dev_inst *structure) :
359         Configurable(structure->driver, structure, NULL),
360         structure(structure)
361 {
362         for (GSList *entry = structure->channels; entry; entry = entry->next)
363         {
364                 auto channel = (struct sr_channel *) entry->data;
365                 channels[channel] = new Channel(channel);
366         }
367
368         for (GSList *entry = structure->channel_groups; entry; entry = entry->next)
369         {
370                 auto group = (struct sr_channel_group *) entry->data;
371                 channel_groups[group->name] = new ChannelGroup(this, group);
372         }
373 }
374
375 Device::~Device()
376 {
377         for (auto entry : channels)
378                 delete entry.second;
379         for (auto entry : channel_groups)
380                 delete entry.second;
381 }
382
383 string Device::get_description()
384 {
385         ostringstream s;
386
387         vector<string> parts =
388                 {get_vendor(), get_model(), get_version()};
389
390         for (string part : parts)
391                 if (part.length() > 0)
392                         s << part;
393
394         return s.str();
395 }
396
397 string Device::get_vendor()
398 {
399         return valid_string(structure->vendor);
400 }
401
402 string Device::get_model()
403 {
404         return valid_string(structure->model);
405 }
406
407 string Device::get_version()
408 {
409         return valid_string(structure->version);
410 }
411
412 vector<shared_ptr<Channel>> Device::get_channels()
413 {
414         vector<shared_ptr<Channel>> result;
415         for (auto entry : channels)
416                 result.push_back(entry.second->get_shared_pointer(get_shared_from_this()));
417         return result;
418 }
419
420 shared_ptr<Channel> Device::get_channel(struct sr_channel *ptr)
421 {
422         return channels[ptr]->get_shared_pointer(get_shared_from_this());
423 }
424
425 map<string, shared_ptr<ChannelGroup>>
426 Device::get_channel_groups()
427 {
428         map<string, shared_ptr<ChannelGroup>> result;
429         for (auto entry: channel_groups)
430         {
431                 auto name = entry.first;
432                 auto channel_group = entry.second;
433                 result[name] = channel_group->get_shared_pointer(get_shared_from_this());
434         }
435         return result;
436 }
437
438 void Device::open()
439 {
440         check(sr_dev_open(structure));
441 }
442
443 void Device::close()
444 {
445         check(sr_dev_close(structure));
446 }
447
448 HardwareDevice::HardwareDevice(Driver *driver, struct sr_dev_inst *structure) :
449         ParentOwned(structure),
450         Device(structure),
451         driver(driver)
452 {
453 }
454
455 HardwareDevice::~HardwareDevice()
456 {
457 }
458
459 shared_ptr<Device> HardwareDevice::get_shared_from_this()
460 {
461         return static_pointer_cast<Device>(shared_from_this());
462 }
463
464 shared_ptr<Driver> HardwareDevice::get_driver()
465 {
466         return driver->get_shared_pointer(parent);
467 }
468
469 Channel::Channel(struct sr_channel *structure) :
470         ParentOwned(structure),
471         type(ChannelType::get(structure->type))
472 {
473 }
474
475 Channel::~Channel()
476 {
477 }
478
479 string Channel::get_name()
480 {
481         return valid_string(structure->name);
482 }
483
484 void Channel::set_name(string name)
485 {
486         check(sr_dev_channel_name_set(parent->structure, structure->index, name.c_str()));
487 }
488
489 const ChannelType *Channel::get_type()
490 {
491         return ChannelType::get(structure->type);
492 }
493
494 bool Channel::get_enabled()
495 {
496         return structure->enabled;
497 }
498
499 void Channel::set_enabled(bool value)
500 {
501         check(sr_dev_channel_enable(parent->structure, structure->index, value));
502 }
503
504 unsigned int Channel::get_index()
505 {
506         return structure->index;
507 }
508
509 ChannelGroup::ChannelGroup(Device *device,
510                 struct sr_channel_group *structure) :
511         ParentOwned(structure),
512         Configurable(device->structure->driver, device->structure, structure)
513 {
514         for (GSList *entry = structure->channels; entry; entry = entry->next)
515                 channels.push_back(device->channels[(struct sr_channel *)entry->data]);
516 }
517
518 ChannelGroup::~ChannelGroup()
519 {
520 }
521
522 string ChannelGroup::get_name()
523 {
524         return valid_string(structure->name);
525 }
526
527 vector<shared_ptr<Channel>> ChannelGroup::get_channels()
528 {
529         vector<shared_ptr<Channel>> result;
530         for (auto channel : channels)
531                 result.push_back(channel->get_shared_pointer(parent));
532         return result;
533 }
534
535 Trigger::Trigger(shared_ptr<Context> context, string name) : 
536         structure(sr_trigger_new(name.c_str())), context(context)
537 {
538         for (auto stage = structure->stages; stage; stage = stage->next)
539                 stages.push_back(new TriggerStage((struct sr_trigger_stage *) stage->data));
540 }
541
542 Trigger::~Trigger()
543 {
544         for (auto stage: stages)
545                 delete stage;
546
547         sr_trigger_free(structure);
548 }
549
550 string Trigger::get_name()
551 {
552         return structure->name;
553 }
554
555 vector<shared_ptr<TriggerStage>> Trigger::get_stages()
556 {
557         vector<shared_ptr<TriggerStage>> result;
558         for (auto stage : stages)
559                 result.push_back(stage->get_shared_pointer(this));
560         return result;
561 }
562
563 shared_ptr<TriggerStage> Trigger::add_stage()
564 {
565         auto stage = new TriggerStage(sr_trigger_stage_add(structure));
566         stages.push_back(stage);
567         return stage->get_shared_pointer(this);
568 }
569
570 TriggerStage::TriggerStage(struct sr_trigger_stage *structure) : 
571         ParentOwned(structure)
572 {
573 }
574
575 TriggerStage::~TriggerStage()
576 {
577         for (auto match : matches)
578                 delete match;
579 }
580         
581 int TriggerStage::get_number()
582 {
583         return structure->stage;
584 }
585
586 vector<shared_ptr<TriggerMatch>> TriggerStage::get_matches()
587 {
588         vector<shared_ptr<TriggerMatch>> result;
589         for (auto match : matches)
590                 result.push_back(match->get_shared_pointer(this));
591         return result;
592 }
593
594 void TriggerStage::add_match(shared_ptr<Channel> channel, const TriggerMatchType *type, float value)
595 {
596         check(sr_trigger_match_add(structure, channel->structure, type->get_id(), value));
597         matches.push_back(new TriggerMatch(
598                 (struct sr_trigger_match *) g_slist_last(structure->matches)->data, channel));
599 }
600
601 void TriggerStage::add_match(shared_ptr<Channel> channel, const TriggerMatchType *type)
602 {
603         add_match(channel, type, NAN);
604 }
605
606 TriggerMatch::TriggerMatch(struct sr_trigger_match *structure, shared_ptr<Channel> channel) :
607         ParentOwned(structure), channel(channel)
608 {
609 }
610
611 TriggerMatch::~TriggerMatch()
612 {
613 }
614
615 shared_ptr<Channel> TriggerMatch::get_channel()
616 {
617         return channel;
618 }
619
620 const TriggerMatchType *TriggerMatch::get_type()
621 {
622         return TriggerMatchType::get(structure->match);
623 }
624
625 float TriggerMatch::get_value()
626 {
627         return structure->value;
628 }
629
630 DatafeedCallbackData::DatafeedCallbackData(Session *session,
631                 DatafeedCallbackFunction callback) :
632         callback(callback), session(session)
633 {
634 }
635
636 void DatafeedCallbackData::run(const struct sr_dev_inst *sdi,
637         const struct sr_datafeed_packet *pkt)
638 {
639         auto device = session->devices[sdi];
640         auto packet = shared_ptr<Packet>(new Packet(device, pkt), Packet::Deleter());
641         callback(device, packet);
642 }
643
644 SourceCallbackData::SourceCallbackData(shared_ptr<EventSource> source) :
645         source(source)
646 {
647 }
648
649 bool SourceCallbackData::run(int revents)
650 {
651         return source->callback((Glib::IOCondition) revents);
652 }
653
654 shared_ptr<EventSource> EventSource::create(int fd, Glib::IOCondition events,
655         int timeout, SourceCallbackFunction callback)
656 {
657         auto result = new EventSource(timeout, callback);
658         result->type = EventSource::SOURCE_FD;
659         result->fd = fd;
660         result->events = events;
661         return shared_ptr<EventSource>(result, EventSource::Deleter());
662 }
663
664 shared_ptr<EventSource> EventSource::create(Glib::PollFD pollfd, int timeout,
665         SourceCallbackFunction callback)
666 {
667         auto result = new EventSource(timeout, callback);
668         result->type = EventSource::SOURCE_POLLFD;
669         result->pollfd = pollfd;
670         return shared_ptr<EventSource>(result, EventSource::Deleter());
671 }
672
673 shared_ptr<EventSource> EventSource::create(Glib::RefPtr<Glib::IOChannel> channel,
674         Glib::IOCondition events, int timeout, SourceCallbackFunction callback)
675 {
676         auto result = new EventSource(timeout, callback);
677         result->type = EventSource::SOURCE_IOCHANNEL;
678         result->channel = channel;
679         result->events = events;
680         return shared_ptr<EventSource>(result, EventSource::Deleter());
681 }
682
683 EventSource::EventSource(int timeout, SourceCallbackFunction callback) :
684         timeout(timeout), callback(callback)
685 {
686 }
687
688 EventSource::~EventSource()
689 {
690 }
691
692 Session::Session(shared_ptr<Context> context) :
693         context(context), saving(false)
694 {
695         check(sr_session_new(&structure));
696         context->session = this;
697 }
698
699 Session::Session(shared_ptr<Context> context, string filename) :
700         context(context), saving(false)
701 {
702         check(sr_session_load(filename.c_str(), &structure));
703         context->session = this;
704 }
705
706 Session::~Session()
707 {
708         check(sr_session_destroy(structure));
709
710         for (auto callback : datafeed_callbacks)
711                 delete callback;
712
713         for (auto entry : source_callbacks)
714                 delete entry.second;
715 }
716
717 void Session::add_device(shared_ptr<Device> device)
718 {
719         check(sr_session_dev_add(structure, device->structure));
720         devices[device->structure] = device;
721 }
722
723 vector<shared_ptr<Device>> Session::get_devices()
724 {
725         GSList *dev_list;
726         check(sr_session_dev_list(structure, &dev_list));
727         vector<shared_ptr<Device>> result;
728         for (GSList *dev = dev_list; dev; dev = dev->next)
729         {
730                 auto sdi = (struct sr_dev_inst *) dev->data;
731                 result.push_back(devices[sdi]);
732         }
733         return result;
734 }
735
736 void Session::remove_devices()
737 {
738         devices.clear();
739         check(sr_session_dev_remove_all(structure));
740 }
741
742 void Session::start()
743 {
744         check(sr_session_start(structure));
745 }
746
747 void Session::run()
748 {
749         check(sr_session_run(structure));
750 }
751
752 void Session::stop()
753 {
754         check(sr_session_stop(structure));
755 }
756
757 void Session::begin_save(string filename)
758 {
759         saving = true;
760         save_initialized = false;
761         save_filename = filename;
762         save_samplerate = 0;
763 }
764
765 void Session::append(shared_ptr<Packet> packet)
766 {
767         if (!saving)
768                 throw Error(SR_ERR);
769
770         switch (packet->structure->type)
771         {
772                 case SR_DF_META:
773                 {
774                         auto meta = (const struct sr_datafeed_meta *)
775                                 packet->structure->payload;
776
777                         for (auto l = meta->config; l; l = l->next)
778                         {
779                                 auto config = (struct sr_config *) l->data;
780                                 if (config->key == SR_CONF_SAMPLERATE)
781                                         save_samplerate = g_variant_get_uint64(config->data);
782                         }
783
784                         break;
785                 }
786                 case SR_DF_LOGIC:
787                 {
788                         if (save_samplerate == 0)
789                         {
790                                 GVariant *samplerate;
791
792                                 check(sr_config_get(packet->device->structure->driver,
793                                         packet->device->structure, NULL, SR_CONF_SAMPLERATE,
794                                         &samplerate));
795
796                                 save_samplerate = g_variant_get_uint64(samplerate);
797
798                                 g_variant_unref(samplerate);
799                         }
800
801                         if (!save_initialized)
802                         {
803                                 vector<shared_ptr<Channel>> save_channels;
804
805                                 for (auto channel : packet->device->get_channels())
806                                         if (channel->structure->enabled &&
807                                                         channel->structure->type == SR_CHANNEL_LOGIC)
808                                                 save_channels.push_back(channel);
809
810                                 auto channels = g_new(char *, save_channels.size());
811
812                                 int i = 0;
813                                 for (auto channel : save_channels)
814                                                 channels[i++] = channel->structure->name;
815                                 channels[i] = NULL;
816
817                                 int ret = sr_session_save_init(structure, save_filename.c_str(),
818                                                 save_samplerate, channels);
819
820                                 g_free(channels);
821
822                                 if (ret != SR_OK)
823                                         throw Error(ret);
824
825                                 save_initialized = true;
826                         }
827
828                         auto logic = (const struct sr_datafeed_logic *)
829                                 packet->structure->payload;
830
831                         check(sr_session_append(structure, save_filename.c_str(),
832                                 (uint8_t *) logic->data, logic->unitsize,
833                                 logic->length / logic->unitsize));
834                 }
835         }
836 }
837
838 void Session::append(void *data, size_t length, unsigned int unit_size)
839 {
840         check(sr_session_append(structure, save_filename.c_str(),
841                 (uint8_t *) data, unit_size, length));
842 }
843
844 static void datafeed_callback(const struct sr_dev_inst *sdi,
845         const struct sr_datafeed_packet *pkt, void *cb_data)
846 {
847         auto callback = static_cast<DatafeedCallbackData *>(cb_data);
848         callback->run(sdi, pkt);
849 }
850         
851 void Session::add_datafeed_callback(DatafeedCallbackFunction callback)
852 {
853         auto cb_data = new DatafeedCallbackData(this, callback);
854         check(sr_session_datafeed_callback_add(structure, datafeed_callback, cb_data));
855         datafeed_callbacks.push_back(cb_data);
856 }
857
858 void Session::remove_datafeed_callbacks(void)
859 {
860         check(sr_session_datafeed_callback_remove_all(structure));
861         for (auto callback : datafeed_callbacks)
862                 delete callback;
863         datafeed_callbacks.clear();
864 }
865
866 static int source_callback(int fd, int revents, void *cb_data)
867 {
868         (void) fd;
869         auto callback = (SourceCallbackData *) cb_data;
870         return callback->run(revents);
871 }
872
873 void Session::add_source(shared_ptr<EventSource> source)
874 {
875         if (source_callbacks.count(source) == 1)
876                 throw Error(SR_ERR_ARG);
877
878         auto cb_data = new SourceCallbackData(source);
879
880         switch (source->type)
881         {
882                 case EventSource::SOURCE_FD:
883                         check(sr_session_source_add(structure, source->fd, source->events,
884                                 source->timeout, source_callback, cb_data));
885                         break;
886                 case EventSource::SOURCE_POLLFD:
887                         check(sr_session_source_add_pollfd(structure,
888                                 source->pollfd.gobj(), source->timeout, source_callback,
889                                 cb_data));
890                         break;
891                 case EventSource::SOURCE_IOCHANNEL:
892                         check(sr_session_source_add_channel(structure,
893                                 source->channel->gobj(), source->events, source->timeout,
894                                 source_callback, cb_data));
895                         break;
896         }
897
898         source_callbacks[source] = cb_data;
899 }
900
901 void Session::remove_source(shared_ptr<EventSource> source)
902 {
903         if (source_callbacks.count(source) == 0)
904                 throw Error(SR_ERR_ARG);
905
906         switch (source->type)
907         {
908                 case EventSource::SOURCE_FD:
909                         check(sr_session_source_remove(structure, source->fd));
910                         break;
911                 case EventSource::SOURCE_POLLFD:
912                         check(sr_session_source_remove_pollfd(structure,
913                                 source->pollfd.gobj()));
914                         break;
915                 case EventSource::SOURCE_IOCHANNEL:
916                         check(sr_session_source_remove_channel(structure,
917                                 source->channel->gobj()));
918                         break;
919         }
920
921         delete source_callbacks[source];
922
923         source_callbacks.erase(source);
924 }
925
926 shared_ptr<Trigger> Session::get_trigger()
927 {
928         return trigger;
929 }
930
931 void Session::set_trigger(shared_ptr<Trigger> trigger)
932 {
933         check(sr_session_trigger_set(structure, trigger->structure));
934         this->trigger = trigger;
935 }
936
937 Packet::Packet(shared_ptr<Device> device,
938         const struct sr_datafeed_packet *structure) :
939         structure(structure),
940         device(device)
941 {
942         switch (structure->type)
943         {
944                 case SR_DF_HEADER:
945                         payload = new Header(
946                                 static_cast<const struct sr_datafeed_header *>(
947                                         structure->payload));
948                         break;
949                 case SR_DF_META:
950                         payload = new Meta(
951                                 static_cast<const struct sr_datafeed_meta *>(
952                                         structure->payload));
953                         break;
954                 case SR_DF_LOGIC:
955                         payload = new Logic(
956                                 static_cast<const struct sr_datafeed_logic *>(
957                                         structure->payload));
958                         break;
959                 case SR_DF_ANALOG:
960                         payload = new Analog(
961                                 static_cast<const struct sr_datafeed_analog *>(
962                                         structure->payload));
963                         break;
964                 default:
965                         payload = nullptr;
966                         break;
967         }
968 }
969
970 Packet::~Packet()
971 {
972         if (payload)
973                 delete payload;
974 }
975
976 const PacketType *Packet::get_type()
977 {
978         return PacketType::get(structure->type);
979 }
980
981 shared_ptr<PacketPayload> Packet::get_payload()
982 {
983         if (payload)
984                 return payload->get_shared_pointer(this);
985         else
986                 throw Error(SR_ERR_NA);
987 }
988
989 PacketPayload::PacketPayload()
990 {
991 }
992
993 PacketPayload::~PacketPayload()
994 {
995 }
996
997 Header::Header(const struct sr_datafeed_header *structure) :
998         ParentOwned(structure),
999         PacketPayload()
1000 {
1001 }
1002
1003 Header::~Header()
1004 {
1005 }
1006
1007 shared_ptr<PacketPayload> Header::get_shared_pointer(Packet *parent)
1008 {
1009         return static_pointer_cast<PacketPayload>(get_shared_pointer(parent));
1010 }
1011
1012 int Header::get_feed_version()
1013 {
1014         return structure->feed_version;
1015 }
1016
1017 Glib::TimeVal Header::get_start_time()
1018 {
1019         return Glib::TimeVal(
1020                 structure->starttime.tv_sec,
1021                 structure->starttime.tv_usec);
1022 }
1023
1024 Meta::Meta(const struct sr_datafeed_meta *structure) :
1025         ParentOwned(structure),
1026         PacketPayload()
1027 {
1028 }
1029
1030 Meta::~Meta()
1031 {
1032 }
1033
1034 shared_ptr<PacketPayload> Meta::get_shared_pointer(Packet *parent)
1035 {
1036         return static_pointer_cast<PacketPayload>(get_shared_pointer(parent));
1037 }
1038
1039 map<const ConfigKey *, Glib::VariantBase> Meta::get_config()
1040 {
1041         map<const ConfigKey *, Glib::VariantBase> result;
1042         for (auto l = structure->config; l; l = l->next)
1043         {
1044                 auto config = (struct sr_config *) l->data;
1045                 result[ConfigKey::get(config->key)] = Glib::VariantBase(config->data);
1046         }
1047         return result;
1048 }
1049
1050 Logic::Logic(const struct sr_datafeed_logic *structure) :
1051         ParentOwned(structure),
1052         PacketPayload()
1053 {
1054 }
1055
1056 Logic::~Logic()
1057 {
1058 }
1059
1060 shared_ptr<PacketPayload> Logic::get_shared_pointer(Packet *parent)
1061 {
1062         return static_pointer_cast<PacketPayload>(get_shared_pointer(parent));
1063 }
1064
1065 void *Logic::get_data_pointer()
1066 {
1067         return structure->data;
1068 }
1069
1070 size_t Logic::get_data_length()
1071 {
1072         return structure->length;
1073 }
1074
1075 unsigned int Logic::get_unit_size()
1076 {
1077         return structure->unitsize;
1078 }
1079
1080 Analog::Analog(const struct sr_datafeed_analog *structure) :
1081         ParentOwned(structure),
1082         PacketPayload()
1083 {
1084 }
1085
1086 Analog::~Analog()
1087 {
1088 }
1089
1090 shared_ptr<PacketPayload> Analog::get_shared_pointer(Packet *parent)
1091 {
1092         return static_pointer_cast<PacketPayload>(get_shared_pointer(parent));
1093 }
1094
1095 float *Analog::get_data_pointer()
1096 {
1097         return structure->data;
1098 }
1099
1100 unsigned int Analog::get_num_samples()
1101 {
1102         return structure->num_samples;
1103 }
1104
1105 vector<shared_ptr<Channel>> Analog::get_channels()
1106 {
1107         vector<shared_ptr<Channel>> result;
1108         for (auto l = structure->channels; l; l = l->next)
1109                 result.push_back(parent->device->get_channel(
1110                         (struct sr_channel *)l->data));
1111         return result;
1112 }
1113
1114 const Quantity *Analog::get_mq()
1115 {
1116         return Quantity::get(structure->mq);
1117 }
1118
1119 const Unit *Analog::get_unit()
1120 {
1121         return Unit::get(structure->unit);
1122 }
1123
1124 vector<const QuantityFlag *> Analog::get_mq_flags()
1125 {
1126         return QuantityFlag::flags_from_mask(structure->mqflags);
1127 }
1128
1129 InputFormat::InputFormat(const struct sr_input_module *structure) :
1130         ParentOwned(structure)
1131 {
1132 }
1133
1134 InputFormat::~InputFormat()
1135 {
1136 }
1137
1138 string InputFormat::get_name()
1139 {
1140         return valid_string(sr_input_id_get(structure));
1141 }
1142
1143 string InputFormat::get_description()
1144 {
1145         return valid_string(sr_input_description_get(structure));
1146 }
1147
1148 map<string, shared_ptr<Option>> InputFormat::get_options()
1149 {
1150         const struct sr_option **options = sr_input_options_get(structure);
1151         auto option_array = shared_ptr<const struct sr_option *>(
1152                 options, sr_input_options_free);
1153         map<string, shared_ptr<Option>> result;
1154         for (int i = 0; options[i]; i++)
1155                 result[options[i]->id] = shared_ptr<Option>(
1156                         new Option(options[i], option_array), Option::Deleter());
1157         return result;
1158 }
1159
1160 shared_ptr<Input> InputFormat::create_input(
1161         map<string, Glib::VariantBase> options)
1162 {
1163         auto input = sr_input_new(structure, map_to_hash_variant(options));
1164         if (!input)
1165                 throw Error(SR_ERR_ARG);
1166         return shared_ptr<Input>(
1167                 new Input(parent->shared_from_this(), input), Input::Deleter());
1168 }
1169
1170 Input::Input(shared_ptr<Context> context, const struct sr_input *structure) :
1171         structure(structure),
1172         context(context),
1173         device(nullptr)
1174 {
1175 }
1176
1177 shared_ptr<InputDevice> Input::get_device()
1178 {
1179         if (!device)
1180         {
1181                 auto sdi = sr_input_dev_inst_get(structure);
1182                 if (!sdi)
1183                         throw Error(SR_ERR_NA);
1184                 device = new InputDevice(shared_from_this(), sdi);
1185         }
1186
1187         return device->get_shared_pointer(shared_from_this());
1188 }
1189
1190 void Input::send(string data)
1191 {
1192         auto gstr = g_string_new(data.c_str());
1193         auto ret = sr_input_send(structure, gstr);
1194         g_string_free(gstr, false);
1195         check(ret);
1196 }
1197
1198 Input::~Input()
1199 {
1200         if (device)
1201                 delete device;
1202         check(sr_input_free(structure));
1203 }
1204
1205 InputDevice::InputDevice(shared_ptr<Input> input,
1206                 struct sr_dev_inst *structure) :
1207         ParentOwned(structure),
1208         Device(structure),
1209         input(input)
1210 {
1211 }
1212
1213 InputDevice::~InputDevice()
1214 {
1215 }
1216
1217 shared_ptr<Device> InputDevice::get_shared_from_this()
1218 {
1219         return static_pointer_cast<Device>(shared_from_this());
1220 }
1221
1222 Option::Option(const struct sr_option *structure,
1223                 shared_ptr<const struct sr_option *> structure_array) :
1224         structure(structure),
1225         structure_array(structure_array)
1226 {
1227 }
1228
1229 Option::~Option()
1230 {
1231 }
1232
1233 string Option::get_id()
1234 {
1235         return valid_string(structure->id);
1236 }
1237
1238 string Option::get_name()
1239 {
1240         return valid_string(structure->name);
1241 }
1242
1243 string Option::get_description()
1244 {
1245         return valid_string(structure->desc);
1246 }
1247
1248 Glib::VariantBase Option::get_default_value()
1249 {
1250         return Glib::VariantBase(structure->def, true);
1251 }
1252
1253 vector<Glib::VariantBase> Option::get_values()
1254 {
1255         vector<Glib::VariantBase> result;
1256         for (auto l = structure->values; l; l = l->next)
1257                 result.push_back(Glib::VariantBase((GVariant *) l->data, true));
1258         return result;
1259 }
1260
1261 OutputFormat::OutputFormat(const struct sr_output_module *structure) :
1262         ParentOwned(structure)
1263 {
1264 }
1265
1266 OutputFormat::~OutputFormat()
1267 {
1268 }
1269
1270 string OutputFormat::get_name()
1271 {
1272         return valid_string(sr_output_id_get(structure));
1273 }
1274
1275 string OutputFormat::get_description()
1276 {
1277         return valid_string(sr_output_description_get(structure));
1278 }
1279
1280 map<string, shared_ptr<Option>> OutputFormat::get_options()
1281 {
1282         const struct sr_option **options = sr_output_options_get(structure);
1283         auto option_array = shared_ptr<const struct sr_option *>(
1284                 options, sr_output_options_free);
1285         map<string, shared_ptr<Option>> result;
1286         for (int i = 0; options[i]; i++)
1287                 result[options[i]->id] = shared_ptr<Option>(
1288                         new Option(options[i], option_array), Option::Deleter());
1289         return result;
1290 }
1291
1292 shared_ptr<Output> OutputFormat::create_output(
1293         shared_ptr<Device> device, map<string, Glib::VariantBase> options)
1294 {
1295         return shared_ptr<Output>(
1296                 new Output(shared_from_this(), device, options),
1297                 Output::Deleter());
1298 }
1299
1300 Output::Output(shared_ptr<OutputFormat> format,
1301                 shared_ptr<Device> device, map<string, Glib::VariantBase> options) :
1302         structure(sr_output_new(format->structure,
1303                 map_to_hash_variant(options), device->structure)),
1304         format(format), device(device), options(options)
1305 {
1306 }
1307
1308 Output::~Output()
1309 {
1310         check(sr_output_free(structure));
1311 }
1312
1313 string Output::receive(shared_ptr<Packet> packet)
1314 {
1315         GString *out;
1316         check(sr_output_send(structure, packet->structure, &out));
1317         if (out)
1318         {
1319                 auto result = string(out->str, out->str + out->len);
1320                 g_string_free(out, true);
1321                 return result;
1322         }
1323         else
1324         {
1325                 return string();
1326         }
1327 }
1328
1329 #include "enums.cpp"
1330
1331 }