如何在C中实现循环缓冲区?

时间:2009-05-06 01:52:52

标签: c data-structures circular-buffer

我需要一个固定大小(在创建它时可在运行时选择,而不是编译时)循环缓冲区,它可以容纳任何类型的对象,它需要非常高性能。我不认为会有资源争用问题,因为虽然它是在多任务嵌入式环境中,但它是一个合作的,所以任务本身可以管理它。

我最初的想法是在缓冲区中存储一个简单的结构,它包含类型(简单的枚举/定义)和指向有效负载的void指针,但我希望它尽可能快,所以我愿意接受建议涉及绕过堆。

实际上我很高兴绕过任何标准库的原始速度 - 从我所看到的代码来看,它没有针对CPU进行大量优化:看起来他们只是为{{{{{{{ 1}}等等,没有手工编码的装配。

任何代码或想法将不胜感激。所需的操作是:

  • 创建具有特定大小的缓冲区。
  • 放在尾巴上。
  • 从头上开始。
  • 返回计数。
  • 删除缓冲区。

9 个答案:

答案 0 :(得分:72)

最简单的解决方案是跟踪项目大小和项目数,然后创建适当字节数的缓冲区:

typedef struct circular_buffer
{
    void *buffer;     // data buffer
    void *buffer_end; // end of data buffer
    size_t capacity;  // maximum number of items in the buffer
    size_t count;     // number of items in the buffer
    size_t sz;        // size of each item in the buffer
    void *head;       // pointer to head
    void *tail;       // pointer to tail
} circular_buffer;

void cb_init(circular_buffer *cb, size_t capacity, size_t sz)
{
    cb->buffer = malloc(capacity * sz);
    if(cb->buffer == NULL)
        // handle error
    cb->buffer_end = (char *)cb->buffer + capacity * sz;
    cb->capacity = capacity;
    cb->count = 0;
    cb->sz = sz;
    cb->head = cb->buffer;
    cb->tail = cb->buffer;
}

void cb_free(circular_buffer *cb)
{
    free(cb->buffer);
    // clear out other fields too, just to be safe
}

void cb_push_back(circular_buffer *cb, const void *item)
{
    if(cb->count == cb->capacity){
        // handle error
    }
    memcpy(cb->head, item, cb->sz);
    cb->head = (char*)cb->head + cb->sz;
    if(cb->head == cb->buffer_end)
        cb->head = cb->buffer;
    cb->count++;
}

void cb_pop_front(circular_buffer *cb, void *item)
{
    if(cb->count == 0){
        // handle error
    }
    memcpy(item, cb->tail, cb->sz);
    cb->tail = (char*)cb->tail + cb->sz;
    if(cb->tail == cb->buffer_end)
        cb->tail = cb->buffer;
    cb->count--;
}

答案 1 :(得分:15)

// Note power of two buffer size
#define kNumPointsInMyBuffer 1024 

typedef struct _ringBuffer {
    UInt32 currentIndex;
    UInt32 sizeOfBuffer;
    double data[kNumPointsInMyBuffer];
} ringBuffer;

// Initialize the ring buffer
ringBuffer *myRingBuffer = (ringBuffer *)calloc(1, sizeof(ringBuffer));
myRingBuffer->sizeOfBuffer = kNumPointsInMyBuffer;
myRingBuffer->currentIndex = 0;

// A little function to write into the buffer
// N.B. First argument of writeIntoBuffer() just happens to have the
// same as the one calloc'ed above. It will only point to the same
// space in memory if the calloc'ed pointer is passed to
// writeIntoBuffer() as an arg when the function is called. Consider
// using another name for clarity
void writeIntoBuffer(ringBuffer *myRingBuffer, double *myData, int numsamples) {
    // -1 for our binary modulo in a moment
    int buffLen = myRingBuffer->sizeOfBuffer - 1;
    int lastWrittenSample = myRingBuffer->currentIndex;

    int idx;
    for (int i=0; i < numsamples; ++i) {
        // modulo will automagically wrap around our index
        idx = (i + lastWrittenSample) & buffLen; 
        myRingBuffer->data[idx] = myData[i];
    }

    // Update the current index of our ring buffer.
    myRingBuffer->currentIndex += numsamples;
    myRingBuffer->currentIndex &= myRingBuffer->sizeOfBuffer - 1;
}

只要您的环形缓冲区的长度是2的幂,就会非常快速的二进制“&amp;”操作将为您包装索引。 对于我的应用程序,我正在从麦克风获取的音频环形缓冲区向用户显示一段音频。

我始终确保屏幕上显示的最大音频量远小于环形缓冲区的大小。否则你可能正在从同一个块读取和写入。这可能会给你带来奇怪的显示效果。

答案 2 :(得分:11)

首先,标题。如果使用位整数来保持头部和放大器,则不需要模运算来包装缓冲区。尾部“指针”,并调整它们的大小,使它们完全同步。 IE:4096塞进一个12位无符号int本身就是0,无论如何都不受干扰。即使对于2的幂,消除模运算也会使速度加倍 - 几乎完全相同。

使用带有默认内联的Visual Studio 2010 C ++编译器,在我的第3代i7 Dell XPS 8500上填充和排空任何类型数据元素的4096缓冲区需要52秒,需要52秒,并且需要1/8192秒才能为数据提供服务

我会在main()中重写测试循环,这样他们就不再控制流 - 这应该是由指示缓冲区已满或空的返回值控制的,并且是随之而来的;声明。 IE:填料和沥水器应该能够相互撞击而不会腐败或不稳定。在某些时候,我希望多线程化这段代码,因此这种行为至关重要。

QUEUE_DESC(队列描述符)和初始化函数强制此代码中的所有缓冲区都是2的幂。否则上述方案将不起作用。关于这个问题,请注意QUEUE_DESC不是硬编码的,它使用清单常量(#define BITS_ELE_KNT)来构造它。 (我假设2的幂是足够的灵活性)

为了使缓冲区大小运行时可选,我尝试了不同的方法(此处未显示),并使用USHRTs来解决Head,Tail,EleKnt能够管理FIFO缓冲区[USHRT]的问题。为了避免模运算,我为&amp;&amp;创建了一个掩码。与Head,Tail,但那个面具原来是(EleKnt -1),所以就这样使用它。在安静的机器上使用USHRTS而不是bit ints可以提高性能~15%。英特尔CPU内核总是比它们的总线更快,因此在繁忙的共享机器上,打包数据结构可以让您在其他竞争线程之前加载和执行。权衡。

请注意,缓冲区的实际存储空间是使用calloc()在堆上分配的,并且指针位于结构的基础上,因此struct和指针的地址完全相同。 IE浏览器;不需要将偏移量添加到结构地址以绑定寄存器。

同样,服务于缓冲区的所有变量都在物理上与缓冲区相邻,绑定到同一个结构中,因此编译器可以制作漂亮的汇编语言。你必须杀死内联优化才能看到任何程序集,否则它会被摧毁掉。

为了支持任何数据类型的多态性,我使用了memcpy()而不是赋值。如果您只需要灵活地为每个编译支持一个随机变量类型,那么此代码可以完美地运行。

对于多态性,您只需要知道类型及其存储要求。 DATA_DESC描述符数组提供了一种跟踪放入QUEUE_DESC.pB​​uffer的每个数据的方法,以便可以正确检索它。我只是分配足够的pBuffer内存来保存最大数据类型的所有元素,但是跟踪给定数据在DATA_DESC.dBytes中实际使用的存储量。另一种方法是重新发明堆管理器。

这意味着QUEUE_DESC的UCHAR * pBuffer将有一个并行伴随数组来跟踪数据类型和大小,而pBuffer中的数据存储位置将保持原样。新成员可能是DATA_DESC * pDataDesc,或者可能是DATA_DESC DataDesc [2 ^ BITS_ELE_KNT],如果你能找到一种方法来通过这样的前向引用来击败你的编译器。在这些情况下,Calloc()总是更灵活。

你仍然在Q_Put(),Q_Get中使用memcpy(),但实际复制的字节数将由DATA_DESC.dBytes确定,而不是QUEUE_DESC.EleBytes。对于任何给定的put或get,元素可能都是不同的类型/大小。

我相信这段代码满足速度和缓冲区大小的要求,可以满足6种不同数据类型的要求。我已经以printf()语句的形式留下了许多测试装置,因此您可以满足(或不)您的代码正常工作。随机数生成器演示该代码适用于任何随机头/尾组合。

enter code here
// Queue_Small.cpp : Defines the entry point for the console application.
//
#include "stdafx.h"
#include <stdio.h>
#include <time.h>
#include <limits.h>
#include <stdlib.h>
#include <malloc.h>
#include <memory.h>
#include <math.h>

#define UCHAR unsigned char
#define ULONG unsigned long
#define USHRT unsigned short
#define dbl   double
/* Queue structure */
#define QUEUE_FULL_FLAG 1
#define QUEUE_EMPTY_FLAG -1
#define QUEUE_OK 0
//  
#define BITS_ELE_KNT    12  //12 bits will create 4.096 elements numbered 0-4095
//
//typedef struct    {
//  USHRT dBytes:8;     //amount of QUEUE_DESC.EleBytes storage used by datatype
//  USHRT dType :3; //supports 8 possible data types (0-7)
//  USHRT dFoo  :5; //unused bits of the unsigned short host's storage
// }    DATA_DESC;
//  This descriptor gives a home to all the housekeeping variables
typedef struct  {
    UCHAR   *pBuffer;   //  pointer to storage, 16 to 4096 elements
    ULONG Tail  :BITS_ELE_KNT;  //  # elements, with range of 0-4095
    ULONG Head  :BITS_ELE_KNT;  //  # elements, with range of 0-4095
    ULONG EleBytes  :8;     //  sizeof(elements) with range of 0-256 bytes
    // some unused bits will be left over if BITS_ELE_KNT < 12
    USHRT EleKnt    :BITS_ELE_KNT +1;// 1 extra bit for # elements (1-4096)
    //USHRT Flags   :(8*sizeof(USHRT) - BITS_ELE_KNT +1);   //  flags you can use
    USHRT   IsFull  :1;     // queue is full
    USHRT   IsEmpty :1;     // queue is empty
    USHRT   Unused  :1;     // 16th bit of USHRT
}   QUEUE_DESC;

//  ---------------------------------------------------------------------------
//  Function prototypes
QUEUE_DESC *Q_Init(QUEUE_DESC *Q, int BitsForEleKnt, int DataTypeSz);
int Q_Put(QUEUE_DESC *Q, UCHAR *pNew);
int Q_Get(UCHAR *pOld, QUEUE_DESC *Q);
//  ---------------------------------------------------------------------------
QUEUE_DESC *Q_Init(QUEUE_DESC *Q, int BitsForEleKnt, int DataTypeSz)    {
    memset((void *)Q, 0, sizeof(QUEUE_DESC));//init flags and bit integers to zero
    //select buffer size from powers of 2 to receive modulo 
    //                arithmetic benefit of bit uints overflowing
    Q->EleKnt   =   (USHRT)pow(2.0, BitsForEleKnt);
    Q->EleBytes =   DataTypeSz; // how much storage for each element?
    //  Randomly generated head, tail a test fixture only. 
    //      Demonstrates that the queue can be entered at a random point 
    //      and still perform properly. Normally zero
    srand(unsigned(time(NULL)));    // seed random number generator with current time
    Q->Head = Q->Tail = rand(); // supposed to be set to zero here, or by memset
    Q->Head = Q->Tail = 0;
    //  allocate queue's storage
    if(NULL == (Q->pBuffer = (UCHAR *)calloc(Q->EleKnt, Q->EleBytes)))  {
        return NULL;
    }   else    {
        return Q;
    }
}
//  ---------------------------------------------------------------------------
int Q_Put(QUEUE_DESC *Q, UCHAR *pNew)   
{
    memcpy(Q->pBuffer + (Q->Tail * Q->EleBytes), pNew, Q->EleBytes);
    if(Q->Tail == (Q->Head + Q->EleKnt)) {
        //  Q->IsFull = 1;
        Q->Tail += 1;   
        return QUEUE_FULL_FLAG; //  queue is full
    }
    Q->Tail += 1;   //  the unsigned bit int MUST wrap around, just like modulo
    return QUEUE_OK; // No errors
}
//  ---------------------------------------------------------------------------
int Q_Get(UCHAR *pOld, QUEUE_DESC *Q)   
{
    memcpy(pOld, Q->pBuffer + (Q->Head * Q->EleBytes), Q->EleBytes);
    Q->Head += 1;   //  the bit int MUST wrap around, just like modulo

    if(Q->Head == Q->Tail)      {
        //  Q->IsEmpty = 1;
        return QUEUE_EMPTY_FLAG; // queue Empty - nothing to get
    }
    return QUEUE_OK; // No errors
}
//
//  ---------------------------------------------------------------------------
int _tmain(int argc, _TCHAR* argv[])    {
//  constrain buffer size to some power of 2 to force faux modulo arithmetic
    int LoopKnt = 1000000;  //  for benchmarking purposes only
    int k, i=0, Qview=0;
    time_t start;
    QUEUE_DESC Queue, *Q;
    if(NULL == (Q = Q_Init(&Queue, BITS_ELE_KNT, sizeof(int)))) {
        printf("\nProgram failed to initialize. Aborting.\n\n");
        return 0;
    }

    start = clock();
    for(k=0; k<LoopKnt; k++)    {
        //printf("\n\n Fill'er up please...\n");
        //Q->Head = Q->Tail = rand();
        for(i=1; i<= Q->EleKnt; i++)    {
            Qview = i*i;
            if(QUEUE_FULL_FLAG == Q_Put(Q, (UCHAR *)&Qview))    {
                //printf("\nQueue is full at %i \n", i);
                //printf("\nQueue value of %i should be %i squared", Qview, i);
                break;
            }
            //printf("\nQueue value of %i should be %i squared", Qview, i);
        }
        //  Get data from queue until completely drained (empty)
        //
        //printf("\n\n Step into the lab, and see what's on the slab... \n");
        Qview = 0;
        for(i=1; i; i++)    {
            if(QUEUE_EMPTY_FLAG == Q_Get((UCHAR *)&Qview, Q))   {
                //printf("\nQueue value of %i should be %i squared", Qview, i);
                //printf("\nQueue is empty at %i", i);
                break;
            }
            //printf("\nQueue value of %i should be %i squared", Qview, i);
        }
        //printf("\nQueue head value is %i, tail is %i\n", Q->Head, Q->Tail);
    }
    printf("\nQueue time was %5.3f to fill & drain %i element queue  %i times \n", 
                     (dbl)(clock()-start)/(dbl)CLOCKS_PER_SEC,Q->EleKnt, LoopKnt);
    printf("\nQueue head value is %i, tail is %i\n", Q->Head, Q->Tail);
    getchar();
    return 0;
}

答案 3 :(得分:9)

您可以枚举编码缓冲区时所需的类型,还是需要能够通过动态调用在运行时添加类型?如果是前者,那么我将创建缓冲区作为n个结构的堆分配数组,其中每个结构由两个元素组成:标识数据类型的枚举标记和所有数据类型的并集。你在小元素的额外存储方面失去了什么,你可以弥补不必处理分配/释放以及由此产生的内存碎片。然后你只需要跟踪定义缓冲区头部和尾部元素的开始和结束索引,并确保在递增/递减索引时计算mod n。

答案 4 :(得分:8)

这是C中的一个简单解决方案。假设每个功能都关闭中断。 没有多态性&amp;东西,只是常识。


#define BUFSIZE 128
char buf[BUFSIZE];
char *pIn, *pOut, *pEnd;
char full;

// init
void buf_init()
{
    pIn = pOut = buf;       // init to any slot in buffer
    pEnd = &buf[BUFSIZE];   // past last valid slot in buffer
    full = 0;               // buffer is empty
}

// add char 'c' to buffer
int buf_put(char c)
{
    if (pIn == pOut  &&  full)
        return 0;           // buffer overrun

    *pIn++ = c;             // insert c into buffer
    if (pIn >= pEnd)        // end of circular buffer?
        pIn = buf;          // wrap around

    if (pIn == pOut)        // did we run into the output ptr?
        full = 1;           // can't add any more data into buffer
    return 1;               // all OK
}

// get a char from circular buffer
int buf_get(char *pc)
{
    if (pIn == pOut  &&  !full)
        return 0;           // buffer empty  FAIL

    *pc = *pOut++;              // pick up next char to be returned
    if (pOut >= pEnd)       // end of circular buffer?
        pOut = buf;         // wrap around

    full = 0;               // there is at least 1 slot
    return 1;               // *pc has the data to be returned
}

答案 5 :(得分:2)

一个简单的实现可以包括:

  • 一个缓冲区,实现为大小为n的数组,无论您需要什么类型
  • 读取指针或索引(对处理器而言更有效)
  • 写指针或索引
  • 一个计数器,指示缓冲区中有多少数据(可从读取和写入指针中导出,但单独跟踪它的速度更快)

每次写入数据时,都会使写指针前进并递增计数器。读取数据时,增加读指针并减少计数器。如果任一指​​针达到n,则将其设置为零。

如果counter = n,你不能写。如果counter = 0,则无法读取。

答案 6 :(得分:2)

C风格,整数的简单环形缓冲区。首先使用init而不是使用put和get。如果缓冲区不包含任何数据,则返回&#34; 0&#34;零。

//=====================================
// ring buffer address based
//=====================================
#define cRingBufCount   512
int     sRingBuf[cRingBufCount];    // Ring Buffer
int     sRingBufPut;                // Input index address
int     sRingBufGet;                // Output index address
Bool    sRingOverWrite;

void    GetRingBufCount(void)
{
int     r;
`       r= sRingBufPut - sRingBufGet;
        if ( r < cRingBufCount ) r+= cRingBufCount;
        return r; 
}

void    InitRingBuffer(void)
{
        sRingBufPut= 0;
        sRingBufGet= 0;
}       

void    PutRingBuffer(int d)
{
        sRingBuffer[sRingBufPut]= d;
        if (sRingBufPut==sRingBufGet)// both address are like ziro
        {
            sRingBufPut= IncRingBufferPointer(sRingBufPut);
            sRingBufGet= IncRingBufferPointer(sRingBufGet);
        }
        else //Put over write a data
        {
            sRingBufPut= IncRingBufferPointer(sRingBufPut);
            if (sRingBufPut==sRingBufGet)
            {
                sRingOverWrite= Ture;
                sRingBufGet= IncRingBufferPointer(sRingBufGet);
            }
        }
}

int     GetRingBuffer(void)
{
int     r;
        if (sRingBufGet==sRingBufPut) return 0;
        r= sRingBuf[sRingBufGet];
        sRingBufGet= IncRingBufferPointer(sRingBufGet);
        sRingOverWrite=False;
        return r;
}

int     IncRingBufferPointer(int a)
{
        a+= 1;
        if (a>= cRingBufCount) a= 0;
        return a;
}

答案 7 :(得分:0)

扩展adam-rosenfield的解决方案,我认为以下内容将适用于多线程单生产者-单消费者方案。

int cb_push_back(circular_buffer *cb, const void *item)
{
  void *new_head = (char *)cb->head + cb->sz;
  if (new_head == cb>buffer_end) {
      new_head = cb->buffer;
  }
  if (new_head == cb->tail) {
    return 1;
  }
  memcpy(cb->head, item, cb->sz);
  cb->head = new_head;
  return 0;
}

int cb_pop_front(circular_buffer *cb, void *item)
{
  void *new_tail = cb->tail + cb->sz;
  if (cb->head == cb->tail) {
    return 1;
  }
  memcpy(item, cb->tail, cb->sz);
  if (new_tail == cb->buffer_end) {
    new_tail = cb->buffer;
  }
  cb->tail = new_tail;
  return 0;
}

答案 8 :(得分:0)

@Adam Rosenfield 的 solution 虽然是正确的,但可以使用不涉及 circular_buffercount 的更轻量级的 capacity 结构来实现。

该结构只能容纳以下 4 个指针:

  • buffer:指向内存中缓冲区的开始。
  • buffer_end:指向内存中缓冲区的末尾。
  • head:指向存储数据的结尾。
  • tail:指向存储数据的开始。

我们可以保留 sz 属性以允许对存储单元进行参数化。

countcapacity 值都应该可以使用上述指针派生。

容量

capacity 是直截了当的,因为它可以通过将 buffer_end 指针和 buffer 指针之间的距离除以存储单位 sz(下面的片段是伪代码):

capacity = (buffer_end - buffer) / sz

计数

不过,对于计数,事情变得有点复杂。例如,在headtail指向同一个位置的情况下,无法判断缓冲区是空还是满。

为了解决这个问题,缓冲区应该为额外的元素分配内存。例如,如果我们的循环缓冲区所需的容量是 10 * sz,那么我们需要分配 11 * sz

容量公式将变为(以下代码段为伪代码):

capacity_bytes = buffer_end - buffer - sz
capacity = capacity_bytes / sz

这个额外的元素语义允许我们构建条件来评估缓冲区是空还是满。

空状态条件

为了使缓冲区为空,head 指针指向与 tail 指针相同的位置:

head == tail

如果上述结果为真,则缓冲区为空。

全状态条件

为了使缓冲区已满,head 指针应位于 tail 指针后面的 1 个元素。因此,从head位置跳转到tail位置所需覆盖的空间应该等于1 * sz

如果 tailhead 大:

tail - head == sz

如果上述结果为真,则缓冲区已满。

如果 headtail 大:

  1. buffer_end - head 返回从 head 跳转到缓冲区末尾的空间。
  2. tail - buffer 返回从缓冲区的开头跳转到 `tail 所需的空间。
  3. 加上上面的2应该等于从head跳转到tail所需的空间
  4. 第 3 步得出的空间,不得超过 1 * sz
(buffer_end - head) + (tail - buffer) == sz
=> buffer_end - buffer - head + tail == sz
=> buffer_end - buffer - sz == head - tail
=> head - tail == buffer_end - buffer - sz
=> head - tail == capacity_bytes

如果上述结果为真,则缓冲区已满。

实践中

修改@Adam Rosenfield's 以使用上述 circular_buffer 结构:

#include <string.h>

#define CB_SUCCESS 0        /* CB operation was successful */
#define CB_MEMORY_ERROR 1   /* Failed to allocate memory */
#define CB_OVERFLOW_ERROR 2 /* CB is full. Cannot push more items. */
#define CB_EMPTY_ERROR 3    /* CB is empty. Cannot pop more items. */

typedef struct circular_buffer {
  void *buffer;
  void *buffer_end;
  size_t sz;
  void *head;
  void *tail;
} circular_buffer;

int cb_init(circular_buffer *cb, size_t capacity, size_t sz) {
  const int incremented_capacity = capacity + 1; // Add extra element to evaluate count
  cb->buffer = malloc(incremented_capacity * sz);
  if (cb->buffer == NULL)
    return CB_MEMORY_ERROR;
  cb->buffer_end = (char *)cb->buffer + incremented_capacity * sz;
  cb->sz = sz;
  cb->head = cb->buffer;
  cb->tail = cb->buffer;
  return CB_SUCCESS;
}

int cb_free(circular_buffer *cb) {
  free(cb->buffer);
  return CB_SUCCESS;
}

const int _cb_length(circular_buffer *cb) {
  return (char *)cb->buffer_end - (char *)cb->buffer;
}

int cb_push_back(circular_buffer *cb, const void *item) {
  const int buffer_length = _cb_length(cb);
  const int capacity_length = buffer_length - cb->sz;

  if ((char *)cb->tail - (char *)cb->head == cb->sz ||
      (char *)cb->head - (char *)cb->tail == capacity_length)
    return CB_OVERFLOW_ERROR;

  memcpy(cb->head, item, cb->sz);

  cb->head = (char*)cb->head + cb->sz;
  if(cb->head == cb->buffer_end)
    cb->head = cb->buffer;

  return CB_SUCCESS;
}

int cb_pop_front(circular_buffer *cb, void *item) {
  if (cb->head == cb->tail)
    return CB_EMPTY_ERROR;

  memcpy(item, cb->tail, cb->sz);

  cb->tail = (char*)cb->tail + cb->sz;
  if(cb->tail == cb->buffer_end)
    cb->tail = cb->buffer;

  return CB_SUCCESS;
}