#pragma once
#include "helper.hpp"
#include "imgui.h"


class Application2 : public Application {
public:
    Application2(int argc, char** argv) : Application(argc, argv, "Assignment 1") {
    }

private:
    Vec3fa renderPixel(float x, float y, const ISPCCamera& camera, RayStats& stats, RandomSampler& sampler) override;
    Vec3fa renderPixelOrig(float x, float y, const ISPCCamera& camera, RayStats& stats, RandomSampler& sampler);
    Vec3fa renderPixelHomogeneous(float x, float y, const ISPCCamera& camera, RayStats& stats, RandomSampler& sampler);

    void drawGUI() override {
        ImGui::Checkbox("Bounding Box", &boundingBox);
        ImGui::InputInt("Ray depth", &ray_depth);
        if (ray_depth < 1) {
            ray_depth = 1;
        }

        const char* items[] = {"Original", "Homogeneous"};
        ImGui::Combo("Version", &selected, items, 2);

        ImGui::SliderFloat("mu_a", &mu_a, 0.0, 1.0);
        ImGui::SliderFloat("mu_s", &mu_s, 0.0, 1.0);
        ImGui::SliderFloat("scattering parameter", &scattering_parameter, 0.0, 1.0);

        const char* scenes[] = {"Gnome", "Horse", "Heterogenous"};
        int oldscene = scene;
        ImGui::Combo("Scenes", &scene, scenes, 3);
        if (scene != oldscene) {
            Data_Destructor(&data);
            Data_Constructor(&data, 1, 8);
            if (scene == 0)
                gnomeScene();
            if (scene == 1)
                horseScene();
            if (scene == 2)
                heterogenousScene();
        }
    }

    void initScene() override;

    void emptyScene();

    void gnomeScene();

    void horseScene();

    void heterogenousScene();

    int ray_depth = 15;
    bool boundingBox = true;
    int selected = 0;
    int scene = 0;
    float mu_a = 0.0;
    float mu_s = 0.0;
    float scattering_parameter = 0.4;
};