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