]> sigrok.org Git - pulseview.git/blobdiff - pv/view/analogsignal.cpp
Moved all sr_probe modification into pv::DevInst
[pulseview.git] / pv / view / analogsignal.cpp
index a6f032b5793aa0422e8e71b07f5e5b2f56fb3e2d..a64cf70f430dc4fd7c8d8d47432c24f593aa63ed 100644 (file)
 #include "analogsignal.h"
 #include "pv/data/analog.h"
 #include "pv/data/analogsnapshot.h"
+#include "pv/view/view.h"
 
-using namespace boost;
-using namespace std;
+using boost::shared_ptr;
+using std::max;
+using std::min;
+using std::deque;
 
 namespace pv {
 namespace view {
 
-AnalogSignal::AnalogSignal(QString name, shared_ptr<data::Analog> data) :
-       Signal(name),
-       _data(data)
+const QColor AnalogSignal::SignalColours[4] = {
+       QColor(0xC4, 0xA0, 0x00),       // Yellow
+       QColor(0x87, 0x20, 0x7A),       // Magenta
+       QColor(0x20, 0x4A, 0x87),       // Blue
+       QColor(0x4E, 0x9A, 0x06)        // Green
+};
+
+const float AnalogSignal::EnvelopeThreshold = 256.0f;
+
+AnalogSignal::AnalogSignal(shared_ptr<pv::DevInst> dev_inst,
+       const sr_probe *const probe, shared_ptr<data::Analog> data) :
+       Signal(dev_inst, probe),
+       _data(data),
+       _scale(1.0f)
 {
-       _colour = Qt::blue;
+       _colour = SignalColours[probe->index % countof(SignalColours)];
 }
 
-void AnalogSignal::paint(QPainter &p, int y, int left, int right, double scale,
-       double offset)
+AnalogSignal::~AnalogSignal()
+{
+}
+
+shared_ptr<pv::data::SignalData> AnalogSignal::data() const
+{
+       return _data;
+}
+
+shared_ptr<pv::data::Analog> AnalogSignal::analog_data() const
+{
+       return _data;
+}
+
+void AnalogSignal::set_scale(float scale)
+{
+       _scale = scale;
+}
+
+void AnalogSignal::paint_back(QPainter &p, int left, int right)
+{
+       if (_probe->enabled)
+               paint_axis(p, get_y(), left, right);
+}
+
+void AnalogSignal::paint_mid(QPainter &p, int left, int right)
 {
-       assert(scale > 0);
        assert(_data);
        assert(right >= left);
 
-       paint_axis(p, y, left, right);
+       assert(_view);
+       const int y = _v_offset - _view->v_offset();
+
+       const double scale = _view->scale();
+       assert(scale > 0);
+
+       const double offset = _view->offset();
+
+       if (!_probe->enabled)
+               return;
 
        const deque< shared_ptr<pv::data::AnalogSnapshot> > &snapshots =
                _data->get_snapshots();
@@ -57,7 +103,7 @@ void AnalogSignal::paint(QPainter &p, int y, int left, int right, double scale,
                snapshots.front();
 
        const double pixels_offset = offset / scale;
-       const double samplerate = _data->get_samplerate();
+       const double samplerate = _data->samplerate();
        const double start_time = _data->get_start_time();
        const int64_t last_sample = snapshot->get_sample_count() - 1;
        const double samples_per_pixel = samplerate * scale;
@@ -66,30 +112,91 @@ void AnalogSignal::paint(QPainter &p, int y, int left, int right, double scale,
 
        const int64_t start_sample = min(max((int64_t)floor(start),
                (int64_t)0), last_sample);
-       const int64_t end_sample = min(max((int64_t)ceil(end),
+       const int64_t end_sample = min(max((int64_t)ceil(end) + 1,
                (int64_t)0), last_sample);
-       const int64_t sample_count = end_sample - start_sample;
 
-       const float* samples = snapshot->get_samples(
-               start_sample, end_sample);
+       if (samples_per_pixel < EnvelopeThreshold)
+               paint_trace(p, snapshot, y, left,
+                       start_sample, end_sample,
+                       pixels_offset, samples_per_pixel);
+       else
+               paint_envelope(p, snapshot, y, left,
+                       start_sample, end_sample,
+                       pixels_offset, samples_per_pixel);
+}
+
+void AnalogSignal::paint_trace(QPainter &p,
+       const shared_ptr<pv::data::AnalogSnapshot> &snapshot,
+       int y, int left, const int64_t start, const int64_t end,
+       const double pixels_offset, const double samples_per_pixel)
+{
+       const int64_t sample_count = end - start;
+
+       const float *const samples = snapshot->get_samples(start, end);
        assert(samples);
 
+       p.setPen(_colour);
+
        QPointF *points = new QPointF[sample_count];
        QPointF *point = points;
 
-       for (int64_t sample = start_sample;
-               sample != end_sample; sample++) {
+       for (int64_t sample = start; sample != end; sample++) {
                const float x = (sample / samples_per_pixel -
                        pixels_offset) + left;
                *point++ = QPointF(x,
-                       samples[sample - start_sample] + y);
+                       y - samples[sample - start] * _scale);
        }
 
-       p.setPen(_colour);
-       p.drawPoints(points, point - points);
+       p.drawPolyline(points, point - points);
 
+       delete[] samples;
        delete[] points;
 }
 
+void AnalogSignal::paint_envelope(QPainter &p,
+       const shared_ptr<pv::data::AnalogSnapshot> &snapshot,
+       int y, int left, const int64_t start, const int64_t end,
+       const double pixels_offset, const double samples_per_pixel)
+{
+       using pv::data::AnalogSnapshot;
+
+       AnalogSnapshot::EnvelopeSection e;
+       snapshot->get_envelope_section(e, start, end, samples_per_pixel);
+
+       if (e.length < 2)
+               return;
+
+       p.setPen(QPen(Qt::NoPen));
+       p.setBrush(_colour);
+
+       QRectF *const rects = new QRectF[e.length];
+       QRectF *rect = rects;
+
+       for(uint64_t sample = 0; sample < e.length-1; sample++) {
+               const float x = ((e.scale * sample + e.start) /
+                       samples_per_pixel - pixels_offset) + left;
+               const AnalogSnapshot::EnvelopeSample *const s =
+                       e.samples + sample;
+
+               // We overlap this sample with the next so that vertical
+               // gaps do not appear during steep rising or falling edges
+               const float b = y - max(s->max, (s+1)->min) * _scale;
+               const float t = y - min(s->min, (s+1)->max) * _scale;
+
+               float h = b - t;
+               if(h >= 0.0f && h <= 1.0f)
+                       h = 1.0f;
+               if(h <= 0.0f && h >= -1.0f)
+                       h = -1.0f;
+
+               *rect++ = QRectF(x, t, 1.0f, h);
+       }
+
+       p.drawRects(rects, e.length);
+
+       delete[] rects;
+       delete[] e.samples;
+}
+
 } // namespace view
 } // namespace pv