diff options
Diffstat (limited to 'src/ringbuffer.h')
| -rw-r--r-- | src/ringbuffer.h | 100 |
1 files changed, 100 insertions, 0 deletions
diff --git a/src/ringbuffer.h b/src/ringbuffer.h new file mode 100644 index 0000000..9480043 --- /dev/null +++ b/src/ringbuffer.h | |||
| @@ -0,0 +1,100 @@ | |||
| 1 | #ifndef RING_BUFFER_H | ||
| 2 | #define RING_BUFFER_H | ||
| 3 | |||
| 4 | #include <memory> | ||
| 5 | #include "bu/exceptionbase.h" | ||
| 6 | |||
| 7 | namespace Bu | ||
| 8 | { | ||
| 9 | template<typename value, typename valuealloc=std::allocator<value> > | ||
| 10 | class RingBuffer | ||
| 11 | { | ||
| 12 | public: | ||
| 13 | RingBuffer( int nCapacity ) : | ||
| 14 | nCapacity( nCapacity ), | ||
| 15 | nStart( -1 ), | ||
| 16 | nEnd( -2 ) | ||
| 17 | { | ||
| 18 | aData = va.allocate( nCapacity ); | ||
| 19 | for( int j = 0; j < nCapacity; j++ ) | ||
| 20 | { | ||
| 21 | va.construct( &aData[j], value() ); | ||
| 22 | } | ||
| 23 | } | ||
| 24 | |||
| 25 | virtual ~RingBuffer() | ||
| 26 | { | ||
| 27 | for( int j = 0; j < nCapacity; j++ ) | ||
| 28 | { | ||
| 29 | va.destroy( &aData[j] ); | ||
| 30 | } | ||
| 31 | va.deallocate( aData, nCapacity ); | ||
| 32 | } | ||
| 33 | |||
| 34 | int getCapacity() | ||
| 35 | { | ||
| 36 | return nCapacity; | ||
| 37 | } | ||
| 38 | |||
| 39 | bool isFilled() | ||
| 40 | { | ||
| 41 | return (nStart == nEnd); | ||
| 42 | } | ||
| 43 | |||
| 44 | bool isEmpty() | ||
| 45 | { | ||
| 46 | return (nStart == -1); | ||
| 47 | } | ||
| 48 | |||
| 49 | void enqueue( const value &v ) | ||
| 50 | { | ||
| 51 | if( nStart == -1 ) | ||
| 52 | { | ||
| 53 | nStart = 0; | ||
| 54 | nEnd = 1; | ||
| 55 | aData[0] = v; | ||
| 56 | } | ||
| 57 | else if( nStart == nEnd ) | ||
| 58 | { | ||
| 59 | throw ExceptionBase("Hey, it's full!"); | ||
| 60 | } | ||
| 61 | else | ||
| 62 | { | ||
| 63 | aData[nEnd] = v; | ||
| 64 | nEnd = (nEnd+1)%nCapacity; | ||
| 65 | } | ||
| 66 | } | ||
| 67 | |||
| 68 | value dequeue() | ||
| 69 | { | ||
| 70 | if( nStart == -1 ) | ||
| 71 | { | ||
| 72 | throw ExceptionBase("No data"); | ||
| 73 | } | ||
| 74 | else | ||
| 75 | { | ||
| 76 | value &v = aData[nStart]; | ||
| 77 | nStart = (nStart+1)%nCapacity; | ||
| 78 | if( nStart == nEnd ) | ||
| 79 | { | ||
| 80 | nStart = -1; | ||
| 81 | nEnd = -2; | ||
| 82 | } | ||
| 83 | return v; | ||
| 84 | } | ||
| 85 | } | ||
| 86 | |||
| 87 | value &operator[]( int nIndex ) | ||
| 88 | { | ||
| 89 | return aData[(nIndex+nStart)%nCapacity]; | ||
| 90 | } | ||
| 91 | |||
| 92 | private: | ||
| 93 | int nCapacity; | ||
| 94 | value *aData; | ||
| 95 | valuealloc va; | ||
| 96 | int nStart, nEnd; | ||
| 97 | }; | ||
| 98 | } | ||
| 99 | |||
| 100 | #endif | ||
