2 * This file is part of the PulseView project.
4 * Copyright (C) 2012 Joel Holdsworth <joel@airwebreathe.org.uk>
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, see <http://www.gnu.org/licenses/>.
24 #include "tracegroup.hpp"
29 #include <boost/iterator/filter_iterator.hpp>
31 #include <QApplication>
33 #include <QMouseEvent>
37 #include <pv/session.hpp>
38 #include <pv/widgets/popup.hpp>
40 using boost::make_filter_iterator;
43 using std::dynamic_pointer_cast;
44 using std::shared_ptr;
45 using std::stable_sort;
52 const int Header::Padding = 12;
54 static bool item_selected(shared_ptr<TraceTreeItem> r)
59 Header::Header(View &parent) :
64 QSize Header::sizeHint() const
66 QRectF max_rect(-Padding, 0, Padding, 0);
67 const vector<shared_ptr<TraceTreeItem>> items(
68 view_.list_by_type<TraceTreeItem>());
71 max_rect = max_rect.united(i->label_rect(QRect()));
72 return QSize(max_rect.width() + Padding, 0);
75 QSize Header::extended_size_hint() const
77 return sizeHint() + QSize(ViewItem::HighlightRadius, 0);
80 vector< shared_ptr<ViewItem> > Header::items()
82 const vector<shared_ptr<TraceTreeItem>> items(
83 view_.list_by_type<TraceTreeItem>());
84 return vector< shared_ptr<ViewItem> >(items.begin(), items.end());
87 shared_ptr<ViewItem> Header::get_mouse_over_item(const QPoint &pt)
89 const QRect r(0, 0, width(), height());
90 const vector<shared_ptr<TraceTreeItem>> items(
91 view_.list_by_type<TraceTreeItem>());
92 for (auto i = items.rbegin(); i != items.rend(); i++)
93 if ((*i)->enabled() && (*i)->label_rect(r).contains(pt))
95 return shared_ptr<TraceTreeItem>();
98 void Header::paintEvent(QPaintEvent*)
100 const QRect rect(0, 0, width(), height());
102 vector< shared_ptr<RowItem> > items(view_.list_by_type<RowItem>());
104 stable_sort(items.begin(), items.end(),
105 [](const shared_ptr<RowItem> &a, const shared_ptr<RowItem> &b) {
106 return a->drag_point(QRect()).y() < b->drag_point(QRect()).y(); });
108 QPainter painter(this);
109 painter.setRenderHint(QPainter::Antialiasing);
111 for (const shared_ptr<RowItem> r : items) {
114 const bool highlight = !item_dragging_ &&
115 r->label_rect(rect).contains(mouse_point_);
116 r->paint_label(painter, rect, highlight);
122 void Header::contextMenuEvent(QContextMenuEvent *event)
124 const shared_ptr<ViewItem> r = get_mouse_over_item(mouse_point_);
128 QMenu *menu = r->create_context_menu(this);
130 menu = new QMenu(this);
132 const vector< shared_ptr<TraceTreeItem> > items(
133 view_.list_by_type<TraceTreeItem>());
134 if (count_if(items.begin(), items.end(), item_selected) > 1) {
135 menu->addSeparator();
137 QAction *const group = new QAction(tr("Group"), this);
138 QList<QKeySequence> shortcuts;
139 shortcuts.append(QKeySequence(Qt::ControlModifier | Qt::Key_G));
140 group->setShortcuts(shortcuts);
141 connect(group, SIGNAL(triggered()), this, SLOT(on_group()));
142 menu->addAction(group);
145 menu->exec(event->globalPos());
148 void Header::keyPressEvent(QKeyEvent *event)
152 MarginWidget::keyPressEvent(event);
154 if (event->key() == Qt::Key_G && event->modifiers() == Qt::ControlModifier)
156 else if (event->key() == Qt::Key_U && event->modifiers() == Qt::ControlModifier)
160 void Header::on_group()
162 const vector< shared_ptr<TraceTreeItem> > items(
163 view_.list_by_type<TraceTreeItem>());
164 vector< shared_ptr<TraceTreeItem> > selected_items(
165 make_filter_iterator(item_selected, items.begin(), items.end()),
166 make_filter_iterator(item_selected, items.end(), items.end()));
167 stable_sort(selected_items.begin(), selected_items.end(),
168 [](const shared_ptr<TraceTreeItem> &a, const shared_ptr<TraceTreeItem> &b) {
169 return a->visual_v_offset() < b->visual_v_offset(); });
171 shared_ptr<TraceGroup> group(new TraceGroup());
172 shared_ptr<TraceTreeItem> mouse_down_item(
173 dynamic_pointer_cast<TraceTreeItem>(mouse_down_item_));
174 shared_ptr<TraceTreeItem> focus_item(
175 mouse_down_item ? mouse_down_item : selected_items.front());
178 assert(focus_item->owner());
179 focus_item->owner()->add_child_item(group);
181 // Set the group v_offset here before reparenting
182 group->force_to_v_offset(focus_item->layout_v_offset() +
183 focus_item->v_extents().first);
185 for (size_t i = 0; i < selected_items.size(); i++) {
186 const shared_ptr<TraceTreeItem> &r = selected_items[i];
188 r->owner()->remove_child_item(r);
189 group->add_child_item(r);
191 // Put the items at 1-pixel offsets, so that restack will
192 // stack them in the right order
193 r->set_layout_v_offset(i);
197 void Header::on_ungroup()
202 const vector< shared_ptr<TraceGroup> > groups(
203 view_.list_by_type<TraceGroup>());
204 for (const shared_ptr<TraceGroup> tg : groups)
205 if (tg->selected()) {