#include "WaterRasterizer.h"

#include "SoftwareRenderer.h"
#include "WaterRenderer.h"
#include "ParallelQueue.h"

WaterRasterizer::WaterRasterizer(SoftwareRenderer* renderer, int client_id):
    Rasterizer(renderer, client_id)
{
}

static Color _postProcessFragment(SoftwareRenderer* renderer, const Vector3 &location, void*)
{
    return renderer->getWaterRenderer()->getResult(location.x, location.z).final;
}

static inline Vector3 _getFirstPassVertex(SoftwareRenderer* renderer, double x, double z)
{
    Vector3 result;

    result.x = x;
    result.y = renderer->getWaterRenderer()->getHeight(x, z);
    result.z = z;

    return result;
}

void WaterRasterizer::rasterizeQuad(CanvasPortion* canvas, double x, double z, double size)
{
    Vector3 v1, v2, v3, v4;

    v1 = _getFirstPassVertex(renderer, x, z);
    v2 = _getFirstPassVertex(renderer, x, z + size);
    v3 = _getFirstPassVertex(renderer, x + size, z + size);
    v4 = _getFirstPassVertex(renderer, x + size, z);

    pushQuad(canvas, v1, v2, v3, v4);
}

void WaterRasterizer::rasterizeToCanvas(CanvasPortion *canvas)
{
    int chunk_factor, chunk_count, i;
    Vector3 cam = renderer->getCameraLocation(VECTOR_ZERO);
    double radius_int, radius_ext, base_chunk_size, chunk_size;

    base_chunk_size = 2.0 / (double)renderer->render_quality;
    if (renderer->render_quality > 7)
    {
        base_chunk_size *= 0.5;
    }

    chunk_factor = 1;
    chunk_count = 2;
    radius_int = 0.0;
    radius_ext = base_chunk_size;
    chunk_size = base_chunk_size;

    double cx = cam.x - fmod(cam.x, base_chunk_size);
    double cz = cam.z - fmod(cam.x, base_chunk_size);

    while (radius_int < 20000.0)
    {
        if (!renderer->addRenderProgress(0.0))
        {
            return;
        }

        for (i = 0; i < chunk_count - 1; i++)
        {
            rasterizeQuad(canvas, cx - radius_ext + chunk_size * i, cz - radius_ext, chunk_size);
            rasterizeQuad(canvas, cx + radius_int, cz - radius_ext + chunk_size * i, chunk_size);
            rasterizeQuad(canvas, cx + radius_int - chunk_size * i, cz + radius_int, chunk_size);
            rasterizeQuad(canvas, cx - radius_ext, cz + radius_int - chunk_size * i, chunk_size);
        }

        if (radius_int > 20.0 && chunk_count % 64 == 0 && (double)chunk_factor < radius_int / 20.0)
        {
            chunk_count /= 2;
            chunk_factor *= 2;
        }
        chunk_count += 2;
        chunk_size = base_chunk_size * chunk_factor;
        radius_int = radius_ext;
        radius_ext += chunk_size;
    }
}