RemoveIf.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreTypes.h"
#include "Templates/Invoke.h"
 
namespace Algo
{
    template <typename RangeType, typename Predicate>
    [[nodiscard]] int32 RemoveIf(RangeType& Range, Predicate Pred)
    {
        auto* First = GetData(Range);
        auto* Last  = First + GetNum(Range);
 
        auto* IterStart = First;
        auto* IterEnd   = Last;
        for (;;)
        {
            // Skip non-removed elements at the start
            for (;;)
            {
                if (IterStart == IterEnd)
                {
                    return UE_PTRDIFF_TO_INT32(IterStart - First);
                }
 
                if (Invoke(Pred, *IterStart))
                {
                    break;
                }
 
                ++IterStart;
            }
 
            // Skip removed elements at the end
            for (;;)
            {
                --IterEnd;
 
                if (IterStart == IterEnd)
                {
                    return UE_PTRDIFF_TO_INT32(IterStart - First);
                }
 
                if (!Invoke(Pred, *IterEnd))
                {
                    break;
                }
            }
 
            *IterStart = MoveTemp(*IterEnd);
 
            ++IterStart;
        }
    }
 
    template <typename RangeType, typename Predicate>
    [[nodiscard]] int32 StableRemoveIf(RangeType&& Range, Predicate Pred)
    {
        auto* First = GetData(Range);
        auto* Last  = First + GetNum(Range);
 
        auto* IterStart = First;
 
        // Skip non-removed elements at the start
        for (;;)
        {
            if (IterStart == Last)
            {
                return UE_PTRDIFF_TO_INT32(IterStart - First);
            }
 
            if (Invoke(Pred, *IterStart))
            {
                break;
            }
 
            ++IterStart;
        }
 
        auto* IterKeep = IterStart;
        ++IterKeep;
 
        for (;;)
        {
            if (IterKeep == Last)
            {
                return UE_PTRDIFF_TO_INT32(IterStart - First);
            }
 
            if (!Invoke(Pred, *IterKeep))
            {
                *IterStart++ = MoveTemp(*IterKeep);
            }
 
            ++IterKeep;
        }
    }
}