#include "SoftwareRenderer.h"

#include "CameraDefinition.h"
#include "Scenery.h"
#include "FluidMediumManager.h"
#include "AtmosphereRenderer.h"
#include "AtmosphereDefinition.h"
#include "AtmosphereResult.h"
#include "CloudsRenderer.h"
#include "TerrainRenderer.h"
#include "TexturesRenderer.h"
#include "WaterRenderer.h"
#include "SkyRasterizer.h"
#include "TerrainRasterizer.h"
#include "WaterRasterizer.h"
#include "LightStatus.h"
#include "LightingManager.h"
#include "System.h"
#include "Thread.h"

SoftwareRenderer::SoftwareRenderer(Scenery* scenery)
{
    RenderArea::RenderParams params = {1, 1, 1, 5};

    render_quality = 5;
    render_width = 1;
    render_height = 1;
    render_interrupt = 0;
    render_progress = 0.0;
    is_rendering = 0;
    render_camera = new CameraDefinition;
    render_area = new RenderArea(this);
    render_area->setParams(params);

    atmosphere_renderer = new BaseAtmosphereRenderer(this);
    clouds_renderer = new CloudsRenderer(this);
    terrain_renderer = new TerrainRenderer(this);
    textures_renderer = new TexturesRenderer(this);
    water_renderer = new WaterRenderer(this);

    fluid_medium = new FluidMediumManager(this);
    lighting = new LightingManager();

    if (scenery)
    {
        this->scenery = scenery;
        own_scenery = false;
    }
    else
    {
        this->scenery = new Scenery;
        own_scenery = true;
    }
}

SoftwareRenderer::~SoftwareRenderer()
{
    delete render_camera;
    delete render_area;

    delete atmosphere_renderer;
    delete clouds_renderer;
    delete terrain_renderer;
    delete textures_renderer;
    delete water_renderer;

    delete fluid_medium;
    delete lighting;

    if (own_scenery)
    {
        delete scenery;
    }
}

void SoftwareRenderer::setScenery(Scenery* scenery)
{
    if (!own_scenery)
    {
        this->scenery = new Scenery;
        own_scenery = true;
    }
    scenery->copy(this->scenery);
}

void SoftwareRenderer::prepare()
{
    // Prepare sub renderers
    delete atmosphere_renderer;
    atmosphere_renderer = new SoftwareBrunetonAtmosphereRenderer(this);

    delete clouds_renderer;
    clouds_renderer = new CloudsRenderer(this);
    clouds_renderer->update();

    delete terrain_renderer;
    terrain_renderer = new TerrainRenderer(this);

    delete textures_renderer;
    textures_renderer = new TexturesRenderer(this);

    delete water_renderer;
    water_renderer = new WaterRenderer(this);

    // Prepare global tools
    fluid_medium->clearMedia();
    //fluid_medium->registerMedium(water_renderer);
}

void SoftwareRenderer::rasterize()
{
    TerrainRasterizer terrain(this);
    terrain.renderSurface();

    WaterRasterizer water(this);
    water.renderSurface();

    SkyRasterizer sky(this);
    sky.rasterize();
}

void SoftwareRenderer::setPreviewCallbacks(RenderArea::RenderCallbackStart start, RenderArea::RenderCallbackDraw draw, RenderArea::RenderCallbackUpdate update)
{
    render_area->setPreviewCallbacks(start, draw, update);
}

static void* _renderFirstPass(void* data)
{
    SoftwareRenderer* renderer = (SoftwareRenderer*)data;
    renderer->rasterize();
    renderer->is_rendering = 0;
    return NULL;
}

void SoftwareRenderer::start(RenderArea::RenderParams params)
{
    Thread thread(_renderFirstPass);
    int loops;
    int core_count = System::getCoreCount();

    params.antialias = (params.antialias < 1) ? 1 : params.antialias;
    params.antialias = (params.antialias > 4) ? 4 : params.antialias;

    render_quality = params.quality;
    render_width = params.width * params.antialias;
    render_height = params.height * params.antialias;
    render_interrupt = 0;
    render_progress = 0.0;

    render_camera->setRenderSize(render_width, render_height);

    render_area->setBackgroundColor(COLOR_BLACK);
    render_area->setParams(params);
    render_area->clear();

    prepare();

    is_rendering = 1;
    thread.start(this);
    loops = 0;

    while (is_rendering)
    {
        Thread::timeSleepMs(100);

        if (++loops >= 10)
        {
            render_area->update();
            loops = 0;
        }
    }
    thread.join();

    is_rendering = 1;
    render_area->postProcess(core_count);
    is_rendering = 0;
}

