bindings/cxx/classes.cpp

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