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

#pragma once

#include "CoreTypes.h"

// TraceInsights
#include "Insights/ViewModels/GraphSeries.h"
#include "Insights/ViewModels/GraphTrack.h"

namespace UE::Insights::TimingProfiler
{
	struct FFrameStatsCachedEvent;
	class STimingView;
}

////////////////////////////////////////////////////////////////////////////////////////////////////

/** The delegate to be invoked when a series visibility is changed. */
DECLARE_MULTICAST_DELEGATE_OneParam(FSeriesVisibilityChangedDelegate, bool bOnOff);

class FTimingGraphSeries : public FGraphSeries
{
	INSIGHTS_DECLARE_RTTI(FTimingGraphSeries, FGraphSeries)

public:
	enum class ESeriesType
	{
		Frame,
		Timer,
		StatsCounter,
		FrameStatsTimer
	};

	struct FSimpleTimingEvent
	{
		double StartTime;
		double Duration;
	};

public:
	explicit FTimingGraphSeries(FTimingGraphSeries::ESeriesType Type);
	virtual ~FTimingGraphSeries();

	virtual FString FormatValue(double Value) const override;

	static bool CompareEventsByStartTime(const FSimpleTimingEvent& EventA, const FSimpleTimingEvent& EventB)
	{
		return EventA.StartTime < EventB.StartTime;
	}

	virtual void SetVisibility(bool bOnOff) override;

public:
	ESeriesType Type;
	union
	{
		uint32 TimerId;
		uint32 CounterId;
	};

	ETraceFrameType FrameType;
	double CachedSessionDuration;
	TArray<FSimpleTimingEvent> CachedEvents; // used by Timer series
	TArray<UE::Insights::TimingProfiler::FFrameStatsCachedEvent> FrameStatsCachedEvents; // used by Frame Stats Timer series
	uint32 CachedTimelinesNum = 0; // the number of timelines used to gather the data

	bool bIsTime; // the unit for values is [second]
	bool bIsMemory; // the unit for value is [byte]
	bool bIsFloatingPoint; // for stats counters

	FSeriesVisibilityChangedDelegate VisibilityChangedDelegate;
};

////////////////////////////////////////////////////////////////////////////////////////////////////

class FTimingGraphTrack : public FGraphTrack
{
	INSIGHTS_DECLARE_RTTI(FTimingGraphTrack, FGraphTrack)

public:
	FTimingGraphTrack(TSharedPtr<UE::Insights::TimingProfiler::STimingView> InTimingView);
	virtual ~FTimingGraphTrack();

	virtual void Update(const ITimingTrackUpdateContext& Context) override;

	void AddDefaultFrameSeries();

	TSharedPtr<FTimingGraphSeries> GetFrameSeries(ETraceFrameType FrameType);

	TSharedPtr<FTimingGraphSeries> GetTimerSeries(uint32 TimerId);
	TSharedPtr<FTimingGraphSeries> AddTimerSeries(uint32 TimerId, FLinearColor Color);
	void RemoveTimerSeries(uint32 TimerId);

	TSharedPtr<FTimingGraphSeries> GetFrameStatsTimerSeries(uint32 TimerId, ETraceFrameType FrameType);
	TSharedPtr<FTimingGraphSeries> AddFrameStatsTimerSeries(uint32 TimerId, ETraceFrameType FrameType, FLinearColor Color);
	void RemoveFrameStatsTimerSeries(uint32 TimerId, ETraceFrameType FrameType);

	TSharedPtr<FTimingGraphSeries> GetStatsCounterSeries(uint32 CounterId);
	TSharedPtr<FTimingGraphSeries> AddStatsCounterSeries(uint32 CounterId, FLinearColor Color);
	void RemoveStatsCounterSeries(uint32 CounterId);

	bool HasAnySeriesForTimer(uint32 TimerId) const;
	uint32 GetNumSeriesForTimer(uint32 TimerId) const;

protected:
	void UpdateFrameSeries(FTimingGraphSeries& Series, const FTimingTrackViewport& Viewport);
	void UpdateTimerSeries(FTimingGraphSeries& Series, const FTimingTrackViewport& Viewport);
	void UpdateFrameStatsTimerSeries(FTimingGraphSeries& Series, const FTimingTrackViewport& Viewport);
	void UpdateStatsCounterSeries(FTimingGraphSeries& Series, const FTimingTrackViewport& Viewport);

	void GetVisibleTimelineIndexes(TSet<uint32>& TimelineIndexes);

	virtual void DrawVerticalAxisGrid(const ITimingTrackDrawContext& Context) const override;

	void LoadDefaultSettings();

private:
	virtual void ContextMenu_ToggleOption_Execute(EGraphOptions Option);

private:
	FDelegateHandle OnTrackVisibilityChangedHandle;
	FDelegateHandle OnTrackAddedHandle;
	FDelegateHandle OnTrackRemovedHandle;

	FDelegateHandle GameFrameSeriesVisibilityHandle;
	FDelegateHandle RenderingFrameSeriesVisibilityHandle;

	TWeakPtr<UE::Insights::TimingProfiler::STimingView> TimingView;
	bool bNotifyTimersOnDestruction;
};

////////////////////////////////////////////////////////////////////////////////////////////////////