Ability/Ability/Plugins/EzAbility/Source/EzAbilityEditor/Private/SEzAbleTreeViewWidget.cpp

// Fill out your copyright notice in the Description page of Project Settings.


#include "SEzAbleTreeViewWidget.h"
#include "SlateOptMacros.h"
#include "SPositiveActionButton.h"
#include "Widgets/Layout/SScrollBox.h"
#include "SEzAbleTreeViewRowWidget.h"
#include "AbleTreeEditorCommands.h"

#define LOCTEXT_NAMESPACE "AbleStateTreeEditor"

BEGIN_SLATE_FUNCTION_BUILD_OPTIMIZATION
void SEzAbleTreeViewWidget::Construct(const FArguments& InArgs, TSharedPtr<FEzAbleTreeViewOperationMode> InAbleTreeViewOperationMode)
{
	AbleTreeViewOperationMode = InAbleTreeViewOperationMode;

	AbleTreeViewOperationMode->GetOnStateAdded().AddSP(this, &SEzAbleTreeViewWidget::HandleModelStateAdded);

	TSharedRef<SScrollBar> HorizontalScrollBar = SNew(SScrollBar)
		.Orientation(Orient_Horizontal)
		.Thickness(FVector2D(12.0f, 12.0f));

	TSharedRef<SScrollBar> VerticalScrollBar = SNew(SScrollBar)
		.Orientation(Orient_Vertical)
		.Thickness(FVector2D(12.0f, 12.0f));

	AbleTreeViewOperationMode->GetSubTrees(Subtrees);

// 	UEzAbilityState* AbilityState = NewObject<UEzAbilityState>();
// 	UEzAbilityState* AbilitySub = NewObject<UEzAbilityState>();
// 	TestStrongState.Add(AbilityState);
// 	TestState.Add(TestStrongState[0]);

	TreeView = SNew(STreeView<TWeakObjectPtr<UEzAbilityState>>)
		.OnGenerateRow(this, &SEzAbleTreeViewWidget::HandleGenerateRow)
		.OnGetChildren(this, &SEzAbleTreeViewWidget::HandleGetChildren)
		.OnSelectionChanged(this, &SEzAbleTreeViewWidget::HandleTreeSelectionChanged)
		.OnExpansionChanged(this, &SEzAbleTreeViewWidget::HandleTreeExpansionChanged)
		.OnContextMenuOpening(this, &SEzAbleTreeViewWidget::HandleContextMenuOpening )
		.TreeItemsSource(&Subtrees)
		.ItemHeight(32)
		.AllowOverscroll(EAllowOverscroll::No)
		.ExternalScrollbar(VerticalScrollBar);
	ChildSlot
		[
			SNew(SVerticalBox)

			+ SVerticalBox::Slot()
		.VAlign(VAlign_Center)
		.AutoHeight()
		[
			SNew(SBorder)
			.BorderImage(FAppStyle::GetBrush("ToolPanel.GroupBorder"))
		.Padding(2.0f)
		[
			SNew(SHorizontalBox)

			// New State
		+ SHorizontalBox::Slot()
		.VAlign(VAlign_Center)
		.Padding(4.0f, 2.0f)
		.AutoWidth()
		[
			SNew(SPositiveActionButton)
			.ToolTipText(LOCTEXT("AddStateToolTip", "Add New State"))
		.Icon(FAppStyle::Get().GetBrush("Icons.Plus"))
		.Text(LOCTEXT("AddState", "Add State"))
		.OnClicked(this, &SEzAbleTreeViewWidget::HandleAddStateButton)
		]
		]
		]

	+ SVerticalBox::Slot()
		.Padding(0.0f, 6.0f, 0.0f, 0.0f)
		[
			SNew(SHorizontalBox)
			+ SHorizontalBox::Slot()
		.FillWidth(1.0f)
		.Padding(0.0f)
		[
			SAssignNew(ViewBox, SScrollBox)
			.Orientation(Orient_Horizontal)
		.ExternalScrollbar(HorizontalScrollBar)
		+ SScrollBox::Slot()
		.FillSize(1.0f)
		[
			TreeView.ToSharedRef()
		]
		]

	+ SHorizontalBox::Slot()
		.AutoWidth()
		[
			VerticalScrollBar
		]
		]
	+ SVerticalBox::Slot()
		.AutoHeight()
		[
			HorizontalScrollBar
		]
		];


		if (CommandList = AbleTreeViewOperationMode->AbleTreeViewCommandList)
		{
			BindCommands();	
		}

	UpdateTree(true);
	
}
END_SLATE_FUNCTION_BUILD_OPTIMIZATION

