Tagger.h

Go to the documentation of this file.

// Copyright (c) 2023 FrostBit Software Lab at the Lapland University of Applied Sciences
//
// This work is licensed under the terms of the MIT license.
// For a copy, see <https://opensource.org/licenses/MIT>.
 
#pragma once
 
#include "CoreMinimal.h"
#include "GameFramework/Actor.h"
#include "Kismet/GameplayStatics.h"
#include "ObjectLabel.h"
#include "PhysicsEngine/PhysicsAsset.h"
 
#include "ObjectLabel.h"
 
#include "Tagger.generated.h"
 
class UStaticMeshComponent;
class USkeletalMeshComponent;
 
/*
* ATagger - Actor responsible for tagging static and skeletal mesh Actors.
* During BeginPlay runtime, this Actor iterates through all Actors in the level, setting stencil value for object if it's not set.
* This stencil value is used by SemanicSegmentationCamera in its post processing material.
* Additionally, each time a new Actor is spawned, this Actor will automatically check it as well.
*/
UCLASS()
class AGRARSENSE_API ATagger : public AActor
{
    GENERATED_BODY()
 
public:
 
    ATagger();
 
protected:
 
    virtual void BeginPlay() override;
 
    virtual void EndPlay(const EEndPlayReason::Type EndPlayReason) override;
 
private:
 
    void TagExistingActors();
 
    void OnActorSpawned(AActor* Actor);
 
    void TagActor(AActor& Actor, bool TagForSemanticSegmentation);
 
    void SetStencilValue(UPrimitiveComponent& Component, const ELabels& Label, const bool SetRenderCustomDepth);
 
    ELabels GetLabelFromString(const FString& String);
 
    ELabels GetLabelFromStaticComponent(UStaticMeshComponent* StaticMeshComponentPtr);
 
    ELabels GetLabelFromSkeletalMeshComponent(USkeletalMeshComponent* USkeletalMeshComponentPtr);
 
    ELabels GetLabelFromPath(const FString& Path, FString& FolderName);
 
    template <typename T>
    ELabels GetLabelFromTag(const T* Object)
    {
        if (!Object->ComponentTags.IsEmpty())
        {
            return GetLabelFromString(Object->ComponentTags[0].ToString());
        }
 
        return ELabels::None;
    }
 
    FDelegateHandle ActorSpawnedDelegateHandle;
 
    TMap<FString, ELabels> LabelMap;
};