diff options
author | bsalomon <bsalomon@google.com> | 2016-02-01 13:16:14 -0800 |
---|---|---|
committer | Commit bot <commit-bot@chromium.org> | 2016-02-01 13:16:14 -0800 |
commit | f267c1efe7de7a8e71404afde6cbf93c3808d267 (patch) | |
tree | 3ca07e8c6e5526ccfe4166953631006dd8b71845 /src/gpu/effects/GrYUVtoRGBEffect.cpp | |
parent | 9d02b264b72be64702e43cb797b7899a8a6926ca (diff) |
Add ability to extract YUV planes from SkImage
GOLD_TRYBOT_URL= https://gold.skia.org/search2?unt=true&query=source_type%3Dgm&master=false&issue=1513393002
Review URL: https://codereview.chromium.org/1513393002
Diffstat (limited to 'src/gpu/effects/GrYUVtoRGBEffect.cpp')
-rw-r--r-- | src/gpu/effects/GrYUVtoRGBEffect.cpp | 185 |
1 files changed, 0 insertions, 185 deletions
diff --git a/src/gpu/effects/GrYUVtoRGBEffect.cpp b/src/gpu/effects/GrYUVtoRGBEffect.cpp deleted file mode 100644 index 16dad8635c..0000000000 --- a/src/gpu/effects/GrYUVtoRGBEffect.cpp +++ /dev/null @@ -1,185 +0,0 @@ -/* - * 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 "GrYUVtoRGBEffect.h" - -#include "GrCoordTransform.h" -#include "GrFragmentProcessor.h" -#include "GrInvariantOutput.h" -#include "GrProcessor.h" -#include "glsl/GrGLSLFragmentProcessor.h" -#include "glsl/GrGLSLFragmentShaderBuilder.h" -#include "glsl/GrGLSLProgramDataManager.h" -#include "glsl/GrGLSLUniformHandler.h" - -namespace { - -class YUVtoRGBEffect : public GrFragmentProcessor { -public: - static GrFragmentProcessor* Create(GrTexture* yTexture, GrTexture* uTexture, - GrTexture* vTexture, const SkISize sizes[3], - SkYUVColorSpace colorSpace) { - SkScalar w[3], h[3]; - w[0] = SkIntToScalar(sizes[0].fWidth) / SkIntToScalar(yTexture->width()); - h[0] = SkIntToScalar(sizes[0].fHeight) / SkIntToScalar(yTexture->height()); - w[1] = SkIntToScalar(sizes[1].fWidth) / SkIntToScalar(uTexture->width()); - h[1] = SkIntToScalar(sizes[1].fHeight) / SkIntToScalar(uTexture->height()); - w[2] = SkIntToScalar(sizes[2].fWidth) / SkIntToScalar(vTexture->width()); - h[2] = SkIntToScalar(sizes[2].fHeight) / SkIntToScalar(vTexture->height()); - SkMatrix yuvMatrix[3]; - yuvMatrix[0] = GrCoordTransform::MakeDivByTextureWHMatrix(yTexture); - yuvMatrix[1] = yuvMatrix[0]; - yuvMatrix[1].preScale(w[1] / w[0], h[1] / h[0]); - yuvMatrix[2] = yuvMatrix[0]; - yuvMatrix[2].preScale(w[2] / w[0], h[2] / h[0]); - GrTextureParams::FilterMode uvFilterMode = - ((sizes[1].fWidth != sizes[0].fWidth) || - (sizes[1].fHeight != sizes[0].fHeight) || - (sizes[2].fWidth != sizes[0].fWidth) || - (sizes[2].fHeight != sizes[0].fHeight)) ? - GrTextureParams::kBilerp_FilterMode : - GrTextureParams::kNone_FilterMode; - return new YUVtoRGBEffect(yTexture, uTexture, vTexture, yuvMatrix, uvFilterMode, - colorSpace); - } - - const char* name() const override { return "YUV to RGB"; } - - SkYUVColorSpace getColorSpace() const { - return fColorSpace; - } - - class GLSLProcessor : public GrGLSLFragmentProcessor { - public: - static const float kJPEGConversionMatrix[16]; - static const float kRec601ConversionMatrix[16]; - static const float kRec709ConversionMatrix[16]; - - // this class always generates the same code. - static void GenKey(const GrProcessor&, const GrGLSLCaps&, GrProcessorKeyBuilder*) {} - - GLSLProcessor(const GrProcessor&) {} - - virtual void emitCode(EmitArgs& args) override { - GrGLSLFragmentBuilder* fragBuilder = args.fFragBuilder; - - const char* yuvMatrix = nullptr; - fMatrixUni = args.fUniformHandler->addUniform( - GrGLSLUniformHandler::kFragment_Visibility, - kMat44f_GrSLType, kDefault_GrSLPrecision, - "YUVMatrix", &yuvMatrix); - fragBuilder->codeAppendf("\t%s = vec4(\n\t\t", args.fOutputColor); - fragBuilder->appendTextureLookup(args.fSamplers[0], args.fCoords[0].c_str(), - args.fCoords[0].getType()); - fragBuilder->codeAppend(".r,\n\t\t"); - fragBuilder->appendTextureLookup(args.fSamplers[1], args.fCoords[1].c_str(), - args.fCoords[1].getType()); - fragBuilder->codeAppend(".r,\n\t\t"); - fragBuilder->appendTextureLookup(args.fSamplers[2], args.fCoords[2].c_str(), - args.fCoords[2].getType()); - fragBuilder->codeAppendf(".r,\n\t\t1.0) * %s;\n", yuvMatrix); - } - - protected: - virtual void onSetData(const GrGLSLProgramDataManager& pdman, - const GrProcessor& processor) override { - const YUVtoRGBEffect& yuvEffect = processor.cast<YUVtoRGBEffect>(); - switch (yuvEffect.getColorSpace()) { - case kJPEG_SkYUVColorSpace: - pdman.setMatrix4f(fMatrixUni, kJPEGConversionMatrix); - break; - case kRec601_SkYUVColorSpace: - pdman.setMatrix4f(fMatrixUni, kRec601ConversionMatrix); - break; - case kRec709_SkYUVColorSpace: - pdman.setMatrix4f(fMatrixUni, kRec709ConversionMatrix); - break; - } - } - - private: - GrGLSLProgramDataManager::UniformHandle fMatrixUni; - - typedef GrGLSLFragmentProcessor INHERITED; - }; - -private: - YUVtoRGBEffect(GrTexture* yTexture, GrTexture* uTexture, GrTexture* vTexture, - const SkMatrix yuvMatrix[3], GrTextureParams::FilterMode uvFilterMode, - SkYUVColorSpace colorSpace) - : fYTransform(kLocal_GrCoordSet, yuvMatrix[0], yTexture, GrTextureParams::kNone_FilterMode) - , fYAccess(yTexture) - , fUTransform(kLocal_GrCoordSet, yuvMatrix[1], uTexture, uvFilterMode) - , fUAccess(uTexture, uvFilterMode) - , fVTransform(kLocal_GrCoordSet, yuvMatrix[2], vTexture, uvFilterMode) - , fVAccess(vTexture, uvFilterMode) - , fColorSpace(colorSpace) { - this->initClassID<YUVtoRGBEffect>(); - this->addCoordTransform(&fYTransform); - this->addTextureAccess(&fYAccess); - this->addCoordTransform(&fUTransform); - this->addTextureAccess(&fUAccess); - this->addCoordTransform(&fVTransform); - this->addTextureAccess(&fVAccess); - } - - GrGLSLFragmentProcessor* onCreateGLSLInstance() const override { - return new GLSLProcessor(*this); - } - - virtual void onGetGLSLProcessorKey(const GrGLSLCaps& caps, - GrProcessorKeyBuilder* b) const override { - GLSLProcessor::GenKey(*this, caps, b); - } - - bool onIsEqual(const GrFragmentProcessor& sBase) const override { - const YUVtoRGBEffect& s = sBase.cast<YUVtoRGBEffect>(); - return fColorSpace == s.getColorSpace(); - } - - void onComputeInvariantOutput(GrInvariantOutput* inout) const override { - // YUV is opaque - inout->setToOther(kA_GrColorComponentFlag, 0xFF << GrColor_SHIFT_A, - GrInvariantOutput::kWillNot_ReadInput); - } - - GrCoordTransform fYTransform; - GrTextureAccess fYAccess; - GrCoordTransform fUTransform; - GrTextureAccess fUAccess; - GrCoordTransform fVTransform; - GrTextureAccess fVAccess; - SkYUVColorSpace fColorSpace; - - typedef GrFragmentProcessor INHERITED; -}; - -const float YUVtoRGBEffect::GLSLProcessor::kJPEGConversionMatrix[16] = { - 1.0f, 0.0f, 1.402f, -0.701f, - 1.0f, -0.34414f, -0.71414f, 0.529f, - 1.0f, 1.772f, 0.0f, -0.886f, - 0.0f, 0.0f, 0.0f, 1.0}; -const float YUVtoRGBEffect::GLSLProcessor::kRec601ConversionMatrix[16] = { - 1.164f, 0.0f, 1.596f, -0.87075f, - 1.164f, -0.391f, -0.813f, 0.52925f, - 1.164f, 2.018f, 0.0f, -1.08175f, - 0.0f, 0.0f, 0.0f, 1.0}; -const float YUVtoRGBEffect::GLSLProcessor::kRec709ConversionMatrix[16] = { - 1.164f, 0.0f, 1.793f, -0.96925f, - 1.164f, -0.213f, -0.533f, 0.30025f, - 1.164f, 2.112f, 0.0f, -1.12875f, - 0.0f, 0.0f, 0.0f, 1.0f}; -} - -////////////////////////////////////////////////////////////////////////////// - -GrFragmentProcessor* -GrYUVtoRGBEffect::Create(GrTexture* yTexture, GrTexture* uTexture, GrTexture* vTexture, - const SkISize sizes[3], SkYUVColorSpace colorSpace) { - SkASSERT(yTexture && uTexture && vTexture && sizes); - return YUVtoRGBEffect::Create(yTexture, uTexture, vTexture, sizes, colorSpace); -} |