UnrealEngineUWP/Engine/Source/Editor/ToolMenusEditor/Private/ToolMenusEditor.cpp

// Copyright Epic Games, Inc. All Rights Reserved.

#include "ToolMenusEditor.h"
#include "ToolMenus.h"

void UToolMenuEditorDialogMenu::Init(UToolMenu* InMenu, const FName InName)
{
	Menu = InMenu;
	Name = InName;

	LoadState();
}

void UToolMenuEditorDialogSection::Init(UToolMenu* InMenu, const FName InName)
{
	Name = InName;
	Type = ESelectedEditMenuEntryType::Section;
	Menu = InMenu;

	LoadState();
}

void UToolMenuEditorDialogSection::LoadState()
{
	Super::LoadState();

	Visibility = ECustomizedToolMenuVisibility::None;

	if (Menu && Name != NAME_None)
	{
		if (FCustomizedToolMenu* CustomizedToolMenu = Menu->FindMenuCustomization())
		{
			if (FCustomizedToolMenuSection* Found = CustomizedToolMenu->Sections.Find(Name))
			{
				Visibility = Found->Visibility;
			}
		}
	}
}

void UToolMenuEditorDialogSection::PostEditChangeProperty(FPropertyChangedEvent& PropertyChangedEvent)
{
	Super::PostEditChangeProperty(PropertyChangedEvent);

	const FName PropertyName = PropertyChangedEvent.GetPropertyName();
	if (PropertyName == GET_MEMBER_NAME_CHECKED(UToolMenuEditorDialogEntry, Visibility))
	{
		if (Menu)
		{
			FCustomizedToolMenu* CustomizedToolMenu = Menu->AddMenuCustomization();
			CustomizedToolMenu->AddSection(Name)->Visibility = Visibility;
		}
	}
}

void UToolMenuEditorDialogEntry::Init(UToolMenu* InMenu, const FName InName)
{
	Name = InName;
	Type = ESelectedEditMenuEntryType::Entry;
	Menu = InMenu;

	LoadState();
}

void UToolMenuEditorDialogEntry::LoadState()
{
	Super::LoadState();

	Visibility = ECustomizedToolMenuVisibility::None;

	if (Menu && Name != NAME_None)
	{
		if (FCustomizedToolMenu* CustomizedToolMenu = Menu->FindMenuCustomization())
		{
			if (FCustomizedToolMenuEntry* Found = CustomizedToolMenu->Entries.Find(Name))
			{
				Visibility = Found->Visibility;
			}
		}
	}
}

void UToolMenuEditorDialogEntry::PostEditChangeProperty(FPropertyChangedEvent& PropertyChangedEvent)
{
	Super::PostEditChangeProperty(PropertyChangedEvent);

	const FName PropertyName = PropertyChangedEvent.GetPropertyName();
	if (PropertyName == GET_MEMBER_NAME_CHECKED(UToolMenuEditorDialogEntry, Visibility))
	{
		if (Menu)
		{
			FCustomizedToolMenu* CustomizedToolMenu = Menu->AddMenuCustomization();
			CustomizedToolMenu->AddEntry(Name)->Visibility = Visibility;
		}
	}
}