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

#pragma once
#include "CoreMinimal.h"
#include "UObject/Object.h"
#include "UObject/Interface.h"
#include "Components/SkeletalMeshComponent.h"


struct FSkeletalMeshEditorParams
{
	void SaveState(USkeletalMeshComponent* InSkelMeshComp)
	{
		if (InSkelMeshComp)
		{
			ChildSkelMesh = InSkelMeshComp;
			VisibilityBasedAnimTickOption = InSkelMeshComp->VisibilityBasedAnimTickOption;
			InSkelMeshComp->VisibilityBasedAnimTickOption = EVisibilityBasedAnimTickOption::AlwaysTickPoseAndRefreshBones;
			PredictedLODLevel = InSkelMeshComp->GetPredictedLODLevel();
#if WITH_EDITOR
			bUpdateAnimationInEditor = InSkelMeshComp->GetUpdateAnimationInEditor();
			bUpdateClothInEditor = InSkelMeshComp->GetUpdateClothInEditor();
			InSkelMeshComp->SetUpdateAnimationInEditor(true);
			InSkelMeshComp->SetUpdateClothInEditor(true);
#endif
		}
	}

	void RestoreState()
	{
		if (ChildSkelMesh.IsValid())
		{
			ChildSkelMesh->VisibilityBasedAnimTickOption = VisibilityBasedAnimTickOption;
			ChildSkelMesh->SetPredictedLODLevel(PredictedLODLevel);
#if WITH_EDITOR

			ChildSkelMesh->SetUpdateAnimationInEditor(bUpdateAnimationInEditor);
			ChildSkelMesh->SetUpdateClothInEditor(bUpdateClothInEditor);
#endif
		}
	}
	void RestoreLOD()
	{
		if (ChildSkelMesh.IsValid())
		{
			ChildSkelMesh->SetPredictedLODLevel(PredictedLODLevel);
		}
	}
	TWeakObjectPtr<USkeletalMeshComponent> ChildSkelMesh;
	EVisibilityBasedAnimTickOption VisibilityBasedAnimTickOption;
	int32 PredictedLODLevel;
#if WITH_EDITOR
	bool bUpdateAnimationInEditor;
	bool bUpdateClothInEditor;
#endif
};

struct FSkeletalMeshRestoreState
{
	void SaveState(USkeletalMeshComponent* InComponent)
	{
		SkeletalMeshCompEditorParams.SetNum(0);
		FSkeletalMeshEditorParams Parent;
		Parent.SaveState(InComponent);
		SkeletalMeshCompEditorParams.Add(Parent);
		TArray<USceneComponent*> ChildComponents;
		InComponent->GetChildrenComponents(true, ChildComponents);
		for (USceneComponent* ChildComponent : ChildComponents)
		{
			USkeletalMeshComponent* SkelMeshComp = Cast<USkeletalMeshComponent>(ChildComponent);
			if (SkelMeshComp)
			{
				FSkeletalMeshEditorParams Params;
				Params.SaveState(SkelMeshComp);
				SkeletalMeshCompEditorParams.Add(Params);
			}
		}
	}

	UE_DEPRECATED(5.6, "InComponent is not required")
	void RestoreState(USkeletalMeshComponent* InComponent)
	{
		RestoreState();
	}
	UE_DEPRECATED(5.6, "InComponent is not required")
	void RestoreLOD(USkeletalMeshComponent* InComponent)
	{
		RestoreLOD();
	}

	void RestoreState()
	{
		for (FSkeletalMeshEditorParams& ChildParams : SkeletalMeshCompEditorParams)
		{
			ChildParams.RestoreState();
		}
	}
	void RestoreLOD()
	{
		for (FSkeletalMeshEditorParams& ChildParams : SkeletalMeshCompEditorParams)
		{
			ChildParams.RestoreLOD();
		}
	}
	TArray<FSkeletalMeshEditorParams> SkeletalMeshCompEditorParams;
	
};