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         QWidget(&parent),
  47         _view(parent),
  48         _action_set_name(new QAction(tr("Set &Name..."), this)),
  49         _action_set_colour(new QAction(tr("Set &Colour..."), this))
  50 {
  51         setMouseTracking(true);
  52
  53         connect(_action_set_name, SIGNAL(triggered()),
  54                 this, SLOT(on_action_set_name_triggered()));
  55         connect(_action_set_colour, SIGNAL(triggered()),
  56                 this, SLOT(on_action_set_colour_triggered()));
  57
  58         connect(&_view, SIGNAL(signals_moved()),
  59                 this, SLOT(on_signals_moved()));
  60 }
  61
  62 boost::shared_ptr<pv::view::Signal> Header::get_mouse_over_signal(
  63         const QPoint &pt)
  64 {
  65         const int w = width();
  66         const vector< shared_ptr<Signal> > sigs(
  67                 _view.session().get_signals());
  68
  69         const int v_offset = _view.v_offset();
  70         BOOST_FOREACH(const shared_ptr<Signal> s, sigs)
  71         {
  72                 assert(s);
  73
  74                 const QRect signal_heading_rect(
  75                         0, s->get_v_offset() - v_offset,
  76                         w, View::SignalHeight);
  77
  78                 if (s->pt_in_label_rect(signal_heading_rect, pt))
  79                         return s;
  80         }
  81
  82         return shared_ptr<Signal>();
  83 }
  84
  85 void Header::paintEvent(QPaintEvent *event)
  86 {
  87         const int w = width();
  88         const vector< shared_ptr<Signal> > sigs(
  89                 _view.session().get_signals());
  90
  91         QPainter painter(this);
  92         painter.setRenderHint(QPainter::Antialiasing);
  93
  94         const int v_offset = _view.v_offset();
  95         const bool dragging = !_drag_sigs.empty();
  96         BOOST_FOREACH(const shared_ptr<Signal> s, sigs)
  97         {
  98                 assert(s);
  99
 100                 const QRect signal_heading_rect(
 101                         0, s->get_v_offset() - v_offset,
 102                         w, View::SignalHeight);
 103
 104                 const bool highlight = !dragging && s->pt_in_label_rect(
 105                         signal_heading_rect, _mouse_point);
 106                 s->paint_label(painter, signal_heading_rect, highlight);
 107         }
 108
 109         painter.end();
 110 }
 111
 112 void Header::mousePressEvent(QMouseEvent *event)
 113 {
 114         assert(event);
 115
 116         const vector< shared_ptr<Signal> > sigs(
 117                 _view.session().get_signals());
 118
 119         if (event->button() & Qt::LeftButton) {
 120                 _mouse_down_point = event->pos();
 121
 122                 // Save the offsets of any signals which will be dragged
 123                 BOOST_FOREACH(const shared_ptr<Signal> s, sigs)
 124                         if (s->selected())
 125                                 _drag_sigs.push_back(
 126                                         make_pair(s, s->get_v_offset()));
 127         }
 128
 129         // Select the signal if it has been clicked
 130         const shared_ptr<Signal> mouse_over_signal =
 131                 get_mouse_over_signal(event->pos());
 132         if (mouse_over_signal) {
 133                 if (mouse_over_signal->selected())
 134                         mouse_over_signal->select(false);
 135                 else {
 136                         mouse_over_signal->select(true);
 137
 138                         if (~QApplication::keyboardModifiers() &
 139                                 Qt::ControlModifier)
 140                                 _drag_sigs.clear();
 141
 142                         // Add the signal to the drag list
 143                         if (event->button() & Qt::LeftButton)
 144                                 _drag_sigs.push_back(
 145                                         make_pair(mouse_over_signal,
 146                                         mouse_over_signal->get_v_offset()));
 147                 }
 148         }
 149
 150         if (~QApplication::keyboardModifiers() & Qt::ControlModifier) {
 151                 // Unselect all other signals because the Ctrl is not
 152                 // pressed
 153                 BOOST_FOREACH(const shared_ptr<Signal> s, sigs)
 154                         if (s != mouse_over_signal)
 155                                 s->select(false);
 156         }
 157
 158         update();
 159 }
 160
 161 void Header::mouseReleaseEvent(QMouseEvent *event)
 162 {
 163         assert(event);
 164         if (event->button() == Qt::LeftButton) {
 165                 _drag_sigs.clear();
 166                 _view.normalize_layout();
 167         }
 168 }
 169
 170 void Header::mouseMoveEvent(QMouseEvent *event)
 171 {
 172         assert(event);
 173         _mouse_point = event->pos();
 174
 175         // Move the signals if we are dragging
 176         if (!_drag_sigs.empty()) {
 177                 const int delta = event->pos().y() - _mouse_down_point.y();
 178
 179                 for (std::list<std::pair<boost::weak_ptr<Signal>,
 180                         int> >::iterator i = _drag_sigs.begin();
 181                         i != _drag_sigs.end(); i++) {
 182                         const boost::shared_ptr<Signal> sig((*i).first);
 183                         if (sig) {
 184                                 const int y = (*i).second + delta;
 185                                 const int y_snap =
 186                                         ((y + View::SignalSnapGridSize / 2) /
 187                                                 View::SignalSnapGridSize) *
 188                                                 View::SignalSnapGridSize;
 189                                 sig->set_v_offset(y_snap);
 190
 191                                 // Ensure the signal is selected
 192                                 sig->select();
 193                         }
 194
 195                 }
 196
 197                 signals_moved();
 198         }
 199
 200         update();
 201 }
 202
 203 void Header::leaveEvent(QEvent *event)
 204 {
 205         _mouse_point = QPoint(-1, -1);
 206         update();
 207 }
 208
 209 void Header::contextMenuEvent(QContextMenuEvent *event)
 210 {
 211         const shared_ptr<Signal> s = get_mouse_over_signal(_mouse_point);
 212
 213         if (!s)
 214                 return;
 215
 216         QMenu menu(this);
 217         menu.addAction(_action_set_name);
 218         menu.addAction(_action_set_colour);
 219
 220         _context_signal = s;
 221         menu.exec(event->globalPos());
 222         _context_signal.reset();
 223 }
 224
 225 void Header::on_action_set_name_triggered()
 226 {
 227         shared_ptr<view::Signal> context_signal = _context_signal;
 228         if (!context_signal)
 229                 return;
 230
 231         const QString new_label = QInputDialog::getText(this, tr("Set Name"),
 232                 tr("Name"), QLineEdit::Normal, context_signal->get_name());
 233
 234         if (!new_label.isEmpty())
 235                 context_signal->set_name(new_label);
 236 }
 237
 238 void Header::on_action_set_colour_triggered()
 239 {
 240         shared_ptr<view::Signal> context_signal = _context_signal;
 241         if (!context_signal)
 242                 return;
 243
 244         const QColor new_colour = QColorDialog::getColor(
 245                 context_signal->get_colour(), this, tr("Set Colour"));
 246
 247         if (new_colour.isValid())
 248                 context_signal->set_colour(new_colour);
 249 }
 250
 251 void Header::on_signals_moved()
 252 {
 253         update();
 254 }
 255
 256
 257 } // namespace view
 258 } // namespace pv