]> sigrok.org Git - pulseview.git/blame - pv/views/trace/decodetrace.cpp
Implement annotation export from the DecodeTrace context menu
[pulseview.git] / pv / views / trace / decodetrace.cpp
CommitLineData
55d3603d
JH
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
efdec55a 17 * along with this program; if not, see <http://www.gnu.org/licenses/>.
55d3603d
JH
18 */
19
20extern "C" {
21#include <libsigrokdecode/libsigrokdecode.h>
22}
23
be843692 24#include <climits>
c3a740dd 25#include <mutex>
5c48ce32 26#include <tuple>
c3a740dd 27
06bb4e6a
JH
28#include <extdef.h>
29
a855d71e 30#include <boost/functional/hash.hpp>
b213ef09 31
c51482b3 32#include <QAction>
d7c0ca4a 33#include <QApplication>
4e5a4405 34#include <QComboBox>
be843692 35#include <QFileDialog>
4e5a4405
JH
36#include <QFormLayout>
37#include <QLabel>
b213ef09 38#include <QMenu>
be843692 39#include <QMessageBox>
ce94e4fd 40#include <QPushButton>
be843692 41#include <QTextStream>
e2f90c50 42#include <QToolTip>
c51482b3 43
2acdb232 44#include "decodetrace.hpp"
1573bf16
SA
45#include "view.hpp"
46#include "viewport.hpp"
2acdb232 47
1cc1c8de 48#include <pv/globalsettings.hpp>
ad908057
SA
49#include <pv/session.hpp>
50#include <pv/strnatcmp.hpp>
51#include <pv/data/decodesignal.hpp>
aca9aa83 52#include <pv/data/decode/annotation.hpp>
2acdb232
JH
53#include <pv/data/decode/decoder.hpp>
54#include <pv/data/logic.hpp>
f3d66e52 55#include <pv/data/logicsegment.hpp>
2acdb232
JH
56#include <pv/widgets/decodergroupbox.hpp>
57#include <pv/widgets/decodermenu.hpp>
119aff65 58
5c48ce32 59using std::abs;
7f8517f6 60using std::make_pair;
819f4c25 61using std::max;
819f4c25 62using std::min;
6f925ba9 63using std::out_of_range;
7f8517f6 64using std::pair;
f9abf97e 65using std::shared_ptr;
53e35b2d 66using std::tie;
819f4c25 67using std::vector;
55d3603d 68
ecd07c20
SA
69using pv::data::decode::Annotation;
70using pv::data::decode::Row;
9f97b357
SA
71using pv::data::DecodeChannel;
72using pv::data::DecodeSignal;
ecd07c20 73
55d3603d 74namespace pv {
f4e57597 75namespace views {
1573bf16 76namespace trace {
55d3603d 77
9ba13f5e
SA
78
79#define DECODETRACE_COLOR_SATURATION (180) /* 0-255 */
80#define DECODETRACE_COLOR_VALUE (170) /* 0-255 */
06bb4e6a 81
641574bc
SA
82const QColor DecodeTrace::ErrorBgColor = QColor(0xEF, 0x29, 0x29);
83const QColor DecodeTrace::NoDecodeColor = QColor(0x88, 0x8A, 0x85);
ad50ac1a 84
88908838 85const int DecodeTrace::ArrowSize = 4;
06e810f2 86const double DecodeTrace::EndCapWidth = 5;
aee9dcf3 87const int DecodeTrace::RowTitleMargin = 10;
06e810f2
JH
88const int DecodeTrace::DrawPadding = 100;
89
5b6ae103
SA
90const int DecodeTrace::MaxTraceUpdateRate = 1; // No more than 1 Hz
91
2b81ae46 92DecodeTrace::DecodeTrace(pv::Session &session,
bb7dd726 93 shared_ptr<data::SignalBase> signalbase, int index) :
bf0edd2b 94 Trace(signalbase),
8dbbc7f0 95 session_(session),
8dbbc7f0 96 row_height_(0),
eee89ff8 97 max_visible_rows_(0),
8dbbc7f0
JH
98 delete_mapper_(this),
99 show_hide_mapper_(this)
55d3603d 100{
ad908057 101 decode_signal_ = dynamic_pointer_cast<data::DecodeSignal>(base_);
e0fc5810 102
752281db
SA
103 // Determine shortest string we want to see displayed in full
104 QFontMetrics m(QApplication::font());
105 min_useful_label_width_ = m.width("XX"); // e.g. two hex characters
106
9ba13f5e
SA
107 // For the base color, we want to start at a very different color for
108 // every decoder stack, so multiply the index with a number that is
109 // rather close to 180 degrees of the color circle but not a dividend of 360
110 // Note: The offset equals the color of the first annotation
111 QColor color;
112 const int h = (120 + 160 * index) % 360;
113 const int s = DECODETRACE_COLOR_SATURATION;
114 const int v = DECODETRACE_COLOR_VALUE;
115 color.setHsv(h, s, v);
116 base_->set_color(color);
9cef9567 117
ad908057
SA
118 connect(decode_signal_.get(), SIGNAL(new_annotations()),
119 this, SLOT(on_new_annotations()));
eee3eab9
SA
120 connect(decode_signal_.get(), SIGNAL(decode_reset()),
121 this, SLOT(on_decode_reset()));
1b56c646
SA
122 connect(decode_signal_.get(), SIGNAL(decode_finished()),
123 this, SLOT(on_decode_finished()));
9f97b357
SA
124 connect(decode_signal_.get(), SIGNAL(channels_updated()),
125 this, SLOT(on_channels_updated()));
126
8dbbc7f0 127 connect(&delete_mapper_, SIGNAL(mapped(int)),
613d097c 128 this, SLOT(on_delete_decoder(int)));
8dbbc7f0 129 connect(&show_hide_mapper_, SIGNAL(mapped(int)),
dd048a7e 130 this, SLOT(on_show_hide_decoder(int)));
5b6ae103
SA
131
132 connect(&delayed_trace_updater_, SIGNAL(timeout()),
133 this, SLOT(on_delayed_trace_update()));
134 delayed_trace_updater_.setSingleShot(true);
135 delayed_trace_updater_.setInterval(1000 / MaxTraceUpdateRate);
55d3603d
JH
136}
137
b9329558 138bool DecodeTrace::enabled() const
55d3603d
JH
139{
140 return true;
141}
142
6f925ba9 143shared_ptr<data::SignalBase> DecodeTrace::base() const
b6b267bb 144{
bb7dd726 145 return base_;
b6b267bb
JH
146}
147
a5d93c27
JH
148pair<int, int> DecodeTrace::v_extents() const
149{
5b5fa4da 150 const int row_height = (ViewItemPaintParams::text_height() * 6) / 4;
796e1360 151
e40a79cb
SA
152 // Make an empty decode trace appear symmetrical
153 const int row_count = max(1, max_visible_rows_);
154
155 return make_pair(-row_height, row_height * row_count);
a5d93c27
JH
156}
157
60938e04 158void DecodeTrace::paint_back(QPainter &p, ViewItemPaintParams &pp)
fe08b6e8 159{
3eb29afd 160 Trace::paint_back(p, pp);
97904bf7 161 paint_axis(p, pp, get_visual_y());
fe08b6e8
JH
162}
163
60938e04 164void DecodeTrace::paint_mid(QPainter &p, ViewItemPaintParams &pp)
55d3603d 165{
0ce3d18c
JH
166 const int text_height = ViewItemPaintParams::text_height();
167 row_height_ = (text_height * 6) / 4;
168 const int annotation_height = (text_height * 5) / 4;
5dfeb70f 169
cd0c558b
SA
170 // Set default pen to allow for text width calculation
171 p.setPen(Qt::black);
172
f9101a91 173 // Iterate through the rows
be9e7b4b 174 int y = get_visual_y();
b82908ab
SA
175 pair<uint64_t, uint64_t> sample_range = get_sample_range(pp.left(), pp.right());
176
177 // Just because the view says we see a certain sample range it
178 // doesn't mean we have this many decoded samples, too, so crop
179 // the range to what has been decoded already
180 sample_range.second = min((int64_t)sample_range.second,
181 decode_signal_->get_decoded_sample_count(current_segment_, false));
5dfeb70f 182
ecd07c20 183 const vector<Row> rows = decode_signal_->visible_rows();
7f8517f6 184
8dbbc7f0 185 visible_rows_.clear();
4fb5fb99 186 for (const Row& row : rows) {
aee9dcf3
SA
187 // Cache the row title widths
188 int row_title_width;
189 try {
190 row_title_width = row_title_widths_.at(row);
30677c13 191 } catch (out_of_range&) {
aee9dcf3
SA
192 const int w = p.boundingRect(QRectF(), 0, row.title()).width() +
193 RowTitleMargin;
194 row_title_widths_[row] = w;
195 row_title_width = w;
196 }
197
f9101a91 198 vector<Annotation> annotations;
ecd07c20 199 decode_signal_->get_annotation_subset(annotations, row,
72435789 200 current_segment_, sample_range.first, sample_range.second);
f9101a91 201 if (!annotations.empty()) {
50631798 202 draw_annotations(annotations, p, annotation_height, pp, y,
f9807084 203 get_row_color(row.index()), row_title_width);
8dbbc7f0 204 y += row_height_;
4fb5fb99 205 visible_rows_.push_back(row);
f9101a91 206 }
7e674e43 207 }
5dfeb70f 208
3eb29afd 209 draw_unresolved_period(p, annotation_height, pp.left(), pp.right());
eee89ff8 210
6d2802aa
SA
211 if ((int)visible_rows_.size() > max_visible_rows_) {
212 max_visible_rows_ = (int)visible_rows_.size();
a303c2d8 213
6d2802aa
SA
214 // Call order is important, otherwise the lazy event handler won't work
215 owner_->extents_changed(false, true);
216 owner_->row_item_appearance_changed(false, true);
217 }
1ae18301
SA
218
219 const QString err = decode_signal_->error_message();
220 if (!err.isEmpty())
221 draw_error(p, err, pp);
55d3603d
JH
222}
223
60938e04 224void DecodeTrace::paint_fore(QPainter &p, ViewItemPaintParams &pp)
88908838 225{
8dbbc7f0 226 assert(row_height_);
88908838 227
2ad82c2e 228 for (size_t i = 0; i < visible_rows_.size(); i++) {
8dbbc7f0 229 const int y = i * row_height_ + get_visual_y();
88908838
JH
230
231 p.setPen(QPen(Qt::NoPen));
232 p.setBrush(QApplication::palette().brush(QPalette::WindowText));
233
2ad82c2e 234 if (i != 0) {
88908838 235 const QPointF points[] = {
3eb29afd
JH
236 QPointF(pp.left(), y - ArrowSize),
237 QPointF(pp.left() + ArrowSize, y),
238 QPointF(pp.left(), y + ArrowSize)
88908838
JH
239 };
240 p.drawPolygon(points, countof(points));
241 }
242
3eb29afd
JH
243 const QRect r(pp.left() + ArrowSize * 2, y - row_height_ / 2,
244 pp.right() - pp.left(), row_height_);
8dbbc7f0 245 const QString h(visible_rows_[i].title());
88908838
JH
246 const int f = Qt::AlignLeft | Qt::AlignVCenter |
247 Qt::TextDontClip;
248
249 // Draw the outline
250 p.setPen(QApplication::palette().color(QPalette::Base));
251 for (int dx = -1; dx <= 1; dx++)
252 for (int dy = -1; dy <= 1; dy++)
253 if (dx != 0 && dy != 0)
254 p.drawText(r.translated(dx, dy), f, h);
255
256 // Draw the text
257 p.setPen(QApplication::palette().color(QPalette::WindowText));
258 p.drawText(r, f, h);
259 }
1931b5f9
SA
260
261 if (show_hover_marker_)
262 paint_hover_marker(p);
88908838
JH
263}
264
b9329558 265void DecodeTrace::populate_popup_form(QWidget *parent, QFormLayout *form)
4e5a4405 266{
613d097c
JH
267 using pv::data::decode::Decoder;
268
4e5a4405 269 assert(form);
4e5a4405 270
7491a29f 271 // Add the standard options
4e5a4405
JH
272 Trace::populate_popup_form(parent, form);
273
7491a29f 274 // Add the decoder options
8dbbc7f0 275 bindings_.clear();
9f97b357
SA
276 channel_id_map_.clear();
277 init_state_map_.clear();
8dbbc7f0 278 decoder_forms_.clear();
4e5a4405 279
47747218 280 const vector< shared_ptr<Decoder> > &stack = decode_signal_->decoder_stack();
4e5a4405 281
2ad82c2e 282 if (stack.empty()) {
5069084a
JH
283 QLabel *const l = new QLabel(
284 tr("<p><i>No decoders in the stack</i></p>"));
285 l->setAlignment(Qt::AlignCenter);
286 form->addRow(l);
2ad82c2e 287 } else {
f46e495e 288 auto iter = stack.cbegin();
5069084a
JH
289 for (int i = 0; i < (int)stack.size(); i++, iter++) {
290 shared_ptr<Decoder> dec(*iter);
291 create_decoder_form(i, dec, parent, form);
292 }
293
294 form->addRow(new QLabel(
8bd26d8b 295 tr("<i>* Required channels</i>"), parent));
5069084a 296 }
4e5a4405 297
ce94e4fd 298 // Add stacking button
ce94e4fd
JH
299 pv::widgets::DecoderMenu *const decoder_menu =
300 new pv::widgets::DecoderMenu(parent);
7491a29f
JH
301 connect(decoder_menu, SIGNAL(decoder_selected(srd_decoder*)),
302 this, SLOT(on_stack_decoder(srd_decoder*)));
303
304 QPushButton *const stack_button =
305 new QPushButton(tr("Stack Decoder"), parent);
ce94e4fd 306 stack_button->setMenu(decoder_menu);
b14a9371 307 stack_button->setToolTip(tr("Stack a higher-level decoder on top of this one"));
ce94e4fd
JH
308
309 QHBoxLayout *stack_button_box = new QHBoxLayout;
310 stack_button_box->addWidget(stack_button, 0, Qt::AlignRight);
311 form->addRow(stack_button_box);
4e5a4405
JH
312}
313
9e773fec 314QMenu* DecodeTrace::create_header_context_menu(QWidget *parent)
c51482b3 315{
9e773fec 316 QMenu *const menu = Trace::create_header_context_menu(parent);
c51482b3
JH
317
318 menu->addSeparator();
319
320 QAction *const del = new QAction(tr("Delete"), this);
a2d21018 321 del->setShortcuts(QKeySequence::Delete);
c51482b3
JH
322 connect(del, SIGNAL(triggered()), this, SLOT(on_delete()));
323 menu->addAction(del);
324
325 return menu;
326}
327
be843692
SA
328QMenu* DecodeTrace::create_view_context_menu(QWidget *parent, QPoint &click_pos)
329{
330 try {
331 selected_row_ = &visible_rows_[get_row_at_point(click_pos)];
332 } catch (out_of_range&) {
333 selected_row_ = nullptr;
334 }
335
336 const pair<uint64_t, uint64_t> sample_range =
337 get_sample_range(click_pos.x(), click_pos.x() + 1);
338 selected_samplepos_ = sample_range.first;
339
340 QMenu *const menu = new QMenu(parent);
341
342 QAction *const export_row =
343 new QAction(tr("Export all annotations for this row"), this);
344 export_row->setIcon(QIcon::fromTheme("document-save-as",
345 QIcon(":/icons/document-save-as.png")));
346 connect(export_row, SIGNAL(triggered()),
347 this, SLOT(on_export_row()));
348 menu->addAction(export_row);
349
350 QAction *const export_row_from_here =
351 new QAction(tr("Export annotations for this row, starting here"), this);
352 export_row_from_here->setIcon(QIcon::fromTheme("document-save-as",
353 QIcon(":/icons/document-save-as.png")));
354 connect(export_row_from_here, SIGNAL(triggered()),
355 this, SLOT(on_export_row_from_here()));
356 menu->addAction(export_row_from_here);
357
358 return menu;
359}
360
50631798
SA
361void DecodeTrace::draw_annotations(vector<pv::data::decode::Annotation> annotations,
362 QPainter &p, int h, const ViewItemPaintParams &pp, int y,
9ba13f5e 363 QColor row_color, int row_title_width)
50631798
SA
364{
365 using namespace pv::data::decode;
366
20d90d22
SA
367 Annotation::Class block_class = 0;
368 bool block_class_uniform = true;
5c48ce32 369 qreal block_start = 0;
20d90d22
SA
370 int block_ann_count = 0;
371
372 const Annotation *prev_ann;
5c48ce32 373 qreal prev_end = INT_MIN;
20d90d22 374
5c48ce32 375 qreal a_end;
50631798
SA
376
377 double samples_per_pixel, pixels_offset;
378 tie(pixels_offset, samples_per_pixel) =
379 get_pixels_offset_samples_per_pixel();
380
bdc2a99b
SA
381 // Sort the annotations by start sample so that decoders
382 // can't confuse us by creating annotations out of order
383 stable_sort(annotations.begin(), annotations.end(),
384 [](const Annotation &a, const Annotation &b) {
385 return a.start_sample() < b.start_sample(); });
386
50631798
SA
387 // Gather all annotations that form a visual "block" and draw them as such
388 for (const Annotation &a : annotations) {
389
5c48ce32
SA
390 const qreal abs_a_start = a.start_sample() / samples_per_pixel;
391 const qreal abs_a_end = a.end_sample() / samples_per_pixel;
bdc2a99b 392
5c48ce32 393 const qreal a_start = abs_a_start - pixels_offset;
20d90d22
SA
394 a_end = abs_a_end - pixels_offset;
395
5c48ce32
SA
396 const qreal a_width = a_end - a_start;
397 const qreal delta = a_end - prev_end;
bdc2a99b
SA
398
399 bool a_is_separate = false;
400
401 // Annotation wider than the threshold for a useful label width?
752281db 402 if (a_width >= min_useful_label_width_) {
bdc2a99b 403 for (const QString &ann_text : a.annotations()) {
5c48ce32 404 const qreal w = p.boundingRect(QRectF(), 0, ann_text).width();
bdc2a99b
SA
405 // Annotation wide enough to fit a label? Don't put it in a block then
406 if (w <= a_width) {
407 a_is_separate = true;
408 break;
409 }
410 }
411 }
50631798 412
bdc2a99b
SA
413 // Were the previous and this annotation more than a pixel apart?
414 if ((abs(delta) > 1) || a_is_separate) {
415 // Block was broken, draw annotations that form the current block
20d90d22
SA
416 if (block_ann_count == 1)
417 draw_annotation(*prev_ann, p, h, pp, y, row_color,
aee9dcf3 418 row_title_width);
20d90d22
SA
419 else if (block_ann_count > 0)
420 draw_annotation_block(block_start, prev_end, block_class,
421 block_class_uniform, p, h, y, row_color);
50631798 422
20d90d22 423 block_ann_count = 0;
50631798
SA
424 }
425
bdc2a99b 426 if (a_is_separate) {
9ba13f5e 427 draw_annotation(a, p, h, pp, y, row_color, row_title_width);
bdc2a99b 428 // Next annotation must start a new block. delta will be > 1
20d90d22
SA
429 // because we set prev_end to INT_MIN but that's okay since
430 // block_ann_count will be 0 and nothing will be drawn
431 prev_end = INT_MIN;
5c48ce32 432 block_ann_count = 0;
bdc2a99b 433 } else {
20d90d22
SA
434 prev_end = a_end;
435 prev_ann = &a;
436
437 if (block_ann_count == 0) {
438 block_start = a_start;
439 block_class = a.ann_class();
440 block_class_uniform = true;
441 } else
442 if (a.ann_class() != block_class)
443 block_class_uniform = false;
444
445 block_ann_count++;
bdc2a99b 446 }
50631798
SA
447 }
448
20d90d22
SA
449 if (block_ann_count == 1)
450 draw_annotation(*prev_ann, p, h, pp, y, row_color, row_title_width);
451 else if (block_ann_count > 0)
452 draw_annotation_block(block_start, prev_end, block_class,
453 block_class_uniform, p, h, y, row_color);
50631798
SA
454}
455
287d607f 456void DecodeTrace::draw_annotation(const pv::data::decode::Annotation &a,
5b5fa4da 457 QPainter &p, int h, const ViewItemPaintParams &pp, int y,
9ba13f5e 458 QColor row_color, int row_title_width) const
06e810f2 459{
53e35b2d
JH
460 double samples_per_pixel, pixels_offset;
461 tie(pixels_offset, samples_per_pixel) =
462 get_pixels_offset_samples_per_pixel();
7f8517f6 463
06e810f2
JH
464 const double start = a.start_sample() / samples_per_pixel -
465 pixels_offset;
c063290a 466 const double end = a.end_sample() / samples_per_pixel - pixels_offset;
287d607f 467
f228f00e 468 QColor color = get_annotation_color(row_color, a.ann_class());
9ba13f5e
SA
469 p.setPen(color.darker());
470 p.setBrush(color);
06e810f2 471
3eb29afd 472 if (start > pp.right() + DrawPadding || end < pp.left() - DrawPadding)
06e810f2
JH
473 return;
474
475 if (a.start_sample() == a.end_sample())
f765c3db 476 draw_instant(a, p, h, start, y);
06e810f2 477 else
c063290a 478 draw_range(a, p, h, start, end, y, pp, row_title_width);
06e810f2
JH
479}
480
5c48ce32 481void DecodeTrace::draw_annotation_block(qreal start, qreal end,
20d90d22 482 Annotation::Class ann_class, bool use_ann_format, QPainter &p, int h,
9ba13f5e 483 int y, QColor row_color) const
50631798 484{
33707990
SA
485 const double top = y + .5 - h / 2;
486 const double bottom = y + .5 + h / 2;
33707990 487
3082ee93
JH
488 const QRectF rect(start, top, end - start, bottom - top);
489 const int r = h / 4;
490
491 p.setPen(QPen(Qt::NoPen));
492 p.setBrush(Qt::white);
493 p.drawRoundedRect(rect, r, r);
494
20d90d22
SA
495 // If all annotations in this block are of the same type, we can use the
496 // one format that all of these annotations have. Otherwise, we should use
497 // a neutral color (i.e. gray)
498 if (use_ann_format) {
499 const QColor color = get_annotation_color(row_color, ann_class);
500 p.setPen(color.darker());
501 p.setBrush(QBrush(color, Qt::Dense4Pattern));
502 } else {
503 p.setPen(Qt::gray);
504 p.setBrush(QBrush(Qt::gray, Qt::Dense4Pattern));
505 }
506
3082ee93 507 p.drawRoundedRect(rect, r, r);
50631798
SA
508}
509
06e810f2 510void DecodeTrace::draw_instant(const pv::data::decode::Annotation &a, QPainter &p,
5c48ce32 511 int h, qreal x, int y) const
06e810f2
JH
512{
513 const QString text = a.annotations().empty() ?
514 QString() : a.annotations().back();
5c48ce32 515 const qreal w = min((qreal)p.boundingRect(QRectF(), 0, text).width(),
06e810f2
JH
516 0.0) + h;
517 const QRectF rect(x - w / 2, y - h / 2, w, h);
518
06e810f2
JH
519 p.drawRoundedRect(rect, h / 2, h / 2);
520
2a56e448 521 p.setPen(Qt::black);
06e810f2
JH
522 p.drawText(rect, Qt::AlignCenter | Qt::AlignVCenter, text);
523}
524
525void DecodeTrace::draw_range(const pv::data::decode::Annotation &a, QPainter &p,
5c48ce32 526 int h, qreal start, qreal end, int y, const ViewItemPaintParams &pp,
f765c3db 527 int row_title_width) const
06e810f2 528{
5c48ce32
SA
529 const qreal top = y + .5 - h / 2;
530 const qreal bottom = y + .5 + h / 2;
06e810f2
JH
531 const vector<QString> annotations = a.annotations();
532
06e810f2 533 // If the two ends are within 1 pixel, draw a vertical line
2ad82c2e 534 if (start + 1.0 > end) {
06e810f2
JH
535 p.drawLine(QPointF(start, top), QPointF(start, bottom));
536 return;
537 }
538
5c48ce32 539 const qreal cap_width = min((end - start) / 4, EndCapWidth);
06e810f2
JH
540
541 QPointF pts[] = {
542 QPointF(start, y + .5f),
543 QPointF(start + cap_width, top),
544 QPointF(end - cap_width, top),
545 QPointF(end, y + .5f),
546 QPointF(end - cap_width, bottom),
547 QPointF(start + cap_width, bottom)
548 };
549
550 p.drawConvexPolygon(pts, countof(pts));
551
552 if (annotations.empty())
553 return;
554
7352be72
SA
555 const int ann_start = start + cap_width;
556 const int ann_end = end - cap_width;
557
6f925ba9
UH
558 const int real_start = max(ann_start, pp.left() + row_title_width);
559 const int real_end = min(ann_end, pp.right());
7352be72
SA
560 const int real_width = real_end - real_start;
561
562 QRectF rect(real_start, y - h / 2, real_width, h);
0f290e9b
JH
563 if (rect.width() <= 4)
564 return;
565
2a56e448 566 p.setPen(Qt::black);
06e810f2
JH
567
568 // Try to find an annotation that will fit
569 QString best_annotation;
570 int best_width = 0;
571
d9aecf1f 572 for (const QString &a : annotations) {
06e810f2
JH
573 const int w = p.boundingRect(QRectF(), 0, a).width();
574 if (w <= rect.width() && w > best_width)
575 best_annotation = a, best_width = w;
576 }
577
578 if (best_annotation.isEmpty())
579 best_annotation = annotations.back();
580
581 // If not ellide the last in the list
582 p.drawText(rect, Qt::AlignCenter, p.fontMetrics().elidedText(
583 best_annotation, Qt::ElideRight, rect.width()));
584}
585
b9329558 586void DecodeTrace::draw_error(QPainter &p, const QString &message,
5b5fa4da 587 const ViewItemPaintParams &pp)
ad50ac1a 588{
be9e7b4b 589 const int y = get_visual_y();
ad50ac1a 590
1ae18301
SA
591 double samples_per_pixel, pixels_offset;
592 tie(pixels_offset, samples_per_pixel) = get_pixels_offset_samples_per_pixel();
593
641574bc
SA
594 p.setPen(ErrorBgColor.darker());
595 p.setBrush(ErrorBgColor);
ad50ac1a 596
1ae18301
SA
597 const QRectF bounding_rect = QRectF(pp.left(), INT_MIN / 2 + y, pp.right(), INT_MAX);
598
599 const QRectF text_rect = p.boundingRect(bounding_rect, Qt::AlignCenter, message);
5c48ce32 600 const qreal r = text_rect.height() / 4;
ad50ac1a 601
1ae18301 602 p.drawRoundedRect(text_rect.adjusted(-r, -r, r, r), r, r, Qt::AbsoluteSize);
ad50ac1a 603
2a56e448 604 p.setPen(Qt::black);
ad50ac1a
JH
605 p.drawText(text_rect, message);
606}
607
ff83d980 608void DecodeTrace::draw_unresolved_period(QPainter &p, int h, int left, int right) const
5dfeb70f
JH
609{
610 using namespace pv::data;
611 using pv::data::decode::Decoder;
612
53e35b2d
JH
613 double samples_per_pixel, pixels_offset;
614
5ecf957f 615 const int64_t sample_count = decode_signal_->get_working_sample_count(current_segment_);
5dfeb70f
JH
616 if (sample_count == 0)
617 return;
618
b82908ab 619 const int64_t samples_decoded = decode_signal_->get_decoded_sample_count(current_segment_, true);
5dfeb70f
JH
620 if (sample_count == samples_decoded)
621 return;
622
be9e7b4b 623 const int y = get_visual_y();
7f8517f6 624
ff83d980 625 tie(pixels_offset, samples_per_pixel) = get_pixels_offset_samples_per_pixel();
7f8517f6 626
5dfeb70f
JH
627 const double start = max(samples_decoded /
628 samples_per_pixel - pixels_offset, left - 1.0);
629 const double end = min(sample_count / samples_per_pixel -
630 pixels_offset, right + 1.0);
e06cf18d 631 const QRectF no_decode_rect(start, y - (h / 2) - 0.5, end - start, h);
5dfeb70f
JH
632
633 p.setPen(QPen(Qt::NoPen));
634 p.setBrush(Qt::white);
635 p.drawRect(no_decode_rect);
636
641574bc
SA
637 p.setPen(NoDecodeColor);
638 p.setBrush(QBrush(NoDecodeColor, Qt::Dense6Pattern));
5dfeb70f
JH
639 p.drawRect(no_decode_rect);
640}
641
53e35b2d 642pair<double, double> DecodeTrace::get_pixels_offset_samples_per_pixel() const
7f8517f6 643{
8dbbc7f0 644 assert(owner_);
7f8517f6 645
8dbbc7f0 646 const View *view = owner_->view();
eae6e30a
JH
647 assert(view);
648
649 const double scale = view->scale();
7f8517f6
SA
650 assert(scale > 0);
651
53e35b2d 652 const double pixels_offset =
ff83d980 653 ((view->offset() - decode_signal_->start_time()) / scale).convert_to<double>();
7f8517f6 654
ff83d980 655 double samplerate = decode_signal_->samplerate();
7f8517f6
SA
656
657 // Show sample rate as 1Hz when it is unknown
658 if (samplerate == 0.0)
659 samplerate = 1.0;
660
53e35b2d 661 return make_pair(pixels_offset, samplerate * scale);
7f8517f6
SA
662}
663
db1bf6bf
JH
664pair<uint64_t, uint64_t> DecodeTrace::get_sample_range(
665 int x_start, int x_end) const
7f8517f6 666{
53e35b2d
JH
667 double samples_per_pixel, pixels_offset;
668 tie(pixels_offset, samples_per_pixel) =
669 get_pixels_offset_samples_per_pixel();
7f8517f6 670
db1bf6bf
JH
671 const uint64_t start = (uint64_t)max(
672 (x_start + pixels_offset) * samples_per_pixel, 0.0);
673 const uint64_t end = (uint64_t)max(
674 (x_end + pixels_offset) * samples_per_pixel, 0.0);
7f8517f6
SA
675
676 return make_pair(start, end);
677}
678
9ba13f5e
SA
679QColor DecodeTrace::get_row_color(int row_index) const
680{
681 // For each row color, use the base color hue and add an offset that's
682 // not a dividend of 360
683
684 QColor color;
685 const int h = (base_->color().toHsv().hue() + 20 * row_index) % 360;
686 const int s = DECODETRACE_COLOR_SATURATION;
687 const int v = DECODETRACE_COLOR_VALUE;
688 color.setHsl(h, s, v);
689
690 return color;
691}
692
693QColor DecodeTrace::get_annotation_color(QColor row_color, int annotation_index) const
694{
695 // For each row color, use the base color hue and add an offset that's
696 // not a dividend of 360 and not a multiple of the row offset
697
698 QColor color(row_color);
699 const int h = (color.toHsv().hue() + 55 * annotation_index) % 360;
700 const int s = DECODETRACE_COLOR_SATURATION;
701 const int v = DECODETRACE_COLOR_VALUE;
702 color.setHsl(h, s, v);
703
704 return color;
705}
706
117cdea3 707int DecodeTrace::get_row_at_point(const QPoint &point)
e2f90c50 708{
8dbbc7f0 709 if (!row_height_)
117cdea3 710 return -1;
e2f90c50 711
99029fda
SA
712 const int y = (point.y() - get_visual_y() + row_height_ / 2);
713
714 /* Integer divison of (x-1)/x would yield 0, so we check for this. */
715 if (y < 0)
716 return -1;
717
718 const int row = y / row_height_;
719
720 if (row >= (int)visible_rows_.size())
117cdea3 721 return -1;
e2f90c50 722
117cdea3 723 return row;
e2f90c50
SA
724}
725
117cdea3 726const QString DecodeTrace::get_annotation_at_point(const QPoint &point)
e2f90c50
SA
727{
728 using namespace pv::data::decode;
729
117cdea3
JH
730 if (!enabled())
731 return QString();
e2f90c50 732
117cdea3
JH
733 const pair<uint64_t, uint64_t> sample_range =
734 get_sample_range(point.x(), point.x() + 1);
735 const int row = get_row_at_point(point);
736 if (row < 0)
737 return QString();
e2f90c50 738
be843692 739 vector<Annotation> annotations;
e2f90c50 740
ff83d980 741 decode_signal_->get_annotation_subset(annotations, visible_rows_[row],
72435789 742 current_segment_, sample_range.first, sample_range.second);
e2f90c50
SA
743
744 return (annotations.empty()) ?
745 QString() : annotations[0].annotations().front();
746}
747
873e8035 748void DecodeTrace::hover_point_changed(const QPoint &hp)
117cdea3 749{
0cbadf1c
SA
750 Trace::hover_point_changed(hp);
751
8dbbc7f0 752 assert(owner_);
eae6e30a 753
8dbbc7f0 754 const View *const view = owner_->view();
eae6e30a
JH
755 assert(view);
756
53b652bd
SA
757 if (hp.x() == 0) {
758 QToolTip::hideText();
759 return;
760 }
761
117cdea3 762 QString ann = get_annotation_at_point(hp);
e2f90c50 763
eae6e30a 764 assert(view);
e2f90c50 765
8b9df0ad 766 if (!row_height_ || ann.isEmpty()) {
ebdfa094 767 QToolTip::hideText();
117cdea3
JH
768 return;
769 }
6e6881e2 770
117cdea3 771 const int hover_row = get_row_at_point(hp);
6e6881e2 772
117cdea3
JH
773 QFontMetrics m(QToolTip::font());
774 const QRect text_size = m.boundingRect(QRect(), 0, ann);
e2f90c50 775
117cdea3
JH
776 // This is OS-specific and unfortunately we can't query it, so
777 // use an approximation to at least try to minimize the error.
778 const int padding = 8;
6e6881e2 779
117cdea3
JH
780 // Make sure the tool tip doesn't overlap with the mouse cursor.
781 // If it did, the tool tip would constantly hide and re-appear.
782 // We also push it up by one row so that it appears above the
783 // decode trace, not below.
873e8035
SA
784 QPoint p = hp;
785 p.setX(hp.x() - (text_size.width() / 2) - padding);
6e6881e2 786
873e8035 787 p.setY(get_visual_y() - (row_height_ / 2) +
8dbbc7f0 788 (hover_row * row_height_) -
99029fda 789 row_height_ - text_size.height() - padding);
e2f90c50 790
873e8035 791 QToolTip::showText(view->viewport()->mapToGlobal(p), ann);
9555ca8b
SA
792}
793
613d097c
JH
794void DecodeTrace::create_decoder_form(int index,
795 shared_ptr<data::decode::Decoder> &dec, QWidget *parent,
796 QFormLayout *form)
7491a29f 797{
1cc1c8de 798 GlobalSettings settings;
7491a29f
JH
799
800 assert(dec);
801 const srd_decoder *const decoder = dec->decoder();
802 assert(decoder);
803
ff59fa2c
SA
804 const bool decoder_deletable = index > 0;
805
204bae45 806 pv::widgets::DecoderGroupBox *const group =
27e8df22 807 new pv::widgets::DecoderGroupBox(
580b4f25
UH
808 QString::fromUtf8(decoder->name),
809 tr("%1:\n%2").arg(QString::fromUtf8(decoder->longname),
810 QString::fromUtf8(decoder->desc)),
811 nullptr, decoder_deletable);
dd048a7e 812 group->set_decoder_visible(dec->shown());
613d097c 813
ff59fa2c
SA
814 if (decoder_deletable) {
815 delete_mapper_.setMapping(group, index);
816 connect(group, SIGNAL(delete_decoder()), &delete_mapper_, SLOT(map()));
817 }
613d097c 818
8dbbc7f0 819 show_hide_mapper_.setMapping(group, index);
dd048a7e 820 connect(group, SIGNAL(show_hide_decoder()),
8dbbc7f0 821 &show_hide_mapper_, SLOT(map()));
dd048a7e 822
204bae45
JH
823 QFormLayout *const decoder_form = new QFormLayout;
824 group->add_layout(decoder_form);
7491a29f 825
47747218 826 const vector<DecodeChannel> channels = decode_signal_->get_channels();
407c9ebe 827
9f97b357
SA
828 // Add the channels
829 for (DecodeChannel ch : channels) {
830 // Ignore channels not part of the decoder we create the form for
831 if (ch.decoder_ != dec)
832 continue;
407c9ebe 833
9f97b357
SA
834 QComboBox *const combo = create_channel_selector(parent, &ch);
835 QComboBox *const combo_init_state = create_channel_selector_init_state(parent, &ch);
407c9ebe 836
9f97b357
SA
837 channel_id_map_[combo] = ch.id;
838 init_state_map_[combo_init_state] = ch.id;
407c9ebe 839
7491a29f 840 connect(combo, SIGNAL(currentIndexChanged(int)),
6ac6242b 841 this, SLOT(on_channel_selected(int)));
9f97b357
SA
842 connect(combo_init_state, SIGNAL(currentIndexChanged(int)),
843 this, SLOT(on_init_state_changed(int)));
407c9ebe
UH
844
845 QHBoxLayout *const hlayout = new QHBoxLayout;
846 hlayout->addWidget(combo);
9f97b357 847 hlayout->addWidget(combo_init_state);
407c9ebe 848
1cc1c8de 849 if (!settings.value(GlobalSettings::Key_Dec_InitialStateConfigurable).toBool())
9f97b357 850 combo_init_state->hide();
7491a29f 851
9f97b357
SA
852 const QString required_flag = ch.is_optional ? QString() : QString("*");
853 decoder_form->addRow(tr("<b>%1</b> (%2) %3")
854 .arg(ch.name, ch.desc, required_flag), hlayout);
7491a29f
JH
855 }
856
857 // Add the options
3cc9ad7b 858 shared_ptr<binding::Decoder> binding(
946b52e1 859 new binding::Decoder(decode_signal_, dec));
204bae45 860 binding->add_properties_to_form(decoder_form, true);
7491a29f 861
8dbbc7f0 862 bindings_.push_back(binding);
204bae45
JH
863
864 form->addRow(group);
8dbbc7f0 865 decoder_forms_.push_back(group);
7491a29f
JH
866}
867
9f97b357 868QComboBox* DecodeTrace::create_channel_selector(QWidget *parent, const DecodeChannel *ch)
4e5a4405 869{
47e9e7bb 870 const auto sigs(session_.signalbases());
78b0af3e 871
9f97b357 872 // Sort signals in natural order
47e9e7bb 873 vector< shared_ptr<data::SignalBase> > sig_list(sigs.begin(), sigs.end());
6f925ba9 874 sort(sig_list.begin(), sig_list.end(),
47e9e7bb
SA
875 [](const shared_ptr<data::SignalBase> &a,
876 const shared_ptr<data::SignalBase> &b) {
877 return strnatcasecmp(a->name().toStdString(),
878 b->name().toStdString()) < 0; });
4e5a4405 879
4e5a4405
JH
880 QComboBox *selector = new QComboBox(parent);
881
4c60462b 882 selector->addItem("-", qVariantFromValue((void*)nullptr));
4e5a4405 883
9f97b357 884 if (!ch->assigned_signal)
4e5a4405
JH
885 selector->setCurrentIndex(0);
886
47e9e7bb
SA
887 for (const shared_ptr<data::SignalBase> &b : sig_list) {
888 assert(b);
79c4a9c8 889 if (b->logic_data() && b->enabled()) {
47e9e7bb
SA
890 selector->addItem(b->name(),
891 qVariantFromValue((void*)b.get()));
5da5d081 892
9f97b357
SA
893 if (ch->assigned_signal == b.get())
894 selector->setCurrentIndex(selector->count() - 1);
4e5a4405
JH
895 }
896 }
897
898 return selector;
899}
900
9f97b357
SA
901QComboBox* DecodeTrace::create_channel_selector_init_state(QWidget *parent,
902 const DecodeChannel *ch)
407c9ebe
UH
903{
904 QComboBox *selector = new QComboBox(parent);
905
906 selector->addItem("0", qVariantFromValue((int)SRD_INITIAL_PIN_LOW));
907 selector->addItem("1", qVariantFromValue((int)SRD_INITIAL_PIN_HIGH));
7df44935 908 selector->addItem("X", qVariantFromValue((int)SRD_INITIAL_PIN_SAME_AS_SAMPLE0));
407c9ebe 909
9f97b357 910 selector->setCurrentIndex(ch->initial_pin_state);
407c9ebe
UH
911
912 selector->setToolTip("Initial (assumed) pin value before the first sample");
913
914 return selector;
915}
916
ad908057 917void DecodeTrace::on_new_annotations()
5b6ae103
SA
918{
919 if (!delayed_trace_updater_.isActive())
920 delayed_trace_updater_.start();
921}
922
923void DecodeTrace::on_delayed_trace_update()
9cef9567 924{
8dbbc7f0 925 if (owner_)
6e2c3c85 926 owner_->row_item_appearance_changed(false, true);
9cef9567
JH
927}
928
eee3eab9
SA
929void DecodeTrace::on_decode_reset()
930{
931 visible_rows_.clear();
932 max_visible_rows_ = 0;
933
934 if (owner_)
935 owner_->row_item_appearance_changed(false, true);
936}
937
1b56c646
SA
938void DecodeTrace::on_decode_finished()
939{
940 if (owner_)
941 owner_->row_item_appearance_changed(false, true);
942}
943
b9329558 944void DecodeTrace::delete_pressed()
5ed1adf5
JH
945{
946 on_delete();
947}
948
b9329558 949void DecodeTrace::on_delete()
c51482b3 950{
ad908057 951 session_.remove_decode_signal(decode_signal_);
c51482b3
JH
952}
953
6ac6242b 954void DecodeTrace::on_channel_selected(int)
4e5a4405 955{
9f97b357
SA
956 QComboBox *cb = qobject_cast<QComboBox*>(QObject::sender());
957
958 // Determine signal that was selected
959 const data::SignalBase *signal =
960 (data::SignalBase*)cb->itemData(cb->currentIndex()).value<void*>();
961
962 // Determine decode channel ID this combo box is the channel selector for
963 const uint16_t id = channel_id_map_.at(cb);
964
965 decode_signal_->assign_signal(id, signal);
966}
967
968void DecodeTrace::on_channels_updated()
969{
970 if (owner_)
971 owner_->row_item_appearance_changed(false, true);
4e5a4405
JH
972}
973
9f97b357 974void DecodeTrace::on_init_state_changed(int)
407c9ebe 975{
9f97b357
SA
976 QComboBox *cb = qobject_cast<QComboBox*>(QObject::sender());
977
978 // Determine inital pin state that was selected
979 int init_state = cb->itemData(cb->currentIndex()).value<int>();
980
981 // Determine decode channel ID this combo box is the channel selector for
982 const uint16_t id = init_state_map_.at(cb);
983
984 decode_signal_->set_initial_pin_state(id, init_state);
407c9ebe
UH
985}
986
7491a29f
JH
987void DecodeTrace::on_stack_decoder(srd_decoder *decoder)
988{
ad908057 989 decode_signal_->stack_decoder(decoder);
37fd11b1
JH
990
991 create_popup_form();
7491a29f
JH
992}
993
613d097c
JH
994void DecodeTrace::on_delete_decoder(int index)
995{
ad908057 996 decode_signal_->remove_decoder(index);
613d097c 997
ded43869
SA
998 // Force re-calculation of the trace height, see paint_mid()
999 max_visible_rows_ = 0;
1000 owner_->extents_changed(false, true);
1001
613d097c 1002 // Update the popup
c063290a 1003 create_popup_form();
613d097c
JH
1004}
1005
dd048a7e
JH
1006void DecodeTrace::on_show_hide_decoder(int index)
1007{
ad908057 1008 const bool state = decode_signal_->toggle_decoder_visibility(index);
dd048a7e 1009
8dbbc7f0 1010 assert(index < (int)decoder_forms_.size());
ad908057 1011 decoder_forms_[index]->set_decoder_visible(state);
dd048a7e 1012
ded43869
SA
1013 if (!state) {
1014 // Force re-calculation of the trace height, see paint_mid()
1015 max_visible_rows_ = 0;
1016 owner_->extents_changed(false, true);
1017 }
1018
8dbbc7f0 1019 if (owner_)
6e2c3c85 1020 owner_->row_item_appearance_changed(false, true);
dd048a7e
JH
1021}
1022
be843692
SA
1023void DecodeTrace::on_export_row()
1024{
1025 selected_samplepos_ = 0;
1026 on_export_row_from_here();
1027}
1028
1029void DecodeTrace::on_export_row_from_here()
1030{
1031 using namespace pv::data::decode;
1032
1033 if (!selected_row_)
1034 return;
1035
1036 vector<Annotation> annotations;
1037
1038 decode_signal_->get_annotation_subset(annotations, *selected_row_,
1039 current_segment_, selected_samplepos_, ULLONG_MAX);
1040
1041 if (annotations.empty())
1042 return;
1043
1044 QSettings settings;
1045 const QString dir = settings.value(SettingSaveDirectory).toString();
1046
1047 const QString file_name = QFileDialog::getSaveFileName(
1048 owner_->view(), tr("Export annotations"), dir, tr("Text Files (*.txt);;All Files (*)"));
1049
1050 if (file_name.isEmpty())
1051 return;
1052
1053 QFile file(file_name);
1054 if (file.open(QIODevice::WriteOnly | QIODevice::Truncate | QIODevice::Text)) {
1055 QTextStream out_stream(&file);
1056
1057 for (Annotation &ann : annotations)
1058 out_stream << ann.annotations().front() << '\n';
1059
1060 if (out_stream.status() == QTextStream::Ok)
1061 return;
1062 }
1063
1064 QMessageBox msg(owner_->view());
1065 msg.setText(tr("Error"));
1066 msg.setInformativeText(tr("File %1 could not be written to.").arg(file_name));
1067 msg.setStandardButtons(QMessageBox::Ok);
1068 msg.setIcon(QMessageBox::Warning);
1069 msg.exec();
1070}
1071
1573bf16 1072} // namespace trace
f4e57597 1073} // namespace views
55d3603d 1074} // namespace pv