]>
Commit | Line | Data |
---|---|---|
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 <extdef.h> | |
22 | ||
23 | #include <math.h> | |
24 | ||
25 | #include "analogsignal.h" | |
26 | #include "pv/data/analog.h" | |
27 | #include "pv/data/analogsnapshot.h" | |
28 | #include "pv/view/view.h" | |
29 | ||
30 | using boost::shared_ptr; | |
31 | using std::max; | |
32 | using std::min; | |
33 | using std::deque; | |
34 | ||
35 | namespace pv { | |
36 | namespace view { | |
37 | ||
38 | const QColor AnalogSignal::SignalColours[4] = { | |
39 | QColor(0xC4, 0xA0, 0x00), // Yellow | |
40 | QColor(0x87, 0x20, 0x7A), // Magenta | |
41 | QColor(0x20, 0x4A, 0x87), // Blue | |
42 | QColor(0x4E, 0x9A, 0x06) // Green | |
43 | }; | |
44 | ||
45 | const float AnalogSignal::EnvelopeThreshold = 256.0f; | |
46 | ||
47 | AnalogSignal::AnalogSignal(pv::SigSession &session, sr_probe *const probe, | |
48 | shared_ptr<data::Analog> data) : | |
49 | Signal(session, probe), | |
50 | _data(data), | |
51 | _scale(1.0f) | |
52 | { | |
53 | _colour = SignalColours[probe->index % countof(SignalColours)]; | |
54 | } | |
55 | ||
56 | AnalogSignal::~AnalogSignal() | |
57 | { | |
58 | } | |
59 | ||
60 | boost::shared_ptr<pv::data::SignalData> AnalogSignal::data() const | |
61 | { | |
62 | return _data; | |
63 | } | |
64 | ||
65 | void AnalogSignal::set_scale(float scale) | |
66 | { | |
67 | _scale = scale; | |
68 | } | |
69 | ||
70 | void AnalogSignal::paint_back(QPainter &p, int left, int right) | |
71 | { | |
72 | if (_probe->enabled) | |
73 | paint_axis(p, get_y(), left, right); | |
74 | } | |
75 | ||
76 | void AnalogSignal::paint_mid(QPainter &p, int left, int right) | |
77 | { | |
78 | assert(_data); | |
79 | assert(right >= left); | |
80 | ||
81 | assert(_view); | |
82 | const int y = _v_offset - _view->v_offset(); | |
83 | ||
84 | const double scale = _view->scale(); | |
85 | assert(scale > 0); | |
86 | ||
87 | const double offset = _view->offset(); | |
88 | ||
89 | if (!_probe->enabled) | |
90 | return; | |
91 | ||
92 | const deque< shared_ptr<pv::data::AnalogSnapshot> > &snapshots = | |
93 | _data->get_snapshots(); | |
94 | if (snapshots.empty()) | |
95 | return; | |
96 | ||
97 | const shared_ptr<pv::data::AnalogSnapshot> &snapshot = | |
98 | snapshots.front(); | |
99 | ||
100 | const double pixels_offset = offset / scale; | |
101 | const double samplerate = _data->samplerate(); | |
102 | const double start_time = _data->get_start_time(); | |
103 | const int64_t last_sample = snapshot->get_sample_count() - 1; | |
104 | const double samples_per_pixel = samplerate * scale; | |
105 | const double start = samplerate * (offset - start_time); | |
106 | const double end = start + samples_per_pixel * (right - left); | |
107 | ||
108 | const int64_t start_sample = min(max((int64_t)floor(start), | |
109 | (int64_t)0), last_sample); | |
110 | const int64_t end_sample = min(max((int64_t)ceil(end) + 1, | |
111 | (int64_t)0), last_sample); | |
112 | ||
113 | if (samples_per_pixel < EnvelopeThreshold) | |
114 | paint_trace(p, snapshot, y, left, | |
115 | start_sample, end_sample, | |
116 | pixels_offset, samples_per_pixel); | |
117 | else | |
118 | paint_envelope(p, snapshot, y, left, | |
119 | start_sample, end_sample, | |
120 | pixels_offset, samples_per_pixel); | |
121 | } | |
122 | ||
123 | void AnalogSignal::paint_trace(QPainter &p, | |
124 | const shared_ptr<pv::data::AnalogSnapshot> &snapshot, | |
125 | int y, int left, const int64_t start, const int64_t end, | |
126 | const double pixels_offset, const double samples_per_pixel) | |
127 | { | |
128 | const int64_t sample_count = end - start; | |
129 | ||
130 | const float *const samples = snapshot->get_samples(start, end); | |
131 | assert(samples); | |
132 | ||
133 | p.setPen(_colour); | |
134 | ||
135 | QPointF *points = new QPointF[sample_count]; | |
136 | QPointF *point = points; | |
137 | ||
138 | for (int64_t sample = start; sample != end; sample++) { | |
139 | const float x = (sample / samples_per_pixel - | |
140 | pixels_offset) + left; | |
141 | *point++ = QPointF(x, | |
142 | y - samples[sample - start] * _scale); | |
143 | } | |
144 | ||
145 | p.drawPolyline(points, point - points); | |
146 | ||
147 | delete[] samples; | |
148 | delete[] points; | |
149 | } | |
150 | ||
151 | void AnalogSignal::paint_envelope(QPainter &p, | |
152 | const shared_ptr<pv::data::AnalogSnapshot> &snapshot, | |
153 | int y, int left, const int64_t start, const int64_t end, | |
154 | const double pixels_offset, const double samples_per_pixel) | |
155 | { | |
156 | using pv::data::AnalogSnapshot; | |
157 | ||
158 | AnalogSnapshot::EnvelopeSection e; | |
159 | snapshot->get_envelope_section(e, start, end, samples_per_pixel); | |
160 | ||
161 | if (e.length < 2) | |
162 | return; | |
163 | ||
164 | p.setPen(QPen(Qt::NoPen)); | |
165 | p.setBrush(_colour); | |
166 | ||
167 | QRectF *const rects = new QRectF[e.length]; | |
168 | QRectF *rect = rects; | |
169 | ||
170 | for(uint64_t sample = 0; sample < e.length-1; sample++) { | |
171 | const float x = ((e.scale * sample + e.start) / | |
172 | samples_per_pixel - pixels_offset) + left; | |
173 | const AnalogSnapshot::EnvelopeSample *const s = | |
174 | e.samples + sample; | |
175 | ||
176 | // We overlap this sample with the next so that vertical | |
177 | // gaps do not appear during steep rising or falling edges | |
178 | const float b = y - max(s->max, (s+1)->min) * _scale; | |
179 | const float t = y - min(s->min, (s+1)->max) * _scale; | |
180 | ||
181 | float h = b - t; | |
182 | if(h >= 0.0f && h <= 1.0f) | |
183 | h = 1.0f; | |
184 | if(h <= 0.0f && h >= -1.0f) | |
185 | h = -1.0f; | |
186 | ||
187 | *rect++ = QRectF(x, t, 1.0f, h); | |
188 | } | |
189 | ||
190 | p.drawRects(rects, e.length); | |
191 | ||
192 | delete[] rects; | |
193 | delete[] e.samples; | |
194 | } | |
195 | ||
196 | } // namespace view | |
197 | } // namespace pv |