#include <stdio.h>
#include <assert.h>
#include <string.h>
#include "../src/types.h"
#include "../src/number_parsing.h"
#include "../src/utils.h"
#include "../src/evaluator.h"
#include "../src/state.h"

void test_evaluation()
{
    printf("== test evaluation == \n");

    struct StateContainer* state = state_init();

    // test int parsing
    int resVal = 0;
    unsigned char resType = 0;

    evaluate(state, "-4", &resVal, &resType);
    assert(resType == TYPE_INT);
    assert(-4 == resVal);

    evaluate(state, "-(4+9)+1", &resVal, &resType);
    assert(resType == TYPE_INT);
    assert(-12 == resVal);

    evaluate(state, "(-(8-9+5))+8", &resVal, &resType);
    assert(resType == TYPE_INT);
    assert(4 == resVal);

    evaluate(state, "2^6", &resVal, &resType);
    assert(resType == TYPE_INT);
    assert(64 == resVal);

    evaluate(state, "3 + 4", &resVal, &resType);
    assert(resType == TYPE_INT);
    assert(7 == resVal);

    evaluate(state, "(2*4)+0+0", &resVal, &resType);
    assert(resType == TYPE_INT);
    assert(8 == resVal);

    evaluate(state, "2.5-(2+0.1)", &resVal, &resType);
    assert(resType == TYPE_FLOAT);
    assert(float_almost_equal(0.4, get_float_from_int_rep(resVal)));

    evaluate(state, "1^0 +  (7*(5 +2))", &resVal, &resType);
    assert(resType == TYPE_INT);
    assert(50 == resVal);

    evaluate(state, "- ( 0.1+ 0.3 )", &resVal, &resType);
    assert(resType == TYPE_FLOAT);
    assert(float_almost_equal(-0.4, get_float_from_int_rep(resVal)));

    evaluate(state, "sqrt(2)-1", &resVal, &resType);
    assert(resType == TYPE_FLOAT);
    assert(float_almost_equal(0.41421, get_float_from_int_rep(resVal)));

    evaluate(state, "(abs((0-1)*2)) + abs(2)", &resVal, &resType);
    assert(resType == TYPE_INT);
    assert(4 == resVal);

    evaluate(state, "exp(2)-1", &resVal, &resType);
    assert(resType == TYPE_FLOAT);
    assert(float_almost_equal(6.389, get_float_from_int_rep(resVal)));
    
    evaluate(state, "(cos(2)^2)+(sin(2)^2)", &resVal, &resType);
    assert(resType == TYPE_FLOAT);
    assert(float_almost_equal(1, get_float_from_int_rep(resVal)));
    
    evaluate(state, "random_int(1, 100)", &resVal, &resType);
    assert(resType == TYPE_INT);
    printf(" - random int: %d \n", resVal);

    evaluate(state, "abs(2)+abs(-2)", &resVal, &resType);
    assert(resType == TYPE_INT);
    assert(4 == resVal);

    // testing function composition is important
    evaluate(state, "abs(abs(-2))", &resVal, &resType);
    assert(resType == TYPE_INT);
    assert(2 == resVal);
}