You've already forked UnrealEngineUWP
mirror of
https://github.com/izzy2lost/UnrealEngineUWP.git
synced 2026-03-26 18:15:20 -07:00
a43fd72984
- Updated add button to new positive button style - Change the add-state logic to add to root or subtrees if context is present (e.g. add after) - Renamed UStateTreeState::LinkedState to LinkedSubtree to indicate what kind of state is expected - Removed strong reference from the treeview (was preventing a StateTree asset being deleted) - Fixed bad colors when renaming a state #jira UE-151757, UE-151874, UE-163569, UE-150568 #rb Mieszko.Zielinski #preflight 63369601f76de2d4d5977b94 [CL 22271644 by mikko mononen in ue5-main branch]
110 lines
4.2 KiB
C++
110 lines
4.2 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<TWeakObjectPtr<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<TWeakObjectPtr<UStateTreeState>>& InSelection);
|
|
bool IsSelected(const UStateTreeState* State) const;
|
|
bool IsChildOfSelection(const UStateTreeState* State) const;
|
|
void GetSelectedStates(TArray<UStateTreeState*>& OutSelectedStates);
|
|
void GetSelectedStates(TArray<TWeakObjectPtr<UStateTreeState>>& OutSelectedStates);
|
|
bool HasSelection() const;
|
|
|
|
// Returns array of subtrees to edit.
|
|
TArray<UStateTreeState*>* GetSubTrees() const;
|
|
int32 GetSubTreeCount() const;
|
|
void GetSubTrees(TArray<TWeakObjectPtr<UStateTreeState>>& OutSubtrees) const;
|
|
|
|
// Gets and sets StateTree view expansion state store in the asset.
|
|
void SetPersistentExpandedStates(TSet<TWeakObjectPtr<UStateTreeState>>& InExpandedStates);
|
|
void GetPersistentExpandedStates(TSet<TWeakObjectPtr<UStateTreeState>>& OutExpandedStates);
|
|
|
|
// 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<TWeakObjectPtr<UStateTreeState>>& ExpandedStates);
|
|
void MoveSelectedStates(UStateTreeState* TargetState, const FStateTreeViewModelInsert RelativeLocation);
|
|
|
|
void HandleIdentifierChanged(const UStateTree& StateTree) const;
|
|
|
|
TWeakObjectPtr<UStateTreeEditorData> TreeDataWeak;
|
|
TSet<TWeakObjectPtr<UStateTreeState>> SelectedStates;
|
|
FOnAssetChanged OnAssetChanged;
|
|
FOnStatesChanged OnStatesChanged;
|
|
FOnStateAdded OnStateAdded;
|
|
FOnStatesRemoved OnStatesRemoved;
|
|
FOnStatesMoved OnStatesMoved;
|
|
FOnSelectionChanged OnSelectionChanged;
|
|
};
|