160 lines
6.2 KiB
C++
160 lines
6.2 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#include "SCreateAssetFromObject.h"
|
|
#include "Modules/ModuleManager.h"
|
|
#include "Widgets/SBoxPanel.h"
|
|
#include "Widgets/Text/STextBlock.h"
|
|
#include "Widgets/Input/SEditableTextBox.h"
|
|
#include "Widgets/Input/SButton.h"
|
|
#include "EditorStyleSet.h"
|
|
#include "GameFramework/Actor.h"
|
|
#include "AssetData.h"
|
|
#include "Engine/Selection.h"
|
|
#include "Editor.h"
|
|
#include "Misc/FileHelper.h"
|
|
#include "IAssetTools.h"
|
|
#include "IContentBrowserSingleton.h"
|
|
#include "ContentBrowserModule.h"
|
|
#include "AssetToolsModule.h"
|
|
#include "AssetRegistryModule.h"
|
|
#include "PackageTools.h"
|
|
|
|
#define LOCTEXT_NAMESPACE "SCreateAssetFromActor"
|
|
|
|
void SCreateAssetFromObject::Construct(const FArguments& InArgs, TSharedPtr<SWindow> InParentWindow)
|
|
{
|
|
AssetFilenameSuffix = InArgs._AssetFilenameSuffix;
|
|
HeadingText = InArgs._HeadingText;
|
|
CreateButtonText = InArgs._CreateButtonText;
|
|
OnCreateAssetAction = InArgs._OnCreateAssetAction;
|
|
|
|
if (InArgs._AssetPath.IsEmpty())
|
|
{
|
|
AssetPath = FString("/Game");
|
|
}
|
|
else
|
|
{
|
|
AssetPath = InArgs._AssetPath;
|
|
}
|
|
|
|
bIsReportingError = false;
|
|
|
|
FPathPickerConfig PathPickerConfig;
|
|
PathPickerConfig.DefaultPath = AssetPath;
|
|
PathPickerConfig.OnPathSelected = FOnPathSelected::CreateRaw(this, &SCreateAssetFromObject::OnSelectAssetPath);
|
|
|
|
SelectionDelegateHandle = USelection::SelectionChangedEvent.AddSP(this, &SCreateAssetFromObject::OnLevelSelectionChanged);
|
|
|
|
// Set up PathPickerConfig.
|
|
FContentBrowserModule& ContentBrowserModule = FModuleManager::LoadModuleChecked<FContentBrowserModule>("ContentBrowser");
|
|
|
|
ParentWindow = InParentWindow;
|
|
|
|
FString PackageName;
|
|
ActorInstanceLabel.Empty();
|
|
|
|
if (InArgs._DefaultNameOverride.IsEmpty())
|
|
{
|
|
USelection* SelectedActors = GEditor->GetSelectedActors();
|
|
for (FSelectionIterator Iter(*SelectedActors); Iter; ++Iter)
|
|
{
|
|
AActor* Actor = Cast<AActor>(*Iter);
|
|
if (Actor)
|
|
{
|
|
ActorInstanceLabel += Actor->GetActorLabel();
|
|
ActorInstanceLabel += TEXT("_");
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
ActorInstanceLabel = InArgs._DefaultNameOverride.ToString();
|
|
}
|
|
|
|
ActorInstanceLabel = UPackageTools::SanitizePackageName(ActorInstanceLabel + AssetFilenameSuffix);
|
|
|
|
FString AssetName = ActorInstanceLabel;
|
|
FString BasePath = AssetPath / AssetName;
|
|
FAssetToolsModule& AssetToolsModule = FModuleManager::LoadModuleChecked<FAssetToolsModule>("AssetTools");
|
|
AssetToolsModule.Get().CreateUniqueAssetName(BasePath, TEXT(""), PackageName, AssetName);
|
|
|
|
ChildSlot
|
|
[SNew(SVerticalBox) + SVerticalBox::Slot().FillHeight(1.0f)[ContentBrowserModule.Get().CreatePathPicker(PathPickerConfig)] + SVerticalBox::Slot().AutoHeight()[SNew(SHorizontalBox) + SHorizontalBox::Slot().Padding(0.0f, 0.0f, 4.0f, 0.0f).AutoWidth()[SNew(STextBlock).Text(HeadingText)] + SHorizontalBox::Slot().FillWidth(1.0f)[SAssignNew(FileNameWidget, SEditableTextBox).Text(FText::FromString(AssetName)).OnTextChanged(this, &SCreateAssetFromObject::OnFilenameChanged)]] + SVerticalBox::Slot().HAlign(HAlign_Right).Padding(0, 20, 0, 0).AutoHeight()[SNew(SHorizontalBox) + SHorizontalBox::Slot().Padding(0, 2, 6, 0).AutoWidth()[SNew(SButton).VAlign(VAlign_Bottom).ContentPadding(FMargin(8, 2, 8, 2)).OnClicked(this, &SCreateAssetFromObject::OnCreateAssetFromActorClicked).IsEnabled(this, &SCreateAssetFromObject::IsCreateAssetFromActorEnabled).ButtonStyle(FEditorStyle::Get(), "FlatButton.Success").TextStyle(FEditorStyle::Get(), "FlatButton.DefaultTextStyle").Text(CreateButtonText)] + SHorizontalBox::Slot().Padding(0, 2, 0, 0).AutoWidth()[SNew(SButton).VAlign(VAlign_Bottom).ContentPadding(FMargin(8, 2, 8, 2)).OnClicked(this, &SCreateAssetFromObject::OnCancelCreateAssetFromActor).ButtonStyle(FEditorStyle::Get(), "FlatButton.Default").TextStyle(FEditorStyle::Get(), "FlatButton.DefaultTextStyle").Text(LOCTEXT("CancelButtonText", "Cancel"))]]];
|
|
|
|
OnFilenameChanged(FText::FromString(AssetName));
|
|
}
|
|
|
|
void SCreateAssetFromObject::RequestDestroyParentWindow()
|
|
{
|
|
USelection::SelectionChangedEvent.Remove(SelectionDelegateHandle);
|
|
|
|
if (ParentWindow.IsValid())
|
|
{
|
|
ParentWindow.Pin()->RequestDestroyWindow();
|
|
}
|
|
}
|
|
|
|
FReply SCreateAssetFromObject::OnCreateAssetFromActorClicked()
|
|
{
|
|
RequestDestroyParentWindow();
|
|
OnCreateAssetAction.ExecuteIfBound(AssetPath / FileNameWidget->GetText().ToString());
|
|
return FReply::Handled();
|
|
}
|
|
|
|
FReply SCreateAssetFromObject::OnCancelCreateAssetFromActor()
|
|
{
|
|
RequestDestroyParentWindow();
|
|
return FReply::Handled();
|
|
}
|
|
|
|
void SCreateAssetFromObject::OnSelectAssetPath(const FString& Path)
|
|
{
|
|
AssetPath = Path;
|
|
OnFilenameChanged(FileNameWidget->GetText());
|
|
}
|
|
|
|
void SCreateAssetFromObject::OnLevelSelectionChanged(UObject* InObjectSelected)
|
|
{
|
|
// When actor selection changes, this window should be destroyed.
|
|
RequestDestroyParentWindow();
|
|
}
|
|
|
|
void SCreateAssetFromObject::OnFilenameChanged(const FText& InNewName)
|
|
{
|
|
TArray<FAssetData> AssetData;
|
|
FAssetRegistryModule& AssetRegistryModule = FModuleManager::LoadModuleChecked<FAssetRegistryModule>("AssetRegistry");
|
|
AssetRegistryModule.Get().GetAssetsByPath(FName(*AssetPath), AssetData);
|
|
|
|
FText ErrorText;
|
|
if (!FFileHelper::IsFilenameValidForSaving(InNewName.ToString(), ErrorText) || !FName(*InNewName.ToString()).IsValidObjectName(ErrorText))
|
|
{
|
|
FileNameWidget->SetError(ErrorText);
|
|
bIsReportingError = true;
|
|
return;
|
|
}
|
|
else
|
|
{
|
|
// Check to see if the name conflicts
|
|
for (auto Iter = AssetData.CreateConstIterator(); Iter; ++Iter)
|
|
{
|
|
if (Iter->AssetName.ToString() == InNewName.ToString())
|
|
{
|
|
FileNameWidget->SetError(LOCTEXT("AssetInUseError", "Asset name already in use!"));
|
|
bIsReportingError = true;
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
FileNameWidget->SetError(FText::FromString(TEXT("")));
|
|
bIsReportingError = false;
|
|
}
|
|
|
|
bool SCreateAssetFromObject::IsCreateAssetFromActorEnabled() const
|
|
{
|
|
return !bIsReportingError;
|
|
}
|
|
|
|
#undef LOCTEXT_NAMESPACE
|