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