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