X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=pv%2Fdata%2Fanalogsnapshot.cpp;h=5e995158ee1dd672163c741538768d767599749a;hb=ff008de665c7990d5f3408f918ff090d8e6c60b2;hp=c052782ed6cc6842f5029c80b6de3223ee76d50c;hpb=fda5b6e0e0ebbe55f3d173051b8800293f87cd09;p=pulseview.git

diff --git a/pv/data/analogsnapshot.cpp b/pv/data/analogsnapshot.cpp
index c052782e..5e995158 100644
--- a/pv/data/analogsnapshot.cpp
+++ b/pv/data/analogsnapshot.cpp
@@ -27,12 +27,14 @@
 
 #include <algorithm>
 
-#include <boost/foreach.hpp>
+#include "analogsnapshot.hpp"
 
-#include "analogsnapshot.h"
-
-using namespace boost;
-using namespace std;
+using std::lock_guard;
+using std::recursive_mutex;
+using std::max;
+using std::max_element;
+using std::min;
+using std::min_element;
 
 namespace pv {
 namespace data {
@@ -43,26 +45,41 @@ const float AnalogSnapshot::LogEnvelopeScaleFactor =
 	logf(EnvelopeScaleFactor);
 const uint64_t AnalogSnapshot::EnvelopeDataUnit = 64*1024;	// bytes
 
-AnalogSnapshot::AnalogSnapshot(const sr_datafeed_analog &analog) :
-	Snapshot(sizeof(float))
+AnalogSnapshot::AnalogSnapshot(
+	uint64_t samplerate, const uint64_t expected_num_samples) :
+	Snapshot(samplerate, sizeof(float))
 {
-	lock_guard<recursive_mutex> lock(_mutex);
-	memset(_envelope_levels, 0, sizeof(_envelope_levels));
-	append_payload(analog);
+	set_capacity(expected_num_samples);
+
+	lock_guard<recursive_mutex> lock(mutex_);
+	memset(envelope_levels_, 0, sizeof(envelope_levels_));
 }
 
 AnalogSnapshot::~AnalogSnapshot()
 {
-	lock_guard<recursive_mutex> lock(_mutex);
-	BOOST_FOREACH(Envelope &e, _envelope_levels)
+	lock_guard<recursive_mutex> lock(mutex_);
+	for (Envelope &e : envelope_levels_)
 		free(e.samples);
 }
 
-void AnalogSnapshot::append_payload(
-	const sr_datafeed_analog &analog)
+void AnalogSnapshot::append_interleaved_samples(const float *data,
+	size_t sample_count, size_t stride)
 {
-	lock_guard<recursive_mutex> lock(_mutex);
-	append_data(analog.data, analog.num_samples);
+	assert(unit_size_ == sizeof(float));
+
+	lock_guard<recursive_mutex> lock(mutex_);
+
+	data_.resize((sample_count_ + sample_count) * sizeof(float));
+
+	float *dst = (float*)data_.data() + sample_count_;
+	const float *dst_end = dst + sample_count;
+	while (dst != dst_end)
+	{
+		*dst++ = *data;
+		data += stride;
+	}
+
+	sample_count_ += sample_count;
 
 	// Generate the first mip-map from the data
 	append_payload_to_envelope_levels();
@@ -72,19 +89,43 @@ const float* AnalogSnapshot::get_samples(
 	int64_t start_sample, int64_t end_sample) const
 {
 	assert(start_sample >= 0);
-	assert(start_sample < (int64_t)_sample_count);
+	assert(start_sample < (int64_t)sample_count_);
 	assert(end_sample >= 0);
-	assert(end_sample < (int64_t)_sample_count);
+	assert(end_sample < (int64_t)sample_count_);
 	assert(start_sample <= end_sample);
 
-	lock_guard<recursive_mutex> lock(_mutex);
+	lock_guard<recursive_mutex> lock(mutex_);
 
 	float *const data = new float[end_sample - start_sample];
-	memcpy(data, (float*)_data + start_sample, sizeof(float) *
+	memcpy(data, (float*)data_.data() + start_sample, sizeof(float) *
 		(end_sample - start_sample));
 	return data;
 }
 
+void AnalogSnapshot::get_envelope_section(EnvelopeSection &s,
+	uint64_t start, uint64_t end, float min_length) const
+{
+	assert(end <= get_sample_count());
+	assert(start <= end);
+	assert(min_length > 0);
+
+	lock_guard<recursive_mutex> lock(mutex_);
+
+	const unsigned int min_level = max((int)floorf(logf(min_length) /
+		LogEnvelopeScaleFactor) - 1, 0);
+	const unsigned int scale_power = (min_level + 1) *
+		EnvelopeScalePower;
+	start >>= scale_power;
+	end >>= scale_power;
+
+	s.start = start << scale_power;
+	s.scale = 1 << scale_power;
+	s.length = end - start;
+	s.samples = new EnvelopeSample[s.length];
+	memcpy(s.samples, envelope_levels_[min_level].samples + start,
+		s.length * sizeof(EnvelopeSample));
+}
+
 void AnalogSnapshot::reallocate_envelope(Envelope &e)
 {
 	const uint64_t new_data_length = ((e.length + EnvelopeDataUnit - 1) /
@@ -99,13 +140,13 @@ void AnalogSnapshot::reallocate_envelope(Envelope &e)
 
 void AnalogSnapshot::append_payload_to_envelope_levels()
 {
-	Envelope &e0 = _envelope_levels[0];
+	Envelope &e0 = envelope_levels_[0];
 	uint64_t prev_length;
 	EnvelopeSample *dest_ptr;
 
 	// Expand the data buffer to fit the new samples
 	prev_length = e0.length;
-	e0.length = _sample_count / EnvelopeScaleFactor;
+	e0.length = sample_count_ / EnvelopeScaleFactor;
 
 	// Break off if there are no new samples to compute
 	if (e0.length == prev_length)
@@ -116,9 +157,9 @@ void AnalogSnapshot::append_payload_to_envelope_levels()
 	dest_ptr = e0.samples + prev_length;
 
 	// Iterate through the samples to populate the first level mipmap
-	const float *const end_src_ptr = (float*)_data +
+	const float *const end_src_ptr = (float*)data_.data() +
 		e0.length * EnvelopeScaleFactor;
-	for (const float *src_ptr = (float*)_data +
+	for (const float *src_ptr = (float*)data_.data() +
 		prev_length * EnvelopeScaleFactor;
 		src_ptr < end_src_ptr; src_ptr += EnvelopeScaleFactor)
 	{
@@ -133,8 +174,8 @@ void AnalogSnapshot::append_payload_to_envelope_levels()
 	// Compute higher level mipmaps
 	for (unsigned int level = 1; level < ScaleStepCount; level++)
 	{
-		Envelope &e = _envelope_levels[level];
-		const Envelope &el = _envelope_levels[level-1];
+		Envelope &e = envelope_levels_[level];
+		const Envelope &el = envelope_levels_[level-1];
 
 		// Expand the data buffer to fit the new samples
 		prev_length = e.length;