X-Git-Url: https://sigrok.org/gitweb/?p=pulseview.git;a=blobdiff_plain;f=pv%2Fdata%2Fsegment.hpp;h=4c6c36af3c3b0c0b6245ae0e257a2f1000992ae2;hp=db34b7687cdd098179b55a483be5ecab7c7c9650;hb=65c92359634f672e5f472a5214719dabc7e20883;hpb=efdec55aec1a137460fa362a381ed1904182bfed

diff --git a/pv/data/segment.hpp b/pv/data/segment.hpp
index db34b768..4c6c36af 100644
--- a/pv/data/segment.hpp
+++ b/pv/data/segment.hpp
@@ -1,6 +1,7 @@
 /*
  * This file is part of the PulseView project.
  *
+ * Copyright (C) 2017 Soeren Apel <soeren@apelpie.net>
  * Copyright (C) 2012 Joel Holdsworth <joel@airwebreathe.org.uk>
  *
  * This program is free software; you can redistribute it and/or modify
@@ -22,17 +23,45 @@
 
 #include "pv/util.hpp"
 
-#include <thread>
 #include <mutex>
+#include <thread>
 #include <vector>
 
+#include <QObject>
+
+using std::recursive_mutex;
+using std::vector;
+
+namespace SegmentTest {
+struct SmallSize8Single;
+struct MediumSize8Single;
+struct MaxSize8Single;
+struct MediumSize24Single;
+struct MediumSize32Single;
+struct MaxSize32Single;
+struct MediumSize32Multi;
+struct MaxSize32Multi;
+struct MaxSize32MultiAtOnce;
+struct MaxSize32MultiIterated;
+}  // namespace SegmentTest
+
 namespace pv {
 namespace data {
 
-class Segment
+typedef struct {
+	uint64_t sample_index, chunk_num, chunk_offs;
+	uint8_t* chunk;
+} SegmentDataIterator;
+
+class Segment : public QObject
 {
+	Q_OBJECT
+
+private:
+	static const uint64_t MaxChunkSize;
+
 public:
-	Segment(uint64_t samplerate, unsigned int unit_size);
+	Segment(uint32_t segment_id, uint64_t samplerate, unsigned int unit_size);
 
 	virtual ~Segment();
 
@@ -45,44 +74,47 @@ public:
 
 	unsigned int unit_size() const;
 
-	/**
-	 * @brief Increase the capacity of the segment.
-	 *
-	 * Increasing the capacity allows samples to be appended without needing
-	 * to reallocate memory.
-	 *
-	 * For the best efficiency @c set_capacity() should be called once before
-	 * @c append_data() is called to set up the segment with the expected number
-	 * of samples that will be appended in total.
-	 *
-	 * @note The capacity will automatically be increased when @c append_data()
-	 * is called if there is not enough capacity in the buffer to store the samples.
-	 *
-	 * @param[in] new_capacity The new capacity of the segment. If this value is
-	 * 	smaller or equal than the current capacity then the method has no effect.
-	 */
-	void set_capacity(uint64_t new_capacity);
-
-	/**
-	 * @brief Get the current capacity of the segment.
-	 *
-	 * The capacity can be increased by calling @c set_capacity().
-	 *
-	 * @return The current capacity of the segment.
-	 */
-	uint64_t capacity() const;
+	uint32_t segment_id() const;
 
-protected:
-	void append_data(void *data, uint64_t samples);
+	void set_complete();
+	bool is_complete() const;
+
+	void free_unused_memory();
 
 protected:
-	mutable std::recursive_mutex mutex_;
-	std::vector<uint8_t> data_;
+	void append_single_sample(void *data);
+	void append_samples(void *data, uint64_t samples);
+	void get_raw_samples(uint64_t start, uint64_t count, uint8_t *dest) const;
+
+	SegmentDataIterator* begin_sample_iteration(uint64_t start);
+	void continue_sample_iteration(SegmentDataIterator* it, uint64_t increase);
+	void end_sample_iteration(SegmentDataIterator* it);
+	uint8_t* get_iterator_value(SegmentDataIterator* it);
+
+	uint32_t segment_id_;
+	mutable recursive_mutex mutex_;
+	vector<uint8_t*> data_chunks_;
+	uint8_t* current_chunk_;
+	uint64_t used_samples_, unused_samples_;
 	uint64_t sample_count_;
 	pv::util::Timestamp start_time_;
 	double samplerate_;
-	uint64_t capacity_;
+	uint64_t chunk_size_;
 	unsigned int unit_size_;
+	int iterator_count_;
+	bool mem_optimization_requested_;
+	bool is_complete_;
+
+	friend struct SegmentTest::SmallSize8Single;
+	friend struct SegmentTest::MediumSize8Single;
+	friend struct SegmentTest::MaxSize8Single;
+	friend struct SegmentTest::MediumSize24Single;
+	friend struct SegmentTest::MediumSize32Single;
+	friend struct SegmentTest::MaxSize32Single;
+	friend struct SegmentTest::MediumSize32Multi;
+	friend struct SegmentTest::MaxSize32Multi;
+	friend struct SegmentTest::MaxSize32MultiAtOnce;
+	friend struct SegmentTest::MaxSize32MultiIterated;
 };
 
 } // namespace data