#include <stdio.h>
#define LOG(args...) printf(args)
#include "../lib/scan/scan.cpp"
#include <unity.h>

struct TestScan : Scan
{
    TestScan(float *ctx, int sz) : ctx(ctx), sz(sz), idx(0) {}

    float *ctx;
    int sz;
    int idx;
};

float getRSSI(void *param)
{
    TestScan *r = (TestScan *)param;

    if (r->idx >= r->sz)
    {
        return -1000000;
    }

    return r->ctx[r->idx++];
}

constexpr int test_sz = 13;
constexpr int inputs_sz = 13;

void test_rssi(void)
{
    uint16_t samples[test_sz];
    float inputs[inputs_sz] = {-40.0, -100.0, -200.0, -50.0, -400.0, -60.0, -20,
                               -75.5, -70,    -80,    -90,   -55.9,  -110};

    TestScan t = TestScan(inputs, inputs_sz);

    uint16_t r = t.rssiMethod(getRSSI, &t, inputs_sz, samples, test_sz);

    uint16_t expect[test_sz] = {20, 50, 55, 60, 0, 70, 75, 80, 0, 90, 0, 100, 110};

    TEST_ASSERT_EQUAL_INT16(20, r);
    TEST_ASSERT_EQUAL_INT16_ARRAY(expect, samples, test_sz);
}

void test_detect()
{
    uint16_t samples[test_sz] = {20, 50, 55, 60, 0, 70, 75, 80, 0, 90, 0, 100, 110};
    bool result[test_sz];

    TestScan test_scan({}, 0);
    Event e = test_scan.detect(samples, result, test_sz, 1);
    size_t r = e.detected.detected_at;

    bool expect[test_sz] = {1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 0, 1, 1};

    TEST_ASSERT_EQUAL_INT16(0, r);
    TEST_ASSERT_EQUAL_INT8_ARRAY(expect, result, test_sz);

    Event e2 = test_scan.detect(samples, result, test_sz, 2);
    r = e2.detected.detected_at;

    bool expect2[test_sz] = {0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0};

    TEST_ASSERT_EQUAL_INT16(1, r);
    TEST_ASSERT_EQUAL_INT8_ARRAY(expect2, result, test_sz);
}