// #include "NewPlacement.hpp"

#include "Rayt.hpp"

#include "CommandLine.hpp"
#include "Timing.hpp"

#define STB_IMAGE_WRITE_IMPLEMENTATION
#include "stb_image_write.h"

using namespace tp;

void writeImage(const RayTracer::RenderBuffer& output, const char* name) {
  // Save the data to a PNG file
  struct urgb {
    uint1 r, g, b, a;
  };

  Buffer2D<urgb> converted;
  converted.reserve(output.size());

  for (RayTracer::RenderBuffer::Index i = 0; i < output.size().x; i++) {
    for (RayTracer::RenderBuffer::Index j = 0; j < output.size().y; j++) {
      converted.get({i, j}).r = uint1(output.get({i, j}).r * 255);
      converted.get({i, j}).g = uint1(output.get({i, j}).g * 255);
      converted.get({i, j}).b = uint1(output.get({i, j}).b * 255);
      converted.get({i, j}).a = uint1(output.get({i, j}).a * 255);
    }
  }

  if (stbi_write_png(name, converted.size().x, converted.size().y, 4, converted.getBuff(), converted.size().x * 4) != 0) {
    // Image saved successfully
    printf("Image saved successfully.\n");
  } else {
    printf("Error saving the image.\n");
  }
}

static void* printStatus(void* arg) {
  auto percentage = (const halnf*) arg;
  auto old = *percentage;
  while (*percentage < 0.99f) {
    sleep(500);
    if (old != *percentage && (*percentage - old) > 0.01) {
      printf("Progress - %f\n", (*percentage) * 100);
      old = *percentage;
    }
  }
  pthread_exit(NULL);
}

void renderCommand(const String& scenePath) {
  Scene scene;

  RayTracer::RenderSettings settings;

  loadScene(scene, scenePath, settings);

  RayTracer::OutputBuffers output;
  RayTracer rayt;

  pthread_t my_thread;
  int ret;

  printf("\nStarting Render:\n\n");

  ret = pthread_create(&my_thread, NULL, &printStatus, &rayt.mProgress.percentage);
  if (ret != 0) {
    printf("Error: pthread_create() failed\n");
    exit(EXIT_FAILURE);
  }

  auto start = get_time();

  rayt.render(scene, output, settings);

  auto end = get_time();

  pthread_join(my_thread, NULL);

  printf("\nRender finished with average render time per sample - %i (ms)\n", end - start);

  writeImage(output.normals, "normals.png");
  writeImage(output.color, "color.png");
  writeImage(output.depth, "depth.png");
}

int main(int argc, const char** argv) {
  tp::ModuleManifest* deps[] = {&gModuleRayTracer, nullptr};
  tp::ModuleManifest module("Rayt", nullptr, nullptr, deps);

  if (module.initialize()) {
    renderCommand("scene.lua");

    module.deinitialize();
  }

  return 0;
}