64 lines
1.5 KiB
C++
64 lines
1.5 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#include "TraceServices/Model/Screenshot.h"
|
|
#include "Model/ScreenshotProviderPrivate.h"
|
|
|
|
#include "AnalysisServicePrivate.h"
|
|
#include "Common/FormatArgs.h"
|
|
|
|
namespace TraceServices
|
|
{
|
|
|
|
FScreenshotProvider::FScreenshotProvider(IAnalysisSession& InSession)
|
|
: Session(InSession)
|
|
{
|
|
}
|
|
|
|
TSharedPtr<FScreenshot> FScreenshotProvider::AddScreenshot(uint32 Id)
|
|
{
|
|
TSharedPtr<FScreenshot> NewScreenshot = Screenshots.Add(Id, MakeShared<FScreenshot>());
|
|
return NewScreenshot;
|
|
}
|
|
|
|
void FScreenshotProvider::AddScreenshotChunk(uint32 Id, uint32 ChunkNum, uint16 Size, const TArrayView<const uint8>& ChunkData)
|
|
{
|
|
TSharedPtr<FScreenshot> *ScreenshotPtr = Screenshots.Find(Id);
|
|
|
|
if (ScreenshotPtr == nullptr)
|
|
{
|
|
return;
|
|
}
|
|
|
|
TSharedPtr<FScreenshot> Screenshot = *ScreenshotPtr;
|
|
|
|
constexpr uint32 ChunkSize = TNumericLimits<uint16>::Max();
|
|
check(Screenshot->Data.Num() == ChunkNum * ChunkSize);
|
|
|
|
Screenshot->Data.Append(ChunkData.GetData(), Size);
|
|
}
|
|
|
|
const TSharedPtr<const FScreenshot> FScreenshotProvider::GetScreenshot(uint32 Id) const
|
|
{
|
|
const TSharedPtr<FScreenshot>* ScreenshotPtr = Screenshots.Find(Id);
|
|
|
|
if (ScreenshotPtr == nullptr)
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
return *ScreenshotPtr;
|
|
}
|
|
|
|
FName GetScreenshotProviderName()
|
|
{
|
|
static const FName Name("ScreenshotProvider");
|
|
return Name;
|
|
}
|
|
|
|
const IScreenshotProvider& ReadScreenshotProvider(const IAnalysisSession& Session)
|
|
{
|
|
return *Session.ReadProvider<IScreenshotProvider>(GetScreenshotProviderName());
|
|
}
|
|
|
|
} // namespace TraceServices
|