]> sigrok.org Git - libsigrokflow.git/blobdiff - src/main.cpp
Implement LegacyOutput.
[libsigrokflow.git] / src / main.cpp
index 96d720e235a9d62228e50d7c620cfada076dd8b5..79af3ec39d45aa455d25cc50e00fa32e3d2c48cf 100644 (file)
 #include <config.h>
 #include <libsigrokflow/libsigrokflow.hpp>
 
+#include <iostream>
+
 namespace Srf
 {
 
+using namespace std;
+using namespace std::placeholders;
+
 void init()
 {
+       Gst::Plugin::register_static(GST_VERSION_MAJOR, GST_VERSION_MINOR,
+                       "sigrok_legacy_capture_device",
+                       "Wrapper for capture devices using legacy libsigrok APIs",
+                       sigc::ptr_fun(&LegacyCaptureDevice::register_element),
+                       "0.01", "GPL", "sigrok", "libsigrokflow", "http://sigrok.org");
+       Gst::Plugin::register_static(GST_VERSION_MAJOR, GST_VERSION_MINOR,
+                       "sigrok_legacy_output",
+                       "Wrapper for outputs using legacy libsigrok APIs",
+                       sigc::ptr_fun(&LegacyOutput::register_element),
+                       "0.01", "GPL", "sigrok", "libsigrokflow", "http://sigrok.org");
+}
+
+Sink::Sink(GstBaseSink *gobj) :
+       Gst::BaseSink(gobj)
+{
+}
+
+Device::Device(GstElement *gobj) :
+       Gst::Element(gobj)
+{
+}
+
+CaptureDevice::CaptureDevice(GstElement *gobj) :
+       Device(gobj)
+{
+}
+
+void LegacyCaptureDevice::class_init(Gst::ElementClass<LegacyCaptureDevice> *klass)
+{
+       klass->set_metadata("sigrok legacy capture device",
+                       "Source", "Wrapper for capture devices using legacy libsigrok APIs",
+                       "Martin Ling");
+
+       klass->add_pad_template(Gst::PadTemplate::create(
+                       "src",
+                       Gst::PAD_SRC,
+                       Gst::PAD_ALWAYS,
+                       Gst::Caps::create_any()));
+}
+
+bool LegacyCaptureDevice::register_element(Glib::RefPtr<Gst::Plugin> plugin)
+{
+       Gst::ElementFactory::register_element(plugin, "sigrok_legacy_capture_device",
+                       0, Gst::register_mm_type<LegacyCaptureDevice>(
+                               "sigrok_legacy_capture_device"));
+       return true;
+}
+
+LegacyCaptureDevice::LegacyCaptureDevice(GstElement *gobj) :
+       Glib::ObjectBase(typeid(LegacyCaptureDevice)),
+       CaptureDevice(gobj)
+{
+       add_pad(_src_pad = Gst::Pad::create(get_pad_template("src"), "src"));
+}
+
+Glib::RefPtr<LegacyCaptureDevice>LegacyCaptureDevice::create(
+       shared_ptr<sigrok::HardwareDevice> libsigrok_device)
+{
+       auto element = Gst::ElementFactory::create_element("sigrok_legacy_capture_device");
+       if (!element)
+               throw runtime_error("Failed to create element - plugin not registered?");
+       auto device = Glib::RefPtr<LegacyCaptureDevice>::cast_static(element);
+       device->_libsigrok_device = libsigrok_device;
+       return device;
+}
+
+shared_ptr<sigrok::HardwareDevice> LegacyCaptureDevice::libsigrok_device()
+{
+       return _libsigrok_device;
+}
+
+Gst::StateChangeReturn LegacyCaptureDevice::change_state_vfunc(Gst::StateChange transition)
+{
+       switch (transition)
+       {
+               case Gst::STATE_CHANGE_READY_TO_PAUSED:
+                       return Gst::StateChangeReturn::STATE_CHANGE_NO_PREROLL;
+               case Gst::STATE_CHANGE_PAUSED_TO_PLAYING:
+                       _task = Gst::Task::create(std::bind(&LegacyCaptureDevice::_run, this));
+                       _task->set_lock(_mutex);
+                       _src_pad->set_active(true);
+                       _task->start();
+                       return Gst::STATE_CHANGE_SUCCESS;
+               default:
+                       return Gst::STATE_CHANGE_SUCCESS;
+       }
+}
+
+void LegacyCaptureDevice::_datafeed_callback(
+       shared_ptr<sigrok::Device> device,
+       shared_ptr<sigrok::Packet> packet)
+{
+       (void) device;
+       switch (packet->type()->id()) {
+               case SR_DF_LOGIC:
+               {
+                       auto logic = static_pointer_cast<sigrok::Logic>(packet->payload());
+                       auto mem = Gst::Memory::create(
+                                       Gst::MEMORY_FLAG_READONLY,
+                                       logic->data_pointer(),
+                                       logic->data_length(),
+                                       0,
+                                       logic->data_length());
+                       auto buf = Gst::Buffer::create();
+                       buf->append_memory(move(mem));
+                       _src_pad->push(move(buf));
+                       break;
+               }
+               case SR_DF_END:
+                       _session->stop();
+                       _src_pad->push_event(Gst::EventEos::create());
+                       break;
+               default:
+                       break;
+       }
+}
+
+void LegacyCaptureDevice::_run()
+{
+       _session = _libsigrok_device->driver()->parent()->create_session();
+       _session->add_device(_libsigrok_device);
+       _session->add_datafeed_callback(bind(&LegacyCaptureDevice::_datafeed_callback, this, _1, _2));
+       _session->start();
+       _session->run();
+       _task->stop();
+}
+
+void LegacyOutput::class_init(Gst::ElementClass<LegacyOutput> *klass)
+{
+       klass->set_metadata("sigrok legacy output",
+                       "Sink", "Wrapper for outputs using legacy libsigrok APIs",
+                       "Martin Ling");
+
+       klass->add_pad_template(Gst::PadTemplate::create(
+                       "sink",
+                       Gst::PAD_SINK,
+                       Gst::PAD_ALWAYS,
+                       Gst::Caps::create_any()));
+}
+
+bool LegacyOutput::register_element(Glib::RefPtr<Gst::Plugin> plugin)
+{
+       Gst::ElementFactory::register_element(plugin, "sigrok_legacy_output",
+                       0, Gst::register_mm_type<LegacyOutput>(
+                               "sigrok_legacy_output"));
+       return true;
+}
+
+LegacyOutput::LegacyOutput(GstBaseSink *gobj) :
+       Glib::ObjectBase(typeid(LegacyOutput)),
+       Sink(gobj)
+{
+       add_pad(_sink_pad = Gst::Pad::create(get_pad_template("sink"), "sink"));
+}
+
+Glib::RefPtr<LegacyOutput>LegacyOutput::create(
+       shared_ptr<sigrok::Output> libsigrok_output)
+{
+       auto element = Gst::ElementFactory::create_element("sigrok_legacy_output");
+       if (!element)
+               throw runtime_error("Failed to create element - plugin not registered?");
+       auto output = Glib::RefPtr<LegacyOutput>::cast_static(element);
+       output->_libsigrok_output = libsigrok_output;
+       return output;
+}
+
+shared_ptr<sigrok::Output> LegacyOutput::libsigrok_output()
+{
+       return _libsigrok_output;
+}
+
+Gst::FlowReturn LegacyOutput::render_vfunc(const Glib::RefPtr<Gst::Buffer> &buffer)
+{
+       Gst::MapInfo info;
+       buffer->map(info, Gst::MAP_READ);
+       auto context = _libsigrok_output->format()->parent();
+       auto packet = context->create_logic_packet(
+                       info.get_data(), info.get_size(), 2);
+       auto result = _libsigrok_output->receive(packet);
+       cout << result;
+       buffer->unmap(info);
+       return Gst::FLOW_OK;
+}
+
+bool LegacyOutput::stop_vfunc()
+{
+       auto context = _libsigrok_output->format()->parent();
+       auto end_packet = context->create_end_packet();
+       auto result = _libsigrok_output->receive(end_packet);
+       cout << result;
+       return true;
 }
 
 }