X-Git-Url: https://sigrok.org/gitweb/?p=pulseview.git;a=blobdiff_plain;f=pv%2Ftoolbars%2Fmainbar.cpp;h=b35e14661373d88325eb4a28c1d73a580d587e4f;hp=b798683c33a0fee228553501cadd528c48704a41;hb=b0773a8aa01735d7220284ab7a3e8b5d02b48e9e;hpb=e6d85b8936b5b7aee80b03834f095fd5b52fe6f4

diff --git a/pv/toolbars/mainbar.cpp b/pv/toolbars/mainbar.cpp
index b798683c..b35e1466 100644
--- a/pv/toolbars/mainbar.cpp
+++ b/pv/toolbars/mainbar.cpp
@@ -35,6 +35,7 @@
 
 #include <boost/algorithm/string/join.hpp>
 
+#include <pv/data/mathsignal.hpp>
 #include <pv/devicemanager.hpp>
 #include <pv/devices/hardwaredevice.hpp>
 #include <pv/devices/inputfile.hpp>
@@ -59,6 +60,7 @@ using std::back_inserter;
 using std::copy;
 using std::list;
 using std::make_pair;
+using std::make_shared;
 using std::map;
 using std::max;
 using std::min;
@@ -108,10 +110,11 @@ MainBar::MainBar(Session &session, QWidget *parent, pv::views::trace::View *view
 	sample_rate_("Hz", this),
 	updating_sample_rate_(false),
 	updating_sample_count_(false),
-	sample_count_supported_(false)
+	sample_count_supported_(false),
 #ifdef ENABLE_DECODE
-	, add_decoder_button_(new QToolButton())
+	add_decoder_button_(new QToolButton()),
 #endif
+	add_math_signal_button_(new QToolButton())
 {
 	setObjectName(QString::fromUtf8("MainBar"));
 
@@ -240,6 +243,16 @@ MainBar::MainBar(Session &session, QWidget *parent, pv::views::trace::View *view
 		this, SLOT(on_add_decoder_clicked()));
 #endif
 
+	// Setup the math signal button
+	add_math_signal_button_->setIcon(QIcon(":/icons/add-math-signal.svg"));
+	add_math_signal_button_->setPopupMode(QToolButton::InstantPopup);
+	add_math_signal_button_->setToolTip(tr("Add math signal"));
+	add_math_signal_button_->setShortcut(QKeySequence(Qt::Key_M));
+
+	connect(add_math_signal_button_, SIGNAL(clicked()),
+		this, SLOT(on_add_math_signal_clicked()));
+
+
 	connect(&sample_count_, SIGNAL(value_changed()),
 		this, SLOT(on_sample_count_changed()));
 	connect(&sample_rate_, SIGNAL(value_changed()),
@@ -689,11 +702,10 @@ void MainBar::export_file(shared_ptr<OutputFormat> format, bool selection_only,
 	}
 
 	if (!selection_only) {
-		session_.set_name(QFileInfo(file_name).fileName());
-
-		if (format == session_.device_manager().context()->output_formats()["srzip"])
+		if (format == session_.device_manager().context()->output_formats()["srzip"]) {
 			session_.set_save_path(QFileInfo(file_name).absolutePath());
-		else
+			session_.set_name(QFileInfo(file_name).fileName());
+		} else
 			session_.set_save_path("");
 	}
 
@@ -825,7 +837,7 @@ void MainBar::on_actionSave_triggered()
 		return;
 	}
 
-	QFileInfo fi = QFileInfo(session_.save_path(), session_.name());
+	QFileInfo fi = QFileInfo(QDir(session_.save_path()), session_.name());
 	export_file(session_.device_manager().context()->output_formats()["srzip"], false,
 		fi.absoluteFilePath());
 }
@@ -894,6 +906,12 @@ void MainBar::on_add_decoder_clicked()
 	show_decoder_selector(&session_);
 }
 
+void MainBar::on_add_math_signal_clicked()
+{
+	shared_ptr<data::SignalBase> signal = make_shared<data::MathSignal>(session_);
+	session_.add_generated_signal(signal);
+}
+
 void MainBar::add_toolbar_widgets()
 {
 	addWidget(new_view_button_);
@@ -913,6 +931,7 @@ void MainBar::add_toolbar_widgets()
 	addSeparator();
 	addWidget(add_decoder_button_);
 #endif
+	addWidget(add_math_signal_button_);
 }
 
 bool MainBar::eventFilter(QObject *watched, QEvent *event)