89 lines
2.4 KiB
C++
89 lines
2.4 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "Generators/MeshShapeGenerator.h"
|
|
#include "MeshProcessingNodes/MeshProcessingBaseNodes.h"
|
|
#include "DynamicMesh/DynamicMeshAABBTree3.h"
|
|
#include "Implicit/Solidify.h"
|
|
#include "Spatial/FastWinding.h"
|
|
#include "Spatial/MeshAABBTree3.h"
|
|
|
|
#include "MeshSolidifyNode.generated.h"
|
|
|
|
USTRUCT()
|
|
struct FMeshSolidifySettings
|
|
{
|
|
GENERATED_USTRUCT_BODY()
|
|
|
|
DECLARE_GEOMETRYFLOW_DATA_TYPE_IDENTIFIER(UE::GeometryFlow::EMeshProcessingDataTypes::SolidifySettings);
|
|
|
|
UPROPERTY(EditAnywhere, Category = "Geometry Flow")
|
|
int VoxelResolution = 64;
|
|
|
|
UPROPERTY(EditAnywhere, Category = "Geometry Flow")
|
|
float WindingThreshold = 0.5;
|
|
|
|
UPROPERTY(EditAnywhere, Category = "Geometry Flow")
|
|
int SurfaceConvergeSteps = 5;
|
|
|
|
UPROPERTY(EditAnywhere, Category = "Geometry Flow")
|
|
float ExtendBounds = 2.0f;
|
|
};
|
|
|
|
|
|
|
|
namespace UE
|
|
{
|
|
namespace GeometryFlow
|
|
{
|
|
|
|
using namespace UE::Geometry;
|
|
|
|
|
|
// bringing this into the namespace so other code that relies on UE::GeometryFlow::FMeshSolidifySettings will just work
|
|
// @todo, update client code and remove this type def
|
|
typedef FMeshSolidifySettings FMeshSolidifySettings;
|
|
|
|
GEOMETRYFLOW_DECLARE_USTRUCT_SETTINGS_TYPES(FMeshSolidifySettings, Solidify, 1);
|
|
|
|
|
|
class FSolidifyMeshNode : public TProcessMeshWithSettingsBaseNode<FMeshSolidifySettings>
|
|
{
|
|
static constexpr int Version = 1;
|
|
GEOMETRYFLOW_NODE_INTERNAL(FSolidifyMeshNode, Version, FNode)
|
|
public:
|
|
|
|
virtual void ProcessMesh(
|
|
const FNamedDataMap& DatasIn,
|
|
const FMeshSolidifySettings& SettingsIn,
|
|
const FDynamicMesh3& MeshIn,
|
|
FDynamicMesh3& MeshOut,
|
|
TUniquePtr<FEvaluationInfo>& EvaluationInfo) override
|
|
{
|
|
ApplySolidify(MeshIn, SettingsIn, MeshOut);
|
|
}
|
|
|
|
void ApplySolidify(const FDynamicMesh3& MeshIn, const FMeshSolidifySettings& Settings, FDynamicMesh3& MeshOut)
|
|
{
|
|
FAxisAlignedBox3d Bounds = MeshIn.GetBounds();
|
|
|
|
FDynamicMeshAABBTree3 MeshBVTree(&MeshIn);
|
|
TFastWindingTree<FDynamicMesh3> FastWinding(&MeshBVTree);
|
|
|
|
TImplicitSolidify<FDynamicMesh3> SolidifyCalc(&MeshIn, &MeshBVTree, &FastWinding);
|
|
|
|
SolidifyCalc.SetCellSizeAndExtendBounds(Bounds, Settings.ExtendBounds, Settings.VoxelResolution);
|
|
SolidifyCalc.WindingThreshold = Settings.WindingThreshold;
|
|
SolidifyCalc.SurfaceSearchSteps = Settings.SurfaceConvergeSteps;
|
|
SolidifyCalc.bSolidAtBoundaries = true;
|
|
SolidifyCalc.ExtendBounds = Settings.ExtendBounds;
|
|
|
|
MeshOut.Copy(&SolidifyCalc.Generate());
|
|
}
|
|
};
|
|
|
|
|
|
} // end namespace GeometryFlow
|
|
} // end namespace UE
|