#pragma once

#include "RetroTypes.hpp"
#include "zeus/CAABox.hpp"
#include "CAreaOctTree.hpp"

namespace urde
{
class CCollisionInfoList;
class CCollisionInfo;
class CMaterialList;

class CAABoxAreaCache
{
    friend class CMetroidAreaCollider;
    const zeus::CAABox& x0_aabb;
    const zeus::CPlane* x4_planes;
    const CMaterialFilter& x8_filter;
    const CMaterialList& xc_material;
    CCollisionInfoList& x10_collisionList;
    zeus::CVector3f x14_center;
    zeus::CVector3f x20_halfExtent;
public:
    CAABoxAreaCache(const zeus::CAABox& aabb, const zeus::CPlane* pl, const CMaterialFilter& filter,
                    const CMaterialList& material, CCollisionInfoList& collisionList);
};

class CBooleanAABoxAreaCache
{
    friend class CMetroidAreaCollider;
    const zeus::CAABox& x0_aabb;
    const CMaterialFilter& x4_filter;
    zeus::CVector3f x8_center;
    zeus::CVector3f x14_halfExtent;
public:
    CBooleanAABoxAreaCache(const zeus::CAABox& aabb, const CMaterialFilter& filter);
};

class CSphereAreaCache
{
    friend class CMetroidAreaCollider;
    const zeus::CAABox& x0_aabb;
    const zeus::CSphere& x4_sphere;
    const CMaterialFilter& x8_filter;
    const CMaterialList& xc_material;
    CCollisionInfoList& x10_collisionList;
public:
    CSphereAreaCache(const zeus::CAABox& aabb, const zeus::CSphere& sphere, const CMaterialFilter& filter,
                     const CMaterialList& material, CCollisionInfoList& collisionList);
};

class CBooleanSphereAreaCache
{
    friend class CMetroidAreaCollider;
    const zeus::CAABox& x0_aabb;
    const zeus::CSphere& x4_sphere;
    const CMaterialFilter& x8_filter;
public:
    CBooleanSphereAreaCache(const zeus::CAABox& aabb, const zeus::CSphere& sphere,
                            const CMaterialFilter& filter);
};

struct SBoxEdge
{
    zeus::CLineSeg x0_seg;
    zeus::CVector3d x28_start;
    zeus::CVector3d x40_end;
    zeus::CVector3d x58_delta;
    zeus::CVector3d x70_coDir;
    double x88_dirCoDirDot;
    SBoxEdge(const zeus::CAABox& aabb, int idx, const zeus::CVector3f& dir);
};

class CMovingAABoxComponents
{
    friend class CMetroidAreaCollider;
    friend class CCollidableOBBTree;
    rstl::reserved_vector<SBoxEdge, 12> x0_edges;
    rstl::reserved_vector<u32, 8> x6c4_vertIdxs;
    zeus::CAABox x6e8_aabb;
public:
    CMovingAABoxComponents(const zeus::CAABox& aabb, const zeus::CVector3f& dir);
};

class CMetroidAreaCollider
{
    friend class CCollidableOBBTree;
    static u32 g_CalledClip;
    static u32 g_RejectedByClip;
    static u32 g_TrianglesProcessed;
    static u32 g_DupTrianglesProcessed;
    static u16 g_DupPrimitiveCheckCount;
    static u16 g_DupVertexList[0x5000];
    static u16 g_DupEdgeList[0xC000];
    static u16 g_DupTriangleList[0x4000];
    static bool AABoxCollisionCheckBoolean_Internal(const CAreaOctTree::Node& node,
                                                    const CBooleanAABoxAreaCache& cache);
    static bool AABoxCollisionCheck_Internal(const CAreaOctTree::Node& node,
                                             const CAABoxAreaCache& cache);

    static bool SphereCollisionCheckBoolean_Internal(const CAreaOctTree::Node& node,
                                                     const CBooleanSphereAreaCache& cache);
    static bool SphereCollisionCheck_Internal(const CAreaOctTree::Node& node,
                                              const CSphereAreaCache& cache);

