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