void SoftwareRenderer::interrupt()
{
    render_interrupt = 1;
}

Color SoftwareRenderer::applyLightingToSurface(const Vector3 &location, const Vector3 &normal, const SurfaceMaterial &material)
{
    LightStatus status(lighting, location, getCameraLocation(location));
    atmosphere_renderer->getLightingStatus(&status, normal, 0);
    return status.apply(normal, material);
}

Color SoftwareRenderer::applyMediumTraversal(Vector3 location, Color color)
{
    color = atmosphere_renderer->applyAerialPerspective(location, color).final;
    color = clouds_renderer->getColor(getCameraLocation(location), location, color);
    return color;

    /*Vector3 eye = cameraGetLocation(scenery->getCamera());
    return fluid_medium->applyTraversal(eye, location, color);*/
}

RayCastingResult SoftwareRenderer::rayWalking(const Vector3 &location, const Vector3 &direction, int, int, int, int)
{
    RayCastingResult result;
    Color sky_color;

    result = terrain_renderer->castRay(location, direction);
    if (!result.hit)
    {
        sky_color = atmosphere_renderer->getSkyColor(direction).final;

        result.hit = 1;
        result.hit_location = location.add(direction.scale(1000.0));
        result.hit_color = clouds_renderer->getColor(location, result.hit_location, sky_color);
    }

    return result;
}

int SoftwareRenderer::addRenderProgress(double)
{
    return not render_interrupt;
}

Vector3 SoftwareRenderer::getCameraLocation(Vector3)
{
    return render_camera->getLocation();
}

Vector3 SoftwareRenderer::getCameraDirection(Vector3)
{
    return render_camera->getDirectionNormalized();
}

double SoftwareRenderer::getPrecision(Vector3 location)
{
    Vector3 projected;

    projected = render_camera->project(location);
    projected.x += 1.0;
    //projected.y += 1.0;

    return render_camera->unproject(projected).sub(location).getNorm(); // / (double)render_quality;
}

Vector3 SoftwareRenderer::projectPoint(Vector3 point)
{
    return render_camera->project(point);
}

Vector3 SoftwareRenderer::unprojectPoint(Vector3 point)
{
    return render_camera->unproject(point);
}

void SoftwareRenderer::pushTriangle(Vector3 v1, Vector3 v2, Vector3 v3, RenderArea::f_RenderFragmentCallback callback, void* callback_data)
{
    Vector3 p1, p2, p3;

    p1 = projectPoint(v1);
    p2 = projectPoint(v2);
    p3 = projectPoint(v3);

    render_area->pushTriangle(p1, p2, p3, v1, v2, v3, callback, callback_data);
}

void SoftwareRenderer::pushQuad(Vector3 v1, Vector3 v2, Vector3 v3, Vector3 v4, RenderArea::f_RenderFragmentCallback callback, void* callback_data)
{
    pushTriangle(v2, v3, v1, callback, callback_data);
    pushTriangle(v4, v1, v3, callback, callback_data);
}

void SoftwareRenderer::pushDisplacedTriangle(Vector3 v1, Vector3 v2, Vector3 v3, Vector3 ov1, Vector3 ov2, Vector3 ov3, RenderArea::f_RenderFragmentCallback callback, void* callback_data)
{
    Vector3 p1, p2, p3;

    p1 = projectPoint(v1);
    p2 = projectPoint(v2);
    p3 = projectPoint(v3);

    render_area->pushTriangle(p1, p2, p3, ov1, ov2, ov3, callback, callback_data);
}

void SoftwareRenderer::pushDisplacedQuad(Vector3 v1, Vector3 v2, Vector3 v3, Vector3 v4, Vector3 ov1, Vector3 ov2, Vector3 ov3, Vector3 ov4, RenderArea::f_RenderFragmentCallback callback, void* callback_data)
{
    pushDisplacedTriangle(v2, v3, v1, ov2, ov3, ov1, callback, callback_data);
    pushDisplacedTriangle(v4, v1, v3, ov4, ov1, ov3, callback, callback_data);
}