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/SkImage.h"
15 #include "include/core/SkPaint.h"
16 #include "include/core/SkSize.h"
17 #include "include/core/SkString.h"
18 #include "include/gpu/GrDirectContext.h"
19 #include "tools/gpu/vk/VkYcbcrSamplerHelper.h"
20
release_ycbcrhelper(void * releaseContext)21 static void release_ycbcrhelper(void* releaseContext) {
22 VkYcbcrSamplerHelper* ycbcrHelper = reinterpret_cast<VkYcbcrSamplerHelper*>(releaseContext);
23 delete ycbcrHelper;
24 }
25
26 namespace skiagm {
27
28 // This GM exercises the native YCbCr image format on Vulkan
29 class YCbCrImageGM : public GM {
30 public:
YCbCrImageGM()31 YCbCrImageGM() {
32 this->setBGColor(0xFFCCCCCC);
33 }
34
35 protected:
36
onShortName()37 SkString onShortName() override {
38 return SkString("ycbcrimage");
39 }
40
onISize()41 SkISize onISize() override {
42 return SkISize::Make(2*kPad+kImageSize, 2*kPad+kImageSize);
43 }
44
createYCbCrImage(GrDirectContext * dContext,SkString * errorMsg)45 DrawResult createYCbCrImage(GrDirectContext* dContext, SkString* errorMsg) {
46 std::unique_ptr<VkYcbcrSamplerHelper> ycbcrHelper(new VkYcbcrSamplerHelper(dContext));
47
48 if (!ycbcrHelper->isYCbCrSupported()) {
49 *errorMsg = "YCbCr sampling not supported.";
50 return skiagm::DrawResult::kSkip;
51 }
52
53 if (!ycbcrHelper->createBackendTexture(kImageSize, kImageSize)) {
54 *errorMsg = "Failed to create I420 backend texture.";
55 return skiagm::DrawResult::kFail;
56 }
57
58 SkASSERT(!fYCbCrImage);
59 fYCbCrImage = SkImage::MakeFromTexture(dContext, ycbcrHelper->backendTexture(),
60 kTopLeft_GrSurfaceOrigin, kRGB_888x_SkColorType,
61 kPremul_SkAlphaType, nullptr,
62 release_ycbcrhelper, ycbcrHelper.get());
63 ycbcrHelper.release();
64 if (!fYCbCrImage) {
65 *errorMsg = "Failed to create I420 image.";
66 return DrawResult::kFail;
67 }
68
69 return DrawResult::kOk;
70 }
71
onGpuSetup(GrDirectContext * dContext,SkString * errorMsg)72 DrawResult onGpuSetup(GrDirectContext* dContext, SkString* errorMsg) override {
73 if (!dContext || dContext->abandoned()) {
74 return DrawResult::kSkip;
75 }
76
77 if (dContext->backend() != GrBackendApi::kVulkan) {
78 *errorMsg = "This GM requires a Vulkan context.";
79 return DrawResult::kSkip;
80 }
81
82 DrawResult result = this->createYCbCrImage(dContext, errorMsg);
83 if (result != DrawResult::kOk) {
84 return result;
85 }
86
87 return DrawResult::kOk;
88 }
89
onGpuTeardown()90 void onGpuTeardown() override {
91 fYCbCrImage = nullptr;
92 }
93
onDraw(SkCanvas * canvas,SkString *)94 DrawResult onDraw(SkCanvas* canvas, SkString*) override {
95 SkASSERT(fYCbCrImage);
96
97 canvas->drawImage(fYCbCrImage, kPad, kPad, SkSamplingOptions(SkFilterMode::kLinear));
98 return DrawResult::kOk;
99 }
100
101 private:
102 static const int kImageSize = 112;
103 static const int kPad = 8;
104
105 sk_sp<SkImage> fYCbCrImage;
106
107 using INHERITED = GpuGM;
108 };
109
110 //////////////////////////////////////////////////////////////////////////////
111
112 DEF_GM(return new YCbCrImageGM;)
113
114 } // namespace skiagm
115
116 #endif // SK_VULKAN
117