1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "RendererScreenshotGrabber.h"
#include "RendererOGL.h"
#include "mozilla/gfx/2D.h"
using mozilla::layers::ProfilerScreenshots;
namespace mozilla {
namespace wr {
RendererScreenshotGrabber::RendererScreenshotGrabber() {
mMaxScreenshotSize = ProfilerScreenshots::ScreenshotSize();
}
void RendererScreenshotGrabber::MaybeGrabScreenshot(
RendererOGL* aRendererOGL, const gfx::IntSize& aWindowSize) {
bool isEnabled =
ProfilerScreenshots::IsEnabled() && aRendererOGL->EnsureAsyncScreenshot();
if (isEnabled) {
if (!mProfilerScreenshots) {
mProfilerScreenshots = new ProfilerScreenshots();
}
GrabScreenshot(aRendererOGL->GetRenderer(), aWindowSize);
} else if (mProfilerScreenshots) {
Destroy(aRendererOGL->GetRenderer());
}
}
void RendererScreenshotGrabber::MaybeProcessQueue(RendererOGL* aRendererOGL) {
bool isEnabled =
ProfilerScreenshots::IsEnabled() && aRendererOGL->EnsureAsyncScreenshot();
if (isEnabled) {
if (!mProfilerScreenshots) {
mProfilerScreenshots = new ProfilerScreenshots();
}
ProcessQueue(aRendererOGL->GetRenderer());
} else if (mProfilerScreenshots) {
Destroy(aRendererOGL->GetRenderer());
}
}
void RendererScreenshotGrabber::Destroy(Renderer* aRenderer) {
mQueue.Clear();
mCurrentFrameQueueItem = Nothing();
mProfilerScreenshots = nullptr;
wr_renderer_release_profiler_structures(aRenderer);
}
void RendererScreenshotGrabber::GrabScreenshot(
Renderer* aRenderer, const gfx::IntSize& aWindowSize) {
gfx::IntSize screenshotSize;
AsyncScreenshotHandle handle = wr_renderer_get_screenshot_async(
aRenderer, 0, 0, aWindowSize.width, aWindowSize.height,
mMaxScreenshotSize.width, mMaxScreenshotSize.height, ImageFormat::BGRA8,
&screenshotSize.width, &screenshotSize.height);
mCurrentFrameQueueItem = Some(QueueItem{
TimeStamp::Now(),
handle,
screenshotSize,
aWindowSize,
reinterpret_cast<uintptr_t>(this),
});
}
void RendererScreenshotGrabber::ProcessQueue(Renderer* aRenderer) {
for (const auto& item : mQueue) {
mProfilerScreenshots->SubmitScreenshot(
item.mWindowIdentifier, item.mWindowSize, item.mScreenshotSize,
item.mTimeStamp,
[&item, aRenderer](gfx::DataSourceSurface* aTargetSurface) {
gfx::DataSourceSurface::ScopedMap map(aTargetSurface,
gfx::DataSourceSurface::WRITE);
int32_t destStride = map.GetStride();
bool success = wr_renderer_map_and_recycle_screenshot(
aRenderer, item.mHandle, map.GetData(),
destStride * aTargetSurface->GetSize().height, destStride);
return success;
});
}
mQueue.Clear();
if (mCurrentFrameQueueItem) {
mQueue.AppendElement(*mCurrentFrameQueueItem);
mCurrentFrameQueueItem = Nothing();
}
}
} // namespace wr
} // namespace mozilla
|
