1 /*
2 * Copyright 2024 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 "tests/Test.h"
9
10 #include "include/core/SkBitmap.h"
11 #include "include/core/SkCanvas.h"
12 #include "include/core/SkPaint.h"
13 #include "include/effects/SkRuntimeEffect.h"
14
15 #if defined(SK_GANESH)
16 #include "include/gpu/ganesh/SkSurfaceGanesh.h"
17 #endif
18
19 #if defined(SK_GRAPHITE)
20 #include "include/gpu/graphite/Context.h"
21 #include "include/gpu/graphite/Surface.h"
22 #endif
23
24 namespace {
25
26 // Tests that draws to an F16 surface blend as expected.
test_f16(skiatest::Reporter * reporter,std::function<sk_sp<SkSurface> (const SkImageInfo &)> createSurface)27 void test_f16(skiatest::Reporter* reporter,
28 std::function<sk_sp<SkSurface>(const SkImageInfo&)> createSurface) {
29 // Some blend modes and their corresponding expected red channel output when blending premul src
30 // (2, 0, 0, 0) with dst (0, 0, 0, 0) on an F16 surface.
31 constexpr uint16_t kHalfFloat0 = 0x0000;
32 constexpr uint16_t kHalfFloat1 = 0x3c00;
33 constexpr uint16_t kHalfFloat2 = 0x4000;
34 constexpr struct Expectation {
35 SkBlendMode fBlendMode;
36 uint16_t fRed;
37 } kExpectations[19] = {
38 { SkBlendMode::kClear, kHalfFloat0 },
39 { SkBlendMode::kSrc, kHalfFloat2 },
40 { SkBlendMode::kDst, kHalfFloat0 },
41 { SkBlendMode::kSrcOver, kHalfFloat2 },
42 { SkBlendMode::kDstOver, kHalfFloat2 },
43 { SkBlendMode::kSrcIn, kHalfFloat0 },
44 { SkBlendMode::kDstIn, kHalfFloat0 },
45 { SkBlendMode::kSrcOut, kHalfFloat2 },
46 { SkBlendMode::kDstOut, kHalfFloat0 },
47 { SkBlendMode::kPlus, kHalfFloat1 },
48 { SkBlendMode::kScreen, kHalfFloat2 },
49 };
50
51 // Create an F16 surface, if possible.
52 SkImageInfo imageInfo = SkImageInfo::Make(
53 SkISize::Make(1, 1), kRGBA_F16_SkColorType, SkAlphaType::kPremul_SkAlphaType);
54 sk_sp<SkSurface> surface = createSurface(imageInfo);
55 if (!surface) {
56 return;
57 }
58
59 for (const Expectation& expectation : kExpectations) {
60 // Draw to the F16 surface.
61 SkPaint paint;
62 auto [effect, error] = SkRuntimeEffect::MakeForShader(SkString(R"(
63 float4 main(vec2 xy) {
64 return float4(2.0, 0.0, 0.0, 0.0);
65 }
66 )"));
67 paint.setShader(effect->makeShader(nullptr, {}));
68 paint.setBlendMode(expectation.fBlendMode);
69 surface->getCanvas()->clear(SkColors::kTransparent);
70 surface->getCanvas()->drawPaint(paint);
71
72 // Read pixels.
73 SkBitmap bitmap;
74 SkPixmap pixmap;
75 bitmap.allocPixels(imageInfo);
76 SkAssertResult(bitmap.peekPixels(&pixmap));
77 if (!surface->readPixels(pixmap, 0, 0)) {
78 ERRORF(reporter, "readPixels failed");
79 return;
80 }
81
82 // Check that the correct color was drawn.
83 const uint16_t* channels = static_cast<const uint16_t*>(pixmap.addr());
84 const uint16_t actual = channels[0];
85 const uint16_t expected = expectation.fRed;
86 REPORTER_ASSERT(reporter,
87 actual == expected,
88 "Wrong color with blend mode %s, expected %04x, found %04x",
89 SkBlendMode_Name(expectation.fBlendMode),
90 expected,
91 actual);
92 }
93 }
94
95 } // namespace
96
97 #if defined(SK_GANESH)
DEF_GANESH_TEST_FOR_RENDERING_CONTEXTS(F16DrawTest_Ganesh,reporter,contextInfo,CtsEnforcement::kNextRelease)98 DEF_GANESH_TEST_FOR_RENDERING_CONTEXTS(F16DrawTest_Ganesh,
99 reporter,
100 contextInfo,
101 CtsEnforcement::kNextRelease) {
102 GrRecordingContext* context = contextInfo.directContext();
103 test_f16(reporter, [context](const SkImageInfo& imageInfo) {
104 return SkSurfaces::RenderTarget(context, skgpu::Budgeted::kNo, imageInfo);
105 });
106 }
107 #endif
108
109 #if defined(SK_GRAPHITE)
DEF_GRAPHITE_TEST_FOR_RENDERING_CONTEXTS(F16DrawTest_Graphite,reporter,context,CtsEnforcement::kNextRelease)110 DEF_GRAPHITE_TEST_FOR_RENDERING_CONTEXTS(F16DrawTest_Graphite,
111 reporter,
112 context,
113 CtsEnforcement::kNextRelease) {
114 std::unique_ptr<skgpu::graphite::Recorder> recorder = context->makeRecorder();
115 skgpu::graphite::Recorder* recorderPtr = recorder.get();
116 test_f16(reporter, [recorderPtr](const SkImageInfo& imageInfo) {
117 return SkSurfaces::RenderTarget(recorderPtr, imageInfo);
118 });
119 }
120 #endif
121