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] = static_pointer_cast<Driver>(
 112                         driver->get_shared_pointer(this));
 113         }
 114         return result;
 115 }
 116
 117 map<string, shared_ptr<InputFormat>> Context::get_input_formats()
 118 {
 119         map<string, shared_ptr<InputFormat>> result;
 120         for (auto entry: input_formats)
 121         {
 122                 auto name = entry.first;
 123                 auto input_format = entry.second;
 124                 result[name] = static_pointer_cast<InputFormat>(
 125                         input_format->get_shared_pointer(this));
 126         }
 127         return result;
 128 }
 129
 130 map<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;
 137                 result[name] = static_pointer_cast<OutputFormat>(
 138                         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::get_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->get_id()));
 162 }
 163
 164 string Context::get_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         StructureWrapper<Context, struct sr_dev_driver>(structure),
 252         initialized(false)
 253 {
 254 }
 255
 256 Driver::~Driver()
 257 {
 258         for (auto device : devices)
 259                 delete device;
 260 }
 261
 262 string Driver::get_name()
 263 {
 264         return valid_string(structure->name);
 265 }
 266
 267 string Driver::get_long_name()
 268 {
 269         return valid_string(structure->longname);
 270 }
 271
 272 vector<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)
 318                 result.push_back(static_pointer_cast<HardwareDevice>(
 319                         device->get_shared_pointer(parent)));
 320         return result;
 321 }
 322
 323 Configurable::Configurable(
 324                 struct sr_dev_driver *driver,
 325                 struct sr_dev_inst *sdi,
 326                 struct sr_channel_group *cg) :
 327         config_driver(driver),
 328         config_sdi(sdi),
 329         config_channel_group(cg)
 330 {
 331 }
 332
 333 Configurable::~Configurable()
 334 {
 335 }
 336
 337 Glib::VariantBase Configurable::config_get(const ConfigKey *key)
 338 {
 339         GVariant *data;
 340         check(sr_config_get(
 341                 config_driver, config_sdi, config_channel_group,
 342                 key->get_id(), &data));
 343         return Glib::VariantBase(data);
 344 }
 345
 346 void Configurable::config_set(const ConfigKey *key, Glib::VariantBase value)
 347 {
 348         check(sr_config_set(
 349                 config_sdi, config_channel_group,
 350                 key->get_id(), value.gobj()));
 351 }
 352
 353 Glib::VariantContainerBase Configurable::config_list(const ConfigKey *key)
 354 {
 355         GVariant *data;
 356         check(sr_config_list(
 357                 config_driver, config_sdi, config_channel_group,
 358                 key->get_id(), &data));
 359         return Glib::VariantContainerBase(data);
 360 }
 361
 362 Device::Device(struct sr_dev_inst *structure) :
 363         Configurable(structure->driver, structure, NULL),
 364         structure(structure)
 365 {
 366         for (GSList *entry = structure->channels; entry; entry = entry->next)
 367         {
 368                 auto channel = (struct sr_channel *) entry->data;
 369                 channels[channel] = new Channel(channel);
 370         }
 371
 372         for (GSList *entry = structure->channel_groups; entry; entry = entry->next)
 373         {
 374                 auto group = (struct sr_channel_group *) entry->data;
 375                 channel_groups[group->name] = new ChannelGroup(this, group);
 376         }
 377 }
 378
 379 Device::~Device()
 380 {
 381         for (auto entry : channels)
 382                 delete entry.second;
 383         for (auto entry : channel_groups)
 384                 delete entry.second;
 385 }
 386
 387 string Device::get_description()
 388 {
 389         ostringstream s;
 390
 391         vector<string> parts =
 392                 {get_vendor(), get_model(), get_version()};
 393
 394         for (string part : parts)
 395                 if (part.length() > 0)
 396                         s << part;
 397
 398         return s.str();
 399 }
 400
 401 string Device::get_vendor()
 402 {
 403         return valid_string(structure->vendor);
 404 }
 405
 406 string Device::get_model()
 407 {
 408         return valid_string(structure->model);
 409 }
 410
 411 string Device::get_version()
 412 {
 413         return valid_string(structure->version);
 414 }
 415
 416 vector<shared_ptr<Channel>> Device::get_channels()
 417 {
 418         vector<shared_ptr<Channel>> result;
 419         for (auto entry : channels)
 420                 result.push_back(static_pointer_cast<Channel>(
 421                         entry.second->get_shared_pointer(get_shared_from_this())));
 422         return result;
 423 }
 424
 425 shared_ptr<Channel> Device::get_channel(struct sr_channel *ptr)
 426 {
 427         return static_pointer_cast<Channel>(
 428                 channels[ptr]->get_shared_pointer(get_shared_from_this()));
 429 }
 430
 431 map<string, shared_ptr<ChannelGroup>>
 432 Device::get_channel_groups()
 433 {
 434         map<string, shared_ptr<ChannelGroup>> result;
 435         for (auto entry: channel_groups)
 436         {
 437                 auto name = entry.first;
 438                 auto channel_group = entry.second;
 439                 result[name] = static_pointer_cast<ChannelGroup>(
 440                         channel_group->get_shared_pointer(get_shared_from_this()));
 441         }
 442         return result;
 443 }
 444
 445 void Device::open()
 446 {
 447         check(sr_dev_open(structure));
 448 }
 449
 450 void Device::close()
 451 {
 452         check(sr_dev_close(structure));
 453 }
 454
 455 HardwareDevice::HardwareDevice(Driver *driver, struct sr_dev_inst *structure) :
 456         StructureWrapper(structure),
 457         Device(structure),
 458         driver(driver)
 459 {
 460 }
 461
 462 HardwareDevice::~HardwareDevice()
 463 {
 464 }
 465
 466 shared_ptr<Device> HardwareDevice::get_shared_from_this()
 467 {
 468         return static_pointer_cast<Device>(
 469                 static_pointer_cast<HardwareDevice>(shared_from_this()));
 470 }
 471
 472 shared_ptr<Driver> HardwareDevice::get_driver()
 473 {
 474         return static_pointer_cast<Driver>(driver->get_shared_pointer(parent));
 475 }
 476
 477 Channel::Channel(struct sr_channel *structure) :
 478         StructureWrapper<Device, struct sr_channel>(structure),
 479         type(ChannelType::get(structure->type))
 480 {
 481 }
 482
 483 Channel::~Channel()
 484 {
 485 }
 486
 487 string Channel::get_name()
 488 {
 489         return valid_string(structure->name);
 490 }
 491
 492 void Channel::set_name(string name)
 493 {
 494         check(sr_dev_channel_name_set(parent->structure, structure->index, name.c_str()));
 495 }
 496
 497 const ChannelType *Channel::get_type()
 498 {
 499         return ChannelType::get(structure->type);
 500 }
 501
 502 bool Channel::get_enabled()
 503 {
 504         return structure->enabled;
 505 }
 506
 507 void Channel::set_enabled(bool value)
 508 {
 509         check(sr_dev_channel_enable(parent->structure, structure->index, value));
 510 }
 511
 512 unsigned int Channel::get_index()
 513 {
 514         return structure->index;
 515 }
 516
 517 ChannelGroup::ChannelGroup(Device *device,
 518                 struct sr_channel_group *structure) :
 519         StructureWrapper<Device, struct sr_channel_group>(structure),
 520         Configurable(device->structure->driver, device->structure, structure)
 521 {
 522         for (GSList *entry = structure->channels; entry; entry = entry->next)
 523                 channels.push_back(device->channels[(struct sr_channel *)entry->data]);
 524 }
 525
 526 ChannelGroup::~ChannelGroup()
 527 {
 528 }
 529
 530 string ChannelGroup::get_name()
 531 {
 532         return valid_string(structure->name);
 533 }
 534
 535 vector<shared_ptr<Channel>> ChannelGroup::get_channels()
 536 {
 537         vector<shared_ptr<Channel>> result;
 538         for (auto channel : channels)
 539                 result.push_back(static_pointer_cast<Channel>(
 540                         channel->get_shared_pointer(parent)));
 541         return result;
 542 }
 543
 544 Trigger::Trigger(shared_ptr<Context> context, string name) :
 545         structure(sr_trigger_new(name.c_str())), context(context)
 546 {
 547         for (auto stage = structure->stages; stage; stage = stage->next)
 548                 stages.push_back(new TriggerStage((struct sr_trigger_stage *) stage->data));
 549 }
 550
 551 Trigger::~Trigger()
 552 {
 553         for (auto stage: stages)
 554                 delete stage;
 555
 556         sr_trigger_free(structure);
 557 }
 558
 559 string Trigger::get_name()
 560 {
 561         return structure->name;
 562 }
 563
 564 vector<shared_ptr<TriggerStage>> Trigger::get_stages()
 565 {
 566         vector<shared_ptr<TriggerStage>> result;
 567         for (auto stage : stages)
 568                 result.push_back(static_pointer_cast<TriggerStage>(
 569                         stage->get_shared_pointer(this)));
 570         return result;
 571 }
 572
 573 shared_ptr<TriggerStage> Trigger::add_stage()
 574 {
 575         auto stage = new TriggerStage(sr_trigger_stage_add(structure));
 576         stages.push_back(stage);
 577         return static_pointer_cast<TriggerStage>(
 578                 stage->get_shared_pointer(this));
 579 }
 580
 581 TriggerStage::TriggerStage(struct sr_trigger_stage *structure) :
 582         StructureWrapper<Trigger, struct sr_trigger_stage>(structure)
 583 {
 584 }
 585
 586 TriggerStage::~TriggerStage()
 587 {
 588         for (auto match : matches)
 589                 delete match;
 590 }
 591
 592 int TriggerStage::get_number()
 593 {
 594         return structure->stage;
 595 }
 596
 597 vector<shared_ptr<TriggerMatch>> TriggerStage::get_matches()
 598 {
 599         vector<shared_ptr<TriggerMatch>> result;
 600         for (auto match : matches)
 601                 result.push_back(static_pointer_cast<TriggerMatch>(
 602                         match->get_shared_pointer(this)));
 603         return result;
 604 }
 605
 606 void TriggerStage::add_match(shared_ptr<Channel> channel, const TriggerMatchType *type, float value)
 607 {
 608         check(sr_trigger_match_add(structure, channel->structure, type->get_id(), value));
 609         matches.push_back(new TriggerMatch(
 610                 (struct sr_trigger_match *) g_slist_last(structure->matches)->data, channel));
 611 }
 612
 613 void TriggerStage::add_match(shared_ptr<Channel> channel, const TriggerMatchType *type)
 614 {
 615         add_match(channel, type, NAN);
 616 }
 617
 618 TriggerMatch::TriggerMatch(struct sr_trigger_match *structure, shared_ptr<Channel> channel) :
 619         StructureWrapper<TriggerStage, struct sr_trigger_match>(structure), channel(channel)
 620 {
 621 }
 622
 623 TriggerMatch::~TriggerMatch()
 624 {
 625 }
 626
 627 shared_ptr<Channel> TriggerMatch::get_channel()
 628 {
 629         return channel;
 630 }
 631
 632 const TriggerMatchType *TriggerMatch::get_type()
 633 {
 634         return TriggerMatchType::get(structure->match);
 635 }
 636
 637 float TriggerMatch::get_value()
 638 {
 639         return structure->value;
 640 }
 641
 642 DatafeedCallbackData::DatafeedCallbackData(Session *session,
 643                 DatafeedCallbackFunction callback) :
 644         callback(callback), session(session)
 645 {
 646 }
 647
 648 void DatafeedCallbackData::run(const struct sr_dev_inst *sdi,
 649         const struct sr_datafeed_packet *pkt)
 650 {
 651         auto device = session->devices[sdi];
 652         auto packet = shared_ptr<Packet>(new Packet(device, pkt), Packet::Deleter());
 653         callback(device, packet);
 654 }
 655
 656 SourceCallbackData::SourceCallbackData(shared_ptr<EventSource> source) :
 657         source(source)
 658 {
 659 }
 660
 661 bool SourceCallbackData::run(int revents)
 662 {
 663         return source->callback((Glib::IOCondition) revents);
 664 }
 665
 666 shared_ptr<EventSource> EventSource::create(int fd, Glib::IOCondition events,
 667         int timeout, SourceCallbackFunction callback)
 668 {
 669         auto result = new EventSource(timeout, callback);
 670         result->type = EventSource::SOURCE_FD;
 671         result->fd = fd;
 672         result->events = events;
 673         return shared_ptr<EventSource>(result, EventSource::Deleter());
 674 }
 675
 676 shared_ptr<EventSource> EventSource::create(Glib::PollFD pollfd, int timeout,
 677         SourceCallbackFunction callback)
 678 {
 679         auto result = new EventSource(timeout, callback);
 680         result->type = EventSource::SOURCE_POLLFD;
 681         result->pollfd = pollfd;
 682         return shared_ptr<EventSource>(result, EventSource::Deleter());
 683 }
 684
 685 shared_ptr<EventSource> EventSource::create(Glib::RefPtr<Glib::IOChannel> channel,
 686         Glib::IOCondition events, int timeout, SourceCallbackFunction callback)
 687 {
 688         auto result = new EventSource(timeout, callback);
 689         result->type = EventSource::SOURCE_IOCHANNEL;
 690         result->channel = channel;
 691         result->events = events;
 692         return shared_ptr<EventSource>(result, EventSource::Deleter());
 693 }
 694
 695 EventSource::EventSource(int timeout, SourceCallbackFunction callback) :
 696         timeout(timeout), callback(callback)
 697 {
 698 }
 699
 700 EventSource::~EventSource()
 701 {
 702 }
 703
 704 Session::Session(shared_ptr<Context> context) :
 705         context(context), saving(false)
 706 {
 707         check(sr_session_new(&structure));
 708         context->session = this;
 709 }
 710
 711 Session::Session(shared_ptr<Context> context, string filename) :
 712         context(context), saving(false)
 713 {
 714         check(sr_session_load(filename.c_str(), &structure));
 715         context->session = this;
 716 }
 717
 718 Session::~Session()
 719 {
 720         check(sr_session_destroy(structure));
 721
 722         for (auto callback : datafeed_callbacks)
 723                 delete callback;
 724
 725         for (auto entry : source_callbacks)
 726                 delete entry.second;
 727 }
 728
 729 void Session::add_device(shared_ptr<Device> device)
 730 {
 731         check(sr_session_dev_add(structure, device->structure));
 732         devices[device->structure] = device;
 733 }
 734
 735 vector<shared_ptr<Device>> Session::get_devices()
 736 {
 737         GSList *dev_list;
 738         check(sr_session_dev_list(structure, &dev_list));
 739         vector<shared_ptr<Device>> result;
 740         for (GSList *dev = dev_list; dev; dev = dev->next)
 741         {
 742                 auto sdi = (struct sr_dev_inst *) dev->data;
 743                 result.push_back(devices[sdi]);
 744         }
 745         return result;
 746 }
 747
 748 void Session::remove_devices()
 749 {
 750         devices.clear();
 751         check(sr_session_dev_remove_all(structure));
 752 }
 753
 754 void Session::start()
 755 {
 756         check(sr_session_start(structure));
 757 }
 758
 759 void Session::run()
 760 {
 761         check(sr_session_run(structure));
 762 }
 763
 764 void Session::stop()
 765 {
 766         check(sr_session_stop(structure));
 767 }
 768
 769 void Session::begin_save(string filename)
 770 {
 771         saving = true;
 772         save_initialized = false;
 773         save_filename = filename;
 774         save_samplerate = 0;
 775 }
 776
 777 void Session::append(shared_ptr<Packet> packet)
 778 {
 779         if (!saving)
 780                 throw Error(SR_ERR);
 781
 782         switch (packet->structure->type)
 783         {
 784                 case SR_DF_META:
 785                 {
 786                         auto meta = (const struct sr_datafeed_meta *)
 787                                 packet->structure->payload;
 788
 789                         for (auto l = meta->config; l; l = l->next)
 790                         {
 791                                 auto config = (struct sr_config *) l->data;
 792                                 if (config->key == SR_CONF_SAMPLERATE)
 793                                         save_samplerate = g_variant_get_uint64(config->data);
 794                         }
 795
 796                         break;
 797                 }
 798                 case SR_DF_LOGIC:
 799                 {
 800                         if (save_samplerate == 0)
 801                         {
 802                                 GVariant *samplerate;
 803
 804                                 check(sr_config_get(packet->device->structure->driver,
 805                                         packet->device->structure, NULL, SR_CONF_SAMPLERATE,
 806                                         &samplerate));
 807
 808                                 save_samplerate = g_variant_get_uint64(samplerate);
 809
 810                                 g_variant_unref(samplerate);
 811                         }
 812
 813                         if (!save_initialized)
 814                         {
 815                                 vector<shared_ptr<Channel>> save_channels;
 816
 817                                 for (auto channel : packet->device->get_channels())
 818                                         if (channel->structure->enabled &&
 819                                                         channel->structure->type == SR_CHANNEL_LOGIC)
 820                                                 save_channels.push_back(channel);
 821
 822                                 auto channels = g_new(char *, save_channels.size());
 823
 824                                 int i = 0;
 825                                 for (auto channel : save_channels)
 826                                                 channels[i++] = channel->structure->name;
 827                                 channels[i] = NULL;
 828
 829                                 int ret = sr_session_save_init(structure, save_filename.c_str(),
 830                                                 save_samplerate, channels);
 831
 832                                 g_free(channels);
 833
 834                                 if (ret != SR_OK)
 835                                         throw Error(ret);
 836
 837                                 save_initialized = true;
 838                         }
 839
 840                         auto logic = (const struct sr_datafeed_logic *)
 841                                 packet->structure->payload;
 842
 843                         check(sr_session_append(structure, save_filename.c_str(),
 844                                 (uint8_t *) logic->data, logic->unitsize,
 845                                 logic->length / logic->unitsize));
 846                 }
 847         }
 848 }
 849
 850 void Session::append(void *data, size_t length, unsigned int unit_size)
 851 {
 852         check(sr_session_append(structure, save_filename.c_str(),
 853                 (uint8_t *) data, unit_size, length));
 854 }
 855
 856 static void datafeed_callback(const struct sr_dev_inst *sdi,
 857         const struct sr_datafeed_packet *pkt, void *cb_data)
 858 {
 859         auto callback = static_cast<DatafeedCallbackData *>(cb_data);
 860         callback->run(sdi, pkt);
 861 }
 862
 863 void Session::add_datafeed_callback(DatafeedCallbackFunction callback)
 864 {
 865         auto cb_data = new DatafeedCallbackData(this, callback);
 866         check(sr_session_datafeed_callback_add(structure, datafeed_callback, cb_data));
 867         datafeed_callbacks.push_back(cb_data);
 868 }
 869
 870 void Session::remove_datafeed_callbacks(void)
 871 {
 872         check(sr_session_datafeed_callback_remove_all(structure));
 873         for (auto callback : datafeed_callbacks)
 874                 delete callback;
 875         datafeed_callbacks.clear();
 876 }
 877
 878 static int source_callback(int fd, int revents, void *cb_data)
 879 {
 880         (void) fd;
 881         auto callback = (SourceCallbackData *) cb_data;
 882         return callback->run(revents);
 883 }
 884
 885 void Session::add_source(shared_ptr<EventSource> source)
 886 {
 887         if (source_callbacks.count(source) == 1)
 888                 throw Error(SR_ERR_ARG);
 889
 890         auto cb_data = new SourceCallbackData(source);
 891
 892         switch (source->type)
 893         {
 894                 case EventSource::SOURCE_FD:
 895                         check(sr_session_source_add(structure, source->fd, source->events,
 896                                 source->timeout, source_callback, cb_data));
 897                         break;
 898                 case EventSource::SOURCE_POLLFD:
 899                         check(sr_session_source_add_pollfd(structure,
 900                                 source->pollfd.gobj(), source->timeout, source_callback,
 901                                 cb_data));
 902                         break;
 903                 case EventSource::SOURCE_IOCHANNEL:
 904                         check(sr_session_source_add_channel(structure,
 905                                 source->channel->gobj(), source->events, source->timeout,
 906                                 source_callback, cb_data));
 907                         break;
 908         }
 909
 910         source_callbacks[source] = cb_data;
 911 }
 912
 913 void Session::remove_source(shared_ptr<EventSource> source)
 914 {
 915         if (source_callbacks.count(source) == 0)
 916                 throw Error(SR_ERR_ARG);
 917
 918         switch (source->type)
 919         {
 920                 case EventSource::SOURCE_FD:
 921                         check(sr_session_source_remove(structure, source->fd));
 922                         break;
 923                 case EventSource::SOURCE_POLLFD:
 924                         check(sr_session_source_remove_pollfd(structure,
 925                                 source->pollfd.gobj()));
 926                         break;
 927                 case EventSource::SOURCE_IOCHANNEL:
 928                         check(sr_session_source_remove_channel(structure,
 929                                 source->channel->gobj()));
 930                         break;
 931         }
 932
 933         delete source_callbacks[source];
 934
 935         source_callbacks.erase(source);
 936 }
 937
 938 shared_ptr<Trigger> Session::get_trigger()
 939 {
 940         return trigger;
 941 }
 942
 943 void Session::set_trigger(shared_ptr<Trigger> trigger)
 944 {
 945         check(sr_session_trigger_set(structure, trigger->structure));
 946         this->trigger = trigger;
 947 }
 948
 949 Packet::Packet(shared_ptr<Device> device,
 950         const struct sr_datafeed_packet *structure) :
 951         structure(structure),
 952         device(device)
 953 {
 954         switch (structure->type)
 955         {
 956                 case SR_DF_HEADER:
 957                         payload = new Header(
 958                                 static_cast<const struct sr_datafeed_header *>(
 959                                         structure->payload));
 960                         break;
 961                 case SR_DF_META:
 962                         payload = new Meta(
 963                                 static_cast<const struct sr_datafeed_meta *>(
 964                                         structure->payload));
 965                         break;
 966                 case SR_DF_LOGIC:
 967                         payload = new Logic(
 968                                 static_cast<const struct sr_datafeed_logic *>(
 969                                         structure->payload));
 970                         break;
 971                 case SR_DF_ANALOG:
 972                         payload = new Analog(
 973                                 static_cast<const struct sr_datafeed_analog *>(
 974                                         structure->payload));
 975                         break;
 976         }
 977 }
 978
 979 Packet::~Packet()
 980 {
 981         if (payload)
 982                 delete payload;
 983 }
 984
 985 const PacketType *Packet::get_type()
 986 {
 987         return PacketType::get(structure->type);
 988 }
 989
 990 shared_ptr<PacketPayload> Packet::get_payload()
 991 {
 992         return payload->get_shared_pointer(this);
 993 }
 994
 995 PacketPayload::PacketPayload()
 996 {
 997 }
 998
 999 PacketPayload::~PacketPayload()
