X-Git-Url: https://sigrok.org/gitweb/?p=pulseview.git;a=blobdiff_plain;f=pv%2Fview%2Fheader.cpp;h=a2a7c505cfd420c5b6b66e3e6224795c7ecdabe9;hp=ba16247a8c8e5609245bc8233943295ff068b32b;hb=bc01bb31fb03f69c8ceef57676b3399d7a08dcd9;hpb=91e8bf08c9799cb1041d26bf2ff6b67abf9a420b

diff --git a/pv/view/header.cpp b/pv/view/header.cpp
index ba16247a..a2a7c505 100644
--- a/pv/view/header.cpp
+++ b/pv/view/header.cpp
@@ -18,15 +18,16 @@
  * 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>
@@ -34,217 +35,262 @@
 #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(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()
@@ -252,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