NaniteAssemblyData.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "UObject/ObjectMacros.h"
#include "UObject/UObjectGlobals.h"
#include "UObject/Object.h"
 
#include "NaniteAssemblyData.generated.h"
 
class UStaticMesh;
 
USTRUCT(BlueprintType)
struct FNaniteAssemblyBoneInfluence
{
    GENERATED_USTRUCT_BODY()
 
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = Default)
    int32 BoneIndex = INDEX_NONE;
 
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = Default)
    float BoneWeight = 1.0f;
 
    bool operator==(const FNaniteAssemblyBoneInfluence& Other) const
    {
        return BoneIndex == Other.BoneIndex && FMath::IsNearlyEqual(BoneWeight, Other.BoneWeight);
    }
    bool operator!=(const FNaniteAssemblyBoneInfluence& Other) const { return !(*this == Other); }
};
 
UENUM(BlueprintType)
enum class ENaniteAssemblyNodeTransformSpace : uint8
{
    // Local (mesh) space
    Local,
    // Relative to the bone influences
    BoneRelative
};
 
USTRUCT()
struct FNaniteAssemblyNode
{
    GENERATED_USTRUCT_BODY()
    
    
    UPROPERTY()
    int32 PartIndex = INDEX_NONE;
 
    UPROPERTY()
    ENaniteAssemblyNodeTransformSpace TransformSpace = ENaniteAssemblyNodeTransformSpace::Local;
 
    UPROPERTY()
    FTransform3f Transform = FTransform3f::Identity;
 
    UPROPERTY()
    TArray<FNaniteAssemblyBoneInfluence> BoneInfluences;
 
    bool operator==(const FNaniteAssemblyNode& Other) const
    {
        return PartIndex == Other.PartIndex &&
            TransformSpace == Other.TransformSpace &&
            Transform.Equals(Other.Transform) &&
            BoneInfluences == Other.BoneInfluences;
    }
    bool operator!=(const FNaniteAssemblyNode& Other) const { return !(*this == Other); }
};
 
USTRUCT()
struct FNaniteAssemblyPart
{
    GENERATED_USTRUCT_BODY()
 
    
    UPROPERTY(VisibleAnywhere, Category = General)
    FSoftObjectPath MeshObjectPath;
 
    UPROPERTY()
    TArray<int32> MaterialRemap;
 
    bool operator==(const FNaniteAssemblyPart& Other) const
    {
        return MeshObjectPath == Other.MeshObjectPath &&
            MaterialRemap == Other.MaterialRemap;
    }
    bool operator!=(const FNaniteAssemblyPart& Other) const { return !(*this == Other); }
};
 
USTRUCT()
struct FNaniteAssemblyData
{
    GENERATED_USTRUCT_BODY()
 
    
    UPROPERTY(VisibleAnywhere, Category = General)
    TArray<FNaniteAssemblyPart> Parts;
 
    UPROPERTY()
    TArray<FNaniteAssemblyNode> Nodes;
 
    const bool IsValid() const { return Parts.Num() > 0 && Nodes.Num() > 0; }
    bool operator==(const FNaniteAssemblyData& Other) const {  return Parts == Other.Parts && Nodes == Other.Nodes; }
    bool operator!=(const FNaniteAssemblyData& Other) const { return !(*this == Other); }
 
    void SerializeForDDC(FArchive& Ar)
    {
        bool bValid = IsValid();
        Ar << bValid;
        if (bValid)
        {
            for (FNaniteAssemblyPart& Part : Parts)
            {
                FString Path = Part.MeshObjectPath.ToString();
                Ar << Path;
                Ar << Part.MaterialRemap;
            }
            Ar << Nodes;
        }
 
    }
};
 
inline FArchive& operator<<(FArchive& Ar, FNaniteAssemblyBoneInfluence& Influence)
{
    Ar << Influence.BoneIndex;
    Ar << Influence.BoneWeight;
 
    return Ar;
}
 
inline FArchive& operator<<(FArchive& Ar, FNaniteAssemblyNode& Node)
{
    Ar << Node.PartIndex;
    Ar << Node.TransformSpace;
    Ar << Node.Transform;
    Ar << Node.BoneInfluences;
 
    return Ar;
}
 
inline FArchive & operator<<(FArchive& Ar, FNaniteAssemblyPart& Part)
{
    Ar << Part.MeshObjectPath;
    Ar << Part.MaterialRemap;
 
    return Ar;
}
 
inline FArchive& operator<<(FArchive& Ar, FNaniteAssemblyData& Data)
{
    Ar << Data.Parts;
    Ar << Data.Nodes;
    
    return Ar;
}