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