pv/view/header.cpp

   1 /*
   2  * This file is part of the PulseView project.
   3  *
   4  * Copyright (C) 2012 Joel Holdsworth <joel@airwebreathe.org.uk>
   5  *
   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.
  10  *
  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.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  19  */
  20
  21 #include "header.h"
  22 #include "view.h"
  23
  24 #include "signal.h"
  25 #include "../sigsession.h"
  26
  27 #include <assert.h>
  28
  29 #include <boost/foreach.hpp>
  30
  31 #include <QApplication>
  32 #include <QColorDialog>
  33 #include <QInputDialog>
  34 #include <QMenu>
  35 #include <QMouseEvent>
  36 #include <QPainter>
  37 #include <QRect>
  38
  39 using namespace boost;
  40 using namespace std;
  41
  42 namespace pv {
  43 namespace view {
  44
  45 Header::Header(View &parent) :
  46         MarginWidget(parent),
  47         _action_set_name(new QAction(tr("Set &Name..."), this)),
  48         _action_set_colour(new QAction(tr("Set &Colour..."), this))
  49 {
  50         setMouseTracking(true);
  51
  52         connect(_action_set_name, SIGNAL(triggered()),
  53                 this, SLOT(on_action_set_name_triggered()));
  54         connect(_action_set_colour, SIGNAL(triggered()),
  55                 this, SLOT(on_action_set_colour_triggered()));
  56
  57         connect(&_view.session(), SIGNAL(signals_changed()),
  58                 this, SLOT(on_signals_changed()));
  59
  60         connect(&_view, SIGNAL(signals_moved()),
  61                 this, SLOT(on_signals_moved()));
  62 }
  63
  64 shared_ptr<Trace> Header::get_mouse_over_trace(const QPoint &pt)
  65 {
  66         const int w = width();
  67         const vector< shared_ptr<Trace> > traces(_view.get_traces());
  68
  69         const int v_offset = _view.v_offset();
  70         BOOST_FOREACH(const shared_ptr<Trace> t, traces)
  71         {
  72                 assert(t);
  73
  74                 if (t->pt_in_label_rect(t->get_v_offset() - v_offset,
  75                         0, w, pt))
  76                         return t;
  77         }
  78
  79         return shared_ptr<Trace>();
  80 }
  81
  82 void Header::clear_selection()
  83 {
  84         const vector< shared_ptr<Trace> > traces(_view.get_traces());
  85         BOOST_FOREACH(const shared_ptr<Trace> t, traces) {
  86                 assert(t);
  87                 t->select(false);
  88         }
  89
  90         update();
  91 }
  92
  93 void Header::paintEvent(QPaintEvent*)
  94 {
  95         const int w = width();
  96         const vector< shared_ptr<Trace> > traces(_view.get_traces());
  97
  98         QPainter painter(this);
  99         painter.setRenderHint(QPainter::Antialiasing);
 100
 101         const int v_offset = _view.v_offset();
 102         const bool dragging = !_drag_traces.empty();
 103         BOOST_FOREACH(const shared_ptr<Trace> t, traces)
 104         {
 105                 assert(t);
 106
 107                 const int y = t->get_v_offset() - v_offset;
 108                 const bool highlight = !dragging && t->pt_in_label_rect(
 109                         y, 0, w, _mouse_point);
 110                 t->paint_label(painter, y, w, highlight);
 111         }
 112
 113         painter.end();
 114 }
 115
 116 void Header::mousePressEvent(QMouseEvent *event)
 117 {
 118         assert(event);
 119
 120         const vector< shared_ptr<Trace> > traces(_view.get_traces());
 121
 122         if (event->button() & Qt::LeftButton) {
 123                 _mouse_down_point = event->pos();
 124
 125                 // Save the offsets of any signals which will be dragged
 126                 BOOST_FOREACH(const shared_ptr<Trace> t, traces)
 127                         if (t->selected())
 128                                 _drag_traces.push_back(
 129                                         make_pair(t, t->get_v_offset()));
 130         }
 131
 132         // Select the signal if it has been clicked
 133         const shared_ptr<Trace> mouse_over_trace =
 134                 get_mouse_over_trace(event->pos());
 135         if (mouse_over_trace) {
 136                 if (mouse_over_trace->selected())
 137                         mouse_over_trace->select(false);
 138                 else {
 139                         mouse_over_trace->select(true);
 140
 141                         if (~QApplication::keyboardModifiers() &
 142                                 Qt::ControlModifier)
 143                                 _drag_traces.clear();
 144
 145                         // Add the signal to the drag list
 146                         if (event->button() & Qt::LeftButton)
 147                                 _drag_traces.push_back(
 148                                         make_pair(mouse_over_trace,
 149                                         mouse_over_trace->get_v_offset()));
 150                 }
 151         }
 152
 153         if (~QApplication::keyboardModifiers() & Qt::ControlModifier) {
 154                 // Unselect all other signals because the Ctrl is not
 155                 // pressed
 156                 BOOST_FOREACH(const shared_ptr<Trace> t, traces)
 157                         if (t != mouse_over_trace)
 158                                 t->select(false);
 159         }
 160
 161         selection_changed();
 162         update();
 163 }
 164
 165 void Header::mouseReleaseEvent(QMouseEvent *event)
 166 {
 167         assert(event);
 168         if (event->button() == Qt::LeftButton) {
 169                 _drag_traces.clear();
 170                 _view.normalize_layout();
 171         }
 172 }
 173
 174 void Header::mouseMoveEvent(QMouseEvent *event)
 175 {
 176         assert(event);
 177         _mouse_point = event->pos();
 178
 179         // Move the signals if we are dragging
 180         if (!_drag_traces.empty()) {
 181                 const int delta = event->pos().y() - _mouse_down_point.y();
 182
 183                 for (std::list<std::pair<boost::weak_ptr<Trace>,
 184                         int> >::iterator i = _drag_traces.begin();
 185                         i != _drag_traces.end(); i++) {
 186                         const boost::shared_ptr<Trace> trace((*i).first);
 187                         if (trace) {
 188                                 const int y = (*i).second + delta;
 189                                 const int y_snap =
 190                                         ((y + View::SignalSnapGridSize / 2) /
 191                                                 View::SignalSnapGridSize) *
 192                                                 View::SignalSnapGridSize;
 193                                 trace->set_v_offset(y_snap);
 194
 195                                 // Ensure the trace is selected
 196                                 trace->select();
 197                         }
 198
 199                 }
 200
 201                 signals_moved();
 202         }
 203
 204         update();
 205 }
 206
 207 void Header::leaveEvent(QEvent*)
 208 {
 209         _mouse_point = QPoint(-1, -1);
 210         update();
 211 }
 212
 213 void Header::contextMenuEvent(QContextMenuEvent *event)
 214 {
 215         const shared_ptr<Trace> t = get_mouse_over_trace(_mouse_point);
 216
 217         if (!t)
 218                 return;
 219
 220         QMenu menu(this);
 221         menu.addAction(_action_set_name);
 222         menu.addAction(_action_set_colour);
 223
 224         _context_trace = t;
 225         menu.exec(event->globalPos());
 226         _context_trace.reset();
 227 }
 228
 229 void Header::on_action_set_name_triggered()
 230 {
 231         bool ok = false;
 232
 233         shared_ptr<view::Trace> context_trace = _context_trace;
 234         if (!context_trace)
 235                 return;
 236
 237         const QString new_label = QInputDialog::getText(this, tr("Set Name"),
 238                 tr("Name"), QLineEdit::Normal, context_trace->get_name(), &ok);
 239
 240         if (ok)
 241                 context_trace->set_name(new_label);
 242 }
 243
 244 void Header::on_action_set_colour_triggered()
 245 {
 246         shared_ptr<view::Trace> context_trace = _context_trace;
 247         if (!context_trace)
 248                 return;
 249
 250         const QColor new_colour = QColorDialog::getColor(
 251                 context_trace->get_colour(), this, tr("Set Colour"));
 252
 253         if (new_colour.isValid())
 254                 context_trace->set_colour(new_colour);
 255 }
 256
 257 void Header::on_signals_changed()
 258 {
 259         const vector< shared_ptr<Trace> > traces(_view.get_traces());
 260         BOOST_FOREACH(shared_ptr<Trace> t, traces) {
 261                 assert(t);
 262                 connect(t.get(), SIGNAL(text_changed()), this, SLOT(update()));
 263         }
 264 }
 265
 266 void Header::on_signals_moved()
 267 {
 268         update();
 269 }
 270
 271
 272 } // namespace view
 273 } // namespace pv