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

#pragma once

#include "CoreTypes.h"

#include "Async/TaskTrace.h"

namespace TraceServices { struct FTaskInfo; }

namespace UE::Insights::TaskGraphProfiler
{

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

class FTaskEntry
{
	friend class STaskTableTreeView;

public:
	FTaskEntry(const TraceServices::FTaskInfo &TaskInfo);
	~FTaskEntry() {}

	const TCHAR* GetDebugName() const { return DebugName; }

	double GetCreatedTimestamp() const { return CreatedTimestamp; }
	uint32 GetCreatedThreadId() const { return CreatedThreadId; }

	double GetLaunchedTimestamp() const { return LaunchedTimestamp; }
	uint32 GetLaunchedThreadId() const { return LaunchedThreadId; }

	double GetScheduledTimestamp() const { return ScheduledTimestamp; }
	uint32 GetScheduledThreadId() const { return ScheduledThreadId; }

	double GetStartedTimestamp() const { return StartedTimestamp; }
	uint32 GetStartedThreadId() const { return StartedThreadId; }

	double GetFinishedTimestamp() const { return FinishedTimestamp; }

	double GetCompletedTimestamp() const { return CompletedTimestamp; }
	uint32 GetCompletedThreadId() const { return CompletedThreadId; }

	double GetDestroyedTimestamp() const { return DestroyedTimestamp; }
	uint32 GetDestroyedThreadId() const { return DestroyedThreadId; }

	uint64 GetTaskSize() const { return TaskSize; }

	uint32 GetNumParents() const { return NumParents; }
	uint32 GetNumNested() const { return NumNested; }
	uint32 GetNumPrerequisites() const { return NumPrerequisites; }
	uint32 GetNumSubsequents() const { return NumSubsequents; }

	TaskTrace::FId GetId() const { return Id; }

private:
	TaskTrace::FId Id;

	const TCHAR* DebugName; 
	bool bTracked; 
	int32 ThreadToExecuteOn;

	double CreatedTimestamp;
	uint32 CreatedThreadId;

	double LaunchedTimestamp;
	uint32 LaunchedThreadId;

	double ScheduledTimestamp;
	uint32 ScheduledThreadId;

	double StartedTimestamp;
	uint32 StartedThreadId;

	double FinishedTimestamp;

	double CompletedTimestamp;
	uint32 CompletedThreadId;

	double DestroyedTimestamp;
	uint32 DestroyedThreadId;

	uint64 TaskSize;

	uint32 NumSubsequents;
	uint32 NumPrerequisites;
	uint32 NumParents;
	uint32 NumNested;
};

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

} // namespace UE::Insights::TaskGraphProfiler