//
//    FILE: runningMedianTest1.ino
//  AUTHOR: Rob Tillaart
// PURPOSE: test functionality
//    DATE: 2013-10-28
//     URL: https://github.com/RobTillaart/RunningMedian


#include <RunningMedian.h>


// 50 consecutive samples from Sharp distance sensor model GP2Y0A710K0F while stationary.
const int sourceData[] =
{
  300, 299, 296, 343, 307, 304, 303, 305, 300, 340,
  308, 305, 300, 304, 311, 304, 300, 300, 304, 304,
  284, 319, 306, 304, 300, 302, 305, 310, 306, 304,
  308, 300, 299, 304, 300, 305, 307, 303, 326, 311,
  306, 304, 305, 300, 300, 307, 302, 305, 296, 300
};

const int sourceSize = (sizeof(sourceData)/sizeof(sourceData[0]));

RunningMedian samples = RunningMedian(sourceSize);


void setup()
{
  Serial.begin(115200);
  while (!Serial); // Wait for serial port to connect. Needed for Leonardo + MKR1010.
  delay(1000); // Simply to allow time for the ERW versions of the IDE time to automagically open the Serial Monitor. 1 second chosen arbitrarily.
  Serial.print(F("Running Median Version: "));
  Serial.println(RUNNING_MEDIAN_VERSION);


#ifdef RUNNING_MEDIAN_USE_MALLOC
  Serial.println(F("Dynamic version using malloc() enabled"));
#else
  Serial.print(F("Static version, will always allocate an array of "));
  Serial.print(MEDIAN_MAX_SIZE, DEC);
  Serial.println(F(" floats."));
#endif

  test1();

  Serial.println("\ndone..\n");
}


void loop()
{
}


void test1()
{
  uint32_t start = 0;
  uint32_t stop = 0;
  float result = 0;

  Serial.print(F("Requested median array size = "));
  Serial.println(sourceSize);
  Serial.print(F("      Actual allocated size = "));
  Serial.println(samples.getSize());
  Serial.println();

  // 50 iterations !!
  for (uint8_t i = 0; i <= (sourceSize - 1); i++)
  {
    Serial.print(F("Loop number : "));
    Serial.println(i + 1);

    start = micros();
    samples.add(sourceData[i]);
    stop = micros();
    Serial.print(F("Time to add the next element to the array = "));
    Serial.println(stop - start);


    Serial.println(F("Cumulative source data added:"));
    Serial.print(F("    "));
    for (uint8_t j = 0; j <= i; j++)
    {
      Serial.print(sourceData[j]);
      Serial.print(F(" "));
    }
    Serial.println();


    Serial.println(F("Unsorted accumulated array:"));
    Serial.print(F("    "));
    for (uint8_t j = 0; j < samples.getCount(); j++)
    {
      Serial.print(samples.getElement(j));
      Serial.print(F(" "));
    }
    Serial.println();


    start = micros();
    result = samples.getSortedElement(0);
    stop = micros();
    Serial.print(F("Time to sort array and return element number zero = "));
    Serial.println(stop - start);


    Serial.println(F("Sorted accumulated array:"));
    Serial.print(F("    "));
    for (uint8_t j = 0; j < samples.getCount(); j++)
    {
      Serial.print(samples.getSortedElement(j));
      Serial.print(F(" "));
    }
    Serial.println();


    start = micros();
    result = samples.getMedian();
    stop = micros();
    Serial.print(F("getMedian() result = "));
    Serial.println(result);
    Serial.print(F("Time to execute getMedian() = "));
    Serial.println(stop - start);


    start = micros();
    result = samples.getAverage();
    stop = micros();
    Serial.print(F("getAverage() result = "));
    Serial.println(result);
    Serial.print(F("Time to execute getAverage() = "));
    Serial.println(stop - start);


    Serial.println(F("getAverage(x) results where:"));
    for (uint8_t j = 1; j <= samples.getCount(); j++)
    {
      start = micros();
      result = samples.getAverage(j);
      stop = micros();
      Serial.print(F("  x = "));
      Serial.print(j);
      Serial.print(F(" => "));
      Serial.print(result);
      Serial.print(F(" Time to execute = "));
      Serial.println(stop - start);
    }

    Serial.println(F("predict(x) results where:"));
    for (uint8_t j = 1; j <= (samples.getCount() / 2); j++)
    {
      start = micros();
      result = samples.predict(j);
      stop = micros();
      Serial.print(F("  x = "));
      Serial.print(j);
      Serial.print(F(" => "));
      Serial.print(result);
      Serial.print(F(" Time to execute = "));
      Serial.println(stop - start);
    }

    Serial.println();
    Serial.println();
  }
}


// -- END OF FILE --