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/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,skiatest::graphite::GraphiteTestContext * graphiteTestContext,SkString * errorMsg)58 DrawResult createYCbCrImage(skgpu::graphite::Recorder* recorder,
59 skiatest::graphite::GraphiteTestContext* graphiteTestContext,
60 SkString* errorMsg) {
61 if (!graphiteTestContext || !recorder) {
62 *errorMsg = "Cannot generate a YCbCr image without a valid GraphiteTestContext and "
63 "recorder.";
64 return skiagm::DrawResult::kSkip;
65 }
66
67 SkASSERT_RELEASE(recorder->backend() == skgpu::BackendApi::kVulkan);
68
69 VulkanTestContext* vkTestCtxt = static_cast<VulkanTestContext*>(graphiteTestContext);
70
71 const VulkanSharedContext* vulkanSharedCtxt =
72 static_cast<const VulkanSharedContext*>(recorder->priv().sharedContext());
73 SkASSERT(vulkanSharedCtxt);
74
75 std::unique_ptr<VkYcbcrSamplerHelper> ycbcrHelper(
76 new VkYcbcrSamplerHelper(vulkanSharedCtxt,
77 vkTestCtxt->getBackendContext().fPhysicalDevice));
78 if (!ycbcrHelper) {
79 *errorMsg = "Failed to create VkYcbcrSamplerHelper.";
80 return skiagm::DrawResult::kFail;
81 }
82 if (!ycbcrHelper->isYCbCrSupported()) {
83 *errorMsg = "YCbCr sampling is not supported.";
84 return skiagm::DrawResult::kSkip;
85 }
86 if (!ycbcrHelper->createBackendTexture(kImageSize, kImageSize)) {
87 *errorMsg = "Failed to create I420 backend texture.";
88 return skiagm::DrawResult::kFail;
89 }
90
91 SkASSERT(!fYCbCrImage);
92
93 // TODO(b/311392779): Once graphite supports YCbCr sampling, actually create the image and
94 // return either DrawResult::Ok or DrawResult::kFail. For now, clean up the helper and
95 // texture manually.
96 recorder->deleteBackendTexture(ycbcrHelper->backendTexture());
97 ycbcrHelper.release();
98 return skiagm::DrawResult::kSkip;
99 // fYCbCrImage = SkImages::WrapTexture(recorder,
100 // ycbcrHelper->backendTexture(),
101 // kRGB_888x_SkColorType,
102 // kPremul_SkAlphaType,
103 // /*colorSpace=*/nullptr,
104 // release_ycbcrhelper,
105 // ycbcrHelper.get());
106 // SkASSERT(fYCbCrImage);
107 // ycbcrHelper.release();
108 // if (!fYCbCrImage) {
109 // *errorMsg = "Failed to create I420 SkImage.";
110 // return DrawResult::kFail;
111 // }
112 // return DrawResult::kOk;
113 }
114 #endif // SK_GRAPHITE
115
createYCbCrImage(GrDirectContext * dContext,SkString * errorMsg)116 DrawResult createYCbCrImage(GrDirectContext* dContext, SkString* errorMsg) {
117 std::unique_ptr<VkYcbcrSamplerHelper> ycbcrHelper(new VkYcbcrSamplerHelper(dContext));
118
119 if (!ycbcrHelper->isYCbCrSupported()) {
120 *errorMsg = "YCbCr sampling not supported.";
121 return skiagm::DrawResult::kSkip;
122 }
123
124 if (!ycbcrHelper->createGrBackendTexture(kImageSize, kImageSize)) {
125 *errorMsg = "Failed to create I420 backend texture.";
126 return skiagm::DrawResult::kFail;
127 }
128
129 SkASSERT(!fYCbCrImage);
130 fYCbCrImage = SkImages::BorrowTextureFrom(dContext,
131 ycbcrHelper->grBackendTexture(),
132 kTopLeft_GrSurfaceOrigin,
133 kRGB_888x_SkColorType,
134 kPremul_SkAlphaType,
135 nullptr,
136 release_ycbcrhelper,
137 ycbcrHelper.get());
138 ycbcrHelper.release();
139 if (!fYCbCrImage) {
140 *errorMsg = "Failed to create I420 image.";
141 return DrawResult::kFail;
142 }
143
144 return DrawResult::kOk;
145 }
146
onGpuSetup(SkCanvas * canvas,SkString * errorMsg,GraphiteTestContext * graphiteTestContext)147 DrawResult onGpuSetup(SkCanvas* canvas,
148 SkString* errorMsg,
149 GraphiteTestContext* graphiteTestContext) override {
150 #if defined(SK_GRAPHITE)
151 skgpu::graphite::Recorder* recorder = canvas->recorder();
152
153 if (recorder) {
154 if (recorder->backend() != skgpu::BackendApi::kVulkan) {
155 *errorMsg = "This GM requires using Vulkan.";
156 return DrawResult::kSkip;
157 }
158
159 return this->createYCbCrImage(recorder, graphiteTestContext, errorMsg);
160 } else
161 #endif
162 {
163 GrDirectContext* dContext = GrAsDirectContext(canvas->recordingContext());
164
165 if (!dContext || dContext->abandoned()) {
166 return DrawResult::kSkip;
167 }
168
169 if (dContext->backend() != GrBackendApi::kVulkan) {
170 *errorMsg = "This GM requires a Vulkan context.";
171 return DrawResult::kSkip;
172 }
173
174 DrawResult result = this->createYCbCrImage(dContext, errorMsg);
175 if (result != DrawResult::kOk) {
176 return result;
177 }
178
179 return DrawResult::kOk;
180 }
181 }
182
onGpuTeardown()183 void onGpuTeardown() override {
184 fYCbCrImage = nullptr;
185 }
186
onDraw(SkCanvas * canvas,SkString *)187 DrawResult onDraw(SkCanvas* canvas, SkString*) override {
188 SkASSERT(fYCbCrImage);
189 canvas->drawImage(fYCbCrImage, kPad, kPad, SkSamplingOptions(SkFilterMode::kLinear));
190 return DrawResult::kOk;
191 }
192
193 private:
194 static const int kImageSize = 112;
195 static const int kPad = 8;
196
197 sk_sp<SkImage> fYCbCrImage;
198
199 using INHERITED = GpuGM;
200 };
201
202 //////////////////////////////////////////////////////////////////////////////
203
204 DEF_GM(return new YCbCrImageGM;)
205
206 } // namespace skiagm
207
208 #endif // SK_VULKAN
209