/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
// Original file Copyright Crytek GMBH or its affiliates, used under license.

#ifndef CRYINCLUDE_CRYPHYSICS_OBBTREE_H
#define CRYINCLUDE_CRYPHYSICS_OBBTREE_H
#pragma once

#if ENABLE_CRY_PHYSICS

struct OBBnode
{
    Vec3 axes[3];
    Vec3 center;
    Vec3 size;
    int iparent;
    int ichild;
    int ntris;

    AUTO_STRUCT_INFO
};

class CTriMesh;

class COBBTree
    : public CBVTree
{
public:
    COBBTree() { m_nMinTrisPerNode = 2; m_nMaxTrisPerNode = 4; m_maxSkipDim = 0; m_pNodes = 0; m_pTri2Node = 0; }
    virtual ~COBBTree()
    {
        if (m_pNodes)
        {
            delete[] m_pNodes;
        }
        m_pNodes = 0;
        if (m_pTri2Node)
        {
            delete[] m_pTri2Node;
        }
        m_pTri2Node = 0;
    }
    virtual int GetType() { return BVT_OBB; }

    virtual float Build(CGeometry* pMesh);
    virtual void SetGeomConvex();

    void SetParams(int nMinTrisPerNode, int nMaxTrisPerNode, float skipdim);
    float BuildNode(int iNode, int iTriStart, int nTris, int nDepth);
    virtual int PrepareForIntersectionTest(geometry_under_test* pGTest, CGeometry* pCollider, geometry_under_test* pGTestColl);
    virtual void CleanupAfterIntersectionTest(geometry_under_test* pGTest);
    virtual void GetBBox(box* pbox);
    virtual int MaxPrimsInNode() { return m_nMaxTrisInNode; }
    virtual void GetNodeBV(BV*& pBV, int iNode = 0, int iCaller = 0);
    virtual void GetNodeBV(BV*& pBV, const Vec3& sweepdir, float sweepstep, int iNode = 0, int iCaller = 0);
    virtual void GetNodeBV(const Matrix33& Rw, const Vec3& offsw, float scalew, BV*& pBV, int iNode = 0, int iCaller = 0);
    virtual void GetNodeBV(const Matrix33& Rw, const Vec3& offsw, float scalew, BV*& pBV, const Vec3& sweepdir, float sweepstep, int iNode = 0, int iCaller = 0) {}
    virtual float SplitPriority(const BV* pBV);
    virtual void GetNodeChildrenBVs(const Matrix33& Rw, const Vec3& offsw, float scalew, const BV* pBV_parent, BV*& pBV_child1, BV*& pBV_child2, int iCaller = 0);
    virtual void GetNodeChildrenBVs(const BV* pBV_parent, BV*& pBV_child1, BV*& pBV_child2, int iCaller = 0);
    virtual void GetNodeChildrenBVs(const BV* pBV_parent, const Vec3& sweepdir, float sweepstep, BV*& pBV_child1, BV*& pBV_child2, int iCaller = 0);
    virtual void ReleaseLastBVs(int iCaller = 0);
    virtual void ReleaseLastSweptBVs(int iCaller = 0);
    virtual int GetNodeContents(int iNode, BV* pBVCollider, int bColliderUsed, int bColliderLocal,
        geometry_under_test* pGTest, geometry_under_test* pGTestOp);
    virtual int GetNodeContentsIdx(int iNode, int& iStartPrim) { iStartPrim = m_pNodes[iNode].ichild; return m_pNodes[iNode].ntris; }
    virtual void MarkUsedTriangle(int itri, geometry_under_test* pGTest);
    virtual float GetMaxSkipDim() { return m_maxSkipDim / max(max(m_pNodes[0].size.x, m_pNodes[0].size.y), m_pNodes[0].size.z); }

    virtual void GetMemoryStatistics(ICrySizer* pSizer);
    virtual void Save(CMemStream& stm);
    virtual void Load(CMemStream& stm, CGeometry* pGeom);

    virtual int SanityCheck();

    CTriMesh* m_pMesh;
    OBBnode* m_pNodes;
    int m_nNodes;
    int m_nNodesAlloc;
    index_t* m_pTri2Node;
    int m_nMaxTrisInNode;

    int m_nMinTrisPerNode;
    int m_nMaxTrisPerNode;
    float m_maxSkipDim;
    int* m_pMapVtxUsed;
    Vec3* m_pVtxUsed;
};


#endif // ENABLE_CRY_PHYSICS

#endif // CRYINCLUDE_CRYPHYSICS_OBBTREE_H