C++版循环缓冲

CCycleBuffer.h

#ifndef __test__CCycleBuffer__
#define __test__CCycleBuffer__
 
#include <iostream>
#include <assert.h>
class CCycleBuffer
{
public:
    bool isFull();
    bool isEmpty();
    void Empty();
    int GetLength();
    CCycleBuffer(int size);
    virtual ~CCycleBuffer();
    int Write(char* buf, int count);
    int Read(char* buf, int count);
private:
    bool m_bEmpty, m_bFull;
    char * m_pBuf;
    int m_nBufSize;
    int m_nReadPos;
    int m_nWritePos;
    
public:
    int GetReadPos() { return m_nReadPos; }
    int GetWritePos() { return m_nWritePos; }
};
#endif /* defined(__test__CCycleBuffer__) */

CCycleBuffer.c

//环形缓冲区源文件
//尽量用C运行时库代码,改变了原作者基于WindowsAPI的代码。
//修改了原作者的两处错误,1是read函数最后的else 少了一个leftcount
//第二个错误,是write函数中,m_nWritePos 变量,他重新定义了一个。这些错误编译不会出错。
//但是运行起来,要了亲命啊。
 
#include "CCycleBuffer.h"
#include <assert.h>
#include <memory.h>
// 定义 
CCycleBuffer::CCycleBuffer(int size)
 
