GY-63_MS5611/libraries/GAMMA/gamma.h
2021-01-29 12:31:58 +01:00

122 lines
2.1 KiB
C++

#pragma once
//
// FILE: gamma.h
// AUTHOR: Rob Tillaart
// VERSION: 0.1.1
// DATE: 2020-08-08
// PURPOSE: Arduino Library to efficiently hold a gamma lookup table
// 0.1.0 2020-08-08 initial release
// 0.1.1 2020-12-24 arduino-ci + unit test
#include "Arduino.h"
#define GAMMA_LIB_VERSION (F("0.1.1"))
#if defined(ESP8266) || defined(ESP32)
#include "types.h"
#endif
class GAMMA
{
public:
GAMMA(uint16_t size = 32)
{
_shift = 7;
// force power of 2; get shift & mask right
for (uint16_t s = 2; s < 512; s <<= 1)
{
if (size <= s)
{
_size = s;
break;
}
_shift--;
}
_mask = (1 << _shift) - 1;
_interval = 256 / _size;
_table = (uint8_t *)malloc(_size + 1);
setGamma(2.8);
}
~GAMMA()
{
if (_table) free(_table);
}
void setGamma(float gamma)
{
if (_gamma != gamma)
{
yield(); // keep ESP happy
_gamma = gamma;
for (uint16_t i = 0; i < _size; i++)
{
_table[i] = pow(i * _interval * (1.0 / 255), _gamma) * 255 + 0.5;
}
_table[_size] = 255; // anchor for interpolation..
}
}
float getGamma()
{
return _gamma;
};
uint8_t operator[] (uint8_t idx)
{
if (_interval == 1) return _table[idx];
// else interpolate
uint8_t i = idx >> _shift;
uint8_t m = idx & _mask;
// exact element shortcut
if ( m == 0 ) return _table[i];
// interpolation
uint16_t delta = _table[i+1] - _table[i];
delta = (delta * m + _interval/2) >> _shift; // == /_interval;
return _table[i] + delta;
}
uint16_t size()
{
return _size + 1;
};
int distinct()
{
int last = _table[0];
int cnt = 1;
for (uint16_t i = 1; i < _size; i++)
{
if (_table[i] == last) continue;
last = _table[i];
cnt++;
}
return cnt;
}
void dump()
{
for (uint16_t i = 0; i < _size; i++)
{
Serial.println(_table[i]);
}
}
private:
uint8_t _shift = 0;
uint8_t _mask = 0;
uint16_t _size = 0;
uint8_t _interval = 0;
float _gamma = 0;
uint8_t * _table;
};
// -- END OF FILE --