5d0a242aaf00f0634339eb25eb9f48644381f3e5
[pulseview.git] / pv / data / segment.hpp
1 /*
2  * This file is part of the PulseView project.
3  *
4  * Copyright (C) 2012 Joel Holdsworth <joel@airwebreathe.org.uk>
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
19  */
20
21 #ifndef PULSEVIEW_PV_DATA_SEGMENT_HPP
22 #define PULSEVIEW_PV_DATA_SEGMENT_HPP
23
24 #include "pv/util.hpp"
25
26 #include <thread>
27 #include <mutex>
28 #include <vector>
29
30 namespace pv {
31 namespace data {
32
33 class Segment
34 {
35 public:
36         Segment(uint64_t samplerate, unsigned int unit_size);
37
38         virtual ~Segment();
39
40         uint64_t get_sample_count() const;
41
42         const pv::util::Timestamp& start_time() const;
43
44         double samplerate() const;
45         void set_samplerate(double samplerate);
46
47         unsigned int unit_size() const;
48
49         /**
50          * @brief Increase the capacity of the segment.
51          *
52          * Increasing the capacity allows samples to be appended without needing
53          * to reallocate memory.
54          *
55          * For the best efficiency @c set_capacity() should be called once before
56          * @c append_data() is called to set up the segment with the expected number
57          * of samples that will be appended in total.
58          *
59          * @note The capacity will automatically be increased when @c append_data()
60          * is called if there is not enough capacity in the buffer to store the samples.
61          *
62          * @param[in] new_capacity The new capacity of the segment. If this value is
63          *      smaller or equal than the current capacity then the method has no effect.
64          */
65         void set_capacity(uint64_t new_capacity);
66
67         /**
68          * @brief Get the current capacity of the segment.
69          *
70          * The capacity can be increased by calling @c set_capacity().
71          *
72          * @return The current capacity of the segment.
73          */
74         uint64_t capacity() const;
75
76 protected:
77         void append_data(void *data, uint64_t samples);
78
79 protected:
80         mutable std::recursive_mutex mutex_;
81         std::vector<uint8_t> data_;
82         uint64_t sample_count_;
83         pv::util::Timestamp start_time_;
84         double samplerate_;
85         uint64_t capacity_;
86         unsigned int unit_size_;
87 };
88
89 } // namespace data
90 } // namespace pv
91
92 #endif // PULSEVIEW_PV_DATA_SEGMENT_HPP