Files
UnrealEngineUWP/Engine/Plugins/Runtime/MeshModelingToolset/Source/MeshModelingTools/Private/Commands/DeleteGeometrySelectionCommand.cpp
2022-07-15 17:53:52 -04:00

90 lines
2.9 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#include "Commands/DeleteGeometrySelectionCommand.h"
#include "ToolContextInterfaces.h"
#include "UDynamicMesh.h"
#include "DynamicMesh/DynamicMesh3.h"
#include "DynamicMesh/DynamicMeshChangeTracker.h"
#include "DynamicMeshEditor.h"
#include "Changes/MeshChange.h"
#include "Selections/GeometrySelectionUtil.h"
using namespace UE::Geometry;
#define LOCTEXT_NAMESPACE "UDeleteGeometrySelectionCommand"
FText UDeleteGeometrySelectionCommand::GetCommandShortString() const
{
return LOCTEXT("ShortString", "Delete Selection");
}
bool UDeleteGeometrySelectionCommand::CanExecuteCommandForSelection(UGeometrySelectionEditCommandArguments* SelectionArgs)
{
return SelectionArgs->IsMatchingType(FGeometryIdentifier::ETargetType::MeshContainer, FGeometryIdentifier::EObjectType::DynamicMesh);
}
void UDeleteGeometrySelectionCommand::ExecuteCommandForSelection(UGeometrySelectionEditCommandArguments* SelectionArgs)
{
// should have been verified by CanExecute
check(SelectionArgs->IsMatchingType(FGeometryIdentifier::ETargetType::MeshContainer, FGeometryIdentifier::EObjectType::DynamicMesh));
// collect up all our inputs
UDynamicMesh* MeshObject = SelectionArgs->SelectionHandle.Identifier.GetAsObjectType<UDynamicMesh>();
check(MeshObject != nullptr);
const FGeometrySelection* Selection = SelectionArgs->SelectionHandle.Selection;
if (Selection->Selection.Num() == 0)
{
return;
}
bool bTrackChanges = SelectionArgs->HasTransactionsAPI();
TUniquePtr<FDynamicMeshChange> DynamicMeshChange; // only initialized if bTrackChanges == true
// apply the delete operation
MeshObject->EditMesh([&](FDynamicMesh3& EditMesh)
{
// build list of triangles from whatever the selection contains
TSet<int32> TriangleList;
//UE::Geometry::FPolygroupSet UsePolygroupSet = ...; // need to support this eventually
UE::Geometry::EnumerateSelectionTriangles(*Selection, EditMesh,
[&](int32 TriangleID) { TriangleList.Add(TriangleID); });
// mark triangles for change
FDynamicMeshChangeTracker ChangeTracker(&EditMesh);
if (bTrackChanges)
{
ChangeTracker.BeginChange();
ChangeTracker.SaveTriangles(TriangleList, true);
}
// actually delete them
FDynamicMeshEditor Editor(&EditMesh);
Editor.RemoveTriangles(TriangleList.Array(), true);
// extract the change record
if (bTrackChanges)
{
DynamicMeshChange = ChangeTracker.EndChange();
}
}, EDynamicMeshChangeType::GeneralEdit, EDynamicMeshAttributeChangeFlags::Unknown, false);
// emit change
if ( bTrackChanges && DynamicMeshChange.IsValid() )
{
SelectionArgs->GetTransactionsAPI()->BeginUndoTransaction(GetCommandShortString());
SelectionArgs->GetTransactionsAPI()->AppendChange(MeshObject,
MakeUnique<FMeshChange>(MoveTemp(DynamicMeshChange)), GetCommandShortString());
SelectionArgs->GetTransactionsAPI()->EndUndoTransaction();
}
}
#undef LOCTEXT_NAMESPACE