GeometryProcessing: Move plugin from Experimental to Runtime

#jira UETOOL-3823 #rb ryan.schmidt [CL 16958628 by michael balzer in ue5-main branch]
2026-03-26 18:15:20 -07:00 · 2021-07-26 14:06:18 -04:00
parent 7cd3754962
commit 9b4922340e
609 changed files with 10 additions and 10 deletions
@@ -0,0 +1,118 @@
+// Copyright Epic Games, Inc. All Rights Reserved.
+
+#pragma once
+
+#include "MathUtil.h"
+#include "DynamicMesh/DynamicMesh3.h"
+#include "DynamicMesh/DynamicMeshAttributeSet.h"
+
+
+namespace UE
+{
+namespace Geometry
+{
+
+
+/**
+ * Polygroup sets can be stored in multiple places. The default location is in the per-triangle group integer stored
+ * directly on a FDynamicMesh3. Additional layers may be stored in the FDynamicMeshAttributeSet. Future iterations
+ * could store packed polygroups in other places, store them in separate arrays, and so on.
+ * FPolygroupSet can be used to abstract these different cases, by providing a standard Polygroup Get/Set API.
+ * 
+ * To support unique Polygroup ID allocation, FPolygroupSet calculates the maximum GroupID on creation, and 
+ * updates this maximum across SetGroup() calls. AllocateNewGroupID() can be used to provide new unused GroupIDs.
+ * For consistency with FDynamicMesh3, MaxGroupID is set such that all GroupIDs are less than MaxGroupID
+ *
+ */
+struct DYNAMICMESH_API FPolygroupSet
+{
+	const FDynamicMesh3* Mesh = nullptr;
+	const FDynamicMeshPolygroupAttribute* PolygroupAttrib = nullptr;
+	int32 GroupLayerIndex = -1;
+	int32 MaxGroupID = 0; // Note: all group IDs are less than MaxGroupID
+
+	/** Initialize a PolygroupSet for the given Mesh, and standard triangle group layer */
+	explicit FPolygroupSet(const FDynamicMesh3* MeshIn);
+
+	/** Initialize a PolygroupSet for given Mesh and specific Polygroup attribute layer */
+	explicit FPolygroupSet(const FDynamicMesh3* MeshIn, const FDynamicMeshPolygroupAttribute* PolygroupAttribIn);
+
+	/** Initialize a PolygroupSet for given Mesh and specific Polygroup attribute layer, found by index. If not valid, fall back to standard triangle group layer. */
+	explicit FPolygroupSet(const FDynamicMesh3* MeshIn, int32 PolygroupLayerIndex);
+
+	/** Initialize a PolygroupSet for given Mesh and specific Polygroup attribute layer, found by name. If not valid, fall back to standard triangle group layer. */
+	explicit FPolygroupSet(const FDynamicMesh3* MeshIn, FName AttribName);
+
+	/** Initialize a PolygroupSet by copying an existing PolygroupSet */
+	explicit FPolygroupSet(const FPolygroupSet* CopyIn);
+
+
+	/** @return Mesh this PolygroupSet references  */
+	const FDynamicMesh3* GetMesh() { return Mesh; }
+
+	/** @return PolygroupAttribute this PolygroupSet references, or null if no PolygroupAttribute is in use */
+	const FDynamicMeshPolygroupAttribute* GetPolygroup() { return PolygroupAttrib; }
+
+	/** @return index of current PolygroupAttribute into Mesh AttributeSet, or -1 if this information does not exist */
+	int32 GetPolygroupIndex() const { return GroupLayerIndex; }
+
+
+	/**
+	 * @return PolygroupID for a TriangleID
+	 */
+	int32 GetGroup(int32 TriangleID) const
+	{
+		return (PolygroupAttrib) ? PolygroupAttrib->GetValue(TriangleID) : Mesh->GetTriangleGroup(TriangleID);
+	}
+
+	/**
+	 * @return PolygroupID for a TriangleID
+	 */
+	int32 GetTriangleGroup(int32 TriangleID) const
+	{
+		return (PolygroupAttrib) ? PolygroupAttrib->GetValue(TriangleID) : Mesh->GetTriangleGroup(TriangleID);
+	}
+
+	/**
+	 * Set the PolygroupID for a TriangleID
+	 */
+	void SetGroup(int32 TriangleID, int32 NewGroupID, FDynamicMesh3& WritableMesh)
+	{
+		checkSlow(&WritableMesh == this->Mesh);		// require the same mesh
+		if (WritableMesh.IsTriangle(TriangleID))
+		{
+			if (PolygroupAttrib)
+			{
+				FDynamicMeshPolygroupAttribute* WritableGroupAttrib = WritableMesh.Attributes()->GetPolygroupLayer(GroupLayerIndex);
+				checkSlow(WritableGroupAttrib == PolygroupAttrib)
+				WritableGroupAttrib->SetValue(TriangleID, NewGroupID);
+			}
+			else
+			{
+				WritableMesh.SetTriangleGroup(TriangleID, NewGroupID);
+			}
+		}
+		MaxGroupID = FMath::Max(MaxGroupID, NewGroupID + 1);
+	}
+
+	/**
+	 * Calculate the current maximum PolygroupID used in the active set and store in MaxGroupID member
+	 */
+	void RecalculateMaxGroupID();
+
+	/**
+	 * Allocate a new unused PolygroupID by incrementing the MaxGroupID member
+	 */
+	int32 AllocateNewGroupID()
+	{
+		return MaxGroupID++;
+	}
+};
+
+
+
+
+
+
+}	// end namespace Geometry
+}	// end namespace UE
@@ -0,0 +1,42 @@
+// Copyright Epic Games, Inc. All Rights Reserved.
+
+#pragma once
+
+#include "DynamicMesh/DynamicMesh3.h"
+#include "DynamicMesh/DynamicMeshAttributeSet.h"
+
+
+namespace UE
+{
+namespace Geometry
+{
+
+/**
+ * Find the first FDynamicMeshPolygroupAttribute with the given FName in a the AttributeSet of a Mesh.
+ * @return nullptr if no Polygroup layer is found
+ */
+DYNAMICMESH_API FDynamicMeshPolygroupAttribute* FindPolygroupLayerByName(FDynamicMesh3& Mesh, FName Name);
+
+/**
+ * Find the first FDynamicMeshPolygroupAttribute with the given FName in a the AttributeSet of a Mesh.
+ * @return nullptr if no Polygroup layer is found
+ */
+DYNAMICMESH_API const FDynamicMeshPolygroupAttribute* FindPolygroupLayerByName(const FDynamicMesh3& Mesh, FName Name);
+
+/**
+ * @return index of the first Layer with the given FName in Mesh AttributeSet, or -1 if not found
+ */
+DYNAMICMESH_API int32 FindPolygroupLayerIndexByName(const FDynamicMesh3& Mesh, FName Name);
+
+/**
+ * @return index of Layer in Mesh AttributeSet, or -1 if not found
+ */
+DYNAMICMESH_API int32 FindPolygroupLayerIndex(const FDynamicMesh3& Mesh, const FDynamicMeshPolygroupAttribute* Layer);
+
+/**
+ * @return Compute the upper bound of the group IDs for the given layer (one greater than the highest group ID)
+ */
+DYNAMICMESH_API int32 ComputeGroupIDBound(const FDynamicMesh3& Mesh, const FDynamicMeshPolygroupAttribute* Layer);
+
+}	// end namespace Geometry
+}	// end namespace UE