1b7325c318ecf01d4c82391c1f0a63090c8de0144Colin Cross/* 2b7325c318ecf01d4c82391c1f0a63090c8de0144Colin Cross * Copyright 2014 Google Inc. 336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines * 436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines * Use of this source code is governed by a BSD-style license that can be 5e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao * found in the LICENSE file. 6e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao */ 7a523d505e6da8650dd5a86474cec28d03109f346Shih-wei Liao 8a523d505e6da8650dd5a86474cec28d03109f346Shih-wei Liao#include "gl/builders/GrGLProgramBuilder.h" 9a523d505e6da8650dd5a86474cec28d03109f346Shih-wei Liao#include "GrBicubicEffect.h" 10e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao 11e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao 12e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao#define DS(x) SkDoubleToScalar(x) 13e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao 14e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liaoconst SkScalar GrBicubicEffect::gMitchellCoefficients[16] = { 15e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao DS( 1.0 / 18.0), DS(-9.0 / 18.0), DS( 15.0 / 18.0), DS( -7.0 / 18.0), 1636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines DS(16.0 / 18.0), DS( 0.0 / 18.0), DS(-36.0 / 18.0), DS( 21.0 / 18.0), 17e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao DS( 1.0 / 18.0), DS( 9.0 / 18.0), DS( 27.0 / 18.0), DS(-21.0 / 18.0), 18e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao DS( 0.0 / 18.0), DS( 0.0 / 18.0), DS( -6.0 / 18.0), DS( 7.0 / 18.0), 197fa502c8157d3e9584da0bf3c6f87f7ca70e5b69Jing Yu}; 207fa502c8157d3e9584da0bf3c6f87f7ca70e5b69Jing Yu 217fa502c8157d3e9584da0bf3c6f87f7ca70e5b69Jing Yu 227fa502c8157d3e9584da0bf3c6f87f7ca70e5b69Jing Yuclass GrGLBicubicEffect : public GrGLFragmentProcessor { 237fa502c8157d3e9584da0bf3c6f87f7ca70e5b69Jing Yupublic: 247fa502c8157d3e9584da0bf3c6f87f7ca70e5b69Jing Yu GrGLBicubicEffect(const GrBackendProcessorFactory& factory, 255183d48d560ae7c0c548bc23eb0101784cfd664dYing Wang const GrProcessor&); 26369c4a40cd8c97e1e513ec8c6eaeee26e6adfc9bDaniel Malea 27369c4a40cd8c97e1e513ec8c6eaeee26e6adfc9bDaniel Malea virtual void emitCode(GrGLProgramBuilder*, 28369c4a40cd8c97e1e513ec8c6eaeee26e6adfc9bDaniel Malea const GrFragmentProcessor&, 29369c4a40cd8c97e1e513ec8c6eaeee26e6adfc9bDaniel Malea const GrProcessorKey&, 30369c4a40cd8c97e1e513ec8c6eaeee26e6adfc9bDaniel Malea const char* outputColor, 3144ce444e5ee78c3230e92439366b48a13559ccd6Shih-wei Liao const char* inputColor, 3244ce444e5ee78c3230e92439366b48a13559ccd6Shih-wei Liao const TransformedCoordsArray&, 33fd6aea25184fd0735922461b88f059ee547844c3Stephen Hines const TextureSamplerArray&) SK_OVERRIDE; 3444ce444e5ee78c3230e92439366b48a13559ccd6Shih-wei Liao 35fd6aea25184fd0735922461b88f059ee547844c3Stephen Hines virtual void setData(const GrGLProgramDataManager&, const GrProcessor&) SK_OVERRIDE; 3644ce444e5ee78c3230e92439366b48a13559ccd6Shih-wei Liao 3744ce444e5ee78c3230e92439366b48a13559ccd6Shih-wei Liao static inline void GenKey(const GrProcessor& effect, const GrGLCaps&, 38e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao GrProcessorKeyBuilder* b) { 39e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao const GrTextureDomain& domain = effect.cast<GrBicubicEffect>().domain(); 40e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao b->add32(GrTextureDomain::GLDomain::DomainKey(domain)); 41c156bc696311928af01132d159b9e307436779bbShih-wei Liao } 42e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao 43e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liaoprivate: 44e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao typedef GrGLProgramDataManager::UniformHandle UniformHandle; 45e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao 46e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao UniformHandle fCoefficientsUni; 47c364886a209cb2712a4a8e55ba84a53e780fa78aPatrick Scott UniformHandle fImageIncrementUni; 48e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao GrTextureDomain::GLDomain fDomain; 49e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao 50e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao typedef GrGLFragmentProcessor INHERITED; 51e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao}; 527abe37e4aee38cc79d91dd069a37d7e91d5bef53Shih-wei Liao 537abe37e4aee38cc79d91dd069a37d7e91d5bef53Shih-wei LiaoGrGLBicubicEffect::GrGLBicubicEffect(const GrBackendProcessorFactory& factory, const GrProcessor&) 544671b174806e6c8577125a644aa85999598e4971jush : INHERITED(factory) { 5536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines} 5636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines 57c156bc696311928af01132d159b9e307436779bbShih-wei Liaovoid GrGLBicubicEffect::emitCode(GrGLProgramBuilder* builder, 58e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao const GrFragmentProcessor& effect, 59e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao const GrProcessorKey& key, 608aca41132270925dfb55502ea2126e2c464ac0ebJeff Brown const char* outputColor, 6136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines const char* inputColor, 62e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao const TransformedCoordsArray& coords, 63e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao const TextureSamplerArray& samplers) { 64e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao const GrTextureDomain& domain = effect.cast<GrBicubicEffect>().domain(); 65e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao 66e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao fCoefficientsUni = builder->addUniform(GrGLProgramBuilder::kFragment_Visibility, 67cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines kMat44f_GrSLType, "Coefficients"); 68cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines fImageIncrementUni = builder->addUniform(GrGLProgramBuilder::kFragment_Visibility, 69e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao kVec2f_GrSLType, "ImageIncrement"); 70e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao 71e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao const char* imgInc = builder->getUniformCStr(fImageIncrementUni); 72e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao const char* coeff = builder->getUniformCStr(fCoefficientsUni); 73e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao 7456837864265a27fdecc85c85f4c1cec73845cc94Logan Chien SkString cubicBlendName; 757b17ef61ee92cbea0cc4d321e95046d3179e21adStephen Hines 76e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao static const GrGLShaderVar gCubicBlendArgs[] = { 77e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao GrGLShaderVar("coefficients", kMat44f_GrSLType), 78e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao GrGLShaderVar("t", kFloat_GrSLType), 79e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao GrGLShaderVar("c0", kVec4f_GrSLType), 80e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao GrGLShaderVar("c1", kVec4f_GrSLType), 81ca65f4e8b7d5566d70a99273a4baabf4dbf08c05Ying Wang GrGLShaderVar("c2", kVec4f_GrSLType), 82e4454320b3cfffe926a487c33fbeb454366de2f8Shih-wei Liao GrGLShaderVar("c3", kVec4f_GrSLType), 83 }; 84 GrGLFragmentShaderBuilder* fsBuilder = builder->getFragmentShaderBuilder(); 85 SkString coords2D = fsBuilder->ensureFSCoords2D(coords, 0); 86 fsBuilder->emitFunction(kVec4f_GrSLType, 87 "cubicBlend", 88 SK_ARRAY_COUNT(gCubicBlendArgs), 89 gCubicBlendArgs, 90 "\tvec4 ts = vec4(1.0, t, t * t, t * t * t);\n" 91 "\tvec4 c = coefficients * ts;\n" 92 "\treturn c.x * c0 + c.y * c1 + c.z * c2 + c.w * c3;\n", 93 &cubicBlendName); 94 fsBuilder->codeAppendf("\tvec2 coord = %s - %s * vec2(0.5);\n", coords2D.c_str(), imgInc); 95 // We unnormalize the coord in order to determine our fractional offset (f) within the texel 96 // We then snap coord to a texel center and renormalize. The snap prevents cases where the 97 // starting coords are near a texel boundary and accumulations of imgInc would cause us to skip/ 98 // double hit a texel. 99 fsBuilder->codeAppendf("\tcoord /= %s;\n", imgInc); 100 fsBuilder->codeAppend("\tvec2 f = fract(coord);\n"); 101 fsBuilder->codeAppendf("\tcoord = (coord - f + vec2(0.5)) * %s;\n", imgInc); 102 fsBuilder->codeAppend("\tvec4 rowColors[4];\n"); 103 for (int y = 0; y < 4; ++y) { 104 for (int x = 0; x < 4; ++x) { 105 SkString coord; 106 coord.printf("coord + %s * vec2(%d, %d)", imgInc, x - 1, y - 1); 107 SkString sampleVar; 108 sampleVar.printf("rowColors[%d]", x); 109 fDomain.sampleTexture(fsBuilder, domain, sampleVar.c_str(), coord, samplers[0]); 110 } 111 fsBuilder->codeAppendf("\tvec4 s%d = %s(%s, f.x, rowColors[0], rowColors[1], rowColors[2], rowColors[3]);\n", y, cubicBlendName.c_str(), coeff); 112 } 113 SkString bicubicColor; 114 bicubicColor.printf("%s(%s, f.y, s0, s1, s2, s3)", cubicBlendName.c_str(), coeff); 115 fsBuilder->codeAppendf("\t%s = %s;\n", outputColor, (GrGLSLExpr4(bicubicColor.c_str()) * GrGLSLExpr4(inputColor)).c_str()); 116} 117 118void GrGLBicubicEffect::setData(const GrGLProgramDataManager& pdman, 119 const GrProcessor& processor) { 120 const GrBicubicEffect& bicubicEffect = processor.cast<GrBicubicEffect>(); 121 const GrTexture& texture = *processor.texture(0); 122 float imageIncrement[2]; 123 imageIncrement[0] = 1.0f / texture.width(); 124 imageIncrement[1] = 1.0f / texture.height(); 125 pdman.set2fv(fImageIncrementUni, 1, imageIncrement); 126 pdman.setMatrix4f(fCoefficientsUni, bicubicEffect.coefficients()); 127 fDomain.setData(pdman, bicubicEffect.domain(), texture.origin()); 128} 129 130static inline void convert_row_major_scalar_coeffs_to_column_major_floats(float dst[16], 131 const SkScalar src[16]) { 132 for (int y = 0; y < 4; y++) { 133 for (int x = 0; x < 4; x++) { 134 dst[x * 4 + y] = SkScalarToFloat(src[y * 4 + x]); 135 } 136 } 137} 138 139GrBicubicEffect::GrBicubicEffect(GrTexture* texture, 140 const SkScalar coefficients[16], 141 const SkMatrix &matrix, 142 const SkShader::TileMode tileModes[2]) 143 : INHERITED(texture, matrix, GrTextureParams(tileModes, GrTextureParams::kNone_FilterMode)) 144 , fDomain(GrTextureDomain::IgnoredDomain()) { 145 convert_row_major_scalar_coeffs_to_column_major_floats(fCoefficients, coefficients); 146} 147 148GrBicubicEffect::GrBicubicEffect(GrTexture* texture, 149 const SkScalar coefficients[16], 150 const SkMatrix &matrix, 151 const SkRect& domain) 152 : INHERITED(texture, matrix, GrTextureParams(SkShader::kClamp_TileMode, 153 GrTextureParams::kNone_FilterMode)) 154 , fDomain(domain, GrTextureDomain::kClamp_Mode) { 155 convert_row_major_scalar_coeffs_to_column_major_floats(fCoefficients, coefficients); 156} 157 158GrBicubicEffect::~GrBicubicEffect() { 159} 160 161const GrBackendFragmentProcessorFactory& GrBicubicEffect::getFactory() const { 162 return GrTBackendFragmentProcessorFactory<GrBicubicEffect>::getInstance(); 163} 164 165bool GrBicubicEffect::onIsEqual(const GrProcessor& sBase) const { 166 const GrBicubicEffect& s = sBase.cast<GrBicubicEffect>(); 167 return this->textureAccess(0) == s.textureAccess(0) && 168 !memcmp(fCoefficients, s.coefficients(), 16) && 169 fDomain == s.fDomain; 170} 171 172void GrBicubicEffect::getConstantColorComponents(GrColor* color, uint32_t* validFlags) const { 173 // FIXME: Perhaps we can do better. 174 *validFlags = 0; 175 return; 176} 177 178GR_DEFINE_FRAGMENT_PROCESSOR_TEST(GrBicubicEffect); 179 180GrFragmentProcessor* GrBicubicEffect::TestCreate(SkRandom* random, 181 GrContext* context, 182 const GrDrawTargetCaps&, 183 GrTexture* textures[]) { 184 int texIdx = random->nextBool() ? GrProcessorUnitTest::kSkiaPMTextureIdx : 185 GrProcessorUnitTest::kAlphaTextureIdx; 186 SkScalar coefficients[16]; 187 for (int i = 0; i < 16; i++) { 188 coefficients[i] = random->nextSScalar1(); 189 } 190 return GrBicubicEffect::Create(textures[texIdx], coefficients); 191} 192 193////////////////////////////////////////////////////////////////////////////// 194 195bool GrBicubicEffect::ShouldUseBicubic(const SkMatrix& matrix, 196 GrTextureParams::FilterMode* filterMode) { 197 if (matrix.isIdentity()) { 198 *filterMode = GrTextureParams::kNone_FilterMode; 199 return false; 200 } 201 202 SkScalar scales[2]; 203 if (!matrix.getMinMaxScales(scales) || scales[0] < SK_Scalar1) { 204 // Bicubic doesn't handle arbitrary minimization well, as src texels can be skipped 205 // entirely, 206 *filterMode = GrTextureParams::kMipMap_FilterMode; 207 return false; 208 } 209 // At this point if scales[1] == SK_Scalar1 then the matrix doesn't do any scaling. 210 if (scales[1] == SK_Scalar1) { 211 if (matrix.rectStaysRect() && SkScalarIsInt(matrix.getTranslateX()) && 212 SkScalarIsInt(matrix.getTranslateY())) { 213 *filterMode = GrTextureParams::kNone_FilterMode; 214 } else { 215 // Use bilerp to handle rotation or fractional translation. 216 *filterMode = GrTextureParams::kBilerp_FilterMode; 217 } 218 return false; 219 } 220 // When we use the bicubic filtering effect each sample is read from the texture using 221 // nearest neighbor sampling. 222 *filterMode = GrTextureParams::kNone_FilterMode; 223 return true; 224} 225