]> sigrok.org Git - pulseview.git/blob - pv/view/decode/annotation.cpp
0005847655fcfbc77d1451c2cc0f7d232f54e7c4
[pulseview.git] / pv / view / decode / annotation.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 extern "C" {
22 #include <libsigrokdecode/libsigrokdecode.h>
23 }
24
25 #include <extdef.h>
26
27 #include <algorithm>
28
29 #include <boost/foreach.hpp>
30
31 #include <QPainter>
32
33 #include "annotation.h"
34
35 using namespace std;
36
37 namespace pv {
38 namespace view {
39 namespace decode {
40
41 const double Annotation::EndCapWidth = 5;
42 const int Annotation::DrawPadding = 100;
43
44 const QColor Annotation::Colours[7] = {
45         QColor(0xFC, 0xE9, 0x4F),       // Light Butter
46         QColor(0xFC, 0xAF, 0x3E),       // Light Orange
47         QColor(0xE9, 0xB9, 0x6E),       // Light Chocolate
48         QColor(0x8A, 0xE2, 0x34),       // Light Green
49         QColor(0x72, 0x9F, 0xCF),       // Light Blue
50         QColor(0xAD, 0x7F, 0xA8),       // Light Plum
51         QColor(0xEF, 0x29, 0x29)        // Light Red
52 };
53
54 Annotation::Annotation(const srd_proto_data *const pdata) :
55         _start_sample(pdata->start_sample),
56         _end_sample(pdata->end_sample)
57 {
58         assert(pdata);
59         const srd_proto_data_annotation *const pda =
60                 (const srd_proto_data_annotation*)pdata->data;
61         assert(pda);
62
63         const char *const *annotations = (char**)pda->ann_text;
64         while(*annotations) {
65                 _annotations.push_back(QString(*annotations));
66                 annotations++;
67         }
68 }
69
70 uint64_t Annotation::start_sample() const
71 {
72         return _start_sample;
73 }
74
75 uint64_t Annotation::end_sample() const
76 {
77         return _end_sample;
78 }
79
80 void Annotation::paint(QPainter &p, QColor text_color, int h,
81         int left, int right, double samples_per_pixel, double pixels_offset,
82         int y)
83 {
84         const double start = _start_sample / samples_per_pixel -
85                 pixels_offset;
86         const double end = _end_sample / samples_per_pixel -
87                 pixels_offset;
88         const QColor fill = Colours[(_format * (countof(Colours) / 2 + 1)) %
89                 countof(Colours)];
90         const QColor outline(fill.darker());
91
92         if (start > right + DrawPadding || end < left - DrawPadding)
93                 return;
94
95         if (_start_sample == _end_sample)
96                 draw_instant(p, fill, outline, text_color, h,
97                         start, y);
98         else
99                 draw_range(p, fill, outline, text_color, h,
100                         start, end, y);
101 }
102
103 void Annotation::draw_instant(QPainter &p, QColor fill, QColor outline,
104         QColor text_color, int h, double x, int y)
105 {
106         const QString text = _annotations.empty() ?
107                 QString() : _annotations.back();
108         const double w = min(p.boundingRect(QRectF(), 0, text).width(),
109                 0.0) + h;
110         const QRectF rect(x - w / 2, y - h / 2, w, h);
111
112         p.setPen(outline);
113         p.setBrush(fill);
114         p.drawRoundedRect(rect, h / 2, h / 2);
115
116         p.setPen(text_color);
117         p.drawText(rect, Qt::AlignCenter | Qt::AlignVCenter, text);
118 }
119
120 void Annotation::draw_range(QPainter &p, QColor fill, QColor outline,
121         QColor text_color, int h, double start, double end, int y)
122 {
123         const double top = y + .5 - h / 2;
124         const double bottom = y + .5 + h / 2;
125
126         p.setPen(outline);
127         p.setBrush(fill);
128
129         // If the two ends are within 1 pixel, draw a vertical line
130         if (start + 1.0 > end)
131         {
132                 p.drawLine(QPointF(start, top), QPointF(start, bottom));
133                 return;
134         }
135
136         const double cap_width = min((end - start) / 2, EndCapWidth);
137
138         QPointF pts[] = {
139                 QPointF(start, y + .5f),
140                 QPointF(start + cap_width, top),
141                 QPointF(end - cap_width, top),
142                 QPointF(end, y + .5f),
143                 QPointF(end - cap_width, bottom),
144                 QPointF(start + cap_width, bottom)
145         };
146
147         p.drawConvexPolygon(pts, countof(pts));
148
149         if (_annotations.empty())
150                 return;
151
152         QRectF rect(start + cap_width, y - h / 2,
153                 end - start - cap_width * 2, h);
154         p.setPen(text_color);
155
156         // Try to find an annotation that will fit
157         QString best_annotation;
158         int best_width = 0;
159
160         BOOST_FOREACH(QString &a, _annotations) {
161                 const int w = p.boundingRect(QRectF(), 0, a).width();
162                 if (w <= rect.width() && w > best_width)
163                         best_annotation = a, best_width = w;
164         }
165
166         if (best_annotation.isEmpty())
167                 best_annotation = _annotations.back();
168
169         // If not ellide the last in the list
170         p.drawText(rect, Qt::AlignCenter, p.fontMetrics().elidedText(
171                 best_annotation, Qt::ElideRight, rect.width()));
172 }
173
174 } // namespace decode
175 } // namespace view
176 } // namespace pv