FReply SEzAbleTreeViewWidget::HandleAddStateButton()
{
	if (AbleTreeViewOperationMode == nullptr)
	{
		return FReply::Handled();
	}

	TArray<UEzAbilityState*> SelectedStates;
	//AbleTreeViewOperationMode->GetSelectedStates(SelectedStates);
	//UStateTreeState* FirstSelectedState = SelectedStates.Num() > 0 ? SelectedStates[0] : nullptr;

// 	if (FirstSelectedState != nullptr)
// 	{
// 		// If the state is root, add child state, else sibling.
// 		if (FirstSelectedState->Parent == nullptr)
// 		{
// 			StateTreeViewModel->AddChildState(FirstSelectedState);
// 			TreeView->SetItemExpansion(FirstSelectedState, true);
// 		}
// 		else
// 		{
// 			StateTreeViewModel->AddState(FirstSelectedState);
// 		}
// 	}
// 	else
// 	{
// 		// Add root state at the lowest level.
// 		StateTreeViewModel->AddState(nullptr);
// 	}
	
	AbleTreeViewOperationMode->AddState(nullptr);

	return FReply::Handled();
}

TSharedRef<ITableRow> SEzAbleTreeViewWidget::HandleGenerateRow(TWeakObjectPtr<UEzAbilityState> InState, const TSharedRef<STableViewBase>& InOwnerTableView)
{
	return SNew(SEzAbleTreeViewRowWidget, InOwnerTableView, InState, ViewBox);
}

void SEzAbleTreeViewWidget::HandleGetChildren(TWeakObjectPtr<UEzAbilityState> InParent, TArray<TWeakObjectPtr<UEzAbilityState>>& OutChildren)
{
	//Test Code.
	//UE_LOG(LogTemp, Log, TEXT("EZAbleTreeViewWidget HandleGetChildren"));

	//InParent->AddChildState(FName());

	if (const UEzAbilityState* Parent = InParent.Get())
	{
		OutChildren.Append(Parent->Children);
	}

}

void SEzAbleTreeViewWidget::UpdateTree(bool bExpandPersistent /*= false*/)
{
	TreeView->RequestTreeRefresh();
}

void SEzAbleTreeViewWidget::HandleTreeSelectionChanged(TWeakObjectPtr<UEzAbilityState> InSelectedItem, ESelectInfo::Type SelectionType)
{
	if (!AbleTreeViewOperationMode)
	{
		return;
	}

	// Do not report code based selection changes.
	if (SelectionType == ESelectInfo::Direct)
	{
		return;
	}

	TArray<TWeakObjectPtr<UEzAbilityState>> SelectedItems = TreeView->GetSelectedItems();

	bUpdatingSelection = true;
	AbleTreeViewOperationMode->SetSelection(SelectedItems);
	bUpdatingSelection = false;
}

void SEzAbleTreeViewWidget::HandleTreeExpansionChanged(TWeakObjectPtr<UEzAbilityState> InSelectedItem, bool bExpanded)
{
	
}

