1 files changed, 11 insertions, 12 deletions
diff --git a/source/blender/gpu/shaders/material/gpu_shader_material_wireframe.glsl b/source/blender/gpu/shaders/material/gpu_shader_material_wireframe.glsl
index e2789e046e1..0c02dab3ae4 100644
--- a/source/blender/gpu/shaders/material/gpu_shader_material_wireframe.glsl
+++ b/source/blender/gpu/shaders/material/gpu_shader_material_wireframe.glsl
@@ -1,20 +1,21 @@
-#ifndef VOLUMETRICS
-void node_wireframe(float size, vec2 barycentric, vec3 barycentric_dist, out float fac)
+
+void node_wireframe(float size, out float fac)
 {
-  vec3 barys = barycentric.xyy;
-  barys.z = 1.0 - barycentric.x - barycentric.y;
+  vec3 barys = g_data.barycentric_coords.xyy;
+  barys.z = 1.0 - barys.x - barys.y;
 
   size *= 0.5;
-  vec3 s = step(-size, -barys * barycentric_dist);
+  vec3 s = step(-size, -barys * g_data.barycentric_dists);
 
   fac = max(s.x, max(s.y, s.z));
 }
 
-void node_wireframe_screenspace(float size, vec2 barycentric, out float fac)
+void node_wireframe_screenspace(float size, out float fac)
 {
-  vec3 barys = barycentric.xyy;
-  barys.z = 1.0 - barycentric.x - barycentric.y;
+  vec3 barys = g_data.barycentric_coords.xyy;
+  barys.z = 1.0 - barys.x - barys.y;
 
+#ifdef GPU_FRAGMENT_SHADER
   size *= (1.0 / 3.0);
   vec3 dx = dFdx(barys);
   vec3 dy = dFdy(barys);
@@ -23,9 +24,7 @@ void node_wireframe_screenspace(float size, vec2 barycentric, out float fac)
   vec3 s = step(-deltas * size, -barys);
 
   fac = max(s.x, max(s.y, s.z));
-}
 #else
-/* Stub wireframe because it is not compatible with volumetrics. */
-#  define node_wireframe(a, b, c, d) (d = 0.0)
-#  define node_wireframe_screenspace(a, b, c) (c = 0.0)
+  fac = 1.0;
 #endif
+}