#include "Clouds_Test.h"
CPPUNIT_TEST_SUITE_REGISTRATION(Clouds_Test);

#include "rendering/renderer.h"
#include "rendering/tools.h"
#include "rendering/clouds/public.h"
#include "rendering/clouds/clo_density.h"
#include "rendering/clouds/clo_walking.h"

Clouds_Test::test_clouds_density()
{
    /* Setup */
    double x, y, z;
    CloudsLayerDefinition* layer;
    layer = cloudsGetLayerType().callback_create();

    /* Test default coverage (empty) */
    for (x = -10.0; x < 10.0; x += 10.0)
    {
        for (y = -10.0; y < 10.0; y += 10.0)
        {
            for (z = -10.0; z < 10.0; z += 10.0)
            {
                ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, y, z)), 0.0);
            }
        }
    }

    /* Test coverage by altitude */
    layer->base_coverage = 1.0;
    layer->lower_altitude = -1.0;
    layer->thickness = 2.0;
    cloudsGetLayerType().callback_validate(layer);
    layer->base_coverage = 1.0;
    noiseForceValue(layer->_coverage_noise, 1.0);
    for (x = -10.0; x < 10.0; x += 10.0)
    {
        for (z = -10.0; z < 10.0; z += 10.0)
        {
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, 0.0, z)), 1.0);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, 0.5, z)), 0.5);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, 1.0, z)), 0.0);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, 1.5, z)), 0.0);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, -0.5, z)), 0.5);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, -1.0, z)), 0.0);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, -1.5, z)), 0.0);
        }
    }
    layer->base_coverage = 0.5;
    noiseForceValue(layer->_coverage_noise, 1.0);
    for (x = -10.0; x < 10.0; x += 10.0)
    {
        for (z = -10.0; z < 10.0; z += 10.0)
        {
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, 0.0, z)), 0.5);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, 0.5, z)), 0.25);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, 1.0, z)), 0.0);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, 1.5, z)), 0.0);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, -0.5, z)), 0.25);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, -1.0, z)), 0.0);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, -1.5, z)), 0.0);
        }
    }
    layer->base_coverage = 1.0;
    noiseForceValue(layer->_coverage_noise, 0.5);
    for (x = -10.0; x < 10.0; x += 10.0)
    {
        for (z = -10.0; z < 10.0; z += 10.0)
        {
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, 0.0, z)), 0.5);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, 0.5, z)), 0.25);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, 1.0, z)), 0.0);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, 1.5, z)), 0.0);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, -0.5, z)), 0.25);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, -1.0, z)), 0.0);
            ck_assert_double_eq(cloudsGetLayerCoverage(layer, v3(x, -1.5, z)), 0.0);
        }
    }

    /* TODO Test fake renderer */

    /* TODO Test real renderer */

    /* Teardown */
    cloudsGetLayerType().callback_delete(layer);
}
END_TEST

Clouds_Test::test_clouds_walking_boundaries()
{
    Vector3 start, end;
    int result;
    CloudsLayerDefinition* layer;
    layer = cloudsGetLayerType().callback_create();
    layer->base_coverage = 1.0;
    layer->lower_altitude = -1.0;
    layer->thickness = 2.0;
    cloudsGetLayerType().callback_validate(layer);

    /* Basic cases */
    start = v3(0.0, -3.0, 0.0);
    end = v3(0.0, -2.0, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 0);

    start = v3(0.0, 2.0, 0.0);
    end = v3(0.0, 3.0, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 0);

    start = v3(0.0, -2.0, 0.0);
    end = v3(0.0, 2.0, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 1);
    ck_assert_vector_values(start, 0.0, -1.0, 0.0);
    ck_assert_vector_values(end, 0.0, 1.0, 0.0);

    start = v3(0.0, 0.0, 0.0);
    end = v3(0.0, 2.0, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 1);
    ck_assert_vector_values(start, 0.0, 0.0, 0.0);
    ck_assert_vector_values(end, 0.0, 1.0, 0.0);

    start = v3(0.0, -2.0, 0.0);
    end = v3(0.0, 0.0, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 1);
    ck_assert_vector_values(start, 0.0, -1.0, 0.0);
    ck_assert_vector_values(end, 0.0, 0.0, 0.0);

    /* Basic cases (inverted) */
    start = v3(0.0, -2.0, 0.0);
    end = v3(0.0, -3.0, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 0);

    start = v3(0.0, 3.0, 0.0);
    end = v3(0.0, 2.0, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 0);

    start = v3(0.0, 2.0, 0.0);
    end = v3(0.0, -2.0, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 1);
    ck_assert_vector_values(start, 0.0, 1.0, 0.0);
    ck_assert_vector_values(end, 0.0, -1.0, 0.0);

    start = v3(0.0, 2.0, 0.0);
    end = v3(0.0, 0.0, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 1);
    ck_assert_vector_values(start, 0.0, 1.0, 0.0);
    ck_assert_vector_values(end, 0.0, 0.0, 0.0);

    start = v3(0.0, 0.0, 0.0);
    end = v3(0.0, -2.0, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 1);
    ck_assert_vector_values(start, 0.0, 0.0, 0.0);
    ck_assert_vector_values(end, 0.0, -1.0, 0.0);

    /* Horizontal cases */
    start = v3(0.0, 2.0, 0.0);
    end = v3(10.0, 2.0, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 0);

    start = v3(0.0, 1.00001, 0.0);
    end = v3(10.0, 1.00001, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 0);

    start = v3(0.0, -1.00001, 0.0);
    end = v3(10.0, -1.00001, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 0);

    start = v3(0.0, -2.0, 0.0);
    end = v3(10.0, -2.0, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 0);

    start = v3(0.0, 0.0, 0.0);
    end = v3(10.0, 0.0, 0.0);
    result = cloudsOptimizeWalkingBounds(layer, &start, &end);
    ck_assert_int_eq(result, 1);
    ck_assert_vector_values(start, 0.0, 0.0, 0.0);
    ck_assert_vector_values(end, 10.0, 0.0, 0.0);

    cloudsGetLayerType().callback_delete(layer);
}

