X-Git-Url: http://git.code-monkey.de/?a=blobdiff_plain;f=pv%2Fviews%2Ftrace%2Fdecodetrace.cpp;h=865c492604fea09e0647defbb6bd5fb95f123d5c;hb=1ed996b433f3391f54846784d76da4fe6b07500f;hp=9eeea1c29ec20beaa7139eb8ea1d9b835754b109;hpb=9e773fecf2d4ab8f2fc6e1cc880b9102c22ec228;p=pulseview.git

diff --git a/pv/views/trace/decodetrace.cpp b/pv/views/trace/decodetrace.cpp
index 9eeea1c..865c492 100644
--- a/pv/views/trace/decodetrace.cpp
+++ b/pv/views/trace/decodetrace.cpp
@@ -21,6 +21,7 @@ extern "C" {
 #include <libsigrokdecode/libsigrokdecode.h>
 }
 
+#include <climits>
 #include <mutex>
 #include <tuple>
 
@@ -31,10 +32,13 @@ extern "C" {
 #include <QAction>
 #include <QApplication>
 #include <QComboBox>
+#include <QFileDialog>
 #include <QFormLayout>
 #include <QLabel>
 #include <QMenu>
+#include <QMessageBox>
 #include <QPushButton>
+#include <QTextStream>
 #include <QToolTip>
 
 #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<uint64_t, uint64_t> 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<pv::data::decode::Annotation> 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<pv::data::decode::Annotation> annotations;
+	vector<Annotation> annotations;
 
 	decode_signal_->get_annotation_subset(annotations, visible_rows_[row],
 		current_segment_, sample_range.first, sample_range.second);
@@ -983,6 +1020,76 @@ 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<Annotation> annotations;
+
+	decode_signal_->get_annotation_subset(annotations, *selected_row_,
+		current_segment_, selected_samplepos_, ULLONG_MAX);
+
+	if (annotations.empty())
+		return;
+
+	GlobalSettings settings;
+	const QString dir = settings.value("MainWindow/SaveDirectory").toString();
+
+	const QString file_name = QFileDialog::getSaveFileName(
+		owner_->view(), tr("Export annotations"), dir, tr("Text Files (*.txt);;All Files (*)"));
+
+	if (file_name.isEmpty())
+		return;
+
+	const QString format = settings.value(GlobalSettings::Key_Dec_ExportFormat).toString();
+	const QString quote = format.contains("%q") ? "\"" : "";
+	const QString class_name = selected_row_->class_name();
+
+	QFile file(file_name);
+	if (file.open(QIODevice::WriteOnly | QIODevice::Truncate | QIODevice::Text)) {
+		QTextStream out_stream(&file);
+
+		for (Annotation &ann : annotations) {
+			const QString sample_range = QString("%1-%2").arg(ann.start_sample()).arg(
+				ann.end_sample());
+
+			QString all_ann_text;
+			for (const QString &s : ann.annotations())
+				all_ann_text = all_ann_text + quote + s + quote + ",";
+			all_ann_text.chop(1);
+
+			const QString first_ann_text = quote + ann.annotations().front() + quote;
+
+			QString out_text = format;
+			out_text = out_text.replace("%s", sample_range);
+			out_text = out_text.replace("%d", decode_signal_->name());
+			out_text = out_text.replace("%c", class_name);
+			out_text = out_text.replace("%1", first_ann_text);
+			out_text = out_text.replace("%a", all_ann_text);
+			out_stream << out_text << '\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