]> sigrok.org Git - pulseview.git/blobdiff - pv/view/header.cpp
Header: Added on_ungroup method
[pulseview.git] / pv / view / header.cpp
index 1c7903dbc372d38d55366b771c49fe3854de9bdf..23cbb0ffa79979d08c764e1809c8248cef406ae3 100644 (file)
 #include "header.h"
 #include "view.h"
 
-#include "../signal.h"
+#include "signal.h"
+#include "tracegroup.h"
 #include "../sigsession.h"
 
-#include <assert.h>
+#include <cassert>
+#include <algorithm>
 
-#include <boost/foreach.hpp>
+#include <boost/iterator/filter_iterator.hpp>
 
-#include <QInputDialog>
+#include <QApplication>
 #include <QMenu>
 #include <QMouseEvent>
 #include <QPainter>
 #include <QRect>
 
-using namespace boost;
-using namespace std;
+#include <pv/widgets/popup.h>
+
+using boost::make_filter_iterator;
+using std::dynamic_pointer_cast;
+using std::max;
+using std::make_pair;
+using std::min;
+using std::pair;
+using std::shared_ptr;
+using std::stable_sort;
+using std::vector;
 
 namespace pv {
 namespace view {
 
+const int Header::Padding = 12;
+const int Header::BaselineOffset = 5;
+
+static bool item_selected(shared_ptr<RowItem> r)
+{
+       return r->selected();
+}
+
 Header::Header(View &parent) :
-       QWidget(&parent),
-       _view(parent),
-       _action_set_name(new QAction(tr("Set &Name..."), this))
+       MarginWidget(parent),
+       _dragging(false)
 {
+       setFocusPolicy(Qt::ClickFocus);
        setMouseTracking(true);
 
-       connect(_action_set_name, SIGNAL(triggered()),
-               this, SLOT(on_action_set_name_triggered()));
+       connect(&_view, SIGNAL(signals_moved()),
+               this, SLOT(on_signals_moved()));
+}
+
+QSize Header::sizeHint() const
+{
+       QRectF max_rect(-Padding, 0, Padding, 0);
+       for (auto &i : _view)
+               if (i->enabled())
+                       max_rect = max_rect.united(i->label_rect(0));
+       return QSize(max_rect.width() + Padding + BaselineOffset, 0);
+}
+
+shared_ptr<RowItem> Header::get_mouse_over_row_item(const QPoint &pt)
+{
+       const int w = width() - BaselineOffset;
+       for (auto &i : _view)
+               if (i->enabled() && i->label_rect(w).contains(pt))
+                       return i;
+       return shared_ptr<RowItem>();
+}
+
+void Header::clear_selection()
+{
+       for (auto &i : _view)
+               i->select(false);
+       update();
+}
+
+void Header::show_popup(const shared_ptr<RowItem> &item)
+{
+       using pv::widgets::Popup;
+
+       Popup *const p = item->create_popup(&_view);
+       if (!p)
+               return;
+
+       const QPoint pt(width() - BaselineOffset, item->get_visual_y());
+       p->set_position(mapToGlobal(pt), Popup::Right);
+       p->show();
 }
 
-void Header::paintEvent(QPaintEvent *event)
+void Header::paintEvent(QPaintEvent*)
 {
-       const int w = width();
-       const vector< shared_ptr<Signal> > &sigs =
-               _view.session().get_signals();
+       // The trace labels are not drawn with the arrows exactly on the
+       // left edge of the widget, because then the selection shadow
+       // would be clipped away.
+       const int w = width() - BaselineOffset;
+
+       vector< shared_ptr<RowItem> > row_items(
+               _view.begin(), _view.end());
+
+       stable_sort(row_items.begin(), row_items.end(),
+               [](const shared_ptr<RowItem> &a, const shared_ptr<RowItem> &b) {
+                       return a->visual_v_offset() < b->visual_v_offset(); });
 
        QPainter painter(this);
        painter.setRenderHint(QPainter::Antialiasing);
 
-       int offset = -_view.v_offset();
-       BOOST_FOREACH(const shared_ptr<Signal> s, sigs)
+       for (const shared_ptr<RowItem> r : row_items)
        {
-               assert(s);
+               assert(r);
+
+               const bool highlight = !_dragging &&
+                       r->label_rect(w).contains(_mouse_point);
+               r->paint_label(painter, w, highlight);
+       }
+
+       painter.end();
+}
+
+void Header::mouseLeftPressEvent(QMouseEvent *event)
+{
+       (void)event;
 
-               const QRect signal_heading_rect(
-                       0, offset, w, View::SignalHeight);
+       const bool ctrl_pressed =
+               QApplication::keyboardModifiers() & Qt::ControlModifier;
 
-               s->paint_label(painter, signal_heading_rect,
-                       s->pt_in_label_rect(signal_heading_rect, _mouse_point));
+       // Clear selection if control is not pressed and this item is unselected
+       if ((!_mouse_down_item || !_mouse_down_item->selected()) &&
+               !ctrl_pressed)
+               for (shared_ptr<RowItem> r : _view)
+                       r->select(false);
 
-               offset += View::SignalHeight;
+       // Set the signal selection state if the item has been clicked
+       if (_mouse_down_item) {
+               if (ctrl_pressed)
+                       _mouse_down_item->select(!_mouse_down_item->selected());
+               else
+                       _mouse_down_item->select(true);
        }
 
-       painter.end();
+       // Save the offsets of any signals which will be dragged
+       for (const shared_ptr<RowItem> r : _view)
+               if (r->selected())
+                       r->drag();
+
+       selection_changed();
+       update();
+}
+
+void Header::mousePressEvent(QMouseEvent *event)
+{
+       assert(event);
+
+       _mouse_down_point = event->pos();
+       _mouse_down_item = get_mouse_over_row_item(event->pos());
+
+       if (event->button() & Qt::LeftButton)
+               mouseLeftPressEvent(event);
+}
+
+void Header::mouseLeftReleaseEvent(QMouseEvent *event)
+{
+       assert(event);
+
+       const bool ctrl_pressed =
+               QApplication::keyboardModifiers() & Qt::ControlModifier;
+
+       // Unselect everything if control is not pressed
+       const shared_ptr<RowItem> mouse_over =
+               get_mouse_over_row_item(event->pos());
+
+       for (auto &r : _view)
+               r->drag_release();
+
+       if (_dragging)
+               _view.restack_all_row_items();
+       else
+       {
+               if (!ctrl_pressed) {
+                       for (shared_ptr<RowItem> r : _view)
+                               if (_mouse_down_item != r)
+                                       r->select(false);
+
+                       if (_mouse_down_item)
+                               show_popup(_mouse_down_item);
+               }
+       }
+
+       _dragging = false;
+}
+
+void Header::mouseReleaseEvent(QMouseEvent *event)
+{
+       assert(event);
+       if (event->button() & Qt::LeftButton)
+               mouseLeftReleaseEvent(event);
+
+       _mouse_down_item = nullptr;
 }
 
 void Header::mouseMoveEvent(QMouseEvent *event)
 {
        assert(event);
        _mouse_point = event->pos();
+
+       if (!(event->buttons() & Qt::LeftButton))
+               return;
+
+       if ((event->pos() - _mouse_down_point).manhattanLength() <
+               QApplication::startDragDistance())
+               return;
+
+       // Check all the drag items share a common owner
+       RowItemOwner *item_owner = nullptr;
+       for (shared_ptr<RowItem> r : _view)
+               if (r->dragging()) {
+                       if (!item_owner)
+                               item_owner = r->owner();
+                       else if(item_owner != r->owner())
+                               return;
+               }
+
+       if (!item_owner)
+               return;
+
+       // Do the drag
+       _dragging = true;
+
+       const int delta = event->pos().y() - _mouse_down_point.y();
+
+       for (std::shared_ptr<RowItem> r : _view)
+               if (r->dragging()) {
+                       r->force_to_v_offset(r->drag_point().y() + delta);
+
+                       // Ensure the trace is selected
+                       r->select();
+               }
+
+       item_owner->restack_items();
+       for (const auto &r : *item_owner)
+               r->animate_to_layout_v_offset();
+       signals_moved();
+
        update();
 }
 
-void Header::leaveEvent(QEvent *event)
+void Header::leaveEvent(QEvent*)
 {
        _mouse_point = QPoint(-1, -1);
        update();
@@ -92,44 +272,98 @@ void Header::leaveEvent(QEvent *event)
 
 void Header::contextMenuEvent(QContextMenuEvent *event)
 {
-       const int w = width();
-       const vector< shared_ptr<Signal> > &sigs =
-               _view.session().get_signals();
+       const shared_ptr<RowItem> r = get_mouse_over_row_item(_mouse_point);
+       if (!r)
+               return;
 
-       int offset = -_view.v_offset();
-       BOOST_FOREACH(const shared_ptr<Signal> s, sigs)
-       {
-               assert(s);
+       QMenu *menu = r->create_context_menu(this);
+       if (!menu)
+               menu = new QMenu(this);
 
-               const QRect signal_heading_rect(
-                       0, offset, w, View::SignalHeight);
+       if (std::count_if(_view.begin(), _view.end(), item_selected) > 1)
+       {
+               menu->addSeparator();
 
-               if(s->pt_in_label_rect(signal_heading_rect, _mouse_point)) {
-                       QMenu menu(this);
-                       menu.addAction(_action_set_name);
+               QAction *const group = new QAction(tr("Group"), this);
+               QList<QKeySequence> shortcuts;
+               shortcuts.append(QKeySequence(Qt::ControlModifier | Qt::Key_G));
+               group->setShortcuts(shortcuts);
+               connect(group, SIGNAL(triggered()), this, SLOT(on_group()));
+               menu->addAction(group);
+       }
 
-                       _context_signal = s;
-                       menu.exec(event->globalPos());
-                       _context_signal.reset();
+       menu->exec(event->globalPos());
+}
 
-                       break;
-               }
+void Header::keyPressEvent(QKeyEvent *e)
+{
+       assert(e);
 
-               offset += View::SignalHeight;
+       switch (e->key())
+       {
+       case Qt::Key_Delete:
+       {
+               for (const shared_ptr<RowItem> r : _view)
+                       if (r->selected())
+                               r->delete_pressed();
+               break;
+       }
        }
 }
 
-void Header::on_action_set_name_triggered()
+void Header::on_signals_moved()
 {
-       boost::shared_ptr<Signal> context_signal = _context_signal;
-       if(!context_signal)
-               return;
+       update();
+}
+
+void Header::on_group()
+{
+       vector< shared_ptr<RowItem> > selected_items(
+               make_filter_iterator(item_selected, _view.begin(), _view.end()),
+               make_filter_iterator(item_selected, _view.end(), _view.end()));
+       stable_sort(selected_items.begin(), selected_items.end(),
+               [](const shared_ptr<RowItem> &a, const shared_ptr<RowItem> &b) {
+                       return a->visual_v_offset() < b->visual_v_offset(); });
+
+       shared_ptr<TraceGroup> group(new TraceGroup());
+       shared_ptr<RowItem> focus_item(
+               _mouse_down_item ? _mouse_down_item : selected_items.front());
+
+       assert(focus_item);
+       assert(focus_item->owner());
+       focus_item->owner()->add_child_item(group);
+
+       // Set the group v_offset here before reparenting
+       group->force_to_v_offset(focus_item->layout_v_offset() +
+               focus_item->v_extents().first);
 
-       const QString new_label = QInputDialog::getText(this, tr("Set Name"),
-               tr("Name"), QLineEdit::Normal, context_signal->get_name());
+       for (size_t i = 0; i < selected_items.size(); i++) {
+               const shared_ptr<RowItem> &r = selected_items[i];
+               assert(r->owner());
+               r->owner()->remove_child_item(r);
+               group->add_child_item(r);
 
-       if(!new_label.isEmpty())
-               context_signal->set_name(new_label);
+               // Put the items at 1-pixel offsets, so that restack will
+               // stack them in the right order
+               r->set_layout_v_offset(i);
+       }
+}
+
+void Header::on_ungroup()
+{
+       bool restart;
+       do {
+               restart = false;
+               for (const shared_ptr<RowItem> r : _view) {
+                       const shared_ptr<TraceGroup> tg =
+                               dynamic_pointer_cast<TraceGroup>(r);
+                       if (tg && tg->selected()) {
+                               tg->ungroup();
+                               restart = true;
+                               break;
+                       }
+               }
+       } while(restart);
 }
 
 } // namespace view