UnrealEngine/Engine/Plugins/Chooser/Source/ChooserEditor/Private/ChooserFactory.cpp

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

#include "ChooserFactory.h"
#include "Chooser.h"
#include "ChooserInitializer.h"
#include "ChooserEditorSettings.h"
#include "Framework/Application/SlateApplication.h"
#include "Widgets/Input/SButton.h"
#include "Widgets/Layout/SUniformGridPanel.h"
#include "Widgets/SCompoundWidget.h"
#include "Widgets/SWindow.h"
#include "Editor.h"
#include "ObjectChooserClassFilter.h"
#include "StructViewerModule.h"
#include "PropertyEditorModule.h"
#include "Modules/ModuleManager.h"

#define LOCTEXT_NAMESPACE "CreateChooser"

/*------------------------------------------------------------------------------
	Dialog to configure creation properties
------------------------------------------------------------------------------*/
class SChooserCreateDialog : public SCompoundWidget
{
public:
	SLATE_BEGIN_ARGS( SChooserCreateDialog ){}

	SLATE_END_ARGS()

	/** Constructs this widget with InArgs */
	void Construct( const FArguments& InArgs )
	{
		bOkClicked = false;
		
		FPropertyEditorModule& PropertyEditorModule = FModuleManager::GetModuleChecked<FPropertyEditorModule>( "PropertyEditor" );
		FDetailsViewArgs DetailsViewArgs;
		// DetailsViewArgs.NotifyHook = this;
		DetailsViewArgs.NameAreaSettings = FDetailsViewArgs::HideNameArea;
		DetailsViewArgs.bAllowSearch = false;
		DetailsViewArgs.bShowOptions = false;
		DetailsViewArgs.bAllowFavoriteSystem = false;
		DetailsViewArgs.bShowLooseProperties = true;
		DetailsView = PropertyEditorModule.CreateDetailView( DetailsViewArgs );
		DetailsView->SetObject(ChooserFactory.Get());

		ChildSlot
		[
			SNew(SVerticalBox)
			+SVerticalBox::Slot().FillHeight(1)
			[
				DetailsView.ToSharedRef()
			]

			// Ok/Cancel buttons
			+SVerticalBox::Slot()
			.AutoHeight()
			.HAlign(HAlign_Right)
			.VAlign(VAlign_Bottom)
			.Padding(10.0f)
			[
				SNew(SUniformGridPanel)
				.SlotPadding(FAppStyle::GetMargin("StandardDialog.SlotPadding"))
				.MinDesiredSlotWidth(FAppStyle::GetFloat("StandardDialog.MinDesiredSlotWidth"))
				.MinDesiredSlotHeight(FAppStyle::GetFloat("StandardDialog.MinDesiredSlotHeight"))
				+SUniformGridPanel::Slot(0,0)
				[
					SNew(SButton)
					.ToolTipText(LOCTEXT("CreateAnimBlueprintCreate_Tooltip", "Create a new Chooser Table Asset.."))
					.IsEnabled_Lambda([this]()
					{
						return true;
					})
					.HAlign(HAlign_Center)
					.ContentPadding( FAppStyle::GetMargin("StandardDialog.ContentPadding") )
					.OnClicked(this, &SChooserCreateDialog::OkClicked)
					.IsEnabled_Lambda([this]() { return ChooserFactory->ChooserInitializer.IsValid(); } )
					.Text(LOCTEXT("Create", "Create"))
				]
				+SUniformGridPanel::Slot(1,0)
				[
					SNew(SButton)
					.HAlign(HAlign_Center)
					.ContentPadding( FAppStyle::GetMargin("StandardDialog.ContentPadding") )
					.OnClicked(this, &SChooserCreateDialog::CancelClicked)
					.Text(LOCTEXT("Cancel", "Cancel"))
				]
			]
		];
	}
	
	/** Sets properties for the supplied AnimBlueprintFactory */
	bool ConfigureProperties(TWeakObjectPtr<UChooserTableFactory> InChooserFactory)
	{
		ChooserFactory = InChooserFactory;
		if (!UChooserEditorSettings::Get().DefaultCreateType.IsEmpty())
		{
			FTopLevelAssetPath StructPath(UChooserEditorSettings::Get().DefaultCreateType);
			if (UScriptStruct* DefaultInitializer = FindObject<UScriptStruct>(StructPath))
			{
				ChooserFactory->ChooserInitializer.InitializeAs(DefaultInitializer);
			}
		}
		
		DetailsView->SetObject(ChooserFactory.Get());
		
		const float AppScale = FSlateApplication::Get().GetApplicationScale();

		Window = SNew(SWindow)
		.Title( LOCTEXT("Create Chooser Options", "Create Chooser Table") )
		.SizingRule(ESizingRule::FixedSize)
		.ClientSize(FVector2f(AppScale*550, AppScale*300))
		.SupportsMinimize(false)
		.SupportsMaximize(false)
		[
			AsShared()
		];

		GEditor->EditorAddModalWindow(Window.ToSharedRef());
		ChooserFactory.Reset();

		return bOkClicked;
	}

private:

	/** Handler for when ok is clicked */
	FReply OkClicked()
	{
		CloseDialog(true);

		return FReply::Handled();
	}

	void CloseDialog(bool bWasPicked=false)
	{
		bOkClicked = bWasPicked;
		if (Window.IsValid())
		{
			Window->RequestDestroyWindow();
		}
	}

	/** Handler for when cancel is clicked */
	FReply CancelClicked()
	{
		CloseDialog();
		return FReply::Handled();
	}

	FReply OnKeyDown( const FGeometry& MyGeometry, const FKeyEvent& InKeyEvent )
	{
		if (InKeyEvent.GetKey() == EKeys::Escape)
		{
			CloseDialog();
			return FReply::Handled();
		}
		return SWidget::OnKeyDown(MyGeometry, InKeyEvent);
	}

private:
	/** The factory for which we are setting up properties */
	TWeakObjectPtr<UChooserTableFactory> ChooserFactory;
	TSharedPtr<SWindow> Window;
	TSharedPtr<IDetailsView> DetailsView;

	/** True if Ok was clicked */
	bool bOkClicked = false;
};

UChooserTableFactory::UChooserTableFactory()
{
	bCreateNew = true;
	bEditAfterNew = true;
	SupportedClass = UChooserTable::StaticClass();
}

bool UChooserTableFactory::ConfigureProperties()
{
	TSharedRef<SChooserCreateDialog> Dialog = SNew(SChooserCreateDialog);
	return Dialog->ConfigureProperties(this);
}

UObject* UChooserTableFactory::FactoryCreateNew(UClass* Class, UObject* InParent, FName Name, EObjectFlags Flags, UObject* Context, FFeedbackContext* Warn, FName CallingContext)
{
	UChooserTable* NewChooser =  NewObject<UChooserTable>(InParent, Class, Name, Flags | RF_Public | RF_Standalone | RF_Transactional | RF_LoadCompleted);
	if (ChooserInitializer.IsValid())
	{
		ChooserInitializer.Get<FChooserInitializer>().Initialize(NewChooser);
		UChooserEditorSettings::Get().DefaultCreateType = ChooserInitializer.GetScriptStruct()->GetStructPathName().ToString();
	}
	return NewChooser;
}

#undef LOCTEXT_NAMESPACE