#include <ungrateful.h>

int main() {
    un_init(UN_MB(4));
    /* 2d */
    {
        float2 a = { 1.0f, 2.0f };
        float2 b = { 3.0f, 4.0f };
        float2 n = { 0.0f, 1.0f };
        float2 v;

        v = un_m_add2f(a, b);
        assert(fabsf(v.x - 4.0f) < EPSILON);
        assert(fabsf(v.y - 6.0f) < EPSILON);

        v = un_m_sub2f(a, b);
        assert(fabsf(v.x + 2.0f) < EPSILON);
        assert(fabsf(v.y + 2.0f) < EPSILON);

        v = un_m_add_scalar2f(a, 1);
        assert(fabsf(v.x - 2.0f) < EPSILON);
        assert(fabsf(v.y - 3.0f) < EPSILON);

        v = un_m_sub_scalar2f(a, 1);
        assert(fabsf(v.x)        < EPSILON);
        assert(fabsf(v.y - 1.0f) < EPSILON);

        v = un_m_mul_scalar2f(a, 2);
        assert(fabsf(v.x - 2.0f) < EPSILON);
        assert(fabsf(v.y - 4.0f) < EPSILON);

        v = un_m_div_scalar2f(a, 2);
        assert(fabsf(v.x - 0.5f) < EPSILON);
        assert(fabsf(v.y - 1.0f) < EPSILON);

        v.x = un_m_dot2f(a, b);
        assert(fabsf(v.x - 11.0f) < EPSILON);

        v = un_m_hadamard2f(a, b);
        assert(fabsf(v.x - 3.0f) < EPSILON);
        assert(fabsf(v.y - 8.0f) < EPSILON);

        v = un_m_project2f(a, b);
        assert(fabsf(v.x - 1.32f) < EPSILON);
        assert(fabsf(v.y - 1.76f) < EPSILON);

        v = un_m_cross2f(a);
        assert(fabsf(v.x - 2.0f) < EPSILON);
        assert(fabsf(v.y + 1.0f) < EPSILON);

        v.x = un_m_magnitude2f(b);
        assert(fabsf(v.x - 5.0f) < EPSILON);

        v.x = un_m_magnitude_sqr2f(b);
        assert(fabsf(v.x - 25.0f) < EPSILON);

        v.x = un_m_distance2f(a, b);
        assert(fabsf(v.x - sqrtf(8.0f)) < EPSILON);

        v.x = un_m_distance_sqr2f(a, b);
        assert(fabsf(v.x - 8.0f) < EPSILON);

        v = un_m_normalize2f(b);
        assert(fabsf(v.x - 0.6f) < EPSILON);
        assert(fabsf(v.y - 0.8f) < EPSILON);

        v = un_m_reflect2f(a, n);
        assert(fabsf(v.x + a.x) < EPSILON);
        assert(fabsf(v.y - a.y) < EPSILON);
    }

    /* 3d */
    {
        float3 a = { 1.0f, 2.0f, 3.0f };
        float3 b = { 3.0f, 4.0f, 5.0f };
        float3 n = { 0.0f, 1.0f, 0.0f };
        float3 v;

        v = un_m_add3f(a, b);
        assert(fabsf(v.x - 4.0f) < EPSILON);
        assert(fabsf(v.y - 6.0f) < EPSILON);
        assert(fabsf(v.z - 8.0f) < EPSILON);

        v = un_m_sub3f(a, b);
        assert(fabsf(v.x + 2.0f) < EPSILON);
        assert(fabsf(v.y + 2.0f) < EPSILON);
        assert(fabsf(v.z + 2.0f) < EPSILON);

        v = un_m_add_scalar3f(a, 1);
        assert(fabsf(v.x - 2.0f) < EPSILON);
        assert(fabsf(v.y - 3.0f) < EPSILON);
        assert(fabsf(v.z - 4.0f) < EPSILON);

        v = un_m_sub_scalar3f(a, 1);
        assert(fabsf(v.x)        < EPSILON);
        assert(fabsf(v.y - 1.0f) < EPSILON);
        assert(fabsf(v.z - 2.0f) < EPSILON);

        v = un_m_mul_scalar3f(a, 2);
        assert(fabsf(v.x - 2.0f)  < EPSILON);
        assert(fabsf(v.y - 4.0f)  < EPSILON);
        assert(fabsf(v.z - 6.0f) < EPSILON);

        v = un_m_div_scalar3f(a, 2);
        assert(fabsf(v.x - 0.5f) < EPSILON);
        assert(fabsf(v.y - 1.0f) < EPSILON);
        assert(fabsf(v.z - 1.5f) < EPSILON);

        v.x = un_m_dot3f(a, b);
        assert(fabsf(v.x - 26.0f) < EPSILON);

        v = un_m_hadamard3f(a, b);
        assert(fabsf(v.x - 3.0f)  < EPSILON);
        assert(fabsf(v.y - 8.0f)  < EPSILON);
        assert(fabsf(v.z - 15.0f) < EPSILON);

        v = un_m_project3f(a, b);
        assert(fabsf(v.x - 1.56f) < EPSILON);
        assert(fabsf(v.y - 2.08f) < EPSILON);
        assert(fabsf(v.z - 2.6f) < EPSILON);

        v = un_m_cross3f(a, b);
        assert(fabsf(v.x + 2.0f) < EPSILON);
        assert(fabsf(v.y - 4.0f) < EPSILON);
        assert(fabsf(v.z + 2.0f) < EPSILON);

        v.x = un_m_magnitude3f(b);
        assert(fabsf(v.x - sqrtf(50.0f)) < EPSILON);

        v.x = un_m_magnitude_sqr3f(b);
        assert(fabsf(v.x - 50.0f) < EPSILON);

        v.x = un_m_distance3f(a, b);
        assert(fabsf(v.x - sqrtf(12.0f)) < EPSILON);

        v.x = un_m_distance_sqr3f(a, b);
        assert(fabsf(v.x - 12.0f) < EPSILON);

        v = un_m_normalize3f(b);
        assert(fabsf(v.x - 0.424264f) < EPSILON);
        assert(fabsf(v.y - 0.565685f) < EPSILON);
        assert(fabsf(v.z - 0.707107f) < EPSILON);

        v = un_m_reflect3f(a, n);
        assert(fabsf(v.x + a.x) < EPSILON);
        assert(fabsf(v.y - a.y) < EPSILON);
        assert(fabsf(v.z + a.z) < EPSILON);
    }

    /* 4d */
    {
        float4 a = {1.0f, 2.0f, 3.0f, 4.0f};
        float4 b = {3.0f, 4.0f, 5.0f, 6.0f};
        float4 n = {0.0f, 1.0f, 0.0f, 0.0f};
        float4 v;

        v = un_m_add4f(a, b);
        assert(fabsf(v.x - 4.0f) < EPSILON);
        assert(fabsf(v.y - 6.0f) < EPSILON);
        assert(fabsf(v.z - 8.0f) < EPSILON);
        assert(fabsf(v.w - 10.0f) < EPSILON);

        v = un_m_sub4f(a, b);
        assert(fabsf(v.x + 2.0f) < EPSILON);
        assert(fabsf(v.y + 2.0f) < EPSILON);
        assert(fabsf(v.z + 2.0f) < EPSILON);
        assert(fabsf(v.w + 2.0f) < EPSILON);

        v = un_m_add_scalar4f(a, 1);
        assert(fabsf(v.x - 2.0f) < EPSILON);
        assert(fabsf(v.y - 3.0f) < EPSILON);
        assert(fabsf(v.z - 4.0f) < EPSILON);
        assert(fabsf(v.w - 5.0f) < EPSILON);

        v = un_m_sub_scalar4f(a, 1);
        assert(fabsf(v.x)        < EPSILON);
        assert(fabsf(v.y - 1.0f) < EPSILON);
        assert(fabsf(v.z - 2.0f) < EPSILON);
        assert(fabsf(v.w - 3.0f) < EPSILON);

        v = un_m_mul_scalar4f(a, 2);
        assert(fabsf(v.x - 2.0f)  < EPSILON);
        assert(fabsf(v.y - 4.0f)  < EPSILON);
        assert(fabsf(v.z - 6.0f) < EPSILON);
        assert(fabsf(v.w - 8.0f) < EPSILON);

        v = un_m_div_scalar4f(a, 2);
        assert(fabsf(v.x - 0.5f) < EPSILON);
        assert(fabsf(v.y - 1.0f) < EPSILON);
        assert(fabsf(v.z - 1.5f) < EPSILON);
        assert(fabsf(v.w - 2.0f) < EPSILON);

        v.x = un_m_dot4f(a, b);
        assert(fabsf(v.x - 50.0f) < EPSILON);

        v = un_m_hadamard4f(a, b);
        assert(fabsf(v.x - 3.0f)  < EPSILON);
        assert(fabsf(v.y - 8.0f)  < EPSILON);
        assert(fabsf(v.z - 15.0f) < EPSILON);
        assert(fabsf(v.w - 24.0f) < EPSILON);

        v = un_m_project4f(a, b);
        assert(fabsf(v.x - 1.744186f) < EPSILON);
        assert(fabsf(v.y - 2.325581f) < EPSILON);
        assert(fabsf(v.z - 2.906976f) < EPSILON);
        assert(fabsf(v.w - 3.488372f) < EPSILON);

        v.x = un_m_magnitude4f(b);
        assert(fabsf(v.x - sqrtf(86.0f)) < EPSILON);

        v.x = un_m_magnitude_sqr4f(b);
        assert(fabsf(v.x - 86.0f) < EPSILON);

        v.x = un_m_distance4f(a, b);
        assert(fabsf(v.x - 4.0f) < EPSILON);

        v.x = un_m_distance_sqr4f(a, b);
        assert(fabsf(v.x - 16.0f) < EPSILON);

        v = un_m_normalize4f(b);
        assert(fabsf(v.x - 0.323498f) < EPSILON);
        assert(fabsf(v.y - 0.431331f) < EPSILON);
        assert(fabsf(v.z - 0.539164f) < EPSILON);
        assert(fabsf(v.w - 0.646997f) < EPSILON);

        v = un_m_reflect4f(a, n);
        assert(fabsf(v.x + a.x) < EPSILON);
        assert(fabsf(v.y - a.y) < EPSILON);
        assert(fabsf(v.z + a.z) < EPSILON);
        assert(fabsf(v.w + a.w) < EPSILON);
    }

    { // Complex
        float2 a = { 1, 2 };
        float2 b = { 3, 4 };
        float2 c = { 1, 0 };
        float2 d = { 0, 1 };
        float2 v;

        v = un_m_complex_addf(a, b);
        assert((v.x - 4.0) < EPSILON);
        assert((v.y - 6.0) < EPSILON);

        v = un_m_complex_subf(a, b);
        assert((v.x + 2.0) < EPSILON);
        assert((v.y + 2.0) < EPSILON);

        v = un_m_complex_mulf(a, b);
        assert((v.x +  5.0) < EPSILON);
        assert((v.y - 10.0) < EPSILON);

        v = un_m_complex_divf(c, d);
        assert(v.x       < EPSILON);
        assert((v.y + 1.0) < EPSILON);

        v.x = un_m_complex_magnitudef(c);
        assert((v.x - 1.0) < EPSILON);
    }

    return 0;
}