| /* |
| * Copyright 2018 Google Inc. |
| * |
| * Use of this source code is governed by a BSD-style license that can be |
| * found in the LICENSE file. |
| */ |
| |
| #include "gm/gm.h" |
| #include "include/core/SkBlendMode.h" |
| #include "include/core/SkCanvas.h" |
| #include "include/core/SkColor.h" |
| #include "include/core/SkMatrix.h" |
| #include "include/core/SkPoint.h" |
| #include "include/core/SkRect.h" |
| #include "include/core/SkRefCnt.h" |
| #include "include/core/SkString.h" |
| #include "include/gpu/GrRecordingContext.h" |
| #include "include/private/gpu/ganesh/GrTypesPriv.h" |
| #include "src/core/SkCanvasPriv.h" |
| #include "src/gpu/ganesh/GrBuffer.h" |
| #include "src/gpu/ganesh/GrCanvas.h" |
| #include "src/gpu/ganesh/GrCaps.h" |
| #include "src/gpu/ganesh/GrDirectContextPriv.h" |
| #include "src/gpu/ganesh/GrGeometryProcessor.h" |
| #include "src/gpu/ganesh/GrGpuBuffer.h" |
| #include "src/gpu/ganesh/GrMemoryPool.h" |
| #include "src/gpu/ganesh/GrOpFlushState.h" |
| #include "src/gpu/ganesh/GrOpsRenderPass.h" |
| #include "src/gpu/ganesh/GrPipeline.h" |
| #include "src/gpu/ganesh/GrProcessor.h" |
| #include "src/gpu/ganesh/GrProcessorSet.h" |
| #include "src/gpu/ganesh/GrProgramInfo.h" |
| #include "src/gpu/ganesh/GrRecordingContextPriv.h" |
| #include "src/gpu/ganesh/GrResourceProvider.h" |
| #include "src/gpu/ganesh/GrShaderCaps.h" |
| #include "src/gpu/ganesh/GrShaderVar.h" |
| #include "src/gpu/ganesh/SurfaceDrawContext.h" |
| #include "src/gpu/ganesh/glsl/GrGLSLFragmentShaderBuilder.h" |
| #include "src/gpu/ganesh/glsl/GrGLSLProgramDataManager.h" |
| #include "src/gpu/ganesh/glsl/GrGLSLUniformHandler.h" |
| #include "src/gpu/ganesh/glsl/GrGLSLVarying.h" |
| #include "src/gpu/ganesh/glsl/GrGLSLVertexGeoBuilder.h" |
| #include "src/gpu/ganesh/ops/GrDrawOp.h" |
| #include "src/gpu/ganesh/ops/GrOp.h" |
| #include "tools/gpu/ProxyUtils.h" |
| |
| #include <memory> |
| #include <utility> |
| |
| class GrAppliedClip; |
| |
| /** |
| * This test ensures that fwidth() works properly on GPU configs by drawing a squircle. |
| */ |
| namespace { |
| |
| static constexpr GrGeometryProcessor::Attribute gVertex = |
| {"bboxcoord", kFloat2_GrVertexAttribType, SkSLType::kFloat2}; |
| |
| //////////////////////////////////////////////////////////////////////////////////////////////////// |
| // SkSL code. |
| |
| class FwidthSquircleTestProcessor : public GrGeometryProcessor { |
| public: |
| static GrGeometryProcessor* Make(SkArenaAlloc* arena, const SkMatrix& viewMatrix) { |
| return arena->make([&](void* ptr) { |
| return new (ptr) FwidthSquircleTestProcessor(viewMatrix); |
| }); |
| } |
| |
| const char* name() const override { return "FwidthSquircleTestProcessor"; } |
| |
| void addToKey(const GrShaderCaps&, skgpu::KeyBuilder*) const final {} |
| |
| std::unique_ptr<ProgramImpl> makeProgramImpl(const GrShaderCaps&) const final; |
| |
| private: |
| FwidthSquircleTestProcessor(const SkMatrix& viewMatrix) |
| : GrGeometryProcessor(kFwidthSquircleTestProcessor_ClassID) |
| , fViewMatrix(viewMatrix) { |
| this->setVertexAttributesWithImplicitOffsets(&gVertex, 1); |
| } |
| |
| const SkMatrix fViewMatrix; |
| |
| using INHERITED = GrGeometryProcessor; |
| }; |
| |
| std::unique_ptr<GrGeometryProcessor::ProgramImpl> FwidthSquircleTestProcessor::makeProgramImpl( |
| const GrShaderCaps&) const { |
| class Impl : public ProgramImpl { |
| public: |
| void setData(const GrGLSLProgramDataManager& pdman, |
| const GrShaderCaps&, |
| const GrGeometryProcessor& geomProc) override { |
| const auto& proc = geomProc.cast<FwidthSquircleTestProcessor>(); |
| pdman.setSkMatrix(fViewMatrixHandle, proc.fViewMatrix); |
| } |
| |
| private: |
| void onEmitCode(EmitArgs& args, GrGPArgs* gpArgs) override { |
| const auto& proc = args.fGeomProc.cast<FwidthSquircleTestProcessor>(); |
| |
| auto* uniforms = args.fUniformHandler; |
| fViewMatrixHandle = uniforms->addUniform(nullptr, |
| kVertex_GrShaderFlag, |
| SkSLType::kFloat3x3, |
| "viewmatrix"); |
| |
| auto* varyings = args.fVaryingHandler; |
| varyings->emitAttributes(proc); |
| |
| GrGLSLVarying squircleCoord(SkSLType::kFloat2); |
| varyings->addVarying("bboxcoord", &squircleCoord); |
| |
| auto* v = args.fVertBuilder; |
| v->codeAppendf("float2x2 R = float2x2(cos(.05), sin(.05), -sin(.05), cos(.05));"); |
| |
| v->codeAppendf("%s = bboxcoord * 1.25;", squircleCoord.vsOut()); |
| v->codeAppendf("float3 vertexpos = float3(bboxcoord * 100 * R + 100, 1);"); |
| v->codeAppendf("vertexpos = %s * vertexpos;", |
| uniforms->getUniformCStr(fViewMatrixHandle)); |
| gpArgs->fPositionVar.set(SkSLType::kFloat3, "vertexpos"); |
| |
| auto* f = args.fFragBuilder; |
| f->codeAppendf("float golden_ratio = 1.61803398875;"); |
| f->codeAppendf("float pi = 3.141592653589793;"); |
| f->codeAppendf("float x = abs(%s.x), y = abs(%s.y);", |
| squircleCoord.fsIn(), squircleCoord.fsIn()); |
| |
| // Squircle function! |
| f->codeAppendf("float fn = half(pow(x, golden_ratio*pi) + " |
| "pow(y, golden_ratio*pi) - 1);"); |
| f->codeAppendf("float fnwidth = fwidth(fn);"); |
| f->codeAppendf("fnwidth += 1e-10;"); // Guard against divide-by-zero. |
| f->codeAppendf("half coverage = clamp(half(.5 - fn/fnwidth), 0, 1);"); |
| |
| f->codeAppendf("half4 %s = half4(.51, .42, .71, 1) * .89;", args.fOutputColor); |
| f->codeAppendf("half4 %s = half4(coverage);", args.fOutputCoverage); |
| } |
| |
| UniformHandle fViewMatrixHandle; |
| }; |
| |
| return std::make_unique<Impl>(); |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////////////////////////// |
| // Draw Op. |
| |
| class FwidthSquircleTestOp : public GrDrawOp { |
| public: |
| DEFINE_OP_CLASS_ID |
| |
| static GrOp::Owner Make(GrRecordingContext* ctx, const SkMatrix& viewMatrix) { |
| return GrOp::Make<FwidthSquircleTestOp>(ctx, viewMatrix); |
| } |
| |
| private: |
| FwidthSquircleTestOp(const SkMatrix& viewMatrix) |
| : GrDrawOp(ClassID()) |
| , fViewMatrix(viewMatrix) { |
| this->setBounds(SkRect::MakeIWH(kWidth, kHeight), HasAABloat::kNo, IsHairline::kNo); |
| } |
| |
| const char* name() const override { return "FwidthSquircleTestOp"; } |
| FixedFunctionFlags fixedFunctionFlags() const override { return FixedFunctionFlags::kNone; } |
| GrProcessorSet::Analysis finalize(const GrCaps&, const GrAppliedClip*, GrClampType) override { |
| return GrProcessorSet::EmptySetAnalysis(); |
| } |
| |
| GrProgramInfo* createProgramInfo(const GrCaps* caps, |
| SkArenaAlloc* arena, |
| const GrSurfaceProxyView& writeView, |
| bool usesMSAASurface, |
| GrAppliedClip&& appliedClip, |
| const GrDstProxyView& dstProxyView, |
| GrXferBarrierFlags renderPassXferBarriers, |
| GrLoadOp colorLoadOp) const { |
| GrGeometryProcessor* geomProc = FwidthSquircleTestProcessor::Make(arena, fViewMatrix); |
| |
| return sk_gpu_test::CreateProgramInfo(caps, arena, writeView, usesMSAASurface, |
| std::move(appliedClip), dstProxyView, |
| geomProc, SkBlendMode::kSrcOver, |
| GrPrimitiveType::kTriangleStrip, |
| renderPassXferBarriers, colorLoadOp); |
| } |
| |
| GrProgramInfo* createProgramInfo(GrOpFlushState* flushState) const { |
| return this->createProgramInfo(&flushState->caps(), |
| flushState->allocator(), |
| flushState->writeView(), |
| flushState->usesMSAASurface(), |
| flushState->detachAppliedClip(), |
| flushState->dstProxyView(), |
| flushState->renderPassBarriers(), |
| flushState->colorLoadOp()); |
| } |
| |
| void onPrePrepare(GrRecordingContext* context, |
| const GrSurfaceProxyView& writeView, |
| GrAppliedClip* clip, |
| const GrDstProxyView& dstProxyView, |
| GrXferBarrierFlags renderPassXferBarriers, |
| GrLoadOp colorLoadOp) final { |
| SkArenaAlloc* arena = context->priv().recordTimeAllocator(); |
| |
| // DMSAA is not supported on DDL. |
| bool usesMSAASurface = writeView.asRenderTargetProxy()->numSamples() > 1; |
| |
| // This is equivalent to a GrOpFlushState::detachAppliedClip |
| GrAppliedClip appliedClip = clip ? std::move(*clip) : GrAppliedClip::Disabled(); |
| |
| fProgramInfo = this->createProgramInfo(context->priv().caps(), arena, writeView, |
| usesMSAASurface, std::move(appliedClip), |
| dstProxyView, renderPassXferBarriers, colorLoadOp); |
| |
| context->priv().recordProgramInfo(fProgramInfo); |
| } |
| |
| void onPrepare(GrOpFlushState* flushState) final { |
| SkPoint vertices[4] = { |
| {-1, -1}, |
| {+1, -1}, |
| {-1, +1}, |
| {+1, +1}, |
| }; |
| fVertexBuffer = flushState->resourceProvider()->createBuffer(vertices, |
| sizeof(vertices), |
| GrGpuBufferType::kVertex, |
| kStatic_GrAccessPattern); |
| } |
| |
| void onExecute(GrOpFlushState* flushState, const SkRect& chainBounds) final { |
| if (!fVertexBuffer) { |
| return; |
| } |
| |
| if (!fProgramInfo) { |
| fProgramInfo = this->createProgramInfo(flushState); |
| } |
| |
| flushState->bindPipeline(*fProgramInfo, SkRect::MakeIWH(kWidth, kHeight)); |
| flushState->bindBuffers(nullptr, nullptr, std::move(fVertexBuffer)); |
| flushState->draw(4, 0); |
| |
| } |
| |
| static const int kWidth = 200; |
| static const int kHeight = 200; |
| |
| sk_sp<GrBuffer> fVertexBuffer; |
| const SkMatrix fViewMatrix; |
| |
| // The program info (and both the GrPipeline and GrGeometryProcessor it relies on), when |
| // allocated, are allocated in either the ddl-record-time or flush-time arena. It is the |
| // arena's job to free up their memory so we just have a bare programInfo pointer here. We |
| // don't even store the GrPipeline and GrGeometryProcessor pointers here bc they are |
| // guaranteed to have the same lifetime as the program info. |
| GrProgramInfo* fProgramInfo = nullptr; |
| |
| friend class ::GrOp; // for ctor |
| |
| using INHERITED = GrDrawOp; |
| }; |
| |
| } // namespace |
| |
| //////////////////////////////////////////////////////////////////////////////////////////////////// |
| // Test. |
| |
| namespace skiagm { |
| |
| DEF_SIMPLE_GPU_GM_CAN_FAIL(fwidth_squircle, rContext, canvas, errorMsg, 200, 200) { |
| if (!rContext->priv().caps()->shaderCaps()->fShaderDerivativeSupport) { |
| *errorMsg = "Shader derivatives not supported."; |
| return DrawResult::kSkip; |
| } |
| |
| auto sdc = skgpu::ganesh::TopDeviceSurfaceDrawContext(canvas); |
| if (!sdc) { |
| *errorMsg = GM::kErrorMsg_DrawSkippedGpuOnly; |
| return DrawResult::kSkip; |
| } |
| |
| // Draw the test directly to the frame buffer. |
| canvas->clear(SK_ColorWHITE); |
| sdc->addDrawOp(FwidthSquircleTestOp::Make(rContext, canvas->getTotalMatrix())); |
| return skiagm::DrawResult::kOk; |
| } |
| |
| } // namespace skiagm |