sdrangel/sdrbase/util/doublebuffer.h

99 lines
3.0 KiB
C++

///////////////////////////////////////////////////////////////////////////////////
// Copyright (C) 2016 Edouard Griffiths, F4EXB //
// //
// This program is free software; you can redistribute it and/or modify //
// it under the terms of the GNU General Public License as published by //
// the Free Software Foundation as version 3 of the License, or //
// //
// This program is distributed in the hope that it will be useful, //
// but WITHOUT ANY WARRANTY; without even the implied warranty of //
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the //
// GNU General Public License V3 for more details. //
// //
// You should have received a copy of the GNU General Public License //
// along with this program. If not, see <http://www.gnu.org/licenses/>. //
///////////////////////////////////////////////////////////////////////////////////
#ifndef SDRBASE_UTIL_DOUBLEBUFFER_H_
#define SDRBASE_UTIL_DOUBLEBUFFER_H_
#include <vector>
#include <algorithm>
template<typename T>
class DoubleBufferSimple
{
public:
DoubleBufferSimple()
{
m_size = 0;
m_current = m_data.end();
}
~DoubleBufferSimple() {}
DoubleBufferSimple(const DoubleBufferSimple& other)
{
m_size = other.m_size;
m_data = other.m_data;
m_current = m_data.begin();
}
DoubleBufferSimple& operator=(const DoubleBufferSimple& other)
{
if (&other == this) {
return *this;
}
m_size = other.m_size;
m_data = other.m_data;
m_current = m_data.begin();
return *this;
}
void resize(int size)
{
m_size = size;
m_data.resize(2*size);
m_current = m_data.begin();
}
void write(const typename std::vector<T>::const_iterator& begin, const typename std::vector<T>::const_iterator& cend)
{
typename std::vector<T>::const_iterator end = cend;
if ((end - begin) > m_size)
{
end = begin + m_size;
}
int insize = end - begin;
std::copy(begin, end, m_current);
if (((m_current - m_data.begin()) + insize) > m_size)
{
int sizeLeft = m_size - (m_current - m_data.begin());
std::copy(begin, begin + sizeLeft, m_current + m_size);
std::copy(begin + sizeLeft, end, m_data.begin());
m_current = m_data.begin() + (insize - sizeLeft);
}
else
{
std::copy(begin, end, m_current + m_size);
m_current += insize;
}
}
typename std::vector<T>::iterator getCurrent() const { return m_current + m_size; }
unsigned int absoluteFill() const { return m_current - m_data.begin(); }
void reset() { m_current = m_data.begin(); }
private:
int m_size;
std::vector<T> m_data;
typename std::vector<T>::iterator m_current;
};
#endif /* SDRBASE_UTIL_DOUBLEBUFFER_H_ */