#pragma once

#include <string_view>
#include <vector>
#include <cmath>
#include <functional>

#include "pass.hpp"
#include "matrix.hpp"
#include "object.hpp"
#include "common.hpp"
#include "render_options.hpp"
#include "path.hpp"
#include "shadercompiler.hpp"
#include "rendertarget.hpp"
#include "platform.hpp"

class GFX;
class GFXBuffer;
class GFXCommandBuffer;
class GFXFramebuffer;
class GFXPipeline;
class GFXRenderPass;
class GFXTexture;
class SMAAPass;
class ShadowPass;
class SceneCapture;
class DoFPass;

class Scene;
struct Camera;

struct render_screen_options {
    bool render_world = false;
    Matrix4x4 mvp;
};

class Material;

namespace prism {
    class renderer {
    public:
        explicit renderer(GFX* gfx, bool enable_imgui = true);

        ~renderer();

        RenderTarget* allocate_render_target(prism::Extent extent);

        void resize_render_target(RenderTarget& target, prism::Extent extent);

        void recreate_all_render_targets();

        struct controller_continuity {
            int elementOffset = 0;
        };

        void render(GFXCommandBuffer* command_buffer, Scene* scene, RenderTarget& target, platform::window_ptr index);

        void render_camera(GFXCommandBuffer* command_buffer, Scene& scene, Object camera_object, Camera& camera,
                           prism::Extent extent, RenderTarget& target, controller_continuity &continuity);

        void create_mesh_pipeline(Material& material) const;

        // passes
        template<class T, typename... Args>
        T* addPass(Args &&... args) {
            auto t = std::make_unique<T>(args...);
            t->initialize();

            return static_cast<T*>(passes.emplace_back(std::move(t)).get());
        }

        GFXTexture* get_requested_texture(PassTextureType type) {
            for (auto& pass : passes) {
                auto texture = pass->get_requested_texture(type);
                if (texture != nullptr)
                    return texture;
            }

            return nullptr;
        }

        GFXRenderPass* offscreen_render_pass = nullptr;

        std::unique_ptr<ShadowPass> shadow_pass;
        std::unique_ptr<SceneCapture> scene_capture;

        GFXTexture* dummy_texture = nullptr;
        GFXRenderPass* unorm_render_pass = nullptr;
        GFXPipeline* render_to_unorm_texture_pipeline = nullptr;

        ShaderSource register_shader(std::string_view shader_file);

        void associate_shader_reload(std::string_view shader_file, const std::function<void()> &reload_function);

        void reload_shader(std::string_view shader_file, std::string_view shader_source);

        struct RegisteredShader {
            std::string filename;
            std::string injected_shader_source;
            std::function<void()> reload_function;
        };

        std::vector<RegisteredShader> registered_shaders;

        bool reloading_shader = false;

        [[nodiscard]] std::vector<RenderTarget*> get_all_render_targets() const {
            return render_targets;
        }

    private:
        void create_dummy_texture();

        void create_render_target_resources(RenderTarget& target);

        void create_post_pipelines();

        void create_sky_pipeline();

        void generate_brdf();

        void create_histogram_resources();

        GFX* gfx = nullptr;

        std::vector<RenderTarget*> render_targets;

        // sky
        GFXPipeline* sky_pipeline = nullptr;

        // post
        GFXPipeline* post_pipeline = nullptr;

        // brdf
        GFXPipeline* brdf_pipeline = nullptr;
        GFXTexture* brdf_texture = nullptr;
        GFXFramebuffer* brdf_framebuffer = nullptr;
        GFXRenderPass* brdf_render_pass = nullptr;

        // histogram compute
        GFXPipeline* histogram_pipeline = nullptr, *histogram_average_pipeline = nullptr;
        GFXBuffer* histogram_buffer = nullptr;
        GFXTexture* average_luminance_texture = nullptr;

        std::unique_ptr<SMAAPass> smaa_pass;
        std::unique_ptr<DoFPass> dof_pass;

        std::vector<std::unique_ptr<Pass>> passes;
    };
}