// Copyright Epic Games, Inc. All Rights Reserved.

#include "MassDebugVisualizationTrait.h"
#include "MassDebuggerSubsystem.h"
#include "MassDebugVisualizationComponent.h"
#include "MassEntityTemplateRegistry.h"
#include "MassCommonFragments.h"
#include "Engine/World.h"

void UMassDebugVisualizationTrait::BuildTemplate(FMassEntityTemplateBuildContext& BuildContext, const UWorld& World) const
{
#if WITH_EDITORONLY_DATA
	const UStaticMesh* const DebugMesh = DebugShape.Mesh;
#else
	const UStaticMesh* const DebugMesh = nullptr;
#endif
	
	if (DebugMesh || BuildContext.IsInspectingData())
	{
#if WITH_EDITORONLY_DATA
		FSimDebugVisFragment& DebugVisFragment = BuildContext.AddFragment_GetRef<FSimDebugVisFragment>();
		if (!BuildContext.IsInspectingData())
		{
			UMassDebuggerSubsystem* Debugger = World.GetSubsystem<UMassDebuggerSubsystem>();
			if (ensure(Debugger))
			{
				UMassDebugVisualizationComponent* DebugVisComponent = Debugger->GetVisualizationComponent();
				if (ensure(DebugVisComponent))
				{
					DebugVisFragment.VisualType = DebugVisComponent->AddDebugVisType(DebugShape);
				}
				// @todo this path requires a fragment destructor that will remove the mesh from the debugger.
			}
		}
#endif // WITH_EDITORONLY_DATA
	}
	// add fragments needed whenever we have debugging capabilities
#if !(UE_BUILD_SHIPPING || UE_BUILD_TEST)
	BuildContext.AddTag<FMassDebuggableTag>();
#if WITH_EDITORONLY_DATA
	BuildContext.AddFragment_GetRef<FDataFragment_DebugVis>().Shape = DebugShape.WireShape;
#else
	// DebugShape unavailable, will used default instead
	BuildContext.AddFragment<FDataFragment_DebugVis>();
#endif // WITH_EDITORONLY_DATA
	BuildContext.AddFragment<FAgentRadiusFragment>();

	BuildContext.AddFragment<FTransformFragment>();
#endif // if !(UE_BUILD_SHIPPING || UE_BUILD_TEST)

}