[Bf-blender-cvs] [df192657da1] viewport-compositor: Viewport Compositor: Port Ellipse Mask node

[Omar Emara]

Commit: df192657da1b249217ae25d001f7373bd50fdc37
Author: Omar Emara
Date:   Sun Nov 14 21:15:18 2021 +0200
Branches: viewport-compositor
https://developer.blender.org/rBdf192657da1b249217ae25d001f7373bd50fdc37

Viewport Compositor: Port Ellipse Mask node

This patch ports the Ellipse Mask node to the viewport compositor.

===================================================================

M	source/blender/gpu/CMakeLists.txt
M	source/blender/gpu/intern/gpu_material_library.c
A	source/blender/gpu/shaders/composite/gpu_shader_composite_ellipse_mask.glsl
M	source/blender/nodes/composite/nodes/node_composite_ellipsemask.cc

===================================================================

diff --git a/source/blender/gpu/CMakeLists.txt b/source/blender/gpu/CMakeLists.txt
index 1d2b5b1c8dc..344f33b74bf 100644
--- a/source/blender/gpu/CMakeLists.txt
+++ b/source/blender/gpu/CMakeLists.txt
@@ -290,6 +290,7 @@ data_to_c_simple(shaders/composite/gpu_shader_composite_color_balance.glsl SRC)
 data_to_c_simple(shaders/composite/gpu_shader_composite_color_correction.glsl SRC)
 data_to_c_simple(shaders/composite/gpu_shader_composite_color_to_luminance.glsl SRC)
 data_to_c_simple(shaders/composite/gpu_shader_composite_composite.glsl SRC)
+data_to_c_simple(shaders/composite/gpu_shader_composite_ellipse_mask.glsl SRC)
 data_to_c_simple(shaders/composite/gpu_shader_composite_exposure.glsl SRC)
 data_to_c_simple(shaders/composite/gpu_shader_composite_gamma.glsl SRC)
 data_to_c_simple(shaders/composite/gpu_shader_composite_hue_correct.glsl SRC)
diff --git a/source/blender/gpu/intern/gpu_material_library.c b/source/blender/gpu/intern/gpu_material_library.c
index 3cb9e586d12..c90dffbc98d 100644
--- a/source/blender/gpu/intern/gpu_material_library.c
+++ b/source/blender/gpu/intern/gpu_material_library.c
@@ -56,6 +56,7 @@ extern char datatoc_gpu_shader_composite_color_balance_glsl[];
 extern char datatoc_gpu_shader_composite_color_correction_glsl[];
 extern char datatoc_gpu_shader_composite_color_to_luminance_glsl[];
 extern char datatoc_gpu_shader_composite_composite_glsl[];
+extern char datatoc_gpu_shader_composite_ellipse_mask_glsl[];
 extern char datatoc_gpu_shader_composite_exposure_glsl[];
 extern char datatoc_gpu_shader_composite_gamma_glsl[];
 extern char datatoc_gpu_shader_composite_hue_correct_glsl[];
@@ -228,6 +229,11 @@ static GPUMaterialLibrary gpu_shader_composite_composite_library = {
     .dependencies = {NULL},
 };
 
