X-Git-Url: https://sigrok.org/gitweb/?p=pulseview.git;a=blobdiff_plain;f=pv%2Fview%2Fanalogsignal.cpp;h=53af2643132ee895c65e5bd8df10e4bc655a3061;hp=209512b595880d9be9e55aed4947935e89b5ae22;hb=03ce95a9bb81c05bae0da9de189d280214db67c3;hpb=043ce0607fad7d72bf1e1488639e81f4adfd6613

diff --git a/pv/view/analogsignal.cpp b/pv/view/analogsignal.cpp
index 209512b5..53af2643 100644
--- a/pv/view/analogsignal.cpp
+++ b/pv/view/analogsignal.cpp
@@ -25,6 +25,7 @@
 #include "analogsignal.h"
 #include "pv/data/analog.h"
 #include "pv/data/analogsnapshot.h"
+#include "pv/view/view.h"
 
 using namespace boost;
 using namespace std;
@@ -32,21 +33,54 @@ using namespace std;
 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(pv::SigSession &session, const sr_probe *const probe,
+	shared_ptr<data::Analog> data) :
+	Signal(session, 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()
+{
+}
+
+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();
@@ -66,27 +100,92 @@ 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 float* samples = snapshot->get_samples();
+	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);
 
-	QPointF *points = new QPointF[end_sample - start_sample];
+	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] + y);
+		*point++ = QPointF(x,
+			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 namespace Qt;
+	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(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