108 lines
4.0 KiB
C++
108 lines
4.0 KiB
C++
|
// Copyright Epic Games, Inc. All Rights Reserved.
|
||
|
|
|
||
|
|
/*=============================================================================
|
||
|
|
SkeletonFactory.cpp: Factory for Skeletons
|
||
|
|
=============================================================================*/
|
||
|
|
|
||
|
|
#include "Factories/SkeletonFactory.h"
|
||
|
|
#include "Misc/MessageDialog.h"
|
||
|
|
#include "Modules/ModuleManager.h"
|
||
|
|
#include "Widgets/DeclarativeSyntaxSupport.h"
|
||
|
|
#include "Widgets/SWindow.h"
|
||
|
|
#include "Widgets/Layout/SBorder.h"
|
||
|
|
#include "EditorStyleSet.h"
|
||
|
|
#include "Animation/Skeleton.h"
|
||
|
|
#include "Engine/SkeletalMesh.h"
|
||
|
|
#include "Editor.h"
|
||
|
|
|
||
|
|
#include "IContentBrowserSingleton.h"
|
||
|
|
#include "ContentBrowserModule.h"
|
||
|
|
|
||
|
|
#define LOCTEXT_NAMESPACE "SkeletonFactory"
|
||
|
|
|
||
|
|
/*------------------------------------------------------------------------------
|
||
|
|
USkeletonFactory implementation.
|
||
|
|
------------------------------------------------------------------------------*/
|
||
|
|
|
||
|
|
USkeletonFactory::USkeletonFactory(const FObjectInitializer& ObjectInitializer)
|
||
|
|
: Super(ObjectInitializer)
|
||
|
|
{
|
||
|
|
bCreateNew = false;
|
||
|
|
SupportedClass = USkeleton::StaticClass();
|
||
|
|
}
|
||
|
|
|
||
|
|
bool USkeletonFactory::ConfigureProperties()
|
||
|
|
{
|
||
|
|
// Null the parent class so we can check for selection later
|
||
|
|
TargetSkeletalMesh = NULL;
|
||
|
|
|
||
|
|
// Load the content browser module to display an asset picker
|
||
|
|
FContentBrowserModule& ContentBrowserModule = FModuleManager::LoadModuleChecked<FContentBrowserModule>("ContentBrowser");
|
||
|
|
|
||
|
|
FAssetPickerConfig AssetPickerConfig;
|
||
|
|
|
||
|
|
/** The asset picker will only show skeletal meshes */
|
||
|
|
AssetPickerConfig.Filter.ClassNames.Add(USkeletalMesh::StaticClass()->GetFName());
|
||
|
|
AssetPickerConfig.Filter.bRecursiveClasses = true;
|
||
|
|
|
||
|
|
/** The delegate that fires when an asset was selected */
|
||
|
|
AssetPickerConfig.OnAssetSelected = FOnAssetSelected::CreateUObject(this, &USkeletonFactory::OnTargetSkeletalMeshSelected);
|
||
|
|
|
||
|
|
/** The default view mode should be a list view */
|
||
|
|
AssetPickerConfig.InitialAssetViewType = EAssetViewType::List;
|
||
|
|
|
||
|
|
PickerWindow = SNew(SWindow)
|
||
|
|
.Title(LOCTEXT("CreateSkeletonOptions", "Pick Skeletal Mesh"))
|
||
|
|
.ClientSize(FVector2D(500, 600))
|
||
|
|
.SupportsMinimize(false)
|
||
|
|
.SupportsMaximize(false)
|
||
|
|
[SNew(SBorder)
|
||
|
|
.BorderImage(FEditorStyle::GetBrush("Menu.Background"))
|
||
|
|
[ContentBrowserModule.Get().CreateAssetPicker(AssetPickerConfig)]];
|
||
|
|
|
||
|
|
GEditor->EditorAddModalWindow(PickerWindow.ToSharedRef());
|
||
|
|
PickerWindow.Reset();
|
||
|
|
|
||
|
|
return TargetSkeletalMesh != NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
UObject* USkeletonFactory::FactoryCreateNew(UClass* Class, UObject* InParent, FName Name, EObjectFlags Flags, UObject* Context, FFeedbackContext* Warn)
|
||
|
|
{
|
||
|
|
// Make sure we are trying to factory a skeleton, then create and init one
|
||
|
|
check(Class->IsChildOf(USkeleton::StaticClass()));
|
||
|
|
|
||
|
|
if (TargetSkeletalMesh == NULL)
|
||
|
|
{
|
||
|
|
FMessageDialog::Open(EAppMsgType::Ok, LOCTEXT("InvalidSkeletalMesh", "Must specify a valid skeletal mesh for the skeleton to target."));
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
USkeleton* NewAsset = NewObject<USkeleton>(InParent, Class, Name, Flags);
|
||
|
|
if (NewAsset)
|
||
|
|
{
|
||
|
|
// this should not fail. If so there is problem.
|
||
|
|
if (!NewAsset->MergeAllBonesToBoneTree(TargetSkeletalMesh))
|
||
|
|
{
|
||
|
|
FMessageDialog::Open(EAppMsgType::Ok, LOCTEXT("CreateNewSkeletonFailed_BoneMerge", "Failed to create Skeleton : Could not merge bone."));
|
||
|
|
NewAsset->ClearFlags(RF_Public | RF_Standalone);
|
||
|
|
NewAsset = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (TargetSkeletalMesh->GetSkeleton() != NewAsset)
|
||
|
|
{
|
||
|
|
TargetSkeletalMesh->SetSkeleton(NewAsset);
|
||
|
|
TargetSkeletalMesh->MarkPackageDirty();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return NewAsset;
|
||
|
|
}
|
||
|
|
|
||
|
|
void USkeletonFactory::OnTargetSkeletalMeshSelected(const FAssetData& SelectedAsset)
|
||
|
|
{
|
||
|
|
TargetSkeletalMesh = Cast<USkeletalMesh>(SelectedAsset.GetAsset());
|
||
|
|
PickerWindow->RequestDestroyWindow();
|
||
|
|
}
|
||
|
|
|
||
|
|
#undef LOCTEXT_NAMESPACE
|