1/* 2 * Copyright 2014 Google Inc. 3 * 4 * Use of this source code is governed by a BSD-style license that can be 5 * found in the LICENSE file. 6 */ 7 8#include "GrYUVtoRGBEffect.h" 9 10#include "GrCoordTransform.h" 11#include "GrInvariantOutput.h" 12#include "GrProcessor.h" 13#include "gl/GrGLProcessor.h" 14#include "gl/builders/GrGLProgramBuilder.h" 15 16namespace { 17 18class YUVtoRGBEffect : public GrFragmentProcessor { 19public: 20 static GrFragmentProcessor* Create(GrTexture* yTexture, GrTexture* uTexture, 21 GrTexture* vTexture, SkISize sizes[3], 22 SkYUVColorSpace colorSpace) { 23 SkScalar w[3], h[3]; 24 w[0] = SkIntToScalar(sizes[0].fWidth) / SkIntToScalar(yTexture->width()); 25 h[0] = SkIntToScalar(sizes[0].fHeight) / SkIntToScalar(yTexture->height()); 26 w[1] = SkIntToScalar(sizes[1].fWidth) / SkIntToScalar(uTexture->width()); 27 h[1] = SkIntToScalar(sizes[1].fHeight) / SkIntToScalar(uTexture->height()); 28 w[2] = SkIntToScalar(sizes[2].fWidth) / SkIntToScalar(vTexture->width()); 29 h[2] = SkIntToScalar(sizes[2].fHeight) / SkIntToScalar(vTexture->height()); 30 SkMatrix yuvMatrix[3]; 31 yuvMatrix[0] = GrCoordTransform::MakeDivByTextureWHMatrix(yTexture); 32 yuvMatrix[1] = yuvMatrix[0]; 33 yuvMatrix[1].preScale(w[1] / w[0], h[1] / h[0]); 34 yuvMatrix[2] = yuvMatrix[0]; 35 yuvMatrix[2].preScale(w[2] / w[0], h[2] / h[0]); 36 GrTextureParams::FilterMode uvFilterMode = 37 ((sizes[1].fWidth != sizes[0].fWidth) || 38 (sizes[1].fHeight != sizes[0].fHeight) || 39 (sizes[2].fWidth != sizes[0].fWidth) || 40 (sizes[2].fHeight != sizes[0].fHeight)) ? 41 GrTextureParams::kBilerp_FilterMode : 42 GrTextureParams::kNone_FilterMode; 43 return SkNEW_ARGS(YUVtoRGBEffect, (yTexture, uTexture, vTexture, yuvMatrix, 44 uvFilterMode, colorSpace)); 45 } 46 47 const char* name() const override { return "YUV to RGB"; } 48 49 SkYUVColorSpace getColorSpace() const { 50 return fColorSpace; 51 } 52 53 class GLProcessor : public GrGLFragmentProcessor { 54 public: 55 static const GrGLfloat kJPEGConversionMatrix[16]; 56 static const GrGLfloat kRec601ConversionMatrix[16]; 57 58 // this class always generates the same code. 59 static void GenKey(const GrProcessor&, const GrGLSLCaps&, GrProcessorKeyBuilder*) {} 60 61 GLProcessor(const GrProcessor&) {} 62 63 virtual void emitCode(GrGLFPBuilder* builder, 64 const GrFragmentProcessor&, 65 const char* outputColor, 66 const char* inputColor, 67 const TransformedCoordsArray& coords, 68 const TextureSamplerArray& samplers) override { 69 GrGLFragmentBuilder* fsBuilder = builder->getFragmentShaderBuilder(); 70 71 const char* yuvMatrix = NULL; 72 fMatrixUni = builder->addUniform(GrGLProgramBuilder::kFragment_Visibility, 73 kMat44f_GrSLType, kDefault_GrSLPrecision, 74 "YUVMatrix", &yuvMatrix); 75 fsBuilder->codeAppendf("\t%s = vec4(\n\t\t", outputColor); 76 fsBuilder->appendTextureLookup(samplers[0], coords[0].c_str(), coords[0].getType()); 77 fsBuilder->codeAppend(".r,\n\t\t"); 78 fsBuilder->appendTextureLookup(samplers[1], coords[1].c_str(), coords[1].getType()); 79 fsBuilder->codeAppend(".r,\n\t\t"); 80 fsBuilder->appendTextureLookup(samplers[2], coords[2].c_str(), coords[2].getType()); 81 fsBuilder->codeAppendf(".r,\n\t\t1.0) * %s;\n", yuvMatrix); 82 } 83 84 virtual void setData(const GrGLProgramDataManager& pdman, 85 const GrProcessor& processor) override { 86 const YUVtoRGBEffect& yuvEffect = processor.cast<YUVtoRGBEffect>(); 87 switch (yuvEffect.getColorSpace()) { 88 case kJPEG_SkYUVColorSpace: 89 pdman.setMatrix4f(fMatrixUni, kJPEGConversionMatrix); 90 break; 91 case kRec601_SkYUVColorSpace: 92 pdman.setMatrix4f(fMatrixUni, kRec601ConversionMatrix); 93 break; 94 } 95 } 96 97 private: 98 GrGLProgramDataManager::UniformHandle fMatrixUni; 99 100 typedef GrGLFragmentProcessor INHERITED; 101 }; 102 103 virtual void getGLProcessorKey(const GrGLSLCaps& caps, 104 GrProcessorKeyBuilder* b) const override { 105 GLProcessor::GenKey(*this, caps, b); 106 } 107 108 GrGLFragmentProcessor* createGLInstance() const override { 109 return SkNEW_ARGS(GLProcessor, (*this)); 110 } 111 112private: 113 YUVtoRGBEffect(GrTexture* yTexture, GrTexture* uTexture, GrTexture* vTexture, 114 SkMatrix yuvMatrix[3], GrTextureParams::FilterMode uvFilterMode, 115 SkYUVColorSpace colorSpace) 116 : fYTransform(kLocal_GrCoordSet, yuvMatrix[0], yTexture, GrTextureParams::kNone_FilterMode) 117 , fYAccess(yTexture) 118 , fUTransform(kLocal_GrCoordSet, yuvMatrix[1], uTexture, uvFilterMode) 119 , fUAccess(uTexture, uvFilterMode) 120 , fVTransform(kLocal_GrCoordSet, yuvMatrix[2], vTexture, uvFilterMode) 121 , fVAccess(vTexture, uvFilterMode) 122 , fColorSpace(colorSpace) { 123 this->initClassID<YUVtoRGBEffect>(); 124 this->addCoordTransform(&fYTransform); 125 this->addTextureAccess(&fYAccess); 126 this->addCoordTransform(&fUTransform); 127 this->addTextureAccess(&fUAccess); 128 this->addCoordTransform(&fVTransform); 129 this->addTextureAccess(&fVAccess); 130 } 131 132 bool onIsEqual(const GrFragmentProcessor& sBase) const override { 133 const YUVtoRGBEffect& s = sBase.cast<YUVtoRGBEffect>(); 134 return fColorSpace == s.getColorSpace(); 135 } 136 137 void onComputeInvariantOutput(GrInvariantOutput* inout) const override { 138 // YUV is opaque 139 inout->setToOther(kA_GrColorComponentFlag, 0xFF << GrColor_SHIFT_A, 140 GrInvariantOutput::kWillNot_ReadInput); 141 } 142 143 GrCoordTransform fYTransform; 144 GrTextureAccess fYAccess; 145 GrCoordTransform fUTransform; 146 GrTextureAccess fUAccess; 147 GrCoordTransform fVTransform; 148 GrTextureAccess fVAccess; 149 SkYUVColorSpace fColorSpace; 150 151 typedef GrFragmentProcessor INHERITED; 152}; 153 154const GrGLfloat YUVtoRGBEffect::GLProcessor::kJPEGConversionMatrix[16] = { 155 1.0f, 0.0f, 1.402f, -0.701f, 156 1.0f, -0.34414f, -0.71414f, 0.529f, 157 1.0f, 1.772f, 0.0f, -0.886f, 158 0.0f, 0.0f, 0.0f, 1.0}; 159const GrGLfloat YUVtoRGBEffect::GLProcessor::kRec601ConversionMatrix[16] = { 160 1.164f, 0.0f, 1.596f, -0.87075f, 161 1.164f, -0.391f, -0.813f, 0.52925f, 162 1.164f, 2.018f, 0.0f, -1.08175f, 163 0.0f, 0.0f, 0.0f, 1.0}; 164} 165 166////////////////////////////////////////////////////////////////////////////// 167 168GrFragmentProcessor* 169GrYUVtoRGBEffect::Create(GrTexture* yTexture, GrTexture* uTexture, GrTexture* vTexture, 170 SkISize sizes[3], SkYUVColorSpace colorSpace) { 171 SkASSERT(yTexture && uTexture && vTexture && sizes); 172 return YUVtoRGBEffect::Create(yTexture, uTexture, vTexture, sizes, colorSpace); 173} 174