SkeletalMeshRestoreState.h

Go to the documentation of this file.

// 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;
    
};