1 /*
2 * Copyright 2020 Google LLC
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7
8 #include "gm/gm.h"
9
10 // This test only works with the Vulkan backend.
11 #ifdef SK_VULKAN
12
13 #include "include/core/SkCanvas.h"
14 #include "include/core/SkColorSpace.h"
15 #include "include/core/SkImage.h"
16 #include "include/core/SkPaint.h"
17 #include "include/core/SkSize.h"
18 #include "include/core/SkString.h"
19 #include "include/gpu/ganesh/GrDirectContext.h"
20 #include "include/gpu/ganesh/SkImageGanesh.h"
21 #include "tools/gpu/vk/VkYcbcrSamplerHelper.h"
22
23 #if defined(SK_GRAPHITE)
24 #include "include/gpu/graphite/Image.h"
25 #include "include/gpu/graphite/Recorder.h"
26 #include "include/gpu/vk/VulkanBackendContext.h"
27 #include "src/gpu/graphite/RecorderPriv.h"
28 #include "src/gpu/graphite/vk/VulkanSharedContext.h"
29 #include "tools/graphite/vk/GraphiteVulkanTestContext.h"
30
31 using VulkanTestContext = skiatest::graphite::VulkanTestContext;
32 using SharedContext = skgpu::graphite::SharedContext;
33 using VulkanSharedContext = skgpu::graphite::VulkanSharedContext;
34 #endif
35
release_ycbcrhelper(void * releaseContext)36 static void release_ycbcrhelper(void* releaseContext) {
37 VkYcbcrSamplerHelper* ycbcrHelper = reinterpret_cast<VkYcbcrSamplerHelper*>(releaseContext);
38 delete ycbcrHelper;
39 }
40
41 namespace skiagm {
42
43 // This GM exercises the native YCbCr image format on Vulkan
44 class YCbCrImageGM : public GM {
45 public:
YCbCrImageGM()46 YCbCrImageGM() {
47 this->setBGColor(0xFFCCCCCC);
48 }
49
50 protected:
getName() const51 SkString getName() const override { return SkString("ycbcrimage"); }
52
getISize()53 SkISize getISize() override {
54 return SkISize::Make(2*kPad+kImageSize, 2*kPad+kImageSize);
55 }
56
57 #if defined(SK_GRAPHITE)
createYCbCrImage(skgpu::graphite::Recorder * recorder,SkString * errorMsg)58 DrawResult createYCbCrImage(skgpu::graphite::Recorder* recorder,
59 SkString* errorMsg) {
60 if (!recorder) {
61 *errorMsg = "Cannot generate a YCbCr image without a valid GraphiteTestContext and "
62 "recorder.";
63 return skiagm::DrawResult::kSkip;
64 }
65
66 SkASSERT_RELEASE(recorder->backend() == skgpu::BackendApi::kVulkan);
67
68 const VulkanSharedContext* vulkanSharedCtxt =
69 static_cast<const VulkanSharedContext*>(recorder->priv().sharedContext());
70 SkASSERT(vulkanSharedCtxt);
71
72 std::unique_ptr<VkYcbcrSamplerHelper> ycbcrHelper(
73 new VkYcbcrSamplerHelper(vulkanSharedCtxt));
74 if (!ycbcrHelper) {
75 *errorMsg = "Failed to create VkYcbcrSamplerHelper.";
76 return skiagm::DrawResult::kFail;
77 }
78 if (!ycbcrHelper->isYCbCrSupported()) {
79 *errorMsg = "YCbCr sampling is not supported.";
80 return skiagm::DrawResult::kSkip;
81 }
82 if (!ycbcrHelper->createBackendTexture(kImageSize, kImageSize)) {
83 *errorMsg = "Failed to create I420 backend texture.";
84 return skiagm::DrawResult::kFail;
85 }
86
87 SkASSERT(!fYCbCrImage);
88
89 fYCbCrImage = SkImages::WrapTexture(recorder,
90 ycbcrHelper->backendTexture(),
91 kRGB_888x_SkColorType,
92 kPremul_SkAlphaType,
93 /*colorSpace=*/nullptr,
94 release_ycbcrhelper,
95 ycbcrHelper.get());
96 SkASSERT(fYCbCrImage);
97 ycbcrHelper.release();
98 if (!fYCbCrImage) {
99 *errorMsg = "Failed to create I420 SkImage.";
100 return DrawResult::kFail;
101 }
102 return DrawResult::kOk;
103 }
104 #endif // SK_GRAPHITE
105
createYCbCrImage(GrDirectContext * dContext,SkString * errorMsg)106 DrawResult createYCbCrImage(GrDirectContext* dContext, SkString* errorMsg) {
107 std::unique_ptr<VkYcbcrSamplerHelper> ycbcrHelper(new VkYcbcrSamplerHelper(dContext));
108
109 if (!ycbcrHelper->isYCbCrSupported()) {
110 *errorMsg = "YCbCr sampling not supported.";
111 return skiagm::DrawResult::kSkip;
112 }
113
114 if (!ycbcrHelper->createGrBackendTexture(kImageSize, kImageSize)) {
115 *errorMsg = "Failed to create I420 backend texture.";
116 return skiagm::DrawResult::kFail;
117 }
118
119 SkASSERT(!fYCbCrImage);
120 fYCbCrImage = SkImages::BorrowTextureFrom(dContext,
121 ycbcrHelper->grBackendTexture(),
122 kTopLeft_GrSurfaceOrigin,
123 kRGB_888x_SkColorType,
124 kPremul_SkAlphaType,
125 nullptr,
126 release_ycbcrhelper,
127 ycbcrHelper.get());
128 ycbcrHelper.release();
129 if (!fYCbCrImage) {
130 *errorMsg = "Failed to create I420 image.";
131 return DrawResult::kFail;
132 }
133
134 return DrawResult::kOk;
135 }
136
onGpuSetup(SkCanvas * canvas,SkString * errorMsg,GraphiteTestContext * graphiteTestContext)137 DrawResult onGpuSetup(SkCanvas* canvas,
138 SkString* errorMsg,
139 GraphiteTestContext* graphiteTestContext) override {
140 #if defined(SK_GRAPHITE)
141 skgpu::graphite::Recorder* recorder = canvas->recorder();
142
143 if (recorder) {
144 if (recorder->backend() != skgpu::BackendApi::kVulkan) {
145 *errorMsg = "This GM requires using Vulkan.";
146 return DrawResult::kSkip;
147 }
148
149 return this->createYCbCrImage(recorder, errorMsg);
150 } else
151 #endif
152 {
153 GrDirectContext* dContext = GrAsDirectContext(canvas->recordingContext());
154
155 if (!dContext || dContext->abandoned()) {
156 return DrawResult::kSkip;
157 }
158
159 if (dContext->backend() != GrBackendApi::kVulkan) {
160 *errorMsg = "This GM requires a Vulkan context.";
161 return DrawResult::kSkip;
162 }
163
164 DrawResult result = this->createYCbCrImage(dContext, errorMsg);
165 if (result != DrawResult::kOk) {
166 return result;
167 }
168
169 return DrawResult::kOk;
170 }
171 }
172
onGpuTeardown()173 void onGpuTeardown() override {
174 fYCbCrImage = nullptr;
175 }
176
onDraw(SkCanvas * canvas,SkString *)177 DrawResult onDraw(SkCanvas* canvas, SkString*) override {
178 SkASSERT(fYCbCrImage);
179 canvas->drawImage(fYCbCrImage, kPad, kPad, SkSamplingOptions(SkFilterMode::kLinear));
180 return DrawResult::kOk;
181 }
182
183 private:
184 static const int kImageSize = 112;
185 static const int kPad = 8;
186
187 sk_sp<SkImage> fYCbCrImage;
188
189 using INHERITED = GpuGM;
190 };
191
192 //////////////////////////////////////////////////////////////////////////////
193
194 DEF_GM(return new YCbCrImageGM;)
195
196 } // namespace skiagm
197
198 #endif // SK_VULKAN
199