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

#include "CountersTraceAnalysis.h"

#include "Common/Utils.h"
#include "HAL/LowLevelMemTracker.h"
#include "ProfilingDebugging/CountersTrace.h"
#include "TraceServices/Model/Counters.h"

namespace TraceServices
{

FCountersAnalyzer::FCountersAnalyzer(IAnalysisSession& InSession, IEditableCounterProvider& InEditableCounterProvider)
	: Session(InSession)
	, EditableCounterProvider(InEditableCounterProvider)
{
}

void FCountersAnalyzer::OnAnalysisBegin(const FOnAnalysisContext& Context)
{
	auto& Builder = Context.InterfaceBuilder;

	Builder.RouteEvent(RouteId_Spec, "Counters", "Spec");
	Builder.RouteEvent(RouteId_SetValueInt, "Counters", "SetValueInt");
	Builder.RouteEvent(RouteId_SetValueFloat, "Counters", "SetValueFloat");
}

bool FCountersAnalyzer::OnEvent(uint16 RouteId, EStyle Style, const FOnEventContext& Context)
{
	LLM_SCOPE_BYNAME(TEXT("Insights/FCountersAnalyzer"));

	FAnalysisSessionEditScope _(Session);

	const auto& EventData = Context.EventData;
	switch (RouteId)
	{
	case RouteId_Spec:
	{
		uint16 CounterId = EventData.GetValue<uint16>("Id");
		ETraceCounterType CounterType = static_cast<ETraceCounterType>(EventData.GetValue<uint8>("Type"));
		ETraceCounterDisplayHint CounterDisplayHint = static_cast<ETraceCounterDisplayHint>(EventData.GetValue<uint8>("DisplayHint"));
		IEditableCounter* EditableCounter = EditableCounterProvider.CreateEditableCounter();
		if (CounterType == TraceCounterType_Float)
		{
			EditableCounter->SetIsFloatingPoint(true);
		}
		if (CounterDisplayHint == TraceCounterDisplayHint_Memory)
		{
			EditableCounter->SetDisplayHint(CounterDisplayHint_Memory);
		}
		FString Name = FTraceAnalyzerUtils::LegacyAttachmentString<TCHAR>("Name", Context);
		if (Name.IsEmpty())
		{
			UE_LOG(LogTraceServices, Warning, TEXT("Invalid counter name for counter %u."), uint32(CounterId));
			Name = FString::Printf(TEXT("<noname counter %u>"), uint32(CounterId));
		}
		EditableCounter->SetName(Session.StoreString(*Name));
		EditableCountersMap.Add(CounterId, EditableCounter);
		break;
	}
	case RouteId_SetValueInt:
	{
		const double Timestamp = Context.EventTime.AsSeconds(EventData.GetValue<uint64>("Cycle"));
		const int64 Value = EventData.GetValue<int64>("Value");
		const uint16 CounterId = EventData.GetValue<uint16>("CounterId");
		IEditableCounter* FindEditableCounter = EditableCountersMap.FindRef(CounterId);
		if (ensure(FindEditableCounter))
		{
			Session.UpdateDurationSeconds(Timestamp);
			FindEditableCounter->SetValue(Timestamp, Value);
		}
		break;
	}
	case RouteId_SetValueFloat:
	{
		const double Timestamp = Context.EventTime.AsSeconds(EventData.GetValue<uint64>("Cycle"));
		const float Value = EventData.GetValue<float>("Value");
		const uint16 CounterId = EventData.GetValue<uint16>("CounterId");
		IEditableCounter* FindEditableCounter = EditableCountersMap.FindRef(CounterId);
		if (ensure(FindEditableCounter))
		{
			Session.UpdateDurationSeconds(Timestamp);
			FindEditableCounter->SetValue(Timestamp, Value);
		}
		break;
	}
	}

	return true;
}

} // namespace TraceServices