#include "CMetaAnimPlay.hpp"
#include "CAnimSysContext.hpp"
#include "CSimplePool.hpp"
#include "CAllFormatsAnimSource.hpp"
#include "CAnimTreeAnimReaderContainer.hpp"

namespace urde
{

CMetaAnimPlay::CMetaAnimPlay(CInputStream& in)
: x4_primitive(in), x1c_startTime(in) {}

void CMetaAnimPlay::GetUniquePrimitives(std::set<CPrimitive>& primsOut) const
{
    primsOut.insert(x4_primitive);
}

std::shared_ptr<CAnimTreeNode>
CMetaAnimPlay::VGetAnimationTree(const CAnimSysContext& animSys,
                                 const CMetaAnimTreeBuildOrders& orders) const
{
    if (orders.x0_recursiveAdvance)
    {
        CMetaAnimTreeBuildOrders modOrders;
        modOrders.PreAdvanceForAll(*orders.x0_recursiveAdvance);
        return GetAnimationTree(animSys, modOrders);
    }

    TLockedToken<CAllFormatsAnimSource> prim =
        animSys.xc_store.GetObj(SObjectTag{FOURCC('ANIM'), x4_primitive.GetAnimResId()});
    std::shared_ptr<CAnimTreeNode> ret =
        std::make_shared<CAnimTreeAnimReaderContainer>(x4_primitive.GetName(),
                                                       CAllFormatsAnimSource::GetNewReader(prim, x1c_startTime),
                                                       x4_primitive.GetAnimDbIdx());
    return ret;
}

}