You've already forked UnrealEngineUWP
mirror of
https://github.com/izzy2lost/UnrealEngineUWP.git
synced 2026-03-26 18:15:20 -07:00
e38a996c3d
- Fix crash when undoing a move or delete - Cleaned up the logic for move and delete #jira UE-149306 #review #preflight 62614403360b45c32a8b2494 [CL 19847123 by mikko mononen in ue5-main branch]
108 lines
4.0 KiB
C++
108 lines
4.0 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "CoreMinimal.h"
|
|
#include "Templates/Casts.h"
|
|
#include "UObject/WeakObjectPtr.h"
|
|
#include "EditorUndoClient.h"
|
|
|
|
class UStateTreeEditorData;
|
|
class UStateTreeState;
|
|
class UStateTree;
|
|
class FMenuBuilder;
|
|
|
|
enum class FStateTreeViewModelInsert : uint8
|
|
{
|
|
Before,
|
|
After,
|
|
Into,
|
|
};
|
|
|
|
/**
|
|
* ModelView for editing StateTreeEditorData.
|
|
*/
|
|
class FStateTreeViewModel : public FEditorUndoClient, public TSharedFromThis<FStateTreeViewModel>
|
|
{
|
|
public:
|
|
|
|
DECLARE_MULTICAST_DELEGATE(FOnAssetChanged);
|
|
DECLARE_MULTICAST_DELEGATE_TwoParams(FOnStatesChanged, const TSet<UStateTreeState*>& /*AffectedStates*/, const FPropertyChangedEvent& /*PropertyChangedEvent*/);
|
|
DECLARE_MULTICAST_DELEGATE_TwoParams(FOnStateAdded, UStateTreeState* /*ParentState*/, UStateTreeState* /*NewState*/);
|
|
DECLARE_MULTICAST_DELEGATE_OneParam(FOnStatesRemoved, const TSet<UStateTreeState*>& /*AffectedParents*/);
|
|
DECLARE_MULTICAST_DELEGATE_TwoParams(FOnStatesMoved, const TSet<UStateTreeState*>& /*AffectedParents*/, const TSet<UStateTreeState*>& /*MovedStates*/);
|
|
DECLARE_MULTICAST_DELEGATE_OneParam(FOnSelectionChanged, const TArray<UStateTreeState*>& /*SelectedStates*/);
|
|
|
|
FStateTreeViewModel();
|
|
virtual ~FStateTreeViewModel() override;
|
|
|
|
void Init(UStateTreeEditorData* InTreeData);
|
|
|
|
//~ FEditorUndoClient
|
|
virtual void PostUndo(bool bSuccess) override;
|
|
virtual void PostRedo(bool bSuccess) override;
|
|
|
|
// Selection handling.
|
|
void ClearSelection();
|
|
void SetSelection(UStateTreeState* Selected);
|
|
void SetSelection(const TArray<UStateTreeState*>& InSelection);
|
|
bool IsSelected(const UStateTreeState* State) const;
|
|
bool IsChildOfSelection(const UStateTreeState* State) const;
|
|
void GetSelectedStates(TArray<UStateTreeState*>& OutSelectedStates);
|
|
bool HasSelection() const;
|
|
|
|
// Returns array of subtrees to edit.
|
|
TArray<UStateTreeState*>* GetSubTrees() const;
|
|
int32 GetSubTreeCount() const;
|
|
|
|
// Gets and sets StateTree view expansion state store in the asset.
|
|
void GetPersistentExpandedStates(TSet<UStateTreeState*>& OutExpandedStates);
|
|
void SetPersistentExpandedStates(TSet<UStateTreeState*>& InExpandedStates);
|
|
|
|
// State manipulation commands
|
|
void AddState(UStateTreeState* AfterState);
|
|
void AddChildState(UStateTreeState* ParentState);
|
|
void RenameState(UStateTreeState* State, FName NewName);
|
|
void RemoveSelectedStates();
|
|
void MoveSelectedStatesBefore(UStateTreeState* TargetState);
|
|
void MoveSelectedStatesAfter(UStateTreeState* TargetState);
|
|
void MoveSelectedStatesInto(UStateTreeState* TargetState);
|
|
|
|
// Force to update the view externally.
|
|
void NotifyAssetChangedExternally() const;
|
|
void NotifyStatesChangedExternally(const TSet<UStateTreeState*>& ChangedStates, const FPropertyChangedEvent& PropertyChangedEvent) const;
|
|
|
|
// Called when the whole asset is updated (i.e. undo/redo).
|
|
FOnAssetChanged& GetOnAssetChanged() { return OnAssetChanged; }
|
|
|
|
// Called when States are changed (i.e. change name or properties).
|
|
FOnStatesChanged& GetOnStatesChanged() { return OnStatesChanged; }
|
|
|
|
// Called each time a state is added.
|
|
FOnStateAdded& GetOnStateAdded() { return OnStateAdded; }
|
|
|
|
// Called each time a states are removed.
|
|
FOnStatesRemoved& GetOnStatesRemoved() { return OnStatesRemoved; }
|
|
|
|
// Called each time a state is removed.
|
|
FOnStatesMoved& GetOnStatesMoved() { return OnStatesMoved; }
|
|
|
|
// Called each time the selection changes.
|
|
FOnSelectionChanged& GetOnSelectionChanged() { return OnSelectionChanged; }
|
|
|
|
protected:
|
|
void GetExpandedStatesRecursive(UStateTreeState* State, TSet<UStateTreeState*>& ExpandedStates);
|
|
void MoveSelectedStates(UStateTreeState* TargetState, const FStateTreeViewModelInsert RelativeLocation);
|
|
|
|
void HandleIdentifierChanged(const UStateTree& StateTree) const;
|
|
|
|
UStateTreeEditorData* TreeData; // TODO: is this safe? should we used TWeakObjectPtr here instead?
|
|
TSet<FWeakObjectPtr> SelectedStates;
|
|
FOnAssetChanged OnAssetChanged;
|
|
FOnStatesChanged OnStatesChanged;
|
|
FOnStateAdded OnStateAdded;
|
|
FOnStatesRemoved OnStatesRemoved;
|
|
FOnStatesMoved OnStatesMoved;
|
|
FOnSelectionChanged OnSelectionChanged;
|
|
};
|