    static bool MovingAABoxCollisionCheck_BoxVertexTri(const CCollisionSurface& surf, const zeus::CAABox& aabb,
                                                       const rstl::reserved_vector<u32, 8>& vertIndices,
                                                       const zeus::CVector3f& dir, double& d,
                                                       zeus::CVector3f& normal, zeus::CVector3f& point);
    static bool MovingAABoxCollisionCheck_TriVertexBox(const zeus::CVector3f& vert, const zeus::CAABox& aabb,
                                                       const zeus::CVector3f& dir, double& d,
                                                       zeus::CVector3f& normal, zeus::CVector3f& point);
    static bool MovingAABoxCollisionCheck_Edge(const zeus::CVector3f& ev0, const zeus::CVector3f& ev1,
                                               const rstl::reserved_vector<SBoxEdge, 12>& edges,
                                               const zeus::CVector3f& dir, double& d,
                                               zeus::CVector3f& normal, zeus::CVector3f& point);
public:
    class COctreeLeafCache
    {
        friend class CMetroidAreaCollider;
        const CAreaOctTree& x0_octTree;
        rstl::reserved_vector<CAreaOctTree::Node, 64> x4_nodeCache;
        bool x908_24_overflow : 1;
    public:
        COctreeLeafCache(const CAreaOctTree& octTree);
        void AddLeaf(const CAreaOctTree::Node& node);
        u32 GetNumLeaves() const { return x4_nodeCache.size(); }
        bool HasCacheOverflowed() const { return x908_24_overflow; }
        const CAreaOctTree& GetOctTree() const { return x0_octTree; }
        rstl::reserved_vector<CAreaOctTree::Node, 64>::iterator begin() { return x4_nodeCache.begin(); }
        rstl::reserved_vector<CAreaOctTree::Node, 64>::iterator end() { return x4_nodeCache.end(); }
    };
    static void BuildOctreeLeafCache(const CAreaOctTree::Node& root, const zeus::CAABox& aabb,
                                     CMetroidAreaCollider::COctreeLeafCache& cache);
    static bool ConvexPolyCollision(const zeus::CPlane* planes, const zeus::CVector3f* verts,
                                    zeus::CAABox& aabb);

    static bool AABoxCollisionCheckBoolean_Cached(const COctreeLeafCache& leafCache, const zeus::CAABox& aabb,
                                                  const CMaterialFilter& filter);
    static bool AABoxCollisionCheckBoolean(const CAreaOctTree& octTree, const zeus::CAABox& aabb,
                                           const CMaterialFilter& filter);

    static bool SphereCollisionCheckBoolean_Cached(const COctreeLeafCache& leafCache, const zeus::CAABox& aabb,
                                                   const zeus::CSphere& sphere, const CMaterialFilter& filter);
    static bool SphereCollisionCheckBoolean(const CAreaOctTree& octTree, const zeus::CAABox& aabb,
                                            const zeus::CSphere& sphere, const CMaterialFilter& filter);

    static bool AABoxCollisionCheck_Cached(const COctreeLeafCache& leafCache, const zeus::CAABox& aabb,
                                           const CMaterialFilter& filter, const CMaterialList& matList,
                                           CCollisionInfoList& list);
    static bool AABoxCollisionCheck(const CAreaOctTree& octTree, const zeus::CAABox& aabb,
                                    const CMaterialFilter& filter, const CMaterialList& matList,
                                    CCollisionInfoList& list);

    static bool SphereCollisionCheck_Cached(const COctreeLeafCache& leafCache, const zeus::CAABox& aabb,
                                            const zeus::CSphere& sphere, const CMaterialList& matList,
                                            const CMaterialFilter& filter, CCollisionInfoList& list);
    static bool SphereCollisionCheck(const CAreaOctTree& octTree, const zeus::CAABox& aabb,
                                     const zeus::CSphere& sphere, const CMaterialList& matList,
                                     const CMaterialFilter& filter, CCollisionInfoList& list);

    static bool MovingAABoxCollisionCheck_Cached(const COctreeLeafCache& leafCache, const zeus::CAABox& aabb,
                                                 const CMaterialFilter& filter, const CMaterialList& matList,
                                                 const zeus::CVector3f& dir, float mag, CCollisionInfo& infoOut,
                                                 double& dOut);
    static bool MovingSphereCollisionCheck_Cached(const COctreeLeafCache& leafCache, const zeus::CAABox& aabb,
                                                  const zeus::CSphere& sphere,
                                                  const CMaterialFilter& filter, const CMaterialList& matList,
                                                  const zeus::CVector3f& dir, float mag, CCollisionInfo& infoOut,
                                                  double& dOut);
    static void ResetInternalCounters();
};

class CAreaCollisionCache
{
    zeus::CAABox x0_aabb;
    rstl::reserved_vector<CMetroidAreaCollider::COctreeLeafCache, 3> x18_leafCaches;
    union
    {
        struct
        {
            bool x1b40_24_leafOverflow : 1;
            bool x1b40_25_cacheOverflow : 1;
        };
        u32 _dummy = 0;
    };
public:
    CAreaCollisionCache(const zeus::CAABox& aabb) : x0_aabb(aabb) {}
    void ClearCache();
    const zeus::CAABox& GetCacheBounds() const { return x0_aabb; }
    void SetCacheBounds(const zeus::CAABox& aabb) { x0_aabb = aabb; }
    void AddOctreeLeafCache(const CMetroidAreaCollider::COctreeLeafCache& leafCache);
    u32 GetNumCaches() const { return x18_leafCaches.size(); }
    const CMetroidAreaCollider::COctreeLeafCache& GetOctreeLeafCache(int idx) { return x18_leafCaches[idx]; }
    bool HasCacheOverflowed() const { return x1b40_24_leafOverflow; }
    rstl::reserved_vector<CMetroidAreaCollider::COctreeLeafCache, 3>::iterator begin() { return x18_leafCaches.begin(); }
    rstl::reserved_vector<CMetroidAreaCollider::COctreeLeafCache, 3>::iterator end() { return x18_leafCaches.end(); }
};

}