TSharedPtr<SWidget> SEzAbleTreeViewWidget::HandleContextMenuOpening()
{
	if (!AbleTreeViewOperationMode)
	{
		return nullptr;
	}

	FMenuBuilder MenuBuilder(true, CommandList);

	MenuBuilder.AddSubMenu(
		LOCTEXT("AddState", "Add State"),
		FText(),
		FNewMenuDelegate::CreateLambda([this](FMenuBuilder& MenuBuilder)
			{							
				MenuBuilder.AddMenuEntry(FAbleTreeEditorCommands::Get().AddSiblingState);
				MenuBuilder.AddMenuEntry(FAbleTreeEditorCommands::Get().AddChildState);
			}),
		/*bInOpenSubMenuOnClick =*/false,
		FSlateIcon(FAppStyle::GetAppStyleSetName(), "Icons.Plus")
	);

	MenuBuilder.AddSeparator();

	MenuBuilder.AddMenuEntry(FAbleTreeEditorCommands::Get().CutStates);
	MenuBuilder.AddMenuEntry(FAbleTreeEditorCommands::Get().CopyStates);

	MenuBuilder.AddSubMenu(
		LOCTEXT("Paste", "Paste"),
		FText(),
		FNewMenuDelegate::CreateLambda([this](FMenuBuilder& MenuBuilder)
			{
				MenuBuilder.AddMenuEntry(FAbleTreeEditorCommands::Get().PasteStatesAsSiblings);
				MenuBuilder.AddMenuEntry(FAbleTreeEditorCommands::Get().PasteStatesAsChildren);
			}),
		/*bInOpenSubMenuOnClick =*/false,
		FSlateIcon(FAppStyle::GetAppStyleSetName(), "GenericCommands.Paste")
	);

	MenuBuilder.AddMenuEntry(FAbleTreeEditorCommands::Get().DuplicateStates);
	MenuBuilder.AddMenuEntry(FAbleTreeEditorCommands::Get().DeleteStates);
	MenuBuilder.AddMenuEntry(FAbleTreeEditorCommands::Get().RenameState);
	MenuBuilder.AddSeparator();
	MenuBuilder.AddMenuEntry(FAbleTreeEditorCommands::Get().EnableStates);

//#if WITH_STATETREE_TRACE_DEBUGGER
	MenuBuilder.AddSeparator();
	MenuBuilder.AddMenuEntry(FAbleTreeEditorCommands::Get().EnableOnEnterStateBreakpoint);
	MenuBuilder.AddMenuEntry(FAbleTreeEditorCommands::Get().EnableOnExitStateBreakpoint);
//#endif // WITH_STATETREE_TRACE_DEBUGGER

	return MenuBuilder.MakeWidget();
}

void SEzAbleTreeViewWidget::HandleModelStateAdded(UEzAbilityState* ParentState, UEzAbilityState* NewState)
{	
	TreeView->RequestTreeRefresh();
}


bool SEzAbleTreeViewWidget::HasSelection() const
{
	return AbleTreeViewOperationMode && AbleTreeViewOperationMode->HasSelection();
}

void SEzAbleTreeViewWidget::BindCommands()
{
	const FAbleTreeEditorCommands& Commands = FAbleTreeEditorCommands::Get();

	CommandList->MapAction(
		Commands.DeleteStates,
		FExecuteAction::CreateSP(this, &SEzAbleTreeViewWidget::HandleDeleteStates),
		FCanExecuteAction::CreateSP(this, &SEzAbleTreeViewWidget::HasSelection));
}

void SEzAbleTreeViewWidget::HandleDeleteStates()
{
	if (AbleTreeViewOperationMode)
	{
		AbleTreeViewOperationMode->RemoveSelectedStates();
		TreeView->RequestTreeRefresh();
	}
}

FReply SEzAbleTreeViewWidget::OnKeyDown(const FGeometry& MyGeometry, const FKeyEvent& InKeyEvent)
{
	if (CommandList->ProcessCommandBindings(InKeyEvent))
	{
		return FReply::Handled();
	}
	else
	{
		return SCompoundWidget::OnKeyDown(MyGeometry, InKeyEvent);
	}
}

#undef LOCTEXT_NAMESPACE