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

#include "CoreMinimal.h"
#include "UObject/Class.h"
#include "UObject/UnrealType.h"
#include "Animation/AnimBlueprintGeneratedClass.h"
#include "BlueprintCompilerCppBackendUtils.h"
#include "Animation/AnimClassData.h"
#include "Animation/AnimNodeBase.h"

void FBackendHelperAnim::AddHeaders(FEmitterLocalContext& Context)
{
	if (Cast<UAnimBlueprintGeneratedClass>(Context.GetCurrentlyGeneratedClass()))
	{
		Context.Header.AddLine(TEXT("#include \"Animation/AnimClassData.h\""));
		Context.Body.AddLine(TEXT("#include \"Animation/BlendProfile.h\""));
	}
}

void FBackendHelperAnim::CreateAnimClassData(FEmitterLocalContext& Context)
{
	if (UAnimBlueprintGeneratedClass* AnimClass = Cast<UAnimBlueprintGeneratedClass>(Context.GetCurrentlyGeneratedClass()))
	{
		const FString LocalNativeName = Context.GenerateUniqueLocalName();
		Context.AddLine(FString::Printf(TEXT("UAnimClassData* %s = NewObject<UAnimClassData>(InDynamicClass, TEXT(\"AnimClassData\"));"), *LocalNativeName));

		UAnimClassData* AnimClassData = NewObject<UAnimClassData>(GetTransientPackage(), TEXT("AnimClassData"));
		AnimClassData->CopyFrom(AnimClass);

		UObject* ObjectArchetype = AnimClassData->GetArchetype();
		for (const FProperty* Property : TFieldRange<const FProperty>(UAnimClassData::StaticClass()))
		{
			FEmitDefaultValueHelper::OuterGenerate(Context, Property, LocalNativeName
				, reinterpret_cast<const uint8*>(AnimClassData)
				, reinterpret_cast<const uint8*>(ObjectArchetype)
				, FEmitDefaultValueHelper::EPropertyAccessOperator::Pointer
				, FEmitDefaultValueHelper::EPropertyGenerationControlFlags::AllowTransient);
		}

		Context.AddLine(FString::Printf(TEXT("InDynamicClass->%s = %s;"), GET_MEMBER_NAME_STRING_CHECKED(UDynamicClass, AnimClassImplementation), *LocalNativeName));
		Context.AddLine(FString::Printf(TEXT("%s->DynamicClassInitialization(InDynamicClass);"), *LocalNativeName));
	}
}

bool FBackendHelperAnim::ShouldAddAnimNodeInitializationFunctionCall(FEmitterLocalContext& Context, const FProperty* InProperty)
{
	if(const FStructProperty* StructProperty = CastField<const FStructProperty>(InProperty))
	{
		if(StructProperty->Struct->IsChildOf(FAnimNode_Base::StaticStruct()))
		{
			return true;
		}
	}

	return false;
}

void FBackendHelperAnim::AddAnimNodeInitializationFunctionCall(FEmitterLocalContext& Context, const FProperty* InProperty)
{
	if(const FStructProperty* StructProperty = CastField<const FStructProperty>(InProperty))
	{
		if(StructProperty->Struct->IsChildOf(FAnimNode_Base::StaticStruct()))
		{
			Context.Body.AddLine(FString::Printf(TEXT("__InitAnimNode__%s();"), *StructProperty->GetName()));
		}
	}
}

void FBackendHelperAnim::AddAnimNodeInitializationFunction(FEmitterLocalContext& Context, const FString& InCppClassName, const FProperty* InProperty, bool bInNewProperty, UObject* InCDO, UObject* InParentCDO)
{
	if(const FStructProperty* StructProperty = CastField<const FStructProperty>(InProperty))
	{
		if(StructProperty->Struct->IsChildOf(FAnimNode_Base::StaticStruct()))
		{
			Context.Header.AddLine(FString::Printf(TEXT("void __InitAnimNode__%s();"), *StructProperty->GetName()));

			Context.Body.AddLine(FString::Printf(TEXT("void %s::__InitAnimNode__%s()"), *InCppClassName, *StructProperty->GetName()));
			Context.Body.AddLine(TEXT("{"));
			Context.Body.IncreaseIndent();

			FEmitDefaultValueHelper::OuterGenerate(Context, InProperty, TEXT(""), reinterpret_cast<const uint8*>(InCDO), bInNewProperty ? nullptr : reinterpret_cast<const uint8*>(InParentCDO), FEmitDefaultValueHelper::EPropertyAccessOperator::None, FEmitDefaultValueHelper::EPropertyGenerationControlFlags::AllowProtected);
			
			// anim nodes constructed, finish anim node initialization:
			// @TODO: if we are to support nativizaton of anim BPs this will need to be reworked
// 			if (UAnimBlueprintGeneratedClass* AnimClass = Cast<UAnimBlueprintGeneratedClass>(Context.GetCurrentlyGeneratedClass()))
// 			{
// 				for (int32 i = 0; i < AnimClass->GetExposedValueHandlers().Num(); ++i)
// 				{
// 					if (AnimClass->GetExposedValueHandlers()[i].ValueHandlerNodeProperty == InProperty)
// 					{
// 						const FString ClassName = FEmitHelper::GetCppName(Context.GetCurrentlyGeneratedClass());
// 						const FString MemberName = FEmitHelper::GetCppName(const_cast<FProperty*>(InProperty));
// 						Context.Body.AddLine(FString::Printf(TEXT("%s.SetExposedValueHandler(&CastChecked<UAnimClassData>(CastChecked<UDynamicClass>(%s::StaticClass())->%s)->GetExposedValueHandlers()[%d]);"), *MemberName, *ClassName, GET_MEMBER_NAME_STRING_CHECKED(UDynamicClass, AnimClassImplementation), i));
// 
// 						break;
// 					}
// 				}
// 			}

			Context.Body.DecreaseIndent();
			Context.Body.AddLine(TEXT("}"));
		}
	}
}

void FBackendHelperAnim::AddAllAnimNodesInitializationFunction(FEmitterLocalContext& Context, const FString& InCppClassName, const TArray<const FProperty*>& InAnimProperties)
{
	Context.Header.AddLine(TEXT("void __InitAllAnimNodes();"));

	Context.Body.AddLine(FString::Printf(TEXT("void %s::__InitAllAnimNodes()"), *InCppClassName));
	Context.Body.AddLine(TEXT("{"));
	Context.Body.IncreaseIndent();

	for(const FProperty* Property : InAnimProperties)
	{
		Context.Body.AddLine(FString::Printf(TEXT("__InitAnimNode__%s();"), *Property->GetName()));
	}

	Context.Body.DecreaseIndent();
	Context.Body.AddLine(TEXT("}"));
}

void FBackendHelperAnim::AddAllAnimNodesInitializationFunctionCall(FEmitterLocalContext& Context)
{
	Context.Body.AddLine(TEXT("__InitAllAnimNodes();"));
}