skia2/include/gpu/GrEffect.h
bsalomon@google.com 838f6e18fb Add deferred count to GrResource.
This will be used to determine whether a resource can either be deleted or reinserted in the cache when its ref count reaches zero.
Review URL: https://codereview.appspot.com/7202046

git-svn-id: http://skia.googlecode.com/svn/trunk@7355 2bbb7eff-a529-9590-31e7-b0007b416f81
2013-01-23 21:37:01 +00:00

262 lines
9.7 KiB
C++

/*
* Copyright 2012 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#ifndef GrEffect_DEFINED
#define GrEffect_DEFINED
#include "GrColor.h"
#include "GrEffectUnitTest.h"
#include "GrNoncopyable.h"
#include "GrRefCnt.h"
#include "GrTexture.h"
#include "GrTextureAccess.h"
class GrBackendEffectFactory;
class GrContext;
class GrEffect;
class SkString;
/**
* A Wrapper class for GrEffect. Its ref-count will track owners that may use effects to enqueue
* new draw operations separately from ownership within a deferred drawing queue. When the
* GrEffectRef ref count reaches zero the scratch GrResources owned by the effect can be recycled
* in service of later draws. However, the deferred draw queue may still own direct references to
* the underlying GrEffect.
*/
class GrEffectRef : public SkRefCnt {
public:
SK_DECLARE_INST_COUNT(GrEffectRef);
GrEffect* get() { return fEffect; }
const GrEffect* get() const { return fEffect; }
const GrEffect* operator-> () { return fEffect; }
const GrEffect* operator-> () const { return fEffect; }
void* operator new(size_t size);
void operator delete(void* target);
private:
friend class GrEffect; // to construct these
explicit GrEffectRef(GrEffect* effect);
virtual ~GrEffectRef();
GrEffect* fEffect;
typedef SkRefCnt INHERITED;
};
/** Provides custom vertex shader, fragment shader, uniform data for a particular stage of the
Ganesh shading pipeline.
Subclasses must have a function that produces a human-readable name:
static const char* Name();
GrEffect objects *must* be immutable: after being constructed, their fields may not change.
GrEffect subclass objects should be created by factory functions that return GrEffectRef.
There is no public way to wrap a GrEffect in a GrEffectRef. Thus, a factory should be a static
member function of a GrEffect subclass.
Because almost no code should ever handle a GrEffect outside of a GrEffectRef, we privately
inherit from GrRefCnt to help prevent accidental direct ref'ing/unref'ing of effects.
*/
class GrEffect : private GrRefCnt {
public:
SK_DECLARE_INST_COUNT(GrEffect)
virtual ~GrEffect();
/**
* Flags for getConstantColorComponents. They are defined so that the bit order reflects the
* GrColor shift order.
*/
enum ValidComponentFlags {
kR_ValidComponentFlag = 1 << (GrColor_SHIFT_R / 8),
kG_ValidComponentFlag = 1 << (GrColor_SHIFT_G / 8),
kB_ValidComponentFlag = 1 << (GrColor_SHIFT_B / 8),
kA_ValidComponentFlag = 1 << (GrColor_SHIFT_A / 8),
kAll_ValidComponentFlags = (kR_ValidComponentFlag | kG_ValidComponentFlag |
kB_ValidComponentFlag | kA_ValidComponentFlag)
};
/**
* This function is used to perform optimizations. When called the color and validFlags params
* indicate whether the input components to this effect in the FS will have known values. The
* function updates both params to indicate known values of its output. A component of the color
* param only has meaning if the corresponding bit in validFlags is set.
*/
virtual void getConstantColorComponents(GrColor* color, uint32_t* validFlags) const = 0;
/** This object, besides creating back-end-specific helper objects, is used for run-time-type-
identification. The factory should be an instance of templated class,
GrTBackendEffectFactory. It is templated on the subclass of GrEffect. The subclass must have
a nested type (or typedef) named GLEffect which will be the subclass of GrGLEffect created
by the factory.
Example:
class MyCustomEffect : public GrEffect {
...
virtual const GrBackendEffectFactory& getFactory() const SK_OVERRIDE {
return GrTBackendEffectFactory<MyCustomEffect>::getInstance();
}
...
};
*/
virtual const GrBackendEffectFactory& getFactory() const = 0;
/** Returns true if this and other effect conservatively draw identically. It can only return
true when the two effects are of the same subclass (i.e. they return the same object from
from getFactory()).
A return value of true from isEqual() should not be used to test whether the effects would
generate the same shader code. To test for identical code generation use the EffectKey
computed by the GrBackendEffectFactory:
effectA.getFactory().glEffectKey(effectA) == effectB.getFactory().glEffectKey(effectB).
*/
bool isEqual(const GrEffectRef& other) const {
return this->isEqual(*other.get());
}
/** Human-meaningful string to identify this effect; may be embedded
in generated shader code. */
const char* name() const;
int numTextures() const { return fTextureAccesses.count(); }
/** Returns the access pattern for the texture at index. index must be valid according to
numTextures(). */
const GrTextureAccess& textureAccess(int index) const { return *fTextureAccesses[index]; }
/** Shortcut for textureAccess(index).texture(); */
GrTexture* texture(int index) const { return this->textureAccess(index).getTexture(); }
/** Useful for effects that want to insert a texture matrix that is implied by the texture
dimensions */
static inline SkMatrix MakeDivByTextureWHMatrix(const GrTexture* texture) {
GrAssert(NULL != texture);
SkMatrix mat;
mat.setIDiv(texture->width(), texture->height());
return mat;
}
void* operator new(size_t size);
void operator delete(void* target);
/** These functions are used when recording effects into a deferred drawing queue. The inc call
keeps the effect alive outside of GrEffectRef while allowing any resources owned by the
effect to be returned to the cache for reuse. The dec call must balance the inc call. */
void incDeferredRefCounts() const {
this->ref();
int count = fTextureAccesses.count();
for (int t = 0; t < count; ++t) {
fTextureAccesses[t]->getTexture()->incDeferredRefCount();
}
}
void decDeferredRefCounts() const {
int count = fTextureAccesses.count();
for (int t = 0; t < count; ++t) {
fTextureAccesses[t]->getTexture()->decDeferredRefCount();
}
this->unref();
}
protected:
/**
* Subclasses call this from their constructor to register GrTextureAcceses. The effect subclass
* manages the lifetime of the accesses (this function only stores a pointer). This must only be
* called from the constructor because GrEffects are supposed to be immutable.
*/
void addTextureAccess(const GrTextureAccess* textureAccess);
GrEffect() : fEffectRef(NULL) {};
/** This should be called by GrEffect subclass factories. See the comment on AutoEffectUnref for
an example factory function. */
static GrEffectRef* CreateEffectRef(GrEffect* effect) {
if (NULL == effect->fEffectRef) {
effect->fEffectRef = SkNEW_ARGS(GrEffectRef, (effect));
} else {
effect->fEffectRef->ref();
}
return effect->fEffectRef;
}
static const GrEffectRef* CreateEffectRef(const GrEffect* effect) {
return CreateEffectRef(const_cast<GrEffect*>(effect));
}
/** Helper used in subclass factory functions to unref the effect after it has been wrapped in a
GrEffectRef. E.g.:
class EffectSubclass : public GrEffect {
public:
GrEffectRef* Create(ParamType1 param1, ParamType2 param2, ...) {
AutoEffectUnref effect(SkNEW_ARGS(EffectSubclass, (param1, param2, ...)));
return CreateEffectRef(effect);
}
*/
class AutoEffectUnref {
public:
AutoEffectUnref(GrEffect* effect) : fEffect(effect) { }
~AutoEffectUnref() { fEffect->unref(); }
operator GrEffect*() { return fEffect; }
private:
GrEffect* fEffect;
};
/** Helper for getting the GrEffect out of a GrEffectRef and down-casting to a GrEffect subclass
*/
template <typename T>
static const T& CastEffect(const GrEffect& effectRef) {
return *static_cast<const T*>(&effectRef);
}
private:
bool isEqual(const GrEffect& other) const {
if (&this->getFactory() != &other.getFactory()) {
return false;
}
bool result = this->onIsEqual(other);
#if GR_DEBUG
if (result) {
GrAssert(this->numTextures() == other.numTextures());
for (int i = 0; i < this->numTextures(); ++i) {
GrAssert(*fTextureAccesses[i] == *other.fTextureAccesses[i]);
}
}
#endif
return result;
}
/** Subclass implements this to support isEqual(). It will only be called if it is known that
the two effects are of the same subclass (i.e. they return the same object from
getFactory()).*/
virtual bool onIsEqual(const GrEffect& other) const = 0;
void EffectRefDestroyed() { fEffectRef = NULL; }
friend class GrEffectRef; // to call EffectRefDestroyed()
friend class GrEffectStage; // to rewrap GrEffect in GrEffectRef when restoring an effect-stage
// from deferred state, to call isEqual on naked GrEffects, and
// to inc/dec deferred ref counts.
SkSTArray<4, const GrTextureAccess*, true> fTextureAccesses;
GrEffectRef* fEffectRef;
typedef GrRefCnt INHERITED;
};
inline GrEffectRef::GrEffectRef(GrEffect* effect) {
GrAssert(NULL != effect);
effect->ref();
fEffect = effect;
}
#endif