PhotoCapture.h

Go to the documentation of this file.

// Copyright (c) 2024 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 "PhotoCapture.generated.h"
 
class ACamera;
 
USTRUCT(Blueprintable)
struct AGRARSENSE_API FCaptureData
{
    GENERATED_BODY()
 
    UPROPERTY(VisibleAnywhere, BlueprintReadWrite)
    TArray<ACamera*> Cameras;
 
    UPROPERTY(VisibleAnywhere, BlueprintReadWrite)
    TArray<FTransform> CapturePositions;
 
    /*
    * If true, captures additional photos at the same location with different yaw rotations (e.g., 360 view without top/bottom).
    */
    UPROPERTY(VisibleAnywhere, BlueprintReadWrite)
    bool CaptureRotatedViews = false;
 
    /*
    * Use can give dataCapture json locations in either World locations or GPS locations.
    */
    UPROPERTY(VisibleAnywhere, BlueprintReadWrite)
    bool UseGPSLocation = false;
 
    UPROPERTY(VisibleAnywhere, BlueprintReadOnly)
    int32 CurrentCaptureIndex = 0;
 
    UPROPERTY(VisibleAnywhere, BlueprintReadOnly)
    int32 FramesToKeepInSamePosition = 4;
};
 
/*
* This Actor automates the process of gathering synthetic camera data by capturing images
* at predefined positions.
* Usage:
* 1) Spawn any Camera (RGB, Depth, Semantic, Thermal or DVS)
* 2) Spawn this Actor and call SetupPhotoCapture with FCaptureData struct
*/
UCLASS()
class AGRARSENSE_API APhotoCapture : public AActor
{
    GENERATED_BODY()
 
public:
 
    APhotoCapture(const FObjectInitializer& ObjectInitializer);
 
    UFUNCTION(BlueprintCallable)
    void SetupPhotoCapture(FCaptureData NewCaptureData);
 
protected:
 
    virtual void BeginPlay() override;
 
    virtual void Tick(float DeltaTime) override;
 
    virtual void EndPlay(const EEndPlayReason::Type EndPlayReason) override;
 
private:
 
    /*
    * Capture one Image each frame and move to next position next frame.
    */
    void Capture();
 
    void UpdateCameraPositions(const FTransform& Transform);
 
    void CaptureImages();
 
    void DestroyCameras();
 
    FCaptureData CaptureData;
 
    int32 FrameCounter = 0;
};