X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=src%2Fmain.cpp;h=79af3ec39d45aa455d25cc50e00fa32e3d2c48cf;hb=e2cfc0efeadec350376bffbe9250a27c6aa9788d;hp=96d720e235a9d62228e50d7c620cfada076dd8b5;hpb=b903bb0a1cd93d39195e812317e20e353b6d1ea8;p=libsigrokflow.git diff --git a/src/main.cpp b/src/main.cpp index 96d720e..79af3ec 100644 --- a/src/main.cpp +++ b/src/main.cpp @@ -20,11 +20,207 @@ #include #include +#include + 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 *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 plugin) +{ + Gst::ElementFactory::register_element(plugin, "sigrok_legacy_capture_device", + 0, Gst::register_mm_type( + "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::RefPtrLegacyCaptureDevice::create( + shared_ptr 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::cast_static(element); + device->_libsigrok_device = libsigrok_device; + return device; +} + +shared_ptr 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 device, + shared_ptr packet) +{ + (void) device; + switch (packet->type()->id()) { + case SR_DF_LOGIC: + { + auto logic = static_pointer_cast(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 *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 plugin) +{ + Gst::ElementFactory::register_element(plugin, "sigrok_legacy_output", + 0, Gst::register_mm_type( + "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::RefPtrLegacyOutput::create( + shared_ptr 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::cast_static(element); + output->_libsigrok_output = libsigrok_output; + return output; +} + +shared_ptr LegacyOutput::libsigrok_output() +{ + return _libsigrok_output; +} + +Gst::FlowReturn LegacyOutput::render_vfunc(const Glib::RefPtr &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; } }