Files
UnrealEngine/Engine/Plugins/Mutable/Source/MutableRuntime/Private/MuR/Instance.cpp
2025-05-18 13:04:45 +08:00

595 lines
21 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#include "MuR/Instance.h"
#include "HAL/LowLevelMemTracker.h"
#include "Misc/AssertionMacros.h"
#include "MuR/InstancePrivate.h"
#include "MuR/MutableMath.h"
namespace mu
{
//---------------------------------------------------------------------------------------------
//---------------------------------------------------------------------------------------------
//---------------------------------------------------------------------------------------------
FInstance::FInstance()
{
LLM_SCOPE_BYNAME(TEXT("MutableRuntime"));
m_pD = new Private();
}
//---------------------------------------------------------------------------------------------
FInstance::~FInstance()
{
LLM_SCOPE_BYNAME(TEXT("MutableRuntime"));
check( m_pD );
delete m_pD;
m_pD = nullptr;
}
//---------------------------------------------------------------------------------------------
FInstance::Private* FInstance::GetPrivate() const
{
return m_pD;
}
//---------------------------------------------------------------------------------------------
TSharedPtr<FInstance> FInstance::Clone() const
{
LLM_SCOPE_BYNAME(TEXT("MutableRuntime"));
TSharedPtr<FInstance> Result = MakeShared<FInstance>();
*Result->GetPrivate() = *m_pD;
return Result;
}
//---------------------------------------------------------------------------------------------
int32 FInstance::GetDataSize() const
{
return 16 + sizeof(Private) + m_pD->Components.GetAllocatedSize() + m_pD->ExtensionData.GetAllocatedSize();
}
//---------------------------------------------------------------------------------------------
FInstance::FID FInstance::GetId() const
{
return m_pD->Id;
}
//---------------------------------------------------------------------------------------------
int32 FInstance::GetComponentCount() const
{
return m_pD->Components.Num();
}
//---------------------------------------------------------------------------------------------
int32 FInstance::GetLODCount( int32 ComponentIndex ) const
{
if (m_pD->Components.IsValidIndex(ComponentIndex))
{
return m_pD->Components[ComponentIndex].LODs.Num();
}
check(false);
return 0;
}
//---------------------------------------------------------------------------------------------
uint16 FInstance::GetComponentId( int32 ComponentIndex ) const
{
if (m_pD->Components.IsValidIndex(ComponentIndex))
{
return m_pD->Components[ComponentIndex].Id;
}
else
{
check(false);
}
return 0;
}
//---------------------------------------------------------------------------------------------
int32 FInstance::GetSurfaceCount( int32 ComponentIndex, int32 LODIndex ) const
{
if (m_pD->Components.IsValidIndex(ComponentIndex) &&
m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex))
{
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.Num();
}
else
{
check(false);
}
return 0;
}
//---------------------------------------------------------------------------------------------
uint32 FInstance::GetSurfaceId(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex ) const
{
if (m_pD->Components.IsValidIndex(ComponentIndex) &&
m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex) &&
m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex) )
{
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].InternalId;
}
else
{
check(false);
}
return 0;
}
//---------------------------------------------------------------------------------------------
int32 FInstance::FindSurfaceById(int32 ComponentIndex, int32 LODIndex, uint32 id ) const
{
if (m_pD->Components.IsValidIndex(ComponentIndex) &&
m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex))
{
for (int32 i = 0; i < m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.Num(); ++i)
{
if (m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[i].InternalId == id)
{
return i;
}
}
}
else
{
check(false);
}
return -1;
}
//---------------------------------------------------------------------------------------------
void FInstance::FindBaseSurfaceBySharedId(int32 CompIndex, int32 SharedId, int32& OutSurfaceIndex, int32& OutLODIndex) const
{
if (m_pD->Components.IsValidIndex(CompIndex))
{
for (int32 LodIndex = 0; LodIndex < m_pD->Components[CompIndex].LODs.Num(); LodIndex++)
{
FInstanceLOD& LOD = m_pD->Components[CompIndex].LODs[LodIndex];
for (int32 SurfaceIndex = 0; SurfaceIndex < LOD.Surfaces.Num(); ++SurfaceIndex)
{
if (LOD.Surfaces[SurfaceIndex].SharedId == SharedId)
{
OutSurfaceIndex = SurfaceIndex;
OutLODIndex = LodIndex;
return;
}
}
}
}
OutSurfaceIndex = INDEX_NONE;
OutLODIndex = INDEX_NONE;
}
//---------------------------------------------------------------------------------------------
int32 FInstance::GetSharedSurfaceId(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex) const
{
if (m_pD->Components.IsValidIndex(ComponentIndex) &&
m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex) &&
m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex))
{
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].SharedId;
}
else
{
check(false);
}
return 0;
}
//---------------------------------------------------------------------------------------------
uint32 FInstance::GetSurfaceCustomId(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex ) const
{
if (m_pD->Components.IsValidIndex(ComponentIndex) &&
m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex) &&
m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex))
{
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].ExternalId;
}
else
{
check(false);
}
return 0;
}
//---------------------------------------------------------------------------------------------
int32 FInstance::GetImageCount(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex ) const
{
check(m_pD->Components.IsValidIndex(ComponentIndex));
check(m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex));
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Images.Num();
}
//---------------------------------------------------------------------------------------------
int32 FInstance::GetVectorCount(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex ) const
{
check(m_pD->Components.IsValidIndex(ComponentIndex));
check(m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex));
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Vectors.Num();
}
//---------------------------------------------------------------------------------------------
int32 FInstance::GetScalarCount(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex ) const
{
check(m_pD->Components.IsValidIndex(ComponentIndex));
check(m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex));
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Scalars.Num();
}
//---------------------------------------------------------------------------------------------
int32 FInstance::GetStringCount(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex ) const
{
check(m_pD->Components.IsValidIndex(ComponentIndex));
check(m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex));
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Strings.Num();
}
//---------------------------------------------------------------------------------------------
FResourceID FInstance::GetMeshId(int32 ComponentIndex, int32 LODIndex ) const
{
check(m_pD->Components.IsValidIndex(ComponentIndex));
check(m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex));
return m_pD->Components[ComponentIndex].LODs[LODIndex].MeshId;
}
//---------------------------------------------------------------------------------------------
FResourceID FInstance::GetImageId(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex, int32 ImageIndex) const
{
check(m_pD->Components.IsValidIndex(ComponentIndex));
check(m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Images.IsValidIndex(ImageIndex));
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Images[ImageIndex].Id;
}
//---------------------------------------------------------------------------------------------
FName FInstance::GetImageName(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex, int32 ImageIndex ) const
{
check(m_pD->Components.IsValidIndex(ComponentIndex));
check(m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Images.IsValidIndex(ImageIndex));
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Images[ImageIndex].Name;
}
//---------------------------------------------------------------------------------------------
FVector4f FInstance::GetVector(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex, int32 VectorIndex) const
{
check(m_pD->Components.IsValidIndex(ComponentIndex));
check(m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Vectors.IsValidIndex(VectorIndex));
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Vectors[VectorIndex].Value;
}
//---------------------------------------------------------------------------------------------
FName FInstance::GetVectorName(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex, int32 VectorIndex) const
{
check(m_pD->Components.IsValidIndex(ComponentIndex));
check(m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Vectors.IsValidIndex(VectorIndex));
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Vectors[VectorIndex].Name;
}
//---------------------------------------------------------------------------------------------
float FInstance::GetScalar(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex, int32 ScalarIndex ) const
{
check(m_pD->Components.IsValidIndex(ComponentIndex));
check(m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Scalars.IsValidIndex(ScalarIndex));
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Scalars[ScalarIndex].Value;
}
//---------------------------------------------------------------------------------------------
FName FInstance::GetScalarName(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex, int32 ScalarIndex) const
{
check(m_pD->Components.IsValidIndex(ComponentIndex));
check(m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Scalars.IsValidIndex(ScalarIndex));
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Scalars[ScalarIndex].Name;
}
//---------------------------------------------------------------------------------------------
FString FInstance::GetString(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex, int32 StringIndex ) const
{
check(m_pD->Components.IsValidIndex(ComponentIndex));
check(m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Strings.IsValidIndex(StringIndex));
bool bValid = m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Strings.IsValidIndex(StringIndex);
if (bValid)
{
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Strings[StringIndex].Value;
}
return "";
}
//---------------------------------------------------------------------------------------------
FName FInstance::GetStringName(int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex, int32 StringIndex) const
{
check(m_pD->Components.IsValidIndex(ComponentIndex));
check(m_pD->Components[ComponentIndex].LODs.IsValidIndex(LODIndex));
check(m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces.IsValidIndex(SurfaceIndex));
bool bValid = m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Strings.IsValidIndex(StringIndex);
if (bValid)
{
return m_pD->Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex].Strings[StringIndex].Name;
}
return NAME_None;
}
int32 FInstance::GetOverlayMaterial(int32 ComponentIndex) const
{
check(m_pD->Components.IsValidIndex(ComponentIndex));
check(m_pD->Components[ComponentIndex].OverlayMaterialId < int32(1 << 22));
return static_cast<int32>(m_pD->Components[ComponentIndex].OverlayMaterialId);
}
//---------------------------------------------------------------------------------------------
int32 FInstance::GetExtensionDataCount() const
{
return m_pD->ExtensionData.Num();
}
//---------------------------------------------------------------------------------------------
void FInstance::GetExtensionData(int32 Index, TSharedPtr<const FExtensionData>& OutExtensionData, FName& OutName) const
{
check(m_pD->ExtensionData.IsValidIndex(Index));
OutExtensionData = m_pD->ExtensionData[Index].Data;
OutName = m_pD->ExtensionData[Index].Name;
}
//---------------------------------------------------------------------------------------------
int32 FInstance::Private::AddComponent()
{
int32 result = Components.Emplace();
return result;
}
//---------------------------------------------------------------------------------------------
int32 FInstance::Private::AddLOD( int32 ComponentIndex )
{
// Automatically create the necessary lods and components
while (ComponentIndex >= Components.Num())
{
AddComponent();
}
return Components[ComponentIndex].LODs.Emplace();
}
//---------------------------------------------------------------------------------------------
int32 FInstance::Private::AddSurface( int32 ComponentIndex, int32 LODIndex )
{
// Automatically create the necessary lods and components
while (ComponentIndex >= Components.Num())
{
AddComponent();
}
while (LODIndex >= Components[ComponentIndex].LODs.Num())
{
AddLOD(ComponentIndex);
}
return Components[ComponentIndex].LODs[LODIndex].Surfaces.Emplace();
}
//---------------------------------------------------------------------------------------------
void FInstance::Private::SetSurfaceName( int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex, FName Name)
{
// Automatically create the necessary lods and components
while (ComponentIndex >= Components.Num())
{
AddComponent();
}
while (LODIndex >= Components[ComponentIndex].LODs.Num())
{
AddLOD(ComponentIndex);
}
while ( SurfaceIndex>=Components[ComponentIndex].LODs[LODIndex].Surfaces.Num() )
{
AddSurface( LODIndex, ComponentIndex );
}
FInstanceSurface& surface = Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex];
surface.Name = Name;
}
//---------------------------------------------------------------------------------------------
void FInstance::Private::SetMesh(int32 ComponentIndex, int32 LODIndex, FResourceID meshId, FName Name)
{
// Automatically create the necessary lods and components
while (ComponentIndex >= Components.Num())
{
AddComponent();
}
while (LODIndex >= Components[ComponentIndex].LODs.Num())
{
AddLOD(ComponentIndex);
}
FInstanceLOD& LOD = Components[ComponentIndex].LODs[LODIndex];
LOD.MeshId = meshId;
LOD.MeshName = Name;
}
//---------------------------------------------------------------------------------------------
int32 FInstance::Private::AddImage( int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex, FResourceID imageId, FName Name)
{
// Automatically create the necessary lods and components
while (ComponentIndex >= Components.Num())
{
AddComponent();
}
while (LODIndex >= Components[ComponentIndex].LODs.Num())
{
AddLOD(ComponentIndex);
}
while (SurfaceIndex >= Components[ComponentIndex].LODs[LODIndex].Surfaces.Num())
{
AddSurface(LODIndex, ComponentIndex);
}
FInstanceSurface& Surface = Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex];
return Surface.Images.Add({ imageId, Name });
}
//---------------------------------------------------------------------------------------------
int32 FInstance::Private::AddVector( int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex, const FVector4f& vec, FName Name)
{
// Automatically create the necessary lods and components
while (ComponentIndex >= Components.Num())
{
AddComponent();
}
while (LODIndex >= Components[ComponentIndex].LODs.Num())
{
AddLOD(ComponentIndex);
}
while (SurfaceIndex >= Components[ComponentIndex].LODs[LODIndex].Surfaces.Num())
{
AddSurface(LODIndex, ComponentIndex);
}
FInstanceSurface& Surface = Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex];
return Surface.Vectors.Add({ vec, Name } );
}
//---------------------------------------------------------------------------------------------
int32 FInstance::Private::AddScalar( int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex, float sca, FName Name)
{
// Automatically create the necessary lods and components
while (ComponentIndex >= Components.Num())
{
AddComponent();
}
while (LODIndex >= Components[ComponentIndex].LODs.Num())
{
AddLOD(ComponentIndex);
}
while (SurfaceIndex >= Components[ComponentIndex].LODs[LODIndex].Surfaces.Num())
{
AddSurface(LODIndex, ComponentIndex);
}
FInstanceSurface& Surface = Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex];
return Surface.Scalars.Add({ sca, Name });
}
//---------------------------------------------------------------------------------------------
int32 FInstance::Private::AddString( int32 ComponentIndex, int32 LODIndex, int32 SurfaceIndex, const FString& Value, FName Name)
{
// Automatically create the necessary lods and components
while (ComponentIndex >= Components.Num())
{
AddComponent();
}
while (LODIndex >= Components[ComponentIndex].LODs.Num())
{
AddLOD(ComponentIndex);
}
while (SurfaceIndex >= Components[ComponentIndex].LODs[LODIndex].Surfaces.Num())
{
AddSurface(LODIndex, ComponentIndex);
}
FInstanceSurface& Surface = Components[ComponentIndex].LODs[LODIndex].Surfaces[SurfaceIndex];
return Surface.Strings.Add({ Value, Name });
}
void FInstance::Private::AddOverlayMaterial(int32 ComponentIndex, float OverlayMaterialId)
{
while (ComponentIndex >= Components.Num())
{
AddComponent();
}
Components[ComponentIndex].OverlayMaterialId = OverlayMaterialId;
}
//---------------------------------------------------------------------------------------------
void FInstance::Private::AddExtensionData(const TSharedPtr<const FExtensionData>& Data, FName Name)
{
check(Data);
NamedExtensionData& Entry = ExtensionData.AddDefaulted_GetRef();
Entry.Data = Data;
Entry.Name = Name;
}
}