static double _getLayerDensitySinX(Renderer* renderer, CloudsLayerDefinition* layer, Vector3 location)
{
    UNUSED(renderer);
    UNUSED(layer);
    double density = sin(location.x * (2.0 * M_PI));
    return (density > 0.0) ? density : 0.0;
}

static double _getEdgeDensitySquared(Renderer* renderer, CloudsLayerDefinition* layer, Vector3 location, double edge_density)
{
    UNUSED(renderer);
    UNUSED(layer);
    UNUSED(location);
    return edge_density * edge_density;
}


Clouds_Test::test_clouds_walking()
{
    /* Init */
    CloudsLayerDefinition* layer;
    layer = cloudsGetLayerType().callback_create();
    layer->lower_altitude = -1.0;
    layer->thickness = 2.0;
    cloudsGetLayerType().callback_validate(layer);

    Renderer* renderer;
    renderer = rendererCreate();

    renderer->render_quality = 8;
    renderer->clouds->getLayerDensity = _getLayerDensitySinX;

    CloudsWalker* walker = cloudsCreateWalker(renderer, layer, v3(-0.4, 0.0, 0.0), v3(1.75, 0.0, 0.0));
    CloudWalkerStepInfo* segment;
    int result;

    /* First step */
    cloudsWalkerSetStepSize(walker, 0.3);
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_false(segment->refined);
    ck_assert_false(segment->subdivided);
    ck_assert_double_eq(segment->length, 0.3);
    ck_assert_double_eq(segment->start.distance_from_start, 0.0);
    ck_assert_vector_values(segment->start.location, -0.4, 0.0, 0.0);
    ck_assert_double_eq(segment->start.global_density, 0.0);
    ck_assert_double_eq(segment->end.distance_from_start, 0.3);
    ck_assert_vector_values(segment->end.location, -0.1, 0.0, 0.0);
    ck_assert_double_eq(segment->end.global_density, 0.0);

    /* Second step */
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_false(segment->refined);
    ck_assert_false(segment->subdivided);
    ck_assert_double_eq(segment->length, 0.3);
    ck_assert_double_eq(segment->start.distance_from_start, 0.3);
    ck_assert_vector_values(segment->start.location, -0.1, 0.0, 0.0);
    ck_assert_double_eq(segment->start.global_density, 0.0);
    ck_assert_double_eq(segment->end.distance_from_start, 0.6);
    ck_assert_vector_values(segment->end.location, 0.2, 0.0, 0.0);
    ck_assert_double_gt(segment->end.global_density, 0.9);

    /* Order to refine second step around the entry point */
    cloudsWalkerOrderRefine(walker, 0.01);
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_true(segment->refined);
    ck_assert_false(segment->subdivided);
    ck_assert_double_in_range(segment->length, 0.19, 0.20);
    ck_assert_double_in_range(segment->start.distance_from_start, 0.40, 0.41);
    ck_assert_double_in_range(segment->start.location.x, 0.0, 0.01);
    ck_assert_double_gt(segment->start.global_density, 0.0);
    ck_assert_double_eq(segment->end.distance_from_start, 0.6);
    ck_assert_vector_values(segment->end.location, 0.2, 0.0, 0.0);
    ck_assert_double_gt(segment->end.global_density, 0.9);

    /* Third step, change step size */
    cloudsWalkerSetStepSize(walker, 0.4);
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_false(segment->refined);
    ck_assert_false(segment->subdivided);
    ck_assert_double_eq(segment->length, 0.4);
    ck_assert_double_eq(segment->start.distance_from_start, 0.6);
    ck_assert_vector_values(segment->start.location, 0.2, 0.0, 0.0);
    ck_assert_double_gt(segment->start.global_density, 0.9);
    ck_assert_double_eq(segment->end.distance_from_start, 1.0);
    ck_assert_vector_values(segment->end.location, 0.6, 0.0, 0.0);
    ck_assert_double_eq(segment->end.global_density, 0.0);

    /* Refine exit point */
    cloudsWalkerOrderRefine(walker, 0.001);
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_true(segment->refined);
    ck_assert_false(segment->subdivided);
    ck_assert_double_in_range(segment->length, 0.3, 0.301);
    ck_assert_double_eq(segment->start.distance_from_start, 0.6);
    ck_assert_vector_values(segment->start.location, 0.2, 0.0, 0.0);
    ck_assert_double_gt(segment->start.global_density, 0.9);
    ck_assert_double_in_range(segment->end.distance_from_start, 0.9, 0.901);
    ck_assert_double_in_range(segment->end.location.x, 0.5, 0.501);
    ck_assert_double_eq(segment->end.global_density, 0.0);

    /* Find next entry point by skipping blank */
    cloudsWalkerSetVoidSkipping(walker, 1);
    cloudsWalkerSetStepSize(walker, 0.2);
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_false(segment->refined);
    ck_assert_false(segment->subdivided);
    ck_assert_double_eq(segment->length, 0.2);
    ck_assert_double_in_range(segment->start.distance_from_start, 1.2, 1.4);
    ck_assert_double_in_range(segment->start.location.x, 0.8, 1.0);
    ck_assert_double_eq(segment->start.global_density, 0.0);
    ck_assert_double_in_range(segment->end.distance_from_start, 1.4, 1.6);
    ck_assert_double_in_range(segment->end.location.x, 1.0, 1.2);
    ck_assert_double_gt(segment->end.global_density, 0.0);

    /* Refine entry point */
    cloudsWalkerOrderRefine(walker, 0.01);
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_true(segment->refined);
    ck_assert_false(segment->subdivided);
    ck_assert_double_in_range(segment->length, 0.0, 0.2);
    ck_assert_double_in_range(segment->start.distance_from_start, 1.4, 1.41);
    ck_assert_double_in_range(segment->start.location.x, 1.0, 1.01);
    ck_assert_double_gt(segment->start.global_density, 0.0);
    ck_assert_double_in_range(segment->end.distance_from_start, 1.41, 1.6);
    ck_assert_double_in_range(segment->end.location.x, 1.01, 1.2);
    ck_assert_double_gt(segment->end.global_density, 0.0);

    /* Subdivide entry for more detail */
    CloudWalkerStepInfo parent = *segment;
    cloudsWalkerOrderSubdivide(walker, 3);
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_false(segment->refined);
    ck_assert_true(segment->subdivided);
    ck_assert_double_eq(segment->length, parent.length / 3.0);
    ck_assert_double_eq(segment->start.distance_from_start, parent.start.distance_from_start);
    ck_assert_double_eq(segment->start.location.x, parent.start.location.x);
    ck_assert_double_eq(segment->end.distance_from_start, parent.start.distance_from_start + segment->length);
    ck_assert_double_eq(segment->end.location.x, parent.start.location.x + segment->length);
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_false(segment->refined);
    ck_assert_true(segment->subdivided);
    ck_assert_double_eq(segment->length, parent.length / 3.0);
    ck_assert_double_eq(segment->start.distance_from_start, parent.start.distance_from_start + segment->length);
    ck_assert_double_eq(segment->start.location.x, parent.start.location.x + segment->length);
    ck_assert_double_eq(segment->end.distance_from_start, parent.start.distance_from_start + 2.0 * segment->length);
    ck_assert_double_eq(segment->end.location.x, parent.start.location.x + 2.0 * segment->length);
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_false(segment->refined);
    ck_assert_true(segment->subdivided);
    ck_assert_double_eq(segment->length, parent.length / 3.0);
    ck_assert_double_eq(segment->start.distance_from_start, parent.start.distance_from_start + 2.0 * segment->length);
    ck_assert_double_eq(segment->start.location.x, parent.start.location.x + 2.0 * segment->length);
    ck_assert_double_eq(segment->end.distance_from_start, parent.end.distance_from_start);
    ck_assert_double_eq(segment->end.location.x, parent.end.location.x);

    /* After subdividing, normal walking resumes */
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_false(segment->refined);
    ck_assert_false(segment->subdivided);
    ck_assert_double_eq(segment->length, 0.2);
    ck_assert_double_in_range(segment->start.distance_from_start, 1.41, 1.6);
    ck_assert_double_in_range(segment->start.location.x, 1.01, 1.2);
    ck_assert_double_gt(segment->start.global_density, 0.0);
    ck_assert_double_in_range(segment->end.distance_from_start, 1.61, 1.8);
    ck_assert_double_in_range(segment->end.location.x, 1.21, 1.4);
    ck_assert_double_gt(segment->end.global_density, 0.0);

    /* Exiting cloud again */
    cloudsWalkerSetStepSize(walker, 0.3);
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_false(segment->refined);
    ck_assert_false(segment->subdivided);
    ck_assert_double_eq(segment->length, 0.3);
    ck_assert_double_in_range(segment->start.distance_from_start, 1.61, 1.8);
    ck_assert_double_in_range(segment->start.location.x, 1.21, 1.4);
    ck_assert_double_gt(segment->start.global_density, 0.0);
    ck_assert_double_in_range(segment->end.distance_from_start, 1.91, 2.1);
    ck_assert_double_in_range(segment->end.location.x, 1.5, 1.7);
    ck_assert_double_eq(segment->end.global_density, 0.0);

    /* A step in the void without skipping */
    cloudsWalkerSetVoidSkipping(walker, 0);
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_false(segment->refined);
    ck_assert_false(segment->subdivided);
    ck_assert_double_eq(segment->length, 0.3);
    ck_assert_double_in_range(segment->start.distance_from_start, 1.91, 2.1);
    ck_assert_double_in_range(segment->start.location.x, 1.5, 1.7);
    ck_assert_double_eq(segment->start.global_density, 0.0);
    ck_assert_double_in_range(segment->end.distance_from_start, 2.21, 2.4);
    ck_assert_double_in_range(segment->end.location.x, 1.8, 2.0);
    ck_assert_double_eq(segment->end.global_density, 0.0);

    /* Walker reached the lookup segment's end, it should stop */
    result = cloudsWalkerPerformStep(walker);
    ck_assert_int_eq(result, 0);

    /* Clean up */
    cloudsDeleteWalker(walker);

    cloudsGetLayerType().callback_delete(layer);
    rendererDelete(renderer);
}