{<br>m_nBufSize = size; 
m_nReadPos =0; 
m_nWritePos =0; 
m_pBuf =newchar[m_nBufSize];
m_bEmpty =true; 
m_bFull =false;
test =0;
} 
CCycleBuffer::~CCycleBuffer() 
{ 
delete[] m_pBuf; 
} 
/************************************************************************/ 
/* 向缓冲区写入数据,返回实际写入的字节数                               */ 
/************************************************************************/ 
int CCycleBuffer::write(char* buf,int count) 
{ 
if(count <=0)
  return0; 
m_bEmpty =false; 
// 缓冲区已满,不能继续写入 
if(m_bFull)  
{ 
  return0; 
} 
else if(m_nReadPos == m_nWritePos)// 缓冲区为空时 
{ 
  /*                          == 内存模型 ==
     (empty)             m_nReadPos                (empty)                      
  |----------------------------------|-----------------------------------------|
         m_nWritePos        m_nBufSize
  */ 
  int leftcount = m_nBufSize - m_nWritePos; 
  if(leftcount > count) 
  { 
   memcpy(m_pBuf + m_nWritePos, buf, count); 
   m_nWritePos += count; 
   m_bFull =(m_nWritePos == m_nReadPos); 
   return count; 
  } 
  else 
  { 
   memcpy(m_pBuf + m_nWritePos, buf, leftcount); 
   m_nWritePos =(m_nReadPos > count - leftcount)? count - leftcount : m_nWritePos; 
   memcpy(m_pBuf, buf + leftcount, m_nWritePos); 
   m_bFull =(m_nWritePos == m_nReadPos); 
   return leftcount + m_nWritePos; 
  } 
}  
else if(m_nReadPos < m_nWritePos)// 有剩余空间可写入 
{ 
  /*                           == 内存模型 ==
   (empty)                 (data)                     (empty)
  |-------------------|----------------------------|---------------------------|
     m_nReadPos                m_nWritePos       (leftcount)           
  */ 
  // 剩余缓冲区大小(从写入位置到缓冲区尾) 
 
  int leftcount = m_nBufSize - m_nWritePos;
  int test = m_nWritePos;
  if(leftcount > count)   // 有足够的剩余空间存放 
  { 
   memcpy(m_pBuf + m_nWritePos, buf, count); 
   m_nWritePos += count; 
   m_bFull =(m_nReadPos == m_nWritePos); 
   assert(m_nReadPos <= m_nBufSize); 
   assert(m_nWritePos <= m_nBufSize);     
   return count; 
  } 
  else       // 剩余空间不足 
  { 
   // 先填充满剩余空间,再回头找空间存放 
   memcpy(m_pBuf + test, buf, leftcount); 
 
   m_nWritePos =(m_nReadPos >= count - leftcount)? count - leftcount : m_nReadPos; 
   memcpy(m_pBuf, buf + leftcount, m_nWritePos); 
   m_bFull =(m_nReadPos == m_nWritePos);    
   assert(m_nReadPos <= m_nBufSize); 
   assert(m_nWritePos <= m_nBufSize);  
   return leftcount + m_nWritePos; 
  } 
} 
else 
{ 
  /*                          == 内存模型 ==
   (unread)                 (read)                     (unread)
  |-------------------|----------------------------|---------------------------|
      m_nWritePos    (leftcount)    m_nReadPos                      
  */ 
  int leftcount = m_nReadPos - m_nWritePos; 
  if(leftcount > count) 
  { 
   // 有足够的剩余空间存放 
   memcpy(m_pBuf + m_nWritePos, buf, count); 
   m_nWritePos += count; 
   m_bFull =(m_nReadPos == m_nWritePos); 
   assert(m_nReadPos <= m_nBufSize); 
   assert(m_nWritePos <= m_nBufSize);  
   return count; 
  } 
  else 
  { 
   // 剩余空间不足时要丢弃后面的数据 
   memcpy(m_pBuf + m_nWritePos, buf, leftcount); 
   m_nWritePos += leftcount; 
   m_bFull =(m_nReadPos == m_nWritePos); 
   assert(m_bFull); 
   assert(m_nReadPos <= m_nBufSize); 
   assert(m_nWritePos <= m_nBufSize);    
   return leftcount; 
  } 
} 
} 
/************************************************************************/ 
/* 从缓冲区读数据,返回实际读取的字节数                                 */ 
/************************************************************************/ 
int CCycleBuffer::read(char* buf,int count) 
{ 
if(count <=0)
  return0; 
m_bFull =false; 
if(m_bEmpty)       // 缓冲区空,不能继续读取数据 
{ 
  return0; 
} 
else if(m_nReadPos == m_nWritePos)   // 缓冲区满时 
{ 
  /*                          == 内存模型 ==
   (data)          m_nReadPos                (data)    
|--------------------------------|--------------------------------------------|
    m_nWritePos         m_nBufSize
  */ 
  int leftcount = m_nBufSize - m_nReadPos; 
  if(leftcount > count) 
  { 
   memcpy(buf, m_pBuf + m_nReadPos, count); 
   m_nReadPos += count; 
   m_bEmpty =(m_nReadPos == m_nWritePos); 
   return count; 
  } 
  else 
  { 
   memcpy(buf, m_pBuf + m_nReadPos, leftcount); 
   m_nReadPos =(m_nWritePos > count - leftcount)? count - leftcount : m_nWritePos; 
   memcpy(buf + leftcount, m_pBuf, m_nReadPos); 
   m_bEmpty =(m_nReadPos == m_nWritePos); 
   return leftcount + m_nReadPos; 
  } 
} 
else if(m_nReadPos < m_nWritePos)   // 写指针在前(未读数据是连接的) 
{ 
  /*                          == 内存模型 ==
   (read)                 (unread)                      (read)    
  |-------------------|----------------------------|---------------------------|
     m_nReadPos                m_nWritePos                     m_nBufSize
  */ 
  int leftcount = m_nWritePos - m_nReadPos; 
  int c =(leftcount > count)? count : leftcount; 
  memcpy(buf, m_pBuf + m_nReadPos, c); 
  m_nReadPos += c; 
  m_bEmpty =(m_nReadPos == m_nWritePos); 
  assert(m_nReadPos <= m_nBufSize); 
  assert(m_nWritePos <= m_nBufSize); 
  return c; 
} 
else          // 读指针在前(未读数据可能是不连接的) 
{ 
  /*                          == 内存模型 ==
     (unread)                (read)                      (unread)
  |-------------------|----------------------------|---------------------------|
      m_nWritePos                  m_nReadPos                  m_nBufSize
 
  */ 
  int leftcount = m_nBufSize - m_nReadPos; 
  if(leftcount > count)   // 未读缓冲区够大,直接读取数据 
  { 
   memcpy(buf, m_pBuf + m_nReadPos, count); 
   m_nReadPos += count; 
   m_bEmpty =(m_nReadPos == m_nWritePos); 
   assert(m_nReadPos <= m_nBufSize); 
   assert(m_nWritePos <= m_nBufSize); 
   return count; 
  } 
  else       // 未读缓冲区不足,需回到缓冲区头开始读 
  { 
   memcpy(buf, m_pBuf + m_nReadPos, leftcount); 
   m_nReadPos =(m_nWritePos >= count - leftcount)? count - leftcount : m_nWritePos; 
   memcpy(buf + leftcount, m_pBuf, m_nReadPos); 
   m_bEmpty =(m_nReadPos == m_nWritePos); 
   assert(m_nReadPos <= m_nBufSize); 
   assert(m_nWritePos <= m_nBufSize); 
   return leftcount + m_nReadPos; 
  }   
} 
} 
/************************************************************************/ 
/* 获取缓冲区有效数据长度                                               */ 
/************************************************************************/ 
int CCycleBuffer::getLength() 
{ 
if(m_bEmpty) 
{ 
  return0; 
} 
else if(m_bFull) 
{ 
  return m_nBufSize; 
} 
else if(m_nReadPos < m_nWritePos) 
{ 
  return m_nWritePos - m_nReadPos; 
} 
else 
{ 
  return m_nBufSize - m_nReadPos + m_nWritePos; 
} 
} 
void CCycleBuffer::empty() 
{ 
m_nReadPos =0; 
m_nWritePos =0; 
m_bEmpty =true; 
m_bFull =false; 
} 
bool CCycleBuffer::isEmpty() 
{ 
return m_bEmpty; 
} 
bool CCycleBuffer::isFull() 
{ 
return m_bFull; 
} 

测试

#include "CCycleBuffer.h"

int main()
{
	CCycleBuffer buffer(100);
	char* datachar = new char[40];
	char* reschar = new char[100];

	int res = 0;

	for (int i = 0; i < 40; ++i)
	{
		datachar[i] = i + 1;
	}
	res = buffer.Write(datachar, 40);
	printf("写入数据个数:%d\n", res);
	res = buffer.Write(datachar, 40);
	printf("写入数据个数:%d\n", res);
	res = buffer.Write(datachar, 40);
	printf("写入数据个数:%d\n", res);
	res = buffer.Write(datachar, 40);
	printf("写入数据个数:%d\n", res);

	res = buffer.Read(reschar, 100);
	printf("读取数据个数:%d\n", res);
	for (int i = 0; i < 100; ++i)
	{
		if (i % 10 == 0)
		{
			printf("\n");
		}
		printf("%2d   ", reschar[i]);
	}

	return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值