CollectionCurveFacade.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "GeometryCollection/ManagedArrayAccessor.h"
#include "GeometryCollection/ManagedArrayCollection.h"
#include "GeometryCollection/GeometryCollection.h"
    
namespace GeometryCollection::Facades
{
    class FCollectionCurveGeometryFacade
    {
    public:
        static CHAOS_API const FName CurvesGroup;
        static CHAOS_API const FName PointsGroup;
        
        static CHAOS_API const FName CurvePointOffsetsAttribute;
        static CHAOS_API const FName GeometryCurveOffsetsAttribute;
        static CHAOS_API const FName PointRestPositionsAttribute;
        static CHAOS_API const FName PointRestOrientationsAttribute;
        static CHAOS_API const FName PointCurveIndicesAttribute;
        static CHAOS_API const FName CurveGeometryIndicesAttribute;
        static CHAOS_API const FName GeometryGroupNamesAttribute;
        static CHAOS_API const FName GeometryCurveThicknessAttribute;
        static CHAOS_API const FName CurveSourceIndicesAttribute;
 
        CHAOS_API FCollectionCurveGeometryFacade(FManagedArrayCollection& InCollection);
        CHAOS_API FCollectionCurveGeometryFacade(const FManagedArrayCollection& InCollection);
 
        CHAOS_API void DefineSchema();
 
        bool IsConst() const { return Collection == nullptr; }
 
        CHAOS_API bool IsValid() const;
        
        int32 GetNumCurves() const { return CurvePointOffsets.Num(); }
 
        int32 GetNumGeometry() const { return GeometryCurveOffsets.Num(); }
 
        int32 GetNumPoints() const { return PointRestPositions.Num(); }
        
        const TArray<FVector3f>& GetPointRestPositions() const { return PointRestPositions.Get().GetConstArray(); }
 
        const TArray<FQuat4f>& GetPointRestOrientations() const { return PointRestOrientations.Get().GetConstArray(); }
 
        const TArray<int32>& GetCurvePointOffsets() const { return CurvePointOffsets.Get().GetConstArray(); }
 
        const TArray<int32>& GetGeometryCurveOffsets() const { return GeometryCurveOffsets.Get().GetConstArray(); }
 
        const TArray<int32>& GetPointCurveIndices() const { return PointCurveIndices.Get().GetConstArray(); }
 
        const TArray<int32>& GetCurveGeometryIndices() const { return CurveGeometryIndices.Get().GetConstArray(); }
 
        const TArray<FString>& GetGeometryGroupNames() const { return GeometryGroupNames.Get().GetConstArray(); }
 
        const TArray<float>& GetGeometryCurveThickness() const { return GeometryCurveThickness.Get().GetConstArray(); }
 
        const TArray<int32>& GetCurveSourceIndices() const { return CurveSourceIndices.Get().GetConstArray(); }
        
        void SetPointRestPositions(const TArray<FVector3f>& InPointRestPositions) { PointRestPositions.Modify() = InPointRestPositions; UpdatePointRestOrientations(); }    
 
        void SetGeometryGroupNames(const TArray<FString>& InGeometryGroupNames) { GeometryGroupNames.Modify() = InGeometryGroupNames; }
        
        void SetGeometryCurveThickness(const TArray<float>& InGeometryCurveThickness) { GeometryCurveThickness.Modify() = InGeometryCurveThickness; }   
        
        void SetCurvePointOffsets(const TArray<int32>& InCurvePointOffsets) { CurvePointOffsets.Modify() = InCurvePointOffsets; UpdatePointCurveIndices(); }
 
        void SetGeometryCurveOffsets(const TArray<int32>& InGeometryCurveOffsets) { GeometryCurveOffsets.Modify() = InGeometryCurveOffsets; UpdateCurveGeometryIndices(); }
 
        void SetCurveSourceIndices(const TArray<int32>& InCurveSourceIndices) { CurveSourceIndices.Modify() = InCurveSourceIndices; }
        
        CHAOS_API void InitCurvesCollection(const TArray<FVector3f>& InPointRestPositions, const TArray<int32>& InCurvePointOffsets, const TArray<int32>& InGeometryCurveOffsets,
            const TArray<FString>& InGeometryGroupNames, const TArray<float>& InGeometryCurveThickness, const TArray<int32>& InCurveSourceIndices);
 
        const FManagedArrayCollection& GetManagedArrayCollection() const {return ConstCollection;}
        
    protected :
 
        void BuildGeometryCollection();
        
        void UpdatePointRestOrientations();
        
        void UpdatePointCurveIndices();
 
        void UpdateCurveGeometryIndices();
        
        const FManagedArrayCollection& ConstCollection;
 
        FManagedArrayCollection* Collection = nullptr;
        
        TManagedArrayAccessor<FQuat4f> PointRestOrientations;
 
        TManagedArrayAccessor<FVector3f> PointRestPositions;
 
        TManagedArrayAccessor<int32> CurvePointOffsets;
                                                 
        TManagedArrayAccessor<int32> GeometryCurveOffsets;
 
        TManagedArrayAccessor<int32> PointCurveIndices;
 
        TManagedArrayAccessor<int32> CurveGeometryIndices;
 
        TManagedArrayAccessor<FString> GeometryGroupNames;
 
        TManagedArrayAccessor<float> GeometryCurveThickness;
 
        TManagedArrayAccessor<int32> CurveSourceIndices;
    };
 
    class FCollectionCurveHierarchyFacade
    {
    public:
 
        static CHAOS_API const FName CurveParentIndicesAttribute;
        static CHAOS_API const FName CurveLodIndicesAttribute;
 
        CHAOS_API FCollectionCurveHierarchyFacade(FManagedArrayCollection& InCollection);
        CHAOS_API FCollectionCurveHierarchyFacade(const FManagedArrayCollection& InCollection);
 
        CHAOS_API void DefineSchema();
 
        bool IsConst() const { return Collection == nullptr; }
 
        CHAOS_API bool IsValid() const;
        
        int32 GetNumCurves() const { return CurveLodIndices.Num(); }
        
        const TArray<int32>& GetCurveLodIndices() const { return CurveLodIndices.Get().GetConstArray(); }
 
        const TArray<int32>& GetCurveParentIndices() const { return CurveParentIndices.Get().GetConstArray(); }
        
        void SetCurveLodIndices(const TArray<int32>& InCurveLodIndices) { CurveLodIndices.Modify() = InCurveLodIndices; }
 
        void SetCurveParentIndices(const TArray<int32>& InCurveParentIndices) { CurveParentIndices.Modify() = InCurveParentIndices; }
 
        const FManagedArrayCollection& GetManagedArrayCollection() const {return ConstCollection;}
        
    protected :
 
        void BuildGeometryCollection();
        
        void UpdatePointRestOrientations();
        
        void UpdatePointCurveIndices();
 
        void UpdateCurveGeometryIndices();
        
        const FManagedArrayCollection& ConstCollection;
 
        FManagedArrayCollection* Collection = nullptr;
 
        TManagedArrayAccessor<int32> CurveParentIndices;
 
        TManagedArrayAccessor<int32> CurveLodIndices;
    };
}