DepthCamera.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 "Camera.h"
#include "DepthCameraParameters.h"
 
#include "DepthCamera.generated.h"
 
UCLASS()
class AGRARSENSE_API ADepthCamera : public ACamera
{
    GENERATED_BODY()
 
public:
 
    ADepthCamera(const FObjectInitializer& ObjectInitializer);
 
    UFUNCTION(BlueprintCallable)
    void ChangeDepthCameraParameters(FDepthCameraParameters Parameters);
 
    virtual void AddProcessingToFrameBuffer(TArray<FColor>& buffer) override;
 
    UFUNCTION(BlueprintPure)
    FDepthCameraParameters GetDepthCameraParameters()
    {
        return DepthCameraParameters;
    }
 
    virtual ESensorTypes GetSensorType() const override
    {
        return ESensorTypes::DepthCamera;
    }
 
    virtual FString GetParametersAsString() const override
    {
        return StructToString(DepthCameraParameters);
    }
 
private:
 
    friend class USensorFactory;
 
    void DepthInit(FDepthCameraParameters Parameters, bool SimulateSensor = true);
 
    void Init(FCameraBaseParameters parameters, bool SimulateSensor = true) override;
 
    virtual void EndPlay(const EEndPlayReason::Type EndPlayReason) override;
 
    void SetupDepthMaterial(const bool UseGrayscale);
 
    FDepthCameraParameters DepthCameraParameters;
 
    TWeakObjectPtr<UMaterial> DepthEffectMaterialGrayscale;
 
    TWeakObjectPtr<UMaterial> DepthEffectMaterial;
 
};