/*
* 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_CRYANIMATION_FACIALANIMATION_FACIALMODEL_H
#define CRYINCLUDE_CRYANIMATION_FACIALANIMATION_FACIALMODEL_H
#pragma once

#include "FaceEffector.h"
#include "CryArray.h"
#include "Skeleton.h"

class CDefaultSkeleton;
class CFacialEffectorsLibrary;
class CCharInstance;
class CFaceState;
class CryModEffMorph;
class CSkeletonPose;


class CFacialDisplaceInfo
{
public:
    CFacialDisplaceInfo(const uint32 count = 0)
        : m_displaceInfo(count, IDENTITY)
        , m_used(count, 0)
        , m_hasUsed(false)
    {
    }

    void Initialize(const size_t count)
    {
        m_displaceInfo.clear();
        m_displaceInfo.resize(count, IDENTITY);

        m_used.clear();
        m_used.resize(count, 0);

        m_hasUsed = false;
    }

    void CopyUsed(const CFacialDisplaceInfo& other)
    {
        m_used = other.m_used;
        m_hasUsed = other.m_hasUsed;
    }

    size_t GetCount() const { return m_displaceInfo.size(); }

    bool IsUsed(const size_t i) const
    {
        CryPrefetch((void*)&m_displaceInfo[i]);
        return (m_used[i] == 1);
    }

    bool HasUsed() const { return m_hasUsed; }

    const QuatT& GetDisplaceInfo(const size_t i) const
    {
        CryPrefetch((void*)&(m_used[i]));
        return m_displaceInfo[i];
    }

    void SetDisplaceInfo(const size_t i, const QuatT& q)
    {
        m_displaceInfo[i] = q;
        m_used[i] = 1;
        m_hasUsed = true;
    }

    void Clear()
    {
        const size_t count = m_displaceInfo.size();
        Initialize(count);
    }

    void ClearFast()
    {
        m_hasUsed = false;
    }

private:
    std::vector<QuatT> m_displaceInfo;
    std::vector<uint32> m_used;
    bool m_hasUsed;
};

//////////////////////////////////////////////////////////////////////////
class CFacialModel
    : public IFacialModel
    , public _i_reference_target_t
{
public:
    struct SMorphTargetInfo
    {
        CFaceIdentifierStorage name;
        int nMorphTargetId;
        bool operator<(const SMorphTargetInfo& m) const { return name.GetCRC32() < m.name.GetCRC32(); }
        bool operator>(const SMorphTargetInfo& m) const { return name.GetCRC32() > m.name.GetCRC32(); }
        bool operator==(const SMorphTargetInfo& m) const { return name.GetCRC32() == m.name.GetCRC32(); }
        bool operator!=(const SMorphTargetInfo& m) const { return name.GetCRC32() != m.name.GetCRC32(); }
        void GetMemoryUsage(ICrySizer* pSizer) const
        {
            pSizer->AddObject(name);
        }
    };




    typedef std::vector<SMorphTargetInfo> MorphTargetsInfo;

    struct SEffectorInfo
    {
        _smart_ptr<CFacialEffector> pEffector;
        string name;        // morph target name.
        int nMorphTargetId; // positive if effector is a morph target.
        void GetMemoryUsage(ICrySizer* pSizer) const
        {
            pSizer->AddObject(name);
        }
    };

    CFacialModel(CDefaultSkeleton* pDefaultSkeleton);

    size_t SizeOfThis();
    void GetMemoryUsage(ICrySizer* pSizer) const;

    //////////////////////////////////////////////////////////////////////////
    // IFacialModel interface
    //////////////////////////////////////////////////////////////////////////
    int GetEffectorCount() const { return m_effectors.size(); }
    CFacialEffector* GetEffector(int nIndex) const { return m_effectors[nIndex].pEffector; }
    virtual void AssignLibrary(IFacialEffectorsLibrary* pLibrary);
    virtual IFacialEffectorsLibrary* GetLibrary();
    virtual int GetMorphTargetCount() const;
    virtual const char* GetMorphTargetName(int morphTargetIndex) const;
    //////////////////////////////////////////////////////////////////////////

    // Creates a new FaceState.
    CFaceState* CreateState();

    SEffectorInfo* GetEffectorsInfo() { return &m_effectors[0]; }
    CFacialEffectorsLibrary* GetFacialEffectorsLibrary() {return m_pEffectorsLibrary; }

    //////////////////////////////////////////////////////////////////////////
    // Apply state to the specified character instance.
    void FillInfoMapFromFaceState(CFacialDisplaceInfo& info, CFaceState* const pFaceState, CCharInstance* const pInstance, int numForcedRotations, const CFacialAnimForcedRotationEntry* const forcedRotations, float blendBoneRotations);

    static void ApplyDisplaceInfoToJoints(const CFacialDisplaceInfo& info, const CDefaultSkeleton* const pSkeleton, Skeleton::CPoseData* const pPoseData, const QuatT* const pJointsRelativeDefault);

    static void ClearResources();
private:
    void InitMorphTargets();

    void TransformTranslations(CFacialDisplaceInfo& info, const CCharInstance* const pInstance);
    int32 ApplyBoneEffector(CCharInstance* pInstance, CFacialEffector* pEffector, float fWeight, QuatT& quatT);
    //  void HandleLookIK(FacialDisplaceInfoMap& Info, CLookAt& lookIK, bool physicsRelinquished, int32 head, int32 leye, int32 reye);
    void PushMorphEffectors(int i, const MorphTargetsInfo& morphTargets, f32 fWeight, f32 fBalance, DynArray<CryModEffMorph>& arrMorphEffectors);


private:
    // Array of facial effectors that apply to this model.
    std::vector<SEffectorInfo> m_effectors;

    _smart_ptr<CFacialEffectorsLibrary> m_pEffectorsLibrary;

    // Maps indices from the face state into the morph target id.
    std::vector<int> m_faceStateIndexToMorphTargetId;
    MorphTargetsInfo m_morphTargets;

    // Character model who owns this facial model.
    CDefaultSkeleton* m_pDefaultSkeleton;

    bool m_bAttachmentChanged;

    static StaticInstance<CFacialDisplaceInfo> s_boneInfoBlending;
};

#endif // CRYINCLUDE_CRYANIMATION_FACIALANIMATION_FACIALMODEL_H