Home | History | Annotate | Line # | Download | only in DotZLib
      1 //
      2 //  Copyright Henrik Ravn 2004
      3 //
      4 // Use, modification and distribution are subject to the Boost Software License, Version 1.0.
      5 // (See accompanying file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
      6 //
      7 
      8 using System;
      9 using System.Diagnostics;
     10 
     11 namespace DotZLib
     12 {
     13 
     14 	/// <summary>
     15 	/// This class implements a circular buffer
     16 	/// </summary>
     17 	internal class CircularBuffer
     18 	{
     19         #region Private data
     20         private int _capacity;
     21         private int _head;
     22         private int _tail;
     23         private int _size;
     24         private byte[] _buffer;
     25         #endregion
     26 
     27         public CircularBuffer(int capacity)
     28         {
     29             Debug.Assert( capacity > 0 );
     30             _buffer = new byte[capacity];
     31             _capacity = capacity;
     32             _head = 0;
     33             _tail = 0;
     34             _size = 0;
     35         }
     36 
     37         public int Size { get { return _size; } }
     38 
     39         public int Put(byte[] source, int offset, int count)
     40         {
     41             Debug.Assert( count > 0 );
     42             int trueCount = Math.Min(count, _capacity - Size);
     43             for (int i = 0; i < trueCount; ++i)
     44                 _buffer[(_tail+i) % _capacity] = source[offset+i];
     45             _tail += trueCount;
     46             _tail %= _capacity;
     47             _size += trueCount;
     48             return trueCount;
     49         }
     50 
     51         public bool Put(byte b)
     52         {
     53             if (Size == _capacity) // no room
     54                 return false;
     55             _buffer[_tail++] = b;
     56             _tail %= _capacity;
     57             ++_size;
     58             return true;
     59         }
     60 
     61         public int Get(byte[] destination, int offset, int count)
     62         {
     63             int trueCount = Math.Min(count,Size);
     64             for (int i = 0; i < trueCount; ++i)
     65                 destination[offset + i] = _buffer[(_head+i) % _capacity];
     66             _head += trueCount;
     67             _head %= _capacity;
     68             _size -= trueCount;
     69             return trueCount;
     70         }
     71 
     72         public int Get()
     73         {
     74             if (Size == 0)
     75                 return -1;
     76 
     77             int result = (int)_buffer[_head++ % _capacity];
     78             --_size;
     79             return result;
     80         }
     81 
     82     }
     83 }
     84