#ifndef BitArray_H
#define BitArray_H
//
//    FILE: BitArray.h
//  AUTHOR: Rob dot Tillaart at gmail dot com
// VERSION: 0.1.05
// 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.05"
#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    0x01
#define BA_IDX_RANGE    0x02


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

    byte begin(const byte 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; };
    byte getError()     { return _error; };
    void clear();

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

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

    uint16_t _size = 0;
    uint16_t _bytes = 0;
    uint16_t _bits = 0;
    uint16_t _segments = 0;
    byte * _ar[BA_MAX_SEGMENTS];
    byte _error = BA_NO_MEMORY;
};

#endif