X-Git-Url: https://sigrok.org/gitweb/?p=pulseview.git;a=blobdiff_plain;f=pv%2Fsession.cpp;h=ae29f6a9c4d5b135242dc4426fc6b1837ba6f89f;hp=b1ce14680bbec848f6ae92fd3ec4010094d8320c;hb=97378470ded88af84edaa0f1063d10d834475665;hpb=a66e286e0276186c41b7ce07cfcbd0c4da2ca1cb diff --git a/pv/session.cpp b/pv/session.cpp index b1ce1468..ae29f6a9 100644 --- a/pv/session.cpp +++ b/pv/session.cpp @@ -28,6 +28,7 @@ #include #include "devicemanager.hpp" +#include "mainwindow.hpp" #include "session.hpp" #include "data/analog.hpp" @@ -51,6 +52,11 @@ #include +#ifdef ENABLE_FLOW +#include +#include +#endif + #ifdef ENABLE_DECODE #include #include "data/decodesignal.hpp" @@ -73,6 +79,9 @@ using std::recursive_mutex; using std::runtime_error; using std::shared_ptr; using std::string; +#ifdef ENABLE_FLOW +using std::unique_lock; +#endif using std::unique_ptr; using std::unordered_set; using std::vector; @@ -90,6 +99,12 @@ using sigrok::Session; using Glib::VariantBase; +#ifdef ENABLE_FLOW +using Gst::Bus; +using Gst::ElementFactory; +using Gst::Pipeline; +#endif + namespace pv { shared_ptr Session::sr_context; @@ -172,11 +187,49 @@ bool Session::data_saved() const return data_saved_; } +void Session::save_setup(QSettings &settings) const +{ + int decode_signals = 0, views = 0; + + // Save channels and decoders + for (const shared_ptr& base : signalbases_) { +#ifdef ENABLE_DECODE + if (base->is_decode_signal()) { + settings.beginGroup("decode_signal" + QString::number(decode_signals++)); + base->save_settings(settings); + settings.endGroup(); + } else +#endif + { + settings.beginGroup(base->internal_name()); + base->save_settings(settings); + settings.endGroup(); + } + } + + settings.setValue("decode_signals", decode_signals); + + // Save view states and their signal settings + // Note: main_view must be saved as view0 + settings.beginGroup("view" + QString::number(views++)); + main_view_->save_settings(settings); + settings.endGroup(); + + for (const shared_ptr& view : views_) { + if (view != main_view_) { + settings.beginGroup("view" + QString::number(views++)); + view->save_settings(settings); + settings.endGroup(); + } + } + + settings.setValue("views", views); +} + void Session::save_settings(QSettings &settings) const { map dev_info; list key_list; - int decode_signals = 0, views = 0; if (device_) { shared_ptr hw_device = @@ -194,7 +247,7 @@ void Session::save_settings(QSettings &settings) const dev_info = device_manager_.get_device_info(device_); - for (string key : key_list) { + for (string& key : key_list) { if (dev_info.count(key)) settings.setValue(QString::fromUtf8(key.c_str()), QString::fromUtf8(dev_info.at(key).c_str())); @@ -206,7 +259,7 @@ void Session::save_settings(QSettings &settings) const } shared_ptr sessionfile_device = - dynamic_pointer_cast< devices::SessionFile >(device_); + dynamic_pointer_cast(device_); if (sessionfile_device) { settings.setValue("device_type", "sessionfile"); @@ -216,39 +269,55 @@ void Session::save_settings(QSettings &settings) const settings.endGroup(); } - // Save channels and decoders - for (shared_ptr base : signalbases_) { -#ifdef ENABLE_DECODE - if (base->is_decode_signal()) { - settings.beginGroup("decode_signal" + QString::number(decode_signals++)); - base->save_settings(settings); - settings.endGroup(); - } else -#endif - { - settings.beginGroup(base->internal_name()); - base->save_settings(settings); - settings.endGroup(); - } + shared_ptr inputfile_device = + dynamic_pointer_cast(device_); + + if (inputfile_device) { + settings.setValue("device_type", "inputfile"); + settings.beginGroup("device"); + inputfile_device->save_meta_to_settings(settings); + settings.endGroup(); } - settings.setValue("decode_signals", decode_signals); + save_setup(settings); + } +} + +void Session::restore_setup(QSettings &settings) +{ + // Restore channels + for (shared_ptr base : signalbases_) { + settings.beginGroup(base->internal_name()); + base->restore_settings(settings); + settings.endGroup(); + } - // Save view states and their signal settings - // Note: main_view must be saved as view0 - settings.beginGroup("view" + QString::number(views++)); - main_view_->save_settings(settings); + // Restore decoders +#ifdef ENABLE_DECODE + int decode_signals = settings.value("decode_signals").toInt(); + + for (int i = 0; i < decode_signals; i++) { + settings.beginGroup("decode_signal" + QString::number(i)); + shared_ptr signal = add_decode_signal(); + signal->restore_settings(settings); settings.endGroup(); + } +#endif - for (shared_ptr view : views_) { - if (view != main_view_) { - settings.beginGroup("view" + QString::number(views++)); - view->save_settings(settings); - settings.endGroup(); - } - } + // Restore views + int views = settings.value("views").toInt(); + + for (int i = 0; i < views; i++) { + settings.beginGroup("view" + QString::number(i)); + + if (i > 0) { + views::ViewType type = (views::ViewType)settings.value("type").toInt(); + add_view(name_, type, this); + views_.back()->restore_settings(settings); + } else + main_view_->restore_settings(settings); - settings.setValue("views", views); + settings.endGroup(); } } @@ -289,59 +358,39 @@ void Session::restore_settings(QSettings &settings) settings.endGroup(); } - if (device_type == "sessionfile") { - settings.beginGroup("device"); - QString filename = settings.value("filename").toString(); - settings.endGroup(); - - if (QFileInfo(filename).isReadable()) { - device = make_shared(device_manager_.context(), - filename.toStdString()); - set_device(device); - - // TODO Perform error handling - start_capture([](QString infoMessage) { (void)infoMessage; }); - - set_name(QFileInfo(filename).fileName()); - } - } - - if (device) { - // Restore channels - for (shared_ptr base : signalbases_) { - settings.beginGroup(base->internal_name()); - base->restore_settings(settings); + if ((device_type == "sessionfile") || (device_type == "inputfile")) { + if (device_type == "sessionfile") { + settings.beginGroup("device"); + QString filename = settings.value("filename").toString(); settings.endGroup(); - } - // Restore decoders -#ifdef ENABLE_DECODE - int decode_signals = settings.value("decode_signals").toInt(); + if (QFileInfo(filename).isReadable()) { + device = make_shared(device_manager_.context(), + filename.toStdString()); + } + } - for (int i = 0; i < decode_signals; i++) { - settings.beginGroup("decode_signal" + QString::number(i)); - shared_ptr signal = add_decode_signal(); - signal->restore_settings(settings); + if (device_type == "inputfile") { + settings.beginGroup("device"); + device = make_shared(device_manager_.context(), + settings); settings.endGroup(); } -#endif - // Restore views - int views = settings.value("views").toInt(); - - for (int i = 0; i < views; i++) { - settings.beginGroup("view" + QString::number(i)); + if (device) { + set_device(device); - if (i > 0) { - views::ViewType type = (views::ViewType)settings.value("type").toInt(); - add_view(name_, type, this); - views_.back()->restore_settings(settings); - } else - main_view_->restore_settings(settings); + start_capture([](QString infoMessage) { + // TODO Emulate noquote() + qDebug() << "Session error:" << infoMessage; }); - settings.endGroup(); + set_name(QString::fromStdString( + dynamic_pointer_cast(device)->display_name(device_manager_))); } } + + if (device) + restore_setup(settings); } void Session::select_device(shared_ptr device) @@ -352,7 +401,7 @@ void Session::select_device(shared_ptr device) else set_default_device(); } catch (const QString &e) { - main_bar_->session_error(tr("Failed to select device"), e); + MainWindow::show_session_error(tr("Failed to select device"), e); } } @@ -372,20 +421,21 @@ void Session::set_device(shared_ptr device) name_ = default_name_; name_changed(); - // Remove all stored data + // Remove all stored data and reset all views for (shared_ptr view : views_) { view->clear_signals(); #ifdef ENABLE_DECODE view->clear_decode_signals(); #endif + view->reset_view_state(); } - for (const shared_ptr d : all_signal_data_) + for (const shared_ptr& d : all_signal_data_) d->clear(); all_signal_data_.clear(); signalbases_.clear(); cur_logic_segment_.reset(); - for (auto entry : cur_analog_segments_) { + for (auto& entry : cur_analog_segments_) { shared_ptr(entry.first).reset(); shared_ptr(entry.second).reset(); } @@ -400,7 +450,7 @@ void Session::set_device(shared_ptr device) device_->open(); } catch (const QString &e) { device_.reset(); - main_bar_->session_error(tr("Failed to open device"), e); + MainWindow::show_session_error(tr("Failed to open device"), e); } if (device_) { @@ -444,7 +494,7 @@ Session::input_format_options(vector user_spec, { map result; - for (auto entry : user_spec) { + for (auto& entry : user_spec) { /* * Split key=value specs. Accept entries without separator * (for simplified boolean specifications). @@ -490,7 +540,7 @@ void Session::load_init_file(const string &file_name, const string &format) [&](const pair > f) { return f.first == user_name; }); if (iter == formats.end()) { - main_bar_->session_error(tr("Error"), + MainWindow::show_session_error(tr("Error"), tr("Unexpected input format: %s").arg(QString::fromStdString(format))); return; } @@ -509,6 +559,10 @@ void Session::load_file(QString file_name, const QString errorMessage( QString("Failed to load file %1").arg(file_name)); + // In the absence of a caller's format spec, try to auto detect. + // Assume "sigrok session file" upon lookup miss. + if (!format) + format = device_manager_.context()->input_format_match(file_name.toStdString()); try { if (format) set_device(shared_ptr( @@ -521,17 +575,26 @@ void Session::load_file(QString file_name, new devices::SessionFile( device_manager_.context(), file_name.toStdString()))); - } catch (Error e) { - main_bar_->session_error(tr("Failed to load ") + file_name, e.what()); + } catch (Error& e) { + MainWindow::show_session_error(tr("Failed to load ") + file_name, e.what()); set_default_device(); main_bar_->update_device_list(); return; } + // Auto-load the setup if one exists + QString setup_file_name = file_name; + setup_file_name.truncate(setup_file_name.lastIndexOf('.')); + setup_file_name.append(".pvs"); + if (QFileInfo::exists(setup_file_name) && QFileInfo(setup_file_name).isReadable()) { + QSettings settings_storage(setup_file_name, QSettings::IniFormat); + restore_setup(settings_storage); + } + main_bar_->update_device_list(); start_capture([&, errorMessage](QString infoMessage) { - main_bar_->session_error(errorMessage, infoMessage); }); + MainWindow::show_session_error(errorMessage, infoMessage); }); set_name(QFileInfo(file_name).fileName()); } @@ -564,7 +627,7 @@ void Session::start_capture(function error_handler) } // Clear signal data - for (const shared_ptr d : all_signal_data_) + for (const shared_ptr& d : all_signal_data_) d->clear(); trigger_list_.clear(); @@ -613,7 +676,7 @@ void Session::register_view(shared_ptr view) qobject_cast(view.get()); if (trace_view) { - for (shared_ptr signalbase : signalbases_) { + for (const shared_ptr& signalbase : signalbases_) { const int sb_exists = count_if( view_signalbases.cbegin(), view_signalbases.cend(), [&](const shared_ptr &sb) { @@ -654,7 +717,7 @@ void Session::deregister_view(shared_ptr view) bool Session::has_view(shared_ptr view) { - for (shared_ptr v : views_) + for (shared_ptr& v : views_) if (v == view) return true; @@ -665,11 +728,11 @@ double Session::get_samplerate() const { double samplerate = 0.0; - for (const shared_ptr d : all_signal_data_) { + for (const shared_ptr& d : all_signal_data_) { assert(d); const vector< shared_ptr > segments = d->segments(); - for (const shared_ptr &s : segments) + for (const shared_ptr& s : segments) samplerate = max(samplerate, s->samplerate()); } // If there is no sample rate given we use samples as unit @@ -684,7 +747,7 @@ uint32_t Session::get_segment_count() const uint32_t value = 0; // Find the highest number of segments - for (shared_ptr data : all_signal_data_) + for (const shared_ptr& data : all_signal_data_) if (data->get_segment_count() > value) value = data->get_segment_count(); @@ -695,7 +758,7 @@ vector Session::get_triggers(uint32_t segment_id) const { vector result; - for (pair entry : trigger_list_) + for (const pair& entry : trigger_list_) if (entry.first == segment_id) result.push_back(entry.second); @@ -711,7 +774,7 @@ bool Session::all_segments_complete(uint32_t segment_id) const { bool all_complete = true; - for (shared_ptr base : signalbases_) + for (const shared_ptr& base : signalbases_) if (!base->segment_is_complete(segment_id)) all_complete = false; @@ -730,9 +793,9 @@ shared_ptr Session::add_decode_signal() signalbases_.insert(signal); // Add the decode signal to all views - for (shared_ptr view : views_) + for (shared_ptr& view : views_) view->add_decode_signal(signal); - } catch (runtime_error e) { + } catch (runtime_error& e) { remove_decode_signal(signal); return nullptr; } @@ -746,7 +809,7 @@ void Session::remove_decode_signal(shared_ptr signal) { signalbases_.erase(signal); - for (shared_ptr view : views_) + for (shared_ptr& view : views_) view->remove_decode_signal(signal); signals_changed(); @@ -772,7 +835,7 @@ void Session::update_signals() if (!device_) { signalbases_.clear(); logic_data_.reset(); - for (shared_ptr view : views_) { + for (shared_ptr& view : views_) { view->clear_signals(); #ifdef ENABLE_DECODE view->clear_decode_signals(); @@ -787,7 +850,7 @@ void Session::update_signals() if (!sr_dev) { signalbases_.clear(); logic_data_.reset(); - for (shared_ptr view : views_) { + for (shared_ptr& view : views_) { view->clear_signals(); #ifdef ENABLE_DECODE view->clear_decode_signals(); @@ -818,7 +881,7 @@ void Session::update_signals() } // Make the signals list - for (shared_ptr viewbase : views_) { + for (shared_ptr& viewbase : views_) { views::trace::View *trace_view = qobject_cast(viewbase.get()); @@ -844,7 +907,7 @@ void Session::update_signals() } else { // Find the signalbase for this channel if possible signalbase.reset(); - for (const shared_ptr b : signalbases_) + for (const shared_ptr& b : signalbases_) if (b->channel() == channel) signalbase = b; @@ -917,10 +980,43 @@ void Session::sample_thread_proc(function error_handler) { assert(error_handler); +#ifdef ENABLE_FLOW + pipeline_ = Pipeline::create(); + + source_ = ElementFactory::create_element("filesrc", "source"); + sink_ = RefPtr::cast_dynamic(ElementFactory::create_element("appsink", "sink")); + + pipeline_->add(source_)->add(sink_); + source_->link(sink_); + + source_->set_property("location", Glib::ustring("/tmp/dummy_binary")); + + sink_->set_property("emit-signals", TRUE); + sink_->signal_new_sample().connect(sigc::mem_fun(*this, &Session::on_gst_new_sample)); + + // Get the bus from the pipeline and add a bus watch to the default main context + RefPtr bus = pipeline_->get_bus(); + bus->add_watch(sigc::mem_fun(this, &Session::on_gst_bus_message)); + + // Start pipeline and Wait until it finished processing + pipeline_done_interrupt_ = false; + pipeline_->set_state(Gst::STATE_PLAYING); + + unique_lock pipeline_done_lock_(pipeline_done_mutex_); + pipeline_done_cond_.wait(pipeline_done_lock_); + + // Let the pipeline free all resources + pipeline_->set_state(Gst::STATE_NULL); + +#else if (!device_) return; - cur_samplerate_ = device_->read_config(ConfigKey::SAMPLERATE); + try { + cur_samplerate_ = device_->read_config(ConfigKey::SAMPLERATE); + } catch (Error& e) { + cur_samplerate_ = 0; + } out_of_memory_ = false; @@ -934,7 +1030,7 @@ void Session::sample_thread_proc(function error_handler) try { device_->start(); - } catch (Error e) { + } catch (Error& e) { error_handler(e.what()); return; } @@ -944,19 +1040,22 @@ void Session::sample_thread_proc(function error_handler) try { device_->run(); - } catch (Error e) { + } catch (Error& e) { error_handler(e.what()); set_capture_state(Stopped); return; + } catch (QString& e) { + error_handler(e); + set_capture_state(Stopped); + return; } set_capture_state(Stopped); // Confirm that SR_DF_END was received - if (cur_logic_segment_) { - qDebug("SR_DF_END was not received."); - assert(false); - } + if (cur_logic_segment_) + qDebug() << "WARNING: SR_DF_END was not received."; +#endif // Optimize memory usage free_unused_memory(); @@ -974,12 +1073,11 @@ void Session::sample_thread_proc(function error_handler) void Session::free_unused_memory() { - for (shared_ptr data : all_signal_data_) { + for (const shared_ptr& data : all_signal_data_) { const vector< shared_ptr > segments = data->segments(); - for (shared_ptr segment : segments) { + for (const shared_ptr& segment : segments) segment->free_unused_memory(); - } } } @@ -1017,7 +1115,7 @@ void Session::signal_segment_completed() { int segment_id = 0; - for (shared_ptr signalbase : signalbases_) { + for (const shared_ptr& signalbase : signalbases_) { // We only care about analog and logic channels, not derived ones if (signalbase->type() == data::SignalBase::AnalogChannel) { segment_id = signalbase->analog_data()->get_segment_count() - 1; @@ -1034,6 +1132,49 @@ void Session::signal_segment_completed() segment_completed(segment_id); } +#ifdef ENABLE_FLOW +bool Session::on_gst_bus_message(const Glib::RefPtr& bus, const Glib::RefPtr& message) +{ + (void)bus; + + if ((message->get_source() == pipeline_) && \ + ((message->get_message_type() == Gst::MESSAGE_EOS))) + pipeline_done_cond_.notify_one(); + + // TODO Also evaluate MESSAGE_STREAM_STATUS to receive error notifications + + return true; +} + +Gst::FlowReturn Session::on_gst_new_sample() +{ + RefPtr sample = sink_->pull_sample(); + RefPtr buf = sample->get_buffer(); + + for (uint32_t block_id = 0; block_id < buf->n_memory(); block_id++) { + RefPtr buf_mem = buf->get_memory(block_id); + Gst::MapInfo mapinfo; + buf_mem->map(mapinfo, Gst::MAP_READ); + + shared_ptr logic_packet = + sr_context->create_logic_packet(mapinfo.get_data(), buf->get_size(), 1); + + try { + feed_in_logic(dynamic_pointer_cast(logic_packet->payload())); + } catch (bad_alloc&) { + out_of_memory_ = true; + device_->stop(); + buf_mem->unmap(mapinfo); + return Gst::FLOW_ERROR; + } + + buf_mem->unmap(mapinfo); + } + + return Gst::FLOW_OK; +} +#endif + void Session::feed_in_header() { // Nothing to do here for now @@ -1041,18 +1182,13 @@ void Session::feed_in_header() void Session::feed_in_meta(shared_ptr meta) { - for (auto entry : meta->config()) { + for (auto& entry : meta->config()) { switch (entry.first->id()) { case SR_CONF_SAMPLERATE: - // We can't rely on the header to always contain the sample rate, - // so in case it's supplied via a meta packet, we use it. - if (!cur_samplerate_) - cur_samplerate_ = g_variant_get_uint64(entry.second.gobj()); - - /// @todo handle samplerate changes + cur_samplerate_ = g_variant_get_uint64(entry.second.gobj()); break; default: - // Unknown metadata is not an error. + qDebug() << "Received meta data key" << entry.first->id() << ", ignoring."; break; } } @@ -1066,7 +1202,7 @@ void Session::feed_in_trigger() uint64_t sample_count = 0; { - for (const shared_ptr d : all_signal_data_) { + for (const shared_ptr& d : all_signal_data_) { assert(d); uint64_t temp_count = 0; @@ -1116,7 +1252,7 @@ void Session::feed_in_frame_end() if (cur_logic_segment_) cur_logic_segment_->set_complete(); - for (auto entry : cur_analog_segments_) { + for (auto& entry : cur_analog_segments_) { shared_ptr segment = entry.second; segment->set_complete(); } @@ -1132,8 +1268,20 @@ void Session::feed_in_frame_end() void Session::feed_in_logic(shared_ptr logic) { + if (logic->data_length() == 0) { + qDebug() << "WARNING: Received logic packet with 0 samples."; + return; + } + + if (logic->unit_size() > 8) + throw QString(tr("Can't handle more than 64 logic channels.")); + if (!cur_samplerate_) - cur_samplerate_ = device_->read_config(ConfigKey::SAMPLERATE); + try { + cur_samplerate_ = device_->read_config(ConfigKey::SAMPLERATE); + } catch (Error& e) { + // Do nothing + } lock_guard lock(data_mutex_); @@ -1164,24 +1312,31 @@ void Session::feed_in_logic(shared_ptr logic) void Session::feed_in_analog(shared_ptr analog) { + if (analog->num_samples() == 0) { + qDebug() << "WARNING: Received analog packet with 0 samples."; + return; + } + if (!cur_samplerate_) - cur_samplerate_ = device_->read_config(ConfigKey::SAMPLERATE); + try { + cur_samplerate_ = device_->read_config(ConfigKey::SAMPLERATE); + } catch (Error& e) { + // Do nothing + } lock_guard lock(data_mutex_); const vector> channels = analog->channels(); - const unsigned int channel_count = channels.size(); - const size_t sample_count = analog->num_samples() / channel_count; bool sweep_beginning = false; - unique_ptr data(new float[analog->num_samples()]); + unique_ptr data(new float[analog->num_samples() * channels.size()]); analog->get_data_as_float(data.get()); if (signalbases_.empty()) update_signals(); float *channel_data = data.get(); - for (auto channel : channels) { + for (auto& channel : channels) { shared_ptr segment; // Try to get the segment of the channel @@ -1216,8 +1371,8 @@ void Session::feed_in_analog(shared_ptr analog) assert(segment); // Append the samples in the segment - segment->append_interleaved_samples(channel_data++, sample_count, - channel_count); + segment->append_interleaved_samples(channel_data++, analog->num_samples(), + channels.size()); } if (sweep_beginning) { @@ -1253,7 +1408,7 @@ void Session::data_feed_in(shared_ptr device, case SR_DF_LOGIC: try { feed_in_logic(dynamic_pointer_cast(packet->payload())); - } catch (bad_alloc) { + } catch (bad_alloc&) { out_of_memory_ = true; device_->stop(); } @@ -1262,7 +1417,7 @@ void Session::data_feed_in(shared_ptr device, case SR_DF_ANALOG: try { feed_in_analog(dynamic_pointer_cast(packet->payload())); - } catch (bad_alloc) { + } catch (bad_alloc&) { out_of_memory_ = true; device_->stop(); } @@ -1282,6 +1437,15 @@ void Session::data_feed_in(shared_ptr device, // devices use frames, and for those devices, we need to do it here. { lock_guard lock(data_mutex_); + + if (cur_logic_segment_) + cur_logic_segment_->set_complete(); + + for (auto& entry : cur_analog_segments_) { + shared_ptr segment = entry.second; + segment->set_complete(); + } + cur_logic_segment_.reset(); cur_analog_segments_.clear(); } @@ -1297,4 +1461,17 @@ void Session::on_data_saved() data_saved_ = true; } +#ifdef ENABLE_DECODE +void Session::on_new_decoders_selected(vector decoders) +{ + assert(decoders.size() > 0); + + shared_ptr signal = add_decode_signal(); + + if (signal) + for (const srd_decoder* d : decoders) + signal->stack_decoder(d); +} +#endif + } // namespace pv