Clouds_Test::test_clouds_walking_local()
{
    /* Init */
    CloudsLayerDefinition* layer;
    layer = cloudsGetLayerType().callback_create();
    layer->lower_altitude = -1.0;
    layer->thickness = 2.0;
    cloudsGetLayerType().callback_validate(layer);

    Renderer* renderer;
    renderer = rendererCreate();

    renderer->render_quality = 8;
    renderer->clouds->getLayerDensity = _getLayerDensitySinX;
    renderer->clouds->getEdgeDensity = _getEdgeDensitySquared;

    CloudsWalker* walker = cloudsCreateWalker(renderer, layer, v3(0.0, 0.0, 0.0), v3(1.0, 0.0, 0.0));
    CloudWalkerStepInfo* segment;
    int result;

    /* Test that local density is off by default */
    cloudsWalkerSetStepSize(walker, 0.3);
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_double_eq(segment->length, 0.3);
    ck_assert_double_eq(segment->start.global_density, 0.0);
    ck_assert_double_eq(segment->start.local_density, 0.0);
    ck_assert_double_eq(segment->end.global_density, 0.951056516295);
    ck_assert_double_eq(segment->end.local_density, 0.0);

    /* Test it's automatically enabled on subdivision */
    cloudsWalkerOrderSubdivide(walker, 2);
    result = cloudsWalkerPerformStep(walker);
    segment = cloudsWalkerGetLastSegment(walker);
    ck_assert_int_eq(result, 1);
    ck_assert_double_eq(segment->length, 0.15);
    ck_assert_double_eq(segment->start.global_density, 0.0);
    ck_assert_double_eq(segment->start.local_density, 0.0);
    ck_assert_double_eq(segment->end.global_density, 0.809016994375);
    ck_assert_double_eq(segment->end.local_density, 0.654508497187);
}