xref: /aosp_15_r20/external/skia/src/gpu/ganesh/ops/FillRRectOp.cpp (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1*c8dee2aaSAndroid Build Coastguard Worker /*
2*c8dee2aaSAndroid Build Coastguard Worker  * Copyright 2018 Google Inc.
3*c8dee2aaSAndroid Build Coastguard Worker  *
4*c8dee2aaSAndroid Build Coastguard Worker  * Use of this source code is governed by a BSD-style license that can be
5*c8dee2aaSAndroid Build Coastguard Worker  * found in the LICENSE file.
6*c8dee2aaSAndroid Build Coastguard Worker  */
7*c8dee2aaSAndroid Build Coastguard Worker 
8*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/ops/FillRRectOp.h"
9*c8dee2aaSAndroid Build Coastguard Worker 
10*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkClipOp.h"
11*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkMatrix.h"
12*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkRRect.h"
13*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkRect.h"
14*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkRefCnt.h"
15*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkScalar.h"
16*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkString.h"
17*c8dee2aaSAndroid Build Coastguard Worker #include "include/gpu/ganesh/GrRecordingContext.h"
18*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/SkColorData.h"
19*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/base/SkAlignedStorage.h"
20*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/base/SkAssert.h"
21*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/base/SkDebug.h"
22*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/base/SkMacros.h"
23*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/base/SkOnce.h"
24*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/base/SkPoint_impl.h"
25*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/base/SkTArray.h"
26*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/gpu/ganesh/GrTypesPriv.h"
27*c8dee2aaSAndroid Build Coastguard Worker #include "src/base/SkArenaAlloc.h"
28*c8dee2aaSAndroid Build Coastguard Worker #include "src/base/SkUtils.h"
29*c8dee2aaSAndroid Build Coastguard Worker #include "src/base/SkVx.h"
30*c8dee2aaSAndroid Build Coastguard Worker #include "src/core/SkRRectPriv.h"
31*c8dee2aaSAndroid Build Coastguard Worker #include "src/core/SkSLTypeShared.h"
32*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/BufferWriter.h"
33*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/KeyBuilder.h"
34*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ResourceKey.h"
35*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrAppliedClip.h"
36*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrBuffer.h"
37*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrCaps.h"
38*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrGeometryProcessor.h"
39*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrMeshDrawTarget.h"
40*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrOpFlushState.h"
41*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrPaint.h"
42*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrProcessorAnalysis.h"
43*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrProcessorSet.h"
44*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrProgramInfo.h"
45*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrRecordingContextPriv.h"
46*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrResourceProvider.h"
47*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrShaderCaps.h"
48*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrShaderVar.h"
49*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/geometry/GrShape.h"
50*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/glsl/GrGLSLFragmentShaderBuilder.h"
51*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/glsl/GrGLSLVarying.h"
52*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/glsl/GrGLSLVertexGeoBuilder.h"
53*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/ops/GrDrawOp.h"
54*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/ops/GrMeshDrawOp.h"
55*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/ops/GrSimpleMeshDrawOpHelper.h"
56*c8dee2aaSAndroid Build Coastguard Worker 
57*c8dee2aaSAndroid Build Coastguard Worker #if defined(GPU_TEST_UTILS)
58*c8dee2aaSAndroid Build Coastguard Worker #include "src/base/SkRandom.h"
59*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrDrawOpTest.h"
60*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrTestUtils.h"
61*c8dee2aaSAndroid Build Coastguard Worker #endif
62*c8dee2aaSAndroid Build Coastguard Worker 
63*c8dee2aaSAndroid Build Coastguard Worker #include <algorithm>
64*c8dee2aaSAndroid Build Coastguard Worker #include <array>
65*c8dee2aaSAndroid Build Coastguard Worker #include <cstddef>
66*c8dee2aaSAndroid Build Coastguard Worker #include <cstdint>
67*c8dee2aaSAndroid Build Coastguard Worker #include <memory>
68*c8dee2aaSAndroid Build Coastguard Worker #include <utility>
69*c8dee2aaSAndroid Build Coastguard Worker 
70*c8dee2aaSAndroid Build Coastguard Worker class GrDstProxyView;
71*c8dee2aaSAndroid Build Coastguard Worker class GrGLSLProgramDataManager;
72*c8dee2aaSAndroid Build Coastguard Worker class GrSurfaceProxyView;
73*c8dee2aaSAndroid Build Coastguard Worker enum class GrXferBarrierFlags;
74*c8dee2aaSAndroid Build Coastguard Worker 
75*c8dee2aaSAndroid Build Coastguard Worker namespace skgpu::ganesh {
76*c8dee2aaSAndroid Build Coastguard Worker class SurfaceDrawContext;
77*c8dee2aaSAndroid Build Coastguard Worker }
78*c8dee2aaSAndroid Build Coastguard Worker 
79*c8dee2aaSAndroid Build Coastguard Worker using namespace skia_private;
80*c8dee2aaSAndroid Build Coastguard Worker 
81*c8dee2aaSAndroid Build Coastguard Worker namespace skgpu::ganesh::FillRRectOp {
82*c8dee2aaSAndroid Build Coastguard Worker 
83*c8dee2aaSAndroid Build Coastguard Worker namespace {
84*c8dee2aaSAndroid Build Coastguard Worker 
85*c8dee2aaSAndroid Build Coastguard Worker // Note: Just checking m.restStaysRect is not sufficient
skews_are_relevant(const SkMatrix & m)86*c8dee2aaSAndroid Build Coastguard Worker bool skews_are_relevant(const SkMatrix& m) {
87*c8dee2aaSAndroid Build Coastguard Worker     SkASSERT(!m.hasPerspective());
88*c8dee2aaSAndroid Build Coastguard Worker 
89*c8dee2aaSAndroid Build Coastguard Worker     if (m[SkMatrix::kMSkewX] == 0.0f && m[SkMatrix::kMSkewY] == 0.0f) {
90*c8dee2aaSAndroid Build Coastguard Worker         return false;
91*c8dee2aaSAndroid Build Coastguard Worker     }
92*c8dee2aaSAndroid Build Coastguard Worker 
93*c8dee2aaSAndroid Build Coastguard Worker     static constexpr float kTol = SK_ScalarNearlyZero;
94*c8dee2aaSAndroid Build Coastguard Worker     float absScaleX = SkScalarAbs(m[SkMatrix::kMScaleX]);
95*c8dee2aaSAndroid Build Coastguard Worker     float absSkewX  = SkScalarAbs(m[SkMatrix::kMSkewX]);
96*c8dee2aaSAndroid Build Coastguard Worker     float absScaleY = SkScalarAbs(m[SkMatrix::kMScaleY]);
97*c8dee2aaSAndroid Build Coastguard Worker     float absSkewY  = SkScalarAbs(m[SkMatrix::kMSkewY]);
98*c8dee2aaSAndroid Build Coastguard Worker 
99*c8dee2aaSAndroid Build Coastguard Worker     // The maximum absolute column sum norm of the upper left 2x2
100*c8dee2aaSAndroid Build Coastguard Worker     float norm = std::max(absScaleX + absSkewY, absSkewX + absScaleY);
101*c8dee2aaSAndroid Build Coastguard Worker 
102*c8dee2aaSAndroid Build Coastguard Worker     return absSkewX > kTol * norm || absSkewY > kTol * norm;
103*c8dee2aaSAndroid Build Coastguard Worker }
104*c8dee2aaSAndroid Build Coastguard Worker 
105*c8dee2aaSAndroid Build Coastguard Worker class FillRRectOpImpl final : public GrMeshDrawOp {
106*c8dee2aaSAndroid Build Coastguard Worker private:
107*c8dee2aaSAndroid Build Coastguard Worker     using Helper = GrSimpleMeshDrawOpHelper;
108*c8dee2aaSAndroid Build Coastguard Worker 
109*c8dee2aaSAndroid Build Coastguard Worker public:
110*c8dee2aaSAndroid Build Coastguard Worker     DEFINE_OP_CLASS_ID
111*c8dee2aaSAndroid Build Coastguard Worker 
112*c8dee2aaSAndroid Build Coastguard Worker     struct LocalCoords {
113*c8dee2aaSAndroid Build Coastguard Worker         enum class Type : bool { kRect, kMatrix };
LocalCoordsskgpu::ganesh::FillRRectOp::__anone688170d0111::FillRRectOpImpl::LocalCoords114*c8dee2aaSAndroid Build Coastguard Worker         LocalCoords(const SkRect& localRect)
115*c8dee2aaSAndroid Build Coastguard Worker                 : fType(Type::kRect)
116*c8dee2aaSAndroid Build Coastguard Worker                 , fRect(localRect) {}
LocalCoordsskgpu::ganesh::FillRRectOp::__anone688170d0111::FillRRectOpImpl::LocalCoords117*c8dee2aaSAndroid Build Coastguard Worker         LocalCoords(const SkMatrix& localMatrix)
118*c8dee2aaSAndroid Build Coastguard Worker                 : fType(Type::kMatrix)
119*c8dee2aaSAndroid Build Coastguard Worker                 , fMatrix(localMatrix) {}
120*c8dee2aaSAndroid Build Coastguard Worker         Type fType;
121*c8dee2aaSAndroid Build Coastguard Worker         union {
122*c8dee2aaSAndroid Build Coastguard Worker             SkRect fRect;
123*c8dee2aaSAndroid Build Coastguard Worker             SkMatrix fMatrix;
124*c8dee2aaSAndroid Build Coastguard Worker         };
125*c8dee2aaSAndroid Build Coastguard Worker     };
126*c8dee2aaSAndroid Build Coastguard Worker 
127*c8dee2aaSAndroid Build Coastguard Worker     static GrOp::Owner Make(GrRecordingContext*,
128*c8dee2aaSAndroid Build Coastguard Worker                             SkArenaAlloc*,
129*c8dee2aaSAndroid Build Coastguard Worker                             GrPaint&&,
130*c8dee2aaSAndroid Build Coastguard Worker                             const SkMatrix& viewMatrix,
131*c8dee2aaSAndroid Build Coastguard Worker                             const SkRRect&,
132*c8dee2aaSAndroid Build Coastguard Worker                             const LocalCoords&,
133*c8dee2aaSAndroid Build Coastguard Worker                             GrAA);
134*c8dee2aaSAndroid Build Coastguard Worker 
name() const135*c8dee2aaSAndroid Build Coastguard Worker     const char* name() const override { return "FillRRectOp"; }
136*c8dee2aaSAndroid Build Coastguard Worker 
fixedFunctionFlags() const137*c8dee2aaSAndroid Build Coastguard Worker     FixedFunctionFlags fixedFunctionFlags() const override { return fHelper.fixedFunctionFlags(); }
138*c8dee2aaSAndroid Build Coastguard Worker 
139*c8dee2aaSAndroid Build Coastguard Worker     ClipResult clipToShape(skgpu::ganesh::SurfaceDrawContext*,
140*c8dee2aaSAndroid Build Coastguard Worker                            SkClipOp,
141*c8dee2aaSAndroid Build Coastguard Worker                            const SkMatrix& clipMatrix,
142*c8dee2aaSAndroid Build Coastguard Worker                            const GrShape&,
143*c8dee2aaSAndroid Build Coastguard Worker                            GrAA) override;
144*c8dee2aaSAndroid Build Coastguard Worker 
145*c8dee2aaSAndroid Build Coastguard Worker     GrProcessorSet::Analysis finalize(const GrCaps&, const GrAppliedClip*, GrClampType) override;
146*c8dee2aaSAndroid Build Coastguard Worker     CombineResult onCombineIfPossible(GrOp*, SkArenaAlloc*, const GrCaps&) override;
147*c8dee2aaSAndroid Build Coastguard Worker 
148*c8dee2aaSAndroid Build Coastguard Worker #if defined(GPU_TEST_UTILS)
149*c8dee2aaSAndroid Build Coastguard Worker     SkString onDumpInfo() const override;
150*c8dee2aaSAndroid Build Coastguard Worker #endif
151*c8dee2aaSAndroid Build Coastguard Worker 
visitProxies(const GrVisitProxyFunc & func) const152*c8dee2aaSAndroid Build Coastguard Worker     void visitProxies(const GrVisitProxyFunc& func) const override {
153*c8dee2aaSAndroid Build Coastguard Worker         if (fProgramInfo) {
154*c8dee2aaSAndroid Build Coastguard Worker             fProgramInfo->visitFPProxies(func);
155*c8dee2aaSAndroid Build Coastguard Worker         } else {
156*c8dee2aaSAndroid Build Coastguard Worker             fHelper.visitProxies(func);
157*c8dee2aaSAndroid Build Coastguard Worker         }
158*c8dee2aaSAndroid Build Coastguard Worker     }
159*c8dee2aaSAndroid Build Coastguard Worker 
160*c8dee2aaSAndroid Build Coastguard Worker     void onPrepareDraws(GrMeshDrawTarget*) override;
161*c8dee2aaSAndroid Build Coastguard Worker 
162*c8dee2aaSAndroid Build Coastguard Worker     void onExecute(GrOpFlushState*, const SkRect& chainBounds) override;
163*c8dee2aaSAndroid Build Coastguard Worker 
164*c8dee2aaSAndroid Build Coastguard Worker private:
165*c8dee2aaSAndroid Build Coastguard Worker     friend class ::GrSimpleMeshDrawOpHelper; // for access to ctor
166*c8dee2aaSAndroid Build Coastguard Worker     friend class ::GrOp;         // for access to ctor
167*c8dee2aaSAndroid Build Coastguard Worker 
168*c8dee2aaSAndroid Build Coastguard Worker     enum class ProcessorFlags {
169*c8dee2aaSAndroid Build Coastguard Worker         kNone             = 0,
170*c8dee2aaSAndroid Build Coastguard Worker         kUseHWDerivatives = 1 << 0,
171*c8dee2aaSAndroid Build Coastguard Worker         kHasLocalCoords   = 1 << 1,
172*c8dee2aaSAndroid Build Coastguard Worker         kWideColor        = 1 << 2,
173*c8dee2aaSAndroid Build Coastguard Worker         kMSAAEnabled      = 1 << 3,
174*c8dee2aaSAndroid Build Coastguard Worker         kFakeNonAA        = 1 << 4,
175*c8dee2aaSAndroid Build Coastguard Worker     };
176*c8dee2aaSAndroid Build Coastguard Worker     constexpr static int kNumProcessorFlags = 5;
177*c8dee2aaSAndroid Build Coastguard Worker 
178*c8dee2aaSAndroid Build Coastguard Worker     SK_DECL_BITFIELD_CLASS_OPS_FRIENDS(ProcessorFlags);
179*c8dee2aaSAndroid Build Coastguard Worker 
180*c8dee2aaSAndroid Build Coastguard Worker     class Processor;
181*c8dee2aaSAndroid Build Coastguard Worker 
182*c8dee2aaSAndroid Build Coastguard Worker     FillRRectOpImpl(GrProcessorSet*,
183*c8dee2aaSAndroid Build Coastguard Worker                     const SkPMColor4f& paintColor,
184*c8dee2aaSAndroid Build Coastguard Worker                     SkArenaAlloc*,
185*c8dee2aaSAndroid Build Coastguard Worker                     const SkMatrix& viewMatrix,
186*c8dee2aaSAndroid Build Coastguard Worker                     const SkRRect&,
187*c8dee2aaSAndroid Build Coastguard Worker                     const LocalCoords&,
188*c8dee2aaSAndroid Build Coastguard Worker                     ProcessorFlags);
189*c8dee2aaSAndroid Build Coastguard Worker 
programInfo()190*c8dee2aaSAndroid Build Coastguard Worker     GrProgramInfo* programInfo() override { return fProgramInfo; }
191*c8dee2aaSAndroid Build Coastguard Worker 
192*c8dee2aaSAndroid Build Coastguard Worker     // Create a GrProgramInfo object in the provided arena
193*c8dee2aaSAndroid Build Coastguard Worker     void onCreateProgramInfo(const GrCaps*,
194*c8dee2aaSAndroid Build Coastguard Worker                              SkArenaAlloc*,
195*c8dee2aaSAndroid Build Coastguard Worker                              const GrSurfaceProxyView& writeView,
196*c8dee2aaSAndroid Build Coastguard Worker                              bool usesMSAASurface,
197*c8dee2aaSAndroid Build Coastguard Worker                              GrAppliedClip&&,
198*c8dee2aaSAndroid Build Coastguard Worker                              const GrDstProxyView&,
199*c8dee2aaSAndroid Build Coastguard Worker                              GrXferBarrierFlags renderPassXferBarriers,
200*c8dee2aaSAndroid Build Coastguard Worker                              GrLoadOp colorLoadOp) override;
201*c8dee2aaSAndroid Build Coastguard Worker 
202*c8dee2aaSAndroid Build Coastguard Worker     Helper         fHelper;
203*c8dee2aaSAndroid Build Coastguard Worker     ProcessorFlags fProcessorFlags;
204*c8dee2aaSAndroid Build Coastguard Worker 
205*c8dee2aaSAndroid Build Coastguard Worker     struct Instance {
Instanceskgpu::ganesh::FillRRectOp::__anone688170d0111::FillRRectOpImpl::Instance206*c8dee2aaSAndroid Build Coastguard Worker         Instance(const SkMatrix& viewMatrix,
207*c8dee2aaSAndroid Build Coastguard Worker                  const SkRRect& rrect,
208*c8dee2aaSAndroid Build Coastguard Worker                  const LocalCoords& localCoords,
209*c8dee2aaSAndroid Build Coastguard Worker                  const SkPMColor4f& color)
210*c8dee2aaSAndroid Build Coastguard Worker                 : fViewMatrix(viewMatrix), fRRect(rrect), fLocalCoords(localCoords), fColor(color) {
211*c8dee2aaSAndroid Build Coastguard Worker         }
212*c8dee2aaSAndroid Build Coastguard Worker         SkMatrix fViewMatrix;
213*c8dee2aaSAndroid Build Coastguard Worker         SkRRect fRRect;
214*c8dee2aaSAndroid Build Coastguard Worker         LocalCoords fLocalCoords;
215*c8dee2aaSAndroid Build Coastguard Worker         SkPMColor4f fColor;
216*c8dee2aaSAndroid Build Coastguard Worker         Instance* fNext = nullptr;
217*c8dee2aaSAndroid Build Coastguard Worker     };
218*c8dee2aaSAndroid Build Coastguard Worker 
219*c8dee2aaSAndroid Build Coastguard Worker     Instance* fHeadInstance;
220*c8dee2aaSAndroid Build Coastguard Worker     Instance** fTailInstance;
221*c8dee2aaSAndroid Build Coastguard Worker     int fInstanceCount = 1;
222*c8dee2aaSAndroid Build Coastguard Worker 
223*c8dee2aaSAndroid Build Coastguard Worker     sk_sp<const GrBuffer> fInstanceBuffer;
224*c8dee2aaSAndroid Build Coastguard Worker     sk_sp<const GrBuffer> fVertexBuffer;
225*c8dee2aaSAndroid Build Coastguard Worker     sk_sp<const GrBuffer> fIndexBuffer;
226*c8dee2aaSAndroid Build Coastguard Worker     int fBaseInstance = 0;
227*c8dee2aaSAndroid Build Coastguard Worker 
228*c8dee2aaSAndroid Build Coastguard Worker     // If this op is prePrepared the created programInfo will be stored here for use in
229*c8dee2aaSAndroid Build Coastguard Worker     // onExecute. In the prePrepared case it will have been stored in the record-time arena.
230*c8dee2aaSAndroid Build Coastguard Worker     GrProgramInfo* fProgramInfo = nullptr;
231*c8dee2aaSAndroid Build Coastguard Worker };
232*c8dee2aaSAndroid Build Coastguard Worker 
233*c8dee2aaSAndroid Build Coastguard Worker SK_MAKE_BITFIELD_CLASS_OPS(FillRRectOpImpl::ProcessorFlags)
234*c8dee2aaSAndroid Build Coastguard Worker 
235*c8dee2aaSAndroid Build Coastguard Worker // Hardware derivatives are not always accurate enough for highly elliptical corners. This method
236*c8dee2aaSAndroid Build Coastguard Worker // checks to make sure the corners will still all look good if we use HW derivatives.
237*c8dee2aaSAndroid Build Coastguard Worker bool can_use_hw_derivatives_with_coverage(const GrShaderCaps&,
238*c8dee2aaSAndroid Build Coastguard Worker                                           const SkMatrix&,
239*c8dee2aaSAndroid Build Coastguard Worker                                           const SkRRect&);
240*c8dee2aaSAndroid Build Coastguard Worker 
Make(GrRecordingContext * ctx,SkArenaAlloc * arena,GrPaint && paint,const SkMatrix & viewMatrix,const SkRRect & rrect,const LocalCoords & localCoords,GrAA aa)241*c8dee2aaSAndroid Build Coastguard Worker GrOp::Owner FillRRectOpImpl::Make(GrRecordingContext* ctx,
242*c8dee2aaSAndroid Build Coastguard Worker                                   SkArenaAlloc* arena,
243*c8dee2aaSAndroid Build Coastguard Worker                                   GrPaint&& paint,
244*c8dee2aaSAndroid Build Coastguard Worker                                   const SkMatrix& viewMatrix,
245*c8dee2aaSAndroid Build Coastguard Worker                                   const SkRRect& rrect,
246*c8dee2aaSAndroid Build Coastguard Worker                                   const LocalCoords& localCoords,
247*c8dee2aaSAndroid Build Coastguard Worker                                   GrAA aa) {
248*c8dee2aaSAndroid Build Coastguard Worker     const GrCaps* caps = ctx->priv().caps();
249*c8dee2aaSAndroid Build Coastguard Worker 
250*c8dee2aaSAndroid Build Coastguard Worker     if (!caps->drawInstancedSupport()) {
251*c8dee2aaSAndroid Build Coastguard Worker         return nullptr;
252*c8dee2aaSAndroid Build Coastguard Worker     }
253*c8dee2aaSAndroid Build Coastguard Worker 
254*c8dee2aaSAndroid Build Coastguard Worker     // We transform into a normalized -1..+1 space to draw the round rect. If the boundaries are too
255*c8dee2aaSAndroid Build Coastguard Worker     // large, the math can overflow. The caller can fall back on path rendering if this is the case.
256*c8dee2aaSAndroid Build Coastguard Worker     if (std::max(rrect.height(), rrect.width()) >= 1e6f) {
257*c8dee2aaSAndroid Build Coastguard Worker         return nullptr;
258*c8dee2aaSAndroid Build Coastguard Worker     }
259*c8dee2aaSAndroid Build Coastguard Worker 
260*c8dee2aaSAndroid Build Coastguard Worker     ProcessorFlags flags = ProcessorFlags::kNone;
261*c8dee2aaSAndroid Build Coastguard Worker     // TODO: Support perspective in a follow-on CL. This shouldn't be difficult, since we already
262*c8dee2aaSAndroid Build Coastguard Worker     // use HW derivatives. The only trick will be adjusting the AA outset to account for
263*c8dee2aaSAndroid Build Coastguard Worker     // perspective. (i.e., outset = 0.5 * z.)
264*c8dee2aaSAndroid Build Coastguard Worker     if (viewMatrix.hasPerspective()) {
265*c8dee2aaSAndroid Build Coastguard Worker         return nullptr;
266*c8dee2aaSAndroid Build Coastguard Worker     }
267*c8dee2aaSAndroid Build Coastguard Worker     if (can_use_hw_derivatives_with_coverage(*caps->shaderCaps(), viewMatrix, rrect)) {
268*c8dee2aaSAndroid Build Coastguard Worker         // HW derivatives (more specifically, fwidth()) are consistently faster on all platforms in
269*c8dee2aaSAndroid Build Coastguard Worker         // coverage mode. We use them as long as the approximation will be accurate enough.
270*c8dee2aaSAndroid Build Coastguard Worker         flags |= ProcessorFlags::kUseHWDerivatives;
271*c8dee2aaSAndroid Build Coastguard Worker     }
272*c8dee2aaSAndroid Build Coastguard Worker     if (aa == GrAA::kNo) {
273*c8dee2aaSAndroid Build Coastguard Worker         flags |= ProcessorFlags::kFakeNonAA;
274*c8dee2aaSAndroid Build Coastguard Worker     }
275*c8dee2aaSAndroid Build Coastguard Worker 
276*c8dee2aaSAndroid Build Coastguard Worker     return Helper::FactoryHelper<FillRRectOpImpl>(ctx, std::move(paint), arena, viewMatrix, rrect,
277*c8dee2aaSAndroid Build Coastguard Worker                                                   localCoords, flags);
278*c8dee2aaSAndroid Build Coastguard Worker }
279*c8dee2aaSAndroid Build Coastguard Worker 
FillRRectOpImpl(GrProcessorSet * processorSet,const SkPMColor4f & paintColor,SkArenaAlloc * arena,const SkMatrix & viewMatrix,const SkRRect & rrect,const LocalCoords & localCoords,ProcessorFlags processorFlags)280*c8dee2aaSAndroid Build Coastguard Worker FillRRectOpImpl::FillRRectOpImpl(GrProcessorSet* processorSet,
281*c8dee2aaSAndroid Build Coastguard Worker                                  const SkPMColor4f& paintColor,
282*c8dee2aaSAndroid Build Coastguard Worker                                  SkArenaAlloc* arena,
283*c8dee2aaSAndroid Build Coastguard Worker                                  const SkMatrix& viewMatrix,
284*c8dee2aaSAndroid Build Coastguard Worker                                  const SkRRect& rrect,
285*c8dee2aaSAndroid Build Coastguard Worker                                  const LocalCoords& localCoords,
286*c8dee2aaSAndroid Build Coastguard Worker                                  ProcessorFlags processorFlags)
287*c8dee2aaSAndroid Build Coastguard Worker         : GrMeshDrawOp(ClassID())
288*c8dee2aaSAndroid Build Coastguard Worker         , fHelper(processorSet,
289*c8dee2aaSAndroid Build Coastguard Worker                   (processorFlags & ProcessorFlags::kFakeNonAA)
290*c8dee2aaSAndroid Build Coastguard Worker                           ? GrAAType::kNone
291*c8dee2aaSAndroid Build Coastguard Worker                           : GrAAType::kCoverage)  // Use analytic AA even if the RT is MSAA.
292*c8dee2aaSAndroid Build Coastguard Worker         , fProcessorFlags(processorFlags & ~(ProcessorFlags::kHasLocalCoords |
293*c8dee2aaSAndroid Build Coastguard Worker                                              ProcessorFlags::kWideColor |
294*c8dee2aaSAndroid Build Coastguard Worker                                              ProcessorFlags::kMSAAEnabled))
295*c8dee2aaSAndroid Build Coastguard Worker         , fHeadInstance(arena->make<Instance>(viewMatrix, rrect, localCoords, paintColor))
296*c8dee2aaSAndroid Build Coastguard Worker         , fTailInstance(&fHeadInstance->fNext) {
297*c8dee2aaSAndroid Build Coastguard Worker     // FillRRectOp::Make fails if there is perspective.
298*c8dee2aaSAndroid Build Coastguard Worker     SkASSERT(!viewMatrix.hasPerspective());
299*c8dee2aaSAndroid Build Coastguard Worker     this->setBounds(viewMatrix.mapRect(rrect.getBounds()),
300*c8dee2aaSAndroid Build Coastguard Worker                     GrOp::HasAABloat(!(processorFlags & ProcessorFlags::kFakeNonAA)),
301*c8dee2aaSAndroid Build Coastguard Worker                     GrOp::IsHairline::kNo);
302*c8dee2aaSAndroid Build Coastguard Worker }
303*c8dee2aaSAndroid Build Coastguard Worker 
clipToShape(skgpu::ganesh::SurfaceDrawContext * sdc,SkClipOp clipOp,const SkMatrix & clipMatrix,const GrShape & shape,GrAA aa)304*c8dee2aaSAndroid Build Coastguard Worker GrDrawOp::ClipResult FillRRectOpImpl::clipToShape(skgpu::ganesh::SurfaceDrawContext* sdc,
305*c8dee2aaSAndroid Build Coastguard Worker                                                   SkClipOp clipOp,
306*c8dee2aaSAndroid Build Coastguard Worker                                                   const SkMatrix& clipMatrix,
307*c8dee2aaSAndroid Build Coastguard Worker                                                   const GrShape& shape,
308*c8dee2aaSAndroid Build Coastguard Worker                                                   GrAA aa) {
309*c8dee2aaSAndroid Build Coastguard Worker     SkASSERT(fInstanceCount == 1);  // This needs to be called before combining.
310*c8dee2aaSAndroid Build Coastguard Worker     SkASSERT(fHeadInstance->fNext == nullptr);
311*c8dee2aaSAndroid Build Coastguard Worker 
312*c8dee2aaSAndroid Build Coastguard Worker     if ((shape.isRect() || shape.isRRect()) &&
313*c8dee2aaSAndroid Build Coastguard Worker         clipOp == SkClipOp::kIntersect &&
314*c8dee2aaSAndroid Build Coastguard Worker         (aa == GrAA::kNo) == (fProcessorFlags & ProcessorFlags::kFakeNonAA)) {
315*c8dee2aaSAndroid Build Coastguard Worker         // The clip shape is a round rect. Attempt to map it to a round rect in "viewMatrix" space.
316*c8dee2aaSAndroid Build Coastguard Worker         SkRRect clipRRect;
317*c8dee2aaSAndroid Build Coastguard Worker         if (clipMatrix == fHeadInstance->fViewMatrix) {
318*c8dee2aaSAndroid Build Coastguard Worker             if (shape.isRect()) {
319*c8dee2aaSAndroid Build Coastguard Worker                 clipRRect.setRect(shape.rect());
320*c8dee2aaSAndroid Build Coastguard Worker             } else {
321*c8dee2aaSAndroid Build Coastguard Worker                 clipRRect = shape.rrect();
322*c8dee2aaSAndroid Build Coastguard Worker             }
323*c8dee2aaSAndroid Build Coastguard Worker         } else {
324*c8dee2aaSAndroid Build Coastguard Worker             // Find a matrix that maps from "clipMatrix" space to "viewMatrix" space.
325*c8dee2aaSAndroid Build Coastguard Worker             SkASSERT(!fHeadInstance->fViewMatrix.hasPerspective());
326*c8dee2aaSAndroid Build Coastguard Worker             if (clipMatrix.hasPerspective()) {
327*c8dee2aaSAndroid Build Coastguard Worker                 return ClipResult::kFail;
328*c8dee2aaSAndroid Build Coastguard Worker             }
329*c8dee2aaSAndroid Build Coastguard Worker             SkMatrix clipToView;
330*c8dee2aaSAndroid Build Coastguard Worker             if (!fHeadInstance->fViewMatrix.invert(&clipToView)) {
331*c8dee2aaSAndroid Build Coastguard Worker                 return ClipResult::kClippedOut;
332*c8dee2aaSAndroid Build Coastguard Worker             }
333*c8dee2aaSAndroid Build Coastguard Worker             clipToView.preConcat(clipMatrix);
334*c8dee2aaSAndroid Build Coastguard Worker             SkASSERT(!clipToView.hasPerspective());
335*c8dee2aaSAndroid Build Coastguard Worker 
336*c8dee2aaSAndroid Build Coastguard Worker             if (skews_are_relevant(clipToView)) {
337*c8dee2aaSAndroid Build Coastguard Worker                 // A rect in "clipMatrix" space is not a rect in "viewMatrix" space.
338*c8dee2aaSAndroid Build Coastguard Worker                 return ClipResult::kFail;
339*c8dee2aaSAndroid Build Coastguard Worker             }
340*c8dee2aaSAndroid Build Coastguard Worker             clipToView.setSkewX(0);
341*c8dee2aaSAndroid Build Coastguard Worker             clipToView.setSkewY(0);
342*c8dee2aaSAndroid Build Coastguard Worker             SkASSERT(clipToView.rectStaysRect());
343*c8dee2aaSAndroid Build Coastguard Worker 
344*c8dee2aaSAndroid Build Coastguard Worker             if (shape.isRect()) {
345*c8dee2aaSAndroid Build Coastguard Worker                 clipRRect.setRect(clipToView.mapRect(shape.rect()));
346*c8dee2aaSAndroid Build Coastguard Worker             } else {
347*c8dee2aaSAndroid Build Coastguard Worker                 if (!shape.rrect().transform(clipToView, &clipRRect)) {
348*c8dee2aaSAndroid Build Coastguard Worker                     // Transforming the rrect failed. This shouldn't generally happen except in
349*c8dee2aaSAndroid Build Coastguard Worker                     // cases of fp32 overflow.
350*c8dee2aaSAndroid Build Coastguard Worker                     return ClipResult::kFail;
351*c8dee2aaSAndroid Build Coastguard Worker                 }
352*c8dee2aaSAndroid Build Coastguard Worker             }
353*c8dee2aaSAndroid Build Coastguard Worker         }
354*c8dee2aaSAndroid Build Coastguard Worker 
355*c8dee2aaSAndroid Build Coastguard Worker         // Intersect our round rect with the clip shape.
356*c8dee2aaSAndroid Build Coastguard Worker         SkRRect isectRRect;
357*c8dee2aaSAndroid Build Coastguard Worker         if (fHeadInstance->fRRect.isRect() && clipRRect.isRect()) {
358*c8dee2aaSAndroid Build Coastguard Worker             SkRect isectRect;
359*c8dee2aaSAndroid Build Coastguard Worker             if (!isectRect.intersect(fHeadInstance->fRRect.rect(), clipRRect.rect())) {
360*c8dee2aaSAndroid Build Coastguard Worker                 return ClipResult::kClippedOut;
361*c8dee2aaSAndroid Build Coastguard Worker             }
362*c8dee2aaSAndroid Build Coastguard Worker             isectRRect.setRect(isectRect);
363*c8dee2aaSAndroid Build Coastguard Worker         } else {
364*c8dee2aaSAndroid Build Coastguard Worker             isectRRect = SkRRectPriv::ConservativeIntersect(fHeadInstance->fRRect, clipRRect);
365*c8dee2aaSAndroid Build Coastguard Worker             if (isectRRect.isEmpty()) {
366*c8dee2aaSAndroid Build Coastguard Worker                 // The round rects did not intersect at all or the intersection was too complicated
367*c8dee2aaSAndroid Build Coastguard Worker                 // to compute quickly.
368*c8dee2aaSAndroid Build Coastguard Worker                 return ClipResult::kFail;
369*c8dee2aaSAndroid Build Coastguard Worker             }
370*c8dee2aaSAndroid Build Coastguard Worker         }
371*c8dee2aaSAndroid Build Coastguard Worker 
372*c8dee2aaSAndroid Build Coastguard Worker         // Don't apply the clip geometrically if it becomes subpixel, since then the hairline
373*c8dee2aaSAndroid Build Coastguard Worker         // rendering may outset beyond the original clip.
374*c8dee2aaSAndroid Build Coastguard Worker         SkRect devISectBounds = fHeadInstance->fViewMatrix.mapRect(isectRRect.rect());
375*c8dee2aaSAndroid Build Coastguard Worker         if (devISectBounds.width() < 1.f || devISectBounds.height() < 1.f) {
376*c8dee2aaSAndroid Build Coastguard Worker             return ClipResult::kFail;
377*c8dee2aaSAndroid Build Coastguard Worker         }
378*c8dee2aaSAndroid Build Coastguard Worker 
379*c8dee2aaSAndroid Build Coastguard Worker         if (fHeadInstance->fLocalCoords.fType == LocalCoords::Type::kRect) {
380*c8dee2aaSAndroid Build Coastguard Worker             // Update the local rect.
381*c8dee2aaSAndroid Build Coastguard Worker             auto rect = sk_bit_cast<skvx::float4>(fHeadInstance->fRRect.rect());
382*c8dee2aaSAndroid Build Coastguard Worker             auto local = sk_bit_cast<skvx::float4>(fHeadInstance->fLocalCoords.fRect);
383*c8dee2aaSAndroid Build Coastguard Worker             auto isect = sk_bit_cast<skvx::float4>(isectRRect.rect());
384*c8dee2aaSAndroid Build Coastguard Worker             auto rectToLocalSize = (local - skvx::shuffle<2,3,0,1>(local)) /
385*c8dee2aaSAndroid Build Coastguard Worker                                    (rect - skvx::shuffle<2,3,0,1>(rect));
386*c8dee2aaSAndroid Build Coastguard Worker             auto localCoordsRect = (isect - rect) * rectToLocalSize + local;
387*c8dee2aaSAndroid Build Coastguard Worker             fHeadInstance->fLocalCoords.fRect.setLTRB(localCoordsRect.x(),
388*c8dee2aaSAndroid Build Coastguard Worker                                                       localCoordsRect.y(),
389*c8dee2aaSAndroid Build Coastguard Worker                                                       localCoordsRect.z(),
390*c8dee2aaSAndroid Build Coastguard Worker                                                       localCoordsRect.w());
391*c8dee2aaSAndroid Build Coastguard Worker         }
392*c8dee2aaSAndroid Build Coastguard Worker 
393*c8dee2aaSAndroid Build Coastguard Worker         // Update the round rect.
394*c8dee2aaSAndroid Build Coastguard Worker         fHeadInstance->fRRect = isectRRect;
395*c8dee2aaSAndroid Build Coastguard Worker         return ClipResult::kClippedGeometrically;
396*c8dee2aaSAndroid Build Coastguard Worker     }
397*c8dee2aaSAndroid Build Coastguard Worker 
398*c8dee2aaSAndroid Build Coastguard Worker     return ClipResult::kFail;
399*c8dee2aaSAndroid Build Coastguard Worker }
400*c8dee2aaSAndroid Build Coastguard Worker 
finalize(const GrCaps & caps,const GrAppliedClip * clip,GrClampType clampType)401*c8dee2aaSAndroid Build Coastguard Worker GrProcessorSet::Analysis FillRRectOpImpl::finalize(const GrCaps& caps, const GrAppliedClip* clip,
402*c8dee2aaSAndroid Build Coastguard Worker                                                    GrClampType clampType) {
403*c8dee2aaSAndroid Build Coastguard Worker     SkASSERT(fInstanceCount == 1);
404*c8dee2aaSAndroid Build Coastguard Worker     SkASSERT(fHeadInstance->fNext == nullptr);
405*c8dee2aaSAndroid Build Coastguard Worker 
406*c8dee2aaSAndroid Build Coastguard Worker     bool isWideColor;
407*c8dee2aaSAndroid Build Coastguard Worker     auto analysis = fHelper.finalizeProcessors(caps, clip, clampType,
408*c8dee2aaSAndroid Build Coastguard Worker                                                GrProcessorAnalysisCoverage::kSingleChannel,
409*c8dee2aaSAndroid Build Coastguard Worker                                                &fHeadInstance->fColor, &isWideColor);
410*c8dee2aaSAndroid Build Coastguard Worker     if (isWideColor) {
411*c8dee2aaSAndroid Build Coastguard Worker         fProcessorFlags |= ProcessorFlags::kWideColor;
412*c8dee2aaSAndroid Build Coastguard Worker     }
413*c8dee2aaSAndroid Build Coastguard Worker     if (analysis.usesLocalCoords()) {
414*c8dee2aaSAndroid Build Coastguard Worker         fProcessorFlags |= ProcessorFlags::kHasLocalCoords;
415*c8dee2aaSAndroid Build Coastguard Worker     }
416*c8dee2aaSAndroid Build Coastguard Worker     return analysis;
417*c8dee2aaSAndroid Build Coastguard Worker }
418*c8dee2aaSAndroid Build Coastguard Worker 
onCombineIfPossible(GrOp * op,SkArenaAlloc *,const GrCaps & caps)419*c8dee2aaSAndroid Build Coastguard Worker GrOp::CombineResult FillRRectOpImpl::onCombineIfPossible(GrOp* op,
420*c8dee2aaSAndroid Build Coastguard Worker                                                          SkArenaAlloc*,
421*c8dee2aaSAndroid Build Coastguard Worker                                                          const GrCaps& caps) {
422*c8dee2aaSAndroid Build Coastguard Worker     auto that = op->cast<FillRRectOpImpl>();
423*c8dee2aaSAndroid Build Coastguard Worker     if (!fHelper.isCompatible(that->fHelper, caps, this->bounds(), that->bounds()) ||
424*c8dee2aaSAndroid Build Coastguard Worker         fProcessorFlags != that->fProcessorFlags) {
425*c8dee2aaSAndroid Build Coastguard Worker         return CombineResult::kCannotCombine;
426*c8dee2aaSAndroid Build Coastguard Worker     }
427*c8dee2aaSAndroid Build Coastguard Worker 
428*c8dee2aaSAndroid Build Coastguard Worker     *fTailInstance = that->fHeadInstance;
429*c8dee2aaSAndroid Build Coastguard Worker     fTailInstance = that->fTailInstance;
430*c8dee2aaSAndroid Build Coastguard Worker     fInstanceCount += that->fInstanceCount;
431*c8dee2aaSAndroid Build Coastguard Worker     return CombineResult::kMerged;
432*c8dee2aaSAndroid Build Coastguard Worker }
433*c8dee2aaSAndroid Build Coastguard Worker 
434*c8dee2aaSAndroid Build Coastguard Worker #if defined(GPU_TEST_UTILS)
onDumpInfo() const435*c8dee2aaSAndroid Build Coastguard Worker SkString FillRRectOpImpl::onDumpInfo() const {
436*c8dee2aaSAndroid Build Coastguard Worker     SkString str = SkStringPrintf("# instances: %d\n", fInstanceCount);
437*c8dee2aaSAndroid Build Coastguard Worker     str += fHelper.dumpInfo();
438*c8dee2aaSAndroid Build Coastguard Worker     int i = 0;
439*c8dee2aaSAndroid Build Coastguard Worker     for (Instance* tmp = fHeadInstance; tmp; tmp = tmp->fNext, ++i) {
440*c8dee2aaSAndroid Build Coastguard Worker         str.appendf("%d: Color: [%.2f, %.2f, %.2f, %.2f] ",
441*c8dee2aaSAndroid Build Coastguard Worker                     i, tmp->fColor.fR, tmp->fColor.fG, tmp->fColor.fB, tmp->fColor.fA);
442*c8dee2aaSAndroid Build Coastguard Worker         SkMatrix m = tmp->fViewMatrix;
443*c8dee2aaSAndroid Build Coastguard Worker         str.appendf("ViewMatrix: [%.2f, %.2f, %.2f, %.2f, %.2f, %.2f, %.2f, %.2f, %.2f] ",
444*c8dee2aaSAndroid Build Coastguard Worker                     m[0], m[1], m[2], m[3], m[4], m[5], m[6], m[7], m[8]);
445*c8dee2aaSAndroid Build Coastguard Worker         SkRect r = tmp->fRRect.rect();
446*c8dee2aaSAndroid Build Coastguard Worker         str.appendf("Rect: [%f %f %f %f]\n", r.fLeft, r.fTop, r.fRight, r.fBottom);
447*c8dee2aaSAndroid Build Coastguard Worker     }
448*c8dee2aaSAndroid Build Coastguard Worker     return str;
449*c8dee2aaSAndroid Build Coastguard Worker }
450*c8dee2aaSAndroid Build Coastguard Worker #endif
451*c8dee2aaSAndroid Build Coastguard Worker 
452*c8dee2aaSAndroid Build Coastguard Worker class FillRRectOpImpl::Processor final : public GrGeometryProcessor {
453*c8dee2aaSAndroid Build Coastguard Worker public:
Make(SkArenaAlloc * arena,GrAAType aaType,ProcessorFlags flags)454*c8dee2aaSAndroid Build Coastguard Worker     static GrGeometryProcessor* Make(SkArenaAlloc* arena, GrAAType aaType, ProcessorFlags flags) {
455*c8dee2aaSAndroid Build Coastguard Worker         return arena->make([&](void* ptr) {
456*c8dee2aaSAndroid Build Coastguard Worker             return new (ptr) Processor(aaType, flags);
457*c8dee2aaSAndroid Build Coastguard Worker         });
458*c8dee2aaSAndroid Build Coastguard Worker     }
459*c8dee2aaSAndroid Build Coastguard Worker 
name() const460*c8dee2aaSAndroid Build Coastguard Worker     const char* name() const override { return "FillRRectOp::Processor"; }
461*c8dee2aaSAndroid Build Coastguard Worker 
addToKey(const GrShaderCaps & caps,KeyBuilder * b) const462*c8dee2aaSAndroid Build Coastguard Worker     void addToKey(const GrShaderCaps& caps, KeyBuilder* b) const override {
463*c8dee2aaSAndroid Build Coastguard Worker         b->addBits(kNumProcessorFlags, (uint32_t)fFlags,  "flags");
464*c8dee2aaSAndroid Build Coastguard Worker     }
465*c8dee2aaSAndroid Build Coastguard Worker 
466*c8dee2aaSAndroid Build Coastguard Worker     std::unique_ptr<ProgramImpl> makeProgramImpl(const GrShaderCaps&) const override;
467*c8dee2aaSAndroid Build Coastguard Worker 
468*c8dee2aaSAndroid Build Coastguard Worker private:
469*c8dee2aaSAndroid Build Coastguard Worker     class Impl;
470*c8dee2aaSAndroid Build Coastguard Worker 
Processor(GrAAType aaType,ProcessorFlags flags)471*c8dee2aaSAndroid Build Coastguard Worker     Processor(GrAAType aaType, ProcessorFlags flags)
472*c8dee2aaSAndroid Build Coastguard Worker             : GrGeometryProcessor(kGrFillRRectOp_Processor_ClassID)
473*c8dee2aaSAndroid Build Coastguard Worker             , fFlags(flags) {
474*c8dee2aaSAndroid Build Coastguard Worker         this->setVertexAttributesWithImplicitOffsets(kVertexAttribs, std::size(kVertexAttribs));
475*c8dee2aaSAndroid Build Coastguard Worker 
476*c8dee2aaSAndroid Build Coastguard Worker         fInstanceAttribs.emplace_back("radii_x", kFloat4_GrVertexAttribType, SkSLType::kFloat4);
477*c8dee2aaSAndroid Build Coastguard Worker         fInstanceAttribs.emplace_back("radii_y", kFloat4_GrVertexAttribType, SkSLType::kFloat4);
478*c8dee2aaSAndroid Build Coastguard Worker         fInstanceAttribs.emplace_back("skew", kFloat4_GrVertexAttribType, SkSLType::kFloat4);
479*c8dee2aaSAndroid Build Coastguard Worker         if (fFlags & ProcessorFlags::kHasLocalCoords) {
480*c8dee2aaSAndroid Build Coastguard Worker             fInstanceAttribs.emplace_back("translate_and_localrotate",
481*c8dee2aaSAndroid Build Coastguard Worker                                           kFloat4_GrVertexAttribType,
482*c8dee2aaSAndroid Build Coastguard Worker                                           SkSLType::kFloat4);
483*c8dee2aaSAndroid Build Coastguard Worker             fInstanceAttribs.emplace_back(
484*c8dee2aaSAndroid Build Coastguard Worker                     "localrect", kFloat4_GrVertexAttribType, SkSLType::kFloat4);
485*c8dee2aaSAndroid Build Coastguard Worker         } else {
486*c8dee2aaSAndroid Build Coastguard Worker             fInstanceAttribs.emplace_back("translate_and_localrotate",
487*c8dee2aaSAndroid Build Coastguard Worker                                           kFloat2_GrVertexAttribType,
488*c8dee2aaSAndroid Build Coastguard Worker                                           SkSLType::kFloat2);
489*c8dee2aaSAndroid Build Coastguard Worker         }
490*c8dee2aaSAndroid Build Coastguard Worker         fColorAttrib = &fInstanceAttribs.push_back(
491*c8dee2aaSAndroid Build Coastguard Worker                 MakeColorAttribute("color", (fFlags & ProcessorFlags::kWideColor)));
492*c8dee2aaSAndroid Build Coastguard Worker         SkASSERT(fInstanceAttribs.size() <= kMaxInstanceAttribs);
493*c8dee2aaSAndroid Build Coastguard Worker         this->setInstanceAttributesWithImplicitOffsets(fInstanceAttribs.begin(),
494*c8dee2aaSAndroid Build Coastguard Worker                                                        fInstanceAttribs.size());
495*c8dee2aaSAndroid Build Coastguard Worker     }
496*c8dee2aaSAndroid Build Coastguard Worker 
497*c8dee2aaSAndroid Build Coastguard Worker     inline static constexpr Attribute kVertexAttribs[] = {
498*c8dee2aaSAndroid Build Coastguard Worker             {"radii_selector", kFloat4_GrVertexAttribType, SkSLType::kFloat4},
499*c8dee2aaSAndroid Build Coastguard Worker             {"corner_and_radius_outsets", kFloat4_GrVertexAttribType, SkSLType::kFloat4},
500*c8dee2aaSAndroid Build Coastguard Worker             // Coverage only.
501*c8dee2aaSAndroid Build Coastguard Worker             {"aa_bloat_and_coverage", kFloat4_GrVertexAttribType, SkSLType::kFloat4}};
502*c8dee2aaSAndroid Build Coastguard Worker 
503*c8dee2aaSAndroid Build Coastguard Worker     const ProcessorFlags fFlags;
504*c8dee2aaSAndroid Build Coastguard Worker 
505*c8dee2aaSAndroid Build Coastguard Worker     constexpr static int kMaxInstanceAttribs = 6;
506*c8dee2aaSAndroid Build Coastguard Worker     STArray<kMaxInstanceAttribs, Attribute> fInstanceAttribs;
507*c8dee2aaSAndroid Build Coastguard Worker     const Attribute* fColorAttrib;
508*c8dee2aaSAndroid Build Coastguard Worker };
509*c8dee2aaSAndroid Build Coastguard Worker 
510*c8dee2aaSAndroid Build Coastguard Worker // Our coverage geometry consists of an inset octagon with solid coverage, surrounded by linear
511*c8dee2aaSAndroid Build Coastguard Worker // coverage ramps on the horizontal and vertical edges, and "arc coverage" pieces on the diagonal
512*c8dee2aaSAndroid Build Coastguard Worker // edges. The Vertex struct tells the shader where to place its vertex within a normalized
513*c8dee2aaSAndroid Build Coastguard Worker // ([l, t, r, b] = [-1, -1, +1, +1]) space, and how to calculate coverage. See onEmitCode.
514*c8dee2aaSAndroid Build Coastguard Worker struct CoverageVertex {
515*c8dee2aaSAndroid Build Coastguard Worker     std::array<float, 4> fRadiiSelector;
516*c8dee2aaSAndroid Build Coastguard Worker     std::array<float, 2> fCorner;
517*c8dee2aaSAndroid Build Coastguard Worker     std::array<float, 2> fRadiusOutset;
518*c8dee2aaSAndroid Build Coastguard Worker     std::array<float, 2> fAABloatDirection;
519*c8dee2aaSAndroid Build Coastguard Worker     float fCoverage;
520*c8dee2aaSAndroid Build Coastguard Worker     float fIsLinearCoverage;
521*c8dee2aaSAndroid Build Coastguard Worker };
522*c8dee2aaSAndroid Build Coastguard Worker 
523*c8dee2aaSAndroid Build Coastguard Worker // This is the offset (when multiplied by radii) from the corners of a bounding box to the vertices
524*c8dee2aaSAndroid Build Coastguard Worker // of its inscribed octagon. We draw the outside portion of arcs with quarter-octagons rather than
525*c8dee2aaSAndroid Build Coastguard Worker // rectangles.
526*c8dee2aaSAndroid Build Coastguard Worker static constexpr float kOctoOffset = 1/(1 + SK_ScalarRoot2Over2);
527*c8dee2aaSAndroid Build Coastguard Worker 
528*c8dee2aaSAndroid Build Coastguard Worker static constexpr CoverageVertex kVertexData[] = {
529*c8dee2aaSAndroid Build Coastguard Worker         // Left inset edge.
530*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,0,1}},  {{-1,+1}},  {{0,-1}},  {{+1,0}},  1,  1},
531*c8dee2aaSAndroid Build Coastguard Worker         {{{1,0,0,0}},  {{-1,-1}},  {{0,+1}},  {{+1,0}},  1,  1},
532*c8dee2aaSAndroid Build Coastguard Worker 
533*c8dee2aaSAndroid Build Coastguard Worker         // Top inset edge.
534*c8dee2aaSAndroid Build Coastguard Worker         {{{1,0,0,0}},  {{-1,-1}},  {{+1,0}},  {{0,+1}},  1,  1},
535*c8dee2aaSAndroid Build Coastguard Worker         {{{0,1,0,0}},  {{+1,-1}},  {{-1,0}},  {{0,+1}},  1,  1},
536*c8dee2aaSAndroid Build Coastguard Worker 
537*c8dee2aaSAndroid Build Coastguard Worker         // Right inset edge.
538*c8dee2aaSAndroid Build Coastguard Worker         {{{0,1,0,0}},  {{+1,-1}},  {{0,+1}},  {{-1,0}},  1,  1},
539*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,1,0}},  {{+1,+1}},  {{0,-1}},  {{-1,0}},  1,  1},
540*c8dee2aaSAndroid Build Coastguard Worker 
541*c8dee2aaSAndroid Build Coastguard Worker         // Bottom inset edge.
542*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,1,0}},  {{+1,+1}},  {{-1,0}},  {{0,-1}},  1,  1},
543*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,0,1}},  {{-1,+1}},  {{+1,0}},  {{0,-1}},  1,  1},
544*c8dee2aaSAndroid Build Coastguard Worker 
545*c8dee2aaSAndroid Build Coastguard Worker 
546*c8dee2aaSAndroid Build Coastguard Worker         // Left outset edge.
547*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,0,1}},  {{-1,+1}},  {{0,-1}},  {{-1,0}},  0,  1},
548*c8dee2aaSAndroid Build Coastguard Worker         {{{1,0,0,0}},  {{-1,-1}},  {{0,+1}},  {{-1,0}},  0,  1},
549*c8dee2aaSAndroid Build Coastguard Worker 
550*c8dee2aaSAndroid Build Coastguard Worker         // Top outset edge.
551*c8dee2aaSAndroid Build Coastguard Worker         {{{1,0,0,0}},  {{-1,-1}},  {{+1,0}},  {{0,-1}},  0,  1},
552*c8dee2aaSAndroid Build Coastguard Worker         {{{0,1,0,0}},  {{+1,-1}},  {{-1,0}},  {{0,-1}},  0,  1},
553*c8dee2aaSAndroid Build Coastguard Worker 
554*c8dee2aaSAndroid Build Coastguard Worker         // Right outset edge.
555*c8dee2aaSAndroid Build Coastguard Worker         {{{0,1,0,0}},  {{+1,-1}},  {{0,+1}},  {{+1,0}},  0,  1},
556*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,1,0}},  {{+1,+1}},  {{0,-1}},  {{+1,0}},  0,  1},
557*c8dee2aaSAndroid Build Coastguard Worker 
558*c8dee2aaSAndroid Build Coastguard Worker         // Bottom outset edge.
559*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,1,0}},  {{+1,+1}},  {{-1,0}},  {{0,+1}},  0,  1},
560*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,0,1}},  {{-1,+1}},  {{+1,0}},  {{0,+1}},  0,  1},
561*c8dee2aaSAndroid Build Coastguard Worker 
562*c8dee2aaSAndroid Build Coastguard Worker 
563*c8dee2aaSAndroid Build Coastguard Worker         // Top-left corner.
564*c8dee2aaSAndroid Build Coastguard Worker         {{{1,0,0,0}},  {{-1,-1}},  {{ 0,+1}},  {{-1, 0}},  0,  0},
565*c8dee2aaSAndroid Build Coastguard Worker         {{{1,0,0,0}},  {{-1,-1}},  {{ 0,+1}},  {{+1, 0}},  1,  0},
566*c8dee2aaSAndroid Build Coastguard Worker         {{{1,0,0,0}},  {{-1,-1}},  {{+1, 0}},  {{ 0,+1}},  1,  0},
567*c8dee2aaSAndroid Build Coastguard Worker         {{{1,0,0,0}},  {{-1,-1}},  {{+1, 0}},  {{ 0,-1}},  0,  0},
568*c8dee2aaSAndroid Build Coastguard Worker         {{{1,0,0,0}},  {{-1,-1}},  {{+kOctoOffset,0}},  {{-1,-1}},  0,  0},
569*c8dee2aaSAndroid Build Coastguard Worker         {{{1,0,0,0}},  {{-1,-1}},  {{0,+kOctoOffset}},  {{-1,-1}},  0,  0},
570*c8dee2aaSAndroid Build Coastguard Worker 
571*c8dee2aaSAndroid Build Coastguard Worker         // Top-right corner.
572*c8dee2aaSAndroid Build Coastguard Worker         {{{0,1,0,0}},  {{+1,-1}},  {{-1, 0}},  {{ 0,-1}},  0,  0},
573*c8dee2aaSAndroid Build Coastguard Worker         {{{0,1,0,0}},  {{+1,-1}},  {{-1, 0}},  {{ 0,+1}},  1,  0},
574*c8dee2aaSAndroid Build Coastguard Worker         {{{0,1,0,0}},  {{+1,-1}},  {{ 0,+1}},  {{-1, 0}},  1,  0},
575*c8dee2aaSAndroid Build Coastguard Worker         {{{0,1,0,0}},  {{+1,-1}},  {{ 0,+1}},  {{+1, 0}},  0,  0},
576*c8dee2aaSAndroid Build Coastguard Worker         {{{0,1,0,0}},  {{+1,-1}},  {{0,+kOctoOffset}},  {{+1,-1}},  0,  0},
577*c8dee2aaSAndroid Build Coastguard Worker         {{{0,1,0,0}},  {{+1,-1}},  {{-kOctoOffset,0}},  {{+1,-1}},  0,  0},
578*c8dee2aaSAndroid Build Coastguard Worker 
579*c8dee2aaSAndroid Build Coastguard Worker         // Bottom-right corner.
580*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,1,0}},  {{+1,+1}},  {{ 0,-1}},  {{+1, 0}},  0,  0},
581*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,1,0}},  {{+1,+1}},  {{ 0,-1}},  {{-1, 0}},  1,  0},
582*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,1,0}},  {{+1,+1}},  {{-1, 0}},  {{ 0,-1}},  1,  0},
583*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,1,0}},  {{+1,+1}},  {{-1, 0}},  {{ 0,+1}},  0,  0},
584*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,1,0}},  {{+1,+1}},  {{-kOctoOffset,0}},  {{+1,+1}},  0,  0},
585*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,1,0}},  {{+1,+1}},  {{0,-kOctoOffset}},  {{+1,+1}},  0,  0},
586*c8dee2aaSAndroid Build Coastguard Worker 
587*c8dee2aaSAndroid Build Coastguard Worker         // Bottom-left corner.
588*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,0,1}},  {{-1,+1}},  {{+1, 0}},  {{ 0,+1}},  0,  0},
589*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,0,1}},  {{-1,+1}},  {{+1, 0}},  {{ 0,-1}},  1,  0},
590*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,0,1}},  {{-1,+1}},  {{ 0,-1}},  {{+1, 0}},  1,  0},
591*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,0,1}},  {{-1,+1}},  {{ 0,-1}},  {{-1, 0}},  0,  0},
592*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,0,1}},  {{-1,+1}},  {{0,-kOctoOffset}},  {{-1,+1}},  0,  0},
593*c8dee2aaSAndroid Build Coastguard Worker         {{{0,0,0,1}},  {{-1,+1}},  {{+kOctoOffset,0}},  {{-1,+1}},  0,  0}};
594*c8dee2aaSAndroid Build Coastguard Worker 
595*c8dee2aaSAndroid Build Coastguard Worker SKGPU_DECLARE_STATIC_UNIQUE_KEY(gVertexBufferKey);
596*c8dee2aaSAndroid Build Coastguard Worker 
597*c8dee2aaSAndroid Build Coastguard Worker static constexpr uint16_t kIndexData[] = {
598*c8dee2aaSAndroid Build Coastguard Worker         // Inset octagon (solid coverage).
599*c8dee2aaSAndroid Build Coastguard Worker         0, 1, 7,
600*c8dee2aaSAndroid Build Coastguard Worker         1, 2, 7,
601*c8dee2aaSAndroid Build Coastguard Worker         7, 2, 6,
602*c8dee2aaSAndroid Build Coastguard Worker         2, 3, 6,
603*c8dee2aaSAndroid Build Coastguard Worker         6, 3, 5,
604*c8dee2aaSAndroid Build Coastguard Worker         3, 4, 5,
605*c8dee2aaSAndroid Build Coastguard Worker 
606*c8dee2aaSAndroid Build Coastguard Worker         // AA borders (linear coverage).
607*c8dee2aaSAndroid Build Coastguard Worker         0, 1, 8, 1, 9, 8,
608*c8dee2aaSAndroid Build Coastguard Worker         2, 3, 10, 3, 11, 10,
609*c8dee2aaSAndroid Build Coastguard Worker         4, 5, 12, 5, 13, 12,
610*c8dee2aaSAndroid Build Coastguard Worker         6, 7, 14, 7, 15, 14,
611*c8dee2aaSAndroid Build Coastguard Worker 
612*c8dee2aaSAndroid Build Coastguard Worker         // Top-left arc.
613*c8dee2aaSAndroid Build Coastguard Worker         16, 17, 21,
614*c8dee2aaSAndroid Build Coastguard Worker         17, 21, 18,
615*c8dee2aaSAndroid Build Coastguard Worker         21, 18, 20,
616*c8dee2aaSAndroid Build Coastguard Worker         18, 20, 19,
617*c8dee2aaSAndroid Build Coastguard Worker 
618*c8dee2aaSAndroid Build Coastguard Worker         // Top-right arc.
619*c8dee2aaSAndroid Build Coastguard Worker         22, 23, 27,
620*c8dee2aaSAndroid Build Coastguard Worker         23, 27, 24,
621*c8dee2aaSAndroid Build Coastguard Worker         27, 24, 26,
622*c8dee2aaSAndroid Build Coastguard Worker         24, 26, 25,
623*c8dee2aaSAndroid Build Coastguard Worker 
624*c8dee2aaSAndroid Build Coastguard Worker         // Bottom-right arc.
625*c8dee2aaSAndroid Build Coastguard Worker         28, 29, 33,
626*c8dee2aaSAndroid Build Coastguard Worker         29, 33, 30,
627*c8dee2aaSAndroid Build Coastguard Worker         33, 30, 32,
628*c8dee2aaSAndroid Build Coastguard Worker         30, 32, 31,
629*c8dee2aaSAndroid Build Coastguard Worker 
630*c8dee2aaSAndroid Build Coastguard Worker         // Bottom-left arc.
631*c8dee2aaSAndroid Build Coastguard Worker         34, 35, 39,
632*c8dee2aaSAndroid Build Coastguard Worker         35, 39, 36,
633*c8dee2aaSAndroid Build Coastguard Worker         39, 36, 38,
634*c8dee2aaSAndroid Build Coastguard Worker         36, 38, 37};
635*c8dee2aaSAndroid Build Coastguard Worker 
636*c8dee2aaSAndroid Build Coastguard Worker SKGPU_DECLARE_STATIC_UNIQUE_KEY(gIndexBufferKey);
637*c8dee2aaSAndroid Build Coastguard Worker 
onPrepareDraws(GrMeshDrawTarget * target)638*c8dee2aaSAndroid Build Coastguard Worker void FillRRectOpImpl::onPrepareDraws(GrMeshDrawTarget* target) {
639*c8dee2aaSAndroid Build Coastguard Worker     if (!fProgramInfo) {
640*c8dee2aaSAndroid Build Coastguard Worker         this->createProgramInfo(target);
641*c8dee2aaSAndroid Build Coastguard Worker     }
642*c8dee2aaSAndroid Build Coastguard Worker 
643*c8dee2aaSAndroid Build Coastguard Worker     size_t instanceStride = fProgramInfo->geomProc().instanceStride();
644*c8dee2aaSAndroid Build Coastguard Worker 
645*c8dee2aaSAndroid Build Coastguard Worker     if (VertexWriter instanceWriter = target->makeVertexWriter(instanceStride, fInstanceCount,
646*c8dee2aaSAndroid Build Coastguard Worker                                                                &fInstanceBuffer, &fBaseInstance)) {
647*c8dee2aaSAndroid Build Coastguard Worker         SkDEBUGCODE(auto end = instanceWriter.mark(instanceStride * fInstanceCount));
648*c8dee2aaSAndroid Build Coastguard Worker         for (Instance* i = fHeadInstance; i; i = i->fNext) {
649*c8dee2aaSAndroid Build Coastguard Worker             auto [l, t, r, b] = i->fRRect.rect();
650*c8dee2aaSAndroid Build Coastguard Worker 
651*c8dee2aaSAndroid Build Coastguard Worker             // Produce a matrix that draws the round rect from normalized [-1, -1, +1, +1] space.
652*c8dee2aaSAndroid Build Coastguard Worker             SkMatrix m;
653*c8dee2aaSAndroid Build Coastguard Worker             // Unmap the normalized rect [-1, -1, +1, +1] back to [l, t, r, b].
654*c8dee2aaSAndroid Build Coastguard Worker             m.setScaleTranslate((r - l)/2, (b - t)/2, (l + r)/2, (t + b)/2);
655*c8dee2aaSAndroid Build Coastguard Worker             // Map to device space.
656*c8dee2aaSAndroid Build Coastguard Worker             m.postConcat(i->fViewMatrix);
657*c8dee2aaSAndroid Build Coastguard Worker 
658*c8dee2aaSAndroid Build Coastguard Worker             // Convert the radii to [-1, -1, +1, +1] space and write their attribs.
659*c8dee2aaSAndroid Build Coastguard Worker             skvx::float4 radiiX, radiiY;
660*c8dee2aaSAndroid Build Coastguard Worker             skvx::strided_load2(&SkRRectPriv::GetRadiiArray(i->fRRect)->fX, radiiX, radiiY);
661*c8dee2aaSAndroid Build Coastguard Worker             radiiX *= 2 / (r - l);
662*c8dee2aaSAndroid Build Coastguard Worker             radiiY *= 2 / (b - t);
663*c8dee2aaSAndroid Build Coastguard Worker 
664*c8dee2aaSAndroid Build Coastguard Worker             instanceWriter << radiiX << radiiY
665*c8dee2aaSAndroid Build Coastguard Worker                            << m.getScaleX() << m.getSkewX() << m.getSkewY() << m.getScaleY()
666*c8dee2aaSAndroid Build Coastguard Worker                            << m.getTranslateX() << m.getTranslateY();
667*c8dee2aaSAndroid Build Coastguard Worker 
668*c8dee2aaSAndroid Build Coastguard Worker             if (fProcessorFlags & ProcessorFlags::kHasLocalCoords) {
669*c8dee2aaSAndroid Build Coastguard Worker                 if (i->fLocalCoords.fType == LocalCoords::Type::kRect) {
670*c8dee2aaSAndroid Build Coastguard Worker                     instanceWriter << 0.f << 0.f  // localrotate
671*c8dee2aaSAndroid Build Coastguard Worker                                    << i->fLocalCoords.fRect;  // localrect
672*c8dee2aaSAndroid Build Coastguard Worker                 } else {
673*c8dee2aaSAndroid Build Coastguard Worker                     SkASSERT(i->fLocalCoords.fType == LocalCoords::Type::kMatrix);
674*c8dee2aaSAndroid Build Coastguard Worker                     const SkRect& bounds = i->fRRect.rect();
675*c8dee2aaSAndroid Build Coastguard Worker                     const SkMatrix& localMatrix = i->fLocalCoords.fMatrix;
676*c8dee2aaSAndroid Build Coastguard Worker                     SkVector u = localMatrix.mapVector(bounds.right() - bounds.left(), 0);
677*c8dee2aaSAndroid Build Coastguard Worker                     SkVector v = localMatrix.mapVector(0, bounds.bottom() - bounds.top());
678*c8dee2aaSAndroid Build Coastguard Worker                     SkPoint l0 = localMatrix.mapPoint({bounds.left(), bounds.top()});
679*c8dee2aaSAndroid Build Coastguard Worker                     instanceWriter << v.x() << u.y()  // localrotate
680*c8dee2aaSAndroid Build Coastguard Worker                                    << l0 << (l0.x() + u.x()) << (l0.y() + v.y());  // localrect
681*c8dee2aaSAndroid Build Coastguard Worker                 }
682*c8dee2aaSAndroid Build Coastguard Worker             }
683*c8dee2aaSAndroid Build Coastguard Worker 
684*c8dee2aaSAndroid Build Coastguard Worker             instanceWriter << VertexColor(i->fColor, fProcessorFlags & ProcessorFlags::kWideColor);
685*c8dee2aaSAndroid Build Coastguard Worker         }
686*c8dee2aaSAndroid Build Coastguard Worker         SkASSERT(instanceWriter.mark() == end);
687*c8dee2aaSAndroid Build Coastguard Worker     }
688*c8dee2aaSAndroid Build Coastguard Worker 
689*c8dee2aaSAndroid Build Coastguard Worker     SKGPU_DEFINE_STATIC_UNIQUE_KEY(gIndexBufferKey);
690*c8dee2aaSAndroid Build Coastguard Worker 
691*c8dee2aaSAndroid Build Coastguard Worker     fIndexBuffer = target->resourceProvider()->findOrMakeStaticBuffer(GrGpuBufferType::kIndex,
692*c8dee2aaSAndroid Build Coastguard Worker                                                                       sizeof(kIndexData),
693*c8dee2aaSAndroid Build Coastguard Worker                                                                       kIndexData, gIndexBufferKey);
694*c8dee2aaSAndroid Build Coastguard Worker 
695*c8dee2aaSAndroid Build Coastguard Worker     SKGPU_DEFINE_STATIC_UNIQUE_KEY(gVertexBufferKey);
696*c8dee2aaSAndroid Build Coastguard Worker 
697*c8dee2aaSAndroid Build Coastguard Worker     fVertexBuffer = target->resourceProvider()->findOrMakeStaticBuffer(GrGpuBufferType::kVertex,
698*c8dee2aaSAndroid Build Coastguard Worker                                                                       sizeof(kVertexData),
699*c8dee2aaSAndroid Build Coastguard Worker                                                                       kVertexData,
700*c8dee2aaSAndroid Build Coastguard Worker                                                                       gVertexBufferKey);
701*c8dee2aaSAndroid Build Coastguard Worker }
702*c8dee2aaSAndroid Build Coastguard Worker 
703*c8dee2aaSAndroid Build Coastguard Worker class FillRRectOpImpl::Processor::Impl : public ProgramImpl {
704*c8dee2aaSAndroid Build Coastguard Worker public:
setData(const GrGLSLProgramDataManager &,const GrShaderCaps &,const GrGeometryProcessor &)705*c8dee2aaSAndroid Build Coastguard Worker     void setData(const GrGLSLProgramDataManager&,
706*c8dee2aaSAndroid Build Coastguard Worker                  const GrShaderCaps&,
707*c8dee2aaSAndroid Build Coastguard Worker                  const GrGeometryProcessor&) override {}
708*c8dee2aaSAndroid Build Coastguard Worker 
709*c8dee2aaSAndroid Build Coastguard Worker private:
onEmitCode(EmitArgs & args,GrGPArgs * gpArgs)710*c8dee2aaSAndroid Build Coastguard Worker     void onEmitCode(EmitArgs& args, GrGPArgs* gpArgs) override {
711*c8dee2aaSAndroid Build Coastguard Worker         GrGLSLVertexBuilder* v = args.fVertBuilder;
712*c8dee2aaSAndroid Build Coastguard Worker         GrGLSLFPFragmentBuilder* f = args.fFragBuilder;
713*c8dee2aaSAndroid Build Coastguard Worker 
714*c8dee2aaSAndroid Build Coastguard Worker         const auto& proc = args.fGeomProc.cast<Processor>();
715*c8dee2aaSAndroid Build Coastguard Worker         bool useHWDerivatives = (proc.fFlags & ProcessorFlags::kUseHWDerivatives);
716*c8dee2aaSAndroid Build Coastguard Worker 
717*c8dee2aaSAndroid Build Coastguard Worker         SkASSERT(proc.vertexStride() == sizeof(CoverageVertex));
718*c8dee2aaSAndroid Build Coastguard Worker 
719*c8dee2aaSAndroid Build Coastguard Worker         GrGLSLVaryingHandler* varyings = args.fVaryingHandler;
720*c8dee2aaSAndroid Build Coastguard Worker         varyings->emitAttributes(proc);
721*c8dee2aaSAndroid Build Coastguard Worker         f->codeAppendf("half4 %s;", args.fOutputColor);
722*c8dee2aaSAndroid Build Coastguard Worker         varyings->addPassThroughAttribute(proc.fColorAttrib->asShaderVar(),
723*c8dee2aaSAndroid Build Coastguard Worker                                           args.fOutputColor,
724*c8dee2aaSAndroid Build Coastguard Worker                                           GrGLSLVaryingHandler::Interpolation::kCanBeFlat);
725*c8dee2aaSAndroid Build Coastguard Worker 
726*c8dee2aaSAndroid Build Coastguard Worker         // Emit the vertex shader.
727*c8dee2aaSAndroid Build Coastguard Worker         // When MSAA is enabled, we need to make sure every sample gets lit up on pixels that have
728*c8dee2aaSAndroid Build Coastguard Worker         // fractional coverage. We do this by making the ramp wider.
729*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppendf("float aa_bloat_multiplier = %i;",
730*c8dee2aaSAndroid Build Coastguard Worker                        (proc.fFlags & ProcessorFlags::kMSAAEnabled)
731*c8dee2aaSAndroid Build Coastguard Worker                                ? 2    // Outset an entire pixel (2 radii).
732*c8dee2aaSAndroid Build Coastguard Worker                        : (!(proc.fFlags & ProcessorFlags::kFakeNonAA))
733*c8dee2aaSAndroid Build Coastguard Worker                                ? 1    // Outset one half pixel (1 radius).
734*c8dee2aaSAndroid Build Coastguard Worker                                : 0);  // No AA bloat.
735*c8dee2aaSAndroid Build Coastguard Worker 
736*c8dee2aaSAndroid Build Coastguard Worker         // Unpack vertex attribs.
737*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float2 corner = corner_and_radius_outsets.xy;");
738*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float2 radius_outset = corner_and_radius_outsets.zw;");
739*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float2 aa_bloat_direction = aa_bloat_and_coverage.xy;");
740*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float is_linear_coverage = aa_bloat_and_coverage.w;");
741*c8dee2aaSAndroid Build Coastguard Worker 
742*c8dee2aaSAndroid Build Coastguard Worker         // Find the amount to bloat each edge for AA (in source space).
743*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float2 pixellength = inversesqrt("
744*c8dee2aaSAndroid Build Coastguard Worker                               "float2(dot(skew.xz, skew.xz), dot(skew.yw, skew.yw)));");
745*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float4 normalized_axis_dirs = skew * pixellength.xyxy;");
746*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float2 axiswidths = (abs(normalized_axis_dirs.xy) + "
747*c8dee2aaSAndroid Build Coastguard Worker                                            "abs(normalized_axis_dirs.zw));");
748*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float2 aa_bloatradius = axiswidths * pixellength * .5;");
749*c8dee2aaSAndroid Build Coastguard Worker 
750*c8dee2aaSAndroid Build Coastguard Worker         // Identify our radii.
751*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float4 radii_and_neighbors = radii_selector"
752*c8dee2aaSAndroid Build Coastguard Worker                               "* float4x4(radii_x, radii_y, radii_x.yxwz, radii_y.wzyx);");
753*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float2 radii = radii_and_neighbors.xy;");
754*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float2 neighbor_radii = radii_and_neighbors.zw;");
755*c8dee2aaSAndroid Build Coastguard Worker 
756*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float coverage_multiplier = 1;");
757*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("if (any(greaterThan(aa_bloatradius, float2(1)))) {");
758*c8dee2aaSAndroid Build Coastguard Worker                           // The rrect is more narrow than a half-pixel AA coverage ramp. We can't
759*c8dee2aaSAndroid Build Coastguard Worker                           // draw as-is or else opposite AA borders will overlap. Instead, fudge the
760*c8dee2aaSAndroid Build Coastguard Worker                           // size up to the width of a coverage ramp, and then reduce total coverage
761*c8dee2aaSAndroid Build Coastguard Worker                           // to make the rect appear more thin.
762*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "corner = max(abs(corner), aa_bloatradius) * sign(corner);");
763*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "coverage_multiplier = 1 / (max(aa_bloatradius.x, 1) * "
764*c8dee2aaSAndroid Build Coastguard Worker                                                      "max(aa_bloatradius.y, 1));");
765*c8dee2aaSAndroid Build Coastguard Worker                           // Set radii to zero to ensure we take the "linear coverage" codepath.
766*c8dee2aaSAndroid Build Coastguard Worker                           // (The "coverage" variable only has effect in the linear codepath.)
767*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "radii = float2(0);");
768*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("}");
769*c8dee2aaSAndroid Build Coastguard Worker 
770*c8dee2aaSAndroid Build Coastguard Worker         // Unpack coverage.
771*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float coverage = aa_bloat_and_coverage.z;");
772*c8dee2aaSAndroid Build Coastguard Worker         if (proc.fFlags & ProcessorFlags::kMSAAEnabled) {
773*c8dee2aaSAndroid Build Coastguard Worker             // MSAA has a wider ramp that goes from -.5 to 1.5 instead of 0 to 1.
774*c8dee2aaSAndroid Build Coastguard Worker             v->codeAppendf("coverage = (coverage - .5) * aa_bloat_multiplier + .5;");
775*c8dee2aaSAndroid Build Coastguard Worker         }
776*c8dee2aaSAndroid Build Coastguard Worker 
777*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("if (any(lessThan(radii, aa_bloatradius * 1.5))) {");
778*c8dee2aaSAndroid Build Coastguard Worker                           // The radii are very small. Demote this arc to a sharp 90 degree corner.
779*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "radii = float2(0);");
780*c8dee2aaSAndroid Build Coastguard Worker                           // Convert to a standard picture frame for an AA rect instead of the round
781*c8dee2aaSAndroid Build Coastguard Worker                           // rect geometry.
782*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "aa_bloat_direction = sign(corner);");
783*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "if (coverage > .5) {");  // Are we an inset edge?
784*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(        "aa_bloat_direction = -aa_bloat_direction;");
785*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "}");
786*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "is_linear_coverage = 1;");
787*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("} else {");
788*c8dee2aaSAndroid Build Coastguard Worker                           // Don't let radii get smaller than a coverage ramp plus an extra half
789*c8dee2aaSAndroid Build Coastguard Worker                           // pixel for MSAA. Always use the same amount so we don't pop when
790*c8dee2aaSAndroid Build Coastguard Worker                           // switching between MSAA and coverage.
791*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "radii = clamp(radii, pixellength * 1.5, 2 - pixellength * 1.5);");
792*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "neighbor_radii = clamp(neighbor_radii, pixellength * 1.5, "
793*c8dee2aaSAndroid Build Coastguard Worker                                                  "2 - pixellength * 1.5);");
794*c8dee2aaSAndroid Build Coastguard Worker                           // Don't let neighboring radii get closer together than 1/16 pixel.
795*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "float2 spacing = 2 - radii - neighbor_radii;");
796*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "float2 extra_pad = max(pixellength * .0625 - spacing, float2(0));");
797*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "radii -= extra_pad * .5;");
798*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("}");
799*c8dee2aaSAndroid Build Coastguard Worker 
800*c8dee2aaSAndroid Build Coastguard Worker         // Find our vertex position, adjusted for radii and bloated for AA. Our rect is drawn in
801*c8dee2aaSAndroid Build Coastguard Worker         // normalized [-1,-1,+1,+1] space.
802*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float2 aa_outset = "
803*c8dee2aaSAndroid Build Coastguard Worker                               "aa_bloat_direction * aa_bloatradius * aa_bloat_multiplier;");
804*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float2 vertexpos = corner + radius_outset * radii + aa_outset;");
805*c8dee2aaSAndroid Build Coastguard Worker 
806*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("if (coverage > .5) {");  // Are we an inset edge?
807*c8dee2aaSAndroid Build Coastguard Worker                           // Don't allow the aa insets to overlap. i.e., Don't let them inset past
808*c8dee2aaSAndroid Build Coastguard Worker                           // the center (x=y=0). Since we don't allow the rect to become thinner
809*c8dee2aaSAndroid Build Coastguard Worker                           // than 1px, this should only happen when using MSAA, where we inset by an
810*c8dee2aaSAndroid Build Coastguard Worker                           // entire pixel instead of half.
811*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "if (aa_bloat_direction.x != 0 && vertexpos.x * corner.x < 0) {");
812*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(        "float backset = abs(vertexpos.x);");
813*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(        "vertexpos.x = 0;");
814*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(        "vertexpos.y += "
815*c8dee2aaSAndroid Build Coastguard Worker                                       "backset * sign(corner.y) * pixellength.y/pixellength.x;");
816*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(        "coverage = (coverage - .5) * abs(corner.x) / "
817*c8dee2aaSAndroid Build Coastguard Worker                                       "(abs(corner.x) + backset) + .5;");
818*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "}");
819*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "if (aa_bloat_direction.y != 0 && vertexpos.y * corner.y < 0) {");
820*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(        "float backset = abs(vertexpos.y);");
821*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(        "vertexpos.y = 0;");
822*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(        "vertexpos.x += "
823*c8dee2aaSAndroid Build Coastguard Worker                                       "backset * sign(corner.x) * pixellength.x/pixellength.y;");
824*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(        "coverage = (coverage - .5) * abs(corner.y) / "
825*c8dee2aaSAndroid Build Coastguard Worker                                       "(abs(corner.y) + backset) + .5;");
826*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "}");
827*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("}");
828*c8dee2aaSAndroid Build Coastguard Worker 
829*c8dee2aaSAndroid Build Coastguard Worker         // Transform to device space.
830*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float2x2 skewmatrix = float2x2(skew.xy, skew.zw);");
831*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("float2 devcoord = vertexpos * skewmatrix + translate_and_localrotate.xy;");
832*c8dee2aaSAndroid Build Coastguard Worker         gpArgs->fPositionVar.set(SkSLType::kFloat2, "devcoord");
833*c8dee2aaSAndroid Build Coastguard Worker 
834*c8dee2aaSAndroid Build Coastguard Worker         // Output local coordinates.
835*c8dee2aaSAndroid Build Coastguard Worker         if (proc.fFlags & ProcessorFlags::kHasLocalCoords) {
836*c8dee2aaSAndroid Build Coastguard Worker             // Do math in a way that preserves exact local coord boundaries when there is no local
837*c8dee2aaSAndroid Build Coastguard Worker             // rotate and vertexpos is on an exact shape boundary.
838*c8dee2aaSAndroid Build Coastguard Worker             v->codeAppend("float2 T = vertexpos * .5 + .5;");
839*c8dee2aaSAndroid Build Coastguard Worker             v->codeAppend("float2 localcoord = localrect.xy * (1 - T) + "
840*c8dee2aaSAndroid Build Coastguard Worker                                                "localrect.zw * T + "
841*c8dee2aaSAndroid Build Coastguard Worker                                                "translate_and_localrotate.zw * T.yx;");
842*c8dee2aaSAndroid Build Coastguard Worker             gpArgs->fLocalCoordVar.set(SkSLType::kFloat2, "localcoord");
843*c8dee2aaSAndroid Build Coastguard Worker         }
844*c8dee2aaSAndroid Build Coastguard Worker 
845*c8dee2aaSAndroid Build Coastguard Worker         // Setup interpolants for coverage.
846*c8dee2aaSAndroid Build Coastguard Worker         GrGLSLVarying arcCoord(useHWDerivatives ? SkSLType::kFloat2 : SkSLType::kFloat4);
847*c8dee2aaSAndroid Build Coastguard Worker         varyings->addVarying("arccoord", &arcCoord);
848*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("if (0 != is_linear_coverage) {");
849*c8dee2aaSAndroid Build Coastguard Worker                            // We are a non-corner piece: Set x=0 to indicate built-in coverage, and
850*c8dee2aaSAndroid Build Coastguard Worker                            // interpolate linear coverage across y.
851*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppendf(    "%s.xy = float2(0, coverage * coverage_multiplier);",
852*c8dee2aaSAndroid Build Coastguard Worker                            arcCoord.vsOut());
853*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("} else {");
854*c8dee2aaSAndroid Build Coastguard Worker                            // Find the normalized arc coordinates for our corner ellipse.
855*c8dee2aaSAndroid Build Coastguard Worker                            // (i.e., the coordinate system where x^2 + y^2 == 1).
856*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend(    "float2 arccoord = 1 - abs(radius_outset) + aa_outset/radii * corner;");
857*c8dee2aaSAndroid Build Coastguard Worker                            // We are a corner piece: Interpolate the arc coordinates for coverage.
858*c8dee2aaSAndroid Build Coastguard Worker                            // Emit x+1 to ensure no pixel in the arc has a x value of 0 (since x=0
859*c8dee2aaSAndroid Build Coastguard Worker                            // instructs the fragment shader to use linear coverage).
860*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppendf(    "%s.xy = float2(arccoord.x+1, arccoord.y);", arcCoord.vsOut());
861*c8dee2aaSAndroid Build Coastguard Worker         if (!useHWDerivatives) {
862*c8dee2aaSAndroid Build Coastguard Worker             // The gradient is order-1: Interpolate it across arccoord.zw.
863*c8dee2aaSAndroid Build Coastguard Worker             v->codeAppendf("float2x2 derivatives = inverse(skewmatrix);");
864*c8dee2aaSAndroid Build Coastguard Worker             v->codeAppendf("%s.zw = derivatives * (arccoord/radii * 2);", arcCoord.vsOut());
865*c8dee2aaSAndroid Build Coastguard Worker         }
866*c8dee2aaSAndroid Build Coastguard Worker         v->codeAppend("}");
867*c8dee2aaSAndroid Build Coastguard Worker 
868*c8dee2aaSAndroid Build Coastguard Worker         // Emit the fragment shader.
869*c8dee2aaSAndroid Build Coastguard Worker         f->codeAppendf("float x_plus_1=%s.x, y=%s.y;", arcCoord.fsIn(), arcCoord.fsIn());
870*c8dee2aaSAndroid Build Coastguard Worker         f->codeAppendf("half coverage;");
871*c8dee2aaSAndroid Build Coastguard Worker         f->codeAppendf("if (0 == x_plus_1) {");
872*c8dee2aaSAndroid Build Coastguard Worker         f->codeAppendf(    "coverage = half(y);");  // We are a non-arc pixel (linear coverage).
873*c8dee2aaSAndroid Build Coastguard Worker         f->codeAppendf("} else {");
874*c8dee2aaSAndroid Build Coastguard Worker         f->codeAppendf(    "float fn = x_plus_1 * (x_plus_1 - 2);");  // fn = (x+1)*(x-1) = x^2-1
875*c8dee2aaSAndroid Build Coastguard Worker         f->codeAppendf(    "fn = fma(y,y, fn);");  // fn = x^2 + y^2 - 1
876*c8dee2aaSAndroid Build Coastguard Worker         if (useHWDerivatives) {
877*c8dee2aaSAndroid Build Coastguard Worker             f->codeAppendf("float fnwidth = fwidth(fn);");
878*c8dee2aaSAndroid Build Coastguard Worker         } else {
879*c8dee2aaSAndroid Build Coastguard Worker             // The gradient is interpolated across arccoord.zw.
880*c8dee2aaSAndroid Build Coastguard Worker             f->codeAppendf("float gx=%s.z, gy=%s.w;", arcCoord.fsIn(), arcCoord.fsIn());
881*c8dee2aaSAndroid Build Coastguard Worker             f->codeAppendf("float fnwidth = abs(gx) + abs(gy);");
882*c8dee2aaSAndroid Build Coastguard Worker         }
883*c8dee2aaSAndroid Build Coastguard Worker         f->codeAppendf(    "coverage = .5 - half(fn/fnwidth);");
884*c8dee2aaSAndroid Build Coastguard Worker         if (proc.fFlags & ProcessorFlags::kMSAAEnabled) {
885*c8dee2aaSAndroid Build Coastguard Worker             // MSAA uses ramps larger than 1px, so we need to clamp in both branches.
886*c8dee2aaSAndroid Build Coastguard Worker             f->codeAppendf("}");
887*c8dee2aaSAndroid Build Coastguard Worker         }
888*c8dee2aaSAndroid Build Coastguard Worker         f->codeAppendf("coverage = clamp(coverage, 0, 1);");
889*c8dee2aaSAndroid Build Coastguard Worker         if (!(proc.fFlags & ProcessorFlags::kMSAAEnabled)) {
890*c8dee2aaSAndroid Build Coastguard Worker             // When not using MSAA, we only need to clamp in the "arc" branch.
891*c8dee2aaSAndroid Build Coastguard Worker             f->codeAppendf("}");
892*c8dee2aaSAndroid Build Coastguard Worker         }
893*c8dee2aaSAndroid Build Coastguard Worker         if (proc.fFlags & ProcessorFlags::kFakeNonAA) {
894*c8dee2aaSAndroid Build Coastguard Worker             f->codeAppendf("coverage = (coverage >= .5) ? 1 : 0;");
895*c8dee2aaSAndroid Build Coastguard Worker         }
896*c8dee2aaSAndroid Build Coastguard Worker         f->codeAppendf("half4 %s = half4(coverage);", args.fOutputCoverage);
897*c8dee2aaSAndroid Build Coastguard Worker     }
898*c8dee2aaSAndroid Build Coastguard Worker };
899*c8dee2aaSAndroid Build Coastguard Worker 
makeProgramImpl(const GrShaderCaps &) const900*c8dee2aaSAndroid Build Coastguard Worker std::unique_ptr<GrGeometryProcessor::ProgramImpl> FillRRectOpImpl::Processor::makeProgramImpl(
901*c8dee2aaSAndroid Build Coastguard Worker         const GrShaderCaps&) const {
902*c8dee2aaSAndroid Build Coastguard Worker     return std::make_unique<Impl>();
903*c8dee2aaSAndroid Build Coastguard Worker }
904*c8dee2aaSAndroid Build Coastguard Worker 
onCreateProgramInfo(const GrCaps * caps,SkArenaAlloc * arena,const GrSurfaceProxyView & writeView,bool usesMSAASurface,GrAppliedClip && appliedClip,const GrDstProxyView & dstProxyView,GrXferBarrierFlags renderPassXferBarriers,GrLoadOp colorLoadOp)905*c8dee2aaSAndroid Build Coastguard Worker void FillRRectOpImpl::onCreateProgramInfo(const GrCaps* caps,
906*c8dee2aaSAndroid Build Coastguard Worker                                           SkArenaAlloc* arena,
907*c8dee2aaSAndroid Build Coastguard Worker                                           const GrSurfaceProxyView& writeView,
908*c8dee2aaSAndroid Build Coastguard Worker                                           bool usesMSAASurface,
909*c8dee2aaSAndroid Build Coastguard Worker                                           GrAppliedClip&& appliedClip,
910*c8dee2aaSAndroid Build Coastguard Worker                                           const GrDstProxyView& dstProxyView,
911*c8dee2aaSAndroid Build Coastguard Worker                                           GrXferBarrierFlags renderPassXferBarriers,
912*c8dee2aaSAndroid Build Coastguard Worker                                           GrLoadOp colorLoadOp) {
913*c8dee2aaSAndroid Build Coastguard Worker     if (usesMSAASurface) {
914*c8dee2aaSAndroid Build Coastguard Worker         fProcessorFlags |= ProcessorFlags::kMSAAEnabled;
915*c8dee2aaSAndroid Build Coastguard Worker     }
916*c8dee2aaSAndroid Build Coastguard Worker     GrGeometryProcessor* gp = Processor::Make(arena, fHelper.aaType(), fProcessorFlags);
917*c8dee2aaSAndroid Build Coastguard Worker     fProgramInfo = fHelper.createProgramInfo(caps, arena, writeView, usesMSAASurface,
918*c8dee2aaSAndroid Build Coastguard Worker                                              std::move(appliedClip), dstProxyView, gp,
919*c8dee2aaSAndroid Build Coastguard Worker                                              GrPrimitiveType::kTriangles, renderPassXferBarriers,
920*c8dee2aaSAndroid Build Coastguard Worker                                              colorLoadOp);
921*c8dee2aaSAndroid Build Coastguard Worker }
922*c8dee2aaSAndroid Build Coastguard Worker 
onExecute(GrOpFlushState * flushState,const SkRect & chainBounds)923*c8dee2aaSAndroid Build Coastguard Worker void FillRRectOpImpl::onExecute(GrOpFlushState* flushState, const SkRect& chainBounds) {
924*c8dee2aaSAndroid Build Coastguard Worker     if (!fInstanceBuffer || !fIndexBuffer || !fVertexBuffer) {
925*c8dee2aaSAndroid Build Coastguard Worker         return;  // Setup failed.
926*c8dee2aaSAndroid Build Coastguard Worker     }
927*c8dee2aaSAndroid Build Coastguard Worker 
928*c8dee2aaSAndroid Build Coastguard Worker     flushState->bindPipelineAndScissorClip(*fProgramInfo, this->bounds());
929*c8dee2aaSAndroid Build Coastguard Worker     flushState->bindTextures(fProgramInfo->geomProc(), nullptr, fProgramInfo->pipeline());
930*c8dee2aaSAndroid Build Coastguard Worker     flushState->bindBuffers(std::move(fIndexBuffer), std::move(fInstanceBuffer),
931*c8dee2aaSAndroid Build Coastguard Worker                             std::move(fVertexBuffer));
932*c8dee2aaSAndroid Build Coastguard Worker     flushState->drawIndexedInstanced(std::size(kIndexData), 0, fInstanceCount, fBaseInstance, 0);
933*c8dee2aaSAndroid Build Coastguard Worker }
934*c8dee2aaSAndroid Build Coastguard Worker 
935*c8dee2aaSAndroid Build Coastguard Worker // Will the given corner look good if we use HW derivatives?
can_use_hw_derivatives_with_coverage(const skvx::float2 & devScale,const skvx::float2 & cornerRadii)936*c8dee2aaSAndroid Build Coastguard Worker bool can_use_hw_derivatives_with_coverage(const skvx::float2& devScale,
937*c8dee2aaSAndroid Build Coastguard Worker                                           const skvx::float2& cornerRadii) {
938*c8dee2aaSAndroid Build Coastguard Worker     skvx::float2 devRadii = devScale * cornerRadii;
939*c8dee2aaSAndroid Build Coastguard Worker     if (devRadii[1] < devRadii[0]) {
940*c8dee2aaSAndroid Build Coastguard Worker         devRadii = skvx::shuffle<1,0>(devRadii);
941*c8dee2aaSAndroid Build Coastguard Worker     }
942*c8dee2aaSAndroid Build Coastguard Worker     float minDevRadius = std::max(devRadii[0], 1.f);  // Shader clamps radius at a minimum of 1.
943*c8dee2aaSAndroid Build Coastguard Worker     // Is the gradient smooth enough for this corner look ok if we use hardware derivatives?
944*c8dee2aaSAndroid Build Coastguard Worker     // This threshold was arrived at subjevtively on an NVIDIA chip.
945*c8dee2aaSAndroid Build Coastguard Worker     return minDevRadius * minDevRadius * 5 > devRadii[1];
946*c8dee2aaSAndroid Build Coastguard Worker }
947*c8dee2aaSAndroid Build Coastguard Worker 
can_use_hw_derivatives_with_coverage(const skvx::float2 & devScale,const SkVector & cornerRadii)948*c8dee2aaSAndroid Build Coastguard Worker bool can_use_hw_derivatives_with_coverage(const skvx::float2& devScale,
949*c8dee2aaSAndroid Build Coastguard Worker                                           const SkVector& cornerRadii) {
950*c8dee2aaSAndroid Build Coastguard Worker     return can_use_hw_derivatives_with_coverage(devScale, skvx::float2::Load(&cornerRadii));
951*c8dee2aaSAndroid Build Coastguard Worker }
952*c8dee2aaSAndroid Build Coastguard Worker 
953*c8dee2aaSAndroid Build Coastguard Worker // Will the given round rect look good if we use HW derivatives?
can_use_hw_derivatives_with_coverage(const GrShaderCaps & shaderCaps,const SkMatrix & viewMatrix,const SkRRect & rrect)954*c8dee2aaSAndroid Build Coastguard Worker bool can_use_hw_derivatives_with_coverage(const GrShaderCaps& shaderCaps,
955*c8dee2aaSAndroid Build Coastguard Worker                                           const SkMatrix& viewMatrix,
956*c8dee2aaSAndroid Build Coastguard Worker                                           const SkRRect& rrect) {
957*c8dee2aaSAndroid Build Coastguard Worker     if (!shaderCaps.fShaderDerivativeSupport) {
958*c8dee2aaSAndroid Build Coastguard Worker         return false;
959*c8dee2aaSAndroid Build Coastguard Worker     }
960*c8dee2aaSAndroid Build Coastguard Worker 
961*c8dee2aaSAndroid Build Coastguard Worker     auto x = skvx::float2(viewMatrix.getScaleX(), viewMatrix.getSkewX());
962*c8dee2aaSAndroid Build Coastguard Worker     auto y = skvx::float2(viewMatrix.getSkewY(), viewMatrix.getScaleY());
963*c8dee2aaSAndroid Build Coastguard Worker     skvx::float2 devScale = sqrt(x*x + y*y);
964*c8dee2aaSAndroid Build Coastguard Worker     switch (rrect.getType()) {
965*c8dee2aaSAndroid Build Coastguard Worker         case SkRRect::kEmpty_Type:
966*c8dee2aaSAndroid Build Coastguard Worker         case SkRRect::kRect_Type:
967*c8dee2aaSAndroid Build Coastguard Worker             return true;
968*c8dee2aaSAndroid Build Coastguard Worker 
969*c8dee2aaSAndroid Build Coastguard Worker         case SkRRect::kOval_Type:
970*c8dee2aaSAndroid Build Coastguard Worker         case SkRRect::kSimple_Type:
971*c8dee2aaSAndroid Build Coastguard Worker             return can_use_hw_derivatives_with_coverage(devScale, rrect.getSimpleRadii());
972*c8dee2aaSAndroid Build Coastguard Worker 
973*c8dee2aaSAndroid Build Coastguard Worker         case SkRRect::kNinePatch_Type: {
974*c8dee2aaSAndroid Build Coastguard Worker             skvx::float2 r0 = skvx::float2::Load(SkRRectPriv::GetRadiiArray(rrect));
975*c8dee2aaSAndroid Build Coastguard Worker             skvx::float2 r1 = skvx::float2::Load(SkRRectPriv::GetRadiiArray(rrect) + 2);
976*c8dee2aaSAndroid Build Coastguard Worker             skvx::float2 minRadii = min(r0, r1);
977*c8dee2aaSAndroid Build Coastguard Worker             skvx::float2 maxRadii = max(r0, r1);
978*c8dee2aaSAndroid Build Coastguard Worker             return can_use_hw_derivatives_with_coverage(devScale,
979*c8dee2aaSAndroid Build Coastguard Worker                                                         skvx::float2(minRadii[0], maxRadii[1])) &&
980*c8dee2aaSAndroid Build Coastguard Worker                    can_use_hw_derivatives_with_coverage(devScale,
981*c8dee2aaSAndroid Build Coastguard Worker                                                         skvx::float2(maxRadii[0], minRadii[1]));
982*c8dee2aaSAndroid Build Coastguard Worker         }
983*c8dee2aaSAndroid Build Coastguard Worker 
984*c8dee2aaSAndroid Build Coastguard Worker         case SkRRect::kComplex_Type: {
985*c8dee2aaSAndroid Build Coastguard Worker             for (int i = 0; i < 4; ++i) {
986*c8dee2aaSAndroid Build Coastguard Worker                 auto corner = static_cast<SkRRect::Corner>(i);
987*c8dee2aaSAndroid Build Coastguard Worker                 if (!can_use_hw_derivatives_with_coverage(devScale, rrect.radii(corner))) {
988*c8dee2aaSAndroid Build Coastguard Worker                     return false;
989*c8dee2aaSAndroid Build Coastguard Worker                 }
990*c8dee2aaSAndroid Build Coastguard Worker             }
991*c8dee2aaSAndroid Build Coastguard Worker             return true;
992*c8dee2aaSAndroid Build Coastguard Worker         }
993*c8dee2aaSAndroid Build Coastguard Worker     }
994*c8dee2aaSAndroid Build Coastguard Worker     SK_ABORT("Invalid round rect type.");
995*c8dee2aaSAndroid Build Coastguard Worker }
996*c8dee2aaSAndroid Build Coastguard Worker 
997*c8dee2aaSAndroid Build Coastguard Worker } // anonymous namespace
998*c8dee2aaSAndroid Build Coastguard Worker 
Make(GrRecordingContext * ctx,SkArenaAlloc * arena,GrPaint && paint,const SkMatrix & viewMatrix,const SkRRect & rrect,const SkRect & localRect,GrAA aa)999*c8dee2aaSAndroid Build Coastguard Worker GrOp::Owner Make(GrRecordingContext* ctx,
1000*c8dee2aaSAndroid Build Coastguard Worker                  SkArenaAlloc* arena,
1001*c8dee2aaSAndroid Build Coastguard Worker                  GrPaint&& paint,
1002*c8dee2aaSAndroid Build Coastguard Worker                  const SkMatrix& viewMatrix,
1003*c8dee2aaSAndroid Build Coastguard Worker                  const SkRRect& rrect,
1004*c8dee2aaSAndroid Build Coastguard Worker                  const SkRect& localRect,
1005*c8dee2aaSAndroid Build Coastguard Worker                  GrAA aa) {
1006*c8dee2aaSAndroid Build Coastguard Worker     return FillRRectOpImpl::Make(ctx, arena, std::move(paint), viewMatrix, rrect, localRect, aa);
1007*c8dee2aaSAndroid Build Coastguard Worker }
1008*c8dee2aaSAndroid Build Coastguard Worker 
Make(GrRecordingContext * ctx,SkArenaAlloc * arena,GrPaint && paint,const SkMatrix & viewMatrix,const SkRRect & rrect,const SkMatrix & localMatrix,GrAA aa)1009*c8dee2aaSAndroid Build Coastguard Worker GrOp::Owner Make(GrRecordingContext* ctx,
1010*c8dee2aaSAndroid Build Coastguard Worker                  SkArenaAlloc* arena,
1011*c8dee2aaSAndroid Build Coastguard Worker                  GrPaint&& paint,
1012*c8dee2aaSAndroid Build Coastguard Worker                  const SkMatrix& viewMatrix,
1013*c8dee2aaSAndroid Build Coastguard Worker                  const SkRRect& rrect,
1014*c8dee2aaSAndroid Build Coastguard Worker                  const SkMatrix& localMatrix,
1015*c8dee2aaSAndroid Build Coastguard Worker                  GrAA aa) {
1016*c8dee2aaSAndroid Build Coastguard Worker     return FillRRectOpImpl::Make(ctx, arena, std::move(paint), viewMatrix, rrect, localMatrix, aa);
1017*c8dee2aaSAndroid Build Coastguard Worker }
1018*c8dee2aaSAndroid Build Coastguard Worker 
1019*c8dee2aaSAndroid Build Coastguard Worker }  // namespace skgpu::ganesh::FillRRectOp
1020*c8dee2aaSAndroid Build Coastguard Worker 
1021*c8dee2aaSAndroid Build Coastguard Worker #if defined(GPU_TEST_UTILS)
1022*c8dee2aaSAndroid Build Coastguard Worker 
GR_DRAW_OP_TEST_DEFINE(FillRRectOp)1023*c8dee2aaSAndroid Build Coastguard Worker GR_DRAW_OP_TEST_DEFINE(FillRRectOp) {
1024*c8dee2aaSAndroid Build Coastguard Worker     SkArenaAlloc arena(64 * sizeof(float));
1025*c8dee2aaSAndroid Build Coastguard Worker     SkMatrix viewMatrix = GrTest::TestMatrix(random);
1026*c8dee2aaSAndroid Build Coastguard Worker     GrAA aa = GrAA(random->nextBool());
1027*c8dee2aaSAndroid Build Coastguard Worker 
1028*c8dee2aaSAndroid Build Coastguard Worker     SkRect rect = GrTest::TestRect(random);
1029*c8dee2aaSAndroid Build Coastguard Worker     float w = rect.width();
1030*c8dee2aaSAndroid Build Coastguard Worker     float h = rect.height();
1031*c8dee2aaSAndroid Build Coastguard Worker 
1032*c8dee2aaSAndroid Build Coastguard Worker     SkRRect rrect;
1033*c8dee2aaSAndroid Build Coastguard Worker     // TODO: test out other rrect configurations
1034*c8dee2aaSAndroid Build Coastguard Worker     rrect.setNinePatch(rect, w / 3.0f, h / 4.0f, w / 5.0f, h / 6.0);
1035*c8dee2aaSAndroid Build Coastguard Worker 
1036*c8dee2aaSAndroid Build Coastguard Worker     return skgpu::ganesh::FillRRectOp::Make(
1037*c8dee2aaSAndroid Build Coastguard Worker             context, &arena, std::move(paint), viewMatrix, rrect, rrect.rect(), aa);
1038*c8dee2aaSAndroid Build Coastguard Worker }
1039*c8dee2aaSAndroid Build Coastguard Worker 
1040*c8dee2aaSAndroid Build Coastguard Worker #endif
1041