Sensor.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 "Sensor.h"
#include "SensorManager.h"
#include "SemanticColors.h"
 
#include "Agrarsense/JSON/SimulatorJsonExporter.h"
#include "Agrarsense/Game/AgrarsenseStatics.h"
#include "Agrarsense/Logging/SimulatorLog.h"
#include "Agrarsense/ROS/ROSHandler.h"
 
#include "Kismet/GameplayStatics.h"
#include "TimerManager.h"
#include "UObject/Class.h"
 
TArray<TWeakObjectPtr<UPrimitiveComponent>> ASensor::ComponentsToHide;
FPrimitiveAdded ASensor::OnPrimitiveAdded;
 
ASensor::ASensor(const FObjectInitializer& ObjectInitializer) : Super(ObjectInitializer)
{
    RootComponent = CreateDefaultSubobject<USceneComponent>(TEXT("SceneRoot"));
}
 
void ASensor::BeginPlay()
{
    Super::BeginPlay();
 
    UWorld* World = GetWorld();
 
    UROSHandler* ROSHandler = UAgrarsenseStatics::GetROSHandle(World);
    if (ROSHandler)
    {
        // Subscribe to ROS changed event
        ROSHandler->OnROSStateChanged.AddUniqueDynamic(this, &ASensor::ROSBridgeStateChanged);
    }
 
    ROSInstance = UAgrarsenseStatics::GetROSGameInstance(World);
    if (ROSInstance)
    {
        // Check if ROS is connected at BeginPlay
        ROSConnected = ROSInstance->IsROSConnected();
    }
 
    // Delay logging sensor information message so each sensor have time setup fully.
    FTimerHandle Handle;
    World->GetTimerManager().SetTimer(Handle, FTimerDelegate::CreateLambda([this]
    {
        if (IsValid(this))
        {
            FString Message = FString("Spawned ") + GetActorInformation_Implementation();
            SimulatorLog::Log(Message);
        }
    }), 0.150f, false);
 
    USensorManager::AddSensor(this);
}
 
void ASensor::EndPlay(const EEndPlayReason::Type EndPlayReason)
{
    // Broadcast that this sensor will be destroyed
    OnSensorDestroy.Broadcast(this);
 
    USensorManager::RemoveSensor(this);
 
    Super::EndPlay(EndPlayReason);
 
    UROSHandler* ROSHandler = UAgrarsenseStatics::GetROSHandle(GetWorld());
    if (ROSHandler)
    {
        ROSHandler->OnROSStateChanged.RemoveDynamic(this, &ASensor::ROSBridgeStateChanged);
    }
 
    FString Sensor = UEnum::GetDisplayValueAsText(GetSensorType()).ToString();
    FString ID = GetSensorIdentifier();
    FString Message = FString::Printf(TEXT("Destroyed %s sensor with ID: %s"), *Sensor, *ID);
    SimulatorLog::Log(Message);
 
    if (EndPlayReason == EEndPlayReason::EndPlayInEditor 
        || EndPlayReason == EEndPlayReason::Quit)
    {
        for (auto& WeakComponentPtr : ComponentsToHide)
        {
            WeakComponentPtr.Reset();
        }
        ComponentsToHide.Empty();
    }
 
    // Destroy all Actors that are attached to this sensor (Sensor model for example)
    TArray<AActor*> AttachedActors;
    GetAttachedActors(AttachedActors);
    for (AActor* Actor : AttachedActors)
    {
        if (Actor) 
        {
            Actor->Destroy();
        }
    }
 
    ROSInstance = nullptr;
 
    if (LogFile)
    {
        LogFile->Destroy();
        LogFile = nullptr;
    }
 
    DestroyROSTopic();
}
 
FString ASensor::ExportToJsonFile(const FString& FileName)
{
    return USimulatorJsonExporter::ExportSensorToJSON(FileName, this);
}
 
void ASensor::ROSBridgeStateChanged(EROSState ROSState)
{
    switch (ROSState)
    {
    case EROSState::Connected:
        ROSConnected = true;
        CreateROSTopic();
        break;
 
    case EROSState::Disconnected:
        ROSConnected = false;
        DestroyROSTopic();
        break;
    }
}
 
void ASensor::DestroyROSTopic()
{
    if (ROSTopic)
    {
        ROSTopic->Unadvertise();
        ROSTopic->Unsubscribe();
        ROSTopic->MarkAsDisconnected();
        ROSTopic->ConditionalBeginDestroy();
        ROSTopic = nullptr;
    }
}
 
