AnimDataModelNotifyCollector.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Animation/AnimData/AnimDataNotifications.h"
#include "Containers/Set.h"
#include "Containers/Array.h"
 
enum class EAnimDataModelNotifyType : uint8;
 
namespace UE {
namespace Anim {
 
#if WITH_EDITORONLY_DATA
 
struct FAnimDataModelNotifyCollector
{
    FAnimDataModelNotifyCollector() : BracketDepth(0), bDataModified(false) {}
 
    void Handle(EAnimDataModelNotifyType NotifyType)
    {
        if (BracketDepth == 0)
        {
            Reset();
            bDataModified = false;
        }
        
        NotifyTypes.Add(NotifyType);
 
        if (NotifyType == EAnimDataModelNotifyType::BracketOpened)
        {
            ++BracketDepth;
        }
        else if (NotifyType == EAnimDataModelNotifyType::BracketClosed)
        {
            --BracketDepth;
        }
    }
    
    bool Contains(EAnimDataModelNotifyType NotifyType) const
    {
        return NotifyTypes.Find(NotifyType) != nullptr;
    }
 
    bool Contains(const TArray<EAnimDataModelNotifyType>& TestNotifyTypes) const
    {
        for (EAnimDataModelNotifyType Notify : TestNotifyTypes)
        {
            if (NotifyTypes.Find(Notify) != nullptr)
            {
                return true;
            }
        }
 
        return false;
    }
 
    bool IsWithinBracket() const { return BracketDepth > 0; }
 
    bool IsNotWithinBracket() const { return BracketDepth == 0; }
 
    void MarkDataModified() { bDataModified = true; }
    bool WasDataModified() { return bDataModified; }
protected:
    void Reset()
    {
        NotifyTypes.Empty();
    }
protected:
    TSet<EAnimDataModelNotifyType> NotifyTypes;
    int32 BracketDepth;
    bool bDataModified;
};
 
#endif // WITH_EDITORONLY_DATA
 
} // namespace Anim 
 
} // namespace UE