[Bf-blender-cvs] [49fa3dd5ff0] sculpt-dev: Face Sets Topology: Initial commit

[Pablo Dobarro]

Commit: 49fa3dd5ff04d930d7359885677b68b229878e21
Author: Pablo Dobarro
Date:   Tue Mar 16 22:03:22 2021 +0100
Branches: sculpt-dev
https://developer.blender.org/rB49fa3dd5ff04d930d7359885677b68b229878e21

Face Sets Topology: Initial commit

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

M	source/blender/editors/sculpt_paint/CMakeLists.txt
A	source/blender/editors/sculpt_paint/sculpt_face_set_topology.c

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

diff --git a/source/blender/editors/sculpt_paint/CMakeLists.txt b/source/blender/editors/sculpt_paint/CMakeLists.txt
index 94272432eab..3edaeb5456f 100644
--- a/source/blender/editors/sculpt_paint/CMakeLists.txt
+++ b/source/blender/editors/sculpt_paint/CMakeLists.txt
@@ -63,6 +63,7 @@ set(SRC
   sculpt_dyntopo.c
   sculpt_expand.c
   sculpt_face_set.c
+  sculpt_face_set_topology.c
   sculpt_filter_color.c
   sculpt_filter_mask.c
   sculpt_filter_mesh.c
diff --git a/source/blender/editors/sculpt_paint/sculpt_face_set_topology.c b/source/blender/editors/sculpt_paint/sculpt_face_set_topology.c
new file mode 100644
index 00000000000..3e349fe366f
--- /dev/null
+++ b/source/blender/editors/sculpt_paint/sculpt_face_set_topology.c
@@ -0,0 +1,156 @@
+/*
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation; either version 2
+ * of the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ *
+ * The Original Code is Copyright (C) 2020 Blender Foundation.
+ * All rights reserved.
+ */
+
+/** \file
+ * \ingroup edsculpt
+ */
+
+#include "MEM_guardedalloc.h"
+
+#include "BLI_blenlib.h"
+#include "BLI_hash.h"
+#include "BLI_math.h"
+#include "BLI_task.h"
+
+#include "DNA_brush_types.h"
+#include "DNA_customdata_types.h"
+#include "DNA_mesh_types.h"
+#include "DNA_meshdata_types.h"
+#include "DNA_object_types.h"
+#include "DNA_scene_types.h"
+
+#include "BKE_brush.h"
+#include "BKE_ccg.h"
+#include "BKE_colortools.h"
+#include "BKE_context.h"
+#include "BKE_customdata.h"
+#include "BKE_mesh.h"
+#include "BKE_mesh_fair.h"
+#include "BKE_mesh_mapping.h"
+#include "BKE_multires.h"
+#include "BKE_node.h"
+#include "BKE_object.h"
+#include "BKE_paint.h"
+#include "BKE_pbvh.h"
+#include "BKE_scene.h"
+
+#include "DEG_depsgraph.h"
+
+#include "WM_api.h"
+#include "WM_message.h"
+#include "WM_toolsystem.h"
+#include "WM_types.h"
+
+#include "ED_object.h"
+#include "ED_screen.h"
+#include "ED_sculpt.h"
+#include "ED_view3d.h"
+#include "paint_intern.h"
+#include "sculpt_intern.h"
+
+#include "RNA_access.h"
+#include "RNA_define.h"
+
+#include "bmesh.h"
+
+#include <math.h>
+#include <stdlib.h>
+
+typedef enum eSculptFaceSetByTopologyMode {
+  SCULPT_FACE_SET_TOPOLOGY_LOOSE_PART = 0,
+  SCULPT_FACE_SET_TOPOLOGY_POLY_LOOP = 1,
+};
+
+
+static EnumPropertyItem prop_sculpt_face_set_by_topology[] = {
+    {
+        SCULPT_FACE_SET_TOPOLOGY_LOOSE_PART,
+        "LOOSE_PART",
+        0,
+        "Loose Part",
+        "",
+    },
+    {
+        SCULPT_FACE_SET_TOPOLOGY_POLY_LOOP,
+        "POLY_LOOP",
+        0,
+        "Face Loop",
+        "",
+    },
+    {0, NULL, 0, NULL, NULL},
+};
+
+static int sculpt_face_set_edit_invoke(bContext *C, wmOperator *op, const wmEvent *event)
+{
+  Object *ob = CTX_data_active_object(C);
+  SculptSession *ss = ob->sculpt;
+  Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C);
+
+  const int mode = RNA_enum_get(op->ptr, "mode");
+  BKE_sculpt_update_object_for_edit(depsgraph, ob, true, false, false);
+
+  /* Update the current active Face Set and Vertex as the operator can be used directly from the
+   * tool without brush cursor. */
+  SculptCursorGeometryInfo sgi;
+  const float mouse[2] = {event->mval[0], event->mval[1]};
+  if (!SCULPT_cursor_geometry_info_update(C, &sgi, mouse, false, false)) {
+    /* The cursor is not over the mesh. Cancel to avoid editing the last updated Face Set ID. */
+    return OPERATOR_CANCELLED;
+  }
+
+  PBVHNode **nodes;
+  int totnode;
+  BKE_pbvh_search_gather(ss->pbvh, NULL, NULL, &nodes, &totnode);
+  SCULPT_undo_push_begin(ob, "face set edit");
+  SCULPT_undo_push_node(ob, nodes[0], SCULPT_UNDO_FACE_SETS);
+  MEM_freeN(nodes);
+
+  Mesh *mesh = BKE_object_get_original_mesh(ob);
+  const int next_face_set = ED_sculpt_face_sets_find_next_available_id(mesh);
+
+  switch (mode) {
+    case SCULPT_FACE_SET_TOPOLOGY_LOOSE_PART:
+      break;
+
+    case SCULPT_FACE_SET_TOPOLOGY_POLY_LOOP:
+      break;
+  }
+
+  SCULPT_undo_push_end();
+  SCULPT_tag_update_overlays(C);
+
+  return OPERATOR_FINISHED;
+}
+
+void SCULPT_OT_face_set_by_topology(struct wmOperatorType *ot)
+{
+  /* Identifiers. */
+  ot->name = "Face Set by Topology";
+  ot->idname = "SCULPT_OT_face_set_by_topology";
+  ot->description = "Create a new Face Set following the mesh topology";
+
+  /* Api callbacks. */
+  ot->invoke = sculpt_face_set_edit_invoke;
+  ot->poll = SCULPT_mode_poll;
+
+  ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
+
+  RNA_def_enum(
+      ot->srna, "mode", prop_sculpt_face_set_by_topology, SCULPT_FACE_SET_TOPOLOGY_POLY_LOOP, "Mode", "");
+}