GY-63_MS5611/libraries/TOPMAX/TOPMAX.cpp

//
//    FILE: TOPMAX.cpp
//  AUTHOR: Rob Tillaart
// VERSION: 0.1.0
//    DATE: 2023-05-18
// PURPOSE: Arduino library to track top n maxima.
//     URL: https://github.com/RobTillaart/TOPMAX


#include "TOPMAX.h"


TOPMAX::TOPMAX(uint8_t size)
{
  _size = size;
  if (_size < 3) _size = 3;
  _count = 0;
  _arr   = (float *) malloc(_size * sizeof(float));
}


TOPMAX::~TOPMAX()
{
  if (_arr) free(_arr);
}


uint8_t TOPMAX::count()
{
  return _count; 
}


uint8_t TOPMAX::size()
{
  return _size; 
}


void TOPMAX::reset()
{
  _count = 0;
}


bool TOPMAX::add(float value)
{
  //  initial
  if (_count == 0)
  {
    _arr[_count] = value;
    _count++;
    return true;
  }

  //  not all elements filled.
  if (_count < _size)
  {
    int i = _count - 1;
    while ((i >= 0) && (value < _arr[i]))
    {
      _arr[i + 1] = _arr[i];
      i--;
    }
    _arr[i + 1] = value;
    _count++;
    return true;
  }

  //  too small (or equal)
  if (value <= _arr[0]) return false;

  //  insert
  int i = 1;
  while ((i < _size) && (value > _arr[i]))
  {
    _arr[i - 1] = _arr[i];
    i++;
  }
  _arr[i-1] = value;
  return true;
}


float TOPMAX::get(uint8_t index)
{
  if (index > _count) return NAN;
  return _arr[index];
}


void TOPMAX::fill(float value)
{
  for (int i = 0; i < _size; i++)
  {
    _arr[i] = value;
  }
  _count = _size;
}


//  -- END OF FILE --