Refugee from Dead Machine 4: MDB Monster

src/gpu/GrGpuResourceRef.cpp

 /*
  * Copyright 2014 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "GrGpuResourceRef.h"
 
 GrGpuResourceRef::GrGpuResourceRef() {
     fResource = nullptr;
     fOwnRef = false;
     fPendingIO = false;
+    fDst = NULL;
 }
 
-GrGpuResourceRef::GrGpuResourceRef(GrGpuResource* resource, GrIOType ioType) {
+GrGpuResourceRef::GrGpuResourceRef(GrGpuResource* resource, GrIOType ioType, GrGpuResource* dst) {
     fResource = nullptr;
     fOwnRef = false;
     fPendingIO = false;
-    this->setResource(resource, ioType);
+    fDst = NULL;
+    this->setResource(resource, ioType, dst);
 }
 
 GrGpuResourceRef::~GrGpuResourceRef() {
     if (fOwnRef) {
         SkASSERT(fResource);
         fResource->unref();
     }
     if (fPendingIO) {
         switch (fIOType) {
             case kRead_GrIOType:
@@ skipping 11 matching lines @@
 }
 
 void GrGpuResourceRef::reset() {
     SkASSERT(!fPendingIO);
     SkASSERT(SkToBool(fResource) == fOwnRef);
     if (fOwnRef) {
         fResource->unref();
         fOwnRef = false;
         fResource = nullptr;
     }
+    fDst = NULL;
 }
 
-void GrGpuResourceRef::setResource(GrGpuResource* resource, GrIOType ioType) {
+void GrGpuResourceRef::setResource(GrGpuResource* resource, GrIOType ioType, GrGpuResource* dst) {
     SkASSERT(!fPendingIO);
     SkASSERT(SkToBool(fResource) == fOwnRef);
     SkSafeUnref(fResource);
     if (nullptr == resource) {
+        SkASSERT(nullptr == dst);
         fResource = nullptr;
         fOwnRef = false;
+        fDst = NULL;
     } else {
         fResource = resource;
         fOwnRef = true;
         fIOType = ioType;
+        if (fIOType == kRead_GrIOType || fIOType == kRW_GrIOType) {
+            SkASSERT(dst);
+            fDst = dst;
+        } else {
+            fDst = NULL;
+        }
     }
 }
 
 void GrGpuResourceRef::markPendingIO() const {
     // This should only be called when the owning GrProgramElement gets its first
     // pendingExecution ref.
     SkASSERT(!fPendingIO);
     SkASSERT(fResource);
     fPendingIO = true;
     switch (fIOType) {
         case kRead_GrIOType:
-            fResource->addPendingRead();
+            SkASSERT(fDst);
+            fResource->addPendingRead(fDst);
             break;
         case kWrite_GrIOType:
             fResource->addPendingWrite();
             break;
         case kRW_GrIOType:
-            fResource->addPendingRead();
+            SkASSERT(fDst);
+            fResource->addPendingRead(fDst);
             fResource->addPendingWrite();
             break;
     }
 }
 
 void GrGpuResourceRef::pendingIOComplete() const {
     // This should only be called when the owner's pending executions have ocurred but it is still
     // reffed.
     SkASSERT(fOwnRef);
     SkASSERT(fPendingIO);
@@ skipping 15 matching lines @@
 
 void GrGpuResourceRef::removeRef() const {
     // This should only be called once, when the owners last ref goes away and
     // there is a pending execution.
     SkASSERT(fOwnRef);
     SkASSERT(fPendingIO);
     SkASSERT(fResource);
     fResource->unref();
     fOwnRef = false;
 }