ThermalCamera.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 "ThermalCamera.h"
 
#include "Components/SceneCaptureComponent2D.h"
#include "Materials/Material.h"
 
AThermalCamera::AThermalCamera(const FObjectInitializer& ObjectInitializer) : Super(ObjectInitializer)
{
    PrimaryActorTick.bCanEverTick = false;
}
 
void AThermalCamera::ThermalInit(FThermalCameraParameters Parameters, bool SimulateSensor)
{
    ThermalCameraParameters = Parameters;
    Init(ThermalCameraParameters.CameraParameters, SimulateSensor);
}
 
void AThermalCamera::Init(FCameraBaseParameters parameters, bool SimulateSensor)
{
    // Defined in Camera.h
    CameraName = "ThermalCamera ";
    FilePrefix = "Data/ThermalCamera_";
 
    SetupMaterial();
 
    // Call Camera base class Init
    Super::Init(parameters, SimulateSensor);
 
    SetShadowRendering(false);
}
 
void AThermalCamera::EndPlay(const EEndPlayReason::Type EndPlayReason)
{
    Super::EndPlay(EndPlayReason);
 
    ThermalMaterial.Reset();
}
 
void AThermalCamera::ChangeThermalCameraParameters(FThermalCameraParameters Parameters)
{
    // Set new Thermal camera parameters
    ThermalCameraParameters = Parameters;
 
    // Update material
    UpdateThermalMaterial();
 
    // Notify Camera base class to change FCameraBaseParameters
    ChangeCameraParameters(Parameters.CameraParameters);
}
 
void AThermalCamera::SetupMaterial()
{
    // Get SceneCaptureComponent2D from the base Camera class
    USceneCaptureComponent2D* SceneCaptureComponent = GetCaptureComponent2D();
 
    if (SceneCaptureComponent)
    {
        FPostProcessSettings& PostProcessSettings = SceneCaptureComponent->PostProcessSettings;
        if (!ThermalMaterial.IsValid())
        {
            FString Path = "/Game/Agrarsense/Materials/PostProcessingMaterials/M_ThermalCameraMaterial.M_ThermalCameraMaterial";
            UMaterial* BaseMaterial = Cast<UMaterial>(StaticLoadObject(UMaterial::StaticClass(), nullptr, *Path));
            if (BaseMaterial)
            {
                UMaterialInstanceDynamic* DynamicMaterialInstance = UMaterialInstanceDynamic::Create(BaseMaterial, this);
                ThermalMaterial = DynamicMaterialInstance;
                PostProcessSettings.AddBlendable(ThermalMaterial.Get(), 1.0f);
                UpdateThermalMaterial();
            }
        }
    }
}
 
void AThermalCamera::UpdateThermalMaterial()
{
    UMaterialInstanceDynamic* Material = ThermalMaterial.Get();
    if (Material)
    {
        float Width, Height;
        if (ThermalCameraParameters.AllowCustomNoiseResolution)
        {
            Width = ThermalCameraParameters.WidthResolutionNoise;
            Height = ThermalCameraParameters.HeightResolutionNoise;
        }
        else
        {
            Width = ThermalCameraParameters.CameraParameters.Width;
            Height = ThermalCameraParameters.CameraParameters.Height;
        }
 
        // Set material instance dynamic parameters.
        Material->SetScalarParameterValue(WidthNoise, Width);
        Material->SetScalarParameterValue(HeightNoise, Height);
 
        Material->SetVectorParameterValue(WarmColor, ThermalCameraParameters.WarmColor);
        Material->SetVectorParameterValue(WarmColor2, ThermalCameraParameters.WarmColor2);
 
        Material->SetVectorParameterValue(ColdColor, ThermalCameraParameters.ColdColor);
        Material->SetVectorParameterValue(ColdColor2, ThermalCameraParameters.ColdColor2);
    }
}