LuckyWorld/Source/LuckyWorldV2/Private/Robot/RobotPawn.cpp

#include "Robot/RobotPawn.h"

#include "LuckySensorPawnBase.h"
#include "Kismet/GameplayStatics.h"
#include "Kismet/KismetMathLibrary.h"
#include "Kismet/KismetSystemLibrary.h"
#include "Robot/PilotComponent/RobotPilotMultiRotorDrone.h"
#include "Robot/PilotComponent/RobotPilotSO100Component.h"

ARobotPawn::ARobotPawn()
{
}

void ARobotPawn::BeginPlay()
{
	Super::BeginPlay();
	// InitRobot(); // TODO Maybe move to GameInstance to control when we initialize the robot completely
}

void ARobotPawn::InitRobot()
{
	InitPilotComponent();
	InitCamera();
}

void ARobotPawn::InitPilotComponent()
{
	// Initialize pilot component based on robot type
	switch (RobotType)
	{
	case ERobotsName::None:
		break;
		
	case ERobotsName::SO100Robot:
		RobotPilotComponent = NewObject<URobotPilotSO100Component>(this);
		break;

	case ERobotsName::DJIDrone:
		RobotPilotComponent = NewObject<URobotPilotMultiRotorDrone>(this);
		break;
	
	case ERobotsName::Luck_e:
		break; // TODO or remove from enum
	case ERobotsName::Stretch:
		break; // TODO or remove from enum
	case ERobotsName::LuckyDrone:
		break; // TODO or remove from enum
	case ERobotsName::ArmLucky:
		break; // TODO or remove from enum
	case ERobotsName::UnitreeG1:
		break; // TODO or remove from enum
	case ERobotsName::StretchRobotV1:
		break; // TODO or remove from enum
	case ERobotsName::PandaArmRobot:
		break; // TODO or remove from enum
	case ERobotsName::PuralinkRobot:
		break; // TODO or remove from enum
	case ERobotsName::UnitreeGo2:
		break; // TODO or remove from enum
	case ERobotsName::RevoluteRobot:
		break; // TODO or remove from enum
	case ERobotsName::BostonSpotRobot:
		break; // TODO or remove from enum
	}

	// Register if this Robot has a Pilot Component 
	if (RobotPilotComponent)
	{
		RobotPilotComponent->RegisterComponent();
	}
}

void ARobotPawn::InitCamera()
{
	// TODO Fix the spawning of sensors in Cpp and spawn them using a config?
	// TODO How people can move the camera themselves?

	// Find all sensors in the scene
	TArray<AActor*> Sensors;
	UGameplayStatics::GetAllActorsOfClass(this->GetWorld(), ALuckySensorPawnBase::StaticClass(), Sensors);
	
	for (const auto Sensor : Sensors)
	{
		if (const auto Camera = Cast<ALuckySensorPawnBase>(Sensor)) Cameras.Add(Camera);
	}
}