LineCallout.glsl.js 8.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131
  1. /*
  2. All material copyright ESRI, All Rights Reserved, unless otherwise specified.
  3. See https://js.arcgis.com/4.25/esri/copyright.txt for details.
  4. */
  5. import{isSome as e}from"../core/maybe.js";import{s as i}from"./vec2.js";import{a as o}from"./vec2f64.js";import{Z as r}from"./vec4f64.js";import{SliceDraw as t}from"../views/3d/webgl-engine/core/shaderLibrary/Slice.glsl.js";import{AlignPixel as a}from"../views/3d/webgl-engine/core/shaderLibrary/hud/AlignPixel.glsl.js";import{HUD as l}from"../views/3d/webgl-engine/core/shaderLibrary/hud/HUD.glsl.js";import{multipassGeometryTest as n}from"../views/3d/webgl-engine/core/shaderLibrary/shading/MultipassGeometryTest.glsl.js";import{addScreenSizePerspectiveAlignment as s}from"../views/3d/webgl-engine/core/shaderLibrary/util/ScreenSizePerspective.glsl.js";import{Float2PassUniform as d}from"../views/3d/webgl-engine/core/shaderModules/Float2PassUniform.js";import{Float4PassUniform as c}from"../views/3d/webgl-engine/core/shaderModules/Float4PassUniform.js";import{FloatPassUniform as f}from"../views/3d/webgl-engine/core/shaderModules/FloatPassUniform.js";import{glsl as p}from"../views/3d/webgl-engine/core/shaderModules/interfaces.js";import{ShaderBuilder as g}from"../views/3d/webgl-engine/core/shaderModules/ShaderBuilder.js";import{VertexAttribute as h}from"../views/3d/webgl-engine/lib/VertexAttribute.js";function S(o){const r=new g;r.include(a),r.include(l,o),r.include(t,o),r.attributes.add(h.UV0,"vec2");const{vertex:S,fragment:u}=r;return S.uniforms.add([new c("viewport",((e,i)=>i.camera.fullViewport)),new f("lineSize",((e,i)=>Math.ceil(e.size)*i.camera.pixelRatio)),new d("pixelToNDC",((e,o)=>i(m,2/o.camera.fullViewport[2],2/o.camera.fullViewport[3]))),new f("borderSize",((i,o)=>e(i.borderColor)?o.camera.pixelRatio:0)),new d("screenOffset",((e,o)=>i(m,e.screenOffset[0]*o.camera.pixelRatio,e.screenOffset[1]*o.camera.pixelRatio)))]),r.varyings.add("coverageSampling","vec4"),r.varyings.add("lineSizes","vec2"),o.hasMultipassGeometry&&r.varyings.add("depth","float"),o.hasScreenSizePerspective&&s(S),S.code.add(p`
  6. void main(void) {
  7. ProjectHUDAux projectAux;
  8. vec4 endPoint = projectPositionHUD(projectAux);
  9. vec3 vpos = projectAux.posModel;
  10. if (rejectBySlice(vpos)) {
  11. gl_Position = vec4(1e38, 1e38, 1e38, 1.0);
  12. return;
  13. }
  14. ${o.occlusionTestEnabled?p`
  15. if (!testVisibilityHUD(endPoint)) {
  16. gl_Position = vec4(1e38, 1e38, 1e38, 1.0);
  17. return;
  18. }`:""}
  19. ${o.hasScreenSizePerspective?p`
  20. vec4 perspectiveFactor = screenSizePerspectiveScaleFactor(projectAux.absCosAngle, projectAux.distanceToCamera, screenSizePerspectiveAlignment);
  21. vec2 screenOffsetScaled = applyScreenSizePerspectiveScaleFactorVec2(screenOffset, perspectiveFactor);
  22. `:p`
  23. vec2 screenOffsetScaled = screenOffset;
  24. `}
  25. // Add view dependent polygon offset to get exact same original starting point. This is mostly
  26. // used to get the correct depth value
  27. vec3 posView = (view * vec4(position, 1.0)).xyz;
  28. ${o.hasMultipassGeometry?"depth = posView.z;":""}
  29. applyHUDViewDependentPolygonOffset(auxpos1.w, projectAux.absCosAngle, posView);
  30. vec4 startPoint = proj * vec4(posView, 1.0);
  31. // Apply screen offset to both start and end point
  32. vec2 screenOffsetNorm = screenOffsetScaled * 2.0 / viewport.zw;
  33. startPoint.xy += screenOffsetNorm * startPoint.w;
  34. endPoint.xy += screenOffsetNorm * endPoint.w;
  35. // Align start and end to pixel origin
  36. vec4 startAligned = alignToPixelOrigin(startPoint, viewport.zw);
  37. vec4 endAligned = alignToPixelOrigin(endPoint, viewport.zw);
  38. ${o.depthHudEnabled?o.depthHudAlignStartEnabled?p`endAligned = vec4(endAligned.xy / endAligned.w * startAligned.w, startAligned.zw);`:p`startAligned = vec4(startAligned.xy / startAligned.w * endAligned.w, endAligned.zw);`:""}
  39. vec4 projectedPosition = mix(startAligned, endAligned, uv0.y);
  40. // The direction of the line in screen space
  41. vec2 screenSpaceDirection = normalize(endAligned.xy / endAligned.w - startAligned.xy / startAligned.w);
  42. vec2 perpendicularScreenSpaceDirection = vec2(screenSpaceDirection.y, -screenSpaceDirection.x);
  43. ${o.hasScreenSizePerspective?p`
  44. float lineSizeScaled = applyScreenSizePerspectiveScaleFactorFloat(lineSize, perspectiveFactor);
  45. float borderSizeScaled = applyScreenSizePerspectiveScaleFactorFloat(borderSize, perspectiveFactor);
  46. `:p`
  47. float lineSizeScaled = lineSize;
  48. float borderSizeScaled = borderSize;
  49. `}
  50. float halfPixelSize = lineSizeScaled * 0.5;
  51. // Calculate a pixel offset from the edge of the pixel, s.t. we keep the line aligned
  52. // to pixels if it has a full pixel size. Since pixel aligned biases to the bottom-left,
  53. // we bias the size to the right (for odd sizes) to balance out the bias. Grow sub-pixel
  54. // sizes towards the left or right s.t. there is a smooth transition (e.g. from 2 to 3 px).
  55. float halfWholePixelSize = floor(lineSizeScaled) * 0.5;
  56. float halfPixelSizeInt = floor(halfWholePixelSize);
  57. // Sub-pixel offset if we need to grow sub-pixels to the left
  58. float subpixelOffset = -fract(lineSizeScaled) * float(halfWholePixelSize > 0.0);
  59. // Pixel offset aligning to whole pixels and adding subpixel offset if needed
  60. float pixelOffset = -halfPixelSizeInt + subpixelOffset;
  61. // Compute full ndc offset, adding 1px padding for doing anti-aliasing and the border size
  62. float padding = 1.0 + borderSizeScaled;
  63. vec2 ndcOffset = (pixelOffset - padding + uv0.x * (lineSizeScaled + padding + padding)) * pixelToNDC;
  64. // Offset x/y from the center of the line in screen space
  65. projectedPosition.xy += perpendicularScreenSpaceDirection * ndcOffset * projectedPosition.w;
  66. // Compute a coverage varying which we can use in the fragment shader to determine
  67. // how much a pixel is actually covered by the line (i.e. to anti alias the line).
  68. // This works by computing two coordinates that can be linearly interpolated and then
  69. // subtracted to find out how far away from the line edge we are.
  70. float edgeDirection = (uv0.x * 2.0 - 1.0);
  71. float halfBorderSize = 0.5 * borderSizeScaled;
  72. float halfPixelSizeAndBorder = halfPixelSize + halfBorderSize;
  73. float outerEdgeCoverageSampler = edgeDirection * (halfPixelSizeAndBorder + halfBorderSize + 1.0);
  74. float isOneSided = float(lineSizeScaled < 2.0 && borderSize < 2.0);
  75. coverageSampling = vec4(
  76. // Edge coordinate
  77. outerEdgeCoverageSampler,
  78. // Border edge coordinate
  79. outerEdgeCoverageSampler - halfPixelSizeAndBorder * isOneSided,
  80. // Line offset
  81. halfPixelSize - 0.5,
  82. // Border offset
  83. halfBorderSize - 0.5 + halfPixelSizeAndBorder * (1.0 - isOneSided)
  84. );
  85. lineSizes = vec2(lineSizeScaled, borderSizeScaled);
  86. gl_Position = projectedPosition;
  87. }
  88. `),u.uniforms.add([new c("uColor",(e=>v(e.color))),new c("borderColor",(e=>v(e.borderColor)))]),o.hasMultipassGeometry&&(u.include(n,o),u.uniforms.add(new d("inverseViewport",((e,i)=>i.inverseViewport)))),u.code.add(p`
  89. void main() {
  90. ${o.hasMultipassGeometry?"if( geometryDepthTest(gl_FragCoord.xy * inverseViewport, depth) ){ discard; }":""}
  91. // Mix between line and border coverage offsets depending on whether we need
  92. // a border (based on the sidedness).
  93. vec2 coverage = min(1.0 - clamp(abs(coverageSampling.xy) - coverageSampling.zw, 0.0, 1.0), lineSizes);
  94. // Mix between border and line color based on the line coverage (conceptually the line
  95. // blends on top of the border background).
  96. //
  97. // Anti-alias by blending final result using the full (including optional border) coverage
  98. // and the color alpha
  99. float borderAlpha = uColor.a * borderColor.a * coverage.y;
  100. float colorAlpha = uColor.a * coverage.x;
  101. float finalAlpha = mix(borderAlpha, 1.0, colorAlpha);
  102. ${o.depthHudEnabled?p`
  103. if (finalAlpha < 0.01) {
  104. discard;
  105. }
  106. `:p`
  107. vec3 finalRgb = mix(borderColor.rgb * borderAlpha, uColor.rgb, colorAlpha);
  108. gl_FragColor = vec4(finalRgb, finalAlpha);
  109. `}
  110. }
  111. `),r}function v(i){return e(i)?i:r}const m=o(),u=Object.freeze(Object.defineProperty({__proto__:null,build:S},Symbol.toStringTag,{value:"Module"}));export{u as L,S as b};