]> sigrok.org Git - pulseview.git/blobdiff - pv/view/header.cpp
Header: Make get_mouse_over_item return shared_ptr<ViewItem>
[pulseview.git] / pv / view / header.cpp
index 77a887f9fe4c9b6b232fa78aaf31cd11b3b5ecd3..a2a7c505cfd420c5b6b66e3e6224795c7ecdabe9 100644 (file)
  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  */
 
-#include "header.h"
-#include "view.h"
+#include "header.hpp"
+#include "view.hpp"
 
-#include "signal.h"
-#include "../sigsession.h"
+#include "signal.hpp"
+#include "tracegroup.hpp"
 
-#include <assert.h>
+#include <cassert>
+#include <algorithm>
 
-#include <boost/foreach.hpp>
+#include <boost/iterator/filter_iterator.hpp>
 
 #include <QApplication>
 #include <QMenu>
 #include <QPainter>
 #include <QRect>
 
-#include <pv/widgets/popup.h>
+#include <pv/session.hpp>
+#include <pv/widgets/popup.hpp>
 
-using namespace boost;
-using namespace std;
+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 {
 
-Header::Header(View &parent) :
-       MarginWidget(parent),
-       _dragging(false)
-{
-       setMouseTracking(true);
+const int Header::Padding = 12;
+const int Header::BaselineOffset = 5;
 
-       connect(&_view.session(), SIGNAL(signals_changed()),
-               this, SLOT(on_signals_changed()));
+static bool item_selected(shared_ptr<RowItem> r)
+{
+       return r->selected();
+}
 
-       connect(&_view, SIGNAL(signals_moved()),
+Header::Header(View &parent) :
+       MarginWidget(parent)
+{
+       connect(&view_, SIGNAL(signals_moved()),
                this, SLOT(on_signals_moved()));
 }
 
-shared_ptr<Trace> Header::get_mouse_over_trace(const QPoint &pt)
+QSize Header::sizeHint() const
 {
-       const int w = width();
-       const vector< shared_ptr<Trace> > traces(_view.get_traces());
+       QRectF max_rect(-Padding, 0, Padding, 0);
+       for (auto &i : view_)
+               if (i->enabled())
+                       max_rect = max_rect.united(i->label_rect(QRect()));
+       return QSize(max_rect.width() + Padding + BaselineOffset, 0);
+}
 
-       BOOST_FOREACH(const shared_ptr<Trace> t, traces)
-       {
-               assert(t);
-               if (t->pt_in_label_rect(0, w, pt))
-                       return t;
-       }
+QSize Header::extended_size_hint() const
+{
+       return sizeHint() + QSize(ViewItem::HighlightRadius, 0);
+}
 
-       return shared_ptr<Trace>();
+shared_ptr<ViewItem> Header::get_mouse_over_item(const QPoint &pt)
+{
+       const QRect r(0, 0, width() - BaselineOffset, height());
+       for (auto &i : view_)
+               if (i->enabled() && i->label_rect(r).contains(pt))
+                       return i;
+       return shared_ptr<RowItem>();
 }
 
 void Header::clear_selection()
 {
-       const vector< shared_ptr<Trace> > traces(_view.get_traces());
-       BOOST_FOREACH(const shared_ptr<Trace> t, traces) {
-               assert(t);
-               t->select(false);
-       }
-
+       for (auto &i : view_)
+               i->select(false);
        update();
 }
 
 void Header::paintEvent(QPaintEvent*)
 {
-       const int w = width();
-       const vector< shared_ptr<Trace> > traces(_view.get_traces());
+       // 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 QRect rect(0, 0, width() - BaselineOffset, height());
+
+       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);
 
-       const bool dragging = !_drag_traces.empty();
-       BOOST_FOREACH(const shared_ptr<Trace> t, traces)
+       for (const shared_ptr<RowItem> r : row_items)
        {
-               assert(t);
+               assert(r);
 
-               const bool highlight = !dragging && t->pt_in_label_rect(
-                       0, w, _mouse_point);
-               t->paint_label(painter, w, highlight);
+               const bool highlight = !dragging_ &&
+                       r->label_rect(rect).contains(mouse_point_);
+               r->paint_label(painter, rect, highlight);
        }
 
        painter.end();
 }
 
-void Header::mousePressEvent(QMouseEvent *event)
+void Header::mouseLeftPressEvent(QMouseEvent *event)
 {
-       assert(event);
+       (void)event;
 
-       const vector< shared_ptr<Trace> > traces(_view.get_traces());
+       const bool ctrl_pressed =
+               QApplication::keyboardModifiers() & Qt::ControlModifier;
 
-       if (event->button() & Qt::LeftButton) {
-               _mouse_down_point = event->pos();
+       // 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);
 
-               // Save the offsets of any signals which will be dragged
-               BOOST_FOREACH(const shared_ptr<Trace> t, traces)
-                       if (t->selected())
-                               _drag_traces.push_back(
-                                       make_pair(t, t->get_v_offset()));
-       }
-
-       // Select the signal if it has been clicked
-       const shared_ptr<Trace> mouse_over_trace =
-               get_mouse_over_trace(event->pos());
-       if (mouse_over_trace) {
-               if (mouse_over_trace->selected())
-                       mouse_over_trace->select(false);
-               else {
-                       mouse_over_trace->select(true);
-
-                       if (~QApplication::keyboardModifiers() &
-                               Qt::ControlModifier)
-                               _drag_traces.clear();
-
-                       // Add the signal to the drag list
-                       if (event->button() & Qt::LeftButton)
-                               _drag_traces.push_back(
-                                       make_pair(mouse_over_trace,
-                                       mouse_over_trace->get_v_offset()));
-               }
+       // 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);
        }
 
-       if (~QApplication::keyboardModifiers() & Qt::ControlModifier) {
-               // Unselect all other signals because the Ctrl is not
-               // pressed
-               BOOST_FOREACH(const shared_ptr<Trace> t, traces)
-                       if (t != mouse_over_trace)
-                               t->select(false);
-       }
+       // 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::mouseReleaseEvent(QMouseEvent *event)
+void Header::mousePressEvent(QMouseEvent *event)
 {
-       using pv::widgets::Popup;
+       assert(event);
+
+       mouse_down_point_ = event->pos();
+       mouse_down_item_ = get_mouse_over_item(event->pos());
+
+       if (event->button() & Qt::LeftButton)
+               mouseLeftPressEvent(event);
+}
 
+void Header::mouseLeftReleaseEvent(QMouseEvent *event)
+{
        assert(event);
-       if (event->button() == Qt::LeftButton) {
-               if (_dragging)
-                       _view.normalize_layout();
-               else
-               {
-                       const shared_ptr<Trace> mouse_over_trace =
-                               get_mouse_over_trace(event->pos());
-                       if (mouse_over_trace) {
-                               Popup *const p =
-                                       mouse_over_trace->create_popup(&_view);
-                               p->set_position(mapToGlobal(QPoint(width(),
-                                       mouse_over_trace->get_y())),
-                                       Popup::Right);
-                               p->show();
-                       }
-               }
 
-               _dragging = false;
-               _drag_traces.clear();
+       const bool ctrl_pressed =
+               QApplication::keyboardModifiers() & Qt::ControlModifier;
+
+       // Unselect everything if control is not pressed
+       const shared_ptr<ViewItem> mouse_over =
+               get_mouse_over_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();
+       mouse_point_ = event->pos();
 
        if (!(event->buttons() & Qt::LeftButton))
                return;
 
-       if ((event->pos() - _mouse_down_point).manhattanLength() <
+       if ((event->pos() - mouse_down_point_).manhattanLength() <
                QApplication::startDragDistance())
                return;
 
-       // Move the signals if we are dragging
-       if (!_drag_traces.empty())
-       {
-               _dragging = true;
-
-               const int delta = event->pos().y() - _mouse_down_point.y();
-
-               for (std::list<std::pair<boost::weak_ptr<Trace>,
-                       int> >::iterator i = _drag_traces.begin();
-                       i != _drag_traces.end(); i++) {
-                       const boost::shared_ptr<Trace> trace((*i).first);
-                       if (trace) {
-                               const int y = (*i).second + delta;
-                               const int y_snap =
-                                       ((y + View::SignalSnapGridSize / 2) /
-                                               View::SignalSnapGridSize) *
-                                               View::SignalSnapGridSize;
-                               trace->set_v_offset(y_snap);
-
-                               // Ensure the trace is selected
-                               trace->select();
-                       }
-                       
+       // 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;
                }
 
-               signals_moved();
-       }
+       if (!item_owner)
+               return;
 
-       update();
-}
+       // Do the drag
+       dragging_ = true;
+
+       const QPoint delta = event->pos() - mouse_down_point_;
+
+       for (std::shared_ptr<RowItem> r : view_)
+               if (r->dragging()) {
+                       r->drag_by(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();
 
-void Header::leaveEvent(QEvent*)
-{
-       _mouse_point = QPoint(-1, -1);
        update();
 }
 
 void Header::contextMenuEvent(QContextMenuEvent *event)
 {
-       const shared_ptr<Trace> t = get_mouse_over_trace(_mouse_point);
+       const shared_ptr<ViewItem> r = get_mouse_over_item(mouse_point_);
+       if (!r)
+               return;
+
+       QMenu *menu = r->create_context_menu(this);
+       if (!menu)
+               menu = new QMenu(this);
 
-       if (t)
-               t->create_context_menu(this)->exec(event->globalPos());
+       if (std::count_if(view_.begin(), view_.end(), item_selected) > 1)
+       {
+               menu->addSeparator();
+
+               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);
+       }
+
+       menu->exec(event->globalPos());
 }
 
-void Header::on_signals_changed()
+void Header::keyPressEvent(QKeyEvent *e)
 {
-       const vector< shared_ptr<Trace> > traces(_view.get_traces());
-       BOOST_FOREACH(shared_ptr<Trace> t, traces) {
-               assert(t);
-               connect(t.get(), SIGNAL(visibility_changed()),
-                       this, SLOT(update()));
-               connect(t.get(), SIGNAL(text_changed()),
-                       this, SLOT(update()));
-               connect(t.get(), SIGNAL(colour_changed()),
-                       this, SLOT(update()));
+       assert(e);
+
+       if (e->key() == Qt::Key_Delete)
+       {
+               for (const shared_ptr<RowItem> r : view_)
+                       if (r->selected())
+                               r->delete_pressed();
        }
+       else if (e->key() == Qt::Key_G && e->modifiers() == Qt::ControlModifier)
+               on_group();
+       else if (e->key() == Qt::Key_U && e->modifiers() == Qt::ControlModifier)
+               on_ungroup();
 }
 
 void Header::on_signals_moved()
@@ -254,6 +298,57 @@ void Header::on_signals_moved()
        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> mouse_down_item(
+               std::dynamic_pointer_cast<RowItem>(mouse_down_item_));
+       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);
+
+       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);
+
+               // 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
 } // namespace pv