1000 {
1001 }
1002
1003 Header::Header(const struct sr_datafeed_header *structure) :
1004         PacketPayload(),
1005         StructureWrapper<Packet, const struct sr_datafeed_header>(structure)
1006 {
1007 }
1008
1009 Header::~Header()
1010 {
1011 }
1012
1013 int Header::get_feed_version()
1014 {
1015         return structure->feed_version;
1016 }
1017
1018 Glib::TimeVal Header::get_start_time()
1019 {
1020         return Glib::TimeVal(
1021                 structure->starttime.tv_sec,
1022                 structure->starttime.tv_usec);
1023 }
1024
1025 Meta::Meta(const struct sr_datafeed_meta *structure) :
1026         PacketPayload(),
1027         StructureWrapper<Packet, const struct sr_datafeed_meta>(structure)
1028 {
1029 }
1030
1031 Meta::~Meta()
1032 {
1033 }
1034
1035 map<const ConfigKey *, Glib::VariantBase> Meta::get_config()
1036 {
1037         map<const ConfigKey *, Glib::VariantBase> result;
1038         for (auto l = structure->config; l; l = l->next)
1039         {
1040                 auto config = (struct sr_config *) l->data;
1041                 result[ConfigKey::get(config->key)] = Glib::VariantBase(config->data);
1042         }
1043         return result;
1044 }
1045
1046 Logic::Logic(const struct sr_datafeed_logic *structure) :
1047         PacketPayload(),
1048         StructureWrapper<Packet, const struct sr_datafeed_logic>(structure)
1049 {
1050 }
1051
1052 Logic::~Logic()
1053 {
1054 }
1055
1056 void *Logic::get_data_pointer()
1057 {
1058         return structure->data;
1059 }
1060
1061 size_t Logic::get_data_length()
1062 {
1063         return structure->length;
1064 }
1065
1066 unsigned int Logic::get_unit_size()
1067 {
1068         return structure->unitsize;
1069 }
1070
1071 Analog::Analog(const struct sr_datafeed_analog *structure) :
1072         PacketPayload(),
1073         StructureWrapper<Packet, const struct sr_datafeed_analog>(structure)
1074 {
1075 }
1076
1077 Analog::~Analog()
1078 {
1079 }
1080
1081 float *Analog::get_data_pointer()
1082 {
1083         return structure->data;
1084 }
1085
1086 unsigned int Analog::get_num_samples()
1087 {
1088         return structure->num_samples;
1089 }
1090
1091 vector<shared_ptr<Channel>> Analog::get_channels()
1092 {
1093         vector<shared_ptr<Channel>> result;
1094         for (auto l = structure->channels; l; l = l->next)
1095                 result.push_back(parent->device->get_channel(
1096                         (struct sr_channel *)l->data));
1097         return result;
1098 }
1099
1100 const Quantity *Analog::get_mq()
1101 {
1102         return Quantity::get(structure->mq);
1103 }
1104
1105 const Unit *Analog::get_unit()
1106 {
1107         return Unit::get(structure->unit);
1108 }
1109
1110 vector<const QuantityFlag *> Analog::get_mq_flags()
1111 {
1112         return QuantityFlag::flags_from_mask(structure->mqflags);
1113 }
1114
1115 InputFormat::InputFormat(const struct sr_input_module *structure) :
1116         StructureWrapper<Context, const struct sr_input_module>(structure)
1117 {
1118 }
1119
1120 InputFormat::~InputFormat()
1121 {
1122 }
1123
1124 string InputFormat::get_name()
1125 {
1126         return valid_string(sr_input_id_get(structure));
1127 }
1128
1129 string InputFormat::get_description()
1130 {
1131         return valid_string(sr_input_description_get(structure));
1132 }
1133
1134 map<string, shared_ptr<Option>> InputFormat::get_options()
1135 {
1136         const struct sr_option **options = sr_input_options_get(structure);
1137         auto option_array = shared_ptr<const struct sr_option *>(
1138                 options, sr_input_options_free);
1139         map<string, shared_ptr<Option>> result;
1140         for (int i = 0; options[i]; i++)
1141                 result[options[i]->id] = shared_ptr<Option>(
1142                         new Option(options[i], option_array), Option::Deleter());
1143         return result;
1144 }
1145
1146 shared_ptr<Input> InputFormat::create_input(
1147         map<string, Glib::VariantBase> options)
1148 {
1149         auto input = sr_input_new(structure, map_to_hash_variant(options));
1150         if (!input)
1151                 throw Error(SR_ERR_ARG);
1152         return shared_ptr<Input>(
1153                 new Input(parent->shared_from_this(), input), Input::Deleter());
1154 }
1155
1156 Input::Input(shared_ptr<Context> context, const struct sr_input *structure) :
1157         structure(structure),
1158         context(context),
1159         device(nullptr)
1160 {
1161 }
1162
1163 shared_ptr<InputDevice> Input::get_device()
1164 {
1165         if (!device)
1166         {
1167                 auto sdi = sr_input_dev_inst_get(structure);
1168                 if (!sdi)
1169                         throw Error(SR_ERR_NA);
1170                 device = new InputDevice(shared_from_this(), sdi);
1171         }
1172
1173         return static_pointer_cast<InputDevice>(
1174                 device->get_shared_pointer(shared_from_this()));
1175 }
1176
1177 void Input::send(string data)
1178 {
1179         auto gstr = g_string_new(data.c_str());
1180         auto ret = sr_input_send(structure, gstr);
1181         g_string_free(gstr, false);
1182         check(ret);
1183 }
1184
1185 Input::~Input()
1186 {
1187         if (device)
1188                 delete device;
1189         check(sr_input_free(structure));
1190 }
1191
1192 InputDevice::InputDevice(shared_ptr<Input> input,
1193                 struct sr_dev_inst *structure) :
1194         StructureWrapper(structure),
1195         Device(structure),
1196         input(input)
1197 {
1198 }
1199
1200 InputDevice::~InputDevice()
1201 {
1202 }
1203
1204 shared_ptr<Device> InputDevice::get_shared_from_this()
1205 {
1206         return static_pointer_cast<Device>(
1207                 static_pointer_cast<InputDevice>(shared_from_this()));
1208 }
1209
1210 Option::Option(const struct sr_option *structure,
1211                 shared_ptr<const struct sr_option *> structure_array) :
1212         structure(structure),
1213         structure_array(structure_array)
1214 {
1215 }
1216
1217 Option::~Option()
1218 {
1219 }
1220
1221 string Option::get_id()
1222 {
1223         return valid_string(structure->id);
1224 }
1225
1226 string Option::get_name()
1227 {
1228         return valid_string(structure->name);
1229 }
1230
1231 string Option::get_description()
1232 {
1233         return valid_string(structure->desc);
1234 }
1235
1236 Glib::VariantBase Option::get_default_value()
1237 {
1238         return Glib::VariantBase(structure->def, true);
1239 }
1240
1241 vector<Glib::VariantBase> Option::get_values()
1242 {
1243         vector<Glib::VariantBase> result;
1244         for (auto l = structure->values; l; l = l->next)
1245                 result.push_back(Glib::VariantBase((GVariant *) l->data, true));
1246         return result;
1247 }
1248
1249 OutputFormat::OutputFormat(const struct sr_output_module *structure) :
1250         StructureWrapper<Context, const struct sr_output_module>(structure)
1251 {
1252 }
1253
1254 OutputFormat::~OutputFormat()
1255 {
1256 }
1257
1258 string OutputFormat::get_name()
1259 {
1260         return valid_string(sr_output_id_get(structure));
1261 }
1262
1263 string OutputFormat::get_description()
1264 {
1265         return valid_string(sr_output_description_get(structure));
1266 }
1267
1268 map<string, shared_ptr<Option>> OutputFormat::get_options()
1269 {
1270         const struct sr_option **options = sr_output_options_get(structure);
1271         auto option_array = shared_ptr<const struct sr_option *>(
1272                 options, sr_output_options_free);
1273         map<string, shared_ptr<Option>> result;
1274         for (int i = 0; options[i]; i++)
1275                 result[options[i]->id] = shared_ptr<Option>(
1276                         new Option(options[i], option_array), Option::Deleter());
1277         return result;
1278 }
1279
1280 shared_ptr<Output> OutputFormat::create_output(
1281         shared_ptr<Device> device, map<string, Glib::VariantBase> options)
1282 {
1283         return shared_ptr<Output>(
1284                 new Output(
1285                         static_pointer_cast<OutputFormat>(shared_from_this()),
1286                                 device, options),
1287                 Output::Deleter());
1288 }
1289
1290 Output::Output(shared_ptr<OutputFormat> format,
1291                 shared_ptr<Device> device, map<string, Glib::VariantBase> options) :
1292         structure(sr_output_new(format->structure,
1293                 map_to_hash_variant(options), device->structure)),
1294         format(format), device(device), options(options)
1295 {
1296 }
1297
1298 Output::~Output()
1299 {
1300         check(sr_output_free(structure));
1301 }
1302
1303 string Output::receive(shared_ptr<Packet> packet)
1304 {
1305         GString *out;
1306         check(sr_output_send(structure, packet->structure, &out));
1307         if (out)
1308         {
1309                 auto result = string(out->str, out->str + out->len);
1310                 g_string_free(out, true);
1311                 return result;
1312         }
1313         else
1314         {
1315                 return string();
1316         }
1317 }
1318
1319 #include "enums.cpp"
1320
1321 }