#pragma once
#include <chrono>
#include <vector>
#include <include/core/SkSurface.h>
#include <unordered_set>
#include "../DrawData.hpp"
#include "../DrawComponents/DrawComponent.hpp"

class DrawingProgram;
class DrawingProgramCache;

class DrawingProgramCacheBVHNode {
    public:
        SCollision::AABB<WorldScalar> bounds;
        std::vector<CollabListType::ObjectInfoPtr> components;
        std::vector<std::shared_ptr<DrawingProgramCacheBVHNode>> children;

        CoordSpaceHelper coords;
        Vector2i resolution;
        struct DrawCacheData {
            uint64_t lastDrawnComponentPlacement;
            uint64_t firstDrawnComponentPlacement;
            sk_sp<SkSurface> surface;
            std::chrono::steady_clock::time_point lastRenderTime;
            DrawingProgramCache* attachedCache;
        };

        std::optional<DrawCacheData> drawCache;
};

class DrawingProgramCache {
    public:
        DrawingProgramCache(DrawingProgram& initDrawP);
        ~DrawingProgramCache();

        static std::unordered_set<std::shared_ptr<DrawingProgramCacheBVHNode>> nodesWithCachedSurfaces;
        bool disableRefresh = false;

        void update();
        void erase_component(const CollabListType::ObjectInfoPtr& c);
        void add_component(const CollabListType::ObjectInfoPtr& c);
        void refresh_all_draw_cache(const DrawData& drawData);
        void traverse_bvh_run_function(const SCollision::AABB<WorldScalar>& aabb, std::function<bool(const std::shared_ptr<DrawingProgramCacheBVHNode>& node, const std::vector<CollabListType::ObjectInfoPtr>& components)> f);
        void traverse_bvh_erase_function(const SCollision::AABB<WorldScalar>& aabb, std::function<bool(const std::shared_ptr<DrawingProgramCacheBVHNode>& node, std::vector<CollabListType::ObjectInfoPtr>& components)> f);
        void force_rebuild(const std::vector<CollabListType::ObjectInfoPtr>& componentsToBuild);
        // Run before actually updating the component
        void preupdate_component(const CollabListType::ObjectInfoPtr& c);
        void invalidate_cache_before_pos(uint64_t placementToInvalidateAt);
        void invalidate_cache_at_aabb_before_pos(const SCollision::AABB<WorldScalar>& aabb, uint64_t placementToInvalidateAt);
        void clear();
        void draw_components_to_canvas(SkCanvas* canvas, const DrawData& drawData, uint64_t* lastDrawnComponentPlacement = nullptr);
        const std::vector<CollabListType::ObjectInfoPtr>& get_unsorted_component_list() const;
        const std::chrono::steady_clock::time_point& get_last_bvh_build_time() const;
        void clear_own_cached_surfaces();
    private:
        bool traverse_bvh_erase_function_recursive(const std::shared_ptr<DrawingProgramCacheBVHNode>& bvhNode, const SCollision::AABB<WorldScalar>& aabb, std::function<bool(const std::shared_ptr<DrawingProgramCacheBVHNode>& node, std::vector<CollabListType::ObjectInfoPtr>& components)> f);
        void build_bvh_node_coords_and_resolution(DrawingProgramCacheBVHNode& node);
        void refresh_draw_cache(const std::shared_ptr<DrawingProgramCacheBVHNode>& bvhNode, const DrawData& drawData);
        void build(std::vector<CollabListType::ObjectInfoPtr> components);
        void traverse_bvh_run_function_recursive(const std::shared_ptr<DrawingProgramCacheBVHNode>& bvhNode, const SCollision::AABB<WorldScalar>& aabb, std::function<bool(const std::shared_ptr<DrawingProgramCacheBVHNode>& node, const std::vector<CollabListType::ObjectInfoPtr>& components)> f);
        std::pair<std::shared_ptr<DrawingProgramCacheBVHNode>, std::vector<CollabListType::ObjectInfoPtr>::iterator> get_bvh_node_fully_containing_recursive(const std::shared_ptr<DrawingProgramCacheBVHNode>& bvhNode, DrawComponent* c);
        void build_bvh_node(const std::shared_ptr<DrawingProgramCacheBVHNode>& bvhNode, std::vector<CollabListType::ObjectInfoPtr> components);

        std::chrono::steady_clock::time_point lastBvhBuildTime;
        std::shared_ptr<DrawingProgramCacheBVHNode> bvhRoot;
        std::vector<CollabListType::ObjectInfoPtr> unsortedComponents;
        DrawingProgram& drawP;
};