Cleanup: style, use braces for editors
[blender.git] / source / blender / editors / mesh / editmesh_tools.c
1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  *
16  * The Original Code is Copyright (C) 2004 by Blender Foundation.
17  * All rights reserved.
18  */
19
20 /** \file
21  * \ingroup edmesh
22  */
23
24 #include <stddef.h>
25
26 #include "MEM_guardedalloc.h"
27
28 #include "DNA_key_types.h"
29 #include "DNA_material_types.h"
30 #include "DNA_mesh_types.h"
31 #include "DNA_meshdata_types.h"
32 #include "DNA_modifier_types.h"
33 #include "DNA_object_types.h"
34 #include "DNA_scene_types.h"
35
36 #include "BLI_bitmap.h"
37 #include "BLI_heap_simple.h"
38 #include "BLI_listbase.h"
39 #include "BLI_linklist.h"
40 #include "BLI_linklist_stack.h"
41 #include "BLI_math.h"
42 #include "BLI_rand.h"
43 #include "BLI_sort_utils.h"
44 #include "BLI_string.h"
45
46 #include "BKE_editmesh.h"
47 #include "BKE_context.h"
48 #include "BKE_deform.h"
49 #include "BKE_key.h"
50 #include "BKE_layer.h"
51 #include "BKE_library.h"
52 #include "BKE_main.h"
53 #include "BKE_material.h"
54 #include "BKE_mesh.h"
55 #include "BKE_report.h"
56 #include "BKE_texture.h"
57
58 #include "DEG_depsgraph.h"
59 #include "DEG_depsgraph_build.h"
60
61 #include "BLT_translation.h"
62
63 #include "RNA_define.h"
64 #include "RNA_access.h"
65 #include "RNA_enum_types.h"
66
67 #include "WM_api.h"
68 #include "WM_types.h"
69
70 #include "ED_mesh.h"
71 #include "ED_object.h"
72 #include "ED_screen.h"
73 #include "ED_transform.h"
74 #include "ED_transform_snap_object_context.h"
75 #include "ED_uvedit.h"
76 #include "ED_view3d.h"
77
78 #include "RE_render_ext.h"
79
80 #include "UI_interface.h"
81 #include "UI_resources.h"
82
83 #include "mesh_intern.h" /* own include */
84
85 #include "bmesh_tools.h"
86
87 #define USE_FACE_CREATE_SEL_EXTEND
88
89 /* -------------------------------------------------------------------- */
90 /** \name Subdivide Operator
91  * \{ */
92
93 static int edbm_subdivide_exec(bContext *C, wmOperator *op)
94 {
95   const int cuts = RNA_int_get(op->ptr, "number_cuts");
96   const float smooth = RNA_float_get(op->ptr, "smoothness");
97   const float fractal = RNA_float_get(op->ptr, "fractal") / 2.5f;
98   const float along_normal = RNA_float_get(op->ptr, "fractal_along_normal");
99
100   if (RNA_boolean_get(op->ptr, "ngon") &&
101       RNA_enum_get(op->ptr, "quadcorner") == SUBD_CORNER_STRAIGHT_CUT) {
102     RNA_enum_set(op->ptr, "quadcorner", SUBD_CORNER_INNERVERT);
103   }
104   const int quad_corner_type = RNA_enum_get(op->ptr, "quadcorner");
105   const bool use_quad_tri = !RNA_boolean_get(op->ptr, "ngon");
106   const int seed = RNA_int_get(op->ptr, "seed");
107
108   ViewLayer *view_layer = CTX_data_view_layer(C);
109   uint objects_len = 0;
110   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
111       view_layer, CTX_wm_view3d(C), &objects_len);
112
113   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
114     Object *obedit = objects[ob_index];
115     BMEditMesh *em = BKE_editmesh_from_object(obedit);
116
117     if (!(em->bm->totedgesel || em->bm->totfacesel)) {
118       continue;
119     }
120
121     BM_mesh_esubdivide(em->bm,
122                        BM_ELEM_SELECT,
123                        smooth,
124                        SUBD_FALLOFF_LIN,
125                        false,
126                        fractal,
127                        along_normal,
128                        cuts,
129                        SUBDIV_SELECT_ORIG,
130                        quad_corner_type,
131                        use_quad_tri,
132                        true,
133                        false,
134                        seed);
135
136     EDBM_update_generic(em, true, true);
137   }
138
139   MEM_freeN(objects);
140
141   return OPERATOR_FINISHED;
142 }
143
144 /* Note, these values must match delete_mesh() event values */
145 static const EnumPropertyItem prop_mesh_cornervert_types[] = {
146     {SUBD_CORNER_INNERVERT, "INNERVERT", 0, "Inner Vert", ""},
147     {SUBD_CORNER_PATH, "PATH", 0, "Path", ""},
148     {SUBD_CORNER_STRAIGHT_CUT, "STRAIGHT_CUT", 0, "Straight Cut", ""},
149     {SUBD_CORNER_FAN, "FAN", 0, "Fan", ""},
150     {0, NULL, 0, NULL, NULL},
151 };
152
153 void MESH_OT_subdivide(wmOperatorType *ot)
154 {
155   PropertyRNA *prop;
156
157   /* identifiers */
158   ot->name = "Subdivide";
159   ot->description = "Subdivide selected edges";
160   ot->idname = "MESH_OT_subdivide";
161
162   /* api callbacks */
163   ot->exec = edbm_subdivide_exec;
164   ot->poll = ED_operator_editmesh;
165
166   /* flags */
167   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
168
169   /* properties */
170   prop = RNA_def_int(ot->srna, "number_cuts", 1, 1, 100, "Number of Cuts", "", 1, 10);
171   /* avoid re-using last var because it can cause
172    * _very_ high poly meshes and annoy users (or worse crash) */
173   RNA_def_property_flag(prop, PROP_SKIP_SAVE);
174
175   RNA_def_float(
176       ot->srna, "smoothness", 0.0f, 0.0f, 1e3f, "Smoothness", "Smoothness factor", 0.0f, 1.0f);
177
178   WM_operatortype_props_advanced_begin(ot);
179
180   RNA_def_boolean(ot->srna,
181                   "ngon",
182                   true,
183                   "Create N-Gons",
184                   "When disabled, newly created faces are limited to 3-4 sided faces");
185   RNA_def_enum(
186       ot->srna,
187       "quadcorner",
188       prop_mesh_cornervert_types,
189       SUBD_CORNER_STRAIGHT_CUT,
190       "Quad Corner Type",
191       "How to subdivide quad corners (anything other than Straight Cut will prevent ngons)");
192
193   RNA_def_float(ot->srna,
194                 "fractal",
195                 0.0f,
196                 0.0f,
197                 1e6f,
198                 "Fractal",
199                 "Fractal randomness factor",
200                 0.0f,
201                 1000.0f);
202   RNA_def_float(ot->srna,
203                 "fractal_along_normal",
204                 0.0f,
205                 0.0f,
206                 1.0f,
207                 "Along Normal",
208                 "Apply fractal displacement along normal only",
209                 0.0f,
210                 1.0f);
211   RNA_def_int(ot->srna,
212               "seed",
213               0,
214               0,
215               INT_MAX,
216               "Random Seed",
217               "Seed for the random number generator",
218               0,
219               255);
220 }
221
222 /** \} */
223
224 /* -------------------------------------------------------------------- */
225 /** \name Edge Ring Subdivide Operator
226  *
227  * Bridge code shares props.
228  *
229  * \{ */
230
231 struct EdgeRingOpSubdProps {
232   int interp_mode;
233   int cuts;
234   float smooth;
235
236   int profile_shape;
237   float profile_shape_factor;
238 };
239
240 static void mesh_operator_edgering_props(wmOperatorType *ot,
241                                          const int cuts_min,
242                                          const int cuts_default)
243 {
244   /* Note, these values must match delete_mesh() event values */
245   static const EnumPropertyItem prop_subd_edgering_types[] = {
246       {SUBD_RING_INTERP_LINEAR, "LINEAR", 0, "Linear", ""},
247       {SUBD_RING_INTERP_PATH, "PATH", 0, "Blend Path", ""},
248       {SUBD_RING_INTERP_SURF, "SURFACE", 0, "Blend Surface", ""},
249       {0, NULL, 0, NULL, NULL},
250   };
251
252   PropertyRNA *prop;
253
254   prop = RNA_def_int(
255       ot->srna, "number_cuts", cuts_default, 0, 1000, "Number of Cuts", "", cuts_min, 64);
256   RNA_def_property_flag(prop, PROP_SKIP_SAVE);
257
258   RNA_def_enum(ot->srna,
259                "interpolation",
260                prop_subd_edgering_types,
261                SUBD_RING_INTERP_PATH,
262                "Interpolation",
263                "Interpolation method");
264
265   RNA_def_float(
266       ot->srna, "smoothness", 1.0f, 0.0f, 1e3f, "Smoothness", "Smoothness factor", 0.0f, 2.0f);
267
268   /* profile-shape */
269   RNA_def_float(ot->srna,
270                 "profile_shape_factor",
271                 0.0f,
272                 -1e3f,
273                 1e3f,
274                 "Profile Factor",
275                 "How much intermediary new edges are shrunk/expanded",
276                 -2.0f,
277                 2.0f);
278
279   prop = RNA_def_property(ot->srna, "profile_shape", PROP_ENUM, PROP_NONE);
280   RNA_def_property_enum_items(prop, rna_enum_proportional_falloff_curve_only_items);
281   RNA_def_property_enum_default(prop, PROP_SMOOTH);
282   RNA_def_property_ui_text(prop, "Profile Shape", "Shape of the profile");
283   RNA_def_property_translation_context(prop, BLT_I18NCONTEXT_ID_CURVE); /* Abusing id_curve :/ */
284 }
285
286 static void mesh_operator_edgering_props_get(wmOperator *op, struct EdgeRingOpSubdProps *op_props)
287 {
288   op_props->interp_mode = RNA_enum_get(op->ptr, "interpolation");
289   op_props->cuts = RNA_int_get(op->ptr, "number_cuts");
290   op_props->smooth = RNA_float_get(op->ptr, "smoothness");
291
292   op_props->profile_shape = RNA_enum_get(op->ptr, "profile_shape");
293   op_props->profile_shape_factor = RNA_float_get(op->ptr, "profile_shape_factor");
294 }
295
296 static int edbm_subdivide_edge_ring_exec(bContext *C, wmOperator *op)
297 {
298
299   ViewLayer *view_layer = CTX_data_view_layer(C);
300   uint objects_len = 0;
301   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
302       view_layer, CTX_wm_view3d(C), &objects_len);
303   struct EdgeRingOpSubdProps op_props;
304
305   mesh_operator_edgering_props_get(op, &op_props);
306
307   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
308     Object *obedit = objects[ob_index];
309     BMEditMesh *em = BKE_editmesh_from_object(obedit);
310
311     if (em->bm->totedgesel == 0) {
312       continue;
313     }
314
315     if (!EDBM_op_callf(em,
316                        op,
317                        "subdivide_edgering edges=%he interp_mode=%i cuts=%i smooth=%f "
318                        "profile_shape=%i profile_shape_factor=%f",
319                        BM_ELEM_SELECT,
320                        op_props.interp_mode,
321                        op_props.cuts,
322                        op_props.smooth,
323                        op_props.profile_shape,
324                        op_props.profile_shape_factor)) {
325       continue;
326     }
327
328     EDBM_update_generic(em, true, true);
329   }
330
331   MEM_freeN(objects);
332   return OPERATOR_FINISHED;
333 }
334
335 void MESH_OT_subdivide_edgering(wmOperatorType *ot)
336 {
337   /* identifiers */
338   ot->name = "Subdivide Edge-Ring";
339   ot->idname = "MESH_OT_subdivide_edgering";
340
341   /* api callbacks */
342   ot->exec = edbm_subdivide_edge_ring_exec;
343   ot->poll = ED_operator_editmesh;
344
345   /* flags */
346   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
347
348   /* properties */
349   mesh_operator_edgering_props(ot, 1, 10);
350 }
351
352 /** \} */
353
354 /* -------------------------------------------------------------------- */
355 /** \name Un-Subdivide Operator
356  * \{ */
357
358 static int edbm_unsubdivide_exec(bContext *C, wmOperator *op)
359 {
360   const int iterations = RNA_int_get(op->ptr, "iterations");
361   ViewLayer *view_layer = CTX_data_view_layer(C);
362   uint objects_len = 0;
363   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode(
364       view_layer, CTX_wm_view3d(C), &objects_len);
365   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
366     Object *obedit = objects[ob_index];
367     BMEditMesh *em = BKE_editmesh_from_object(obedit);
368
369     if ((em->bm->totvertsel == 0) && (em->bm->totedgesel == 0) && (em->bm->totfacesel == 0)) {
370       continue;
371     }
372
373     BMOperator bmop;
374     EDBM_op_init(em, &bmop, op, "unsubdivide verts=%hv iterations=%i", BM_ELEM_SELECT, iterations);
375
376     BMO_op_exec(em->bm, &bmop);
377
378     if (!EDBM_op_finish(em, &bmop, op, true)) {
379       continue;
380     }
381
382     if ((em->selectmode & SCE_SELECT_VERTEX) == 0) {
383       EDBM_selectmode_flush_ex(em, SCE_SELECT_VERTEX); /* need to flush vert->face first */
384     }
385     EDBM_selectmode_flush(em);
386
387     EDBM_update_generic(em, true, true);
388   }
389   MEM_freeN(objects);
390
391   return OPERATOR_FINISHED;
392 }
393
394 void MESH_OT_unsubdivide(wmOperatorType *ot)
395 {
396   /* identifiers */
397   ot->name = "Un-Subdivide";
398   ot->description = "UnSubdivide selected edges & faces";
399   ot->idname = "MESH_OT_unsubdivide";
400
401   /* api callbacks */
402   ot->exec = edbm_unsubdivide_exec;
403   ot->poll = ED_operator_editmesh;
404
405   /* flags */
406   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
407
408   /* props */
409   RNA_def_int(
410       ot->srna, "iterations", 2, 1, 1000, "Iterations", "Number of times to unsubdivide", 1, 100);
411 }
412
413 void EDBM_project_snap_verts(bContext *C, ARegion *ar, BMEditMesh *em)
414 {
415   Main *bmain = CTX_data_main(C);
416   Object *obedit = em->ob;
417   BMIter iter;
418   BMVert *eve;
419
420   ED_view3d_init_mats_rv3d(obedit, ar->regiondata);
421
422   struct SnapObjectContext *snap_context = ED_transform_snap_object_context_create_view3d(
423       bmain, CTX_data_scene(C), CTX_data_depsgraph(C), 0, ar, CTX_wm_view3d(C));
424
425   BM_ITER_MESH (eve, &iter, em->bm, BM_VERTS_OF_MESH) {
426     if (BM_elem_flag_test(eve, BM_ELEM_SELECT)) {
427       float mval[2], co_proj[3];
428       if (ED_view3d_project_float_object(ar, eve->co, mval, V3D_PROJ_TEST_NOP) ==
429           V3D_PROJ_RET_OK) {
430         if (ED_transform_snap_object_project_view3d(snap_context,
431                                                     SCE_SNAP_MODE_FACE,
432                                                     &(const struct SnapObjectParams){
433                                                         .snap_select = SNAP_NOT_ACTIVE,
434                                                         .use_object_edit_cage = false,
435                                                         .use_occlusion_test = true,
436                                                     },
437                                                     mval,
438                                                     NULL,
439                                                     co_proj,
440                                                     NULL)) {
441           mul_v3_m4v3(eve->co, obedit->imat, co_proj);
442         }
443       }
444     }
445   }
446
447   ED_transform_snap_object_context_destroy(snap_context);
448 }
449
450 /** \} */
451
452 /* -------------------------------------------------------------------- */
453 /** \name Delete Operator
454  * \{ */
455
456 /* Note, these values must match delete_mesh() event values */
457 enum {
458   MESH_DELETE_VERT = 0,
459   MESH_DELETE_EDGE = 1,
460   MESH_DELETE_FACE = 2,
461   MESH_DELETE_EDGE_FACE = 3,
462   MESH_DELETE_ONLY_FACE = 4,
463 };
464
465 static void edbm_report_delete_info(ReportList *reports,
466                                     const int totelem_old[3],
467                                     const int totelem_new[3])
468 {
469   BKE_reportf(reports,
470               RPT_INFO,
471               "Removed: %d vertices, %d edges, %d faces",
472               totelem_old[0] - totelem_new[0],
473               totelem_old[1] - totelem_new[1],
474               totelem_old[2] - totelem_new[2]);
475 }
476
477 static int edbm_delete_exec(bContext *C, wmOperator *op)
478 {
479   ViewLayer *view_layer = CTX_data_view_layer(C);
480
481   uint objects_len = 0;
482   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
483       view_layer, CTX_wm_view3d(C), &objects_len);
484   bool changed_multi = false;
485
486   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
487     Object *obedit = objects[ob_index];
488     BMEditMesh *em = BKE_editmesh_from_object(obedit);
489     const int type = RNA_enum_get(op->ptr, "type");
490
491     switch (type) {
492       case MESH_DELETE_VERT: /* Erase Vertices */
493         if (!(em->bm->totvertsel &&
494               EDBM_op_callf(em, op, "delete geom=%hv context=%i", BM_ELEM_SELECT, DEL_VERTS))) {
495           continue;
496         }
497         break;
498       case MESH_DELETE_EDGE: /* Erase Edges */
499         if (!(em->bm->totedgesel &&
500               EDBM_op_callf(em, op, "delete geom=%he context=%i", BM_ELEM_SELECT, DEL_EDGES))) {
501           continue;
502         }
503         break;
504       case MESH_DELETE_FACE: /* Erase Faces */
505         if (!(em->bm->totfacesel &&
506               EDBM_op_callf(em, op, "delete geom=%hf context=%i", BM_ELEM_SELECT, DEL_FACES))) {
507           continue;
508         }
509         break;
510       case MESH_DELETE_EDGE_FACE:
511         /* Edges and Faces */
512         if (!((em->bm->totedgesel || em->bm->totfacesel) &&
513               EDBM_op_callf(
514                   em, op, "delete geom=%hef context=%i", BM_ELEM_SELECT, DEL_EDGESFACES))) {
515           continue;
516         }
517         break;
518       case MESH_DELETE_ONLY_FACE:
519         /* Only faces. */
520         if (!(em->bm->totfacesel &&
521               EDBM_op_callf(
522                   em, op, "delete geom=%hf context=%i", BM_ELEM_SELECT, DEL_ONLYFACES))) {
523           continue;
524         }
525         break;
526       default:
527         BLI_assert(0);
528         break;
529     }
530
531     changed_multi = true;
532
533     EDBM_flag_disable_all(em, BM_ELEM_SELECT);
534
535     EDBM_update_generic(em, true, true);
536   }
537
538   MEM_freeN(objects);
539
540   return changed_multi ? OPERATOR_FINISHED : OPERATOR_CANCELLED;
541 }
542
543 void MESH_OT_delete(wmOperatorType *ot)
544 {
545   static const EnumPropertyItem prop_mesh_delete_types[] = {
546       {MESH_DELETE_VERT, "VERT", 0, "Vertices", ""},
547       {MESH_DELETE_EDGE, "EDGE", 0, "Edges", ""},
548       {MESH_DELETE_FACE, "FACE", 0, "Faces", ""},
549       {MESH_DELETE_EDGE_FACE, "EDGE_FACE", 0, "Only Edges & Faces", ""},
550       {MESH_DELETE_ONLY_FACE, "ONLY_FACE", 0, "Only Faces", ""},
551       {0, NULL, 0, NULL, NULL},
552   };
553
554   /* identifiers */
555   ot->name = "Delete";
556   ot->description = "Delete selected vertices, edges or faces";
557   ot->idname = "MESH_OT_delete";
558
559   /* api callbacks */
560   ot->invoke = WM_menu_invoke;
561   ot->exec = edbm_delete_exec;
562
563   ot->poll = ED_operator_editmesh;
564
565   /* flags */
566   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
567
568   /* props */
569   ot->prop = RNA_def_enum(ot->srna,
570                           "type",
571                           prop_mesh_delete_types,
572                           MESH_DELETE_VERT,
573                           "Type",
574                           "Method used for deleting mesh data");
575   RNA_def_property_flag(ot->prop, PROP_HIDDEN | PROP_SKIP_SAVE);
576 }
577
578 /** \} */
579
580 /* -------------------------------------------------------------------- */
581 /** \name Delete Loose Operator
582  * \{ */
583
584 static bool bm_face_is_loose(BMFace *f)
585 {
586   BMLoop *l_iter, *l_first;
587
588   l_iter = l_first = BM_FACE_FIRST_LOOP(f);
589   do {
590     if (!BM_edge_is_boundary(l_iter->e)) {
591       return false;
592     }
593   } while ((l_iter = l_iter->next) != l_first);
594
595   return true;
596 }
597
598 static int edbm_delete_loose_exec(bContext *C, wmOperator *op)
599 {
600   ViewLayer *view_layer = CTX_data_view_layer(C);
601   int totelem_old_sel[3];
602   int totelem_old[3];
603
604   uint objects_len = 0;
605   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
606       view_layer, CTX_wm_view3d(C), &objects_len);
607
608   EDBM_mesh_stats_multi(objects, objects_len, totelem_old, totelem_old_sel);
609
610   const bool use_verts = (RNA_boolean_get(op->ptr, "use_verts") && totelem_old_sel[0]);
611   const bool use_edges = (RNA_boolean_get(op->ptr, "use_edges") && totelem_old_sel[1]);
612   const bool use_faces = (RNA_boolean_get(op->ptr, "use_faces") && totelem_old_sel[2]);
613
614   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
615     Object *obedit = objects[ob_index];
616
617     BMEditMesh *em = BKE_editmesh_from_object(obedit);
618     BMesh *bm = em->bm;
619     BMIter iter;
620
621     BM_mesh_elem_hflag_disable_all(bm, BM_VERT | BM_EDGE | BM_FACE, BM_ELEM_TAG, false);
622
623     if (use_faces) {
624       BMFace *f;
625
626       BM_ITER_MESH (f, &iter, bm, BM_FACES_OF_MESH) {
627         if (BM_elem_flag_test(f, BM_ELEM_SELECT)) {
628           BM_elem_flag_set(f, BM_ELEM_TAG, bm_face_is_loose(f));
629         }
630       }
631
632       BM_mesh_delete_hflag_context(bm, BM_ELEM_TAG, DEL_FACES);
633     }
634
635     if (use_edges) {
636       BMEdge *e;
637
638       BM_ITER_MESH (e, &iter, bm, BM_EDGES_OF_MESH) {
639         if (BM_elem_flag_test(e, BM_ELEM_SELECT)) {
640           BM_elem_flag_set(e, BM_ELEM_TAG, BM_edge_is_wire(e));
641         }
642       }
643
644       BM_mesh_delete_hflag_context(bm, BM_ELEM_TAG, DEL_EDGES);
645     }
646
647     if (use_verts) {
648       BMVert *v;
649
650       BM_ITER_MESH (v, &iter, bm, BM_VERTS_OF_MESH) {
651         if (BM_elem_flag_test(v, BM_ELEM_SELECT)) {
652           BM_elem_flag_set(v, BM_ELEM_TAG, (v->e == NULL));
653         }
654       }
655
656       BM_mesh_delete_hflag_context(bm, BM_ELEM_TAG, DEL_VERTS);
657     }
658
659     EDBM_flag_disable_all(em, BM_ELEM_SELECT);
660
661     EDBM_update_generic(em, true, true);
662   }
663
664   int totelem_new[3];
665   EDBM_mesh_stats_multi(objects, objects_len, totelem_new, NULL);
666
667   edbm_report_delete_info(op->reports, totelem_old, totelem_new);
668
669   MEM_freeN(objects);
670
671   return OPERATOR_FINISHED;
672 }
673
674 void MESH_OT_delete_loose(wmOperatorType *ot)
675 {
676   /* identifiers */
677   ot->name = "Delete Loose";
678   ot->description = "Delete loose vertices, edges or faces";
679   ot->idname = "MESH_OT_delete_loose";
680
681   /* api callbacks */
682   ot->exec = edbm_delete_loose_exec;
683
684   ot->poll = ED_operator_editmesh;
685
686   /* flags */
687   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
688
689   /* props */
690   RNA_def_boolean(ot->srna, "use_verts", true, "Vertices", "Remove loose vertices");
691   RNA_def_boolean(ot->srna, "use_edges", true, "Edges", "Remove loose edges");
692   RNA_def_boolean(ot->srna, "use_faces", false, "Faces", "Remove loose faces");
693 }
694
695 /** \} */
696
697 /* -------------------------------------------------------------------- */
698 /** \name Collapse Edge Operator
699  * \{ */
700
701 static int edbm_collapse_edge_exec(bContext *C, wmOperator *op)
702 {
703   ViewLayer *view_layer = CTX_data_view_layer(C);
704   uint objects_len = 0;
705   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
706       view_layer, CTX_wm_view3d(C), &objects_len);
707   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
708     Object *obedit = objects[ob_index];
709     BMEditMesh *em = BKE_editmesh_from_object(obedit);
710
711     if (em->bm->totedgesel == 0) {
712       continue;
713     }
714
715     if (!EDBM_op_callf(em, op, "collapse edges=%he uvs=%b", BM_ELEM_SELECT, true)) {
716       continue;
717     }
718
719     EDBM_update_generic(em, true, true);
720   }
721   MEM_freeN(objects);
722
723   return OPERATOR_FINISHED;
724 }
725
726 void MESH_OT_edge_collapse(wmOperatorType *ot)
727 {
728   /* identifiers */
729   ot->name = "Edge Collapse";
730   ot->description = "Collapse selected edges";
731   ot->idname = "MESH_OT_edge_collapse";
732
733   /* api callbacks */
734   ot->exec = edbm_collapse_edge_exec;
735   ot->poll = ED_operator_editmesh;
736
737   /* flags */
738   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
739 }
740
741 /** \} */
742
743 /* -------------------------------------------------------------------- */
744 /** \name Create Edge/Face Operator
745  * \{ */
746
747 static bool edbm_add_edge_face__smooth_get(BMesh *bm)
748 {
749   BMEdge *e;
750   BMIter iter;
751
752   unsigned int vote_on_smooth[2] = {0, 0};
753
754   BM_ITER_MESH (e, &iter, bm, BM_EDGES_OF_MESH) {
755     if (BM_elem_flag_test(e, BM_ELEM_SELECT) && e->l) {
756       vote_on_smooth[BM_elem_flag_test_bool(e->l->f, BM_ELEM_SMOOTH)]++;
757     }
758   }
759
760   return (vote_on_smooth[0] < vote_on_smooth[1]);
761 }
762
763 #ifdef USE_FACE_CREATE_SEL_EXTEND
764 /**
765  * Function used to get a fixed number of edges linked to a vertex that passes a test function.
766  * This is used so we can request all boundary edges connected to a vertex for eg.
767  */
768 static int edbm_add_edge_face_exec__vert_edge_lookup(
769     BMVert *v, BMEdge *e_used, BMEdge **e_arr, const int e_arr_len, bool (*func)(const BMEdge *))
770 {
771   BMIter iter;
772   BMEdge *e_iter;
773   int i = 0;
774   BM_ITER_ELEM (e_iter, &iter, v, BM_EDGES_OF_VERT) {
775     if (BM_elem_flag_test(e_iter, BM_ELEM_HIDDEN) == false) {
776       if ((e_used == NULL) || (e_used != e_iter)) {
777         if (func(e_iter)) {
778           e_arr[i++] = e_iter;
779           if (i >= e_arr_len) {
780             break;
781           }
782         }
783       }
784     }
785   }
786   return i;
787 }
788
789 static BMElem *edbm_add_edge_face_exec__tricky_extend_sel(BMesh *bm)
790 {
791   BMIter iter;
792   bool found = false;
793
794   if (bm->totvertsel == 1 && bm->totedgesel == 0 && bm->totfacesel == 0) {
795     /* first look for 2 boundary edges */
796     BMVert *v;
797
798     BM_ITER_MESH (v, &iter, bm, BM_VERTS_OF_MESH) {
799       if (BM_elem_flag_test(v, BM_ELEM_SELECT)) {
800         found = true;
801         break;
802       }
803     }
804
805     if (found) {
806       BMEdge *ed_pair[3];
807       if (((edbm_add_edge_face_exec__vert_edge_lookup(v, NULL, ed_pair, 3, BM_edge_is_wire) ==
808             2) &&
809            (BM_edge_share_face_check(ed_pair[0], ed_pair[1]) == false)) ||
810
811           ((edbm_add_edge_face_exec__vert_edge_lookup(v, NULL, ed_pair, 3, BM_edge_is_boundary) ==
812             2) &&
813            (BM_edge_share_face_check(ed_pair[0], ed_pair[1]) == false))) {
814         BMEdge *e_other = BM_edge_exists(BM_edge_other_vert(ed_pair[0], v),
815                                          BM_edge_other_vert(ed_pair[1], v));
816         BM_edge_select_set(bm, ed_pair[0], true);
817         BM_edge_select_set(bm, ed_pair[1], true);
818         if (e_other) {
819           BM_edge_select_set(bm, e_other, true);
820         }
821         return (BMElem *)v;
822       }
823     }
824   }
825   else if (bm->totvertsel == 2 && bm->totedgesel == 1 && bm->totfacesel == 0) {
826     /* first look for 2 boundary edges */
827     BMEdge *e;
828
829     BM_ITER_MESH (e, &iter, bm, BM_EDGES_OF_MESH) {
830       if (BM_elem_flag_test(e, BM_ELEM_SELECT)) {
831         found = true;
832         break;
833       }
834     }
835     if (found) {
836       BMEdge *ed_pair_v1[2];
837       BMEdge *ed_pair_v2[2];
838       if (((edbm_add_edge_face_exec__vert_edge_lookup(e->v1, e, ed_pair_v1, 2, BM_edge_is_wire) ==
839             1) &&
840            (edbm_add_edge_face_exec__vert_edge_lookup(e->v2, e, ed_pair_v2, 2, BM_edge_is_wire) ==
841             1) &&
842            (BM_edge_share_face_check(e, ed_pair_v1[0]) == false) &&
843            (BM_edge_share_face_check(e, ed_pair_v2[0]) == false)) ||
844
845 #  if 1 /* better support mixed cases [#37203] */
846           ((edbm_add_edge_face_exec__vert_edge_lookup(e->v1, e, ed_pair_v1, 2, BM_edge_is_wire) ==
847             1) &&
848            (edbm_add_edge_face_exec__vert_edge_lookup(
849                 e->v2, e, ed_pair_v2, 2, BM_edge_is_boundary) == 1) &&
850            (BM_edge_share_face_check(e, ed_pair_v1[0]) == false) &&
851            (BM_edge_share_face_check(e, ed_pair_v2[0]) == false)) ||
852
853           ((edbm_add_edge_face_exec__vert_edge_lookup(
854                 e->v1, e, ed_pair_v1, 2, BM_edge_is_boundary) == 1) &&
855            (edbm_add_edge_face_exec__vert_edge_lookup(e->v2, e, ed_pair_v2, 2, BM_edge_is_wire) ==
856             1) &&
857            (BM_edge_share_face_check(e, ed_pair_v1[0]) == false) &&
858            (BM_edge_share_face_check(e, ed_pair_v2[0]) == false)) ||
859 #  endif
860
861           ((edbm_add_edge_face_exec__vert_edge_lookup(
862                 e->v1, e, ed_pair_v1, 2, BM_edge_is_boundary) == 1) &&
863            (edbm_add_edge_face_exec__vert_edge_lookup(
864                 e->v2, e, ed_pair_v2, 2, BM_edge_is_boundary) == 1) &&
865            (BM_edge_share_face_check(e, ed_pair_v1[0]) == false) &&
866            (BM_edge_share_face_check(e, ed_pair_v2[0]) == false))) {
867         BMVert *v1_other = BM_edge_other_vert(ed_pair_v1[0], e->v1);
868         BMVert *v2_other = BM_edge_other_vert(ed_pair_v2[0], e->v2);
869         BMEdge *e_other = (v1_other != v2_other) ? BM_edge_exists(v1_other, v2_other) : NULL;
870         BM_edge_select_set(bm, ed_pair_v1[0], true);
871         BM_edge_select_set(bm, ed_pair_v2[0], true);
872         if (e_other) {
873           BM_edge_select_set(bm, e_other, true);
874         }
875         return (BMElem *)e;
876       }
877     }
878   }
879
880   return NULL;
881 }
882 static void edbm_add_edge_face_exec__tricky_finalize_sel(BMesh *bm, BMElem *ele_desel, BMFace *f)
883 {
884   /* now we need to find the edge that isnt connected to this element */
885   BM_select_history_clear(bm);
886
887   /* Notes on hidden geometry:
888    * - Un-hide the face since its possible hidden was copied when copying
889    *   surrounding face attributes.
890    * - Un-hide before adding to select history
891    *   since we may extend into an existing, hidden vert/edge.
892    */
893
894   BM_elem_flag_disable(f, BM_ELEM_HIDDEN);
895   BM_face_select_set(bm, f, false);
896
897   if (ele_desel->head.htype == BM_VERT) {
898     BMLoop *l = BM_face_vert_share_loop(f, (BMVert *)ele_desel);
899     BLI_assert(f->len == 3);
900     BM_vert_select_set(bm, (BMVert *)ele_desel, false);
901     BM_edge_select_set(bm, l->next->e, true);
902     BM_select_history_store(bm, l->next->e);
903   }
904   else {
905     BMLoop *l = BM_face_edge_share_loop(f, (BMEdge *)ele_desel);
906     BLI_assert(f->len == 4 || f->len == 3);
907
908     BM_edge_select_set(bm, (BMEdge *)ele_desel, false);
909     if (f->len == 4) {
910       BMEdge *e_active = l->next->next->e;
911       BM_elem_flag_disable(e_active, BM_ELEM_HIDDEN);
912       BM_edge_select_set(bm, e_active, true);
913       BM_select_history_store(bm, e_active);
914     }
915     else {
916       BMVert *v_active = l->next->next->v;
917       BM_elem_flag_disable(v_active, BM_ELEM_HIDDEN);
918       BM_vert_select_set(bm, v_active, true);
919       BM_select_history_store(bm, v_active);
920     }
921   }
922 }
923 #endif /* USE_FACE_CREATE_SEL_EXTEND */
924
925 static int edbm_add_edge_face_exec(bContext *C, wmOperator *op)
926 {
927   /* when this is used to dissolve we could avoid this, but checking isnt too slow */
928
929   ViewLayer *view_layer = CTX_data_view_layer(C);
930   uint objects_len = 0;
931   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
932       view_layer, CTX_wm_view3d(C), &objects_len);
933   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
934     Object *obedit = objects[ob_index];
935     BMEditMesh *em = BKE_editmesh_from_object(obedit);
936
937     if ((em->bm->totvertsel == 0) && (em->bm->totedgesel == 0) && (em->bm->totvertsel == 0)) {
938       continue;
939     }
940
941     bool use_smooth = edbm_add_edge_face__smooth_get(em->bm);
942     int totedge_orig = em->bm->totedge;
943     int totface_orig = em->bm->totface;
944
945     BMOperator bmop;
946 #ifdef USE_FACE_CREATE_SEL_EXTEND
947     BMElem *ele_desel;
948     BMFace *ele_desel_face;
949
950     /* be extra clever, figure out if a partial selection should be extended so we can create
951      * geometry with single vert or single edge selection. */
952     ele_desel = edbm_add_edge_face_exec__tricky_extend_sel(em->bm);
953 #endif
954     if (!EDBM_op_init(em,
955                       &bmop,
956                       op,
957                       "contextual_create geom=%hfev mat_nr=%i use_smooth=%b",
958                       BM_ELEM_SELECT,
959                       em->mat_nr,
960                       use_smooth)) {
961       continue;
962     }
963
964     BMO_op_exec(em->bm, &bmop);
965
966     /* cancel if nothing was done */
967     if ((totedge_orig == em->bm->totedge) && (totface_orig == em->bm->totface)) {
968       EDBM_op_finish(em, &bmop, op, true);
969       continue;
970     }
971 #ifdef USE_FACE_CREATE_SEL_EXTEND
972     /* normally we would want to leave the new geometry selected,
973      * but being able to press F many times to add geometry is too useful! */
974     if (ele_desel && (BMO_slot_buffer_count(bmop.slots_out, "faces.out") == 1) &&
975         (ele_desel_face = BMO_slot_buffer_get_first(bmop.slots_out, "faces.out"))) {
976       edbm_add_edge_face_exec__tricky_finalize_sel(em->bm, ele_desel, ele_desel_face);
977     }
978     else
979 #endif
980     {
981       /* Newly created faces may include existing hidden edges,
982        * copying face data from surrounding, may have copied hidden face flag too.
983        *
984        * Important that faces use flushing since 'edges.out'
985        * wont include hidden edges that already existed.
986        */
987       BMO_slot_buffer_hflag_disable(
988           em->bm, bmop.slots_out, "faces.out", BM_FACE, BM_ELEM_HIDDEN, true);
989       BMO_slot_buffer_hflag_disable(
990           em->bm, bmop.slots_out, "edges.out", BM_EDGE, BM_ELEM_HIDDEN, false);
991
992       BMO_slot_buffer_hflag_enable(
993           em->bm, bmop.slots_out, "faces.out", BM_FACE, BM_ELEM_SELECT, true);
994       BMO_slot_buffer_hflag_enable(
995           em->bm, bmop.slots_out, "edges.out", BM_EDGE, BM_ELEM_SELECT, true);
996     }
997
998     if (!EDBM_op_finish(em, &bmop, op, true)) {
999       continue;
1000     }
1001
1002     EDBM_update_generic(em, true, true);
1003   }
1004   MEM_freeN(objects);
1005
1006   return OPERATOR_FINISHED;
1007 }
1008
1009 void MESH_OT_edge_face_add(wmOperatorType *ot)
1010 {
1011   /* identifiers */
1012   ot->name = "Make Edge/Face";
1013   ot->description = "Add an edge or face to selected";
1014   ot->idname = "MESH_OT_edge_face_add";
1015
1016   /* api callbacks */
1017   ot->exec = edbm_add_edge_face_exec;
1018   ot->poll = ED_operator_editmesh;
1019
1020   /* flags */
1021   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1022 }
1023
1024 /** \} */
1025
1026 /* -------------------------------------------------------------------- */
1027 /** \name Mark Edge (Seam) Operator
1028  * \{ */
1029
1030 static int edbm_mark_seam_exec(bContext *C, wmOperator *op)
1031 {
1032   Scene *scene = CTX_data_scene(C);
1033   ViewLayer *view_layer = CTX_data_view_layer(C);
1034   BMEdge *eed;
1035   BMIter iter;
1036   const bool clear = RNA_boolean_get(op->ptr, "clear");
1037
1038   uint objects_len = 0;
1039   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
1040       view_layer, CTX_wm_view3d(C), &objects_len);
1041   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
1042     Object *obedit = objects[ob_index];
1043     BMEditMesh *em = BKE_editmesh_from_object(obedit);
1044     BMesh *bm = em->bm;
1045
1046     if (bm->totedgesel == 0) {
1047       continue;
1048     }
1049
1050     if (clear) {
1051       BM_ITER_MESH (eed, &iter, bm, BM_EDGES_OF_MESH) {
1052         if (!BM_elem_flag_test(eed, BM_ELEM_SELECT) || BM_elem_flag_test(eed, BM_ELEM_HIDDEN)) {
1053           continue;
1054         }
1055
1056         BM_elem_flag_disable(eed, BM_ELEM_SEAM);
1057       }
1058     }
1059     else {
1060       BM_ITER_MESH (eed, &iter, bm, BM_EDGES_OF_MESH) {
1061         if (!BM_elem_flag_test(eed, BM_ELEM_SELECT) || BM_elem_flag_test(eed, BM_ELEM_HIDDEN)) {
1062           continue;
1063         }
1064         BM_elem_flag_enable(eed, BM_ELEM_SEAM);
1065       }
1066     }
1067   }
1068
1069   ED_uvedit_live_unwrap(scene, objects, objects_len);
1070
1071   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
1072     Object *obedit = objects[ob_index];
1073     BMEditMesh *em = BKE_editmesh_from_object(obedit);
1074     EDBM_update_generic(em, true, false);
1075   }
1076
1077   MEM_freeN(objects);
1078
1079   return OPERATOR_FINISHED;
1080 }
1081
1082 void MESH_OT_mark_seam(wmOperatorType *ot)
1083 {
1084   PropertyRNA *prop;
1085
1086   /* identifiers */
1087   ot->name = "Mark Seam";
1088   ot->idname = "MESH_OT_mark_seam";
1089   ot->description = "(Un)mark selected edges as a seam";
1090
1091   /* api callbacks */
1092   ot->exec = edbm_mark_seam_exec;
1093   ot->poll = ED_operator_editmesh;
1094
1095   /* flags */
1096   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1097
1098   prop = RNA_def_boolean(ot->srna, "clear", 0, "Clear", "");
1099   RNA_def_property_flag(prop, PROP_HIDDEN | PROP_SKIP_SAVE);
1100
1101   WM_operatortype_props_advanced_begin(ot);
1102 }
1103
1104 /** \} */
1105
1106 /* -------------------------------------------------------------------- */
1107 /** \name Mark Edge (Sharp) Operator
1108  * \{ */
1109
1110 static int edbm_mark_sharp_exec(bContext *C, wmOperator *op)
1111 {
1112   BMEdge *eed;
1113   BMIter iter;
1114   const bool clear = RNA_boolean_get(op->ptr, "clear");
1115   const bool use_verts = RNA_boolean_get(op->ptr, "use_verts");
1116   ViewLayer *view_layer = CTX_data_view_layer(C);
1117
1118   uint objects_len = 0;
1119   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
1120       view_layer, CTX_wm_view3d(C), &objects_len);
1121   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
1122     Object *obedit = objects[ob_index];
1123     BMEditMesh *em = BKE_editmesh_from_object(obedit);
1124     BMesh *bm = em->bm;
1125
1126     if (bm->totedgesel == 0) {
1127       continue;
1128     }
1129
1130     BM_ITER_MESH (eed, &iter, bm, BM_EDGES_OF_MESH) {
1131       if (use_verts) {
1132         if (!(BM_elem_flag_test(eed->v1, BM_ELEM_SELECT) ||
1133               BM_elem_flag_test(eed->v2, BM_ELEM_SELECT))) {
1134           continue;
1135         }
1136       }
1137       else if (!BM_elem_flag_test(eed, BM_ELEM_SELECT)) {
1138         continue;
1139       }
1140
1141       BM_elem_flag_set(eed, BM_ELEM_SMOOTH, clear);
1142     }
1143
1144     EDBM_update_generic(em, true, false);
1145   }
1146   MEM_freeN(objects);
1147
1148   return OPERATOR_FINISHED;
1149 }
1150
1151 void MESH_OT_mark_sharp(wmOperatorType *ot)
1152 {
1153   PropertyRNA *prop;
1154
1155   /* identifiers */
1156   ot->name = "Mark Sharp";
1157   ot->idname = "MESH_OT_mark_sharp";
1158   ot->description = "(Un)mark selected edges as sharp";
1159
1160   /* api callbacks */
1161   ot->exec = edbm_mark_sharp_exec;
1162   ot->poll = ED_operator_editmesh;
1163
1164   /* flags */
1165   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1166
1167   prop = RNA_def_boolean(ot->srna, "clear", false, "Clear", "");
1168   RNA_def_property_flag(prop, PROP_HIDDEN | PROP_SKIP_SAVE);
1169   prop = RNA_def_boolean(
1170       ot->srna,
1171       "use_verts",
1172       false,
1173       "Vertices",
1174       "Consider vertices instead of edges to select which edges to (un)tag as sharp");
1175   RNA_def_property_flag(prop, PROP_SKIP_SAVE);
1176 }
1177
1178 static bool edbm_connect_vert_pair(BMEditMesh *em, wmOperator *op)
1179 {
1180   BMesh *bm = em->bm;
1181   BMOperator bmop;
1182   const int verts_len = bm->totvertsel;
1183   bool is_pair = (verts_len == 2);
1184   int len = 0;
1185   bool check_degenerate = true;
1186
1187   BMVert **verts;
1188   bool checks_succeded = true;
1189
1190   /* sanity check */
1191   if (verts_len < 2) {
1192     return false;
1193   }
1194
1195   verts = MEM_mallocN(sizeof(*verts) * verts_len, __func__);
1196   {
1197     BMIter iter;
1198     BMVert *v;
1199     int i = 0;
1200
1201     BM_ITER_MESH (v, &iter, bm, BM_VERTS_OF_MESH) {
1202       if (BM_elem_flag_test(v, BM_ELEM_SELECT)) {
1203         verts[i++] = v;
1204       }
1205     }
1206
1207     if (BM_vert_pair_share_face_check_cb(
1208             verts[0],
1209             verts[1],
1210             BM_elem_cb_check_hflag_disabled_simple(BMFace *, BM_ELEM_HIDDEN))) {
1211       check_degenerate = false;
1212       is_pair = false;
1213     }
1214   }
1215
1216   if (is_pair) {
1217     if (!EDBM_op_init(em,
1218                       &bmop,
1219                       op,
1220                       "connect_vert_pair verts=%eb verts_exclude=%hv faces_exclude=%hf",
1221                       verts,
1222                       verts_len,
1223                       BM_ELEM_HIDDEN,
1224                       BM_ELEM_HIDDEN)) {
1225       checks_succeded = false;
1226     }
1227   }
1228   else {
1229     if (!EDBM_op_init(em,
1230                       &bmop,
1231                       op,
1232                       "connect_verts verts=%eb faces_exclude=%hf check_degenerate=%b",
1233                       verts,
1234                       verts_len,
1235                       BM_ELEM_HIDDEN,
1236                       check_degenerate)) {
1237       checks_succeded = false;
1238     }
1239   }
1240   if (checks_succeded) {
1241     BMO_op_exec(bm, &bmop);
1242     len = BMO_slot_get(bmop.slots_out, "edges.out")->len;
1243
1244     if (len && is_pair) {
1245       /* new verts have been added, we have to select the edges, not just flush */
1246       BMO_slot_buffer_hflag_enable(
1247           em->bm, bmop.slots_out, "edges.out", BM_EDGE, BM_ELEM_SELECT, true);
1248     }
1249
1250     if (!EDBM_op_finish(em, &bmop, op, true)) {
1251       len = 0;
1252     }
1253     else {
1254       /* so newly created edges get the selection state from the vertex */
1255       EDBM_selectmode_flush(em);
1256
1257       EDBM_update_generic(em, true, true);
1258     }
1259   }
1260   MEM_freeN(verts);
1261
1262   return len;
1263 }
1264
1265 static int edbm_vert_connect_exec(bContext *C, wmOperator *op)
1266 {
1267   ViewLayer *view_layer = CTX_data_view_layer(C);
1268   uint objects_len = 0;
1269   uint failed_objects_len = 0;
1270   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
1271       view_layer, CTX_wm_view3d(C), &objects_len);
1272
1273   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
1274     Object *obedit = objects[ob_index];
1275     BMEditMesh *em = BKE_editmesh_from_object(obedit);
1276
1277     if (!edbm_connect_vert_pair(em, op)) {
1278       failed_objects_len++;
1279     }
1280   }
1281   MEM_freeN(objects);
1282   return failed_objects_len == objects_len ? OPERATOR_FINISHED : OPERATOR_CANCELLED;
1283 }
1284
1285 void MESH_OT_vert_connect(wmOperatorType *ot)
1286 {
1287   /* identifiers */
1288   ot->name = "Vertex Connect";
1289   ot->idname = "MESH_OT_vert_connect";
1290   ot->description = "Connect selected vertices of faces, splitting the face";
1291
1292   /* api callbacks */
1293   ot->exec = edbm_vert_connect_exec;
1294   ot->poll = ED_operator_editmesh;
1295
1296   /* flags */
1297   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1298 }
1299
1300 /** \} */
1301
1302 /* -------------------------------------------------------------------- */
1303 /** \name Split Concave Faces Operator
1304  * \{ */
1305
1306 /**
1307  * check that endpoints are verts and only have a single selected edge connected.
1308  */
1309 static bool bm_vert_is_select_history_open(BMesh *bm)
1310 {
1311   BMEditSelection *ele_a = bm->selected.first;
1312   BMEditSelection *ele_b = bm->selected.last;
1313   if ((ele_a->htype == BM_VERT) && (ele_b->htype == BM_VERT)) {
1314     if ((BM_iter_elem_count_flag(BM_EDGES_OF_VERT, (BMVert *)ele_a->ele, BM_ELEM_SELECT, true) ==
1315          1) &&
1316         (BM_iter_elem_count_flag(BM_EDGES_OF_VERT, (BMVert *)ele_b->ele, BM_ELEM_SELECT, true) ==
1317          1)) {
1318       return true;
1319     }
1320   }
1321
1322   return false;
1323 }
1324
1325 static bool bm_vert_connect_pair(BMesh *bm, BMVert *v_a, BMVert *v_b)
1326 {
1327   BMOperator bmop;
1328   BMVert **verts;
1329   const int totedge_orig = bm->totedge;
1330
1331   BMO_op_init(bm, &bmop, BMO_FLAG_DEFAULTS, "connect_vert_pair");
1332
1333   verts = BMO_slot_buffer_alloc(&bmop, bmop.slots_in, "verts", 2);
1334   verts[0] = v_a;
1335   verts[1] = v_b;
1336
1337   BM_vert_normal_update(verts[0]);
1338   BM_vert_normal_update(verts[1]);
1339
1340   BMO_op_exec(bm, &bmop);
1341   BMO_slot_buffer_hflag_enable(bm, bmop.slots_out, "edges.out", BM_EDGE, BM_ELEM_SELECT, true);
1342   BMO_op_finish(bm, &bmop);
1343   return (bm->totedge != totedge_orig);
1344 }
1345
1346 static bool bm_vert_connect_select_history(BMesh *bm)
1347 {
1348   /* Logic is as follows:
1349    *
1350    * - If there are any isolated/wire verts - connect as edges.
1351    * - Otherwise connect faces.
1352    * - If all edges have been created already, closed the loop.
1353    */
1354   if (BLI_listbase_count_at_most(&bm->selected, 2) == 2 && (bm->totvertsel > 2)) {
1355     BMEditSelection *ese;
1356     int tot = 0;
1357     bool changed = false;
1358     bool has_wire = false;
1359     // bool all_verts;
1360
1361     /* ensure all verts have history */
1362     for (ese = bm->selected.first; ese; ese = ese->next, tot++) {
1363       BMVert *v;
1364       if (ese->htype != BM_VERT) {
1365         break;
1366       }
1367       v = (BMVert *)ese->ele;
1368       if ((has_wire == false) && ((v->e == NULL) || BM_vert_is_wire(v))) {
1369         has_wire = true;
1370       }
1371     }
1372     // all_verts = (ese == NULL);
1373
1374     if (has_wire == false) {
1375       /* all verts have faces , connect verts via faces! */
1376       if (tot == bm->totvertsel) {
1377         BMEditSelection *ese_last;
1378         ese_last = bm->selected.first;
1379         ese = ese_last->next;
1380
1381         do {
1382
1383           if (BM_edge_exists((BMVert *)ese_last->ele, (BMVert *)ese->ele)) {
1384             /* pass, edge exists (and will be selected) */
1385           }
1386           else {
1387             changed |= bm_vert_connect_pair(bm, (BMVert *)ese_last->ele, (BMVert *)ese->ele);
1388           }
1389         } while ((void)(ese_last = ese), (ese = ese->next));
1390
1391         if (changed) {
1392           return true;
1393         }
1394       }
1395
1396       if (changed == false) {
1397         /* existing loops: close the selection */
1398         if (bm_vert_is_select_history_open(bm)) {
1399           changed |= bm_vert_connect_pair(bm,
1400                                           (BMVert *)((BMEditSelection *)bm->selected.first)->ele,
1401                                           (BMVert *)((BMEditSelection *)bm->selected.last)->ele);
1402
1403           if (changed) {
1404             return true;
1405           }
1406         }
1407       }
1408     }
1409
1410     else {
1411       /* no faces, simply connect the verts by edges */
1412       BMEditSelection *ese_prev;
1413       ese_prev = bm->selected.first;
1414       ese = ese_prev->next;
1415
1416       do {
1417         if (BM_edge_exists((BMVert *)ese_prev->ele, (BMVert *)ese->ele)) {
1418           /* pass, edge exists (and will be selected) */
1419         }
1420         else {
1421           BMEdge *e;
1422           e = BM_edge_create(bm, (BMVert *)ese_prev->ele, (BMVert *)ese->ele, NULL, 0);
1423           BM_edge_select_set(bm, e, true);
1424           changed = true;
1425         }
1426       } while ((void)(ese_prev = ese), (ese = ese->next));
1427
1428       if (changed == false) {
1429         /* existing loops: close the selection */
1430         if (bm_vert_is_select_history_open(bm)) {
1431           BMEdge *e;
1432           ese_prev = bm->selected.first;
1433           ese = bm->selected.last;
1434           e = BM_edge_create(bm, (BMVert *)ese_prev->ele, (BMVert *)ese->ele, NULL, 0);
1435           BM_edge_select_set(bm, e, true);
1436         }
1437       }
1438
1439       return true;
1440     }
1441   }
1442
1443   return false;
1444 }
1445
1446 /**
1447  * Convert an edge selection to a temp vertex selection
1448  * (which must be cleared after use as a path to connect).
1449  */
1450 static bool bm_vert_connect_select_history_edge_to_vert_path(BMesh *bm, ListBase *r_selected)
1451 {
1452   ListBase selected_orig = {NULL, NULL};
1453   BMEditSelection *ese;
1454   int edges_len = 0;
1455   bool side = false;
1456
1457   /* first check all edges are OK */
1458   for (ese = bm->selected.first; ese; ese = ese->next) {
1459     if (ese->htype == BM_EDGE) {
1460       edges_len += 1;
1461     }
1462     else {
1463       return false;
1464     }
1465   }
1466   /* if this is a mixed selection, bail out! */
1467   if (bm->totedgesel != edges_len) {
1468     return false;
1469   }
1470
1471   SWAP(ListBase, bm->selected, selected_orig);
1472
1473   /* convert edge selection into 2 ordered loops (where the first edge ends up in the middle) */
1474   for (ese = selected_orig.first; ese; ese = ese->next) {
1475     BMEdge *e_curr = (BMEdge *)ese->ele;
1476     BMEdge *e_prev = ese->prev ? (BMEdge *)ese->prev->ele : NULL;
1477     BMLoop *l_curr;
1478     BMLoop *l_prev;
1479     BMVert *v;
1480
1481     if (e_prev) {
1482       BMFace *f = BM_edge_pair_share_face_by_len(e_curr, e_prev, &l_curr, &l_prev, true);
1483       if (f) {
1484         if ((e_curr->v1 != l_curr->v) == (e_prev->v1 != l_prev->v)) {
1485           side = !side;
1486         }
1487       }
1488       else if (is_quad_flip_v3(e_curr->v1->co, e_curr->v2->co, e_prev->v2->co, e_prev->v1->co)) {
1489         side = !side;
1490       }
1491     }
1492
1493     v = (&e_curr->v1)[side];
1494     if (!bm->selected.last || (BMVert *)((BMEditSelection *)bm->selected.last)->ele != v) {
1495       BM_select_history_store_notest(bm, v);
1496     }
1497
1498     v = (&e_curr->v1)[!side];
1499     if (!bm->selected.first || (BMVert *)((BMEditSelection *)bm->selected.first)->ele != v) {
1500       BM_select_history_store_head_notest(bm, v);
1501     }
1502
1503     e_prev = e_curr;
1504   }
1505
1506   *r_selected = bm->selected;
1507   bm->selected = selected_orig;
1508
1509   return true;
1510 }
1511
1512 static int edbm_vert_connect_path_exec(bContext *C, wmOperator *op)
1513 {
1514   ViewLayer *view_layer = CTX_data_view_layer(C);
1515   uint objects_len = 0;
1516   uint failed_selection_order_len = 0;
1517   uint failed_connect_len = 0;
1518   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
1519       view_layer, CTX_wm_view3d(C), &objects_len);
1520
1521   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
1522     Object *obedit = objects[ob_index];
1523     BMEditMesh *em = BKE_editmesh_from_object(obedit);
1524     BMesh *bm = em->bm;
1525     const bool is_pair = (em->bm->totvertsel == 2);
1526     ListBase selected_orig = {NULL, NULL};
1527
1528     if (bm->totvertsel == 0) {
1529       continue;
1530     }
1531
1532     /* when there is only 2 vertices, we can ignore selection order */
1533     if (is_pair) {
1534       if (!edbm_connect_vert_pair(em, op)) {
1535         failed_connect_len++;
1536       }
1537       continue;
1538     }
1539
1540     if (bm->selected.first) {
1541       BMEditSelection *ese = bm->selected.first;
1542       if (ese->htype == BM_EDGE) {
1543         if (bm_vert_connect_select_history_edge_to_vert_path(bm, &selected_orig)) {
1544           SWAP(ListBase, bm->selected, selected_orig);
1545         }
1546       }
1547     }
1548
1549     if (bm_vert_connect_select_history(bm)) {
1550       EDBM_selectmode_flush(em);
1551       EDBM_update_generic(em, true, true);
1552     }
1553     else {
1554       failed_selection_order_len++;
1555     }
1556
1557     if (!BLI_listbase_is_empty(&selected_orig)) {
1558       BM_select_history_clear(bm);
1559       bm->selected = selected_orig;
1560     }
1561   }
1562
1563   MEM_freeN(objects);
1564
1565   if (failed_selection_order_len == objects_len) {
1566     BKE_report(op->reports, RPT_ERROR, "Invalid selection order");
1567     return OPERATOR_CANCELLED;
1568   }
1569   else if (failed_connect_len == objects_len) {
1570     BKE_report(op->reports, RPT_ERROR, "Could not connect vertices");
1571     return OPERATOR_CANCELLED;
1572   }
1573
1574   return OPERATOR_FINISHED;
1575 }
1576
1577 void MESH_OT_vert_connect_path(wmOperatorType *ot)
1578 {
1579   /* identifiers */
1580   ot->name = "Vertex Connect Path";
1581   ot->idname = "MESH_OT_vert_connect_path";
1582   ot->description = "Connect vertices by their selection order, creating edges, splitting faces";
1583
1584   /* api callbacks */
1585   ot->exec = edbm_vert_connect_path_exec;
1586   ot->poll = ED_operator_editmesh;
1587
1588   /* flags */
1589   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1590 }
1591
1592 static int edbm_vert_connect_concave_exec(bContext *C, wmOperator *op)
1593 {
1594   ViewLayer *view_layer = CTX_data_view_layer(C);
1595   uint objects_len = 0;
1596   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
1597       view_layer, CTX_wm_view3d(C), &objects_len);
1598   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
1599     Object *obedit = objects[ob_index];
1600     BMEditMesh *em = BKE_editmesh_from_object(obedit);
1601
1602     if (em->bm->totfacesel == 0) {
1603       continue;
1604     }
1605
1606     if (!EDBM_op_call_and_selectf(
1607             em, op, "faces.out", true, "connect_verts_concave faces=%hf", BM_ELEM_SELECT)) {
1608       continue;
1609     }
1610     EDBM_update_generic(em, true, true);
1611   }
1612
1613   MEM_freeN(objects);
1614   return OPERATOR_FINISHED;
1615 }
1616
1617 void MESH_OT_vert_connect_concave(wmOperatorType *ot)
1618 {
1619   /* identifiers */
1620   ot->name = "Split Concave Faces";
1621   ot->idname = "MESH_OT_vert_connect_concave";
1622   ot->description = "Make all faces convex";
1623
1624   /* api callbacks */
1625   ot->exec = edbm_vert_connect_concave_exec;
1626   ot->poll = ED_operator_editmesh;
1627
1628   /* flags */
1629   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1630 }
1631
1632 /** \} */
1633
1634 /* -------------------------------------------------------------------- */
1635 /** \name Split Non-Planar Faces Operator
1636  * \{ */
1637
1638 static int edbm_vert_connect_nonplaner_exec(bContext *C, wmOperator *op)
1639 {
1640   ViewLayer *view_layer = CTX_data_view_layer(C);
1641   const float angle_limit = RNA_float_get(op->ptr, "angle_limit");
1642   uint objects_len = 0;
1643   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
1644       view_layer, CTX_wm_view3d(C), &objects_len);
1645
1646   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
1647     Object *obedit = objects[ob_index];
1648     BMEditMesh *em = BKE_editmesh_from_object(obedit);
1649
1650     if (em->bm->totfacesel == 0) {
1651       continue;
1652     }
1653
1654     if (!EDBM_op_call_and_selectf(em,
1655                                   op,
1656                                   "faces.out",
1657                                   true,
1658                                   "connect_verts_nonplanar faces=%hf angle_limit=%f",
1659                                   BM_ELEM_SELECT,
1660                                   angle_limit)) {
1661       continue;
1662     }
1663
1664     EDBM_update_generic(em, true, true);
1665   }
1666   MEM_freeN(objects);
1667
1668   return OPERATOR_FINISHED;
1669 }
1670
1671 void MESH_OT_vert_connect_nonplanar(wmOperatorType *ot)
1672 {
1673   PropertyRNA *prop;
1674
1675   /* identifiers */
1676   ot->name = "Split Non-Planar Faces";
1677   ot->idname = "MESH_OT_vert_connect_nonplanar";
1678   ot->description = "Split non-planar faces that exceed the angle threshold";
1679
1680   /* api callbacks */
1681   ot->exec = edbm_vert_connect_nonplaner_exec;
1682   ot->poll = ED_operator_editmesh;
1683
1684   /* flags */
1685   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1686
1687   /* props */
1688   prop = RNA_def_float_rotation(ot->srna,
1689                                 "angle_limit",
1690                                 0,
1691                                 NULL,
1692                                 0.0f,
1693                                 DEG2RADF(180.0f),
1694                                 "Max Angle",
1695                                 "Angle limit",
1696                                 0.0f,
1697                                 DEG2RADF(180.0f));
1698   RNA_def_property_float_default(prop, DEG2RADF(5.0f));
1699 }
1700
1701 /** \} */
1702
1703 /* -------------------------------------------------------------------- */
1704 /** \name Make Planar Faces Operator
1705  * \{ */
1706
1707 static int edbm_face_make_planar_exec(bContext *C, wmOperator *op)
1708 {
1709   ViewLayer *view_layer = CTX_data_view_layer(C);
1710   uint objects_len = 0;
1711   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
1712       view_layer, CTX_wm_view3d(C), &objects_len);
1713
1714   const int repeat = RNA_int_get(op->ptr, "repeat");
1715   const float fac = RNA_float_get(op->ptr, "factor");
1716
1717   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
1718     Object *obedit = objects[ob_index];
1719     BMEditMesh *em = BKE_editmesh_from_object(obedit);
1720     if (em->bm->totfacesel == 0) {
1721       continue;
1722     }
1723
1724     if (!EDBM_op_callf(em,
1725                        op,
1726                        "planar_faces faces=%hf iterations=%i factor=%f",
1727                        BM_ELEM_SELECT,
1728                        repeat,
1729                        fac)) {
1730       continue;
1731     }
1732
1733     EDBM_update_generic(em, true, true);
1734   }
1735   MEM_freeN(objects);
1736
1737   return OPERATOR_FINISHED;
1738 }
1739
1740 void MESH_OT_face_make_planar(wmOperatorType *ot)
1741 {
1742   /* identifiers */
1743   ot->name = "Make Planar Faces";
1744   ot->idname = "MESH_OT_face_make_planar";
1745   ot->description = "Flatten selected faces";
1746
1747   /* api callbacks */
1748   ot->exec = edbm_face_make_planar_exec;
1749   ot->poll = ED_operator_editmesh;
1750
1751   /* flags */
1752   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1753
1754   /* props */
1755   RNA_def_float(ot->srna, "factor", 1.0f, -10.0f, 10.0f, "Factor", "", 0.0f, 1.0f);
1756   RNA_def_int(ot->srna, "repeat", 1, 1, 10000, "Iterations", "", 1, 200);
1757 }
1758
1759 /** \} */
1760
1761 /* -------------------------------------------------------------------- */
1762 /** \name Split Edge Operator
1763  * \{ */
1764
1765 static int edbm_edge_split_exec(bContext *C, wmOperator *op)
1766 {
1767   ViewLayer *view_layer = CTX_data_view_layer(C);
1768   uint objects_len = 0;
1769   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
1770       view_layer, CTX_wm_view3d(C), &objects_len);
1771   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
1772     Object *obedit = objects[ob_index];
1773     BMEditMesh *em = BKE_editmesh_from_object(obedit);
1774     if (em->bm->totedgesel == 0) {
1775       continue;
1776     }
1777
1778     if (!EDBM_op_call_and_selectf(
1779             em, op, "edges.out", false, "split_edges edges=%he", BM_ELEM_SELECT)) {
1780       continue;
1781     }
1782
1783     if (em->selectmode == SCE_SELECT_FACE) {
1784       EDBM_select_flush(em);
1785     }
1786
1787     EDBM_update_generic(em, true, true);
1788   }
1789   MEM_freeN(objects);
1790
1791   return OPERATOR_FINISHED;
1792 }
1793
1794 void MESH_OT_edge_split(wmOperatorType *ot)
1795 {
1796   /* identifiers */
1797   ot->name = "Edge Split";
1798   ot->idname = "MESH_OT_edge_split";
1799   ot->description = "Split selected edges so that each neighbor face gets its own copy";
1800
1801   /* api callbacks */
1802   ot->exec = edbm_edge_split_exec;
1803   ot->poll = ED_operator_editmesh;
1804
1805   /* flags */
1806   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1807 }
1808
1809 /** \} */
1810
1811 /* -------------------------------------------------------------------- */
1812 /** \name Duplicate Operator
1813  * \{ */
1814
1815 static int edbm_duplicate_exec(bContext *C, wmOperator *op)
1816 {
1817   ViewLayer *view_layer = CTX_data_view_layer(C);
1818   uint objects_len = 0;
1819   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
1820       view_layer, CTX_wm_view3d(C), &objects_len);
1821
1822   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
1823     Object *obedit = objects[ob_index];
1824     BMEditMesh *em = BKE_editmesh_from_object(obedit);
1825     if (em->bm->totvertsel == 0) {
1826       continue;
1827     }
1828
1829     BMOperator bmop;
1830     BMesh *bm = em->bm;
1831
1832     EDBM_op_init(em,
1833                  &bmop,
1834                  op,
1835                  "duplicate geom=%hvef use_select_history=%b use_edge_flip_from_face=%b",
1836                  BM_ELEM_SELECT,
1837                  true,
1838                  true);
1839
1840     BMO_op_exec(bm, &bmop);
1841
1842     /* de-select all would clear otherwise */
1843     BM_SELECT_HISTORY_BACKUP(bm);
1844
1845     EDBM_flag_disable_all(em, BM_ELEM_SELECT);
1846
1847     BMO_slot_buffer_hflag_enable(
1848         bm, bmop.slots_out, "geom.out", BM_ALL_NOLOOP, BM_ELEM_SELECT, true);
1849
1850     /* rebuild editselection */
1851     BM_SELECT_HISTORY_RESTORE(bm);
1852
1853     if (!EDBM_op_finish(em, &bmop, op, true)) {
1854       continue;
1855     }
1856     EDBM_update_generic(em, true, true);
1857   }
1858   MEM_freeN(objects);
1859
1860   return OPERATOR_FINISHED;
1861 }
1862
1863 static int edbm_duplicate_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1864 {
1865   WM_cursor_wait(1);
1866   edbm_duplicate_exec(C, op);
1867   WM_cursor_wait(0);
1868
1869   return OPERATOR_FINISHED;
1870 }
1871
1872 void MESH_OT_duplicate(wmOperatorType *ot)
1873 {
1874   /* identifiers */
1875   ot->name = "Duplicate";
1876   ot->description = "Duplicate selected vertices, edges or faces";
1877   ot->idname = "MESH_OT_duplicate";
1878
1879   /* api callbacks */
1880   ot->invoke = edbm_duplicate_invoke;
1881   ot->exec = edbm_duplicate_exec;
1882
1883   ot->poll = ED_operator_editmesh;
1884
1885   /* to give to transform */
1886   RNA_def_int(ot->srna, "mode", TFM_TRANSLATION, 0, INT_MAX, "Mode", "", 0, INT_MAX);
1887 }
1888
1889 /** \} */
1890
1891 /* -------------------------------------------------------------------- */
1892 /** \name Flip Normals Operator
1893  * \{ */
1894 static int edbm_flip_normals_exec(bContext *C, wmOperator *op)
1895 {
1896   ViewLayer *view_layer = CTX_data_view_layer(C);
1897   uint objects_len = 0;
1898   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
1899       view_layer, CTX_wm_view3d(C), &objects_len);
1900
1901   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
1902     Object *obedit = objects[ob_index];
1903     BMEditMesh *em = BKE_editmesh_from_object(obedit);
1904
1905     if (em->bm->totfacesel == 0) {
1906       continue;
1907     }
1908
1909     if (!EDBM_op_callf(em, op, "reverse_faces faces=%hf flip_multires=%b", BM_ELEM_SELECT, true)) {
1910       continue;
1911     }
1912
1913     EDBM_update_generic(em, true, false);
1914   }
1915
1916   MEM_freeN(objects);
1917   return OPERATOR_FINISHED;
1918 }
1919
1920 void MESH_OT_flip_normals(wmOperatorType *ot)
1921 {
1922   /* identifiers */
1923   ot->name = "Flip Normals";
1924   ot->description = "Flip the direction of selected faces' normals (and of their vertices)";
1925   ot->idname = "MESH_OT_flip_normals";
1926
1927   /* api callbacks */
1928   ot->exec = edbm_flip_normals_exec;
1929   ot->poll = ED_operator_editmesh;
1930
1931   /* flags */
1932   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1933 }
1934
1935 /** \} */
1936
1937 /* -------------------------------------------------------------------- */
1938 /** \name Rotate Edge Operator
1939  * \{ */
1940
1941 /**
1942  * Rotate the edges between selected faces, otherwise rotate the selected edges.
1943  */
1944 static int edbm_edge_rotate_selected_exec(bContext *C, wmOperator *op)
1945 {
1946   BMEdge *eed;
1947   BMIter iter;
1948   const bool use_ccw = RNA_boolean_get(op->ptr, "use_ccw");
1949
1950   int tot_rotate_all = 0, tot_failed_all = 0;
1951   bool no_selected_edges = true, invalid_selected_edges = true;
1952
1953   ViewLayer *view_layer = CTX_data_view_layer(C);
1954   uint objects_len = 0;
1955   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
1956       view_layer, CTX_wm_view3d(C), &objects_len);
1957   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
1958     Object *obedit = objects[ob_index];
1959     BMEditMesh *em = BKE_editmesh_from_object(obedit);
1960     int tot = 0;
1961
1962     if (em->bm->totedgesel == 0) {
1963       continue;
1964     }
1965     no_selected_edges = false;
1966
1967     /* first see if we have two adjacent faces */
1968     BM_ITER_MESH (eed, &iter, em->bm, BM_EDGES_OF_MESH) {
1969       BM_elem_flag_disable(eed, BM_ELEM_TAG);
1970       if (BM_elem_flag_test(eed, BM_ELEM_SELECT)) {
1971         BMFace *fa, *fb;
1972         if (BM_edge_face_pair(eed, &fa, &fb)) {
1973           /* if both faces are selected we rotate between them,
1974            * otherwise - rotate between 2 unselected - but not mixed */
1975           if (BM_elem_flag_test(fa, BM_ELEM_SELECT) == BM_elem_flag_test(fb, BM_ELEM_SELECT)) {
1976             BM_elem_flag_enable(eed, BM_ELEM_TAG);
1977             tot++;
1978           }
1979         }
1980       }
1981     }
1982
1983     /* ok, we don't have two adjacent faces, but we do have two selected ones.
1984      * that's an error condition.*/
1985     if (tot == 0) {
1986       continue;
1987     }
1988     invalid_selected_edges = false;
1989
1990     BMOperator bmop;
1991     EDBM_op_init(em, &bmop, op, "rotate_edges edges=%he use_ccw=%b", BM_ELEM_TAG, use_ccw);
1992
1993     /* avoids leaving old verts selected which can be a problem running multiple times,
1994      * since this means the edges become selected around the face
1995      * which then attempt to rotate */
1996     BMO_slot_buffer_hflag_disable(em->bm, bmop.slots_in, "edges", BM_EDGE, BM_ELEM_SELECT, true);
1997
1998     BMO_op_exec(em->bm, &bmop);
1999     /* edges may rotate into hidden vertices, if this does _not_ run we get an illogical state */
2000     BMO_slot_buffer_hflag_disable(
2001         em->bm, bmop.slots_out, "edges.out", BM_EDGE, BM_ELEM_HIDDEN, true);
2002     BMO_slot_buffer_hflag_enable(
2003         em->bm, bmop.slots_out, "edges.out", BM_EDGE, BM_ELEM_SELECT, true);
2004
2005     const int tot_rotate = BMO_slot_buffer_count(bmop.slots_out, "edges.out");
2006     const int tot_failed = tot - tot_rotate;
2007
2008     tot_rotate_all += tot_rotate;
2009     tot_failed_all += tot_failed;
2010
2011     if (tot_failed != 0) {
2012       /* If some edges fail to rotate, we need to re-select them,
2013        * otherwise we can end up with invalid selection
2014        * (unselected edge between 2 selected faces). */
2015       BM_mesh_elem_hflag_enable_test(em->bm, BM_EDGE, BM_ELEM_SELECT, true, false, BM_ELEM_TAG);
2016     }
2017
2018     EDBM_selectmode_flush(em);
2019
2020     if (!EDBM_op_finish(em, &bmop, op, true)) {
2021       continue;
2022     }
2023
2024     EDBM_update_generic(em, true, true);
2025   }
2026   MEM_freeN(objects);
2027
2028   if (no_selected_edges) {
2029     BKE_report(
2030         op->reports, RPT_ERROR, "Select edges or face pairs for edge loops to rotate about");
2031     return OPERATOR_CANCELLED;
2032   }
2033
2034   /* Ok, we don't have two adjacent faces, but we do have two selected ones.
2035    * that's an error condition. */
2036   if (invalid_selected_edges) {
2037     BKE_report(op->reports, RPT_ERROR, "Could not find any selected edges that can be rotated");
2038     return OPERATOR_CANCELLED;
2039   }
2040
2041   if (tot_failed_all != 0) {
2042     BKE_reportf(op->reports, RPT_WARNING, "Unable to rotate %d edge(s)", tot_failed_all);
2043   }
2044
2045   return OPERATOR_FINISHED;
2046 }
2047
2048 void MESH_OT_edge_rotate(wmOperatorType *ot)
2049 {
2050   /* identifiers */
2051   ot->name = "Rotate Selected Edge";
2052   ot->description = "Rotate selected edge or adjoining faces";
2053   ot->idname = "MESH_OT_edge_rotate";
2054
2055   /* api callbacks */
2056   ot->exec = edbm_edge_rotate_selected_exec;
2057   ot->poll = ED_operator_editmesh;
2058
2059   /* flags */
2060   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
2061
2062   /* props */
2063   RNA_def_boolean(ot->srna, "use_ccw", false, "Counter Clockwise", "");
2064 }
2065
2066 /** \} */
2067
2068 /* -------------------------------------------------------------------- */
2069 /** \name Hide Operator
2070  * \{ */
2071
2072 static int edbm_hide_exec(bContext *C, wmOperator *op)
2073 {
2074   const bool unselected = RNA_boolean_get(op->ptr, "unselected");
2075   ViewLayer *view_layer = CTX_data_view_layer(C);
2076   bool changed = false;
2077
2078   uint objects_len = 0;
2079   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
2080       view_layer, CTX_wm_view3d(C), &objects_len);
2081   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
2082     Object *obedit = objects[ob_index];
2083     BMEditMesh *em = BKE_editmesh_from_object(obedit);
2084     BMesh *bm = em->bm;
2085
2086     if (unselected) {
2087       if (bm->totvertsel == bm->totvert) {
2088         continue;
2089       }
2090     }
2091     else {
2092       if (bm->totvertsel == 0) {
2093         continue;
2094       }
2095     }
2096
2097     if (EDBM_mesh_hide(em, unselected)) {
2098       EDBM_update_generic(em, true, false);
2099       changed = true;
2100     }
2101   }
2102   MEM_freeN(objects);
2103
2104   if (!changed) {
2105     return OPERATOR_CANCELLED;
2106   }
2107
2108   return OPERATOR_FINISHED;
2109 }
2110
2111 void MESH_OT_hide(wmOperatorType *ot)
2112 {
2113   /* identifiers */
2114   ot->name = "Hide Selected";
2115   ot->idname = "MESH_OT_hide";
2116   ot->description = "Hide (un)selected vertices, edges or faces";
2117
2118   /* api callbacks */
2119   ot->exec = edbm_hide_exec;
2120   ot->poll = ED_operator_editmesh;
2121
2122   /* flags */
2123   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
2124
2125   /* props */
2126   RNA_def_boolean(
2127       ot->srna, "unselected", false, "Unselected", "Hide unselected rather than selected");
2128 }
2129
2130 /** \} */
2131
2132 /* -------------------------------------------------------------------- */
2133 /** \name Reveal Operator
2134  * \{ */
2135
2136 static int edbm_reveal_exec(bContext *C, wmOperator *op)
2137 {
2138   const bool select = RNA_boolean_get(op->ptr, "select");
2139   ViewLayer *view_layer = CTX_data_view_layer(C);
2140
2141   uint objects_len = 0;
2142   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
2143       view_layer, CTX_wm_view3d(C), &objects_len);
2144   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
2145     Object *obedit = objects[ob_index];
2146     BMEditMesh *em = BKE_editmesh_from_object(obedit);
2147
2148     if (EDBM_mesh_reveal(em, select)) {
2149       EDBM_update_generic(em, true, false);
2150     }
2151   }
2152   MEM_freeN(objects);
2153
2154   return OPERATOR_FINISHED;
2155 }
2156
2157 void MESH_OT_reveal(wmOperatorType *ot)
2158 {
2159   /* identifiers */
2160   ot->name = "Reveal Hidden";
2161   ot->idname = "MESH_OT_reveal";
2162   ot->description = "Reveal all hidden vertices, edges and faces";
2163
2164   /* api callbacks */
2165   ot->exec = edbm_reveal_exec;
2166   ot->poll = ED_operator_editmesh;
2167
2168   /* flags */
2169   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
2170
2171   RNA_def_boolean(ot->srna, "select", true, "Select", "");
2172 }
2173
2174 /** \} */
2175
2176 /* -------------------------------------------------------------------- */
2177 /** \name Recalculate Normals Operator
2178  * \{ */
2179
2180 static int edbm_normals_make_consistent_exec(bContext *C, wmOperator *op)
2181 {
2182   ViewLayer *view_layer = CTX_data_view_layer(C);
2183
2184   uint objects_len = 0;
2185   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
2186       view_layer, CTX_wm_view3d(C), &objects_len);
2187   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
2188     Object *obedit = objects[ob_index];
2189     BMEditMesh *em = BKE_editmesh_from_object(obedit);
2190
2191     if (em->bm->totfacesel == 0) {
2192       continue;
2193     }
2194
2195     if (!EDBM_op_callf(em, op, "recalc_face_normals faces=%hf", BM_ELEM_SELECT)) {
2196       continue;
2197     }
2198     if (RNA_boolean_get(op->ptr, "inside")) {
2199       EDBM_op_callf(em, op, "reverse_faces faces=%hf flip_multires=%b", BM_ELEM_SELECT, true);
2200     }
2201
2202     EDBM_update_generic(em, true, false);
2203   }
2204   MEM_freeN(objects);
2205
2206   return OPERATOR_FINISHED;
2207 }
2208
2209 void MESH_OT_normals_make_consistent(wmOperatorType *ot)
2210 {
2211   /* identifiers */
2212   ot->name = "Make Normals Consistent";
2213   ot->description = "Make face and vertex normals point either outside or inside the mesh";
2214   ot->idname = "MESH_OT_normals_make_consistent";
2215
2216   /* api callbacks */
2217   ot->exec = edbm_normals_make_consistent_exec;
2218   ot->poll = ED_operator_editmesh;
2219
2220   /* flags */
2221   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
2222
2223   RNA_def_boolean(ot->srna, "inside", false, "Inside", "");
2224 }
2225
2226 /** \} */
2227
2228 /* -------------------------------------------------------------------- */
2229 /** \name Smooth Vertices Operator
2230  * \{ */
2231
2232 static int edbm_do_smooth_vertex_exec(bContext *C, wmOperator *op)
2233 {
2234   const float fac = RNA_float_get(op->ptr, "factor");
2235
2236   const bool xaxis = RNA_boolean_get(op->ptr, "xaxis");
2237   const bool yaxis = RNA_boolean_get(op->ptr, "yaxis");
2238   const bool zaxis = RNA_boolean_get(op->ptr, "zaxis");
2239   int repeat = RNA_int_get(op->ptr, "repeat");
2240
2241   if (!repeat) {
2242     repeat = 1;
2243   }
2244
2245   ViewLayer *view_layer = CTX_data_view_layer(C);
2246   uint objects_len = 0;
2247   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
2248       view_layer, CTX_wm_view3d(C), &objects_len);
2249   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
2250     Object *obedit = objects[ob_index];
2251     Mesh *me = obedit->data;
2252     BMEditMesh *em = BKE_editmesh_from_object(obedit);
2253     ModifierData *md;
2254     bool mirrx = false, mirry = false, mirrz = false;
2255     int i;
2256     float clip_dist = 0.0f;
2257     const bool use_topology = (me->editflag & ME_EDIT_MIRROR_TOPO) != 0;
2258
2259     if (em->bm->totvertsel == 0) {
2260       continue;
2261     }
2262
2263     /* mirror before smooth */
2264     if (((Mesh *)obedit->data)->editflag & ME_EDIT_MIRROR_X) {
2265       EDBM_verts_mirror_cache_begin(em, 0, false, true, use_topology);
2266     }
2267
2268     /* if there is a mirror modifier with clipping, flag the verts that
2269      * are within tolerance of the plane(s) of reflection
2270      */
2271     for (md = obedit->modifiers.first; md; md = md->next) {
2272       if (md->type == eModifierType_Mirror && (md->mode & eModifierMode_Realtime)) {
2273         MirrorModifierData *mmd = (MirrorModifierData *)md;
2274
2275         if (mmd->flag & MOD_MIR_CLIPPING) {
2276           if (mmd->flag & MOD_MIR_AXIS_X) {
2277             mirrx = true;
2278           }
2279           if (mmd->flag & MOD_MIR_AXIS_Y) {
2280             mirry = true;
2281           }
2282           if (mmd->flag & MOD_MIR_AXIS_Z) {
2283             mirrz = true;
2284           }
2285
2286           clip_dist = mmd->tolerance;
2287         }
2288       }
2289     }
2290
2291     for (i = 0; i < repeat; i++) {
2292       if (!EDBM_op_callf(
2293               em,
2294               op,
2295               "smooth_vert verts=%hv factor=%f mirror_clip_x=%b mirror_clip_y=%b mirror_clip_z=%b "
2296               "clip_dist=%f use_axis_x=%b use_axis_y=%b use_axis_z=%b",
2297               BM_ELEM_SELECT,
2298               fac,
2299               mirrx,
2300               mirry,
2301               mirrz,
2302               clip_dist,
2303               xaxis,
2304               yaxis,
2305               zaxis)) {
2306         continue;
2307       }
2308     }
2309
2310     /* apply mirror */
2311     if (((Mesh *)obedit->data)->editflag & ME_EDIT_MIRROR_X) {
2312       EDBM_verts_mirror_apply(em, BM_ELEM_SELECT, 0);
2313       EDBM_verts_mirror_cache_end(em);
2314     }
2315
2316     EDBM_update_generic(em, true, false);
2317   }
2318   MEM_freeN(objects);
2319
2320   return OPERATOR_FINISHED;
2321 }
2322
2323 void MESH_OT_vertices_smooth(wmOperatorType *ot)
2324 {
2325   /* identifiers */
2326   ot->name = "Smooth Vertices";
2327   ot->description = "Flatten angles of selected vertices";
2328   ot->idname = "MESH_OT_vertices_smooth";
2329
2330   /* api callbacks */
2331   ot->exec = edbm_do_smooth_vertex_exec;
2332   ot->poll = ED_operator_editmesh;
2333
2334   /* flags */
2335   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
2336
2337   ot->prop = RNA_def_float_factor(
2338       ot->srna, "factor", 0.5f, -10.0f, 10.0f, "Smoothing", "Smoothing factor", 0.0f, 1.0f);
2339   RNA_def_int(
2340       ot->srna, "repeat", 1, 1, 1000, "Repeat", "Number of times to smooth the mesh", 1, 100);
2341
2342   WM_operatortype_props_advanced_begin(ot);
2343
2344   RNA_def_boolean(ot->srna, "xaxis", true, "X-Axis", "Smooth along the X axis");
2345   RNA_def_boolean(ot->srna, "yaxis", true, "Y-Axis", "Smooth along the Y axis");
2346   RNA_def_boolean(ot->srna, "zaxis", true, "Z-Axis", "Smooth along the Z axis");
2347 }
2348
2349 /** \} */
2350
2351 /* -------------------------------------------------------------------- */
2352 /** \name Laplacian Smooth Vertices Operator
2353  * \{ */
2354
2355 static int edbm_do_smooth_laplacian_vertex_exec(bContext *C, wmOperator *op)
2356 {
2357   BMIter fiter;
2358   BMFace *f;
2359   int tot_invalid = 0;
2360   int tot_unselected = 0;
2361   ViewLayer *view_layer = CTX_data_view_layer(C);
2362
2363   const float lambda_factor = RNA_float_get(op->ptr, "lambda_factor");
2364   const float lambda_border = RNA_float_get(op->ptr, "lambda_border");
2365   const bool usex = RNA_boolean_get(op->ptr, "use_x");
2366   const bool usey = RNA_boolean_get(op->ptr, "use_y");
2367   const bool usez = RNA_boolean_get(op->ptr, "use_z");
2368   const bool preserve_volume = RNA_boolean_get(op->ptr, "preserve_volume");
2369   int repeat = RNA_int_get(op->ptr, "repeat");
2370
2371   if (!repeat) {
2372     repeat = 1;
2373   }
2374
2375   uint objects_len = 0;
2376   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
2377       view_layer, CTX_wm_view3d(C), &objects_len);
2378   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
2379     Object *obedit = objects[ob_index];
2380     BMEditMesh *em = BKE_editmesh_from_object(obedit);
2381     Mesh *me = obedit->data;
2382     bool use_topology = (me->editflag & ME_EDIT_MIRROR_TOPO) != 0;
2383
2384     if (em->bm->totvertsel == 0) {
2385       tot_unselected++;
2386       tot_invalid++;
2387       continue;
2388     }
2389
2390     bool is_invalid = false;
2391     /* Check if select faces are triangles. */
2392     BM_ITER_MESH (f, &fiter, em->bm, BM_FACES_OF_MESH) {
2393       if (BM_elem_flag_test(f, BM_ELEM_SELECT)) {
2394         if (f->len > 4) {
2395           tot_invalid++;
2396           is_invalid = true;
2397           break;
2398         }
2399       }
2400     }
2401     if (is_invalid) {
2402       continue;
2403     }
2404
2405     /* Mirror before smooth. */
2406     if (((Mesh *)obedit->data)->editflag & ME_EDIT_MIRROR_X) {
2407       EDBM_verts_mirror_cache_begin(em, 0, false, true, use_topology);
2408     }
2409
2410     bool failed_repeat_loop = false;
2411     for (int i = 0; i < repeat; i++) {
2412       if (!EDBM_op_callf(em,
2413                          op,
2414                          "smooth_laplacian_vert verts=%hv lambda_factor=%f lambda_border=%f "
2415                          "use_x=%b use_y=%b use_z=%b preserve_volume=%b",
2416                          BM_ELEM_SELECT,
2417                          lambda_factor,
2418                          lambda_border,
2419                          usex,
2420                          usey,
2421                          usez,
2422                          preserve_volume)) {
2423         failed_repeat_loop = true;
2424         break;
2425       }
2426     }
2427     if (failed_repeat_loop) {
2428       continue;
2429     }
2430
2431     /* Apply mirror. */
2432     if (((Mesh *)obedit->data)->editflag & ME_EDIT_MIRROR_X) {
2433       EDBM_verts_mirror_apply(em, BM_ELEM_SELECT, 0);
2434       EDBM_verts_mirror_cache_end(em);
2435     }
2436
2437     EDBM_update_generic(em, true, false);
2438   }
2439   MEM_freeN(objects);
2440
2441   if (tot_unselected == objects_len) {
2442     BKE_report(op->reports, RPT_WARNING, "No selected vertex");
2443     return OPERATOR_CANCELLED;
2444   }
2445   else if (tot_invalid == objects_len) {
2446     BKE_report(op->reports, RPT_WARNING, "Selected faces must be triangles or quads");
2447     return OPERATOR_CANCELLED;
2448   }
2449
2450   return OPERATOR_FINISHED;
2451 }
2452
2453 void MESH_OT_vertices_smooth_laplacian(wmOperatorType *ot)
2454 {
2455   /* identifiers */
2456   ot->name = "Laplacian Smooth Vertices";
2457   ot->description = "Laplacian smooth of selected vertices";
2458   ot->idname = "MESH_OT_vertices_smooth_laplacian";
2459
2460   /* api callbacks */
2461   ot->exec = edbm_do_smooth_laplacian_vertex_exec;
2462   ot->poll = ED_operator_editmesh;
2463
2464   /* flags */
2465   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
2466
2467   RNA_def_int(
2468       ot->srna, "repeat", 1, 1, 1000, "Number of iterations to smooth the mesh", "", 1, 200);
2469   RNA_def_float(
2470       ot->srna, "lambda_factor", 1.0f, 1e-7f, 1000.0f, "Lambda factor", "", 1e-7f, 1000.0f);
2471   RNA_def_float(ot->srna,
2472                 "lambda_border",
2473                 5e-5f,
2474                 1e-7f,
2475                 1000.0f,
2476                 "Lambda factor in border",
2477                 "",
2478                 1e-7f,
2479                 1000.0f);
2480
2481   WM_operatortype_props_advanced_begin(ot);
2482
2483   RNA_def_boolean(ot->srna, "use_x", true, "Smooth X Axis", "Smooth object along X axis");
2484   RNA_def_boolean(ot->srna, "use_y", true, "Smooth Y Axis", "Smooth object along Y axis");
2485   RNA_def_boolean(ot->srna, "use_z", true, "Smooth Z Axis", "Smooth object along Z axis");
2486   RNA_def_boolean(ot->srna,
2487                   "preserve_volume",
2488                   true,
2489                   "Preserve Volume",
2490                   "Apply volume preservation after smooth");
2491 }
2492
2493 /** \} */
2494
2495 /* -------------------------------------------------------------------- */
2496 /** \name Set Faces Smooth Shading Operator
2497  * \{ */
2498
2499 static void mesh_set_smooth_faces(BMEditMesh *em, short smooth)
2500 {
2501   BMIter iter;
2502   BMFace *efa;
2503
2504   if (em == NULL) {
2505     return;
2506   }
2507
2508   BM_ITER_MESH (efa, &iter, em->bm, BM_FACES_OF_MESH) {
2509     if (BM_elem_flag_test(efa, BM_ELEM_SELECT)) {
2510       BM_elem_flag_set(efa, BM_ELEM_SMOOTH, smooth);
2511     }
2512   }
2513 }
2514
2515 static int edbm_faces_shade_smooth_exec(bContext *C, wmOperator *UNUSED(op))
2516 {
2517   ViewLayer *view_layer = CTX_data_view_layer(C);
2518   uint objects_len = 0;
2519   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
2520       view_layer, CTX_wm_view3d(C), &objects_len);
2521   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
2522     Object *obedit = objects[ob_index];
2523     BMEditMesh *em = BKE_editmesh_from_object(obedit);
2524
2525     if (em->bm->totfacesel == 0) {
2526       continue;
2527     }
2528
2529     mesh_set_smooth_faces(em, 1);
2530     EDBM_update_generic(em, false, false);
2531   }
2532   MEM_freeN(objects);
2533
2534   return OPERATOR_FINISHED;
2535 }
2536
2537 void MESH_OT_faces_shade_smooth(wmOperatorType *ot)
2538 {
2539   /* identifiers */
2540   ot->name = "Shade Smooth";
2541   ot->description = "Display faces smooth (using vertex normals)";
2542   ot->idname = "MESH_OT_faces_shade_smooth";
2543
2544   /* api callbacks */
2545   ot->exec = edbm_faces_shade_smooth_exec;
2546   ot->poll = ED_operator_editmesh;
2547
2548   /* flags */
2549   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
2550 }
2551
2552 /** \} */
2553
2554 /* -------------------------------------------------------------------- */
2555 /** \name Set Faces Flat Shading Operator
2556  * \{ */
2557
2558 static int edbm_faces_shade_flat_exec(bContext *C, wmOperator *UNUSED(op))
2559 {
2560   ViewLayer *view_layer = CTX_data_view_layer(C);
2561   uint objects_len = 0;
2562   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
2563       view_layer, CTX_wm_view3d(C), &objects_len);
2564   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
2565     Object *obedit = objects[ob_index];
2566     BMEditMesh *em = BKE_editmesh_from_object(obedit);
2567
2568     if (em->bm->totfacesel == 0) {
2569       continue;
2570     }
2571
2572     mesh_set_smooth_faces(em, 0);
2573     EDBM_update_generic(em, false, false);
2574   }
2575   MEM_freeN(objects);
2576
2577   return OPERATOR_FINISHED;
2578 }
2579
2580 void MESH_OT_faces_shade_flat(wmOperatorType *ot)
2581 {
2582   /* identifiers */
2583   ot->name = "Shade Flat";
2584   ot->description = "Display faces flat";
2585   ot->idname = "MESH_OT_faces_shade_flat";
2586
2587   /* api callbacks */
2588   ot->exec = edbm_faces_shade_flat_exec;
2589   ot->poll = ED_operator_editmesh;
2590
2591   /* flags */
2592   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
2593 }
2594
2595 /** \} */
2596
2597 /* -------------------------------------------------------------------- */
2598 /** \name UV/Color Rotate/Reverse Operator
2599  * \{ */
2600
2601 static int edbm_rotate_uvs_exec(bContext *C, wmOperator *op)
2602 {
2603   /* get the direction from RNA */
2604   const bool use_ccw = RNA_boolean_get(op->ptr, "use_ccw");
2605
2606   ViewLayer *view_layer = CTX_data_view_layer(C);
2607   uint objects_len = 0;
2608   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
2609       view_layer, CTX_wm_view3d(C), &objects_len);
2610   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
2611     Object *obedit = objects[ob_index];
2612     BMEditMesh *em = BKE_editmesh_from_object(obedit);
2613
2614     if (em->bm->totfacesel == 0) {
2615       continue;
2616     }
2617
2618     BMOperator bmop;
2619
2620     /* initialize the bmop using EDBM api, which does various ui error reporting and other stuff */
2621     EDBM_op_init(em, &bmop, op, "rotate_uvs faces=%hf use_ccw=%b", BM_ELEM_SELECT, use_ccw);
2622
2623     /* execute the operator */
2624     BMO_op_exec(em->bm, &bmop);
2625
2626     if (!EDBM_op_finish(em, &bmop, op, true)) {
2627       continue;
2628     }
2629
2630     EDBM_update_generic(em, false, false);
2631   }
2632
2633   MEM_freeN(objects);
2634   return OPERATOR_FINISHED;
2635 }
2636
2637 static int edbm_reverse_uvs_exec(bContext *C, wmOperator *op)
2638 {
2639   ViewLayer *view_layer = CTX_data_view_layer(C);
2640   uint objects_len = 0;
2641   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
2642       view_layer, CTX_wm_view3d(C), &objects_len);
2643   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
2644     Object *obedit = objects[ob_index];
2645     BMEditMesh *em = BKE_editmesh_from_object(obedit);
2646
2647     if (em->bm->totfacesel == 0) {
2648       continue;
2649     }
2650
2651     BMOperator bmop;
2652
2653     /* initialize the bmop using EDBM api, which does various ui error reporting and other stuff */
2654     EDBM_op_init(em, &bmop, op, "reverse_uvs faces=%hf", BM_ELEM_SELECT);
2655
2656     /* execute the operator */
2657     BMO_op_exec(em->bm, &bmop);
2658
2659     /* finish the operator */
2660     if (!EDBM_op_finish(em, &bmop, op, true)) {
2661       continue;
2662     }
2663     EDBM_update_generic(em, false, false);
2664   }
2665
2666   MEM_freeN(objects);
2667   return OPERATOR_FINISHED;
2668 }
2669
2670 static int edbm_rotate_colors_exec(bContext *C, wmOperator *op)
2671 {
2672   /* get the direction from RNA */
2673   const bool use_ccw = RNA_boolean_get(op->ptr, "use_ccw");
2674
2675   ViewLayer *view_layer = CTX_data_view_layer(C);
2676   uint objects_len = 0;
2677   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
2678       view_layer, CTX_wm_view3d(C), &objects_len);
2679
2680   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
2681     Object *ob = objects[ob_index];
2682     BMEditMesh *em = BKE_editmesh_from_object(ob);
2683     if (em->bm->totfacesel == 0) {
2684       continue;
2685     }
2686
2687     BMOperator bmop;
2688
2689     /* initialize the bmop using EDBM api, which does various ui error reporting and other stuff */
2690     EDBM_op_init(em, &bmop, op, "rotate_colors faces=%hf use_ccw=%b", BM_ELEM_SELECT, use_ccw);
2691
2692     /* execute the operator */
2693     BMO_op_exec(em->bm, &bmop);
2694
2695     /* finish the operator */
2696     if (!EDBM_op_finish(em, &bmop, op, true)) {
2697       continue;
2698     }
2699
2700     /* dependencies graph and notification stuff */
2701     EDBM_update_generic(em, false, false);
2702   }
2703
2704   MEM_freeN(objects);
2705
2706   return OPERATOR_FINISHED;
2707 }
2708
2709 static int edbm_reverse_colors_exec(bContext *C, wmOperator *op)
2710 {
2711   ViewLayer *view_layer = CTX_data_view_layer(C);
2712   uint objects_len = 0;
2713   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
2714       view_layer, CTX_wm_view3d(C), &objects_len);
2715
2716   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
2717     Object *ob = objects[ob_index];
2718     BMEditMesh *em = BKE_editmesh_from_object(ob);
2719
2720     if (em->bm->totfacesel == 0) {
2721       continue;
2722     }
2723
2724     BMOperator bmop;
2725
2726     /* initialize the bmop using EDBM api, which does various ui error reporting and other stuff */
2727     EDBM_op_init(em, &bmop, op, "reverse_colors faces=%hf", BM_ELEM_SELECT);
2728
2729     /* execute the operator */
2730     BMO_op_exec(em->bm, &bmop);
2731
2732     /* finish the operator */
2733     if (!EDBM_op_finish(em, &bmop, op, true)) {
2734       return OPERATOR_CANCELLED;
2735     }
2736
2737     EDBM_update_generic(em, false, false);
2738   }
2739   MEM_freeN(objects);
2740
2741   return OPERATOR_FINISHED;
2742 }
2743
2744 void MESH_OT_uvs_rotate(wmOperatorType *ot)
2745 {
2746   /* identifiers */
2747   ot->name = "Rotate UVs";
2748   ot->idname = "MESH_OT_uvs_rotate";
2749   ot->description = "Rotate UV coordinates inside faces";
2750
2751   /* api callbacks */
2752   ot->exec = edbm_rotate_uvs_exec;
2753   ot->poll = ED_operator_editmesh;
2754
2755   /* flags */
2756   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
2757
2758   /* props */
2759   RNA_def_boolean(ot->srna, "use_ccw", false, "Counter Clockwise", "");
2760 }
2761
2762 void MESH_OT_uvs_reverse(wmOperatorType *ot)
2763 {
2764   /* identifiers */
2765   ot->name = "Reverse UVs";
2766   ot->idname = "MESH_OT_uvs_reverse";
2767   ot->description = "Flip direction of UV coordinates inside faces";
2768
2769   /* api callbacks */
2770   ot->exec = edbm_reverse_uvs_exec;
2771   ot->poll = ED_operator_editmesh;
2772
2773   /* flags */
2774   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
2775
2776   /* props */
2777   //RNA_def_enum(ot->srna, "axis", axis_items, DIRECTION_CW, "Axis", "Axis to mirror UVs around");
2778 }
2779
2780 void MESH_OT_colors_rotate(wmOperatorType *ot)
2781 {
2782   /* identifiers */
2783   ot->name = "Rotate Colors";
2784   ot->idname = "MESH_OT_colors_rotate";
2785   ot->description = "Rotate vertex colors inside faces";
2786
2787   /* api callbacks */
2788   ot->exec = edbm_rotate_colors_exec;
2789   ot->poll = ED_operator_editmesh;
2790
2791   /* flags */
2792   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
2793
2794   /* props */
2795   RNA_def_boolean(ot->srna, "use_ccw", false, "Counter Clockwise", "");
2796 }
2797
2798 void MESH_OT_colors_reverse(wmOperatorType *ot)
2799 {
2800   /* identifiers */
2801   ot->name = "Reverse Colors";
2802   ot->idname = "MESH_OT_colors_reverse";
2803   ot->description = "Flip direction of vertex colors inside faces";
2804
2805   /* api callbacks */
2806   ot->exec = edbm_reverse_colors_exec;
2807   ot->poll = ED_operator_editmesh;
2808
2809   /* flags */
2810   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
2811
2812   /* props */
2813   //RNA_def_enum(ot->srna, "axis", axis_items, DIRECTION_CW, "Axis", "Axis to mirror colors around");
2814 }
2815
2816 /** \} */
2817
2818 /* -------------------------------------------------------------------- */
2819 /** \name Merge Vertices Operator
2820  * \{ */
2821
2822 enum {
2823   MESH_MERGE_LAST = 1,
2824   MESH_MERGE_CENTER = 3,
2825   MESH_MERGE_CURSOR = 4,
2826   MESH_MERGE_COLLAPSE = 5,
2827   MESH_MERGE_FIRST = 6,
2828 };
2829
2830 static bool merge_firstlast(BMEditMesh *em,
2831                             const bool use_first,
2832                             const bool use_uvmerge,
2833                             wmOperator *wmop)
2834 {
2835   BMVert *mergevert;
2836   BMEditSelection *ese;
2837
2838   /* operator could be called directly from shortcut or python,
2839    * so do extra check for data here
2840    */
2841
2842   /* do sanity check in mergemenu in edit.c ?*/
2843   if (use_first == false) {
2844     if (!em->bm->selected.last || ((BMEditSelection *)em->bm->selected.last)->htype != BM_VERT) {
2845       return false;
2846     }
2847
2848     ese = em->bm->selected.last;
2849     mergevert = (BMVert *)ese->ele;
2850   }
2851   else {
2852     if (!em->bm->selected.first || ((BMEditSelection *)em->bm->selected.first)->htype != BM_VERT) {
2853       return false;
2854     }
2855
2856     ese = em->bm->selected.first;
2857     mergevert = (BMVert *)ese->ele;
2858   }
2859
2860   if (!BM_elem_flag_test(mergevert, BM_ELEM_SELECT)) {
2861     return false;
2862   }
2863
2864   if (use_uvmerge) {
2865     if (!EDBM_op_callf(
2866             em, wmop, "pointmerge_facedata verts=%hv vert_snap=%e", BM_ELEM_SELECT, mergevert)) {
2867       return false;
2868     }
2869   }
2870
2871   if (!EDBM_op_callf(
2872           em, wmop, "pointmerge verts=%hv merge_co=%v", BM_ELEM_SELECT, mergevert->co)) {
2873     return false;
2874   }
2875
2876   return true;
2877 }
2878
2879 static bool merge_target(BMEditMesh *em,
2880                          Scene *scene,
2881                          Object *ob,
2882                          const bool use_cursor,
2883                          const bool use_uvmerge,
2884                          wmOperator *wmop)
2885 {
2886   BMIter iter;
2887   BMVert *v;
2888   float co[3], cent[3] = {0.0f, 0.0f, 0.0f};
2889   const float *vco = NULL;
2890
2891   if (use_cursor) {
2892     vco = scene->cursor.location;
2893     copy_v3_v3(co, vco);
2894     invert_m4_m4(ob->imat, ob->obmat);
2895     mul_m4_v3(ob->imat, co);
2896   }
2897   else {
2898     float fac;
2899     int i = 0;
2900     BM_ITER_MESH (v, &iter, em->bm, BM_VERTS_OF_MESH) {
2901       if (!BM_elem_flag_test(v, BM_ELEM_SELECT)) {
2902         continue;
2903       }
2904       add_v3_v3(cent, v->co);
2905       i++;
2906     }
2907
2908     if (!i) {
2909       return false;
2910     }
2911
2912     fac = 1.0f / (float)i;
2913     mul_v3_fl(cent, fac);
2914     copy_v3_v3(co, cent);
2915     vco = co;
2916   }
2917
2918   if (!vco) {
2919     return false;
2920   }
2921
2922   if (use_uvmerge) {
2923     if (!EDBM_op_callf(em, wmop, "average_vert_facedata verts=%hv", BM_ELEM_SELECT)) {
2924       return false;
2925     }
2926   }
2927
2928   if (!EDBM_op_callf(em, wmop, "pointmerge verts=%hv merge_co=%v", BM_ELEM_SELECT, co)) {
2929     return false;
2930   }
2931
2932   return true;
2933 }
2934
2935 static int edbm_merge_exec(bContext *C, wmOperator *op)
2936 {
2937   Scene *scene = CTX_data_scene(C);
2938   ViewLayer *view_layer = CTX_data_view_layer(C);
2939   uint objects_len = 0;
2940   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
2941       view_layer, CTX_wm_view3d(C), &objects_len);
2942   const int type = RNA_enum_get(op->ptr, "type");
2943   const bool uvs = RNA_boolean_get(op->ptr, "uvs");
2944
2945   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
2946     Object *obedit = objects[ob_index];
2947     BMEditMesh *em = BKE_editmesh_from_object(obedit);
2948
2949     if (em->bm->totvertsel == 0) {
2950       continue;
2951     }
2952
2953     bool ok = false;
2954     switch (type) {
2955       case MESH_MERGE_CENTER:
2956         ok = merge_target(em, scene, obedit, false, uvs, op);
2957         break;
2958       case MESH_MERGE_CURSOR:
2959         ok = merge_target(em, scene, obedit, true, uvs, op);
2960         break;
2961       case MESH_MERGE_LAST:
2962         ok = merge_firstlast(em, false, uvs, op);
2963         break;
2964       case MESH_MERGE_FIRST:
2965         ok = merge_firstlast(em, true, uvs, op);
2966         break;
2967       case MESH_MERGE_COLLAPSE:
2968         ok = EDBM_op_callf(em, op, "collapse edges=%he uvs=%b", BM_ELEM_SELECT, uvs);
2969         break;
2970       default:
2971         BLI_assert(0);
2972         break;
2973     }
2974
2975     if (!ok) {
2976       continue;
2977     }
2978
2979     EDBM_update_generic(em, true, true);
2980
2981     /* once collapsed, we can't have edge/face selection */
2982     if ((em->selectmode & SCE_SELECT_VERTEX) == 0) {
2983       EDBM_flag_disable_all(em, BM_ELEM_SELECT);
2984     }
2985     /* Only active object supported, see comment below. */
2986     if (ELEM(type, MESH_MERGE_FIRST, MESH_MERGE_LAST)) {
2987       break;
2988     }
2989   }
2990
2991   MEM_freeN(objects);
2992
2993   return OPERATOR_FINISHED;
2994 }
2995
2996 static const EnumPropertyItem merge_type_items[] = {
2997     {MESH_MERGE_FIRST, "FIRST", 0, "At First", ""},
2998     {MESH_MERGE_LAST, "LAST", 0, "At Last", ""},
2999     {MESH_MERGE_CENTER, "CENTER", 0, "At Center", ""},
3000     {MESH_MERGE_CURSOR, "CURSOR", 0, "At Cursor", ""},
3001     {MESH_MERGE_COLLAPSE, "COLLAPSE", 0, "Collapse", ""},
3002     {0, NULL, 0, NULL, NULL},
3003 };
3004
3005 static const EnumPropertyItem *merge_type_itemf(bContext *C,
3006                                                 PointerRNA *UNUSED(ptr),
3007                                                 PropertyRNA *UNUSED(prop),
3008                                                 bool *r_free)
3009 {
3010   Object *obedit;
3011   EnumPropertyItem *item = NULL;
3012   int totitem = 0;
3013
3014   if (!C) { /* needed for docs */
3015     return merge_type_items;
3016   }
3017
3018   obedit = CTX_data_edit_object(C);
3019   if (obedit && obedit->type == OB_MESH) {
3020     BMEditMesh *em = BKE_editmesh_from_object(obedit);
3021
3022     /* Only active object supported:
3023      * In practice it doesn't make sense to run this operation on non-active meshes
3024      * since selecting will activate - we could have own code-path for these but it's a hassle
3025      * for now just apply to the active (first) object. */
3026     if (em->selectmode & SCE_SELECT_VERTEX) {
3027       if (em->bm->selected.first && em->bm->selected.last &&
3028           ((BMEditSelection *)em->bm->selected.first)->htype == BM_VERT &&
3029           ((BMEditSelection *)em->bm->selected.last)->htype == BM_VERT) {
3030         RNA_enum_items_add_value(&item, &totitem, merge_type_items, MESH_MERGE_FIRST);
3031         RNA_enum_items_add_value(&item, &totitem, merge_type_items, MESH_MERGE_LAST);
3032       }
3033       else if (em->bm->selected.first &&
3034                ((BMEditSelection *)em->bm->selected.first)->htype == BM_VERT) {
3035         RNA_enum_items_add_value(&item, &totitem, merge_type_items, MESH_MERGE_FIRST);
3036       }
3037       else if (em->bm->selected.last &&
3038                ((BMEditSelection *)em->bm->selected.last)->htype == BM_VERT) {
3039         RNA_enum_items_add_value(&item, &totitem, merge_type_items, MESH_MERGE_LAST);
3040       }
3041     }
3042
3043     RNA_enum_items_add_value(&item, &totitem, merge_type_items, MESH_MERGE_CENTER);
3044     RNA_enum_items_add_value(&item, &totitem, merge_type_items, MESH_MERGE_CURSOR);
3045     RNA_enum_items_add_value(&item, &totitem, merge_type_items, MESH_MERGE_COLLAPSE);
3046     RNA_enum_item_end(&item, &totitem);
3047
3048     *r_free = true;
3049
3050     return item;
3051   }
3052
3053   /* Get all items e.g. when creating keymap item. */
3054   return merge_type_items;
3055 }
3056
3057 void MESH_OT_merge(wmOperatorType *ot)
3058 {
3059   /* identifiers */
3060   ot->name = "Merge";
3061   ot->description = "Merge selected vertices";
3062   ot->idname = "MESH_OT_merge";
3063
3064   /* api callbacks */
3065   ot->exec = edbm_merge_exec;
3066   ot->invoke = WM_menu_invoke;
3067   ot->poll = ED_operator_editmesh;
3068
3069   /* flags */
3070   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
3071
3072   /* properties */
3073   ot->prop = RNA_def_enum(
3074       ot->srna, "type", merge_type_items, MESH_MERGE_CENTER, "Type", "Merge method to use");
3075   RNA_def_enum_funcs(ot->prop, merge_type_itemf);
3076
3077   WM_operatortype_props_advanced_begin(ot);
3078
3079   RNA_def_boolean(ot->srna, "uvs", false, "UVs", "Move UVs according to merge");
3080 }
3081
3082 /** \} */
3083
3084 /* -------------------------------------------------------------------- */
3085 /** \name Remove Doubles Operator
3086  * \{ */
3087
3088 static int edbm_remove_doubles_exec(bContext *C, wmOperator *op)
3089 {
3090   const float threshold = RNA_float_get(op->ptr, "threshold");
3091   const bool use_unselected = RNA_boolean_get(op->ptr, "use_unselected");
3092   int count_multi = 0;
3093
3094   ViewLayer *view_layer = CTX_data_view_layer(C);
3095   uint objects_len = 0;
3096   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
3097       view_layer, CTX_wm_view3d(C), &objects_len);
3098
3099   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
3100     Object *obedit = objects[ob_index];
3101     BMEditMesh *em = BKE_editmesh_from_object(obedit);
3102
3103     /* Selection used as target with 'use_unselected'. */
3104     if (em->bm->totvertsel == 0) {
3105       continue;
3106     }
3107
3108     BMOperator bmop;
3109     const int totvert_orig = em->bm->totvert;
3110
3111     /* avoid losing selection state (select -> tags) */
3112     char htype_select;
3113     if (em->selectmode & SCE_SELECT_VERTEX) {
3114       htype_select = BM_VERT;
3115     }
3116     else if (em->selectmode & SCE_SELECT_EDGE) {
3117       htype_select = BM_EDGE;
3118     }
3119     else {
3120       htype_select = BM_FACE;
3121     }
3122
3123     /* store selection as tags */
3124     BM_mesh_elem_hflag_enable_test(em->bm, htype_select, BM_ELEM_TAG, true, true, BM_ELEM_SELECT);
3125
3126     if (use_unselected) {
3127       EDBM_op_init(em, &bmop, op, "automerge verts=%hv dist=%f", BM_ELEM_SELECT, threshold);
3128       BMO_op_exec(em->bm, &bmop);
3129
3130       if (!EDBM_op_finish(em, &bmop, op, true)) {
3131         continue;
3132       }
3133     }
3134     else {
3135       EDBM_op_init(em, &bmop, op, "find_doubles verts=%hv dist=%f", BM_ELEM_SELECT, threshold);
3136
3137       BMO_op_exec(em->bm, &bmop);
3138
3139       if (!EDBM_op_callf(em, op, "weld_verts targetmap=%S", &bmop, "targetmap.out")) {
3140         BMO_op_finish(em->bm, &bmop);
3141         continue;
3142       }
3143
3144       if (!EDBM_op_finish(em, &bmop, op, true)) {
3145         continue;
3146       }
3147     }
3148
3149     const int count = (totvert_orig - em->bm->totvert);
3150
3151     /* restore selection from tags */
3152     BM_mesh_elem_hflag_enable_test(em->bm, htype_select, BM_ELEM_SELECT, true, true, BM_ELEM_TAG);
3153     EDBM_selectmode_flush(em);
3154
3155     if (count) {
3156       count_multi += count;
3157       EDBM_update_generic(em, true, true);
3158     }
3159   }
3160   MEM_freeN(objects);
3161
3162   BKE_reportf(op->reports, RPT_INFO, "Removed %d vertices", count_multi);
3163
3164   return OPERATOR_FINISHED;
3165 }
3166
3167 void MESH_OT_remove_doubles(wmOperatorType *ot)
3168 {
3169   /* identifiers */
3170   ot->name = "Remove Doubles";
3171   ot->description = "Remove duplicate vertices";
3172   ot->idname = "MESH_OT_remove_doubles";
3173
3174   /* api callbacks */
3175   ot->exec = edbm_remove_doubles_exec;
3176   ot->poll = ED_operator_editmesh;
3177
3178   /* flags */
3179   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
3180
3181   RNA_def_float_distance(ot->srna,
3182                          "threshold",
3183                          1e-4f,
3184                          1e-6f,
3185                          50.0f,
3186                          "Merge Distance",
3187                          "Minimum distance between elements to merge",
3188                          1e-5f,
3189                          10.0f);
3190   RNA_def_boolean(ot->srna,
3191                   "use_unselected",
3192                   false,
3193                   "Unselected",
3194                   "Merge selected to other unselected vertices");
3195 }
3196
3197 /** \} */
3198
3199 /* -------------------------------------------------------------------- */
3200 /** \name Shape Key Propagate Operator
3201  * \{ */
3202
3203 /* BMESH_TODO this should be properly encapsulated in a bmop.  but later.*/
3204 static bool shape_propagate(BMEditMesh *em)
3205 {
3206   BMIter iter;
3207   BMVert *eve = NULL;
3208   float *co;
3209   int totshape = CustomData_number_of_layers(&em->bm->vdata, CD_SHAPEKEY);
3210
3211   if (!CustomData_has_layer(&em->bm->vdata, CD_SHAPEKEY)) {
3212     return false;
3213   }
3214
3215   BM_ITER_MESH (eve, &iter, em->bm, BM_VERTS_OF_MESH) {
3216     if (!BM_elem_flag_test(eve, BM_ELEM_SELECT) || BM_elem_flag_test(eve, BM_ELEM_HIDDEN)) {
3217       continue;
3218     }
3219
3220     for (int i = 0; i < totshape; i++) {
3221       co = CustomData_bmesh_get_n(&em->bm->vdata, eve->head.data, CD_SHAPEKEY, i);
3222       copy_v3_v3(co, eve->co);
3223     }
3224   }
3225   return true;
3226 }
3227
3228 static int edbm_shape_propagate_to_all_exec(bContext *C, wmOperator *op)
3229 {
3230   ViewLayer *view_layer = CTX_data_view_layer(C);
3231   int tot_shapekeys = 0;
3232   int tot_selected_verts_objects = 0;
3233
3234   uint objects_len = 0;
3235   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
3236       view_layer, CTX_wm_view3d(C), &objects_len);
3237   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
3238     Object *obedit = objects[ob_index];
3239     Mesh *me = obedit->data;
3240     BMEditMesh *em = me->edit_mesh;
3241
3242     if (em->bm->totvertsel == 0) {
3243       continue;
3244     }
3245     tot_selected_verts_objects++;
3246
3247     if (shape_propagate(em)) {
3248       tot_shapekeys++;
3249     }
3250
3251     EDBM_update_generic(em, false, false);
3252   }
3253   MEM_freeN(objects);
3254
3255   if (tot_selected_verts_objects == 0) {
3256     BKE_report(op->reports, RPT_ERROR, "No selected vertex");
3257     return OPERATOR_CANCELLED;
3258   }
3259   else if (tot_shapekeys == 0) {
3260     BKE_report(op->reports,
3261                RPT_ERROR,
3262                objects_len > 1 ? "Meshes do not have shape keys" :
3263                                  "Mesh does not have shape keys");
3264     return OPERATOR_CANCELLED;
3265   }
3266
3267   return OPERATOR_FINISHED;
3268 }
3269
3270 void MESH_OT_shape_propagate_to_all(wmOperatorType *ot)
3271 {
3272   /* identifiers */
3273   ot->name = "Shape Propagate";
3274   ot->description = "Apply selected vertex locations to all other shape keys";
3275   ot->idname = "MESH_OT_shape_propagate_to_all";
3276
3277   /* api callbacks */
3278   ot->exec = edbm_shape_propagate_to_all_exec;
3279   ot->poll = ED_operator_editmesh;
3280
3281   /* flags */
3282   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
3283 }
3284
3285 /** \} */
3286
3287 /* -------------------------------------------------------------------- */
3288 /** \name Blend from Shape Operator
3289  * \{ */
3290
3291 /* BMESH_TODO this should be properly encapsulated in a bmop.  but later.*/
3292 static int edbm_blend_from_shape_exec(bContext *C, wmOperator *op)
3293 {
3294   Object *obedit_ref = CTX_data_edit_object(C);
3295   Mesh *me_ref = obedit_ref->data;
3296   Key *key_ref = me_ref->key;
3297   KeyBlock *kb_ref = NULL;
3298   BMEditMesh *em_ref = me_ref->edit_mesh;
3299   BMVert *eve;
3300   BMIter iter;
3301   ViewLayer *view_layer = CTX_data_view_layer(C);
3302   float co[3], *sco;
3303   int totshape_ref = 0;
3304
3305   const float blend = RNA_float_get(op->ptr, "blend");
3306   int shape_ref = RNA_enum_get(op->ptr, "shape");
3307   const bool use_add = RNA_boolean_get(op->ptr, "add");
3308
3309   /* Sanity check. */
3310   totshape_ref = CustomData_number_of_layers(&em_ref->bm->vdata, CD_SHAPEKEY);
3311
3312   if (totshape_ref == 0 || shape_ref < 0) {
3313     BKE_report(op->reports, RPT_ERROR, "Active mesh does not have shape keys");
3314     return OPERATOR_CANCELLED;
3315   }
3316   else if (shape_ref >= totshape_ref) {
3317     /* This case occurs if operator was used before on object with more keys than current one. */
3318     shape_ref = 0; /* default to basis */
3319   }
3320
3321   /* Get shape key - needed for finding reference shape (for add mode only). */
3322   if (key_ref) {
3323     kb_ref = BLI_findlink(&key_ref->block, shape_ref);
3324   }
3325
3326   int tot_selected_verts_objects = 0;
3327   uint objects_len = 0;
3328   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
3329       view_layer, CTX_wm_view3d(C), &objects_len);
3330   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
3331     Object *obedit = objects[ob_index];
3332     Mesh *me = obedit->data;
3333     Key *key = me->key;
3334     KeyBlock *kb = NULL;
3335     BMEditMesh *em = me->edit_mesh;
3336     int shape;
3337
3338     if (em->bm->totvertsel == 0) {
3339       continue;
3340     }
3341     tot_selected_verts_objects++;
3342
3343     if (!key) {
3344       continue;
3345     }
3346     else {
3347       kb = BKE_keyblock_find_name(key, kb_ref->name);
3348       shape = BLI_findindex(&key->block, kb);
3349     }
3350
3351     if (kb) {
3352       /* Perform blending on selected vertices. */
3353       BM_ITER_MESH (eve, &iter, em->bm, BM_VERTS_OF_MESH) {
3354         if (!BM_elem_flag_test(eve, BM_ELEM_SELECT) || BM_elem_flag_test(eve, BM_ELEM_HIDDEN)) {
3355           continue;
3356         }
3357
3358         /* Get coordinates of shapekey we're blending from. */
3359         sco = CustomData_bmesh_get_n(&em->bm->vdata, eve->head.data, CD_SHAPEKEY, shape);
3360         copy_v3_v3(co, sco);
3361
3362         if (use_add) {
3363           /* In add mode, we add relative shape key offset. */
3364           if (kb) {
3365             const float *rco = CustomData_bmesh_get_n(
3366                 &em->bm->vdata, eve->head.data, CD_SHAPEKEY, kb->relative);
3367             sub_v3_v3v3(co, co, rco);
3368           }
3369
3370           madd_v3_v3fl(eve->co, co, blend);
3371         }
3372         else {
3373           /* In blend mode, we interpolate to the shape key. */
3374           interp_v3_v3v3(eve->co, eve->co, co, blend);
3375         }
3376       }
3377       EDBM_update_generic(em, true, false);
3378     }
3379   }
3380   MEM_freeN(objects);
3381
3382   if (tot_selected_verts_objects == 0) {
3383     BKE_report(op->reports, RPT_ERROR, "No selected vertex");
3384     return OPERATOR_CANCELLED;
3385   }
3386
3387   return OPERATOR_FINISHED;
3388 }
3389
3390 static const EnumPropertyItem *shape_itemf(bContext *C,
3391                                            PointerRNA *UNUSED(ptr),
3392                                            PropertyRNA *UNUSED(prop),
3393                                            bool *r_free)
3394 {
3395   Object *obedit = CTX_data_edit_object(C);
3396   BMEditMesh *em;
3397   EnumPropertyItem *item = NULL;
3398   int totitem = 0;
3399
3400   if ((obedit && obedit->type == OB_MESH) && (em = BKE_editmesh_from_object(obedit)) &&
3401       CustomData_has_layer(&em->bm->vdata, CD_SHAPEKEY)) {
3402     EnumPropertyItem tmp = {0, "", 0, "", ""};
3403     int a;
3404
3405     for (a = 0; a < em->bm->vdata.totlayer; a++) {
3406       if (em->bm->vdata.layers[a].type != CD_SHAPEKEY) {
3407         continue;
3408       }
3409
3410       tmp.value = totitem;
3411       tmp.identifier = em->bm->vdata.layers[a].name;
3412       tmp.name = em->bm->vdata.layers[a].name;
3413       /* RNA_enum_item_add sets totitem itself! */
3414       RNA_enum_item_add(&item, &totitem, &tmp);
3415     }
3416   }
3417
3418   RNA_enum_item_end(&item, &totitem);
3419   *r_free = true;
3420
3421   return item;
3422 }
3423
3424 static void edbm_blend_from_shape_ui(bContext *C, wmOperator *op)
3425 {
3426   uiLayout *layout = op->layout;
3427   PointerRNA ptr;
3428   Object *obedit = CTX_data_edit_object(C);
3429   Mesh *me = obedit->data;
3430   PointerRNA ptr_key;
3431
3432   RNA_pointer_create(NULL, op->type->srna, op->properties, &ptr);
3433   RNA_id_pointer_create((ID *)me->key, &ptr_key);
3434
3435   uiItemPointerR(layout, &ptr, "shape", &ptr_key, "key_blocks", "", ICON_SHAPEKEY_DATA);
3436   uiItemR(layout, &ptr, "blend", 0, NULL, ICON_NONE);
3437   uiItemR(layout, &ptr, "add", 0, NULL, ICON_NONE);
3438 }
3439
3440 void MESH_OT_blend_from_shape(wmOperatorType *ot)
3441 {
3442   PropertyRNA *prop;
3443
3444   /* identifiers */
3445   ot->name = "Blend From Shape";
3446   ot->description = "Blend in shape from a shape key";
3447   ot->idname = "MESH_OT_blend_from_shape";
3448
3449   /* api callbacks */
3450   ot->exec = edbm_blend_from_shape_exec;
3451   /* disable because search popup closes too easily */
3452   //  ot->invoke = WM_operator_props_popup_call;
3453   ot->ui = edbm_blend_from_shape_ui;
3454   ot->poll = ED_operator_editmesh;
3455
3456   /* flags */
3457   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
3458
3459   /* properties */
3460   prop = RNA_def_enum(
3461       ot->srna, "shape", DummyRNA_NULL_items, 0, "Shape", "Shape key to use for blending");
3462   RNA_def_enum_funcs(prop, shape_itemf);
3463   RNA_def_property_flag(prop, PROP_ENUM_NO_TRANSLATE | PROP_NEVER_UNLINK);
3464   RNA_def_float(ot->srna, "blend", 1.0f, -1e3f, 1e3f, "Blend", "Blending factor", -2.0f, 2.0f);
3465   RNA_def_boolean(ot->srna, "add", true, "Add", "Add rather than blend between shapes");
3466 }
3467
3468 /** \} */
3469
3470 /* -------------------------------------------------------------------- */
3471 /** \name Solidify Mesh Operator
3472  * \{ */
3473
3474 static int edbm_solidify_exec(bContext *C, wmOperator *op)
3475 {
3476   const float thickness = RNA_float_get(op->ptr, "thickness");
3477
3478   ViewLayer *view_layer = CTX_data_view_layer(C);
3479   uint objects_len = 0;
3480   Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(
3481       view_layer, CTX_wm_view3d(C), &objects_len);
3482   for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
3483     Object *obedit = objects[ob_index];
3484     BMEditMesh *em = BKE_editmesh_from_object(obedit);
3485     BMesh *bm = em->bm;
3486
3487     if (em->bm->totfacesel == 0) {
3488       continue;
3489     }
3490
3491     BMOperator bmop;
3492
3493     if (!EDBM_op_init(
3494             em, &bmop, op, "solidify geom=%hf thickness=%f", BM_ELEM_SELECT, thickness)) {
3495       continue;
3496     }
3497
3498     /* deselect only the faces in the region to be solidified (leave wire
3499      * edges and loose verts selected, as there will be no corresponding
3500      * geometry selected below) */
3501     BMO_slot_buffer_hflag_disable(bm, bmop.slots_in, "geom", BM_FACE, BM_ELEM_SELECT, true);
3502
3503     /* run the solidify operator */
3504     BMO_op_exec(bm, &bmop);
3505
3506     /* select the newly generated faces */
3507     BMO_slot_buffer_hflag_enable(bm, bmop.slots_out, "geom.out", BM_FACE, BM_ELEM_SELECT, true);
3508
3509     if (!EDBM_op_finish(em, &bmop, op, true)) {
3510       continue;
3511     }
3512
3513     EDBM_update_generic(em, true, true);
3514   }
3515
3516   MEM_freeN(objects);
3517   return OPERATOR_FINISHED;
3518 }
3519
3520 void MESH_OT_solidify(wmOperatorType *ot)
3521 {
3522   PropertyRNA *prop;
3523   /* identifiers */
3524   ot->name = "Solidify";
3525   ot->description = "Create a solid skin by extruding, compensating for sharp angles";
3526   ot->idname = "MESH_OT_solidify";
3527
3528   /* api callbacks */
3529   ot->exec = edbm_solidify_exec;
3530   ot->poll = ED_operator_editmesh;
3531
3532   /* flags */
3533   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
3534
3535   prop = RNA_def_float_distance(
3536       ot->srna, "thickness", 0.01f, -1e4f, 1e4f, "Thickness", "", -10.0f, 10.0f);
3537   RNA_def_property_ui_range(prop, -10.0, 10.0, 0.1, 4);
3538 }
3539
3540 /** \} */
3541
3542 /* -------------------------------------------------------------------- */
3543 /** \name Knife Subdivide Operator
3544  * \{ */
3545
3546 #define KNIFE_EXACT 1
3547 #define KNIFE_MIDPOINT 2
3548 #define KNIFE_MULTICUT 3
3549
3550 static const EnumPropertyItem knife_items[] = {
3551     {KNIFE_EXACT, "EXACT", 0, "Exact", ""},
3552     {KNIFE_MIDPOINT, "MIDPOINTS", 0, "Midpoints", ""},
3553     {KNIFE_MULTICUT, "MULTICUT", 0, "Multicut", ""},
3554     {0, NULL, 0, NULL, NULL},
3555 };
3556
3557 /* bm_edge_seg_isect() Determines if and where a mouse trail intersects an BMEdge */
3558
3559 static float bm_edge_seg_isect(const float sco_a[2],
3560                                const float sco_b[2],
3561                                float (*mouse_path)[2],
3562                                int len,
3563                                char mode,
3564                                int *isected)
3565 {
3566 #define MAXSLOPE 100000
3567   float x11, y11, x12 = 0, y12 = 0, x2max, x2min, y2max;
3568   float y2min, dist, lastdist = 0, xdiff2, xdiff1;
3569   float m1, b1, m2, b2, x21, x22, y21, y22, xi;
3570   float yi, x1min, x1max, y1max, y1min, perc = 0;
3571   float threshold = 0.0;
3572   int i;
3573
3574   //threshold = 0.000001; /* tolerance for vertex intersection */
3575   // XXX threshold = scene->toolsettings->select_thresh / 100;
3576
3577   /* Get screen coords of verts */
3578   x21 = sco_a[0];
3579   y21 = sco_a[1];
3580
3581   x22 = sco_b[0];
3582   y22 = sco_b[1];
3583
3584   xdiff2 = (x22 - x21);
3585   if (xdiff2) {
3586     m2 = (y22 - y21) / xdiff2;
3587     b2 = ((x22 * y21) - (x21 * y22)) / xdiff2;
3588   }
3589   else {
3590     m2 = MAXSLOPE; /* Vertical slope  */
3591     b2 = x22;
3592   }
3593
3594   *isected = 0;
3595
3596   /* check for _exact_ vertex intersection first */
3597   if (mode != KNIFE_MULTICUT) {
3598     for (i = 0; i < len; i++) {
3599       if (i > 0) {
3600         x11 = x12;
3601         y11 = y12;
3602       }
3603       else {
3604         x11 = mouse_path[i][0];
3605         y11 = mouse_path[i][1];
3606       }
3607       x12 = mouse_path[i][0];
3608       y12 = mouse_path[i][1];
3609
3610       /* test e->v1 */
3611       if ((x11 == x21 && y11 == y21) || (x12 == x21 && y12 == y21)) {
3612         perc = 0;
3613         *isected = 1;
3614         return perc;
3615       }
3616       /* test e->v2 */
3617       else if ((x11 == x22 && y11 == y22) || (x12 == x22 && y12 == y22)) {
3618         perc = 0;
3619         *isected = 2;
3620         return perc;
3621       }
3622     }
3623   }
3624
3625   /* now check for edge intersect (may produce vertex intersection as well) */
3626   for (i = 0; i < len; i++) {
3627     if (i > 0) {
3628       x11 = x12;
3629       y11 = y12;
3630     }
3631     else {
3632       x11 = mouse_path[i][0];
3633       y11 = mouse_path[i][1];
3634     }
3635     x12 = mouse_path[i][0];
3636     y12 = mouse_path[i][1];
3637
3638     /* Perp. Distance from point to line */
3639     if (m2 != MAXSLOPE) {
3640       /* sqrt(m2 * m2 + 1); Only looking for change in sign.  Skip extra math .*/
3641       dist = (y12 - m2 * x12 - b2);
3642     }
3643     else {
3644       dist = x22 - x12;
3645     }
3646
3647     if (i == 0) {
3648       lastdist = dist;
3649     }
3650
3651     /* if dist changes sign, and intersect point in edge's Bound Box */
3652     if ((lastdist * dist) <= 0) {
3653       xdiff1 = (x12 - x11); /* Equation of line between last 2 points */
3654       if (xdiff1) {
3655         m1 = (y12 - y11) / xdiff1;
3656         b1 = ((x12 * y11) - (x11 * y12)) / xdiff1;
3657       }
3658       else {
3659         m1 = MAXSLOPE;
3660         b1 = x12;
3661       }
3662       x2max = max_ff(x21, x22) + 0.001f; /* prevent missed edges   */
3663       x2min = min_ff(x21, x22) - 0.001f; /* due to round off error */
3664       y2max = max_ff(y21, y22) + 0.001f;
3665       y2min = min_ff(y21, y22) - 0.001f;
3666
3667       /* Found an intersect,  calc intersect point */
3668       if (m1 == m2) { /* co-incident lines */
3669         /* cut at 50% of overlap area */
3670         x1max = max_ff(x11, x12);
3671         x1min = min_ff(x11, x12);
3672         xi = (min_ff(x2max, x1max) + max_ff(x2min, x1min)) / 2.0f;
3673
3674         y1max = max_ff(y11, y12);
3675         y1min = min_ff(y11, y12);
3676         yi = (min_ff(y2max, y1max) + max_ff(y2min, y1min)) / 2.0f;
3677       }
3678       else if (m2 == MAXSLOPE) {
3679         xi = x22;
3680         yi = m1 * x22 + b1;
3681       }
3682       else if (m1 == MAXSLOPE) {
3683         xi = x12;
3684         yi = m2 * x12 + b2;
3685       }
3686       else {
3687         xi = (b1 - b2) / (m2 - m1);
3688         yi = (b1 * m2 - m1 * b2) / (m2 - m1);
3689       }
3690
3691       /* Intersect inside bounding box of edge?*/
3692       if ((xi >= x2min) && (xi <= x2max) && (yi <= y2max) && (yi >= y2min)) {
3693         /* test for vertex intersect that may be 'close enough'*/
3694         if (mode != KNIFE_MULTICUT) {
3695           if (xi <= (x21 + threshold) && xi >= (x21 - threshold)) {
3696             if (yi <= (y21 + threshold) && yi >= (y21 - threshold)) {
3697               *isected = 1;
3698               perc = 0;
3699               break;
3700             }
3701           }
3702           if (xi <= (x22 + threshold) && xi >= (x22 - threshold)) {
3703             if (yi <= (y22 + threshold) && yi >= (y22 - threshold)) {
3704               *isected = 2;
3705               perc = 0;
3706               break;
3707             }
3708           }
3709         }
3710         if ((m2 <= 1.0f) && (m2 >= -1.0f)) {
3711           perc = (xi - x21) / (x22 - x21);
3712         }