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