Make member variable underscores a suffix instead of a prefix

[pulseview.git] / pv / data / snapshot.cpp

diff --git a/pv/data/snapshot.cpp b/pv/data/snapshot.cpp
index 602388412f00624cc6db0cfbce81fff684b1223f..80b9fb3792e09e25a099edd631461ba67f411569 100644 (file)
--- a/pv/data/snapshot.cpp
+++ b/pv/data/snapshot.cpp
@@ -24,45 +24,69 @@
 #include <stdlib.h>
 #include <string.h>
 
-using namespace boost;
+using std::lock_guard;
+using std::recursive_mutex;
 
 namespace pv {
 namespace data {
 
-Snapshot::Snapshot(int unit_size) :
-       _data(NULL),
-       _sample_count(0),
-       _unit_size(unit_size)
+Snapshot::Snapshot(unsigned int unit_size) :
+       sample_count_(0),
+       capacity_(0),
+       unit_size_(unit_size)
 {
-       lock_guard<recursive_mutex> lock(_mutex);
-       assert(_unit_size > 0);
+       lock_guard<recursive_mutex> lock(mutex_);
+       assert(unit_size_ > 0);
 }
 
 Snapshot::~Snapshot()
 {
-       lock_guard<recursive_mutex> lock(_mutex);
-       free(_data);
+       lock_guard<recursive_mutex> lock(mutex_);
 }
 
 uint64_t Snapshot::get_sample_count() const
 {
-       lock_guard<recursive_mutex> lock(_mutex);
-       return _sample_count;
+       lock_guard<recursive_mutex> lock(mutex_);
+       return sample_count_;
 }
 
-int Snapshot::unit_size() const
+unsigned int Snapshot::unit_size() const
 {
-       return _unit_size;
+       return unit_size_;
+}
+
+void Snapshot::set_capacity(const uint64_t new_capacity)
+{
+       lock_guard<recursive_mutex> lock(mutex_);
+
+       assert(capacity_ >= sample_count_);
+       if (new_capacity > capacity_) {
+               capacity_ = new_capacity;
+               data_.resize((new_capacity * unit_size_) + sizeof(uint64_t));
+       }
+}
+
+uint64_t Snapshot::capacity() const
+{
+       lock_guard<recursive_mutex> lock(mutex_);
+       return data_.size();
 }
 
 void Snapshot::append_data(void *data, uint64_t samples)
 {
-       lock_guard<recursive_mutex> lock(_mutex);
-       _data = realloc(_data, (_sample_count + samples) * _unit_size +
-               sizeof(uint64_t));
-       memcpy((uint8_t*)_data + _sample_count * _unit_size,
-               data, samples * _unit_size);
-       _sample_count += samples;
+       lock_guard<recursive_mutex> lock(mutex_);
+
+       assert(capacity_ >= sample_count_);
+
+       // Ensure there's enough capacity to copy.
+       const uint64_t free_space = capacity_ - sample_count_;
+       if (free_space < samples) {
+               set_capacity(sample_count_ + samples);
+       }
+
+       memcpy((uint8_t*)data_.data() + sample_count_ * unit_size_,
+               data, samples * unit_size_);
+       sample_count_ += samples;
 }
 
 } // namespace data