sdrangel/sdrbase/util/doublebuffer.h

148 wiersze
4.5 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 //
// (at your option) any later version. //
// //
// 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>
#include <QByteArray>
#include "simpleserializer.h"
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; }
typename std::vector<T>::const_iterator begin() const { return m_data.begin(); }
typename std::vector<T>::iterator begin() { return m_data.begin(); }
unsigned int absoluteFill() const { return m_current - m_data.begin(); }
void reset() { m_current = m_data.begin(); }
QByteArray serialize() const
{
SimpleSerializer s(1);
QByteArray buf(reinterpret_cast<const char*>(m_data.data()), m_data.size()*sizeof(T));
s.writeS32(1, m_size);
s.writeU32(2, m_current - m_data.begin());
s.writeBlob(3, buf);
return s.final();
}
bool deserialize(const QByteArray& data)
{
SimpleDeserializer d(data);
if(!d.isValid()) {
return false;
}
if (d.getVersion() == 1)
{
unsigned int tmpUInt;
QByteArray buf;
d.readS32(1, &m_size, m_data.size()/2);
m_data.resize(2*m_size);
d.readU32(2, &tmpUInt, 0);
m_current = m_data.begin() + tmpUInt;
d.readBlob(3, &buf);
//qDebug("DoubleBufferSimple::deserialize: m_data.size(): %u buf.size(): %d", m_data.size(), buf.size());
//std::copy(reinterpret_cast<char *>(m_data.data()), buf.data(), buf.data() + buf.size()); // bug
memcpy(reinterpret_cast<char *>(m_data.data()), buf.data(), buf.size());
return true;
}
else
{
return false;
}
}
private:
int m_size;
std::vector<T> m_data;
typename std::vector<T>::iterator m_current;
};
#endif /* SDRBASE_UTIL_DOUBLEBUFFER_H_ */