MassSimple/Source/Xms/EntityRegistry/XmsEntityRegistryProcessors.cpp at main · XistGG/MassSimple · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
// Copyright (c) 2025 Xist.GG

#include "EntityRegistry/XmsEntityRegistryProcessors.h"

#include "MassExecutionContext.h"
#include "XmsEntityMetaData.h"
#include "XmsEntityRegistry.h"
#include "XmsLog.h"

//----------------------------------------------------------------------//
//  UXmsEntityCreated
//----------------------------------------------------------------------//

UXmsEntityCreated::UXmsEntityCreated()
	: Query(*this)
{
	ObservedType = FXmsT_Registry::StaticStruct();
	ObservedOperations = EMassObservedOperationFlags::CreateEntity;
	ExecutionFlags = static_cast<int32>(EProcessorExecutionFlags::AllNetModes);
}

void UXmsEntityCreated::ConfigureQueries(const TSharedRef<FMassEntityManager>& EntityManager)
{
	Query.AddTagRequirement<FXmsT_Registry>(EMassFragmentPresence::All);
	Query.AddConstSharedRequirement<FXmsCSF_MetaData>(EMassFragmentPresence::All);

	ProcessorRequirements.AddSubsystemRequirement<UXmsRegistrySubsystem>(EMassFragmentAccess::ReadWrite);
}

void UXmsEntityCreated::Execute(FMassEntityManager& EntityManager, FMassExecutionContext& Context)
{
	QUICK_SCOPE_CYCLE_COUNTER(UXmsEntityCreated);

	TArray<UXmsRegistrySubsystem::FEntityContext> Entities;
	Query.ForEachEntityChunk(Context,[&Entities](FMassExecutionContext& Context)
	{
		// MetaData is shared by all Entities in this chunk
		const auto& MetaData = Context.GetConstSharedFragment<FXmsCSF_MetaData>();

		// Allocate memory to store data for MassOnEntitiesCreated event
		Entities.Reserve(Entities.Num() + Context.GetNumEntities());

		// Iterate over each Entity in this chunk, copy relevant info
		for (FMassExecutionContext::FEntityIterator EntityIt = Context.CreateEntityIterator(); EntityIt; ++EntityIt)
		{
			UXmsRegistrySubsystem::FEntityContext EntityContext {
				.Entity = Context.GetEntity(*EntityIt),
				.MetaData = MetaData,  // COPY the MetaData
			};
			Entities.Emplace(EntityContext);
		}
	});

	if (Entities.Num() > 0)
	{
#if WITH_XMS_DEBUG
		UE_VLOG_UELOG(this, LogXmsRegistry, Verbose, TEXT("%hs: %i Entities Created"), __FUNCTION__, Entities.Num());
#endif

		UXmsRegistrySubsystem& RegistrySubsystem = Context.GetMutableSubsystemChecked<UXmsRegistrySubsystem>();
		RegistrySubsystem.MassOnEntitiesCreated(Entities);
	}
}

//----------------------------------------------------------------------//
//  UXmsEntityDestroyed
//----------------------------------------------------------------------//

UXmsEntityDestroyed::UXmsEntityDestroyed()
	: Query(*this)
{
	ObservedType = FXmsT_Registry::StaticStruct();
	ObservedOperations = EMassObservedOperationFlags::DestroyEntity;
	ExecutionFlags = static_cast<int32>(EProcessorExecutionFlags::AllNetModes);
}

void UXmsEntityDestroyed::ConfigureQueries(const TSharedRef<FMassEntityManager>& EntityManager)
{
	Query.AddTagRequirement<FXmsT_Registry>(EMassFragmentPresence::All);
	Query.AddConstSharedRequirement<FXmsCSF_MetaData>(EMassFragmentPresence::All);

	ProcessorRequirements.AddSubsystemRequirement<UXmsRegistrySubsystem>(EMassFragmentAccess::ReadWrite);
}

void UXmsEntityDestroyed::Execute(FMassEntityManager& EntityManager, FMassExecutionContext& Context)
{
	QUICK_SCOPE_CYCLE_COUNTER(UXmsEntityDestroyed);

	TArray<UXmsRegistrySubsystem::FEntityContext> Entities;
	Query.ForEachEntityChunk(Context,[&Entities](FMassExecutionContext& Context)
	{
		// MetaData is shared by all Entities in this chunk
		const auto& MetaData = Context.GetConstSharedFragment<FXmsCSF_MetaData>();

		// Allocate memory to store data for MassOnEntitiesDestroyed event
		Entities.Reserve(Entities.Num() + Context.GetNumEntities());

		// Iterate over each Entity in this chunk, copy relevant info
		for (FMassExecutionContext::FEntityIterator EntityIt = Context.CreateEntityIterator(); EntityIt; ++EntityIt)
		{
			UXmsRegistrySubsystem::FEntityContext EntityContext {
				.Entity = Context.GetEntity(*EntityIt),
				.MetaData = MetaData,  // COPY the MetaData
			};
			Entities.Emplace(EntityContext);
		}
	});

	if (Entities.Num() > 0)
	{
#if WITH_XMS_DEBUG
		UE_VLOG_UELOG(this, LogXmsRegistry, Verbose, TEXT("%hs: %i Entities Destroyed"), __FUNCTION__, Entities.Num());
#endif

		UXmsRegistrySubsystem& RegistrySubsystem = Context.GetMutableSubsystemChecked<UXmsRegistrySubsystem>();
		RegistrySubsystem.MassOnEntitiesDestroyed(Entities);
	}
}