重构ringbuffer
parent
29c15e4016
commit
c8da1b73b6
|
|
@ -21,85 +21,71 @@ public:
|
|||
~RingBuffer();
|
||||
|
||||
protected:
|
||||
T *mData;
|
||||
uint64_t mSize;
|
||||
uint64_t mCurrentTail;
|
||||
uint64_t mCurrentHead;
|
||||
uint16_t mRemain;
|
||||
T *m_data;
|
||||
T *m_data_end;
|
||||
uint64_t m_size;
|
||||
T *m_tail;
|
||||
T *m_head;
|
||||
};
|
||||
|
||||
/**
|
||||
* author : caiyuzheng
|
||||
* function: 构造函数
|
||||
*/
|
||||
template<typename T>
|
||||
RingBuffer<T>::RingBuffer(uint64_t size) :
|
||||
mData(nullptr),
|
||||
mSize(0),
|
||||
mCurrentTail(0),
|
||||
mCurrentHead(0)
|
||||
m_data(nullptr),
|
||||
m_size(0),
|
||||
m_head(0),
|
||||
m_tail(0)
|
||||
{
|
||||
mSize = size;
|
||||
mData = new T[size];
|
||||
mRemain = mSize;
|
||||
m_size = size;
|
||||
m_data = new T[size];
|
||||
m_data_end = &m_data[size - 1];
|
||||
m_head = m_tail = m_data;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
RingBuffer<T>::~RingBuffer(){
|
||||
delete[] mData;
|
||||
delete[] m_data;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
void RingBuffer<T>::SetEmpty(){
|
||||
this->mCurrentHead = this->mCurrentTail;
|
||||
this->mRemain = this->mSize;
|
||||
this->m_head = this->m_tail;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
int RingBuffer<T>::Copy(T *data,uint64_t len){
|
||||
if(data == nullptr)
|
||||
return -1;
|
||||
if(mRemain == mSize)
|
||||
return 0;
|
||||
int bytes_read = 0;
|
||||
if(len > (mSize - mRemain)){
|
||||
bytes_read = mSize - mRemain;
|
||||
}else{
|
||||
bytes_read = len;
|
||||
}
|
||||
/// |start|head|.....|tail|end|
|
||||
if(mCurrentTail >= mCurrentHead){
|
||||
memcpy(data,&mData[mCurrentHead],sizeof(T)*bytes_read);
|
||||
return bytes_read;
|
||||
}
|
||||
/// |start|...|tail|.....|head|...|end|
|
||||
if(mCurrentHead > mCurrentTail){
|
||||
int off = mCurrentHead + bytes_read;
|
||||
if(off > mSize){
|
||||
int tmp = mSize - mCurrentHead;
|
||||
memcpy(data,&mData[mCurrentHead],(mSize - mCurrentHead)*sizeof(T));
|
||||
memcpy(&data[mSize - mCurrentHead],&mData[0],(bytes_read - tmp)*sizeof(T));
|
||||
return bytes_read;
|
||||
}
|
||||
if(off <= mSize){
|
||||
memcpy(data,&mData[mCurrentHead],sizeof(T)*bytes_read);
|
||||
return bytes_read;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
uint32_t RingBuffer<T>::Size(){
|
||||
return mSize;
|
||||
return m_size;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
uint32_t RingBuffer<T>::Length(){
|
||||
return mSize - mRemain;
|
||||
if(m_tail > m_head){
|
||||
return uint32_t(m_tail - m_head + 1);
|
||||
}
|
||||
if(m_tail == m_head){
|
||||
return 0;
|
||||
}
|
||||
if(m_head > m_tail){
|
||||
return uint32_t(m_data_end - m_head + 1) + uint32_t(m_tail - m_data); // todo unsure
|
||||
}
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
T RingBuffer<T>::At(uint64_t pos){
|
||||
return mData[pos];
|
||||
if(pos > (this->Length() - 1)){
|
||||
return T{};
|
||||
}
|
||||
T* p = m_head + pos;
|
||||
if(p > m_data_end){
|
||||
p = p - m_size;
|
||||
}
|
||||
return *p;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
|
|
@ -107,127 +93,60 @@ int RingBuffer<T>::Add(T *data,uint64_t len){
|
|||
if(data == nullptr){
|
||||
return -1;
|
||||
}
|
||||
// buffer already full
|
||||
if(mRemain == 0){
|
||||
return -2;
|
||||
int i = 0;
|
||||
std::cout<<"123 "<<m_tail<<" " << m_head <<std::endl;
|
||||
if(m_tail == m_head){
|
||||
std::cout<<"123"<<std::endl;
|
||||
*m_tail = data[0];
|
||||
for(i = 1;i < len;i++){
|
||||
T* tmp = m_tail;
|
||||
if(m_tail < m_data_end) {
|
||||
tmp++;
|
||||
}
|
||||
uint16_t bytes_write;
|
||||
if(len > mRemain){
|
||||
bytes_write = mRemain;
|
||||
}else{
|
||||
bytes_write = len;
|
||||
if(m_tail == m_data_end) {
|
||||
tmp = m_data;
|
||||
}
|
||||
|
||||
if(mCurrentTail == mCurrentHead){
|
||||
/// |start also head also tail|...|end|
|
||||
if(mCurrentHead == 0){
|
||||
memcpy(mData,data,bytes_write*sizeof(T));
|
||||
mCurrentTail = bytes_write;
|
||||
mRemain -= bytes_write;
|
||||
return bytes_write;
|
||||
}else{
|
||||
/// |start |...| head also tail|...|end|
|
||||
if(mCurrentTail + bytes_write < mSize){
|
||||
memcpy(&mData[mCurrentTail],data,(bytes_write)*sizeof(T));
|
||||
mCurrentTail += bytes_write;
|
||||
mRemain -= bytes_write;
|
||||
}else{
|
||||
memcpy(&mData[mCurrentTail],data,(mSize - mCurrentTail)*sizeof(T));
|
||||
std::cout<<"\r\n"<<(mCurrentTail + bytes_write)%mSize<<"\r\n";
|
||||
memcpy(&mData[0],&data[mSize - mCurrentTail],(mCurrentTail + bytes_write)%mSize*sizeof(T));
|
||||
mCurrentTail = (mCurrentTail + bytes_write)%mSize;
|
||||
mRemain -= bytes_write;
|
||||
return bytes_write;
|
||||
if(tmp == m_head) {
|
||||
break;
|
||||
}
|
||||
(*tmp) = data[i];
|
||||
m_tail = tmp;
|
||||
}
|
||||
}else{
|
||||
std::cout<<"123"<<std::endl;
|
||||
for(i = 0;i < len;i++){
|
||||
T* tmp = m_tail;
|
||||
if(m_tail < m_data_end) {
|
||||
tmp++;
|
||||
}
|
||||
if(m_tail == m_data_end) {
|
||||
tmp = m_data;
|
||||
}
|
||||
if(tmp == m_head) {
|
||||
break;
|
||||
}
|
||||
(*tmp) = data[i];
|
||||
m_tail = tmp;
|
||||
}
|
||||
}
|
||||
|
||||
/// |start|head|.....|tail|end|
|
||||
if(mCurrentTail > mCurrentHead){
|
||||
uint16_t off = mCurrentTail + bytes_write;
|
||||
if(off > mSize){
|
||||
memcpy(&mData[mCurrentTail],data,
|
||||
sizeof(T)*(mSize - mCurrentTail));
|
||||
memcpy(mData,data + sizeof(T)*(mSize - mCurrentTail),
|
||||
(bytes_write - (mSize - mCurrentTail))*sizeof(T));
|
||||
mRemain -= bytes_write;
|
||||
mCurrentTail = off - mSize;
|
||||
return bytes_write;
|
||||
}
|
||||
if(off <= mSize){
|
||||
memcpy(&mData[mCurrentTail],data,
|
||||
sizeof(T)*bytes_write);
|
||||
mRemain -= bytes_write;
|
||||
if(off < mSize)
|
||||
mCurrentTail += bytes_write;
|
||||
else
|
||||
mCurrentTail = 0;
|
||||
return bytes_write;
|
||||
}
|
||||
}
|
||||
/// |start|...|tail|.....|head|...|end|
|
||||
if(mCurrentHead > mCurrentTail){
|
||||
memcpy(&mData[mCurrentTail],data,sizeof(T)*(bytes_write));
|
||||
mCurrentTail += bytes_write;
|
||||
mRemain -= bytes_write;
|
||||
return bytes_write;
|
||||
}
|
||||
return i;
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
int RingBuffer<T>::Take(T *data,uint64_t len){
|
||||
if(data == nullptr)
|
||||
return -1;
|
||||
if(mRemain == mSize)
|
||||
return 0;
|
||||
int bytes_read = 0;
|
||||
if(len > (mSize - mRemain)){
|
||||
bytes_read = mSize - mRemain;
|
||||
}else{
|
||||
bytes_read = len;
|
||||
}
|
||||
/// |start|head|.....|tail|end|
|
||||
if(mCurrentTail >= mCurrentHead){
|
||||
memcpy(data,&mData[mCurrentHead],sizeof(T)*bytes_read);
|
||||
mCurrentHead += bytes_read;
|
||||
mRemain = mSize - (mCurrentTail - mCurrentHead);
|
||||
return bytes_read;
|
||||
|
||||
}
|
||||
/// |start|...|tail|.....|head|...|end|
|
||||
if(mCurrentHead > mCurrentTail){
|
||||
int off = mCurrentHead + bytes_read;
|
||||
if(off > mSize){
|
||||
int tmp = mSize - mCurrentHead;
|
||||
memcpy(data,&mData[mCurrentHead],(mSize - mCurrentHead)*sizeof(T));
|
||||
memcpy(&data[mSize - mCurrentHead],&mData[0],(bytes_read - tmp)*sizeof(T));
|
||||
mCurrentHead = off - mSize;
|
||||
mRemain = mSize - (mCurrentTail - mCurrentHead);
|
||||
return bytes_read;
|
||||
}
|
||||
if(off <= mSize){
|
||||
memcpy(data,&mData[mCurrentHead],sizeof(T)*bytes_read);
|
||||
if(off == mSize){
|
||||
mCurrentHead = 0;
|
||||
mRemain = mSize - mCurrentTail;
|
||||
return bytes_read;
|
||||
}
|
||||
else{
|
||||
mCurrentHead += bytes_read;
|
||||
mRemain = mSize - ((mSize - mCurrentHead) + mCurrentTail);
|
||||
return bytes_read;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
uint32_t RingBuffer<T>::CanReadCount(){
|
||||
return mSize - mRemain;
|
||||
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
uint32_t RingBuffer<T>::CanWriteCount(){
|
||||
return mRemain;
|
||||
|
||||
}
|
||||
#endif
|
||||
|
|
|
|||
Loading…
Reference in New Issue