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

#pragma once

#include "symbol_resolver.h"
#include "trace_model.h"

#include <string>
#include <vector>

ZEN_THIRD_PARTY_INCLUDES_START
#include <EASTL/hash_map.h>
#include <EASTL/vector.h>
ZEN_THIRD_PARTY_INCLUDES_END

namespace zen::trace_detail {

struct CallstackFilterOptions
{
	bool					 EnableHeuristic = true;
	std::vector<std::string> SkipPatterns;
};

struct FilteredCallstackFrame
{
	size_t				 OriginalIndex = 0;
	const ResolvedFrame* Frame		   = nullptr;
	std::string			 Display;
};

struct FilteredCallstackView
{
	eastl::vector<FilteredCallstackFrame> Frames;
	uint32_t							  HiddenPrefixCount			 = 0;
	bool								  IncludedThirdPartyBoundary = false;
};

class CallstackFormatter
{
public:
	CallstackFormatter(const TraceModel& InModel, const SymbolResolver* InSymbols);

	const eastl::hash_map<uint64_t, std::string>& GetResolvedCache() const { return m_Cache; }

	const CallstackEntry* FindCallstackEntry(uint32_t CallstackId) const;
	const std::string&	  Describe(const ResolvedFrame& Frame);
	FilteredCallstackView BuildView(const CallstackEntry& Entry, const CallstackFilterOptions& Options);

private:
	const TraceModel&					   m_Model;
	const SymbolResolver*				   m_Symbols = nullptr;
	eastl::hash_map<uint64_t, std::string> m_Cache;
};

}  // namespace zen::trace_detail