1 files changed, 99 insertions, 0 deletions
diff --git a/source/blender/draw/engines/eevee_next/shaders/eevee_depth_of_field_gather_comp.glsl b/source/blender/draw/engines/eevee_next/shaders/eevee_depth_of_field_gather_comp.glsl
new file mode 100644
index 00000000000..cf8dd7a36e6
--- /dev/null
+++ b/source/blender/draw/engines/eevee_next/shaders/eevee_depth_of_field_gather_comp.glsl
@@ -0,0 +1,99 @@
+
+/**
+ * Gather pass: Convolve foreground and background parts in separate passes.
+ *
+ * Using the min&max CoC tile buffer, we select the best appropriate method to blur the scene
+ *color. A fast gather path is taken if there is not many CoC variation inside the tile.
+ *
+ * We sample using an octaweb sampling pattern. We randomize the kernel center and each ring
+ * rotation to ensure maximum coverage.
+ *
+ * Outputs:
+ * - Color * Weight, Weight, Occlusion 'CoC' Depth (mean and variance)
+ **/
+
+#pragma BLENDER_REQUIRE(eevee_depth_of_field_accumulator_lib.glsl)
+
+void main()
+{
+  ivec2 tile_co = ivec2(gl_GlobalInvocationID.xy / DOF_TILES_SIZE);
+  CocTile coc_tile = dof_coc_tile_load(in_tiles_fg_img, in_tiles_bg_img, tile_co);
+  CocTilePrediction prediction = dof_coc_tile_prediction_get(coc_tile);
+
+  float base_radius, min_radius, min_intersectable_radius;
+  bool can_early_out;
+  if (is_foreground) {
+    base_radius = -coc_tile.fg_min_coc;
+    min_radius = -coc_tile.fg_max_coc;
+    min_intersectable_radius = -coc_tile.fg_max_intersectable_coc;
+    can_early_out = !prediction.do_foreground;
+  }
+  else {
+    base_radius = coc_tile.bg_max_coc;
+    min_radius = coc_tile.bg_min_coc;
+    min_intersectable_radius = coc_tile.bg_min_intersectable_coc;
+    can_early_out = !prediction.do_background;
+  }
+
+  bool do_fast_gather = dof_do_fast_gather(base_radius, min_radius, is_foreground);
+
+  /* Gather at half resolution. Divide CoC by 2. */
+  base_radius *= 0.5;
+  min_intersectable_radius *= 0.5;
+
+  bool do_density_change = dof_do_density_change(base_radius, min_intersectable_radius);
+
+  vec4 out_color;
+  float out_weight;
+  vec2 out_occlusion;
+
+  if (can_early_out) {
+    out_color = vec4(0.0);
+    out_weight = 0.0;
+    out_occlusion = vec2(0.0, 0.0);
+  }
+  else if (do_fast_gather) {
+    dof_gather_accumulator(color_tx,
+                           color_bilinear_tx,
+                           coc_tx,
+                           bokeh_lut_tx,
+                           base_radius,
+                           min_intersectable_radius,
+                           true,
+                           false,
+                           out_color,
+                           out_weight,
+                           out_occlusion);
+  }
+  else if (do_density_change) {
+    dof_gather_accumulator(color_tx,
+                           color_bilinear_tx,
+                           coc_tx,
+                           bokeh_lut_tx,
+                           base_radius,
+                           min_intersectable_radius,
+                           false,
+                           true,
+                           out_color,
+                           out_weight,
+                           out_occlusion);
+  }
+  else {
+    dof_gather_accumulator(color_tx,
+                           color_bilinear_tx,
+                           coc_tx,
+                           bokeh_lut_tx,
+                           base_radius,
+                           min_intersectable_radius,
+                           false,
+                           false,
+                           out_color,
+                           out_weight,
+                           out_occlusion);
+  }
+
+  ivec2 out_texel = ivec2(gl_GlobalInvocationID.xy);
+  imageStore(out_color_img, out_texel, out_color);
+  imageStore(out_weight_img, out_texel, vec4(out_weight));
+  imageStore(out_occlusion_img, out_texel, out_occlusion.xyxy);
+}