DepthCamera.cpp

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>.
 
#include "DepthCamera.h"
#include "Materials/Material.h"
 
ADepthCamera::ADepthCamera(const FObjectInitializer& ObjectInitializer) : Super(ObjectInitializer)
{
    // Ticked by Camera base class
    PrimaryActorTick.bCanEverTick = false;
}
 
void ADepthCamera::DepthInit(FDepthCameraParameters Parameters, bool SimulateSensor)
{
    DepthCameraParameters = Parameters;
    Init(DepthCameraParameters.CameraParameters, SimulateSensor);
}
 
void ADepthCamera::Init(FCameraBaseParameters parameters, bool SimulateSensor)
{
    // Set directory and camera name, defined in Camera.h
    CameraName = "DepthCamera ";
    FilePrefix = "Data/DepthCamera_";
 
    // Setup Depth material
    SetupDepthMaterial(DepthCameraParameters.ConvertToGrayscale);
 
    // Call Camera base class Init
    Super::Init(parameters, SimulateSensor);
 
    // This camera doesn't need to render shadows
    SetShadowRendering(false);
}
 
void ADepthCamera::EndPlay(const EEndPlayReason::Type EndPlayReason)
{
    Super::EndPlay(EndPlayReason);
 
    DepthEffectMaterialGrayscale.Reset();
    DepthEffectMaterial.Reset();
}
 
void ADepthCamera::ChangeDepthCameraParameters(FDepthCameraParameters Parameters)
{
    DepthCameraParameters = Parameters;
    SetupDepthMaterial(Parameters.ConvertToGrayscale);
    ChangeCameraParameters(Parameters.CameraParameters);
}
 
void ADepthCamera::SetupDepthMaterial(const bool UseGrayscale)
{
    // Get SceneCaptureComponent2D from the base Camera class
    USceneCaptureComponent2D* SceneCaptureComponent = GetCaptureComponent2D();
 
    if (SceneCaptureComponent)
    {
        // Determine which post-processing material to apply based on whether grayscale is used
        if (UseGrayscale)
        {
            // Load and apply the grayscale material if it hasn't been loaded already
            if (!DepthEffectMaterialGrayscale.IsValid())
            {
                FString Path = "/Game/Agrarsense/Materials/PostProcessingMaterials/DepthEffectMaterialGrayscale.DepthEffectMaterialGrayscale";
                DepthEffectMaterialGrayscale = Cast<UMaterial>(StaticLoadObject(UMaterial::StaticClass(), nullptr, *Path));
                AddPostProcessingMaterial(Path, 1.0f);
            }
 
            // Remove the non-grayscale material if it's currently applied
            if (DepthEffectMaterial.IsValid())
            {
                RemovePostProcessingMaterial(DepthEffectMaterial.Get());
                DepthEffectMaterial.Reset();
            }
        }
        else
        {
            // Load and apply the non-grayscale material if it hasn't been loaded already
            if (!DepthEffectMaterial.IsValid())
            {
                FString Path = "/Game/Agrarsense/Materials/PostProcessingMaterials/DepthEffectMaterial.DepthEffectMaterial";
                DepthEffectMaterial = Cast<UMaterial>(StaticLoadObject(UMaterial::StaticClass(), nullptr, *Path));
                AddPostProcessingMaterial(Path, 1.0f);
            }
 
            // Remove the grayscale material if it's currently applied
            if (DepthEffectMaterialGrayscale.IsValid())
            {
                RemovePostProcessingMaterial(DepthEffectMaterialGrayscale.Get());
                DepthEffectMaterialGrayscale.Reset();
            }
        }
    }
}