GrShadowGeoProc.cpp 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  1. /*
  2. * Copyright 2016 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. #include "src/gpu/effects/GrShadowGeoProc.h"
  8. #include "src/gpu/glsl/GrGLSLFragmentShaderBuilder.h"
  9. #include "src/gpu/glsl/GrGLSLGeometryProcessor.h"
  10. #include "src/gpu/glsl/GrGLSLUniformHandler.h"
  11. #include "src/gpu/glsl/GrGLSLVarying.h"
  12. #include "src/gpu/glsl/GrGLSLVertexGeoBuilder.h"
  13. class GrGLSLRRectShadowGeoProc : public GrGLSLGeometryProcessor {
  14. public:
  15. GrGLSLRRectShadowGeoProc() {}
  16. void onEmitCode(EmitArgs& args, GrGPArgs* gpArgs) override {
  17. const GrRRectShadowGeoProc& rsgp = args.fGP.cast<GrRRectShadowGeoProc>();
  18. GrGLSLVertexBuilder* vertBuilder = args.fVertBuilder;
  19. GrGLSLVaryingHandler* varyingHandler = args.fVaryingHandler;
  20. GrGLSLUniformHandler* uniformHandler = args.fUniformHandler;
  21. GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder;
  22. // emit attributes
  23. varyingHandler->emitAttributes(rsgp);
  24. fragBuilder->codeAppend("half3 shadowParams;");
  25. varyingHandler->addPassThroughAttribute(rsgp.inShadowParams(), "shadowParams");
  26. // setup pass through color
  27. varyingHandler->addPassThroughAttribute(rsgp.inColor(), args.fOutputColor);
  28. // Setup position
  29. this->writeOutputPosition(vertBuilder, gpArgs, rsgp.inPosition().name());
  30. // emit transforms
  31. this->emitTransforms(vertBuilder,
  32. varyingHandler,
  33. uniformHandler,
  34. rsgp.inPosition().asShaderVar(),
  35. args.fFPCoordTransformHandler);
  36. fragBuilder->codeAppend("half d = length(shadowParams.xy);");
  37. fragBuilder->codeAppend("half distance = shadowParams.z * (1.0 - d);");
  38. fragBuilder->codeAppend("half factor = 1.0 - clamp(distance, 0.0, 1.0);");
  39. fragBuilder->codeAppend("factor = exp(-factor * factor * 4.0) - 0.018;");
  40. fragBuilder->codeAppendf("%s = half4(factor);",
  41. args.fOutputCoverage);
  42. }
  43. void setData(const GrGLSLProgramDataManager& pdman, const GrPrimitiveProcessor& proc,
  44. FPCoordTransformIter&& transformIter) override {
  45. this->setTransformDataHelper(SkMatrix::I(), pdman, &transformIter);
  46. }
  47. private:
  48. typedef GrGLSLGeometryProcessor INHERITED;
  49. };
  50. ///////////////////////////////////////////////////////////////////////////////
  51. GrRRectShadowGeoProc::GrRRectShadowGeoProc() : INHERITED(kGrRRectShadowGeoProc_ClassID) {
  52. fInPosition = {"inPosition", kFloat2_GrVertexAttribType, kFloat2_GrSLType};
  53. fInColor = {"inColor", kUByte4_norm_GrVertexAttribType, kHalf4_GrSLType};
  54. fInShadowParams = {"inShadowParams", kFloat3_GrVertexAttribType, kHalf3_GrSLType};
  55. this->setVertexAttributes(&fInPosition, 3);
  56. }
  57. GrGLSLPrimitiveProcessor* GrRRectShadowGeoProc::createGLSLInstance(const GrShaderCaps&) const {
  58. return new GrGLSLRRectShadowGeoProc();
  59. }
  60. ///////////////////////////////////////////////////////////////////////////////
  61. GR_DEFINE_GEOMETRY_PROCESSOR_TEST(GrRRectShadowGeoProc);
  62. #if GR_TEST_UTILS
  63. sk_sp<GrGeometryProcessor> GrRRectShadowGeoProc::TestCreate(GrProcessorTestData* d) {
  64. return GrRRectShadowGeoProc::Make();
  65. }
  66. #endif