55 lines
1.9 KiB
C++
55 lines
1.9 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#include "CsvProfilerModule.h"
|
|
|
|
#include "Analyzers/CsvProfilerTraceAnalysis.h"
|
|
#include "TraceServices/Model/Counters.h"
|
|
#include "TraceServices/Model/Frames.h"
|
|
#include "TraceServices/Model/Threads.h"
|
|
|
|
namespace TraceServices
|
|
{
|
|
|
|
void FCsvProfilerModule::GetModuleInfo(FModuleInfo& OutModuleInfo)
|
|
{
|
|
static const FName CsvProfilerModuleName("TraceModule_CsvProfiler");
|
|
|
|
OutModuleInfo.Name = CsvProfilerModuleName;
|
|
OutModuleInfo.DisplayName = TEXT("CsvProfiler");
|
|
}
|
|
|
|
void FCsvProfilerModule::OnAnalysisBegin(IAnalysisSession& Session)
|
|
{
|
|
const IFrameProvider& FrameProvider = ReadFrameProvider(Session);
|
|
const IThreadProvider& ThreadProvider = ReadThreadProvider(Session);
|
|
IEditableCounterProvider& EditableCounterProvider = EditCounterProvider(Session);
|
|
TSharedPtr<FCsvProfilerProvider> CsvProfilerProvider = MakeShared<FCsvProfilerProvider>(Session);
|
|
Session.AddProvider(GetCsvProfilerProviderName(), CsvProfilerProvider);
|
|
Session.AddAnalyzer(new FCsvProfilerAnalyzer(Session, *CsvProfilerProvider, EditableCounterProvider, FrameProvider, ThreadProvider));
|
|
}
|
|
|
|
void FCsvProfilerModule::GenerateReports(const IAnalysisSession& Session, const TCHAR* CmdLine, const TCHAR* OutputDirectory)
|
|
{
|
|
const ICsvProfilerProvider* CsvProfilerProvider = ReadCsvProfilerProvider(Session);
|
|
if (CsvProfilerProvider)
|
|
{
|
|
CsvProfilerProvider->EnumerateCaptures([CsvProfilerProvider, &OutputDirectory](const FCaptureInfo& CaptureInfo)
|
|
{
|
|
Table2Csv(CsvProfilerProvider->GetTable(CaptureInfo.Id), *(FString(OutputDirectory) / TEXT("CsvProfiler") / CaptureInfo.Filename));
|
|
});
|
|
}
|
|
}
|
|
|
|
FName GetCsvProfilerProviderName()
|
|
{
|
|
static const FName Name("CsvProfilerProvider");
|
|
return Name;
|
|
}
|
|
|
|
const ICsvProfilerProvider* ReadCsvProfilerProvider(const IAnalysisSession& Session)
|
|
{
|
|
return Session.ReadProvider<ICsvProfilerProvider>(GetCsvProfilerProviderName());
|
|
}
|
|
|
|
} // namespace TraceServices
|