#include "Scene.h"
#include <iostream>
#include <chrono>

Scene::Scene() {}

Scene::~Scene() {}

static bool firstTime = true;
void Scene::render(Camera cam, RenderParameter params)
{
    if (!firstTime)
    {
        pendingCancel = true;
        worker.join();
        firstTime = false;
    }
    pendingCancel = false;
    image.clear();
    accumulator.clear();
    image.resize(params.width * params.height);
    accumulator.resize(params.width * params.height);
    worker = std::thread(
        [&, cam, params]()
        {
            for (int samp = 0; samp < params.numSamples; ++samp)
            {
                if (pendingCancel)
                    return;
                Batch batch;
                for (int w = 0; w < params.width; ++w)
                {
                    batch.jobs.push_back(
                        [&, w]()
                        {
                            for (int h = 0; h < params.height; ++h)
                            {
                                // Ray r = Ray();
                                // bvh.traceRay(r);
                                accumulator[w + h * params.width] +=
                                    glm::vec3(w / float(params.width * params.numSamples), h / float(params.height * params.numSamples), 0);
                            }
                        });
                }
                auto start = std::chrono::high_resolution_clock::now();
                threadPool.runBatch(std::move(batch));
                auto end = std::chrono::high_resolution_clock::now();
                std::cout << std::chrono::duration_cast<std::chrono::milliseconds>(end - start).count() << std::endl;
                std::memcpy(image.data(), accumulator.data(), accumulator.size() * sizeof(glm::vec3));
            }
        });
}