/* * Copyright 2018 Google Inc. * * Use of this source code is governed by a BSD-style license that can be * found in the LICENSE file. */ /************************************************************************************************** *** This file was autogenerated from GrYUVtoRGBEffect.fp; do not modify. **************************************************************************************************/ #include "GrYUVtoRGBEffect.h" static const float kJPEGConversionMatrix[16] = { 1.0f, 0.0f, 1.402f, -0.703749f, 1.0f, -0.344136f, -0.714136f, 0.531211f, 1.0f, 1.772f, 0.0f, -0.889475f, 0.0f, 0.0f, 0.0f, 1.0}; static const float 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}; static const float 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}; std::unique_ptr GrYUVtoRGBEffect::Make(sk_sp yProxy, sk_sp uProxy, sk_sp vProxy, SkYUVColorSpace colorSpace, bool nv12) { SkScalar w[3], h[3]; w[0] = SkIntToScalar(yProxy->width()); h[0] = SkIntToScalar(yProxy->height()); w[1] = SkIntToScalar(uProxy->width()); h[1] = SkIntToScalar(uProxy->height()); w[2] = SkIntToScalar(vProxy->width()); h[2] = SkIntToScalar(vProxy->height()); SkMatrix yTransform = SkMatrix::I(); SkMatrix uTransform = SkMatrix::MakeScale(w[1] / w[0], h[1] / h[0]); SkMatrix vTransform = SkMatrix::MakeScale(w[2] / w[0], h[2] / h[0]); GrSamplerState::Filter uvFilterMode = ((uProxy->width() != yProxy->width()) || (uProxy->height() != yProxy->height()) || (vProxy->width() != yProxy->width()) || (vProxy->height() != yProxy->height())) ? GrSamplerState::Filter::kBilerp : GrSamplerState::Filter::kNearest; SkMatrix44 mat(SkMatrix44::kUninitialized_Constructor); switch (colorSpace) { case kJPEG_SkYUVColorSpace: mat.setColMajorf(kJPEGConversionMatrix); break; case kRec601_SkYUVColorSpace: mat.setColMajorf(kRec601ConversionMatrix); break; case kRec709_SkYUVColorSpace: mat.setColMajorf(kRec709ConversionMatrix); break; } return std::unique_ptr(new GrYUVtoRGBEffect( std::move(yProxy), yTransform, std::move(uProxy), uTransform, std::move(vProxy), vTransform, mat, nv12, GrSamplerState(GrSamplerState::WrapMode::kClamp, uvFilterMode))); } SkString GrYUVtoRGBEffect::dumpInfo() const { SkString str; str.appendf("Y: %d %d U: %d %d V: %d %d\n", fYSampler.proxy()->uniqueID().asUInt(), fYSampler.proxy()->underlyingUniqueID().asUInt(), fUSampler.proxy()->uniqueID().asUInt(), fUSampler.proxy()->underlyingUniqueID().asUInt(), fVSampler.proxy()->uniqueID().asUInt(), fVSampler.proxy()->underlyingUniqueID().asUInt()); return str; } #include "glsl/GrGLSLFragmentProcessor.h" #include "glsl/GrGLSLFragmentShaderBuilder.h" #include "glsl/GrGLSLProgramBuilder.h" #include "GrTexture.h" #include "SkSLCPP.h" #include "SkSLUtil.h" class GrGLSLYUVtoRGBEffect : public GrGLSLFragmentProcessor { public: GrGLSLYUVtoRGBEffect() {} void emitCode(EmitArgs& args) override { GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder; const GrYUVtoRGBEffect& _outer = args.fFp.cast(); (void)_outer; auto ySamplerTransform = _outer.ySamplerTransform(); (void)ySamplerTransform; auto uSamplerTransform = _outer.uSamplerTransform(); (void)uSamplerTransform; auto vSamplerTransform = _outer.vSamplerTransform(); (void)vSamplerTransform; auto colorSpaceMatrix = _outer.colorSpaceMatrix(); (void)colorSpaceMatrix; auto nv12 = _outer.nv12(); (void)nv12; fColorSpaceMatrixVar = args.fUniformHandler->addUniform(kFragment_GrShaderFlag, kHalf4x4_GrSLType, kDefault_GrSLPrecision, "colorSpaceMatrix"); SkString sk_TransformedCoords2D_0 = fragBuilder->ensureCoords2D(args.fTransformedCoords[0]); SkString sk_TransformedCoords2D_1 = fragBuilder->ensureCoords2D(args.fTransformedCoords[1]); SkString sk_TransformedCoords2D_2 = fragBuilder->ensureCoords2D(args.fTransformedCoords[2]); fragBuilder->codeAppendf( "@if (%s) {\n %s = half4(texture(%s, %s).%s.x, texture(%s, %s).%s.xy, 1.0) * " "%s;\n} else {\n %s = half4(texture(%s, %s).%s.x, texture(%s, %s).%s.x, " "texture(%s, %s).%s.x, 1.0) * %s;\n}\n", (_outer.nv12() ? "true" : "false"), args.fOutputColor, fragBuilder->getProgramBuilder()->samplerVariable(args.fTexSamplers[0]).c_str(), sk_TransformedCoords2D_0.c_str(), fragBuilder->getProgramBuilder()->samplerSwizzle(args.fTexSamplers[0]).c_str(), fragBuilder->getProgramBuilder()->samplerVariable(args.fTexSamplers[1]).c_str(), sk_TransformedCoords2D_1.c_str(), fragBuilder->getProgramBuilder()->samplerSwizzle(args.fTexSamplers[1]).c_str(), args.fUniformHandler->getUniformCStr(fColorSpaceMatrixVar), args.fOutputColor, fragBuilder->getProgramBuilder()->samplerVariable(args.fTexSamplers[0]).c_str(), sk_TransformedCoords2D_0.c_str(), fragBuilder->getProgramBuilder()->samplerSwizzle(args.fTexSamplers[0]).c_str(), fragBuilder->getProgramBuilder()->samplerVariable(args.fTexSamplers[1]).c_str(), sk_TransformedCoords2D_1.c_str(), fragBuilder->getProgramBuilder()->samplerSwizzle(args.fTexSamplers[1]).c_str(), fragBuilder->getProgramBuilder()->samplerVariable(args.fTexSamplers[2]).c_str(), sk_TransformedCoords2D_2.c_str(), fragBuilder->getProgramBuilder()->samplerSwizzle(args.fTexSamplers[2]).c_str(), args.fUniformHandler->getUniformCStr(fColorSpaceMatrixVar)); } private: void onSetData(const GrGLSLProgramDataManager& pdman, const GrFragmentProcessor& _proc) override { const GrYUVtoRGBEffect& _outer = _proc.cast(); { float colorSpaceMatrixValue[16]; _outer.colorSpaceMatrix().asColMajorf(colorSpaceMatrixValue); pdman.setMatrix4f(fColorSpaceMatrixVar, colorSpaceMatrixValue); } } UniformHandle fColorSpaceMatrixVar; }; GrGLSLFragmentProcessor* GrYUVtoRGBEffect::onCreateGLSLInstance() const { return new GrGLSLYUVtoRGBEffect(); } void GrYUVtoRGBEffect::onGetGLSLProcessorKey(const GrShaderCaps& caps, GrProcessorKeyBuilder* b) const { b->add32((int32_t)fNv12); } bool GrYUVtoRGBEffect::onIsEqual(const GrFragmentProcessor& other) const { const GrYUVtoRGBEffect& that = other.cast(); (void)that; if (fYSampler != that.fYSampler) return false; if (fYSamplerTransform != that.fYSamplerTransform) return false; if (fUSampler != that.fUSampler) return false; if (fUSamplerTransform != that.fUSamplerTransform) return false; if (fVSampler != that.fVSampler) return false; if (fVSamplerTransform != that.fVSamplerTransform) return false; if (fColorSpaceMatrix != that.fColorSpaceMatrix) return false; if (fNv12 != that.fNv12) return false; return true; } GrYUVtoRGBEffect::GrYUVtoRGBEffect(const GrYUVtoRGBEffect& src) : INHERITED(kGrYUVtoRGBEffect_ClassID, src.optimizationFlags()) , fYSampler(src.fYSampler) , fYSamplerTransform(src.fYSamplerTransform) , fUSampler(src.fUSampler) , fUSamplerTransform(src.fUSamplerTransform) , fVSampler(src.fVSampler) , fVSamplerTransform(src.fVSamplerTransform) , fColorSpaceMatrix(src.fColorSpaceMatrix) , fNv12(src.fNv12) , fYSamplerCoordTransform(src.fYSamplerCoordTransform) , fUSamplerCoordTransform(src.fUSamplerCoordTransform) , fVSamplerCoordTransform(src.fVSamplerCoordTransform) { this->addTextureSampler(&fYSampler); this->addTextureSampler(&fUSampler); this->addTextureSampler(&fVSampler); this->addCoordTransform(&fYSamplerCoordTransform); this->addCoordTransform(&fUSamplerCoordTransform); this->addCoordTransform(&fVSamplerCoordTransform); } std::unique_ptr GrYUVtoRGBEffect::clone() const { return std::unique_ptr(new GrYUVtoRGBEffect(*this)); }