Added missing includes and defintions

[pulseview.git] / pv / view / signal.cpp

diff --git a/pv/view/signal.cpp b/pv/view/signal.cpp
index dadba05aec0ed607927142edeab0e7fac1ec7adc..1f00477a2012d9cb5a3a8e11f6cd046ff62f8682 100644 (file)
--- a/pv/view/signal.cpp
+++ b/pv/view/signal.cpp
@@ -24,6 +24,8 @@
 #include <math.h>
 
 #include <QApplication>
+#include <QFormLayout>
+#include <QMenu>
 
 #include "signal.h"
 #include "view.h"
@@ -48,7 +50,7 @@ const char *const ProbeNames[] = {
        "SCL"
 };
 
-Signal::Signal(pv::SigSession &session, const sr_probe *const probe) :
+Signal::Signal(pv::SigSession &session, sr_probe *const probe) :
        Trace(session, probe->name),
        _probe(probe),
        _name_widget(NULL),
@@ -70,6 +72,12 @@ bool Signal::enabled() const
        return _probe->enabled;
 }
 
+void Signal::enable(bool enable)
+{
+       _probe->enabled = enable;
+       visibility_changed();
+}
+
 const sr_probe* Signal::probe() const
 {
        return _probe;
@@ -92,5 +100,29 @@ void Signal::populate_popup_form(QWidget *parent, QFormLayout *form)
        add_colour_option(parent, form);
 }
 
+QMenu* Signal::create_context_menu(QWidget *parent)
+{
+       QMenu *const menu = Trace::create_context_menu(parent);
+
+       menu->addSeparator();
+
+       QAction *const disable = new QAction(tr("Disable"), this);
+       disable->setShortcuts(QKeySequence::Delete);
+       connect(disable, SIGNAL(triggered()), this, SLOT(on_disable()));
+       menu->addAction(disable);
+
+       return menu;
+}
+
+void Signal::delete_pressed()
+{
+       on_disable();
+}
+
+void Signal::on_disable()
+{
+       enable(false);
+}
+
 } // namespace view
 } // namespace pv