UObjectPoolManager.cpp

이 파일의 문서화 페이지로 가기

// Copyright (c) 2025 Doppleddiggong. All rights reserved. Unauthorized copying, modification, or distribution of this file, via any medium is strictly prohibited. Proprietary and confidential.
 
#include "UObjectPoolManager.h"
#include "Engine/World.h"
#include "GameFramework/CharacterMovementComponent.h"
#include "GameFramework/ProjectileMovementComponent.h"
 
 
void UObjectPoolManager::Initialize(FSubsystemCollectionBase& Collection)
{
    Super::Initialize(Collection);
 
    UE_LOG(LogTemp, Log, TEXT("ObjectPoolManager Initialized."));
}
 
void UObjectPoolManager::Deinitialize()
{
    for (auto& Elem : PoolMap)
    {
        for (AActor* Actor : Elem.Value)
        {
            if (IsValid(Actor))
            {
                Actor->Destroy();
            }
        }
    }
    PoolMap.Empty();
    
    Super::Deinitialize();
}
 
AActor* UObjectPoolManager::GetPoolItem(const UObject* WorldContextObject, const TSubclassOf<AActor> InClass )
{
    return GetPoolItemLocationRotator(WorldContextObject, InClass, FVector::ZeroVector, FRotator::ZeroRotator );
}
 
AActor* UObjectPoolManager::GetPoolItemLocationRotator(const UObject* WorldContextObject, const TSubclassOf<AActor> InClass, const FVector Location, const FRotator Rotator  )
{
    if (!InClass)
    {
        UE_LOG(LogTemp, Error, TEXT("GetPoolItem failed: InClass is null."));
        return nullptr;
    }
    
    TArray<AActor*>* Pool = PoolMap.Find(InClass);
    if (Pool && Pool->Num() > 0)
    {
        AActor* Actor = Pool->Pop();
        Actor->SetActorLocationAndRotation(Location, Rotator);
 
        Actor->SetActorHiddenInGame(false);
        Actor->SetActorEnableCollision(true);
        Actor->SetActorTickEnabled(true);
        
        return Actor;
    }
 
    if (UWorld* World = GEngine->GetWorldFromContextObject(WorldContextObject, EGetWorldErrorMode::LogAndReturnNull))
        return World->SpawnActor<AActor>(InClass, Location, Rotator);
 
    return nullptr;
}
 
void UObjectPoolManager::ReturnActorToPool(AActor* Actor)
{
    if ( Actor == nullptr )
        return;
        
    Actor->SetActorHiddenInGame(true);
    Actor->SetActorEnableCollision(false);
    Actor->SetActorTickEnabled(false);
 
    // 타이머/수명 정리
    if (UWorld* World = Actor->GetWorld())
    {
        World->GetTimerManager().ClearAllTimersForObject(Actor);
    }
    Actor->SetLifeSpan(0.f);
 
    if (UProjectileMovementComponent* Proj = Actor->FindComponentByClass<UProjectileMovementComponent>())
    {
        Proj->StopMovementImmediately();
        Proj->Deactivate();
    }
    if (UCharacterMovementComponent* CharMove = Actor->FindComponentByClass<UCharacterMovementComponent>())
    {
        CharMove->StopMovementImmediately();
        CharMove->Velocity = FVector::ZeroVector;
    }
    if (UPrimitiveComponent* Prim = Cast<UPrimitiveComponent>(Actor->GetRootComponent()))
    {
        Prim->SetPhysicsLinearVelocity(FVector::ZeroVector, true);
        Prim->SetPhysicsAngularVelocityInDegrees(FVector::ZeroVector, true);
        Prim->PutRigidBodyToSleep();
    }
 
    TSubclassOf<AActor> ActorClass = Actor->GetClass();
    TArray<AActor*>& Pool = PoolMap.FindOrAdd(ActorClass);
    Pool.Add(Actor);
}