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