Files
ryan schmidt 46630c002f ModelingMode: Selection system improvements. Add Disconnect command and various new Selection-Edit Commands
Add UInteractiveCommandResult, UInteractiveCommand::ExecuteCommand() now optionally can return a result subclass
Add UGeometrySelectionEditCommandResult, UGeometrySelectionEditCommand now can optionally return an "output" selection via this type
UGeometrySelectionManager::ExecuteSelectionCommand optionally can restore a selection after a command, via UGeometrySelectionEditCommandResult

Add UDisconnectGeometrySelectionCommand, implements disconnection of selected triangles (ie separates but not into a new mesh)
Add UModifyGeometrySelectionCommand, implements various selection edits (select all, expand to connected, invert, invert connected, expand, contract)
Add IGeometrySelector::InitializeSelectionFromPredicate() and ::UpdateSelectionFromSelection(), implement in UDynamicMeshSelector, used to implement selection edit commands
Add UI to enable new commands in Modeling Mode

#rb none
#preflight 63c047f4305002c64170f6a2

[CL 23667880 by ryan schmidt in ue5-main branch]
2023-01-12 14:52:52 -05:00

105 lines
3.3 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#include "Commands/DisconnectGeometrySelectionCommand.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"
#include UE_INLINE_GENERATED_CPP_BY_NAME(DisconnectGeometrySelectionCommand)
using namespace UE::Geometry;
#define LOCTEXT_NAMESPACE "UDisconnectGeometrySelectionCommand"
FText UDisconnectGeometrySelectionCommand::GetCommandShortString() const
{
return LOCTEXT("ShortString", "Disconnect Selection");
}
bool UDisconnectGeometrySelectionCommand::CanExecuteCommandForSelection(UGeometrySelectionEditCommandArguments* SelectionArgs)
{
return SelectionArgs->IsMatchingType(FGeometryIdentifier::ETargetType::MeshContainer, FGeometryIdentifier::EObjectType::DynamicMesh);
}
void UDisconnectGeometrySelectionCommand::ExecuteCommandForSelection(UGeometrySelectionEditCommandArguments* SelectionArgs, UInteractiveCommandResult** Result)
{
if (Result != nullptr)
{
*Result = nullptr;
}
// 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 Disconnect 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 Disconnect them
FDynamicMeshEditor Editor(&EditMesh);
Editor.DisconnectTriangles(TriangleList.Array(), false);
// 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();
}
if (Result != nullptr)
{
UGeometrySelectionEditCommandResult* NewResult = NewObject<UGeometrySelectionEditCommandResult>();
NewResult->SourceHandle = SelectionArgs->SelectionHandle;
NewResult->OutputSelection = *NewResult->SourceHandle.Selection;
*Result = NewResult;
}
}
#undef LOCTEXT_NAMESPACE