src/main.cpp

   1 /*
   2  * This file is part of the libsigrokflow project.
   3  *
   4  * Copyright (C) 2018 Uwe Hermann <uwe@hermann-uwe.de>
   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 <config.h>
  21 #include <libsigrokflow/libsigrokflow.hpp>
  22
  23 #include <iostream>
  24
  25 namespace Srf
  26 {
  27
  28 using namespace std;
  29 using namespace std::placeholders;
  30
  31 void init()
  32 {
  33         Gst::Plugin::register_static(GST_VERSION_MAJOR, GST_VERSION_MINOR,
  34                         "sigrok_legacy_capture_device",
  35                         "Wrapper for capture devices using legacy libsigrok APIs",
  36                         sigc::ptr_fun(&LegacyCaptureDevice::register_element),
  37                         "0.01", "GPL", "sigrok", "libsigrokflow", "http://sigrok.org");
  38 }
  39
  40 GstBlock::GstBlock(GstElement *gobj) :
  41         Gst::Element(gobj)
  42 {
  43 }
  44
  45 Device::Device(GstElement *gobj) :
  46         GstBlock(gobj)
  47 {
  48 }
  49
  50 CaptureDevice::CaptureDevice(GstElement *gobj) :
  51         Device(gobj)
  52 {
  53 }
  54
  55 void LegacyCaptureDevice::class_init(Gst::ElementClass<LegacyCaptureDevice> *klass)
  56 {
  57         klass->set_metadata("sigrok legacy capture device",
  58                         "Source", "Wrapper for capture devices using legacy libsigrok APIs",
  59                         "Martin Ling");
  60
  61         klass->add_pad_template(Gst::PadTemplate::create(
  62                         "src",
  63                         Gst::PAD_SRC,
  64                         Gst::PAD_ALWAYS,
  65                         Gst::Caps::create_any()));
  66 }
  67
  68 bool LegacyCaptureDevice::register_element(Glib::RefPtr<Gst::Plugin> plugin)
  69 {
  70         Gst::ElementFactory::register_element(plugin, "sigrok_legacy_capture_device",
  71                         0, Gst::register_mm_type<LegacyCaptureDevice>(
  72                                 "sigrok_legacy_capture_device"));
  73         return true;
  74 }
  75
  76 LegacyCaptureDevice::LegacyCaptureDevice(GstElement *gobj) :
  77         Glib::ObjectBase(typeid(LegacyCaptureDevice)),
  78         CaptureDevice(gobj)
  79 {
  80         add_pad(_src_pad = Gst::Pad::create(get_pad_template("src"), "src"));
  81 }
  82
  83 Glib::RefPtr<LegacyCaptureDevice>LegacyCaptureDevice::create(
  84         shared_ptr<sigrok::HardwareDevice> libsigrok_device)
  85 {
  86         auto element = Gst::ElementFactory::create_element("sigrok_legacy_capture_device");
  87         if (!element)
  88                 throw runtime_error("Failed to create element - plugin not registered?");
  89         auto device = Glib::RefPtr<LegacyCaptureDevice>::cast_static(element);
  90         device->_libsigrok_device = libsigrok_device;
  91         return device;
  92 }
  93
  94 shared_ptr<sigrok::HardwareDevice> LegacyCaptureDevice::libsigrok_device()
  95 {
  96         return _libsigrok_device;
  97 }
  98
  99 Gst::StateChangeReturn LegacyCaptureDevice::change_state_vfunc(Gst::StateChange transition)
 100 {
 101         switch (transition)
 102         {
 103                 case Gst::STATE_CHANGE_READY_TO_PAUSED:
 104                         return Gst::StateChangeReturn::STATE_CHANGE_NO_PREROLL;
 105                 case Gst::STATE_CHANGE_PAUSED_TO_PLAYING:
 106                         _libsigrok_device->open();
 107                         _libsigrok_device->config_set(sigrok::ConfigKey::LIMIT_SAMPLES,
 108                                         Glib::Variant<int>::create(10));
 109                         _task = Gst::Task::create(std::bind(&LegacyCaptureDevice::_run, this));
 110                         _task->set_lock(_mutex);
 111                         _src_pad->set_active(true);
 112                         _task->start();
 113                         return Gst::STATE_CHANGE_SUCCESS;
 114                 default:
 115                         return Gst::STATE_CHANGE_SUCCESS;
 116         }
 117 }
 118
 119 void LegacyCaptureDevice::_datafeed_callback(
 120         shared_ptr<sigrok::Device> device,
 121         shared_ptr<sigrok::Packet> packet)
 122 {
 123         (void) device;
 124         switch (packet->type()->id()) {
 125                 case SR_DF_LOGIC:
 126                 {
 127                         auto logic = static_pointer_cast<sigrok::Logic>(packet->payload());
 128                         auto mem = Gst::Memory::create(
 129                                         Gst::MEMORY_FLAG_READONLY,
 130                                         logic->data_pointer(),
 131                                         logic->data_length(),
 132                                         0,
 133                                         logic->data_length());
 134                         auto buf = Gst::Buffer::create();
 135                         buf->append_memory(move(mem));
 136                         _src_pad->push(move(buf));
 137                         break;
 138                 }
 139                 case SR_DF_END:
 140                         _session->stop();
 141                         _src_pad->push_event(Gst::EventEos::create());
 142                         break;
 143                 default:
 144                         break;
 145         }
 146 }
 147
 148 void LegacyCaptureDevice::_run()
 149 {
 150         _session = _libsigrok_device->driver()->parent()->create_session();
 151         _session->add_device(_libsigrok_device);
 152         _session->add_datafeed_callback(bind(&LegacyCaptureDevice::_datafeed_callback, this, _1, _2));
 153         _session->start();
 154         _session->run();
 155         _task->stop();
 156 }
 157
 158 }