Refugee from Dead Machine 4: MDB Monster

include/gpu/GrGpuResourceRef.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 GrGpuResourceRef_DEFINED
 #define GrGpuResourceRef_DEFINED
 
@@ skipping 21 matching lines @@
  *
  * TODO: Once GrDODrawState no longer exists and therefore GrDrawState and GrOptDrawState no
  * longer share an instance of this class, attempt to make the resource owned by GrGpuResourceRef
  * only settable via the constructor.
  */
 class GrGpuResourceRef : SkNoncopyable {
 public:
     ~GrGpuResourceRef();
 
     GrGpuResource* getResource() const { return fResource; }
+    GrGpuResource* getDst() const {
+        SkASSERT(fIOType == kRead_GrIOType || fIOType == kRW_GrIOType);
+        SkASSERT(fDst);
+        return fDst;
+    }
 
     /** Does this object own a pending read or write on the resource it is wrapping. */
     bool ownsPendingIO() const { return fPendingIO; }
 
     /** Shortcut for calling setResource() with NULL. It cannot be called after markingPendingIO
         is called. */
     void reset();
 
 protected:
     GrGpuResourceRef();
 
     /** Adopts a ref from the caller. ioType expresses what type of IO operations will be marked as
         pending on the resource when markPendingIO is called. */
-    GrGpuResourceRef(GrGpuResource*, GrIOType);
+    GrGpuResourceRef(GrGpuResource*, GrIOType, GrGpuResource* dst);
 
     /** Adopts a ref from the caller. ioType expresses what type of IO operations will be marked as
         pending on the resource when markPendingIO is called. */
-    void setResource(GrGpuResource*, GrIOType);
+    void setResource(GrGpuResource*, GrIOType, GrGpuResource* dst);
 
 private:
     /** Called by owning GrProgramElement when the program element is first scheduled for
         execution. It can only be called once. */
     void markPendingIO() const;
 
     /** Called when the program element/draw state is no longer owned by GrDrawTarget-client code.
         This lets the cache know that the drawing code will no longer schedule additional reads or
         writes to the resource using the program element or draw state. It can only be called once.
       */
     void removeRef() const;
 
     /** Called to indicate that the previous pending IO is complete. Useful when the owning object
         still has refs, so it is not about to destroy this GrGpuResourceRef, but its previously
         pending executions have been complete. Can only be called if removeRef() was not previously
         called. */
     void pendingIOComplete() const;
 
     friend class GrProgramElement;
 
     GrGpuResource*  fResource;
     mutable bool    fOwnRef;
     mutable bool    fPendingIO;
     GrIOType        fIOType;
+    GrGpuResource*  fDst;
 
     typedef SkNoncopyable INHERITED;
 };
 
 /**
  * Templated version of GrGpuResourceRef to enforce type safety.
  */
 template <typename T> class GrTGpuResourceRef : public GrGpuResourceRef {
 public:
     GrTGpuResourceRef() {}
 
     /** Adopts a ref from the caller. ioType expresses what type of IO operations will be marked as
         pending on the resource when markPendingIO is called. */
     GrTGpuResourceRef(T* resource, GrIOType ioType) : INHERITED(resource, ioType) { }
 
     T* get() const { return static_cast<T*>(this->getResource()); }
 
     /** Adopts a ref from the caller. ioType expresses what type of IO operations will be marked as
         pending on the resource when markPendingIO is called. */
-    void set(T* resource, GrIOType ioType) { this->setResource(resource, ioType); }
+    void set(T* resource, GrIOType ioType, T* dst) { this->setResource(resource, ioType, dst); }
 
 private:
     typedef GrGpuResourceRef INHERITED;
 };
 
 // Specializations for GrTexture and GrRenderTarget because they use virtual inheritance.
 template<> class GrTGpuResourceRef<GrTexture> : public GrGpuResourceRef {
 public:
     GrTGpuResourceRef() {}
 
-    GrTGpuResourceRef(GrTexture* texture, GrIOType ioType) : INHERITED(texture, ioType) { }
+    GrTGpuResourceRef(GrTexture* texture, GrIOType ioType, GrRenderTarget* dst) : INHERITED(texture, ioType, dst) { }
 
     GrTexture* get() const {
         GrSurface* surface = static_cast<GrSurface*>(this->getResource());
         if (surface) {
             return surface->asTexture();
         } else {
             return NULL;
         }
     }
 
-    void set(GrTexture* texture, GrIOType ioType) { this->setResource(texture, ioType); }
+    void set(GrTexture* texture, GrIOType ioType, GrRenderTarget* dst) { this->setResource(texture, ioType, dst); }
 
 private:
     typedef GrGpuResourceRef INHERITED;
 };
 
 template<> class GrTGpuResourceRef<GrRenderTarget> : public GrGpuResourceRef {
 public:
     GrTGpuResourceRef() {}
 
-    GrTGpuResourceRef(GrRenderTarget* rt, GrIOType ioType) : INHERITED(rt, ioType) { }
+    GrTGpuResourceRef(GrRenderTarget* rt, GrIOType ioType, GrRenderTarget* dst) : INHERITED(rt, ioType, dst) { }
 
     GrRenderTarget* get() const {
         GrSurface* surface = static_cast<GrSurface*>(this->getResource());
         if (surface) {
             return surface->asRenderTarget();
         } else {
             return NULL;
         }
     }
 
-    void set(GrRenderTarget* rt, GrIOType ioType) { this->setResource(rt, ioType); }
+    void set(GrRenderTarget* rt, GrIOType ioType, GrRenderTarget* dst) { this->setResource(rt, ioType, dst); }
 
 private:
     typedef GrGpuResourceRef INHERITED;
 };
 
 /**
  * This is similar to GrTGpuResourceRef but can only be in the pending IO state. It never owns a
  * ref.
  */
 template <typename T, GrIOType IO_TYPE> class GrPendingIOResource : SkNoncopyable {
 public:
-    GrPendingIOResource(T* resource = NULL) : fResource(NULL) {
-        this->reset(resource);
+    GrPendingIOResource(T* resource = NULL, GrGpuResource* dst = NULL) : fResource(NULL) {
+        this->reset(resource, dst);
     }
 
-    void reset(T* resource) {
+    void reset(T* resource, GrGpuResource* dst) {
         if (resource) {
             switch (IO_TYPE) {
                 case kRead_GrIOType:
-                    resource->addPendingRead();
+                    resource->addPendingRead(dst);
                     break;
                 case kWrite_GrIOType:
                     resource->addPendingWrite();
                     break;
                 case kRW_GrIOType:
-                    resource->addPendingRead();
+                    resource->addPendingRead(dst);
                     resource->addPendingWrite();
                     break;
             }
         }
         this->release();
         fResource = resource;
     }
 
     ~GrPendingIOResource() {
         this->release();
@@ skipping 21 matching lines @@
                     fResource->completedRead();
                     fResource->completedWrite();
                     break;
             }
         }
     }
 
     T* fResource;
 };
 #endif