Altered GrDrawState to always ref texture and render target
http://codereview.appspot.com/6251049/
git-svn-id: http://skia.googlecode.com/svn/trunk@4298 2bbb7eff-a529-9590-31e7-b0007b416f81
diff --git a/src/gpu/GrDrawState.h b/src/gpu/GrDrawState.h
index e77b63a..5feccb5 100644
--- a/src/gpu/GrDrawState.h
+++ b/src/gpu/GrDrawState.h
@@ -15,11 +15,10 @@
#include "GrSamplerState.h"
#include "GrStencil.h"
#include "GrTexture.h"
+#include "GrRenderTarget.h"
#include "SkXfermode.h"
-class GrRenderTarget;
-class GrTexture;
class GrDrawState : public GrRefCnt {
@@ -54,18 +53,39 @@
typedef uint32_t StageMask;
GR_STATIC_ASSERT(sizeof(StageMask)*8 >= GrDrawState::kNumStages);
- GrDrawState() {
+ GrDrawState()
+ : fRenderTarget(NULL) {
+
+ for (int i = 0; i < kNumStages; ++i) {
+ fTextures[i] = NULL;
+ }
+
this->reset();
}
- GrDrawState(const GrDrawState& state) {
+ GrDrawState(const GrDrawState& state)
+ : fRenderTarget(NULL) {
+
+ for (int i = 0; i < kNumStages; ++i) {
+ fTextures[i] = NULL;
+ }
+
*this = state;
}
+ virtual ~GrDrawState() {
+ this->releaseTextures();
+ GrSafeSetNull(fRenderTarget);
+ }
+
/**
* Resets to the default state. Sampler states will not be modified.
*/
void reset() {
+
+ this->releaseTextures();
+ GrSafeSetNull(fRenderTarget);
+
// make sure any pad is zero for memcmp
// all GrDrawState members should default to something valid by the
// the memset except those initialized individually below. There should
@@ -86,14 +106,13 @@
fSrcBlend = kOne_GrBlendCoeff;
fDstBlend = kZero_GrBlendCoeff;
fViewMatrix.reset();
- fBehaviorBits = 0;
// ensure values that will be memcmp'ed in == but not memset in reset()
// are tightly packed
GrAssert(this->memsetSize() + sizeof(fColor) + sizeof(fCoverage) +
sizeof(fFirstCoverageStage) + sizeof(fColorFilterMode) +
- sizeof(fSrcBlend) + sizeof(fDstBlend) ==
- this->podSize());
+ sizeof(fSrcBlend) + sizeof(fDstBlend) + sizeof(fTextures) +
+ sizeof(fRenderTarget) == this->podSize());
}
///////////////////////////////////////////////////////////////////////////
@@ -175,18 +194,7 @@
void setTexture(int stage, GrTexture* texture) {
GrAssert((unsigned)stage < kNumStages);
- if (isBehaviorEnabled(kTexturesNeedRef_BehaviorBit)) {
- // If we don't clear out the current texture before unreffing
- // it we can get into an infinite loop as the GrGLTexture's
- // onRelease method recursively calls setTexture
- GrTexture* temp = fTextures[stage];
- fTextures[stage] = NULL;
-
- SkSafeRef(texture);
- SkSafeUnref(temp);
- }
-
- fTextures[stage] = texture;
+ GrSafeAssign(fTextures[stage], texture);
}
/**
@@ -210,14 +218,14 @@
*/
void releaseTextures() {
for (int i = 0; i < kNumStages; ++i) {
- this->setTexture(i, NULL);
+ GrSafeSetNull(fTextures[i]);
}
}
class AutoTextureRelease : public ::GrNoncopyable {
public:
AutoTextureRelease(GrDrawState* ds) : fDrawState(ds) {}
- ~AutoTextureRelease() {
+ ~AutoTextureRelease() {
if (NULL != fDrawState) {
fDrawState->releaseTextures();
}
@@ -483,7 +491,9 @@
*
* @param target The render target to set.
*/
- void setRenderTarget(GrRenderTarget* target) { fRenderTarget = target; }
+ void setRenderTarget(GrRenderTarget* target) {
+ GrSafeAssign(fRenderTarget, target);
+ }
/**
* Retrieves the currently set rendertarget.
@@ -501,16 +511,26 @@
fSavedTarget = NULL;
this->set(ds, newTarget);
}
- ~AutoRenderTargetRestore() { this->set(NULL, NULL); }
- void set(GrDrawState* ds, GrRenderTarget* newTarget) {
+ ~AutoRenderTargetRestore() { this->restore(); }
+
+ void restore() {
if (NULL != fDrawState) {
fDrawState->setRenderTarget(fSavedTarget);
+ fDrawState = NULL;
}
+ GrSafeSetNull(fSavedTarget);
+ }
+
+ void set(GrDrawState* ds, GrRenderTarget* newTarget) {
+ this->restore();
+
if (NULL != ds) {
+ GrAssert(NULL == fSavedTarget);
fSavedTarget = ds->getRenderTarget();
+ SkSafeRef(fSavedTarget);
ds->setRenderTarget(newTarget);
+ fDrawState = ds;
}
- fDrawState = ds;
}
private:
GrDrawState* fDrawState;
@@ -693,28 +713,6 @@
return 0 != (stateBit & fFlagBits);
}
- /**
- * Flags that do not affect rendering.
- */
- enum GrBehaviorBits {
- /**
- * Calls to setTexture will ref/unref the texture
- */
- kTexturesNeedRef_BehaviorBit = 0x01,
- };
-
- void enableBehavior(uint32_t behaviorBits) {
- fBehaviorBits |= behaviorBits;
- }
-
- void disableBehavior(uint32_t behaviorBits) {
- fBehaviorBits &= ~(behaviorBits);
- }
-
- bool isBehaviorEnabled(uint32_t behaviorBits) const {
- return 0 != (behaviorBits & fBehaviorBits);
- }
-
/// @}
///////////////////////////////////////////////////////////////////////////
@@ -760,14 +758,6 @@
return false;
}
- // kTexturesNeedRef is an internal flag for altering the draw state's
- // behavior rather than a property that will impact drawing - ignore it
- // here
- if ((fBehaviorBits & ~kTexturesNeedRef_BehaviorBit) !=
- (s.fBehaviorBits & ~kTexturesNeedRef_BehaviorBit)) {
- return false;
- }
-
for (int i = 0; i < kNumStages; i++) {
if (fTextures[i] &&
this->fSamplerStates[i] != s.fSamplerStates[i]) {
@@ -792,13 +782,16 @@
memcpy(this->podStart(), s.podStart(), this->podSize());
fViewMatrix = s.fViewMatrix;
- fBehaviorBits = s.fBehaviorBits;
for (int i = 0; i < kNumStages; i++) {
+ SkSafeRef(fTextures[i]); // already copied by memcpy
if (s.fTextures[i]) {
this->fSamplerStates[i] = s.fSamplerStates[i];
}
}
+
+ SkSafeRef(fRenderTarget); // already copied by memcpy
+
if (kColorMatrix_StateBit & s.fFlagBits) {
memcpy(this->fColorMatrix, s.fColorMatrix, sizeof(fColorMatrix));
}
@@ -832,23 +825,25 @@
GrColor fBlendConstant;
GrColor fPodStartMarker;
};
- GrTexture* fTextures[kNumStages];
GrColor fColorFilterColor;
- uint32_t fFlagBits;
DrawFace fDrawFace;
VertexEdgeType fVertexEdgeType;
GrStencilSettings fStencilSettings;
union {
- GrRenderTarget* fRenderTarget;
- GrRenderTarget* fMemsetEndMarker;
+ uint32_t fFlagBits;
+ uint32_t fMemsetEndMarker;
};
// @}
+ int fFirstCoverageStage;
+
// @{ Initialized to values other than zero, but memcmp'ed in operator==
// and memcpy'ed in operator=.
+ GrTexture* fTextures[kNumStages];
+ GrRenderTarget* fRenderTarget;
+
GrColor fColor;
GrColor fCoverage;
- int fFirstCoverageStage;
SkXfermode::Mode fColorFilterMode;
GrBlendCoeff fSrcBlend;
union {
@@ -857,7 +852,6 @@
};
// @}
- uint32_t fBehaviorBits;
GrMatrix fViewMatrix;
// This field must be last; it will not be copied or compared