#include "SoftwareCanvasRenderer.h"

#include "Rasterizer.h"
#include "SoftwareRenderer.h"
#include "Canvas.h"
#include "TerrainRasterizer.h"
#include "WaterRasterizer.h"
#include "SkyRasterizer.h"
#include "CameraDefinition.h"
#include "ParallelWork.h"
#include "CanvasPortion.h"
#include "CanvasPixelShader.h"
#include "RenderConfig.h"
#include "ColorProfile.h"
#include "CanvasPreview.h"
#include "RenderProgress.h"

SoftwareCanvasRenderer::SoftwareCanvasRenderer(Scenery *scenery):
    SoftwareRenderer(scenery)
{
    started = false;
    finished = false;
    interrupted = false;
    canvas = new Canvas();
    progress = new RenderProgress();
    samples = 1;

    rasterizers.push_back(new SkyRasterizer(this, progress, 0));
    rasterizers.push_back(new WaterRasterizer(this, progress, 1));
    rasterizers.push_back(new TerrainRasterizer(this, progress, 2));

    current_work = NULL;
}

SoftwareCanvasRenderer::~SoftwareCanvasRenderer()
{
    delete canvas;
    delete progress;

    for (auto &rasterizer: rasterizers)
    {
        delete rasterizer;
    }
}

double SoftwareCanvasRenderer::getProgress() const
{
    return progress->get();
}

void SoftwareCanvasRenderer::setConfig(const RenderConfig &config)
{
    if (not started)
    {
        setSize(config.width, config.height, config.antialias);
        render_quality = config.quality;
    }
}

void SoftwareCanvasRenderer::setSize(int width, int height, int samples)
{
    if (not started)
    {
        canvas->setSize(width * samples, height * samples);
        this->samples = samples;
    }
}

void SoftwareCanvasRenderer::render()
{
    started = true;
    progress->reset();

    render_camera->setRenderSize(canvas->getWidth(), canvas->getHeight());

    prepare();

    // Iterate portions
    int nx = canvas->getHorizontalPortionCount();
    int ny = canvas->getVerticalPortionCount();
    int n = nx * ny;
    progress->enterSub(n);
    for (int y = 0; y < ny; y++)
    {
        for (int x = 0; x < nx; x++)
        {
            CanvasPortion *portion = canvas->at(x, y);

            progress->enterSub(2);

            if (not interrupted)
            {
                portion->preparePixels();
                rasterize(portion);
            }

            if (not interrupted)
            {
                applyPixelShader(portion);
            }

            portion->discardPixels();

            progress->exitSub();
        }
    }
    progress->exitSub();
    finished = true;
}

void SoftwareCanvasRenderer::interrupt()
{
    interrupted = true;

    if (current_work)
    {
        current_work->interrupt();
    }
    for (auto &rasterizer:rasterizers)
    {
        rasterizer->interrupt();
    }
}

const Rasterizer &SoftwareCanvasRenderer::getRasterizer(int client_id) const
{
    return *(rasterizers[client_id]);
}

bool SoftwareCanvasRenderer::saveToDisk(const std::string &filepath) const
{
    return getCanvas()->saveToDisk(filepath, *getCanvas()->getPreview()->getToneMapping(), samples);
}

void SoftwareCanvasRenderer::rasterize(CanvasPortion *portion)
{
    int count = 0;
    for (auto &rasterizer:rasterizers)
    {
        count += rasterizer->prepareRasterization();
    }

    progress->enterSub(count);
    for (auto &rasterizer:rasterizers)
    {
        rasterizer->rasterizeToCanvas(portion);
    }
    progress->exitSub();
}

void SoftwareCanvasRenderer::applyPixelShader(CanvasPortion *portion)
{
    // Subdivide in chunks
    int chunk_size = 64;
    int chunks_x = (portion->getWidth() - 1) / chunk_size + 1;
    int chunks_y = (portion->getHeight() - 1) / chunk_size + 1;
    int units = chunks_x * chunks_y;

    // Render chunks in parallel
    progress->enterSub(portion->getWidth() * portion->getHeight());
    for (int sub_chunk_size = chunk_size; sub_chunk_size >= 1; sub_chunk_size /= 2)
    {
        if (interrupted)
        {
            break;
        }

        CanvasPixelShader shader(*this, portion, progress, chunk_size, sub_chunk_size, chunks_x, chunks_y);
        ParallelWork work(&shader, units);

        current_work = &work;
        work.perform();
        current_work = NULL;
    }
    progress->exitSub();
}