ToolBuilderUtil.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Algo/Count.h"
#include "Algo/Find.h"
#include "GameFramework/Actor.h"
#include "Engine/Selection.h"
#include "InteractiveToolBuilder.h"
 
namespace ToolBuilderUtil
{
    INTERACTIVETOOLSFRAMEWORK_API
    int CountComponents(const FToolBuilderState& InputState, const TFunction<bool(UActorComponent*)>& Predicate);
 
    INTERACTIVETOOLSFRAMEWORK_API
    UActorComponent* FindFirstComponent(const FToolBuilderState& InputState, const TFunction<bool(UActorComponent*)>& Predicate);
 
    INTERACTIVETOOLSFRAMEWORK_API
    TArray<UActorComponent*> FindAllComponents(const FToolBuilderState& InputState, const TFunction<bool(UActorComponent*)>& Predicate);
 
    // @todo not sure that actors with multiple components are handled properly...
    template<typename ComponentType>
    int CountSelectedComponentsOfType(const FToolBuilderState& InputState);
 
    template<typename ComponentType>
    ComponentType* FindFirstComponentOfType(const FToolBuilderState& InputState);
 
    INTERACTIVETOOLSFRAMEWORK_API
    void EnumerateComponents(const FToolBuilderState& InputState, TFunctionRef<void(UActorComponent*)> ComponentFunc);
 
    INTERACTIVETOOLSFRAMEWORK_API
    int32 CountActors(const FToolBuilderState& InputState, const TFunction<bool(AActor*)>& Predicate);
 
    INTERACTIVETOOLSFRAMEWORK_API
    AActor* FindFirstActor(const FToolBuilderState& InputState, const TFunction<bool(AActor*)>& Predicate);
 
    INTERACTIVETOOLSFRAMEWORK_API
    TArray<AActor*> FindAllActors(const FToolBuilderState& InputState, const TFunction<bool(AActor*)>& Predicate);
 
    template<typename ActorType>
    int CountSelectedActorsOfType(const FToolBuilderState& InputState);
 
    template<typename ActorType>
    ActorType* FindFirstActorOfType(const FToolBuilderState& InputState);
 
    INTERACTIVETOOLSFRAMEWORK_API
    bool ComponentTypeCouldHaveUVs(const UActorComponent& Component);
 
    INTERACTIVETOOLSFRAMEWORK_API
    bool IsVolume(const UActorComponent& Component);
}
 
/*
 * Template Implementations
 */
template<typename ComponentType>
int ToolBuilderUtil::CountSelectedComponentsOfType(const FToolBuilderState& InputState)
{
    return CountComponents(InputState, [](UActorComponent* Component) { return Cast<ComponentType>(Component) != nullptr; });
}
 
template<typename ComponentType>
ComponentType* ToolBuilderUtil::FindFirstComponentOfType(const FToolBuilderState& InputState)
{
    return static_cast<ComponentType*>(FindFirstComponent(InputState, [](UActorComponent* Component) { return Cast<ComponentType>(Component) != nullptr; }));
}
 
template<typename ActorType>
int ToolBuilderUtil::CountSelectedActorsOfType(const FToolBuilderState& InputState)
{
    return CountActors(InputState, [](AActor* Actor) { return Cast<ActorType>(Actor) != nullptr; });
}
 
template<typename ActorType>
ActorType* ToolBuilderUtil::FindFirstActorOfType(const FToolBuilderState& InputState)
{
    AActor* Found = FindFirstActor(InputState, [](AActor* Actor) { return Cast<ActorType>(Actor) != nullptr; });
    return (Found != nullptr) ? Cast<ActorType>(Found) : nullptr;
}