CollectionTransformFacade.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "GeometryCollection/ManagedArrayAccessor.h"
#include "GeometryCollection/ManagedArrayCollection.h"
 
namespace GeometryCollection::Facades
{
    class FCollectionTransformFacade
    {
    public:
        CHAOS_API FCollectionTransformFacade(FManagedArrayCollection& InCollection);
        CHAOS_API FCollectionTransformFacade(const FManagedArrayCollection& InCollection);
 
        void DefineSchema() {}
 
        CHAOS_API bool IsValid() const;
 
        bool IsConst() const { return ParentAttribute.IsConst(); }
 
        CHAOS_API int32 Num() const;
 
 
        CHAOS_API TArray<int32> GetRootIndices() const;
 
        CHAOS_API FTransform GetRootTransform() const;
 
        const TManagedArray< int32 >* GetParents() const { return ParentAttribute.Find(); }
 
        const TManagedArray< TSet<int32> >* FindChildren() const { return ChildrenAttribute.Find(); }
 
        const TManagedArray<FTransform3f>* FindTransforms() const { return TransformAttribute.Find(); }
 
        const TManagedArray<FString>* FindBoneNames() const { return BoneNameAttribute.Find(); }
 
        CHAOS_API TArray<FTransform> ComputeCollectionSpaceTransforms() const;
 
        CHAOS_API FTransform ComputeCollectionSpaceTransform(int32 BoneIdx) const;
 
        CHAOS_API void SetPivot(const FTransform& InTransform);
 
        CHAOS_API void Transform(const FTransform& InTransform);
 
        CHAOS_API void Transform(const FTransform& InTransform, const TArray<int32>& InSelection);
 
        CHAOS_API bool HasBoneNameAttribute() const;
 
        CHAOS_API FString BoneName(int32 Index) const;
 
        CHAOS_API TMap<FString, int32> BoneNameIndexMap() const;
 
        static CHAOS_API FMatrix BuildMatrix(const FVector& Translate,
            const uint8 RotationOrder,
            const FVector& Rotate,
            const FVector& Scale,
            const FVector& Shear,
            const float UniformScale,
            const FVector& RotatePivot,
            const FVector& ScalePivot,
            const bool InvertTransformation);
 
        static CHAOS_API FTransform BuildTransform(const FVector& Translate,
            const uint8 RotationOrder,
            const FVector& Rotate,
            const FVector& Scale,
            const float UniformScale,
            const FVector& RotatePivot,
            const FVector& ScalePivot,
            const bool InvertTransformation);
 
        CHAOS_API void SetBoneTransformToIdentity(int32 BoneIdx);
 
        static CHAOS_API bool HasCycle(const TManagedArray<int32>& Parents, int32 Node);
 
        static CHAOS_API bool HasCycle(const TManagedArray<int32>& Parents, const TArray<int32>& SelectedBones);
 
        CHAOS_API void ParentTransform(const int32 TransformIndex, const int32 ChildIndex);
 
        CHAOS_API void ParentTransforms(const int32 TransformIndex, const TArray<int32>& SelectedBones);
 
        CHAOS_API void UnparentTransform(const int32 ChildIndex);
 
        CHAOS_API void EnforceSingleRoot(const FString & RootName);
 
        CHAOS_API void ComputeLevelsBounds(float& LevelsMin, float& LevelsMax);
 
    private:
        const FManagedArrayCollection& ConstCollection;
        FManagedArrayCollection* Collection = nullptr;
 
        TManagedArrayAccessor<int32>        ParentAttribute;
        TManagedArrayAccessor<TSet<int32>>  ChildrenAttribute;
        TManagedArrayAccessor<FTransform3f> TransformAttribute;
        TManagedArrayAccessor<FString>      BoneNameAttribute;
    };
}