+static GPUMaterialLibrary gpu_shader_composite_ellipse_mask_library = {
+    .code = datatoc_gpu_shader_composite_ellipse_mask_glsl,
+    .dependencies = {NULL},
+};
+
 static GPUMaterialLibrary gpu_shader_composite_exposure_library = {
     .code = datatoc_gpu_shader_composite_exposure_glsl,
     .dependencies = {NULL},
@@ -795,6 +801,7 @@ static GPUMaterialLibrary *gpu_material_libraries[] = {
     &gpu_shader_composite_color_correction_library,
     &gpu_shader_composite_color_to_luminance_library,
     &gpu_shader_composite_composite_library,
+    &gpu_shader_composite_ellipse_mask_library,
     &gpu_shader_composite_exposure_library,
     &gpu_shader_composite_gamma_library,
     &gpu_shader_composite_hue_correct_library,
diff --git a/source/blender/gpu/shaders/composite/gpu_shader_composite_ellipse_mask.glsl b/source/blender/gpu/shaders/composite/gpu_shader_composite_ellipse_mask.glsl
new file mode 100644
index 00000000000..7c596cabdcb
--- /dev/null
+++ b/source/blender/gpu/shaders/composite/gpu_shader_composite_ellipse_mask.glsl
@@ -0,0 +1,40 @@
+#define CMP_NODE_MASKTYPE_ADD 0.0
+#define CMP_NODE_MASKTYPE_SUBTRACT 1.0
+#define CMP_NODE_MASKTYPE_MULTIPLY 2.0
+#define CMP_NODE_MASKTYPE_NOT 3.0
+
+void node_composite_ellipse_mask(float in_mask,
+                                 float value,
+                                 const float mask_type,
+                                 float x_location,
+                                 float y_location,
+                                 float half_width,
+                                 float half_height,
+                                 float cos_angle,
+                                 float sin_angle,
+                                 out float out_mask)
+{
+  vec2 uv = g_data.uv_render_layer.xy;
+  uv -= vec2(x_location, y_location);
+  uv.y *= ViewportSize.y / ViewportSize.x;
+  uv = mat2(cos_angle, -sin_angle, sin_angle, cos_angle) * uv;
+  bool is_inside = length(uv / vec2(half_width, half_height)) < 1.0;
+
+  if (mask_type == CMP_NODE_MASKTYPE_ADD) {
+    out_mask = is_inside ? max(in_mask, value) : in_mask;
+  }
+  else if (mask_type == CMP_NODE_MASKTYPE_SUBTRACT) {
+    out_mask = is_inside ? clamp(in_mask - value, 0.0, 1.0) : in_mask;
+  }
+  else if (mask_type == CMP_NODE_MASKTYPE_MULTIPLY) {
+    out_mask = is_inside ? in_mask * value : 0.0;
+  }
+  else if (mask_type == CMP_NODE_MASKTYPE_NOT) {
+    out_mask = is_inside ? (in_mask > 0.0 ? 0.0 : value) : in_mask;
+  }
+}
+
+#undef CMP_NODE_MASKTYPE_ADD
+#undef CMP_NODE_MASKTYPE_SUBTRACT
+#undef CMP_NODE_MASKTYPE_MULTIPLY
+#undef CMP_NODE_MASKTYPE_NOT
diff --git a/source/blender/nodes/composite/nodes/node_composite_ellipsemask.cc b/source/blender/nodes/composite/nodes/node_composite_ellipsemask.cc
index 67196fb0d35..adb62b3e488 100644
--- a/source/blender/nodes/composite/nodes/node_composite_ellipsemask.cc
+++ b/source/blender/nodes/composite/nodes/node_composite_ellipsemask.cc
@@ -44,6 +44,34 @@ static void node_composit_init_ellipsemask(bNodeTree *UNUSED(ntree), bNode *node
   node->storage = data;
 }
 
+static int node_composite_gpu_ellipsemask(GPUMaterial *mat,
+                                          bNode *node,
+                                          bNodeExecData *UNUSED(execdata),
+                                          GPUNodeStack *in,
+                                          GPUNodeStack *out)
+{
+  const NodeEllipseMask *data = (NodeEllipseMask *)node->storage;
+
+  const float mask_type = (float)node->custom1;
+  const float cos_angle = std::cos(data->rotation);
+  const float sin_angle = std::sin(data->rotation);
+  const float half_width = data->width / 2.0;
+  const float half_height = data->height / 2.0;
+
+  return GPU_stack_link(mat,
+                        node,
+                        "node_composite_ellipse_mask",
+                        in,
+                        out,
+                        GPU_constant(&mask_type),
+                        GPU_uniform(&data->x),
+                        GPU_uniform(&data->y),
+                        GPU_uniform(&half_width),
+                        GPU_uniform(&half_height),
+                        GPU_uniform(&cos_angle),
+                        GPU_uniform(&sin_angle));
+}
+
 void register_node_type_cmp_ellipsemask(void)
 {
   static bNodeType ntype;
@@ -54,6 +82,7 @@ void register_node_type_cmp_ellipsemask(void)
   node_type_init(&ntype, node_composit_init_ellipsemask);
   node_type_storage(
       &ntype, "NodeEllipseMask", node_free_standard_storage, node_copy_standard_storage);
+  node_type_gpu(&ntype, node_composite_gpu_ellipsemask);
 
   nodeRegisterType(&ntype);
 }