#include <ungrateful.h>

int main() {
    /* 2d */
    {
        real a[2] = {1.0, 2.0};
        real b[2] = {3.0, 4.0};
        real n[2] = {0.0, 1.0};
        real v[2];

        un_m_add2r(v, a, b);
        assert(fabs(v[0] - 4.0) < EPSILON);
        assert(fabs(v[1] - 6.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_sub2r(v, a, b);
        assert(fabs(v[0] + 2.0) < EPSILON);
        assert(fabs(v[1] + 2.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_add_scalar2r(v, a, 1);
        assert(fabs(v[0] - 2.0) < EPSILON);
        assert(fabs(v[1] - 3.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_sub_scalar2r(v, a, 1);
        assert(fabs(v[0])        < EPSILON);
        assert(fabs(v[1] - 1.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_mul_scalar2r(v, a, 2);
        assert(fabs(v[0] - 2.0) < EPSILON);
        assert(fabs(v[1] - 4.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_div_scalar2r(v, a, 2);
        assert(fabs(v[0] - 0.5) < EPSILON);
        assert(fabs(v[1] - 1.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_dot2r(v, a, b);
        assert(fabs(v[0] - 11.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_hadamard2r(v, a, b);
        assert(fabs(v[0] - 3.0) < EPSILON);
        assert(fabs(v[1] - 8.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_project2r(v, a, b);
        assert(fabs(v[0] - 1.32) < EPSILON);
        assert(fabs(v[1] - 1.76) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_cross2r(v, a);
        assert(fabs(v[0] - 2.0) < EPSILON);
        assert(fabs(v[1] + 1.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_magnitude2r(v, b);
        assert(fabs(v[0] - 5.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_magnitude_sqr2r(v, b);
        assert(fabs(v[0] - 25.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_distance2r(v, a, b);
        assert(fabs(v[0] - sqrt(8.0)) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_distance_sqr2r(v, a, b);
        assert(fabs(v[0] - 8.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_normalize2r(v, b);
        assert(fabs(v[0] - 0.6) < EPSILON);
        assert(fabs(v[1] - 0.8) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);

        un_m_reflect2r(v, a, n);
        assert(fabs(v[0] + a[0]) < EPSILON);
        assert(fabs(v[1] - a[1]) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 2);
    }

    /* 3d */
    {
        real a[3] = {1.0, 2.0, 3.0};
        real b[3] = {3.0, 4.0, 5.0};
        real n[3] = {0.0, 1.0, 0.0};
        real v[3];

        un_m_add3r(v, a, b);
        assert(fabs(v[0] - 4.0) < EPSILON);
        assert(fabs(v[1] - 6.0) < EPSILON);
        assert(fabs(v[2] - 8.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_sub3r(v, a, b);
        assert(fabs(v[0] + 2.0) < EPSILON);
        assert(fabs(v[1] + 2.0) < EPSILON);
        assert(fabs(v[2] + 2.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_add_scalar3r(v, a, 1);
        assert(fabs(v[0] - 2.0) < EPSILON);
        assert(fabs(v[1] - 3.0) < EPSILON);
        assert(fabs(v[2] - 4.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_sub_scalar3r(v, a, 1);
        assert(fabs(v[0])        < EPSILON);
        assert(fabs(v[1] - 1.0) < EPSILON);
        assert(fabs(v[2] - 2.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_mul_scalar3r(v, a, 2);
        assert(fabs(v[0] - 2.0)  < EPSILON);
        assert(fabs(v[1] - 4.0)  < EPSILON);
        assert(fabs(v[2] - 6.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_div_scalar3r(v, a, 2);
        assert(fabs(v[0] - 0.5) < EPSILON);
        assert(fabs(v[1] - 1.0) < EPSILON);
        assert(fabs(v[2] - 1.5) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_dot3r(v, a, b);
        assert(fabs(v[0] - 26.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_hadamard3r(v, a, b);
        assert(fabs(v[0] - 3.0)  < EPSILON);
        assert(fabs(v[1] - 8.0)  < EPSILON);
        assert(fabs(v[2] - 15.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_project3r(v, a, b);
        assert(fabs(v[0] - 1.56) < EPSILON);
        assert(fabs(v[1] - 2.08) < EPSILON);
        assert(fabs(v[2] - 2.6) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_cross3r(v, a, b);
        assert(fabs(v[0] + 2.0) < EPSILON);
        assert(fabs(v[1] - 4.0) < EPSILON);
        assert(fabs(v[2] + 2.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_magnitude3r(v, b);
        assert(fabs(v[0] - sqrt(50.0)) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_magnitude_sqr3r(v, b);
        assert(fabs(v[0] - 50.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_distance3r(v, a, b);
        assert(fabs(v[0] - sqrt(12.0)) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_distance_sqr3r(v, a, b);
        assert(fabs(v[0] - 12.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_normalize3r(v, b);
        assert(fabs(v[0] - 0.424264) < EPSILON);
        assert(fabs(v[1] - 0.565685) < EPSILON);
        assert(fabs(v[2] - 0.707107) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);

        un_m_reflect3r(v, a, n);
        assert(fabs(v[0] + a[0]) < EPSILON);
        assert(fabs(v[1] - a[1]) < EPSILON);
        assert(fabs(v[2] + a[2]) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 3);
    }

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

        un_m_add4r(v, a, b);
        assert(fabs(v[0] - 4.0) < EPSILON);
        assert(fabs(v[1] - 6.0) < EPSILON);
        assert(fabs(v[2] - 8.0) < EPSILON);
        assert(fabs(v[3] - 10.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);

        un_m_sub4r(v, a, b);
        assert(fabs(v[0] + 2.0) < EPSILON);
        assert(fabs(v[1] + 2.0) < EPSILON);
        assert(fabs(v[2] + 2.0) < EPSILON);
        assert(fabs(v[3] + 2.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);

        un_m_add_scalar4r(v, a, 1);
        assert(fabs(v[0] - 2.0) < EPSILON);
        assert(fabs(v[1] - 3.0) < EPSILON);
        assert(fabs(v[2] - 4.0) < EPSILON);
        assert(fabs(v[3] - 5.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);

        un_m_sub_scalar4r(v, a, 1);
        assert(fabs(v[0])        < EPSILON);
        assert(fabs(v[1] - 1.0) < EPSILON);
        assert(fabs(v[2] - 2.0) < EPSILON);
        assert(fabs(v[3] - 3.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);

        un_m_mul_scalar4r(v, a, 2);
        assert(fabs(v[0] - 2.0)  < EPSILON);
        assert(fabs(v[1] - 4.0)  < EPSILON);
        assert(fabs(v[2] - 6.0) < EPSILON);
        assert(fabs(v[3] - 8.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);

        un_m_div_scalar4r(v, a, 2);
        assert(fabs(v[0] - 0.5) < EPSILON);
        assert(fabs(v[1] - 1.0) < EPSILON);
        assert(fabs(v[2] - 1.5) < EPSILON);
        assert(fabs(v[3] - 2.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);

        un_m_dot4r(v, a, b);
        assert(fabs(v[0] - 50.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);

        un_m_hadamard4r(v, a, b);
        assert(fabs(v[0] - 3.0)  < EPSILON);
        assert(fabs(v[1] - 8.0)  < EPSILON);
        assert(fabs(v[2] - 15.0) < EPSILON);
        assert(fabs(v[3] - 24.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);

        un_m_project4r(v, a, b);
        assert(fabs(v[0] - 1.744186) < EPSILON);
        assert(fabs(v[1] - 2.325581) < EPSILON);
        assert(fabs(v[2] - 2.906976) < EPSILON);
        assert(fabs(v[3] - 3.488372) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);

        un_m_magnitude4r(v, b);
        assert(fabs(v[0] - sqrt(86.0)) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);

        un_m_magnitude_sqr4r(v, b);
        assert(fabs(v[0] - 86.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);

        un_m_distance4r(v, a, b);
        assert(fabs(v[0] - 4.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);

        un_m_distance_sqr4r(v, a, b);
        assert(fabs(v[0] - 16.0) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);

        un_m_normalize4r(v, b);
        assert(fabs(v[0] - 0.323498) < EPSILON);
        assert(fabs(v[1] - 0.431331) < EPSILON);
        assert(fabs(v[2] - 0.539164) < EPSILON);
        assert(fabs(v[3] - 0.646997) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);

        un_m_reflect4r(v, a, n);
        assert(fabs(v[0] + a[0]) < EPSILON);
        assert(fabs(v[1] - a[1]) < EPSILON);
        assert(fabs(v[2] + a[2]) < EPSILON);
        assert(fabs(v[3] + a[3]) < EPSILON);
        un_memory_set((void*)v, 0, sizeof(*v) * 4);
    }

    return 0;
}