From: Soeren Apel Date: Sun, 5 Aug 2018 16:18:22 +0000 (+0200) Subject: Implement annotation export from the DecodeTrace context menu X-Git-Url: https://sigrok.org/gitaction?a=commitdiff_plain;h=be843692f664e44d26458063042cc5d41a13df57;p=pulseview.git Implement annotation export from the DecodeTrace context menu --- diff --git a/pv/views/trace/decodetrace.cpp b/pv/views/trace/decodetrace.cpp index 9eeea1c2..300fd2ec 100644 --- a/pv/views/trace/decodetrace.cpp +++ b/pv/views/trace/decodetrace.cpp @@ -21,6 +21,7 @@ extern "C" { #include } +#include #include #include @@ -31,10 +32,13 @@ extern "C" { #include #include #include +#include #include #include #include +#include #include +#include #include #include "decodetrace.hpp" @@ -321,6 +325,39 @@ QMenu* DecodeTrace::create_header_context_menu(QWidget *parent) return menu; } +QMenu* DecodeTrace::create_view_context_menu(QWidget *parent, QPoint &click_pos) +{ + try { + selected_row_ = &visible_rows_[get_row_at_point(click_pos)]; + } catch (out_of_range&) { + selected_row_ = nullptr; + } + + const pair sample_range = + get_sample_range(click_pos.x(), click_pos.x() + 1); + selected_samplepos_ = sample_range.first; + + QMenu *const menu = new QMenu(parent); + + QAction *const export_row = + new QAction(tr("Export all annotations for this row"), this); + export_row->setIcon(QIcon::fromTheme("document-save-as", + QIcon(":/icons/document-save-as.png"))); + connect(export_row, SIGNAL(triggered()), + this, SLOT(on_export_row())); + menu->addAction(export_row); + + QAction *const export_row_from_here = + new QAction(tr("Export annotations for this row, starting here"), this); + export_row_from_here->setIcon(QIcon::fromTheme("document-save-as", + QIcon(":/icons/document-save-as.png"))); + connect(export_row_from_here, SIGNAL(triggered()), + this, SLOT(on_export_row_from_here())); + menu->addAction(export_row_from_here); + + return menu; +} + void DecodeTrace::draw_annotations(vector annotations, QPainter &p, int h, const ViewItemPaintParams &pp, int y, QColor row_color, int row_title_width) @@ -699,7 +736,7 @@ const QString DecodeTrace::get_annotation_at_point(const QPoint &point) if (row < 0) return QString(); - vector annotations; + vector annotations; decode_signal_->get_annotation_subset(annotations, visible_rows_[row], current_segment_, sample_range.first, sample_range.second); @@ -983,6 +1020,55 @@ void DecodeTrace::on_show_hide_decoder(int index) owner_->row_item_appearance_changed(false, true); } +void DecodeTrace::on_export_row() +{ + selected_samplepos_ = 0; + on_export_row_from_here(); +} + +void DecodeTrace::on_export_row_from_here() +{ + using namespace pv::data::decode; + + if (!selected_row_) + return; + + vector annotations; + + decode_signal_->get_annotation_subset(annotations, *selected_row_, + current_segment_, selected_samplepos_, ULLONG_MAX); + + if (annotations.empty()) + return; + + QSettings settings; + const QString dir = settings.value(SettingSaveDirectory).toString(); + + const QString file_name = QFileDialog::getSaveFileName( + owner_->view(), tr("Export annotations"), dir, tr("Text Files (*.txt);;All Files (*)")); + + if (file_name.isEmpty()) + return; + + QFile file(file_name); + if (file.open(QIODevice::WriteOnly | QIODevice::Truncate | QIODevice::Text)) { + QTextStream out_stream(&file); + + for (Annotation &ann : annotations) + out_stream << ann.annotations().front() << '\n'; + + if (out_stream.status() == QTextStream::Ok) + return; + } + + QMessageBox msg(owner_->view()); + msg.setText(tr("Error")); + msg.setInformativeText(tr("File %1 could not be written to.").arg(file_name)); + msg.setStandardButtons(QMessageBox::Ok); + msg.setIcon(QMessageBox::Warning); + msg.exec(); +} + } // namespace trace } // namespace views } // namespace pv diff --git a/pv/views/trace/decodetrace.hpp b/pv/views/trace/decodetrace.hpp index 4d9e59c1..8f8765c1 100644 --- a/pv/views/trace/decodetrace.hpp +++ b/pv/views/trace/decodetrace.hpp @@ -120,6 +120,8 @@ public: QMenu* create_header_context_menu(QWidget *parent); + virtual QMenu* create_view_context_menu(QWidget *parent, QPoint &click_pos); + void delete_pressed(); private: @@ -198,6 +200,9 @@ private Q_SLOTS: void on_show_hide_decoder(int index); + void on_export_row(); + void on_export_row_from_here(); + private: pv::Session &session_; shared_ptr decode_signal_; @@ -208,6 +213,9 @@ private: map init_state_map_; // init state selector -> decode channel ID list< shared_ptr > bindings_; + data::decode::Row *selected_row_; + uint64_t selected_samplepos_; + vector decoder_forms_; map row_title_widths_; diff --git a/pv/views/trace/view.cpp b/pv/views/trace/view.cpp index 99e75bfd..1e4f07aa 100644 --- a/pv/views/trace/view.cpp +++ b/pv/views/trace/view.cpp @@ -1254,11 +1254,13 @@ bool View::eventFilter(QObject *object, QEvent *event) void View::contextMenuEvent(QContextMenuEvent *event) { - const shared_ptr r = viewport_->get_mouse_over_item(event->pos()); + QPoint pos = event->pos() - QPoint(0, ruler_->sizeHint().height()); + + const shared_ptr r = viewport_->get_mouse_over_item(pos); if (!r) return; - QMenu *menu = r->create_view_context_menu(this); + QMenu *menu = r->create_view_context_menu(this, pos); if (menu) menu->exec(event->globalPos()); } diff --git a/pv/views/trace/viewitem.cpp b/pv/views/trace/viewitem.cpp index ee0720ea..43c9ca26 100644 --- a/pv/views/trace/viewitem.cpp +++ b/pv/views/trace/viewitem.cpp @@ -88,9 +88,10 @@ QMenu* ViewItem::create_header_context_menu(QWidget *parent) return new QMenu(parent); } -QMenu* ViewItem::create_view_context_menu(QWidget *parent) +QMenu* ViewItem::create_view_context_menu(QWidget *parent, QPoint &click_pos) { (void)parent; + (void)click_pos; return nullptr; } diff --git a/pv/views/trace/viewitem.hpp b/pv/views/trace/viewitem.hpp index e07b26b8..89a57e5e 100644 --- a/pv/views/trace/viewitem.hpp +++ b/pv/views/trace/viewitem.hpp @@ -23,6 +23,7 @@ #include #include +#include #include "viewitempaintparams.hpp" @@ -157,7 +158,7 @@ public: public: virtual QMenu* create_header_context_menu(QWidget *parent); - virtual QMenu* create_view_context_menu(QWidget *parent); + virtual QMenu* create_view_context_menu(QWidget *parent, QPoint &click_pos); virtual pv::widgets::Popup* create_popup(QWidget *parent);