ActorIterator.h

#pragma once
 
#include "krpch.h"
 
#include "KarmaUtilities.h"
 
namespace Karma
{
    class AActor;
 
    class FActorIterator : public TActorIteratorBase<FActorIterator>
    {
        friend class TActorIteratorBase<FActorIterator>;
        typedef TActorIteratorBase<FActorIterator> Super;
 
    public:
        explicit FActorIterator(const UWorld* InWorld, const EActorIteratorFlags InFlags = EActorIteratorFlags(OnlyActiveLevels | SkipPendingKill))
            : Super(InWorld, AActor::StaticClass(), InFlags)
        {
            ++(*this);
        }
 
        explicit FActorIterator(const UWorld* InWorld, const TSubclassOf<AActor> InClass, const EActorIteratorFlags InFlags = EActorIteratorFlags(OnlyActiveLevels | SkipPendingKill))
            : Super(InWorld, InClass, InFlags)
        {
            ++(*this);
        }
    };
 
    template <typename ActorType>
    class TActorIterator : public TActorIteratorBase<TActorIterator<ActorType>>
    {
        friend class TActorIteratorBase<TActorIterator>;
        typedef TActorIteratorBase<TActorIterator> Super;
 
    public:
        explicit TActorIterator(const UWorld* InWorld, TSubclassOf<AActor> InClass = ActorType::StaticClass(), EActorIteratorFlags InFlags = EActorIteratorFlags(EActorIteratorFlags::OnlyActiveLevels | EActorIteratorFlags::SkipPendingKill))
            : Super(InWorld, InClass, InFlags)
        {
            ++(*this);
        }
 
        explicit TActorIterator(EActorIteratorType)
            : Super(EActorIteratorType::End)
        {
        }
 
        FORCEINLINE ActorType* operator*() const
        {
            return CastChecked<ActorType>(**static_cast<const Super*>(this));
        }
 
        FORCEINLINE ActorType* operator->() const
        {
            return **this;
        }
    };
}