UE4 Perception AI System: Detect By Affiliation

AAIControllerTeam.cpp

#include "AIControllerTeam.h"

// A Tutorial for this code is available here:
// https://www.thinkandbuild.it/ue4-ai-perception-system/

AAIControllerTeam::AAIControllerTeam()
{
  SetGenericTeamId(FGenericTeamId(5));
}


ETeamAttitude::Type AAIControllerTeam::GetTeamAttitudeTowards(const AActor& Other) const
{
  
  if (const APawn* OtherPawn = Cast<APawn>(&Other)) {
     
    // DEFAULT BEHAVIOR---------------------------------------------------
    if (const IGenericTeamAgentInterface* TeamAgent = Cast<IGenericTeamAgentInterface>(OtherPawn->GetController()))
    {
      return Super::GetTeamAttitudeTowards(*OtherPawn->GetController());
    }
    
    //OR CUSTOM BEHAVIOUR--------------------------------------------------
    if (const IGenericTeamAgentInterface* TeamAgent = Cast<IGenericTeamAgentInterface>(OtherPawn->GetController()))
    {
      //Create an alliance with Team with ID 10 and set all the other teams as Hostiles:
      FGenericTeamId OtherTeamID = TeamAgent->GetGenericTeamId();
      if (OtherTeamID == 10) {
        return ETeamAttitude::Neutral;
      }
      else {
        return ETeamAttitude::Hostile;
      }
    }
  }

  return ETeamAttitude::Neutral;
}

AIControllerTeam.h

#include "CoreMinimal.h"
#include "AIController.h"
#include "GenericTeamAgentInterface.h"
#include "AAIControllerTeam.generated.h"

// A Tutorial for this code is available here:
// https://www.thinkandbuild.it/ue4-ai-perception-system/

UCLASS()
class THEMIRRORSEND_API AAIControllerTeam : public AAIController
{
  GENERATED_BODY()
  ASightAIController();

public:
  // Override this function 
  ETeamAttitude::Type GetTeamAttitudeTowards(const AActor& Other) const override;
};

APlayerControllerTeam.cpp

#include "APlayerControllerTeam.h" 

APlayerControllerTeam::APlayerControllerTeam()
{
  PrimaryActorTick.bCanEverTick = true;
  TeamId = FGenericTeamId(10);
}

FGenericTeamId APlayerControllerTeam::GetGenericTeamId() const
{
  return TeamId;
}

APlayerControllerTeam.h

#include "CoreMinimal.h"
#include "GameFramework/PlayerController.h"
#include "GenericTeamAgentInterface.h"
#include "APlayerControllerTeam.generated.h"

class APlayerControllerTeam;

UCLASS()
class TUTORIAL_API APlayerControllerTeam : public APlayerController, public IGenericTeamAgentInterface
{
  GENERATED_BODY()
  
public:
  APlayerControllerTeam();

private: 
  // Implement The Generic Team Interface 
  FGenericTeamId TeamId;
  FGenericTeamId GetGenericTeamId() const;
};