ring_buffer.rb

# RingBuffer class: 
# https://github.com/celluloid/celluloid/blob/master/lib/celluloid/logging/ring_buffer.rb
# added the moving average method

class RingBuffer

  attr_reader :buffer

  def initialize(size)
    @size = size
    @start = 0
    @count = 0
    @buffer = Array.new(size)
    @mutex = Mutex.new
    end

    def full?
      @count == @size
    end

    # calculates the mean average of all the values in the buffer provided that
    # the buffer is already full
    def average
      if self.full?
        @buffer.inject{ |sum, el| sum + el }.to_f / @size
      end
    end

    def empty?
      @count == 0
    end

    def push(value)
      @mutex.synchronize do
        stop = (@start + @count) % @size
        @buffer[stop] = value
        if full?
          @start = (@start + 1) % @size
        else
          @count += 1
        end
        value
      end
    end
    alias :<< :push

    def shift
      @mutex.synchronize do
        remove_element
      end
    end

    def flush
      values = []
      @mutex.synchronize do
        while !empty?
          values << remove_element
        end
      end
      values
    end
 def clear
      @buffer = Array.new(@size)
      @start = 0
      @count = 0
    end

    private

    def remove_element
      return nil if empty?
      value, @buffer[@start] = @buffer[@start], nil
      @start = (@start + 1) % @size
      @count -= 1
      value
    end

end