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