| /* |
| * Copyright 2015 Google Inc. |
| * |
| * Use of this source code is governed by a BSD-style license that can be |
| * found in the LICENSE file. |
| */ |
| |
| #ifndef GrOpFlushState_DEFINED |
| #define GrOpFlushState_DEFINED |
| |
| #include <utility> |
| #include "src/base/SkArenaAlloc.h" |
| #include "src/base/SkArenaAllocList.h" |
| #include "src/gpu/ganesh/GrAppliedClip.h" |
| #include "src/gpu/ganesh/GrBufferAllocPool.h" |
| #include "src/gpu/ganesh/GrDeferredUpload.h" |
| #include "src/gpu/ganesh/GrMeshDrawTarget.h" |
| #include "src/gpu/ganesh/GrProgramInfo.h" |
| #include "src/gpu/ganesh/GrRenderTargetProxy.h" |
| #include "src/gpu/ganesh/GrSurfaceProxyView.h" |
| |
| class GrGpu; |
| class GrOpsRenderPass; |
| class GrResourceProvider; |
| |
| /** Tracks the state across all the GrOps (really just the GrDrawOps) in a OpsTask flush. */ |
| class GrOpFlushState final : public GrDeferredUploadTarget, public GrMeshDrawTarget { |
| public: |
| // vertexSpace and indexSpace may either be null or an alloation of size |
| // GrBufferAllocPool::kDefaultBufferSize. If the latter, then CPU memory is only allocated for |
| // vertices/indices when a buffer larger than kDefaultBufferSize is required. |
| GrOpFlushState(GrGpu*, GrResourceProvider*, skgpu::TokenTracker*, |
| sk_sp<GrBufferAllocPool::CpuBufferCache> = nullptr); |
| |
| ~GrOpFlushState() final { this->reset(); } |
| |
| /** This is called after each op has a chance to prepare its draws and before the draws are |
| executed. */ |
| void preExecuteDraws(); |
| |
| /** Called to upload data to a texture using the GrDeferredTextureUploadFn. If the uploaded |
| surface needs to be prepared for being sampled in a draw after the upload, the caller |
| should pass in true for shouldPrepareSurfaceForSampling. This feature is needed for Vulkan |
| when doing inline uploads to reset the image layout back to sampled. */ |
| void doUpload(GrDeferredTextureUploadFn&, bool shouldPrepareSurfaceForSampling = false); |
| |
| /** Called as ops are executed. Must be called in the same order as the ops were prepared. */ |
| void executeDrawsAndUploadsForMeshDrawOp(const GrOp* op, const SkRect& chainBounds, |
| const GrPipeline*, const GrUserStencilSettings*); |
| |
| GrOpsRenderPass* opsRenderPass() { return fOpsRenderPass; } |
| void setOpsRenderPass(GrOpsRenderPass* renderPass) { fOpsRenderPass = renderPass; } |
| |
| GrGpu* gpu() { return fGpu; } |
| |
| void reset(); |
| |
| /** Additional data required on a per-op basis when executing GrOps. */ |
| struct OpArgs { |
| // TODO: why does OpArgs have the op we're going to pass it to as a member? Remove it. |
| explicit OpArgs(GrOp* op, const GrSurfaceProxyView& surfaceView, bool usesMSAASurface, |
| GrAppliedClip* appliedClip, const GrDstProxyView& dstProxyView, |
| GrXferBarrierFlags renderPassXferBarriers, GrLoadOp colorLoadOp) |
| : fOp(op) |
| , fSurfaceView(surfaceView) |
| , fRenderTargetProxy(surfaceView.asRenderTargetProxy()) |
| , fUsesMSAASurface(usesMSAASurface) |
| , fAppliedClip(appliedClip) |
| , fDstProxyView(dstProxyView) |
| , fRenderPassXferBarriers(renderPassXferBarriers) |
| , fColorLoadOp(colorLoadOp) { |
| SkASSERT(surfaceView.asRenderTargetProxy()); |
| } |
| |
| GrOp* op() { return fOp; } |
| const GrSurfaceProxyView& writeView() const { return fSurfaceView; } |
| GrRenderTargetProxy* rtProxy() const { return fRenderTargetProxy; } |
| // True if the op under consideration belongs to an opsTask that renders to an MSAA buffer. |
| bool usesMSAASurface() const { return fUsesMSAASurface; } |
| GrAppliedClip* appliedClip() { return fAppliedClip; } |
| const GrAppliedClip* appliedClip() const { return fAppliedClip; } |
| const GrDstProxyView& dstProxyView() const { return fDstProxyView; } |
| GrXferBarrierFlags renderPassBarriers() const { return fRenderPassXferBarriers; } |
| GrLoadOp colorLoadOp() const { return fColorLoadOp; } |
| |
| #ifdef SK_DEBUG |
| void validate() const { |
| SkASSERT(fOp); |
| SkASSERT(fSurfaceView); |
| } |
| #endif |
| |
| private: |
| GrOp* fOp; |
| const GrSurfaceProxyView& fSurfaceView; |
| GrRenderTargetProxy* fRenderTargetProxy; |
| bool fUsesMSAASurface; |
| GrAppliedClip* fAppliedClip; |
| GrDstProxyView fDstProxyView; // TODO: do we still need the dst proxy here? |
| GrXferBarrierFlags fRenderPassXferBarriers; |
| GrLoadOp fColorLoadOp; |
| }; |
| |
| void setOpArgs(OpArgs* opArgs) { fOpArgs = opArgs; } |
| |
| const OpArgs& drawOpArgs() const { |
| SkASSERT(fOpArgs); |
| SkDEBUGCODE(fOpArgs->validate()); |
| return *fOpArgs; |
| } |
| |
| void setSampledProxyArray(SkTArray<GrSurfaceProxy*, true>* sampledProxies) { |
| fSampledProxies = sampledProxies; |
| } |
| |
| SkTArray<GrSurfaceProxy*, true>* sampledProxyArray() override { |
| return fSampledProxies; |
| } |
| |
| /** Overrides of GrDeferredUploadTarget. */ |
| |
| const skgpu::TokenTracker* tokenTracker() final { return fTokenTracker; } |
| skgpu::AtlasToken addInlineUpload(GrDeferredTextureUploadFn&&) final; |
| skgpu::AtlasToken addASAPUpload(GrDeferredTextureUploadFn&&) final; |
| |
| /** Overrides of GrMeshDrawTarget. */ |
| void recordDraw(const GrGeometryProcessor*, |
| const GrSimpleMesh[], |
| int meshCnt, |
| const GrSurfaceProxy* const primProcProxies[], |
| GrPrimitiveType) final; |
| void* makeVertexSpace(size_t vertexSize, int vertexCount, sk_sp<const GrBuffer>*, |
| int* startVertex) final; |
| uint16_t* makeIndexSpace(int indexCount, sk_sp<const GrBuffer>*, int* startIndex) final; |
| void* makeVertexSpaceAtLeast(size_t vertexSize, int minVertexCount, int fallbackVertexCount, |
| sk_sp<const GrBuffer>*, int* startVertex, |
| int* actualVertexCount) final; |
| uint16_t* makeIndexSpaceAtLeast(int minIndexCount, int fallbackIndexCount, |
| sk_sp<const GrBuffer>*, int* startIndex, |
| int* actualIndexCount) final; |
| GrDrawIndirectWriter makeDrawIndirectSpace(int drawCount, sk_sp<const GrBuffer>* buffer, |
| size_t* offset) override { |
| return fDrawIndirectPool.makeSpace(drawCount, buffer, offset); |
| } |
| GrDrawIndexedIndirectWriter makeDrawIndexedIndirectSpace(int drawCount, |
| sk_sp<const GrBuffer>* buffer, |
| size_t* offset) override { |
| return fDrawIndirectPool.makeIndexedSpace(drawCount, buffer, offset); |
| } |
| void putBackIndices(int indexCount) final; |
| void putBackVertices(int vertices, size_t vertexStride) final; |
| void putBackIndirectDraws(int drawCount) final { fDrawIndirectPool.putBack(drawCount); } |
| void putBackIndexedIndirectDraws(int drawCount) final { |
| fDrawIndirectPool.putBackIndexed(drawCount); |
| } |
| const GrSurfaceProxyView& writeView() const final { return this->drawOpArgs().writeView(); } |
| GrRenderTargetProxy* rtProxy() const final { return this->drawOpArgs().rtProxy(); } |
| bool usesMSAASurface() const final { return this->drawOpArgs().usesMSAASurface(); } |
| const GrAppliedClip* appliedClip() const final { return this->drawOpArgs().appliedClip(); } |
| const GrAppliedHardClip& appliedHardClip() const { |
| return (fOpArgs->appliedClip()) ? |
| fOpArgs->appliedClip()->hardClip() : GrAppliedHardClip::Disabled(); |
| } |
| GrAppliedClip detachAppliedClip() final; |
| const GrDstProxyView& dstProxyView() const final { |
| return this->drawOpArgs().dstProxyView(); |
| } |
| |
| GrXferBarrierFlags renderPassBarriers() const final { |
| return this->drawOpArgs().renderPassBarriers(); |
| } |
| |
| GrLoadOp colorLoadOp() const final { |
| return this->drawOpArgs().colorLoadOp(); |
| } |
| |
| GrDeferredUploadTarget* deferredUploadTarget() final { return this; } |
| const GrCaps& caps() const final; |
| GrThreadSafeCache* threadSafeCache() const final; |
| GrResourceProvider* resourceProvider() const final { return fResourceProvider; } |
| |
| sktext::gpu::StrikeCache* strikeCache() const final; |
| |
| // At this point we know we're flushing so full access to the GrAtlasManager and |
| // SmallPathAtlasMgr is required (and permissible). |
| GrAtlasManager* atlasManager() const final; |
| #if !defined(SK_ENABLE_OPTIMIZE_SIZE) |
| skgpu::v1::SmallPathAtlasMgr* smallPathAtlasManager() const final; |
| #endif |
| |
| /** GrMeshDrawTarget override. */ |
| SkArenaAlloc* allocator() override { return &fArena; } |
| |
| // This is a convenience method that binds the given pipeline, and then, if our applied clip has |
| // a scissor, sets the scissor rect from the applied clip. |
| void bindPipelineAndScissorClip(const GrProgramInfo& programInfo, const SkRect& drawBounds) { |
| SkASSERT((programInfo.pipeline().isScissorTestEnabled()) == |
| (this->appliedClip() && this->appliedClip()->scissorState().enabled())); |
| this->bindPipeline(programInfo, drawBounds); |
| if (programInfo.pipeline().isScissorTestEnabled()) { |
| this->setScissorRect(this->appliedClip()->scissorState().rect()); |
| } |
| } |
| |
| // This is a convenience method for when the primitive processor has exactly one texture. It |
| // binds one texture for the primitive processor, and any others for FPs on the pipeline. |
| void bindTextures(const GrGeometryProcessor& geomProc, |
| const GrSurfaceProxy& singleGeomProcTexture, |
| const GrPipeline& pipeline) { |
| SkASSERT(geomProc.numTextureSamplers() == 1); |
| const GrSurfaceProxy* ptr = &singleGeomProcTexture; |
| this->bindTextures(geomProc, &ptr, pipeline); |
| } |
| |
| // Makes the appropriate bindBuffers() and draw*() calls for the provided mesh. |
| void drawMesh(const GrSimpleMesh& mesh); |
| |
| // Pass-through methods to GrOpsRenderPass. |
| void bindPipeline(const GrProgramInfo& programInfo, const SkRect& drawBounds) { |
| fOpsRenderPass->bindPipeline(programInfo, drawBounds); |
| } |
| void setScissorRect(const SkIRect& scissorRect) { |
| fOpsRenderPass->setScissorRect(scissorRect); |
| } |
| void bindTextures(const GrGeometryProcessor& geomProc, |
| const GrSurfaceProxy* const geomProcTextures[], |
| const GrPipeline& pipeline) { |
| fOpsRenderPass->bindTextures(geomProc, geomProcTextures, pipeline); |
| } |
| void bindBuffers(sk_sp<const GrBuffer> indexBuffer, sk_sp<const GrBuffer> instanceBuffer, |
| sk_sp<const GrBuffer> vertexBuffer, |
| GrPrimitiveRestart primitiveRestart = GrPrimitiveRestart::kNo) { |
| fOpsRenderPass->bindBuffers(std::move(indexBuffer), std::move(instanceBuffer), |
| std::move(vertexBuffer), primitiveRestart); |
| } |
| void draw(int vertexCount, int baseVertex) { |
| fOpsRenderPass->draw(vertexCount, baseVertex); |
| } |
| void drawIndexed(int indexCount, int baseIndex, uint16_t minIndexValue, uint16_t maxIndexValue, |
| int baseVertex) { |
| fOpsRenderPass->drawIndexed(indexCount, baseIndex, minIndexValue, maxIndexValue, |
| baseVertex); |
| } |
| void drawInstanced(int instanceCount, int baseInstance, int vertexCount, int baseVertex) { |
| fOpsRenderPass->drawInstanced(instanceCount, baseInstance, vertexCount, baseVertex); |
| } |
| void drawIndexedInstanced(int indexCount, int baseIndex, int instanceCount, int baseInstance, |
| int baseVertex) { |
| fOpsRenderPass->drawIndexedInstanced(indexCount, baseIndex, instanceCount, baseInstance, |
| baseVertex); |
| } |
| void drawIndirect(const GrBuffer* drawIndirectBuffer, size_t offset, int drawCount) { |
| fOpsRenderPass->drawIndirect(drawIndirectBuffer, offset, drawCount); |
| } |
| void drawIndexedIndirect(const GrBuffer* drawIndirectBuffer, size_t offset, int drawCount) { |
| fOpsRenderPass->drawIndexedIndirect(drawIndirectBuffer, offset, drawCount); |
| } |
| void drawIndexPattern(int patternIndexCount, int patternRepeatCount, |
| int maxPatternRepetitionsInIndexBuffer, int patternVertexCount, |
| int baseVertex) { |
| fOpsRenderPass->drawIndexPattern(patternIndexCount, patternRepeatCount, |
| maxPatternRepetitionsInIndexBuffer, patternVertexCount, |
| baseVertex); |
| } |
| |
| private: |
| struct InlineUpload { |
| InlineUpload(GrDeferredTextureUploadFn&& upload, skgpu::AtlasToken token) |
| : fUpload(std::move(upload)), fUploadBeforeToken(token) {} |
| GrDeferredTextureUploadFn fUpload; |
| skgpu::AtlasToken fUploadBeforeToken; |
| }; |
| |
| // A set of contiguous draws that share a draw token, geometry processor, and pipeline. The |
| // meshes for the draw are stored in the fMeshes array. The reason for coalescing meshes |
| // that share a geometry processor into a Draw is that it allows the Gpu object to setup |
| // the shared state once and then issue draws for each mesh. |
| struct Draw { |
| ~Draw(); |
| // The geometry processor is always forced to be in an arena allocation. This object does |
| // not need to manage its lifetime. |
| const GrGeometryProcessor* fGeometryProcessor = nullptr; |
| // Must have GrGeometryProcessor::numTextureSamplers() entries. Can be null if no samplers. |
| const GrSurfaceProxy* const* fGeomProcProxies = nullptr; |
| const GrSimpleMesh* fMeshes = nullptr; |
| const GrOp* fOp = nullptr; |
| int fMeshCnt = 0; |
| GrPrimitiveType fPrimitiveType; |
| }; |
| |
| // Storage for ops' pipelines, draws, and inline uploads. |
| SkArenaAllocWithReset fArena{sizeof(GrPipeline) * 100}; |
| |
| // Store vertex and index data on behalf of ops that are flushed. |
| GrVertexBufferAllocPool fVertexPool; |
| GrIndexBufferAllocPool fIndexPool; |
| GrDrawIndirectBufferAllocPool fDrawIndirectPool; |
| |
| // Data stored on behalf of the ops being flushed. |
| SkArenaAllocList<GrDeferredTextureUploadFn> fASAPUploads; |
| SkArenaAllocList<InlineUpload> fInlineUploads; |
| SkArenaAllocList<Draw> fDraws; |
| |
| // All draws we store have an implicit draw token. This is the draw token for the first draw |
| // in fDraws. |
| skgpu::AtlasToken fBaseDrawToken = skgpu::AtlasToken::InvalidToken(); |
| |
| // Info about the op that is currently preparing or executing using the flush state or null if |
| // an op is not currently preparing of executing. |
| OpArgs* fOpArgs = nullptr; |
| |
| // This field is only transiently set during flush. Each OpsTask will set it to point to an |
| // array of proxies it uses before call onPrepare and onExecute. |
| SkTArray<GrSurfaceProxy*, true>* fSampledProxies; |
| |
| GrGpu* fGpu; |
| GrResourceProvider* fResourceProvider; |
| skgpu::TokenTracker* fTokenTracker; |
| GrOpsRenderPass* fOpsRenderPass = nullptr; |
| |
| // Variables that are used to track where we are in lists as ops are executed |
| SkArenaAllocList<Draw>::Iter fCurrDraw; |
| SkArenaAllocList<InlineUpload>::Iter fCurrUpload; |
| }; |
| |
| #endif |