提交 ee546fb0 编写于 作者: B bernard.xiong@gmail.com

change rt_memcpy to memcpy in ringbuffer implementation.

git-svn-id: https://rt-thread.googlecode.com/svn/trunk@2199 bbd45198-f89e-11dd-88c7-29a3b14d5316
上级 e0402669
...@@ -48,6 +48,7 @@ rt_size_t rt_ringbuffer_putchar(struct rt_ringbuffer* rb, ...@@ -48,6 +48,7 @@ rt_size_t rt_ringbuffer_putchar(struct rt_ringbuffer* rb,
rt_size_t rt_ringbuffer_get(struct rt_ringbuffer* rb, rt_size_t rt_ringbuffer_get(struct rt_ringbuffer* rb,
rt_uint8_t *ptr, rt_uint8_t *ptr,
rt_uint16_t length); rt_uint16_t length);
rt_size_t rt_ringbuffer_getchar(struct rt_ringbuffer* rb, rt_uint8_t *ch);
rt_size_t rt_ringbuffer_available_size(struct rt_ringbuffer* rb); rt_size_t rt_ringbuffer_available_size(struct rt_ringbuffer* rb);
rt_size_t rt_ringbuffer_emptry_size(struct rt_ringbuffer* rb); rt_size_t rt_ringbuffer_emptry_size(struct rt_ringbuffer* rb);
......
...@@ -16,7 +16,7 @@ ...@@ -16,7 +16,7 @@
#include <rthw.h> #include <rthw.h>
#include <rtthread.h> #include <rtthread.h>
#include <drivers/serial.h> #include <rtdevice.h>
rt_inline void serial_ringbuffer_init(struct serial_ringbuffer* rbuffer) rt_inline void serial_ringbuffer_init(struct serial_ringbuffer* rbuffer)
{ {
......
#include <rtthread.h> #include <rtthread.h>
#include <rtdevice.h> #include <rtdevice.h>
#include <string.h>
void rt_ringbuffer_init(struct rt_ringbuffer* rb, rt_uint8_t *pool, rt_uint16_t size) void rt_ringbuffer_init(struct rt_ringbuffer* rb, rt_uint8_t *pool, rt_uint16_t size)
{ {
...@@ -10,47 +11,45 @@ void rt_ringbuffer_init(struct rt_ringbuffer* rb, rt_uint8_t *pool, rt_uint16_t ...@@ -10,47 +11,45 @@ void rt_ringbuffer_init(struct rt_ringbuffer* rb, rt_uint8_t *pool, rt_uint16_t
/* set buffer pool and size */ /* set buffer pool and size */
rb->buffer_ptr = pool; rb->buffer_ptr = pool;
rb->buffer_size = size; rb->buffer_size = RT_ALIGN_DOWN(size, RT_ALIGN_SIZE);
} }
rt_size_t rt_ringbuffer_put(struct rt_ringbuffer* rb, const rt_uint8_t *ptr, rt_uint16_t length) rt_size_t rt_ringbuffer_put(struct rt_ringbuffer* rb, const rt_uint8_t *ptr, rt_uint16_t length)
{ {
rt_size_t size; rt_uint16_t size;
rt_uint16_t mask;
RT_ASSERT(rb != RT_NULL); RT_ASSERT(rb != RT_NULL);
mask = rb->buffer_size - 1;
/* whether has enough space */ /* whether has enough space */
if (rb->read_index > rb->write_index) size = rb->buffer_size - ((rb->write_index - rb->read_index) & mask);
size = rb->read_index - rb->write_index;
else
size = rb->buffer_size - rb->write_index + rb->read_index;
/* no space */ /* no space */
if (size == 0) return 0; if (size == 0) return 0;
/* drop some data */ /* drop some data */
if (size < length) length = size; if (size < length) length = size;
if (rb->read_index > rb->write_index) if (rb->read_index > rb->write_index)
{ {
/* read_index - write_index = empty space */ /* read_index - write_index = empty space */
rt_memcpy(&rb->buffer_ptr[rb->write_index], ptr, length); memcpy(&rb->buffer_ptr[rb->write_index], ptr, length);
rb->write_index += length; rb->write_index += length;
} }
else else
{ {
if (rb->buffer_size - rb->write_index > length) if (rb->buffer_size - rb->write_index >= length)
{ {
/* there is enough space after write_index */ /* there is enough space after write_index */
rt_memcpy(&rb->buffer_ptr[rb->write_index], ptr, length); memcpy(&rb->buffer_ptr[rb->write_index], ptr, length);
rb->write_index += length; rb->write_index = (rb->write_index + length) & mask;
} }
else else
{ {
rt_memcpy(&rb->buffer_ptr[rb->write_index], ptr, memcpy(&rb->buffer_ptr[rb->write_index], ptr,
rb->buffer_size - rb->write_index); rb->buffer_size - rb->write_index);
rt_memcpy(&rb->buffer_ptr[0], &ptr[rb->buffer_size - rb->write_index], memcpy(&rb->buffer_ptr[0], &ptr[rb->buffer_size - rb->write_index],
length - (rb->buffer_size - rb->write_index)); length - (rb->buffer_size - rb->write_index));
rb->write_index = length - (rb->buffer_size - rb->write_index); rb->write_index = length - (rb->buffer_size - rb->write_index);
} }
} }
...@@ -64,11 +63,12 @@ rt_size_t rt_ringbuffer_put(struct rt_ringbuffer* rb, const rt_uint8_t *ptr, rt_ ...@@ -64,11 +63,12 @@ rt_size_t rt_ringbuffer_put(struct rt_ringbuffer* rb, const rt_uint8_t *ptr, rt_
rt_size_t rt_ringbuffer_putchar(struct rt_ringbuffer* rb, const rt_uint8_t ch) rt_size_t rt_ringbuffer_putchar(struct rt_ringbuffer* rb, const rt_uint8_t ch)
{ {
rt_uint16_t next; rt_uint16_t next;
rt_uint16_t mask;
RT_ASSERT(rb != RT_NULL); RT_ASSERT(rb != RT_NULL);
/* whether has enough space */ /* whether has enough space */
next = rb->write_index + 1; mask = rb->buffer_size - 1;
if (next >= rb->buffer_size) next = 0; next = (rb->write_index + 1) & mask;
if (next == rb->read_index) return 0; if (next == rb->read_index) return 0;
...@@ -85,59 +85,78 @@ rt_size_t rt_ringbuffer_putchar(struct rt_ringbuffer* rb, const rt_uint8_t ch) ...@@ -85,59 +85,78 @@ rt_size_t rt_ringbuffer_putchar(struct rt_ringbuffer* rb, const rt_uint8_t ch)
rt_size_t rt_ringbuffer_get(struct rt_ringbuffer* rb, rt_uint8_t *ptr, rt_uint16_t length) rt_size_t rt_ringbuffer_get(struct rt_ringbuffer* rb, rt_uint8_t *ptr, rt_uint16_t length)
{ {
rt_size_t size; rt_size_t size;
rt_uint16_t mask;
RT_ASSERT(rb != RT_NULL); RT_ASSERT(rb != RT_NULL);
/* whether has enough data */ /* whether has enough data */
if (rb->read_index > rb->write_index) mask = rb->buffer_size - 1;
size = rb->buffer_size - rb->read_index + rb->write_index; size = (rb->write_index - rb->read_index) & mask;
else
size = rb->write_index - rb->read_index;
/* no data */ /* no data */
if (size == 0) return 0; if (size == 0) return 0;
/* less data */ /* less data */
if (size < length) length = size; if (size < length) length = size;
if (rb->read_index > rb->write_index) if (rb->read_index > rb->write_index)
{ {
if (rb->buffer_size - rb->read_index > length) if (rb->buffer_size - rb->read_index >= length)
{ {
/* copy directly */ /* copy directly */
rt_memcpy(ptr, &rb->buffer_ptr[rb->read_index], length); memcpy(ptr, &rb->buffer_ptr[rb->read_index], length);
rb->read_index += length; rb->read_index = (rb->read_index + length) & mask;
} }
else else
{ {
/* copy first and second */ /* copy first and second */
rt_memcpy(ptr, &rb->buffer_ptr[rb->read_index], memcpy(ptr, &rb->buffer_ptr[rb->read_index],
rb->buffer_size - rb->read_index); rb->buffer_size - rb->read_index);
rt_memcpy(&ptr[rb->buffer_size - rb->read_index], &rb->buffer_ptr[0], memcpy(&ptr[rb->buffer_size - rb->read_index], &rb->buffer_ptr[0],
length - rb->buffer_size + rb->read_index); length - rb->buffer_size + rb->read_index);
rb->read_index = length - rb->buffer_size + rb->read_index; rb->read_index = length - rb->buffer_size + rb->read_index;
} }
} }
else else
{ {
rt_memcpy(ptr, &rb->buffer_ptr[rb->read_index], length); memcpy(ptr, &rb->buffer_ptr[rb->read_index], length);
rb->read_index += length; rb->read_index += length;
} }
return length; return length;
} }
rt_size_t rt_ringbuffer_getchar(struct rt_ringbuffer* rb, rt_uint8_t *ch)
{
rt_uint16_t next;
rt_uint16_t mask;
RT_ASSERT(rb != RT_NULL);
/* ringbuffer is empty */
if (rb->read_index == rb->write_index) return 0;
/* whether has data */
mask = rb->buffer_size - 1;
next = (rb->read_index + 1) & mask;
/* put character */
*ch = rb->buffer_ptr[rb->read_index];
rb->read_index = next;
return 1;
}
/** /**
* get available data size * get available data size
*/ */
rt_size_t rt_ringbuffer_available_size(struct rt_ringbuffer* rb) rt_size_t rt_ringbuffer_available_size(struct rt_ringbuffer* rb)
{ {
rt_size_t size; rt_size_t size;
rt_uint16_t mask;
RT_ASSERT(rb != RT_NULL); RT_ASSERT(rb != RT_NULL);
if (rb->read_index > rb->write_index)
size = rb->buffer_size - rb->read_index + rb->write_index; mask = rb->buffer_size - 1;
else size = (rb->write_index - rb->read_index) & mask;
size = rb->write_index - rb->read_index;
/* return the available size */ /* return the available size */
return size; return size;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册