Files
UnrealEngine/Engine/Plugins/AI/MassAI/Source/MassNavigation/Private/MassSteeringTrait.cpp
2025-05-18 13:04:45 +08:00

33 lines
1.3 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#include "Steering/MassSteeringTrait.h"
#include "MassEntityTemplateRegistry.h"
#include "MassCommonFragments.h"
#include "MassMovementFragments.h"
#include "MassNavigationFragments.h"
#include "Steering/MassSteeringFragments.h"
#include "Engine/World.h"
#include "MassEntityUtils.h"
void UMassSteeringTrait::BuildTemplate(FMassEntityTemplateBuildContext& BuildContext, const UWorld& World) const
{
FMassEntityManager& EntityManager = UE::Mass::Utils::GetEntityManagerChecked(World);
BuildContext.RequireFragment<FAgentRadiusFragment>();
BuildContext.RequireFragment<FTransformFragment>();
BuildContext.RequireFragment<FMassVelocityFragment>();
BuildContext.RequireFragment<FMassForceFragment>();
BuildContext.AddFragment<FMassMoveTargetFragment>();
BuildContext.AddFragment<FMassSteeringFragment>();
BuildContext.AddFragment<FMassStandingSteeringFragment>();
BuildContext.AddFragment<FMassGhostLocationFragment>();
const FConstSharedStruct MovingSteeringFragment = EntityManager.GetOrCreateConstSharedFragment(MovingSteering);
BuildContext.AddConstSharedFragment(MovingSteeringFragment);
const FConstSharedStruct StandingSteeringFragment = EntityManager.GetOrCreateConstSharedFragment(StandingSteering);
BuildContext.AddConstSharedFragment(StandingSteeringFragment);
}