GrBatchPrototype

src/gpu/GrFlushToGpuDrawTarget.h

 /*
  * Copyright 2014 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #ifndef GrFlushToGpuDrawTarget_DEFINED
 #define GrFlushToGpuDrawTarget_DEFINED
 
@@ skipping 28 matching lines @@
      * before flushing and popped afterwards.
      */
     void flush();
 
     bool geometryHints(size_t vertexStride, int* vertexCount, int* indexCount) const SK_OVERRIDE;
 
 protected:
     GrGpu* getGpu() { return fGpu; }
     const GrGpu* getGpu() const{ return fGpu; }
 
-private:
+    GrVertexBufferAllocPool* getVertexAllocPool() { return fVertexPool; }
+    GrIndexBufferAllocPool* getIndexAllocPool() { return fIndexPool; }
+
+    // TODO all of this goes away when batch is everywhere
     enum {
         kGeoPoolStatePreAllocCnt = 4,
     };
 
     struct GeometryPoolState {
         const GrVertexBuffer*   fPoolVertexBuffer;
         int                     fPoolStartVertex;
         const GrIndexBuffer*    fPoolIndexBuffer;
         int                     fPoolStartIndex;
         // caller may conservatively over reserve vertices / indices.
         // we release unused space back to allocator if possible
         // can only do this if there isn't an intervening pushGeometrySource()
         size_t                  fUsedPoolVertexBytes;
         size_t                  fUsedPoolIndexBytes;
     };
 
     typedef SkSTArray<kGeoPoolStatePreAllocCnt, GeometryPoolState> GeoPoolStateStack;
+    const GeoPoolStateStack& getGeoPoolStateStack() const { return fGeoPoolStateStack; }
 
+private:
     virtual void onReset() = 0;
 
     virtual void onFlush() = 0;
 
     void setDrawBuffers(DrawInfo*, size_t stride) SK_OVERRIDE;
     bool onReserveVertexSpace(size_t vertexSize, int vertexCount, void** vertices) SK_OVERRIDE;
     bool onReserveIndexSpace(int indexCount, void** indices) SK_OVERRIDE;
     void releaseReservedVertexSpace() SK_OVERRIDE;
     void releaseReservedIndexSpace() SK_OVERRIDE;
     void geometrySourceWillPush() SK_OVERRIDE;
@@ skipping 10 matching lines @@
     GeoPoolStateStack                   fGeoPoolStateStack;
     SkAutoTUnref<GrGpu>                 fGpu;
     GrVertexBufferAllocPool*            fVertexPool;
     GrIndexBufferAllocPool*             fIndexPool;
     bool                                fFlushing;
 
     typedef GrClipTarget INHERITED;
 };
 
 #endif