UnrealEngineUWP/Engine/Source/Editor/Persona/Private/AnimInstanceDetails.cpp

// Copyright 1998-2015 Epic Games, Inc. All Rights Reserved.

#include "PersonaPrivatePCH.h"
#include "DetailLayoutBuilder.h"
#include "PropertyHandle.h"
#include "PropertyEditorModule.h"
#include "AnimInstanceDetails.h"
#include "ObjectEditorUtils.h"
#include "DetailCategoryBuilder.h"
#include "DetailWidgetRow.h"
#include "IDetailPropertyRow.h"
#include "IDetailCustomNodeBuilder.h"
#include "PropertyCustomizationHelpers.h"
#include "AssetData.h"

/////////////////////////////////////////////////////
// FAnimInstanceDetails 

TSharedRef<IDetailCustomization> FAnimInstanceDetails::MakeInstance()
{
	return MakeShareable(new FAnimInstanceDetails());
}

void FAnimInstanceDetails::CustomizeDetails(class IDetailLayoutBuilder& DetailBuilder)
{
	TArray<TWeakObjectPtr<UObject>> Objects;
	DetailBuilder.GetObjectsBeingCustomized(Objects);

	TArray<UClass*> ObjectClasses;
	USkeleton* TargetSkeleton = NULL;

	// Grab the skeleton we are displaying for filtering
	for (TWeakObjectPtr<UObject> Object : Objects)
	{
		if (UObject* ObjectPtr = Object.Get())
		{
			ObjectClasses.AddUnique(Object->GetClass());
			if (UAnimInstance* AnimInst = Cast<UAnimInstance>(ObjectPtr))
			{
				if (TargetSkeleton && TargetSkeleton != AnimInst->CurrentSkeleton)
				{
					TargetSkeleton = NULL;
					break;
				}
				else
				{
					TargetSkeleton = AnimInst->CurrentSkeleton;
				}
			}
		}
	}

	// Grab common base class for property population
	UClass* CommonBaseClass = UClass::FindCommonBase(ObjectClasses);

	// If everything uses the same skeleton we can filter based on that skeleton
	if (TargetSkeleton)
	{
		TargetSkeletonName = FString::Printf(TEXT("%s'%s'"), *TargetSkeleton->GetClass()->GetName(), *TargetSkeleton->GetPathName());

		for (TFieldIterator<UProperty> It(CommonBaseClass, EFieldIteratorFlags::IncludeSuper); It; ++It)
		{
			UProperty* TargetProperty = *It;

			TSharedRef<IPropertyHandle> TargetPropertyHandle = DetailBuilder.GetProperty(*TargetProperty->GetName(), CommonBaseClass);
			if ((*TargetPropertyHandle).GetProperty() )
			{
				IDetailCategoryBuilder& CurrentCategory = DetailBuilder.EditCategory(FObjectEditorUtils::GetCategoryFName(TargetProperty));
				
				IDetailPropertyRow& PropertyRow = CurrentCategory.AddProperty(TargetPropertyHandle);

				TSharedPtr<SWidget> NameWidget;
				TSharedPtr<SWidget> ValueWidget;
				FDetailWidgetRow Row;
				PropertyRow.GetDefaultWidgets(NameWidget, ValueWidget, Row);

				TSharedRef<SWidget> TempWidget = CreateFilteredObjectPropertyWidget(TargetProperty, TargetPropertyHandle);
				ValueWidget = (TempWidget == SNullWidget::NullWidget) ? ValueWidget : TempWidget;

				const bool bShowChildren = true;
				PropertyRow.CustomWidget(bShowChildren)
					.NameContent()
					.MinDesiredWidth(Row.NameWidget.MinWidth)
					.MaxDesiredWidth(Row.NameWidget.MaxWidth)
					[
						NameWidget.ToSharedRef()
					]
				.ValueContent()
					.MinDesiredWidth(Row.ValueWidget.MinWidth)
					.MaxDesiredWidth(Row.ValueWidget.MaxWidth)
					[
						ValueWidget.ToSharedRef()
					];
			}
		}
	}
}

TSharedRef<SWidget> FAnimInstanceDetails::CreateFilteredObjectPropertyWidget(UProperty* TargetProperty, TSharedRef<IPropertyHandle> TargetPropertyHandle)
{
	if(const UObjectPropertyBase* ObjectProperty = Cast<const UObjectPropertyBase>( TargetProperty ))
	{
		if(ObjectProperty->PropertyClass->IsChildOf(UAnimationAsset::StaticClass()))
		{
			bool bAllowClear = !(ObjectProperty->PropertyFlags & CPF_NoClear);

			return SNew(SObjectPropertyEntryBox)
				.PropertyHandle(TargetPropertyHandle)
				.AllowedClass(ObjectProperty->PropertyClass)
				.AllowClear(bAllowClear)
				.OnShouldFilterAsset(FOnShouldFilterAsset::CreateSP(this, &FAnimInstanceDetails::OnShouldFilterAnimAsset));
		}
	}

	return SNullWidget::NullWidget;
}

bool FAnimInstanceDetails::OnShouldFilterAnimAsset(const FAssetData& AssetData) const
{
	const FString* SkeletonName = AssetData.TagsAndValues.Find(TEXT("Skeleton"));
	return *SkeletonName != TargetSkeletonName;
}