#pragma once

#include "IOStreams.hpp"
#include "CFactoryMgr.hpp"
#include "COBBTree.hpp"
#include "zeus/CAABox.hpp"
#include "CCollisionPrimitive.hpp"

namespace urde
{
class CCollidableOBBTreeGroupContainer
{
    friend class CCollidableOBBTreeGroup;
    std::vector<std::unique_ptr<COBBTree>> x0_trees;
    std::vector<zeus::CAABox> x10_aabbs;
    zeus::CAABox x20_aabox;
public:
    CCollidableOBBTreeGroupContainer(CInputStream& in);
    CCollidableOBBTreeGroupContainer(const zeus::CVector3f&, const zeus::CVector3f&);
};

class CCollidableOBBTreeGroup : public CCollisionPrimitive
{
    static const Type sType;
    static u32 sTableIndex;
    const CCollidableOBBTreeGroupContainer* x10_container;
public:
    CCollidableOBBTreeGroup(CInputStream& in);
    CCollidableOBBTreeGroup(const CCollidableOBBTreeGroupContainer*, const CMaterialList&);
    virtual ~CCollidableOBBTreeGroup() {}

    void ResetTestStats() const;
    virtual u32 GetTableIndex() const;
    virtual zeus::CAABox CalculateAABox(const zeus::CTransform&) const;
    virtual zeus::CAABox CalculateLocalAABox() const;
    virtual FourCC GetPrimType() const;
    virtual CRayCastResult CastRayInternal(const CInternalRayCastStructure&) const;

    static const Type& GetType();
    static void SetStaticTableIndex(u32 index);
    /* Sphere Collide */
    static bool SphereCollide(const CInternalCollisionStructure&, CCollisionInfoList&);
    static bool SphereCollideBoolean(const CInternalCollisionStructure&);
    static bool CollideMovingSphere(const CInternalCollisionStructure&, const zeus::CVector3f&, double&,
                                    CCollisionInfo&);
    /* AABox Collide */
    static bool AABoxCollide(const CInternalCollisionStructure&, CCollisionInfoList&);
    static bool AABoxCollideBoolean(const CInternalCollisionStructure&);
    static bool CollideMovingAABox(const CInternalCollisionStructure&, const zeus::CVector3f&, double&,
                                    CCollisionInfo&);
};

CFactoryFnReturn FCollidableOBBTreeGroupFactory(const SObjectTag& tag, CInputStream& in, const CVParamTransfer& vparms,
                                                CObjectReference* selfRef);
}