#ifndef BitArray_H
#define BitArray_H
//
//    FILE: BitArray.h
//  AUTHOR: Rob dot Tillaart at gmail dot com
// VERSION: 0.1.06
// PURPOSE: BitArray library for Arduino
// HISTORY: See BitArray.cpp
//
// Released to the public domain
//
// BitArray allows you to make a compact array of objects with a size
// expressed in bits. typically 1..10.
// The interface uses uint32_t as that will be enough for most purposes.
// The main requirement is to optimize storage space
// the total space may not exceed 256 bytes.
//
// Originally created to store lot of numbers between 1..6 dice rolls
// the storage is also usable to store e.g. raw 10 bit analogReads
// see demo sketches.
//

#if defined(ARDUINO) && ARDUINO >= 100
#include "Arduino.h"
#else
#include "WProgram.h"
#endif

#define BITARRAY_LIB_VERSION "0.1.06"
#define BA_SEGMENT_SIZE 200

// max memory is board type dependant
// note the bitArray does not use all of the RAM
// 1K - max 600
#if defined(__AVR_ATmega168__)
#define BA_MAX_SEGMENTS 3
// 2K - max 1600
#elif defined(__AVR_ATmega328P__)
#define BA_MAX_SEGMENTS 8
// 8K - max 7000
#elif defined(__AVR_ATmega1280__)
#define BA_MAX_SEGMENTS 35
// 8K - max 7000
#elif defined(__AVR_ATmega2560__)
#define BA_MAX_SEGMENTS 35
// 1.25K - max 800
#elif defined(__AVR_ATmega16U4__)
#define BA_MAX_SEGMENTS 4
// 2.5K - max 2000
#elif defined(__AVR_ATmega32U4__)
#define BA_MAX_SEGMENTS 10
// 96K (64 + 32) DUE...
#elif defined(__SAM3X8E__)
#define BA_MAX_SEGMENTS 100
// default max 1000
#else
#define BA_MAX_SEGMENTS 5
#endif

#define BA_ERR              0xFFFFFFFF
#define BA_OK               0x00
#define BA_NO_MEMORY_ERR    0x01
#define BA_IDX_RANGE_ERR    0x02
#define BA_ELEMENT_SIZE_ERR 0x03
#define BA_SIZE_ERR         0x04


class BitArray
{
public:
    BitArray() {};
    ~BitArray();

    uint8_t  begin(const uint8_t bits, const uint16_t size);

    uint16_t capacity() { return _bytes * 8 / _bits; };
    uint16_t memory()   { return _bytes; };
    uint16_t bits()     { return _bits; };
    uint16_t segments() { return _segments; };
    uint8_t  getError() { return _error; };
    void     clear();

    uint32_t get(const uint16_t idx);
    uint32_t set(const uint16_t idx, uint32_t value);

private:
    uint8_t _bitget(const uint16_t idx);
    void    _bitset(const uint16_t idx, const uint8_t value);

    uint16_t _bytes = 0;
    uint8_t  _bits = 0;
    uint8_t  _segments = 0;
    uint8_t *_ar[BA_MAX_SEGMENTS];
    uint8_t  _error = BA_NO_MEMORY_ERR;
};

#endif