o3de/Gems/EMotionFX/Code/Source/Integration/Assets/AnimGraphAsset.h

/*
 * Copyright (c) Contributors to the Open 3D Engine Project.
 * For complete copyright and license terms please see the LICENSE at the root of this distribution.
 *
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */


#pragma once

#include <Integration/Assets/AssetCommon.h>

namespace EMotionFX
{
    class AnimGraph;
    class AnimGraphInstance;
    class ActorInstance;
    class MotionSet;

    namespace Integration
    {
        class ActorAsset;

        class AnimGraphAsset 
            : public EMotionFXAsset
        {
        public:

            friend class AnimGraphAssetHandler;

            AZ_CLASS_ALLOCATOR_DECL
            AZ_RTTI(AnimGraphAsset, "{28003359-4A29-41AE-8198-0AEFE9FF5263}", EMotionFXAsset);

            AnimGraphAsset(AZ::Data::AssetId id = AZ::Data::AssetId());

            typedef EMotionFXPtr<EMotionFX::AnimGraphInstance> AnimGraphInstancePtr;
            AnimGraphInstancePtr CreateInstance(
                EMotionFX::ActorInstance* actorInstance,
                EMotionFX::MotionSet* motionSet);

            EMotionFX::AnimGraph* GetAnimGraph() { return m_emfxAnimGraph ? m_emfxAnimGraph.get() : nullptr; }

            void SetData(EMotionFX::AnimGraph* animGraph);

            void SetStatus(AssetStatus newStatus) { m_status = newStatus; }

        private:

            AZStd::unique_ptr<EMotionFX::AnimGraph> m_emfxAnimGraph;
        };

        class AnimGraphAssetHandler : public EMotionFXAssetHandler<AnimGraphAsset>
        {
        public:
            AZ_CLASS_ALLOCATOR_DECL

            bool OnInitAsset(const AZ::Data::Asset<AZ::Data::AssetData>& asset) override final;
            void DestroyAsset(AZ::Data::AssetPtr ptr) override final;
            AZ::Data::AssetType GetAssetType() const override final;
            void GetAssetTypeExtensions(AZStd::vector<AZStd::string>& extensions) override final;
            AZ::Uuid GetComponentTypeId() const override final;
            const char* GetAssetTypeDisplayName() const override final;
            const char* GetBrowserIcon() const override;
        };

        class AnimGraphAssetBuilderHandler : public AnimGraphAssetHandler
        {
        public:
            void InitAsset(const AZ::Data::Asset<AZ::Data::AssetData>& asset, bool loadStageSucceeded, bool isReload) override;
            AZ::Data::AssetHandler::LoadResult LoadAssetData(
                const AZ::Data::Asset<AZ::Data::AssetData>& asset,
                AZStd::shared_ptr<AZ::Data::AssetDataStream> stream,
                const AZ::Data::AssetFilterCB& assetLoadFilterCB) override;
        };
    } // namespace Integration
} // namespace EMotionFX

namespace AZ
{
    AZ_TYPE_INFO_SPECIALIZE(EMotionFX::Integration::EMotionFXPtr<EMotionFX::Integration::AnimGraphAsset>, "{BF1ACFB9-8295-4B55-8B55-DC64BFF36BD3}");
    AZ_TYPE_INFO_SPECIALIZE(EMotionFX::Integration::EMotionFXPtr<EMotionFX::AnimGraphInstance>, "{769ED685-EC18-449D-9453-7D47D9BC1B8A}");
}