InstanceSegmentationCamera.cpp

Go to the documentation of this file.

// Copyright (c) 2025 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>.
 
#include "InstanceSegmentationCamera.h"
#include "Agrarsense/Logging/SimulatorLog.h"
 
#include "Engine/World.h"
 
static int32 InstanceSegmentationCameraCount = 0;
 
AInstanceSegmentationCamera::AInstanceSegmentationCamera(const FObjectInitializer& ObjectInitializer) : Super(ObjectInitializer)
{
    PrimaryActorTick.bCanEverTick = false;
}
 
void AInstanceSegmentationCamera::Init(FCameraBaseParameters parameters, bool SimulateSensor)
{
    // Defined in Camera.h
    CameraName = "InstanceSegmentationCamera ";
    FilePrefix = "Data/InstanceSegmentationCamera_";
 
    AddPostProcessingMaterial("/Game/Agrarsense/Materials/PostProcessingMaterials/M_InstanceSegmentation.M_InstanceSegmentation");
    Super::Init(parameters, SimulateSensor);
 
    // This camera doesn't need to render shadows
    SetShadowRendering(false);
 
    // Or many other advanced show flags.
    DisableShowFlags();
 
#ifdef InstanceSegmentationPass_EXISTS
    // If user is using AGRARSENSE Unreal Engine fork, we can utilize this sensor
    UWorld* World = GetWorld();
    if (GEngine && World)
    {
        GEngine->Exec(World, TEXT("r.EnableSegmentationRendering 1"));
    }
    ++InstanceSegmentationCameraCount;
#else
    // else log and destroy this Actor as the user cannot use this sensor anyway.
    SimulatorLog::Log("InstanceSegmentationCamera.cpp: This sensor is only supported with our AGRARSENSE fork of the engine.");
    Destroy();
#endif
}
 
void AInstanceSegmentationCamera::EndPlay(const EEndPlayReason::Type EndPlayReason)
{
    Super::EndPlay(EndPlayReason);
 
#ifdef InstanceSegmentationPass_EXISTS
    --InstanceSegmentationCameraCount;
    if (InstanceSegmentationCameraCount == 0) 
    {
        UWorld* World = GetWorld();
        if (GEngine && World)
        {
            GEngine->Exec(World, TEXT("r.EnableSegmentationRendering 0"));
        }
    }
#endif
}
 
void AInstanceSegmentationCamera::AddProcessingToFrameBuffer(TArray<FColor>& buffer)
{
    // Old way
    //FColor* BufferPtr = buffer.GetData();
 
    //for (int32 i = 0; i < buffer.Num(); ++i)
    //{
    //  // Convert the current FColor (sRGB 0-255) to FLinearColor (linear 0.0-1.0)
    //  FLinearColor pixel = FLinearColor::FromSRGBColor(BufferPtr[i]);
 
    //  // Process the pixel and convert it back to FColor (sRGB 0-255)
    //  // This step applies gamma correction and ensures values are properly adjusted
    //  BufferPtr[i] = FColor(
    //      FMath::RoundToInt(pixel.R * 255.0f),
    //      FMath::RoundToInt(pixel.G * 255.0f),
    //      FMath::RoundToInt(pixel.B * 255.0f));
    //}
 
    ApplyGammaCorrectionLUT(buffer);
}