void ASensor::CreateROSTopic()
{
    if (!SendDataToROS
        || ROSTopic
        || !ROSInstance
        || !IsROSConnected())
    {
        return;
    }
 
    FString MessageType;
    ESensorTypes SensorType = GetSensorType();
 
    switch (SensorType)
    {
    case ESensorTypes::Lidar:
    case ESensorTypes::Radar:
        MessageType = "sensor_msgs/PointCloud2";
        break;
 
    case ESensorTypes::RGBCamera:
    case ESensorTypes::ThermalCamera:
    case ESensorTypes::DepthCamera:
    case ESensorTypes::SemanticSegmentationCamera:
    case ESensorTypes::DVSCamera: // DVS Camera should override this.
        MessageType = "sensor_msgs/Image";
        break;
 
    case ESensorTypes::Transform:
        MessageType = "geometry_msgs/Transform";
        break;
 
    case ESensorTypes::Collision:
    case ESensorTypes::Overlap:
        MessageType = "std_msgs/String";
        break;
 
    default:
        FString SensorNameString = UEnumUtilities::ConvertSensorTypeToString(SensorType);
        UE_LOG(LogTemp, Warning, TEXT("Unhandled ROS sensor message creation for sensor: %s"), *SensorNameString);
        break;
    }
 
    FString TopicName = FString::Printf(TEXT("/agrarsense/out/sensors/%s"), *GetSensorIdentifier());
 
    ROSTopic = NewObject<UTopic>(UTopic::StaticClass());
    if (ROSTopic)
    {
        ROSTopic->Init(ROSInstance->ROSIntegrationCore, TopicName, MessageType);
        ROSTopic->Advertise();
    }
 
    FString Message = FString::Printf(TEXT("%s sensor created ROS topic. Name: '%s' MessageType: '%s' "), *GetSensorName(), *TopicName, *MessageType);
    SimulatorLog::Log(Message);
}
 
void ASensor::CreateDataSavePath()
{
    if (FileSavePath.IsEmpty())
    {
        FString DataLocation = UAgrarsensePaths::GetDataFolder();
        FString UnixTimeStamp = FString::FromInt(FDateTime::Now().ToUnixTimestamp());
        FileSavePath = DataLocation + GetSensorIdentifier() + "_" + UnixTimeStamp + "/";
    }
}
 
void ASensor::CreateLogFile()
{
    if (IsValid(LogFile))
    {
        // File has already been created, return
        return;
    }
 
    FLogFileSettings Settings;
    Settings.FileCreationOptions = FFileCreationOptions::Overwrite;
    Settings.FileWriteOptions = FFileWriteOptions::Queue;
    Settings.QueueLength = 10;
    Settings.KeepFileOpen = false;
 
    LogFile = NewObject<ULogFile>(ULogFile::StaticClass());
    if (LogFile)
    {
        // FileName is SensorName_ID_UnixTimeStamp.txt
        FString UnixTimeStamp = FString::FromInt(FDateTime::Now().ToUnixTimestamp());
 
        FString ActorID = GetActorID_Implementation();
 
        // Temp fix, Remove all "/" from ActorID
        ActorID.ReplaceInline(TEXT("/"), TEXT(""), ESearchCase::IgnoreCase);
 
        FString FileName = FString::Printf(TEXT("%s_%s_%s"), *GetSensorName(), *ActorID, *UnixTimeStamp);
        LogFile->Create(FileName, Settings);
    }
}
 
void ASensor::WriteToLogFile(const FString& Message)
{
    if (!IsValid(LogFile) && IsValid(this))
    {
        CreateLogFile();
    }
 
    if (IsValid(LogFile))
    {
        LogFile->Write(Message);
    }
}
 
void ASensor::HideComponentForAllCameras(UPrimitiveComponent* PrimitiveComponent)
{
    if (PrimitiveComponent)
    {
        TWeakObjectPtr<UPrimitiveComponent> Component(PrimitiveComponent);
        ComponentsToHide.Add(Component);
        OnPrimitiveAdded.Broadcast(PrimitiveComponent);
    }
}
 
TMap<FString, FColor> ASensor::GetSemanticColors()
{
    return USemanticColors::GetSemanticColors();
}