o3de/Gems/EMotionFX/Code/Tests/BlendTreeMotionFrameNodeTests.cpp

/*
 * 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
 *
 */

#include <Tests/AnimGraphFixture.h>
#include <EMotionFX/Source/AnimGraph.h>
#include <EMotionFX/Source/AnimGraphMotionNode.h>
#include <EMotionFX/Source/AnimGraphStateMachine.h>
#include <EMotionFX/Source/BlendTree.h>
#include <EMotionFX/Source/BlendTreeMotionFrameNode.h>
#include <EMotionFX/Source/Motion.h>
#include <EMotionFX/Source/MotionSet.h>
#include <EMotionFX/Source/MotionData/NonUniformMotionData.h>

namespace EMotionFX
{
    class BlendTreeMotionFrameTests
        : public AnimGraphFixture
    {
    public:
        void ConstructGraph() override
        {
            AnimGraphFixture::ConstructGraph();
            m_blendTreeAnimGraph = AnimGraphFactory::Create<OneBlendTreeNodeAnimGraph>();
            m_rootStateMachine = m_blendTreeAnimGraph->GetRootStateMachine();
            m_blendTree = m_blendTreeAnimGraph->GetBlendTreeNode();

            /*
            +--------+    +--------------+    +------------+
            | Motion +--->+ Motion Frame +--->+ Final Node |
            +--------+    +--------------+    +------------+
            */

            m_motionFrameNode = aznew BlendTreeMotionFrameNode();
            m_blendTree->AddChildNode(m_motionFrameNode);
            BlendTreeFinalNode* finalNode = aznew BlendTreeFinalNode();
            m_blendTree->AddChildNode(finalNode);
            finalNode->AddConnection(m_motionFrameNode, BlendTreeMotionFrameNode::OUTPUTPORT_RESULT, BlendTreeFinalNode::PORTID_INPUT_POSE);

            m_motionNode = aznew AnimGraphMotionNode();
            m_blendTree->AddChildNode(m_motionNode);
            m_motionFrameNode->AddConnection(m_motionNode, AnimGraphMotionNode::PORTID_OUTPUT_MOTION, BlendTreeMotionFrameNode::INPUTPORT_MOTION);
            m_blendTreeAnimGraph->InitAfterLoading();
        }

        void SetUp() override
        {
            AnimGraphFixture::SetUp();
            m_animGraphInstance->Destroy();
            m_animGraphInstance = m_blendTreeAnimGraph->GetAnimGraphInstance(m_actorInstance, m_motionSet);

            const char* motionId = "Test Motion";
            Motion* motion = aznew Motion(motionId);
            motion->SetMotionData(aznew NonUniformMotionData());
            motion->GetMotionData()->SetDuration(m_motionDuration);
            MotionSet::MotionEntry* motionEntry = aznew MotionSet::MotionEntry(motionId, motionId, motion);
            m_motionSet->AddMotionEntry(motionEntry);
            m_motionNode->AddMotionId(motionId);

            m_motionNode->RecursiveOnChangeMotionSet(m_animGraphInstance, m_motionSet);
            m_motionNode->PickNewActiveMotion(m_animGraphInstance, static_cast<AnimGraphMotionNode::UniqueData*>(m_motionNode->FindOrCreateUniqueNodeData(m_animGraphInstance)));

            // The motion node creates its motion instance in the Output() call, which means that at the first
            // evaluation of the motion frame node the duration is 0.0 (due to the missing motion instance) and
            // that results in the normalized time also being 0.0 at that frame which we need for testing.
            GetEMotionFX().Update(0.0f);
        }

        void SetAndTestTimeValue(BlendTreeMotionFrameNode::UniqueData* uniqueData, float newNormalizedTime, bool rewind = false)
        {
            const float prevNewTime = uniqueData->m_newTime;
            m_motionFrameNode->SetNormalizedTimeValue(newNormalizedTime);

            if (rewind)
            {
                m_motionFrameNode->Rewind(m_animGraphInstance);
            }
            GetEMotionFX().Update(0.0f);

            EXPECT_EQ(m_motionFrameNode->GetNormalizedTimeValue(), newNormalizedTime);
            
            float expectedOldTime = prevNewTime;
            if (rewind)
            {
                if (m_motionFrameNode->GetEmitEventsFromStart())
                {
                    expectedOldTime = 0.0f;
                }
                else
                {
                    expectedOldTime = newNormalizedTime * m_motionDuration;
                }
            }
            EXPECT_EQ(uniqueData->m_oldTime, expectedOldTime);
            
            EXPECT_EQ(uniqueData->m_newTime, newNormalizedTime * m_motionDuration);
        }

    public:
        AZStd::unique_ptr<OneBlendTreeNodeAnimGraph> m_blendTreeAnimGraph;
        float m_motionDuration = 1.0f;
        AnimGraphMotionNode* m_motionNode = nullptr;
        BlendTreeMotionFrameNode* m_motionFrameNode = nullptr;
        BlendTreeMotionFrameNode::UniqueData* m_motionFrameNodeUniqueData = nullptr;
        BlendTree* m_blendTree = nullptr;
    };

    TEST_F(BlendTreeMotionFrameTests, SetNormalizedTime)
    {
        BlendTreeMotionFrameNode::UniqueData* uniqueData = static_cast<BlendTreeMotionFrameNode::UniqueData*>(m_motionFrameNode->FindOrCreateUniqueNodeData(m_animGraphInstance));
        SetAndTestTimeValue(uniqueData, 0.2f);
        SetAndTestTimeValue(uniqueData, 0.4f);
        SetAndTestTimeValue(uniqueData, 0.3f);
        SetAndTestTimeValue(uniqueData, 1.0f);
        SetAndTestTimeValue(uniqueData, 0.0f);
    }

    TEST_F(BlendTreeMotionFrameTests, RewindTest)
    {
        BlendTreeMotionFrameNode::UniqueData* uniqueData = static_cast<BlendTreeMotionFrameNode::UniqueData*>(m_motionFrameNode->FindOrCreateUniqueNodeData(m_animGraphInstance));
        SetAndTestTimeValue(uniqueData, 0.2f);
        SetAndTestTimeValue(uniqueData, 0.4f, true);
    }

    TEST_F(BlendTreeMotionFrameTests, RewindTest_SetEmitEventsFromStart)
    {
        BlendTreeMotionFrameNode::UniqueData* uniqueData = static_cast<BlendTreeMotionFrameNode::UniqueData*>(m_motionFrameNode->FindOrCreateUniqueNodeData(m_animGraphInstance));
        m_motionFrameNode->SetEmitEventsFromStart(true);
        SetAndTestTimeValue(uniqueData, 0.2f);
        SetAndTestTimeValue(uniqueData, 0.4f, true);
    }
} // namespace EMotionFX