bindings/cxx/classes.cpp

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