xref: /aosp_15_r20/external/skia/src/gpu/ganesh/ops/LatticeOp.cpp (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1*c8dee2aaSAndroid Build Coastguard Worker /*
2*c8dee2aaSAndroid Build Coastguard Worker  * Copyright 2015 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 #include "src/gpu/ganesh/ops/LatticeOp.h"
8*c8dee2aaSAndroid Build Coastguard Worker 
9*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkAlphaType.h"
10*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkBlendMode.h"
11*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkCanvas.h"
12*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkColor.h"
13*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkMatrix.h"
14*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkRect.h"
15*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkSamplingOptions.h"
16*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkScalar.h"
17*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkSize.h"
18*c8dee2aaSAndroid Build Coastguard Worker #include "include/core/SkString.h"
19*c8dee2aaSAndroid Build Coastguard Worker #include "include/gpu/GpuTypes.h"
20*c8dee2aaSAndroid Build Coastguard Worker #include "include/gpu/ganesh/GrBackendSurface.h"
21*c8dee2aaSAndroid Build Coastguard Worker #include "include/gpu/ganesh/GrRecordingContext.h"
22*c8dee2aaSAndroid Build Coastguard Worker #include "include/gpu/ganesh/GrTypes.h"
23*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/SkColorData.h"
24*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/base/SkAssert.h"
25*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/base/SkDebug.h"
26*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/base/SkPoint_impl.h"
27*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/base/SkTArray.h"
28*c8dee2aaSAndroid Build Coastguard Worker #include "include/private/gpu/ganesh/GrTypesPriv.h"
29*c8dee2aaSAndroid Build Coastguard Worker #include "src/base/SkArenaAlloc.h"
30*c8dee2aaSAndroid Build Coastguard Worker #include "src/base/SkSafeMath.h"
31*c8dee2aaSAndroid Build Coastguard Worker #include "src/base/SkVx.h"
32*c8dee2aaSAndroid Build Coastguard Worker #include "src/core/SkLatticeIter.h"
33*c8dee2aaSAndroid Build Coastguard Worker #include "src/core/SkSLTypeShared.h"
34*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/BufferWriter.h"
35*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/KeyBuilder.h"
36*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/SkBackingFit.h"
37*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrAppliedClip.h"
38*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrCaps.h"
39*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrColorSpaceXform.h"
40*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrGeometryProcessor.h"
41*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrOpFlushState.h"
42*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrPaint.h"
43*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrProcessorAnalysis.h"
44*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrProcessorSet.h"
45*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrProgramInfo.h"
46*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrShaderVar.h"
47*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrSurfaceProxy.h"
48*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrSurfaceProxyView.h"
49*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrUserStencilSettings.h"
50*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/glsl/GrGLSLColorSpaceXformHelper.h"
51*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/glsl/GrGLSLFragmentShaderBuilder.h"
52*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/glsl/GrGLSLVarying.h"
53*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/ops/GrMeshDrawOp.h"
54*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/ops/GrSimpleMeshDrawOpHelper.h"
55*c8dee2aaSAndroid Build Coastguard Worker 
56*c8dee2aaSAndroid Build Coastguard Worker #if defined(GPU_TEST_UTILS)
57*c8dee2aaSAndroid Build Coastguard Worker #include "src/base/SkRandom.h"
58*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrDrawOpTest.h"
59*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrProxyProvider.h"
60*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrRecordingContextPriv.h"
61*c8dee2aaSAndroid Build Coastguard Worker #include "src/gpu/ganesh/GrTestUtils.h"
62*c8dee2aaSAndroid Build Coastguard Worker #endif
63*c8dee2aaSAndroid Build Coastguard Worker 
64*c8dee2aaSAndroid Build Coastguard Worker #include <cstddef>
65*c8dee2aaSAndroid Build Coastguard Worker #include <utility>
66*c8dee2aaSAndroid Build Coastguard Worker 
67*c8dee2aaSAndroid Build Coastguard Worker class GrDstProxyView;
68*c8dee2aaSAndroid Build Coastguard Worker class GrGLSLProgramDataManager;
69*c8dee2aaSAndroid Build Coastguard Worker class GrMeshDrawTarget;
70*c8dee2aaSAndroid Build Coastguard Worker enum class GrXferBarrierFlags;
71*c8dee2aaSAndroid Build Coastguard Worker struct GrShaderCaps;
72*c8dee2aaSAndroid Build Coastguard Worker struct GrSimpleMesh;
73*c8dee2aaSAndroid Build Coastguard Worker 
74*c8dee2aaSAndroid Build Coastguard Worker namespace skgpu::ganesh {
75*c8dee2aaSAndroid Build Coastguard Worker class SurfaceDrawContext;
76*c8dee2aaSAndroid Build Coastguard Worker }
77*c8dee2aaSAndroid Build Coastguard Worker 
78*c8dee2aaSAndroid Build Coastguard Worker using namespace skia_private;
79*c8dee2aaSAndroid Build Coastguard Worker 
80*c8dee2aaSAndroid Build Coastguard Worker namespace skgpu::ganesh::LatticeOp {
81*c8dee2aaSAndroid Build Coastguard Worker 
82*c8dee2aaSAndroid Build Coastguard Worker namespace {
83*c8dee2aaSAndroid Build Coastguard Worker 
84*c8dee2aaSAndroid Build Coastguard Worker class LatticeGP : public GrGeometryProcessor {
85*c8dee2aaSAndroid Build Coastguard Worker public:
Make(SkArenaAlloc * arena,const GrSurfaceProxyView & view,sk_sp<GrColorSpaceXform> csxf,GrSamplerState::Filter filter,bool wideColor)86*c8dee2aaSAndroid Build Coastguard Worker     static GrGeometryProcessor* Make(SkArenaAlloc* arena,
87*c8dee2aaSAndroid Build Coastguard Worker                                      const GrSurfaceProxyView& view,
88*c8dee2aaSAndroid Build Coastguard Worker                                      sk_sp<GrColorSpaceXform> csxf,
89*c8dee2aaSAndroid Build Coastguard Worker                                      GrSamplerState::Filter filter,
90*c8dee2aaSAndroid Build Coastguard Worker                                      bool wideColor) {
91*c8dee2aaSAndroid Build Coastguard Worker         return arena->make([&](void* ptr) {
92*c8dee2aaSAndroid Build Coastguard Worker             return new (ptr) LatticeGP(view, std::move(csxf), filter, wideColor);
93*c8dee2aaSAndroid Build Coastguard Worker         });
94*c8dee2aaSAndroid Build Coastguard Worker     }
95*c8dee2aaSAndroid Build Coastguard Worker 
name() const96*c8dee2aaSAndroid Build Coastguard Worker     const char* name() const override { return "LatticeGP"; }
97*c8dee2aaSAndroid Build Coastguard Worker 
addToKey(const GrShaderCaps &,KeyBuilder * b) const98*c8dee2aaSAndroid Build Coastguard Worker     void addToKey(const GrShaderCaps&, KeyBuilder* b) const override {
99*c8dee2aaSAndroid Build Coastguard Worker         b->add32(GrColorSpaceXform::XformKey(fColorSpaceXform.get()));
100*c8dee2aaSAndroid Build Coastguard Worker     }
101*c8dee2aaSAndroid Build Coastguard Worker 
makeProgramImpl(const GrShaderCaps &) const102*c8dee2aaSAndroid Build Coastguard Worker     std::unique_ptr<ProgramImpl> makeProgramImpl(const GrShaderCaps&) const override {
103*c8dee2aaSAndroid Build Coastguard Worker         class Impl : public ProgramImpl {
104*c8dee2aaSAndroid Build Coastguard Worker         public:
105*c8dee2aaSAndroid Build Coastguard Worker             void setData(const GrGLSLProgramDataManager& pdman,
106*c8dee2aaSAndroid Build Coastguard Worker                          const GrShaderCaps&,
107*c8dee2aaSAndroid Build Coastguard Worker                          const GrGeometryProcessor& geomProc) override {
108*c8dee2aaSAndroid Build Coastguard Worker                 const auto& latticeGP = geomProc.cast<LatticeGP>();
109*c8dee2aaSAndroid Build Coastguard Worker                 fColorSpaceXformHelper.setData(pdman, latticeGP.fColorSpaceXform.get());
110*c8dee2aaSAndroid Build Coastguard Worker             }
111*c8dee2aaSAndroid Build Coastguard Worker 
112*c8dee2aaSAndroid Build Coastguard Worker         private:
113*c8dee2aaSAndroid Build Coastguard Worker             void onEmitCode(EmitArgs& args, GrGPArgs* gpArgs) override {
114*c8dee2aaSAndroid Build Coastguard Worker                 using Interpolation = GrGLSLVaryingHandler::Interpolation;
115*c8dee2aaSAndroid Build Coastguard Worker                 const auto& latticeGP = args.fGeomProc.cast<LatticeGP>();
116*c8dee2aaSAndroid Build Coastguard Worker                 fColorSpaceXformHelper.emitCode(args.fUniformHandler,
117*c8dee2aaSAndroid Build Coastguard Worker                                                 latticeGP.fColorSpaceXform.get());
118*c8dee2aaSAndroid Build Coastguard Worker 
119*c8dee2aaSAndroid Build Coastguard Worker                 args.fVaryingHandler->emitAttributes(latticeGP);
120*c8dee2aaSAndroid Build Coastguard Worker                 WriteOutputPosition(args.fVertBuilder, gpArgs, latticeGP.fInPosition.name());
121*c8dee2aaSAndroid Build Coastguard Worker                 gpArgs->fLocalCoordVar = latticeGP.fInTextureCoords.asShaderVar();
122*c8dee2aaSAndroid Build Coastguard Worker 
123*c8dee2aaSAndroid Build Coastguard Worker                 args.fFragBuilder->codeAppend("float2 textureCoords;");
124*c8dee2aaSAndroid Build Coastguard Worker                 args.fVaryingHandler->addPassThroughAttribute(
125*c8dee2aaSAndroid Build Coastguard Worker                         latticeGP.fInTextureCoords.asShaderVar(),
126*c8dee2aaSAndroid Build Coastguard Worker                         "textureCoords");
127*c8dee2aaSAndroid Build Coastguard Worker                 args.fFragBuilder->codeAppend("float4 textureDomain;");
128*c8dee2aaSAndroid Build Coastguard Worker                 args.fVaryingHandler->addPassThroughAttribute(
129*c8dee2aaSAndroid Build Coastguard Worker                         latticeGP.fInTextureDomain.asShaderVar(),
130*c8dee2aaSAndroid Build Coastguard Worker                         "textureDomain",
131*c8dee2aaSAndroid Build Coastguard Worker                         Interpolation::kCanBeFlat);
132*c8dee2aaSAndroid Build Coastguard Worker                 args.fFragBuilder->codeAppendf("half4 %s;", args.fOutputColor);
133*c8dee2aaSAndroid Build Coastguard Worker                 args.fVaryingHandler->addPassThroughAttribute(latticeGP.fInColor.asShaderVar(),
134*c8dee2aaSAndroid Build Coastguard Worker                                                               args.fOutputColor,
135*c8dee2aaSAndroid Build Coastguard Worker                                                               Interpolation::kCanBeFlat);
136*c8dee2aaSAndroid Build Coastguard Worker                 args.fFragBuilder->codeAppendf("%s = ", args.fOutputColor);
137*c8dee2aaSAndroid Build Coastguard Worker                 args.fFragBuilder->appendTextureLookupAndBlend(
138*c8dee2aaSAndroid Build Coastguard Worker                         args.fOutputColor,
139*c8dee2aaSAndroid Build Coastguard Worker                         SkBlendMode::kModulate,
140*c8dee2aaSAndroid Build Coastguard Worker                         args.fTexSamplers[0],
141*c8dee2aaSAndroid Build Coastguard Worker                         "clamp(textureCoords, textureDomain.xy, textureDomain.zw)",
142*c8dee2aaSAndroid Build Coastguard Worker                         &fColorSpaceXformHelper);
143*c8dee2aaSAndroid Build Coastguard Worker                 args.fFragBuilder->codeAppend(";");
144*c8dee2aaSAndroid Build Coastguard Worker                 args.fFragBuilder->codeAppendf("const half4 %s = half4(1);", args.fOutputCoverage);
145*c8dee2aaSAndroid Build Coastguard Worker             }
146*c8dee2aaSAndroid Build Coastguard Worker 
147*c8dee2aaSAndroid Build Coastguard Worker             GrGLSLColorSpaceXformHelper fColorSpaceXformHelper;
148*c8dee2aaSAndroid Build Coastguard Worker         };
149*c8dee2aaSAndroid Build Coastguard Worker 
150*c8dee2aaSAndroid Build Coastguard Worker         return std::make_unique<Impl>();
151*c8dee2aaSAndroid Build Coastguard Worker     }
152*c8dee2aaSAndroid Build Coastguard Worker 
153*c8dee2aaSAndroid Build Coastguard Worker private:
LatticeGP(const GrSurfaceProxyView & view,sk_sp<GrColorSpaceXform> csxf,GrSamplerState::Filter filter,bool wideColor)154*c8dee2aaSAndroid Build Coastguard Worker     LatticeGP(const GrSurfaceProxyView& view, sk_sp<GrColorSpaceXform> csxf,
155*c8dee2aaSAndroid Build Coastguard Worker               GrSamplerState::Filter filter, bool wideColor)
156*c8dee2aaSAndroid Build Coastguard Worker             : INHERITED(kLatticeGP_ClassID)
157*c8dee2aaSAndroid Build Coastguard Worker             , fColorSpaceXform(std::move(csxf)) {
158*c8dee2aaSAndroid Build Coastguard Worker 
159*c8dee2aaSAndroid Build Coastguard Worker         fSampler.reset(GrSamplerState(GrSamplerState::WrapMode::kClamp, filter),
160*c8dee2aaSAndroid Build Coastguard Worker                        view.proxy()->backendFormat(), view.swizzle());
161*c8dee2aaSAndroid Build Coastguard Worker         this->setTextureSamplerCnt(1);
162*c8dee2aaSAndroid Build Coastguard Worker         fInPosition = {"position", kFloat2_GrVertexAttribType, SkSLType::kFloat2};
163*c8dee2aaSAndroid Build Coastguard Worker         fInTextureCoords = {"textureCoords", kFloat2_GrVertexAttribType, SkSLType::kFloat2};
164*c8dee2aaSAndroid Build Coastguard Worker         fInTextureDomain = {"textureDomain", kFloat4_GrVertexAttribType, SkSLType::kFloat4};
165*c8dee2aaSAndroid Build Coastguard Worker         fInColor = MakeColorAttribute("color", wideColor);
166*c8dee2aaSAndroid Build Coastguard Worker         this->setVertexAttributesWithImplicitOffsets(&fInPosition, 4);
167*c8dee2aaSAndroid Build Coastguard Worker     }
168*c8dee2aaSAndroid Build Coastguard Worker 
onTextureSampler(int) const169*c8dee2aaSAndroid Build Coastguard Worker     const TextureSampler& onTextureSampler(int) const override { return fSampler; }
170*c8dee2aaSAndroid Build Coastguard Worker 
171*c8dee2aaSAndroid Build Coastguard Worker     Attribute fInPosition;
172*c8dee2aaSAndroid Build Coastguard Worker     Attribute fInTextureCoords;
173*c8dee2aaSAndroid Build Coastguard Worker     Attribute fInTextureDomain;
174*c8dee2aaSAndroid Build Coastguard Worker     Attribute fInColor;
175*c8dee2aaSAndroid Build Coastguard Worker 
176*c8dee2aaSAndroid Build Coastguard Worker     sk_sp<GrColorSpaceXform> fColorSpaceXform;
177*c8dee2aaSAndroid Build Coastguard Worker     TextureSampler fSampler;
178*c8dee2aaSAndroid Build Coastguard Worker 
179*c8dee2aaSAndroid Build Coastguard Worker     using INHERITED = GrGeometryProcessor;
180*c8dee2aaSAndroid Build Coastguard Worker };
181*c8dee2aaSAndroid Build Coastguard Worker 
182*c8dee2aaSAndroid Build Coastguard Worker class NonAALatticeOp final : public GrMeshDrawOp {
183*c8dee2aaSAndroid Build Coastguard Worker private:
184*c8dee2aaSAndroid Build Coastguard Worker     using Helper = GrSimpleMeshDrawOpHelper;
185*c8dee2aaSAndroid Build Coastguard Worker 
186*c8dee2aaSAndroid Build Coastguard Worker public:
187*c8dee2aaSAndroid Build Coastguard Worker     DEFINE_OP_CLASS_ID
188*c8dee2aaSAndroid Build Coastguard Worker 
Make(GrRecordingContext * context,GrPaint && paint,const SkMatrix & viewMatrix,GrSurfaceProxyView view,SkAlphaType alphaType,sk_sp<GrColorSpaceXform> colorSpaceXForm,GrSamplerState::Filter filter,std::unique_ptr<SkLatticeIter> iter,const SkRect & dst)189*c8dee2aaSAndroid Build Coastguard Worker     static GrOp::Owner Make(GrRecordingContext* context,
190*c8dee2aaSAndroid Build Coastguard Worker                             GrPaint&& paint,
191*c8dee2aaSAndroid Build Coastguard Worker                             const SkMatrix& viewMatrix,
192*c8dee2aaSAndroid Build Coastguard Worker                             GrSurfaceProxyView view,
193*c8dee2aaSAndroid Build Coastguard Worker                             SkAlphaType alphaType,
194*c8dee2aaSAndroid Build Coastguard Worker                             sk_sp<GrColorSpaceXform> colorSpaceXForm,
195*c8dee2aaSAndroid Build Coastguard Worker                             GrSamplerState::Filter filter,
196*c8dee2aaSAndroid Build Coastguard Worker                             std::unique_ptr<SkLatticeIter> iter,
197*c8dee2aaSAndroid Build Coastguard Worker                             const SkRect& dst) {
198*c8dee2aaSAndroid Build Coastguard Worker         SkASSERT(view.proxy());
199*c8dee2aaSAndroid Build Coastguard Worker         return Helper::FactoryHelper<NonAALatticeOp>(context, std::move(paint), viewMatrix,
200*c8dee2aaSAndroid Build Coastguard Worker                                                      std::move(view), alphaType,
201*c8dee2aaSAndroid Build Coastguard Worker                                                      std::move(colorSpaceXForm), filter,
202*c8dee2aaSAndroid Build Coastguard Worker                                                      std::move(iter), dst);
203*c8dee2aaSAndroid Build Coastguard Worker     }
204*c8dee2aaSAndroid Build Coastguard Worker 
NonAALatticeOp(GrProcessorSet * processorSet,const SkPMColor4f & color,const SkMatrix & viewMatrix,GrSurfaceProxyView view,SkAlphaType alphaType,sk_sp<GrColorSpaceXform> colorSpaceXform,GrSamplerState::Filter filter,std::unique_ptr<SkLatticeIter> iter,const SkRect & dst)205*c8dee2aaSAndroid Build Coastguard Worker     NonAALatticeOp(GrProcessorSet* processorSet, const SkPMColor4f& color,
206*c8dee2aaSAndroid Build Coastguard Worker                    const SkMatrix& viewMatrix, GrSurfaceProxyView view,
207*c8dee2aaSAndroid Build Coastguard Worker                    SkAlphaType alphaType, sk_sp<GrColorSpaceXform> colorSpaceXform,
208*c8dee2aaSAndroid Build Coastguard Worker                    GrSamplerState::Filter filter, std::unique_ptr<SkLatticeIter> iter,
209*c8dee2aaSAndroid Build Coastguard Worker                    const SkRect& dst)
210*c8dee2aaSAndroid Build Coastguard Worker             : INHERITED(ClassID())
211*c8dee2aaSAndroid Build Coastguard Worker             , fHelper(processorSet, GrAAType::kNone)
212*c8dee2aaSAndroid Build Coastguard Worker             , fView(std::move(view))
213*c8dee2aaSAndroid Build Coastguard Worker             , fAlphaType(alphaType)
214*c8dee2aaSAndroid Build Coastguard Worker             , fColorSpaceXform(std::move(colorSpaceXform))
215*c8dee2aaSAndroid Build Coastguard Worker             , fFilter(filter) {
216*c8dee2aaSAndroid Build Coastguard Worker         Patch& patch = fPatches.push_back();
217*c8dee2aaSAndroid Build Coastguard Worker         patch.fViewMatrix = viewMatrix;
218*c8dee2aaSAndroid Build Coastguard Worker         patch.fColor = color;
219*c8dee2aaSAndroid Build Coastguard Worker         patch.fIter = std::move(iter);
220*c8dee2aaSAndroid Build Coastguard Worker         patch.fDst = dst;
221*c8dee2aaSAndroid Build Coastguard Worker 
222*c8dee2aaSAndroid Build Coastguard Worker         // setup bounds
223*c8dee2aaSAndroid Build Coastguard Worker         this->setTransformedBounds(patch.fDst, viewMatrix, HasAABloat::kNo, IsHairline::kNo);
224*c8dee2aaSAndroid Build Coastguard Worker     }
225*c8dee2aaSAndroid Build Coastguard Worker 
name() const226*c8dee2aaSAndroid Build Coastguard Worker     const char* name() const override { return "NonAALatticeOp"; }
227*c8dee2aaSAndroid Build Coastguard Worker 
visitProxies(const GrVisitProxyFunc & func) const228*c8dee2aaSAndroid Build Coastguard Worker     void visitProxies(const GrVisitProxyFunc& func) const override {
229*c8dee2aaSAndroid Build Coastguard Worker         func(fView.proxy(), skgpu::Mipmapped::kNo);
230*c8dee2aaSAndroid Build Coastguard Worker         if (fProgramInfo) {
231*c8dee2aaSAndroid Build Coastguard Worker             fProgramInfo->visitFPProxies(func);
232*c8dee2aaSAndroid Build Coastguard Worker         } else {
233*c8dee2aaSAndroid Build Coastguard Worker             fHelper.visitProxies(func);
234*c8dee2aaSAndroid Build Coastguard Worker         }
235*c8dee2aaSAndroid Build Coastguard Worker     }
236*c8dee2aaSAndroid Build Coastguard Worker 
fixedFunctionFlags() const237*c8dee2aaSAndroid Build Coastguard Worker     FixedFunctionFlags fixedFunctionFlags() const override { return fHelper.fixedFunctionFlags(); }
238*c8dee2aaSAndroid Build Coastguard Worker 
finalize(const GrCaps & caps,const GrAppliedClip * clip,GrClampType clampType)239*c8dee2aaSAndroid Build Coastguard Worker     GrProcessorSet::Analysis finalize(const GrCaps& caps, const GrAppliedClip* clip,
240*c8dee2aaSAndroid Build Coastguard Worker                                       GrClampType clampType) override {
241*c8dee2aaSAndroid Build Coastguard Worker         auto opaque = fPatches[0].fColor.isOpaque() && fAlphaType == kOpaque_SkAlphaType
242*c8dee2aaSAndroid Build Coastguard Worker                               ? GrProcessorAnalysisColor::Opaque::kYes
243*c8dee2aaSAndroid Build Coastguard Worker                               : GrProcessorAnalysisColor::Opaque::kNo;
244*c8dee2aaSAndroid Build Coastguard Worker         auto analysisColor = GrProcessorAnalysisColor(opaque);
245*c8dee2aaSAndroid Build Coastguard Worker         auto result = fHelper.finalizeProcessors(caps, clip, clampType,
246*c8dee2aaSAndroid Build Coastguard Worker                                                  GrProcessorAnalysisCoverage::kNone,
247*c8dee2aaSAndroid Build Coastguard Worker                                                  &analysisColor);
248*c8dee2aaSAndroid Build Coastguard Worker         analysisColor.isConstant(&fPatches[0].fColor);
249*c8dee2aaSAndroid Build Coastguard Worker         fWideColor = !fPatches[0].fColor.fitsInBytes();
250*c8dee2aaSAndroid Build Coastguard Worker         return result;
251*c8dee2aaSAndroid Build Coastguard Worker     }
252*c8dee2aaSAndroid Build Coastguard Worker 
253*c8dee2aaSAndroid Build Coastguard Worker private:
programInfo()254*c8dee2aaSAndroid Build Coastguard Worker     GrProgramInfo* programInfo() override { return fProgramInfo; }
255*c8dee2aaSAndroid Build Coastguard Worker 
onCreateProgramInfo(const GrCaps * caps,SkArenaAlloc * arena,const GrSurfaceProxyView & writeView,bool usesMSAASurface,GrAppliedClip && appliedClip,const GrDstProxyView & dstProxyView,GrXferBarrierFlags renderPassXferBarriers,GrLoadOp colorLoadOp)256*c8dee2aaSAndroid Build Coastguard Worker     void onCreateProgramInfo(const GrCaps* caps,
257*c8dee2aaSAndroid Build Coastguard Worker                              SkArenaAlloc* arena,
258*c8dee2aaSAndroid Build Coastguard Worker                              const GrSurfaceProxyView& writeView,
259*c8dee2aaSAndroid Build Coastguard Worker                              bool usesMSAASurface,
260*c8dee2aaSAndroid Build Coastguard Worker                              GrAppliedClip&& appliedClip,
261*c8dee2aaSAndroid Build Coastguard Worker                              const GrDstProxyView& dstProxyView,
262*c8dee2aaSAndroid Build Coastguard Worker                              GrXferBarrierFlags renderPassXferBarriers,
263*c8dee2aaSAndroid Build Coastguard Worker                              GrLoadOp colorLoadOp) override {
264*c8dee2aaSAndroid Build Coastguard Worker 
265*c8dee2aaSAndroid Build Coastguard Worker         auto gp = LatticeGP::Make(arena, fView, fColorSpaceXform, fFilter, fWideColor);
266*c8dee2aaSAndroid Build Coastguard Worker         if (!gp) {
267*c8dee2aaSAndroid Build Coastguard Worker             return;
268*c8dee2aaSAndroid Build Coastguard Worker         }
269*c8dee2aaSAndroid Build Coastguard Worker 
270*c8dee2aaSAndroid Build Coastguard Worker         fProgramInfo = GrSimpleMeshDrawOpHelper::CreateProgramInfo(caps, arena, writeView,
271*c8dee2aaSAndroid Build Coastguard Worker                                                                    usesMSAASurface,
272*c8dee2aaSAndroid Build Coastguard Worker                                                                    std::move(appliedClip),
273*c8dee2aaSAndroid Build Coastguard Worker                                                                    dstProxyView, gp,
274*c8dee2aaSAndroid Build Coastguard Worker                                                                    fHelper.detachProcessorSet(),
275*c8dee2aaSAndroid Build Coastguard Worker                                                                    GrPrimitiveType::kTriangles,
276*c8dee2aaSAndroid Build Coastguard Worker                                                                    renderPassXferBarriers,
277*c8dee2aaSAndroid Build Coastguard Worker                                                                    colorLoadOp,
278*c8dee2aaSAndroid Build Coastguard Worker                                                                    fHelper.pipelineFlags(),
279*c8dee2aaSAndroid Build Coastguard Worker                                                                    &GrUserStencilSettings::kUnused);
280*c8dee2aaSAndroid Build Coastguard Worker     }
281*c8dee2aaSAndroid Build Coastguard Worker 
onPrepareDraws(GrMeshDrawTarget * target)282*c8dee2aaSAndroid Build Coastguard Worker     void onPrepareDraws(GrMeshDrawTarget* target) override {
283*c8dee2aaSAndroid Build Coastguard Worker         if (!fProgramInfo) {
284*c8dee2aaSAndroid Build Coastguard Worker             this->createProgramInfo(target);
285*c8dee2aaSAndroid Build Coastguard Worker             if (!fProgramInfo) {
286*c8dee2aaSAndroid Build Coastguard Worker                 return;
287*c8dee2aaSAndroid Build Coastguard Worker             }
288*c8dee2aaSAndroid Build Coastguard Worker         }
289*c8dee2aaSAndroid Build Coastguard Worker 
290*c8dee2aaSAndroid Build Coastguard Worker         int patchCnt = fPatches.size();
291*c8dee2aaSAndroid Build Coastguard Worker         int numRects = 0;
292*c8dee2aaSAndroid Build Coastguard Worker 
293*c8dee2aaSAndroid Build Coastguard Worker         SkSafeMath safeMath;
294*c8dee2aaSAndroid Build Coastguard Worker         for (int i = 0; i < patchCnt; i++) {
295*c8dee2aaSAndroid Build Coastguard Worker             numRects = safeMath.addInt(numRects, fPatches[i].fIter->numRectsToDraw());
296*c8dee2aaSAndroid Build Coastguard Worker         }
297*c8dee2aaSAndroid Build Coastguard Worker 
298*c8dee2aaSAndroid Build Coastguard Worker         if (!numRects || !safeMath) {
299*c8dee2aaSAndroid Build Coastguard Worker             return;
300*c8dee2aaSAndroid Build Coastguard Worker         }
301*c8dee2aaSAndroid Build Coastguard Worker 
302*c8dee2aaSAndroid Build Coastguard Worker         const size_t kVertexStride = fProgramInfo->geomProc().vertexStride();
303*c8dee2aaSAndroid Build Coastguard Worker 
304*c8dee2aaSAndroid Build Coastguard Worker         QuadHelper helper(target, kVertexStride, numRects);
305*c8dee2aaSAndroid Build Coastguard Worker 
306*c8dee2aaSAndroid Build Coastguard Worker         VertexWriter vertices{helper.vertices()};
307*c8dee2aaSAndroid Build Coastguard Worker         if (!vertices) {
308*c8dee2aaSAndroid Build Coastguard Worker             SkDebugf("Could not allocate vertices\n");
309*c8dee2aaSAndroid Build Coastguard Worker             return;
310*c8dee2aaSAndroid Build Coastguard Worker         }
311*c8dee2aaSAndroid Build Coastguard Worker 
312*c8dee2aaSAndroid Build Coastguard Worker         for (int i = 0; i < patchCnt; i++) {
313*c8dee2aaSAndroid Build Coastguard Worker             const Patch& patch = fPatches[i];
314*c8dee2aaSAndroid Build Coastguard Worker 
315*c8dee2aaSAndroid Build Coastguard Worker             VertexColor patchColor(patch.fColor, fWideColor);
316*c8dee2aaSAndroid Build Coastguard Worker 
317*c8dee2aaSAndroid Build Coastguard Worker             // Apply the view matrix here if it is scale-translate.  Otherwise, we need to
318*c8dee2aaSAndroid Build Coastguard Worker             // wait until we've created the dst rects.
319*c8dee2aaSAndroid Build Coastguard Worker             bool isScaleTranslate = patch.fViewMatrix.isScaleTranslate();
320*c8dee2aaSAndroid Build Coastguard Worker             if (isScaleTranslate) {
321*c8dee2aaSAndroid Build Coastguard Worker                 patch.fIter->mapDstScaleTranslate(patch.fViewMatrix);
322*c8dee2aaSAndroid Build Coastguard Worker             }
323*c8dee2aaSAndroid Build Coastguard Worker 
324*c8dee2aaSAndroid Build Coastguard Worker             SkIRect srcR;
325*c8dee2aaSAndroid Build Coastguard Worker             SkRect dstR;
326*c8dee2aaSAndroid Build Coastguard Worker             skvx::float4 scales(1.f / fView.proxy()->width(), 1.f / fView.proxy()->height(),
327*c8dee2aaSAndroid Build Coastguard Worker                                 1.f / fView.proxy()->width(), 1.f / fView.proxy()->height());
328*c8dee2aaSAndroid Build Coastguard Worker             static const skvx::float4 kDomainOffsets(0.5f, 0.5f, -0.5f, -0.5f);
329*c8dee2aaSAndroid Build Coastguard Worker             static const skvx::float4 kFlipOffsets(0.f, 1.f, 0.f, 1.f);
330*c8dee2aaSAndroid Build Coastguard Worker             static const skvx::float4 kFlipMuls(1.f, -1.f, 1.f, -1.f);
331*c8dee2aaSAndroid Build Coastguard Worker             while (patch.fIter->next(&srcR, &dstR)) {
332*c8dee2aaSAndroid Build Coastguard Worker                 skvx::float4 coords(SkIntToScalar(srcR.fLeft), SkIntToScalar(srcR.fTop),
333*c8dee2aaSAndroid Build Coastguard Worker                                     SkIntToScalar(srcR.fRight), SkIntToScalar(srcR.fBottom));
334*c8dee2aaSAndroid Build Coastguard Worker                 skvx::float4 domain = coords + kDomainOffsets;
335*c8dee2aaSAndroid Build Coastguard Worker                 coords *= scales;
336*c8dee2aaSAndroid Build Coastguard Worker                 domain *= scales;
337*c8dee2aaSAndroid Build Coastguard Worker                 if (fView.origin() == kBottomLeft_GrSurfaceOrigin) {
338*c8dee2aaSAndroid Build Coastguard Worker                     coords = kFlipMuls * coords + kFlipOffsets;
339*c8dee2aaSAndroid Build Coastguard Worker                     domain = skvx::shuffle<0, 3, 2, 1>(kFlipMuls * domain + kFlipOffsets);
340*c8dee2aaSAndroid Build Coastguard Worker                 }
341*c8dee2aaSAndroid Build Coastguard Worker                 SkRect texDomain;
342*c8dee2aaSAndroid Build Coastguard Worker                 SkRect texCoords;
343*c8dee2aaSAndroid Build Coastguard Worker                 domain.store(&texDomain);
344*c8dee2aaSAndroid Build Coastguard Worker                 coords.store(&texCoords);
345*c8dee2aaSAndroid Build Coastguard Worker 
346*c8dee2aaSAndroid Build Coastguard Worker                 if (isScaleTranslate) {
347*c8dee2aaSAndroid Build Coastguard Worker                     vertices.writeQuad(VertexWriter::TriStripFromRect(dstR),
348*c8dee2aaSAndroid Build Coastguard Worker                                        VertexWriter::TriStripFromRect(texCoords),
349*c8dee2aaSAndroid Build Coastguard Worker                                        texDomain,
350*c8dee2aaSAndroid Build Coastguard Worker                                        patchColor);
351*c8dee2aaSAndroid Build Coastguard Worker                 } else {
352*c8dee2aaSAndroid Build Coastguard Worker                     SkPoint mappedPts[4];
353*c8dee2aaSAndroid Build Coastguard Worker                     patch.fViewMatrix.mapRectToQuad(mappedPts, dstR);
354*c8dee2aaSAndroid Build Coastguard Worker                     // In the above if statement, writeQuad writes the corners as:
355*c8dee2aaSAndroid Build Coastguard Worker                     // left-top, left-bottom, right-top, right-bottom.
356*c8dee2aaSAndroid Build Coastguard Worker                     // However, mapRectToQuad returns them in the order:
357*c8dee2aaSAndroid Build Coastguard Worker                     // left-top, right-top, right-bottom, left-bottom
358*c8dee2aaSAndroid Build Coastguard Worker                     // Thus we write out the vertices to match the writeQuad path.
359*c8dee2aaSAndroid Build Coastguard Worker                     vertices << mappedPts[0]
360*c8dee2aaSAndroid Build Coastguard Worker                              << SkPoint::Make(texCoords.fLeft, texCoords.fTop)
361*c8dee2aaSAndroid Build Coastguard Worker                              << texDomain
362*c8dee2aaSAndroid Build Coastguard Worker                              << patchColor;
363*c8dee2aaSAndroid Build Coastguard Worker                     vertices << mappedPts[3]
364*c8dee2aaSAndroid Build Coastguard Worker                              << SkPoint::Make(texCoords.fLeft, texCoords.fBottom)
365*c8dee2aaSAndroid Build Coastguard Worker                              << texDomain
366*c8dee2aaSAndroid Build Coastguard Worker                              << patchColor;
367*c8dee2aaSAndroid Build Coastguard Worker                     vertices << mappedPts[1]
368*c8dee2aaSAndroid Build Coastguard Worker                              << SkPoint::Make(texCoords.fRight, texCoords.fTop)
369*c8dee2aaSAndroid Build Coastguard Worker                              << texDomain
370*c8dee2aaSAndroid Build Coastguard Worker                              << patchColor;
371*c8dee2aaSAndroid Build Coastguard Worker                     vertices << mappedPts[2]
372*c8dee2aaSAndroid Build Coastguard Worker                              << SkPoint::Make(texCoords.fRight, texCoords.fBottom)
373*c8dee2aaSAndroid Build Coastguard Worker                              << texDomain
374*c8dee2aaSAndroid Build Coastguard Worker                              << patchColor;
375*c8dee2aaSAndroid Build Coastguard Worker                 }
376*c8dee2aaSAndroid Build Coastguard Worker             }
377*c8dee2aaSAndroid Build Coastguard Worker         }
378*c8dee2aaSAndroid Build Coastguard Worker 
379*c8dee2aaSAndroid Build Coastguard Worker         fMesh = helper.mesh();
380*c8dee2aaSAndroid Build Coastguard Worker     }
381*c8dee2aaSAndroid Build Coastguard Worker 
onExecute(GrOpFlushState * flushState,const SkRect & chainBounds)382*c8dee2aaSAndroid Build Coastguard Worker     void onExecute(GrOpFlushState* flushState, const SkRect& chainBounds) override {
383*c8dee2aaSAndroid Build Coastguard Worker         if (!fProgramInfo || !fMesh) {
384*c8dee2aaSAndroid Build Coastguard Worker             return;
385*c8dee2aaSAndroid Build Coastguard Worker         }
386*c8dee2aaSAndroid Build Coastguard Worker 
387*c8dee2aaSAndroid Build Coastguard Worker         flushState->bindPipelineAndScissorClip(*fProgramInfo, chainBounds);
388*c8dee2aaSAndroid Build Coastguard Worker         flushState->bindTextures(fProgramInfo->geomProc(),
389*c8dee2aaSAndroid Build Coastguard Worker                                  *fView.proxy(),
390*c8dee2aaSAndroid Build Coastguard Worker                                  fProgramInfo->pipeline());
391*c8dee2aaSAndroid Build Coastguard Worker         flushState->drawMesh(*fMesh);
392*c8dee2aaSAndroid Build Coastguard Worker     }
393*c8dee2aaSAndroid Build Coastguard Worker 
onCombineIfPossible(GrOp * t,SkArenaAlloc *,const GrCaps & caps)394*c8dee2aaSAndroid Build Coastguard Worker     CombineResult onCombineIfPossible(GrOp* t, SkArenaAlloc*, const GrCaps& caps) override {
395*c8dee2aaSAndroid Build Coastguard Worker         NonAALatticeOp* that = t->cast<NonAALatticeOp>();
396*c8dee2aaSAndroid Build Coastguard Worker         if (fView != that->fView) {
397*c8dee2aaSAndroid Build Coastguard Worker             return CombineResult::kCannotCombine;
398*c8dee2aaSAndroid Build Coastguard Worker         }
399*c8dee2aaSAndroid Build Coastguard Worker         if (fFilter != that->fFilter) {
400*c8dee2aaSAndroid Build Coastguard Worker             return CombineResult::kCannotCombine;
401*c8dee2aaSAndroid Build Coastguard Worker         }
402*c8dee2aaSAndroid Build Coastguard Worker         if (!GrColorSpaceXform::Equals(fColorSpaceXform.get(), that->fColorSpaceXform.get())) {
403*c8dee2aaSAndroid Build Coastguard Worker             return CombineResult::kCannotCombine;
404*c8dee2aaSAndroid Build Coastguard Worker         }
405*c8dee2aaSAndroid Build Coastguard Worker         if (!fHelper.isCompatible(that->fHelper, caps, this->bounds(), that->bounds())) {
406*c8dee2aaSAndroid Build Coastguard Worker             return CombineResult::kCannotCombine;
407*c8dee2aaSAndroid Build Coastguard Worker         }
408*c8dee2aaSAndroid Build Coastguard Worker 
409*c8dee2aaSAndroid Build Coastguard Worker         fPatches.move_back_n(that->fPatches.size(), that->fPatches.begin());
410*c8dee2aaSAndroid Build Coastguard Worker         fWideColor |= that->fWideColor;
411*c8dee2aaSAndroid Build Coastguard Worker         return CombineResult::kMerged;
412*c8dee2aaSAndroid Build Coastguard Worker     }
413*c8dee2aaSAndroid Build Coastguard Worker 
414*c8dee2aaSAndroid Build Coastguard Worker #if defined(GPU_TEST_UTILS)
onDumpInfo() const415*c8dee2aaSAndroid Build Coastguard Worker     SkString onDumpInfo() const override {
416*c8dee2aaSAndroid Build Coastguard Worker         SkString str;
417*c8dee2aaSAndroid Build Coastguard Worker 
418*c8dee2aaSAndroid Build Coastguard Worker         for (int i = 0; i < fPatches.size(); ++i) {
419*c8dee2aaSAndroid Build Coastguard Worker             str.appendf("%d: Color: 0x%08x Dst [L: %.2f, T: %.2f, R: %.2f, B: %.2f]\n", i,
420*c8dee2aaSAndroid Build Coastguard Worker                         fPatches[i].fColor.toBytes_RGBA(), fPatches[i].fDst.fLeft,
421*c8dee2aaSAndroid Build Coastguard Worker                         fPatches[i].fDst.fTop, fPatches[i].fDst.fRight, fPatches[i].fDst.fBottom);
422*c8dee2aaSAndroid Build Coastguard Worker         }
423*c8dee2aaSAndroid Build Coastguard Worker 
424*c8dee2aaSAndroid Build Coastguard Worker         str += fHelper.dumpInfo();
425*c8dee2aaSAndroid Build Coastguard Worker         return str;
426*c8dee2aaSAndroid Build Coastguard Worker     }
427*c8dee2aaSAndroid Build Coastguard Worker #endif
428*c8dee2aaSAndroid Build Coastguard Worker 
429*c8dee2aaSAndroid Build Coastguard Worker     struct Patch {
430*c8dee2aaSAndroid Build Coastguard Worker         SkMatrix fViewMatrix;
431*c8dee2aaSAndroid Build Coastguard Worker         std::unique_ptr<SkLatticeIter> fIter;
432*c8dee2aaSAndroid Build Coastguard Worker         SkRect fDst;
433*c8dee2aaSAndroid Build Coastguard Worker         SkPMColor4f fColor;
434*c8dee2aaSAndroid Build Coastguard Worker     };
435*c8dee2aaSAndroid Build Coastguard Worker 
436*c8dee2aaSAndroid Build Coastguard Worker     Helper fHelper;
437*c8dee2aaSAndroid Build Coastguard Worker     STArray<1, Patch, true> fPatches;
438*c8dee2aaSAndroid Build Coastguard Worker     GrSurfaceProxyView fView;
439*c8dee2aaSAndroid Build Coastguard Worker     SkAlphaType fAlphaType;
440*c8dee2aaSAndroid Build Coastguard Worker     sk_sp<GrColorSpaceXform> fColorSpaceXform;
441*c8dee2aaSAndroid Build Coastguard Worker     GrSamplerState::Filter fFilter;
442*c8dee2aaSAndroid Build Coastguard Worker     bool fWideColor;
443*c8dee2aaSAndroid Build Coastguard Worker 
444*c8dee2aaSAndroid Build Coastguard Worker     GrSimpleMesh*  fMesh = nullptr;
445*c8dee2aaSAndroid Build Coastguard Worker     GrProgramInfo* fProgramInfo = nullptr;
446*c8dee2aaSAndroid Build Coastguard Worker 
447*c8dee2aaSAndroid Build Coastguard Worker     using INHERITED = GrMeshDrawOp;
448*c8dee2aaSAndroid Build Coastguard Worker };
449*c8dee2aaSAndroid Build Coastguard Worker 
450*c8dee2aaSAndroid Build Coastguard Worker }  // anonymous namespace
451*c8dee2aaSAndroid Build Coastguard Worker 
MakeNonAA(GrRecordingContext * context,GrPaint && paint,const SkMatrix & viewMatrix,GrSurfaceProxyView view,SkAlphaType alphaType,sk_sp<GrColorSpaceXform> colorSpaceXform,GrSamplerState::Filter filter,std::unique_ptr<SkLatticeIter> iter,const SkRect & dst)452*c8dee2aaSAndroid Build Coastguard Worker GrOp::Owner MakeNonAA(GrRecordingContext* context,
453*c8dee2aaSAndroid Build Coastguard Worker                       GrPaint&& paint,
454*c8dee2aaSAndroid Build Coastguard Worker                       const SkMatrix& viewMatrix,
455*c8dee2aaSAndroid Build Coastguard Worker                       GrSurfaceProxyView view,
456*c8dee2aaSAndroid Build Coastguard Worker                       SkAlphaType alphaType,
457*c8dee2aaSAndroid Build Coastguard Worker                       sk_sp<GrColorSpaceXform> colorSpaceXform,
458*c8dee2aaSAndroid Build Coastguard Worker                       GrSamplerState::Filter filter,
459*c8dee2aaSAndroid Build Coastguard Worker                       std::unique_ptr<SkLatticeIter> iter,
460*c8dee2aaSAndroid Build Coastguard Worker                       const SkRect& dst) {
461*c8dee2aaSAndroid Build Coastguard Worker     return NonAALatticeOp::Make(context, std::move(paint), viewMatrix, std::move(view), alphaType,
462*c8dee2aaSAndroid Build Coastguard Worker                                 std::move(colorSpaceXform), filter, std::move(iter), dst);
463*c8dee2aaSAndroid Build Coastguard Worker }
464*c8dee2aaSAndroid Build Coastguard Worker 
465*c8dee2aaSAndroid Build Coastguard Worker }  // namespace skgpu::ganesh::LatticeOp
466*c8dee2aaSAndroid Build Coastguard Worker 
467*c8dee2aaSAndroid Build Coastguard Worker #if defined(GPU_TEST_UTILS)
468*c8dee2aaSAndroid Build Coastguard Worker 
469*c8dee2aaSAndroid Build Coastguard Worker /** Randomly divides subset into count divs. */
init_random_divs(int divs[],int count,int subsetStart,int subsetStop,SkRandom * random)470*c8dee2aaSAndroid Build Coastguard Worker static void init_random_divs(int divs[], int count, int subsetStart, int subsetStop,
471*c8dee2aaSAndroid Build Coastguard Worker                              SkRandom* random) {
472*c8dee2aaSAndroid Build Coastguard Worker     // Rules for lattice divs: Must be strictly increasing and in the range
473*c8dee2aaSAndroid Build Coastguard Worker     // [subsetStart, subsetStop.
474*c8dee2aaSAndroid Build Coastguard Worker     // Not terribly efficient alg for generating random divs:
475*c8dee2aaSAndroid Build Coastguard Worker     // 1 Start with minimum legal pixels between each div.
476*c8dee2aaSAndroid Build Coastguard Worker     // 2) Randomly assign the remaining pixels of the subset to divs.
477*c8dee2aaSAndroid Build Coastguard Worker     // 3) Convert from pixel counts to div offsets.
478*c8dee2aaSAndroid Build Coastguard Worker 
479*c8dee2aaSAndroid Build Coastguard Worker     // 1) Initially each divs[i] represents the number of pixels between
480*c8dee2aaSAndroid Build Coastguard Worker     // div i-1 and i. The initial div is allowed to be at subsetStart. There
481*c8dee2aaSAndroid Build Coastguard Worker     // must be one pixel spacing between subsequent divs.
482*c8dee2aaSAndroid Build Coastguard Worker     divs[0] = 0;
483*c8dee2aaSAndroid Build Coastguard Worker     for (int i = 1; i < count; ++i) {
484*c8dee2aaSAndroid Build Coastguard Worker         divs[i] = 1;
485*c8dee2aaSAndroid Build Coastguard Worker     }
486*c8dee2aaSAndroid Build Coastguard Worker     // 2) Assign the remaining subset pixels to fall
487*c8dee2aaSAndroid Build Coastguard Worker     int subsetLength = subsetStop - subsetStart;
488*c8dee2aaSAndroid Build Coastguard Worker     for (int i = 0; i < subsetLength - count; ++i) {
489*c8dee2aaSAndroid Build Coastguard Worker         // +1 because count divs means count+1 intervals.
490*c8dee2aaSAndroid Build Coastguard Worker         int entry = random->nextULessThan(count + 1);
491*c8dee2aaSAndroid Build Coastguard Worker         // We don't have an entry to  to store the count after the last div
492*c8dee2aaSAndroid Build Coastguard Worker         if (entry < count) {
493*c8dee2aaSAndroid Build Coastguard Worker             divs[entry]++;
494*c8dee2aaSAndroid Build Coastguard Worker         }
495*c8dee2aaSAndroid Build Coastguard Worker     }
496*c8dee2aaSAndroid Build Coastguard Worker     // 3) Now convert the counts between divs to pixel indices, incorporating the subset's offset.
497*c8dee2aaSAndroid Build Coastguard Worker     int offset = subsetStart;
498*c8dee2aaSAndroid Build Coastguard Worker     for (int i = 0; i < count; ++i) {
499*c8dee2aaSAndroid Build Coastguard Worker         divs[i] += offset;
500*c8dee2aaSAndroid Build Coastguard Worker         offset = divs[i];
501*c8dee2aaSAndroid Build Coastguard Worker     }
502*c8dee2aaSAndroid Build Coastguard Worker }
503*c8dee2aaSAndroid Build Coastguard Worker 
GR_DRAW_OP_TEST_DEFINE(NonAALatticeOp)504*c8dee2aaSAndroid Build Coastguard Worker GR_DRAW_OP_TEST_DEFINE(NonAALatticeOp) {
505*c8dee2aaSAndroid Build Coastguard Worker     SkCanvas::Lattice lattice;
506*c8dee2aaSAndroid Build Coastguard Worker     // We loop because our random lattice code can produce an invalid lattice in the case where
507*c8dee2aaSAndroid Build Coastguard Worker     // there is a single div separator in both x and y and both are aligned with the left and top
508*c8dee2aaSAndroid Build Coastguard Worker     // edge of the image subset, respectively.
509*c8dee2aaSAndroid Build Coastguard Worker     std::unique_ptr<int[]> xdivs;
510*c8dee2aaSAndroid Build Coastguard Worker     std::unique_ptr<int[]> ydivs;
511*c8dee2aaSAndroid Build Coastguard Worker     std::unique_ptr<SkCanvas::Lattice::RectType[]> flags;
512*c8dee2aaSAndroid Build Coastguard Worker     std::unique_ptr<SkColor[]> colors;
513*c8dee2aaSAndroid Build Coastguard Worker     SkIRect subset;
514*c8dee2aaSAndroid Build Coastguard Worker     SkISize dims;
515*c8dee2aaSAndroid Build Coastguard Worker     dims.fWidth = random->nextRangeU(1, 1000);
516*c8dee2aaSAndroid Build Coastguard Worker     dims.fHeight = random->nextRangeU(1, 1000);
517*c8dee2aaSAndroid Build Coastguard Worker     GrSurfaceOrigin origin = random->nextBool() ? kTopLeft_GrSurfaceOrigin
518*c8dee2aaSAndroid Build Coastguard Worker                                                 : kBottomLeft_GrSurfaceOrigin;
519*c8dee2aaSAndroid Build Coastguard Worker     const GrBackendFormat format =
520*c8dee2aaSAndroid Build Coastguard Worker             context->priv().caps()->getDefaultBackendFormat(GrColorType::kRGBA_8888,
521*c8dee2aaSAndroid Build Coastguard Worker                                                             GrRenderable::kNo);
522*c8dee2aaSAndroid Build Coastguard Worker 
523*c8dee2aaSAndroid Build Coastguard Worker     auto proxy = context->priv().proxyProvider()->createProxy(format,
524*c8dee2aaSAndroid Build Coastguard Worker                                                               dims,
525*c8dee2aaSAndroid Build Coastguard Worker                                                               GrRenderable::kNo,
526*c8dee2aaSAndroid Build Coastguard Worker                                                               1,
527*c8dee2aaSAndroid Build Coastguard Worker                                                               skgpu::Mipmapped::kNo,
528*c8dee2aaSAndroid Build Coastguard Worker                                                               SkBackingFit::kExact,
529*c8dee2aaSAndroid Build Coastguard Worker                                                               skgpu::Budgeted::kYes,
530*c8dee2aaSAndroid Build Coastguard Worker                                                               GrProtected::kNo,
531*c8dee2aaSAndroid Build Coastguard Worker                                                               /*label=*/"LatticeOp");
532*c8dee2aaSAndroid Build Coastguard Worker 
533*c8dee2aaSAndroid Build Coastguard Worker     do {
534*c8dee2aaSAndroid Build Coastguard Worker         if (random->nextBool()) {
535*c8dee2aaSAndroid Build Coastguard Worker             subset.fLeft = random->nextULessThan(dims.fWidth);
536*c8dee2aaSAndroid Build Coastguard Worker             subset.fRight = random->nextRangeU(subset.fLeft + 1, dims.fWidth);
537*c8dee2aaSAndroid Build Coastguard Worker             subset.fTop = random->nextULessThan(dims.fHeight);
538*c8dee2aaSAndroid Build Coastguard Worker             subset.fBottom = random->nextRangeU(subset.fTop + 1, dims.fHeight);
539*c8dee2aaSAndroid Build Coastguard Worker         } else {
540*c8dee2aaSAndroid Build Coastguard Worker             subset.setXYWH(0, 0, dims.fWidth, dims.fHeight);
541*c8dee2aaSAndroid Build Coastguard Worker         }
542*c8dee2aaSAndroid Build Coastguard Worker         // SkCanvas::Lattice allows bounds to be null. However, SkCanvas creates a temp Lattice with
543*c8dee2aaSAndroid Build Coastguard Worker         // a non-null bounds before creating a SkLatticeIter since SkLatticeIter requires a bounds.
544*c8dee2aaSAndroid Build Coastguard Worker         lattice.fBounds = &subset;
545*c8dee2aaSAndroid Build Coastguard Worker         lattice.fXCount = random->nextRangeU(1, subset.width());
546*c8dee2aaSAndroid Build Coastguard Worker         lattice.fYCount = random->nextRangeU(1, subset.height());
547*c8dee2aaSAndroid Build Coastguard Worker         xdivs.reset(new int[lattice.fXCount]);
548*c8dee2aaSAndroid Build Coastguard Worker         ydivs.reset(new int[lattice.fYCount]);
549*c8dee2aaSAndroid Build Coastguard Worker         init_random_divs(xdivs.get(), lattice.fXCount, subset.fLeft, subset.fRight, random);
550*c8dee2aaSAndroid Build Coastguard Worker         init_random_divs(ydivs.get(), lattice.fYCount, subset.fTop, subset.fBottom, random);
551*c8dee2aaSAndroid Build Coastguard Worker         lattice.fXDivs = xdivs.get();
552*c8dee2aaSAndroid Build Coastguard Worker         lattice.fYDivs = ydivs.get();
553*c8dee2aaSAndroid Build Coastguard Worker         bool hasFlags = random->nextBool();
554*c8dee2aaSAndroid Build Coastguard Worker         if (hasFlags) {
555*c8dee2aaSAndroid Build Coastguard Worker             int n = (lattice.fXCount + 1) * (lattice.fYCount + 1);
556*c8dee2aaSAndroid Build Coastguard Worker             flags.reset(new SkCanvas::Lattice::RectType[n]);
557*c8dee2aaSAndroid Build Coastguard Worker             colors.reset(new SkColor[n]);
558*c8dee2aaSAndroid Build Coastguard Worker             for (int i = 0; i < n; ++i) {
559*c8dee2aaSAndroid Build Coastguard Worker                 flags[i] = random->nextBool() ? SkCanvas::Lattice::kTransparent
560*c8dee2aaSAndroid Build Coastguard Worker                                               : SkCanvas::Lattice::kDefault;
561*c8dee2aaSAndroid Build Coastguard Worker             }
562*c8dee2aaSAndroid Build Coastguard Worker             lattice.fRectTypes = flags.get();
563*c8dee2aaSAndroid Build Coastguard Worker             lattice.fColors = colors.get();
564*c8dee2aaSAndroid Build Coastguard Worker         } else {
565*c8dee2aaSAndroid Build Coastguard Worker             lattice.fRectTypes = nullptr;
566*c8dee2aaSAndroid Build Coastguard Worker             lattice.fColors = nullptr;
567*c8dee2aaSAndroid Build Coastguard Worker         }
568*c8dee2aaSAndroid Build Coastguard Worker     } while (!SkLatticeIter::Valid(dims.fWidth, dims.fHeight, lattice));
569*c8dee2aaSAndroid Build Coastguard Worker     SkRect dst;
570*c8dee2aaSAndroid Build Coastguard Worker     dst.fLeft = random->nextRangeScalar(-2000.5f, 1000.f);
571*c8dee2aaSAndroid Build Coastguard Worker     dst.fTop = random->nextRangeScalar(-2000.5f, 1000.f);
572*c8dee2aaSAndroid Build Coastguard Worker     dst.fRight = dst.fLeft + random->nextRangeScalar(0.5f, 1000.f);
573*c8dee2aaSAndroid Build Coastguard Worker     dst.fBottom = dst.fTop + random->nextRangeScalar(0.5f, 1000.f);
574*c8dee2aaSAndroid Build Coastguard Worker     std::unique_ptr<SkLatticeIter> iter(new SkLatticeIter(lattice, dst));
575*c8dee2aaSAndroid Build Coastguard Worker     SkMatrix viewMatrix = GrTest::TestMatrixPreservesRightAngles(random);
576*c8dee2aaSAndroid Build Coastguard Worker     auto csxf = GrTest::TestColorXform(random);
577*c8dee2aaSAndroid Build Coastguard Worker     GrSamplerState::Filter filter =
578*c8dee2aaSAndroid Build Coastguard Worker             random->nextBool() ? GrSamplerState::Filter::kNearest : GrSamplerState::Filter::kLinear;
579*c8dee2aaSAndroid Build Coastguard Worker 
580*c8dee2aaSAndroid Build Coastguard Worker     GrSurfaceProxyView view(
581*c8dee2aaSAndroid Build Coastguard Worker             std::move(proxy), origin,
582*c8dee2aaSAndroid Build Coastguard Worker             context->priv().caps()->getReadSwizzle(format, GrColorType::kRGBA_8888));
583*c8dee2aaSAndroid Build Coastguard Worker 
584*c8dee2aaSAndroid Build Coastguard Worker     return skgpu::ganesh::LatticeOp::NonAALatticeOp::Make(context,
585*c8dee2aaSAndroid Build Coastguard Worker                                                           std::move(paint),
586*c8dee2aaSAndroid Build Coastguard Worker                                                           viewMatrix,
587*c8dee2aaSAndroid Build Coastguard Worker                                                           std::move(view),
588*c8dee2aaSAndroid Build Coastguard Worker                                                           kPremul_SkAlphaType,
589*c8dee2aaSAndroid Build Coastguard Worker                                                           std::move(csxf),
590*c8dee2aaSAndroid Build Coastguard Worker                                                           filter,
591*c8dee2aaSAndroid Build Coastguard Worker                                                           std::move(iter),
592*c8dee2aaSAndroid Build Coastguard Worker                                                           dst);
593*c8dee2aaSAndroid Build Coastguard Worker }
594*c8dee2aaSAndroid Build Coastguard Worker 
595*c8dee2aaSAndroid Build Coastguard Worker #endif
596