Fix for missing id_lib_extern, assigning ID's
[blender-staging.git] / source / blender / editors / mesh / editmesh_knife.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2007 Blender Foundation.
19  * All rights reserved.
20  *
21  * 
22  * Contributor(s): Joseph Eagar, Joshua Leung, Howard Trickey,
23  *                 Campbell Barton
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/editors/mesh/editmesh_knife.c
29  *  \ingroup edmesh
30  *
31  * Interactive editmesh knife tool.
32  */
33
34 #ifdef _MSC_VER
35 #  define _USE_MATH_DEFINES
36 #endif
37
38 #include "MEM_guardedalloc.h"
39
40 #include "BLI_listbase.h"
41 #include "BLI_string.h"
42 #include "BLI_array.h"
43 #include "BLI_alloca.h"
44 #include "BLI_linklist.h"
45 #include "BLI_math.h"
46 #include "BLI_smallhash.h"
47 #include "BLI_memarena.h"
48
49 #include "BLT_translation.h"
50
51 #include "BKE_DerivedMesh.h"
52 #include "BKE_context.h"
53 #include "BKE_editmesh.h"
54 #include "BKE_editmesh_bvh.h"
55 #include "BKE_report.h"
56
57 #include "BIF_gl.h"
58 #include "BIF_glutil.h" /* for paint cursor */
59
60 #include "ED_screen.h"
61 #include "ED_space_api.h"
62 #include "ED_view3d.h"
63 #include "ED_mesh.h"
64
65 #include "WM_api.h"
66 #include "WM_types.h"
67
68 #include "DNA_object_types.h"
69
70 #include "UI_interface.h"
71 #include "UI_resources.h"
72
73 #include "RNA_access.h"
74 #include "RNA_define.h"
75
76 #include "mesh_intern.h"  /* own include */
77
78 #define KMAXDIST    10  /* max mouse distance from edge before not detecting it */
79
80 /* WARNING: knife float precision is fragile:
81  * be careful before making changes here see: (T43229, T42864, T42459, T41164).
82  */
83 #define KNIFE_FLT_EPS          0.00001f
84 #define KNIFE_FLT_EPS_SQUARED  (KNIFE_FLT_EPS * KNIFE_FLT_EPS)
85 #define KNIFE_FLT_EPSBIG       0.0005f
86
87 #define KNIFE_FLT_EPS_PX_VERT  0.5f
88 #define KNIFE_FLT_EPS_PX_EDGE  0.05f
89 #define KNIFE_FLT_EPS_PX_FACE  0.05f
90
91 typedef struct KnifeColors {
92         unsigned char line[3];
93         unsigned char edge[3];
94         unsigned char curpoint[3];
95         unsigned char curpoint_a[4];
96         unsigned char point[3];
97         unsigned char point_a[4];
98 } KnifeColors;
99
100 /* knifetool operator */
101 typedef struct KnifeVert {
102         BMVert *v; /* non-NULL if this is an original vert */
103         ListBase edges;
104         ListBase faces;
105
106         float co[3], cageco[3], sco[2]; /* sco is screen coordinates for cageco */
107         bool is_face, in_space;
108         bool is_cut;  /* along a cut created by user input (will draw too) */
109 } KnifeVert;
110
111 typedef struct Ref {
112         struct Ref *next, *prev;
113         void *ref;
114 } Ref;
115
116 typedef struct KnifeEdge {
117         KnifeVert *v1, *v2;
118         BMFace *basef; /* face to restrict face fill to */
119         ListBase faces;
120
121         BMEdge *e /* , *e_old */; /* non-NULL if this is an original edge */
122         bool is_cut;  /* along a cut created by user input (will draw too) */
123 } KnifeEdge;
124
125 typedef struct KnifeLineHit {
126         float hit[3], cagehit[3];
127         float schit[2];  /* screen coordinates for cagehit */
128         float l; /* lambda along cut line */
129         float perc; /* lambda along hit line */
130         float m; /* depth front-to-back */
131
132         /* Exactly one of kfe, v, or f should be non-NULL,
133          * saying whether cut line crosses and edge,
134          * is snapped to a vert, or is in the middle of some face. */
135         KnifeEdge *kfe;
136         KnifeVert *v;
137         BMFace *f;
138 } KnifeLineHit;
139
140 typedef struct KnifePosData {
141         float co[3];
142         float cage[3];
143
144         /* At most one of vert, edge, or bmface should be non-NULL,
145          * saying whether the point is snapped to a vertex, edge, or in a face.
146          * If none are set, this point is in space and is_space should be true. */
147         KnifeVert *vert;
148         KnifeEdge *edge;
149         BMFace *bmface;
150         bool is_space;
151
152         float mval[2]; /* mouse screen position (may be non-integral if snapped to something) */
153 } KnifePosData;
154
155 /* struct for properties used while drawing */
156 typedef struct KnifeTool_OpData {
157         ARegion *ar;        /* region that knifetool was activated in */
158         void *draw_handle;  /* for drawing preview loop */
159         ViewContext vc;     /* note: _don't_ use 'mval', instead use the one we define below */
160         float mval[2];      /* mouse value with snapping applied */
161         //bContext *C;
162
163         Scene *scene;
164         Object *ob;
165         BMEditMesh *em;
166
167         MemArena *arena;
168
169         GHash *origvertmap;
170         GHash *origedgemap;
171         GHash *kedgefacemap;
172         GHash *facetrimap;
173
174         BMBVHTree *bmbvh;
175
176         BLI_mempool *kverts;
177         BLI_mempool *kedges;
178
179         float vthresh;
180         float ethresh;
181
182         /* used for drag-cutting */
183         KnifeLineHit *linehits;
184         int totlinehit;
185
186         /* Data for mouse-position-derived data */
187         KnifePosData curr;  /* current point under the cursor */
188         KnifePosData prev;  /* last added cut (a line draws from the cursor to this) */
189         KnifePosData init;  /* the first point in the cut-list, used for closing the loop */
190
191         int totkedge, totkvert;
192
193         BLI_mempool *refs;
194
195         float projmat[4][4];
196         float projmat_inv[4][4];
197         /* vector along view z axis (object space, normalized) */
198         float proj_zaxis[3];
199
200         KnifeColors colors;
201
202         /* run by the UI or not */
203         bool is_interactive;
204
205         /* operatpr options */
206         bool cut_through;    /* preference, can be modified at runtime (that feature may go) */
207         bool only_select;    /* set on initialization */
208         bool select_result;  /* set on initialization */
209
210         bool is_ortho;
211         float ortho_extent;
212         float ortho_extent_center[3];
213
214         float clipsta, clipend;
215
216         enum {
217                 MODE_IDLE,
218                 MODE_DRAGGING,
219                 MODE_CONNECT,
220                 MODE_PANNING
221         } mode;
222         bool is_drag_hold;
223
224         int prevmode;
225         bool snap_midpoints;
226         bool ignore_edge_snapping;
227         bool ignore_vert_snapping;
228
229         /* use to check if we're currently dragging an angle snapped line */
230         bool is_angle_snapping;
231         bool angle_snapping;
232         float angle;
233
234         const float (*cagecos)[3];
235 } KnifeTool_OpData;
236
237 enum {
238         KNF_MODAL_CANCEL = 1,
239         KNF_MODAL_CONFIRM,
240         KNF_MODAL_MIDPOINT_ON,
241         KNF_MODAL_MIDPOINT_OFF,
242         KNF_MODAL_NEW_CUT,
243         KNF_MODEL_IGNORE_SNAP_ON,
244         KNF_MODEL_IGNORE_SNAP_OFF,
245         KNF_MODAL_ADD_CUT,
246         KNF_MODAL_ANGLE_SNAP_TOGGLE,
247         KNF_MODAL_CUT_THROUGH_TOGGLE,
248         KNF_MODAL_PANNING,
249         KNF_MODAL_ADD_CUT_CLOSED,
250 };
251
252
253 static ListBase *knife_get_face_kedges(KnifeTool_OpData *kcd, BMFace *f);
254
255 static void knife_input_ray_segment(KnifeTool_OpData *kcd, const float mval[2], const float ofs,
256                                     float r_origin[3], float r_dest[3]);
257
258 static bool knife_verts_edge_in_face(KnifeVert *v1, KnifeVert *v2, BMFace *f);
259
260 static void knifetool_free_bmbvh(KnifeTool_OpData *kcd);
261
262 static void knife_update_header(bContext *C, wmOperator *op, KnifeTool_OpData *kcd)
263 {
264         char header[UI_MAX_DRAW_STR];
265         char buf[UI_MAX_DRAW_STR];
266
267         char *p = buf;
268         int available_len = sizeof(buf);
269
270 #define WM_MODALKEY(_id) \
271         WM_modalkeymap_operator_items_to_string_buf(op->type, (_id), true, UI_MAX_SHORTCUT_STR, &available_len, &p)
272
273         BLI_snprintf(header, sizeof(header), IFACE_("%s: confirm, %s: cancel, "
274                                                     "%s: start/define cut, %s: close cut, %s: new cut, "
275                                                     "%s: midpoint snap (%s), %s: ignore snap (%s), "
276                                                     "%s: angle constraint (%s), %s: cut through (%s), "
277                                                     "%s: panning"),
278                      WM_MODALKEY(KNF_MODAL_CONFIRM), WM_MODALKEY(KNF_MODAL_CANCEL),
279                      WM_MODALKEY(KNF_MODAL_ADD_CUT), WM_MODALKEY(KNF_MODAL_ADD_CUT_CLOSED), WM_MODALKEY(KNF_MODAL_NEW_CUT),
280                      WM_MODALKEY(KNF_MODAL_MIDPOINT_ON), WM_bool_as_string(kcd->snap_midpoints),
281                      WM_MODALKEY(KNF_MODEL_IGNORE_SNAP_ON), WM_bool_as_string(kcd->ignore_edge_snapping),
282                      WM_MODALKEY(KNF_MODAL_ANGLE_SNAP_TOGGLE), WM_bool_as_string(kcd->angle_snapping),
283                      WM_MODALKEY(KNF_MODAL_CUT_THROUGH_TOGGLE), WM_bool_as_string(kcd->cut_through),
284                      WM_MODALKEY(KNF_MODAL_PANNING));
285
286 #undef WM_MODALKEY
287
288         ED_area_headerprint(CTX_wm_area(C), header);
289 }
290
291 static void knife_project_v2(const KnifeTool_OpData *kcd, const float co[3], float sco[2])
292 {
293         ED_view3d_project_float_v2_m4(kcd->ar, co, sco, (float (*)[4])kcd->projmat);
294 }
295
296 /* use when lambda is in screen-space */
297 static void knife_interp_v3_v3v3(
298         const KnifeTool_OpData *kcd,
299         float r_co[3], const float v1[3], const float v2[3], float lambda_ss)
300 {
301         if (kcd->is_ortho) {
302                 interp_v3_v3v3(r_co, v1, v2, lambda_ss);
303         }
304         else {
305                 /* transform into screen-space, interp, then transform back */
306                 float v1_ss[3], v2_ss[3];
307
308                 mul_v3_project_m4_v3(v1_ss, (float (*)[4])kcd->projmat, v1);
309                 mul_v3_project_m4_v3(v2_ss, (float (*)[4])kcd->projmat, v2);
310
311                 interp_v3_v3v3(r_co, v1_ss, v2_ss, lambda_ss);
312
313                 mul_project_m4_v3((float (*)[4])kcd->projmat_inv, r_co);
314         }
315 }
316
317 static void knife_pos_data_clear(KnifePosData *kpd)
318 {
319         zero_v3(kpd->co);
320         zero_v3(kpd->cage);
321         kpd->vert = NULL;
322         kpd->edge = NULL;
323         kpd->bmface = NULL;
324         zero_v2(kpd->mval);
325 }
326
327 static ListBase *knife_empty_list(KnifeTool_OpData *kcd)
328 {
329         ListBase *lst;
330
331         lst = BLI_memarena_alloc(kcd->arena, sizeof(ListBase));
332         BLI_listbase_clear(lst);
333         return lst;
334 }
335
336 static void knife_append_list(KnifeTool_OpData *kcd, ListBase *lst, void *elem)
337 {
338         Ref *ref;
339
340         ref = BLI_mempool_calloc(kcd->refs);
341         ref->ref = elem;
342         BLI_addtail(lst, ref);
343 }
344
345 static Ref *find_ref(ListBase *lb, void *ref)
346 {
347         Ref *ref1;
348
349         for (ref1 = lb->first; ref1; ref1 = ref1->next) {
350                 if (ref1->ref == ref)
351                         return ref1;
352         }
353
354         return NULL;
355 }
356
357 static void knife_append_list_no_dup(KnifeTool_OpData *kcd, ListBase *lst, void *elem)
358 {
359         if (!find_ref(lst, elem))
360                 knife_append_list(kcd, lst, elem);
361 }
362
363 static KnifeEdge *new_knife_edge(KnifeTool_OpData *kcd)
364 {
365         kcd->totkedge++;
366         return BLI_mempool_calloc(kcd->kedges);
367 }
368
369 static void knife_add_to_vert_edges(KnifeTool_OpData *kcd, KnifeEdge *kfe)
370 {
371         knife_append_list(kcd, &kfe->v1->edges, kfe);
372         knife_append_list(kcd, &kfe->v2->edges, kfe);
373 }
374
375 /* Add faces of an edge to a KnifeVert's faces list.  No checks for dups. */
376 static void knife_add_edge_faces_to_vert(KnifeTool_OpData *kcd, KnifeVert *kfv, BMEdge *e)
377 {
378         BMIter bmiter;
379         BMFace *f;
380
381         BM_ITER_ELEM (f, &bmiter, e, BM_FACES_OF_EDGE) {
382                 knife_append_list(kcd, &kfv->faces, f);
383         }
384 }
385
386 /* Find a face in common in the two faces lists.
387  * If more than one, return the first; if none, return NULL */
388 static BMFace *knife_find_common_face(ListBase *faces1, ListBase *faces2)
389 {
390         Ref *ref1, *ref2;
391
392         for (ref1 = faces1->first; ref1; ref1 = ref1->next) {
393                 for (ref2 = faces2->first; ref2; ref2 = ref2->next) {
394                         if (ref1->ref == ref2->ref)
395                                 return (BMFace *)(ref1->ref);
396                 }
397         }
398         return NULL;
399 }
400
401 static KnifeVert *new_knife_vert(KnifeTool_OpData *kcd, const float co[3], const float cageco[3])
402 {
403         KnifeVert *kfv = BLI_mempool_calloc(kcd->kverts);
404
405         kcd->totkvert++;
406
407         copy_v3_v3(kfv->co, co);
408         copy_v3_v3(kfv->cageco, cageco);
409
410         knife_project_v2(kcd, kfv->cageco, kfv->sco);
411
412         return kfv;
413 }
414
415 /* get a KnifeVert wrapper for an existing BMVert */
416 static KnifeVert *get_bm_knife_vert(KnifeTool_OpData *kcd, BMVert *v)
417 {
418         KnifeVert *kfv = BLI_ghash_lookup(kcd->origvertmap, v);
419         const float *cageco;
420
421         if (!kfv) {
422                 BMIter bmiter;
423                 BMFace *f;
424
425                 if (BM_elem_index_get(v) >= 0)
426                         cageco = kcd->cagecos[BM_elem_index_get(v)];
427                 else
428                         cageco = v->co;
429                 kfv = new_knife_vert(kcd, v->co, cageco);
430                 kfv->v = v;
431                 BLI_ghash_insert(kcd->origvertmap, v, kfv);
432                 BM_ITER_ELEM (f, &bmiter, v, BM_FACES_OF_VERT) {
433                         knife_append_list(kcd, &kfv->faces, f);
434                 }
435         }
436
437         return kfv;
438 }
439
440 /* get a KnifeEdge wrapper for an existing BMEdge */
441 static KnifeEdge *get_bm_knife_edge(KnifeTool_OpData *kcd, BMEdge *e)
442 {
443         KnifeEdge *kfe = BLI_ghash_lookup(kcd->origedgemap, e);
444         if (!kfe) {
445                 BMIter bmiter;
446                 BMFace *f;
447
448                 kfe = new_knife_edge(kcd);
449                 kfe->e = e;
450                 kfe->v1 = get_bm_knife_vert(kcd, e->v1);
451                 kfe->v2 = get_bm_knife_vert(kcd, e->v2);
452
453                 knife_add_to_vert_edges(kcd, kfe);
454
455                 BLI_ghash_insert(kcd->origedgemap, e, kfe);
456
457                 BM_ITER_ELEM (f, &bmiter, e, BM_FACES_OF_EDGE) {
458                         knife_append_list(kcd, &kfe->faces, f);
459                 }
460         }
461
462         return kfe;
463 }
464
465 /* Record the index in kcd->em->looptris of first looptri triple for a given face,
466  * given an index for some triple in that array.
467  * This assumes that all of the triangles for a given face are contiguous
468  * in that array (as they are by the current tesselation routines).
469  * Actually store index + 1 in the hash, because 0 looks like "no entry"
470  * to hash lookup routine; will reverse this in the get routine.
471  * Doing this lazily rather than all at once for all faces.
472  */
473 static void set_lowest_face_tri(KnifeTool_OpData *kcd, BMFace *f, int index)
474 {
475         int i;
476
477         if (BLI_ghash_lookup(kcd->facetrimap, f))
478                 return;
479
480         BLI_assert(index >= 0 && index < kcd->em->tottri);
481         BLI_assert(kcd->em->looptris[index][0]->f == f);
482         for (i = index - 1; i >= 0; i--) {
483                 if (kcd->em->looptris[i][0]->f != f) {
484                         i++;
485                         break;
486                 }
487         }
488         if (i == -1)
489                 i++;
490
491         BLI_ghash_insert(kcd->facetrimap, f, SET_INT_IN_POINTER(i + 1));
492 }
493
494 /* This should only be called for faces that have had a lowest face tri set by previous function */
495 static int get_lowest_face_tri(KnifeTool_OpData *kcd, BMFace *f)
496 {
497         int ans;
498
499         ans = GET_INT_FROM_POINTER(BLI_ghash_lookup(kcd->facetrimap, f));
500         BLI_assert(ans != 0);
501         return ans - 1;
502 }
503
504 /* User has just clicked for first time or first time after a restart (E key).
505  * Copy the current position data into prev. */
506 static void knife_start_cut(KnifeTool_OpData *kcd)
507 {
508         kcd->prev = kcd->curr;
509         kcd->curr.is_space = 0; /*TODO: why do we do this? */
510
511         if (kcd->prev.vert == NULL && kcd->prev.edge == NULL) {
512                 float origin[3], origin_ofs[3];
513                 float ofs_local[3];
514
515                 negate_v3_v3(ofs_local, kcd->vc.rv3d->ofs);
516                 invert_m4_m4(kcd->ob->imat, kcd->ob->obmat);
517                 mul_m4_v3(kcd->ob->imat, ofs_local);
518
519                 knife_input_ray_segment(kcd, kcd->curr.mval, 1.0f, origin, origin_ofs);
520
521                 if (!isect_line_plane_v3(kcd->prev.cage, origin, origin_ofs, ofs_local, kcd->proj_zaxis)) {
522                         zero_v3(kcd->prev.cage);
523                 }
524
525                 copy_v3_v3(kcd->prev.co, kcd->prev.cage); /*TODO: do we need this? */
526                 copy_v3_v3(kcd->curr.cage, kcd->prev.cage);
527                 copy_v3_v3(kcd->curr.co, kcd->prev.co);
528         }
529 }
530
531 static ListBase *knife_get_face_kedges(KnifeTool_OpData *kcd, BMFace *f)
532 {
533         ListBase *lst = BLI_ghash_lookup(kcd->kedgefacemap, f);
534
535         if (!lst) {
536                 BMIter bmiter;
537                 BMEdge *e;
538
539                 lst = knife_empty_list(kcd);
540
541                 BM_ITER_ELEM (e, &bmiter, f, BM_EDGES_OF_FACE) {
542                         knife_append_list(kcd, lst, get_bm_knife_edge(kcd, e));
543                 }
544
545                 BLI_ghash_insert(kcd->kedgefacemap, f, lst);
546         }
547
548         return lst;
549 }
550
551 static void knife_edge_append_face(KnifeTool_OpData *kcd, KnifeEdge *kfe, BMFace *f)
552 {
553         knife_append_list(kcd, knife_get_face_kedges(kcd, f), kfe);
554         knife_append_list(kcd, &kfe->faces, f);
555 }
556
557 static KnifeVert *knife_split_edge(
558         KnifeTool_OpData *kcd, KnifeEdge *kfe,
559         const float co[3], const float cageco[3],
560         KnifeEdge **r_kfe)
561 {
562         KnifeEdge *newkfe = new_knife_edge(kcd);
563         Ref *ref;
564         BMFace *f;
565
566         newkfe->v1 = kfe->v1;
567         newkfe->v2 = new_knife_vert(kcd, co, cageco);
568         newkfe->v2->is_cut = true;
569         if (kfe->e) {
570                 knife_add_edge_faces_to_vert(kcd, newkfe->v2, kfe->e);
571         }
572         else {
573                 /* kfe cuts across an existing face.
574                  * If v1 and v2 are in multiple faces together (e.g., if they
575                  * are in doubled polys) then this arbitrarily chooses one of them */
576                 f = knife_find_common_face(&kfe->v1->faces, &kfe->v2->faces);
577                 if (f)
578                         knife_append_list(kcd, &newkfe->v2->faces, f);
579         }
580         newkfe->basef = kfe->basef;
581
582         ref = find_ref(&kfe->v1->edges, kfe);
583         BLI_remlink(&kfe->v1->edges, ref);
584
585         kfe->v1 = newkfe->v2;
586         BLI_addtail(&kfe->v1->edges, ref);
587
588         for (ref = kfe->faces.first; ref; ref = ref->next)
589                 knife_edge_append_face(kcd, newkfe, ref->ref);
590
591         knife_add_to_vert_edges(kcd, newkfe);
592
593         newkfe->is_cut = kfe->is_cut;
594         newkfe->e = kfe->e;
595
596         *r_kfe = newkfe;
597
598         return newkfe->v2;
599 }
600
601 static void linehit_to_knifepos(KnifePosData *kpos, KnifeLineHit *lh)
602 {
603         kpos->bmface = lh->f;
604         kpos->vert = lh->v;
605         kpos->edge = lh->kfe;
606         copy_v3_v3(kpos->cage, lh->cagehit);
607         copy_v3_v3(kpos->co, lh->hit);
608         copy_v2_v2(kpos->mval, lh->schit);
609 }
610
611 /* primary key: lambda along cut
612  * secondary key: lambda along depth
613  * tertiary key: pointer comparisons of verts if both snapped to verts
614  */
615 static int linehit_compare(const void *vlh1, const void *vlh2)
616 {
617         const KnifeLineHit *lh1 = vlh1;
618         const KnifeLineHit *lh2 = vlh2;
619
620         if      (lh1->l < lh2->l) return -1;
621         else if (lh1->l > lh2->l) return  1;
622         else {
623                 if      (lh1->m < lh2->m) return -1;
624                 else if (lh1->m > lh2->m) return  1;
625                 else {
626                         if      (lh1->v < lh2->v) return -1;
627                         else if (lh1->v > lh2->v) return  1;
628                         else return 0;
629                 }
630         }
631 }
632
633 /*
634  * Sort linehits by distance along cut line, and secondarily from
635  * front to back (from eye), and tertiarily by snap vertex,
636  * and remove any duplicates.
637  */
638 static void prepare_linehits_for_cut(KnifeTool_OpData *kcd)
639 {
640         KnifeLineHit *linehits, *lhi, *lhj;
641         int i, j, n;
642         bool is_double = false;
643
644         n = kcd->totlinehit;
645         linehits = kcd->linehits;
646         if (n == 0)
647                 return;
648
649         qsort(linehits, n, sizeof(KnifeLineHit), linehit_compare);
650
651         /* Remove any edge hits that are preceded or followed
652          * by a vertex hit that is very near. Mark such edge hits using
653          * l == -1 and then do another pass to actually remove.
654          * Also remove all but one of a series of vertex hits for the same vertex. */
655         for (i = 0; i < n; i++) {
656                 lhi = &linehits[i];
657                 if (lhi->v) {
658                         for (j = i - 1; j >= 0; j--) {
659                                 lhj = &linehits[j];
660                                 if (!lhj->kfe ||
661                                     fabsf(lhi->l - lhj->l) > KNIFE_FLT_EPSBIG ||
662                                     fabsf(lhi->m - lhj->m) > KNIFE_FLT_EPSBIG)
663                                 {
664                                         break;
665                                 }
666
667                                 if (lhi->kfe == lhj->kfe) {
668                                         lhj->l = -1.0f;
669                                         is_double = true;
670                                 }
671                         }
672                         for (j = i + 1; j < n; j++) {
673                                 lhj = &linehits[j];
674                                 if (fabsf(lhi->l - lhj->l) > KNIFE_FLT_EPSBIG ||
675                                     fabsf(lhi->m - lhj->m) > KNIFE_FLT_EPSBIG)
676                                 {
677                                         break;
678                                 }
679                                 if ((lhj->kfe && (lhi->kfe == lhj->kfe)) ||
680                                     (lhi->v == lhj->v))
681                                 {
682                                         lhj->l = -1.0f;
683                                         is_double = true;
684                                 }
685                         }
686                 }
687         }
688
689         if (is_double) {
690                 /* delete-in-place loop: copying from pos j to pos i+1 */
691                 i = 0;
692                 j = 1;
693                 while (j < n) {
694                         lhi = &linehits[i];
695                         lhj = &linehits[j];
696                         if (lhj->l == -1.0f) {
697                                 j++; /* skip copying this one */
698                         }
699                         else {
700                                 /* copy unless a no-op */
701                                 if (lhi->l == -1.0f) {
702                                         /* could happen if linehits[0] is being deleted */
703                                         memcpy(&linehits[i], &linehits[j], sizeof(KnifeLineHit));
704                                 }
705                                 else {
706                                         if (i + 1 != j)
707                                                 memcpy(&linehits[i + 1], &linehits[j], sizeof(KnifeLineHit));
708                                         i++;
709                                 }
710                                 j++;
711                         }
712                 }
713                 kcd->totlinehit = i + 1;
714         }
715 }
716
717 /* Add hit to list of hits in facehits[f], where facehits is a map, if not already there */
718 static void add_hit_to_facehits(KnifeTool_OpData *kcd, GHash *facehits, BMFace *f, KnifeLineHit *hit)
719 {
720         ListBase *lst = BLI_ghash_lookup(facehits, f);
721
722         if (!lst) {
723                 lst = knife_empty_list(kcd);
724                 BLI_ghash_insert(facehits, f, lst);
725         }
726         knife_append_list_no_dup(kcd, lst, hit);
727 }
728
729 /**
730  * special purpose function, if the linehit is connected to a real edge/vert
731  * return true if \a co is outside the face.
732  */
733 static bool knife_add_single_cut__is_linehit_outside_face(BMFace *f, const KnifeLineHit *lh, const float co[3])
734 {
735
736         if (lh->v && lh->v->v) {
737                 BMLoop *l;  /* side-of-loop */
738                 if ((l = BM_face_vert_share_loop(f, lh->v->v)) &&
739                     (BM_loop_point_side_of_loop_test(l, co) < 0.0f))
740                 {
741                         return true;
742                 }
743         }
744         else if ((lh->kfe && lh->kfe->e)) {
745                 BMLoop *l;  /* side-of-edge */
746                 if ((l = BM_face_edge_share_loop(f, lh->kfe->e)) &&
747                     (BM_loop_point_side_of_edge_test(l, co) < 0.0f))
748                 {
749                         return true;
750                 }
751         }
752
753         return false;
754 }
755
756
757 static void knife_add_single_cut(KnifeTool_OpData *kcd, KnifeLineHit *lh1, KnifeLineHit *lh2, BMFace *f)
758 {
759         KnifeEdge *kfe, *kfe2;
760         BMEdge *e_base;
761
762         if ((lh1->v && lh1->v == lh2->v) ||
763             (lh1->kfe && lh1->kfe == lh2->kfe))
764         {
765                 return;
766         }
767
768         /* if the cut is on an edge, just tag that its a cut and return */
769         if ((lh1->v && lh2->v) &&
770             (lh1->v->v && lh2->v && lh2->v->v) &&
771             (e_base = BM_edge_exists(lh1->v->v, lh2->v->v)))
772         {
773                 kfe = get_bm_knife_edge(kcd, e_base);
774                 kfe->is_cut = true;
775                 kfe->e = e_base;
776                 return;
777         }
778         else {
779                 if (knife_add_single_cut__is_linehit_outside_face(f, lh1, lh2->hit) ||
780                     knife_add_single_cut__is_linehit_outside_face(f, lh2, lh1->hit))
781                 {
782                         return;
783                 }
784         }
785
786
787         /* Check if edge actually lies within face (might not, if this face is concave) */
788         if ((lh1->v && !lh1->kfe) && (lh2->v && !lh2->kfe)) {
789                 if (!knife_verts_edge_in_face(lh1->v, lh2->v, f)) {
790                         return;
791                 }
792         }
793
794         kfe = new_knife_edge(kcd);
795         kfe->is_cut = true;
796         kfe->basef = f;
797
798         if (lh1->v) {
799                 kfe->v1 = lh1->v;
800         }
801         else if (lh1->kfe) {
802                 kfe->v1 = knife_split_edge(kcd, lh1->kfe, lh1->hit, lh1->cagehit, &kfe2);
803                 lh1->v = kfe->v1;  /* record the KnifeVert for this hit  */
804         }
805         else {
806                 BLI_assert(lh1->f);
807                 kfe->v1 = new_knife_vert(kcd, lh1->hit, lh1->cagehit);
808                 kfe->v1->is_cut = true;
809                 kfe->v1->is_face = true;
810                 knife_append_list(kcd, &kfe->v1->faces, lh1->f);
811                 lh1->v = kfe->v1;  /* record the KnifeVert for this hit */
812         }
813
814         if (lh2->v) {
815                 kfe->v2 = lh2->v;
816         }
817         else if (lh2->kfe) {
818                 kfe->v2 = knife_split_edge(kcd, lh2->kfe, lh2->hit, lh2->cagehit, &kfe2);
819                 lh2->v = kfe->v2;  /* future uses of lh2 won't split again */
820         }
821         else {
822                 BLI_assert(lh2->f);
823                 kfe->v2 = new_knife_vert(kcd, lh2->hit, lh2->cagehit);
824                 kfe->v2->is_cut = true;
825                 kfe->v2->is_face = true;
826                 knife_append_list(kcd, &kfe->v2->faces, lh2->f);
827                 lh2->v = kfe->v2;  /* record the KnifeVert for this hit */
828         }
829
830         knife_add_to_vert_edges(kcd, kfe);
831
832         /* TODO: check if this is ever needed */
833         if (kfe->basef && !find_ref(&kfe->faces, kfe->basef))
834                 knife_edge_append_face(kcd, kfe, kfe->basef);
835
836 }
837
838 /* Given a list of KnifeLineHits for one face, sorted by l
839  * and then by m, make the required KnifeVerts and
840  * KnifeEdges.
841  */
842 static void knife_cut_face(KnifeTool_OpData *kcd, BMFace *f, ListBase *hits)
843 {
844         Ref *r;
845
846         if (BLI_listbase_count_ex(hits, 2) != 2)
847                 return;
848
849         for (r = hits->first; r->next; r = r->next) {
850                 knife_add_single_cut(kcd, r->ref, r->next->ref, f);
851         }
852 }
853
854 /* User has just left-clicked after the first time.
855  * Add all knife cuts implied by line from prev to curr.
856  * If that line crossed edges then kcd->linehits will be non-NULL.
857  * Make all of the KnifeVerts and KnifeEdges implied by this cut.
858  */
859 static void knife_add_cut(KnifeTool_OpData *kcd)
860 {
861         int i;
862         KnifeLineHit *lh;
863         GHash *facehits;
864         BMFace *f;
865         Ref *r;
866         GHashIterator giter;
867         ListBase *lst;
868
869         prepare_linehits_for_cut(kcd);
870         if (kcd->totlinehit == 0) {
871                 if (kcd->is_drag_hold == false) {
872                         kcd->prev = kcd->curr;
873                 }
874                 return;
875         }
876
877         /* make facehits: map face -> list of linehits touching it */
878         facehits = BLI_ghash_ptr_new("knife facehits");
879         for (i = 0; i < kcd->totlinehit; i++) {
880                 lh = &kcd->linehits[i];
881                 if (lh->f) {
882                         add_hit_to_facehits(kcd, facehits, lh->f, lh);
883                 }
884                 if (lh->v) {
885                         for (r = lh->v->faces.first; r; r = r->next) {
886                                 add_hit_to_facehits(kcd, facehits, r->ref, lh);
887                         }
888                 }
889                 if (lh->kfe) {
890                         for (r = lh->kfe->faces.first; r; r = r->next) {
891                                 add_hit_to_facehits(kcd, facehits, r->ref, lh);
892                         }
893                 }
894         }
895
896         /* Note: as following loop progresses, the 'v' fields of
897          * the linehits will be filled in (as edges are split or
898          * in-face verts are made), so it may be true that both
899          * the v and the kfe or f fields will be non-NULL. */
900         GHASH_ITER (giter, facehits) {
901                 f = (BMFace *)BLI_ghashIterator_getKey(&giter);
902                 lst = (ListBase *)BLI_ghashIterator_getValue(&giter);
903                 knife_cut_face(kcd, f, lst);
904         }
905
906         /* set up for next cut */
907         kcd->prev = kcd->curr;
908
909
910         if (kcd->prev.bmface) {
911                 KnifeLineHit *lh;
912                 /* was "in face" but now we have a KnifeVert it is snapped to */
913                 lh = &kcd->linehits[kcd->totlinehit - 1];
914                 kcd->prev.vert = lh->v;
915                 kcd->prev.bmface = NULL;
916         }
917
918         if (kcd->is_drag_hold) {
919                 lh = &kcd->linehits[kcd->totlinehit - 1];
920                 linehit_to_knifepos(&kcd->prev, lh);
921         }
922
923         BLI_ghash_free(facehits, NULL, NULL);
924         MEM_freeN(kcd->linehits);
925         kcd->linehits = NULL;
926         kcd->totlinehit = 0;
927 }
928
929 static void knife_finish_cut(KnifeTool_OpData *kcd)
930 {
931         if (kcd->linehits) {
932                 MEM_freeN(kcd->linehits);
933                 kcd->linehits = NULL;
934                 kcd->totlinehit = 0;
935         }
936 }
937
938 static void knifetool_draw_angle_snapping(const KnifeTool_OpData *kcd)
939 {
940         float v1[3], v2[3];
941         float planes[4][4];
942
943         planes_from_projmat(
944                 (float (*)[4])kcd->projmat,
945                 planes[2], planes[0], planes[3], planes[1], NULL, NULL);
946
947         /* ray-cast all planes */
948         {
949                 float ray_dir[3];
950                 float ray_hit_best[2][3] = {{UNPACK3(kcd->prev.cage)}, {UNPACK3(kcd->curr.cage)}};
951                 float lambda_best[2] = {-FLT_MAX, FLT_MAX};
952                 int i;
953
954                 /* we (sometimes) need the lines to be at the same depth before projecting */
955 #if 0
956                 sub_v3_v3v3(ray_dir, kcd->curr.cage, kcd->prev.cage);
957 #else
958                 {
959                         float curr_cage_adjust[3];
960                         float co_depth[3];
961
962                         copy_v3_v3(co_depth, kcd->prev.cage);
963                         mul_m4_v3(kcd->ob->obmat, co_depth);
964                         ED_view3d_win_to_3d(kcd->ar, co_depth, kcd->curr.mval, curr_cage_adjust);
965                         mul_m4_v3(kcd->ob->imat, curr_cage_adjust);
966
967                         sub_v3_v3v3(ray_dir, curr_cage_adjust, kcd->prev.cage);
968                 }
969 #endif
970
971                 for (i = 0; i < 4; i++) {
972                         float ray_hit[3];
973                         float lambda_test;
974                         if (isect_ray_plane_v3(kcd->prev.cage, ray_dir, planes[i], &lambda_test, false)) {
975                                 madd_v3_v3v3fl(ray_hit, kcd->prev.cage, ray_dir, lambda_test);
976                                 if (lambda_test < 0.0f) {
977                                         if (lambda_test > lambda_best[0]) {
978                                                 copy_v3_v3(ray_hit_best[0], ray_hit);
979                                                 lambda_best[0] = lambda_test;
980                                         }
981                                 }
982                                 else {
983                                         if (lambda_test < lambda_best[1]) {
984                                                 copy_v3_v3(ray_hit_best[1], ray_hit);
985                                                 lambda_best[1] = lambda_test;
986                                         }
987                                 }
988                         }
989                 }
990
991                 copy_v3_v3(v1, ray_hit_best[0]);
992                 copy_v3_v3(v2, ray_hit_best[1]);
993         }
994
995         UI_ThemeColor(TH_TRANSFORM);
996         glLineWidth(2.0);
997         glBegin(GL_LINES);
998         glVertex3fv(v1);
999         glVertex3fv(v2);
1000         glEnd();
1001 }
1002
1003 static void knife_init_colors(KnifeColors *colors)
1004 {
1005         /* possible BMESH_TODO: add explicit themes or calculate these by
1006          * figuring out contrasting colors with grid / edges / verts
1007          * a la UI_make_axis_color */
1008         UI_GetThemeColor3ubv(TH_NURB_VLINE, colors->line);
1009         UI_GetThemeColor3ubv(TH_NURB_ULINE, colors->edge);
1010         UI_GetThemeColor3ubv(TH_HANDLE_SEL_VECT, colors->curpoint);
1011         UI_GetThemeColor3ubv(TH_HANDLE_SEL_VECT, colors->curpoint_a);
1012         colors->curpoint_a[3] = 102;
1013         UI_GetThemeColor3ubv(TH_ACTIVE_SPLINE, colors->point);
1014         UI_GetThemeColor3ubv(TH_ACTIVE_SPLINE, colors->point_a);
1015         colors->point_a[3] = 102;
1016 }
1017
1018 /* modal loop selection drawing callback */
1019 static void knifetool_draw(const bContext *C, ARegion *UNUSED(ar), void *arg)
1020 {
1021         View3D *v3d = CTX_wm_view3d(C);
1022         const KnifeTool_OpData *kcd = arg;
1023
1024         if (v3d->zbuf) glDisable(GL_DEPTH_TEST);
1025
1026         glPolygonOffset(1.0f, 1.0f);
1027
1028         glPushMatrix();
1029         glMultMatrixf(kcd->ob->obmat);
1030
1031         if (kcd->mode == MODE_DRAGGING) {
1032                 if (kcd->is_angle_snapping)
1033                         knifetool_draw_angle_snapping(kcd);
1034
1035                 glColor3ubv(kcd->colors.line);
1036                 
1037                 glLineWidth(2.0);
1038
1039                 glBegin(GL_LINES);
1040                 glVertex3fv(kcd->prev.cage);
1041                 glVertex3fv(kcd->curr.cage);
1042                 glEnd();
1043
1044                 glLineWidth(1.0);
1045         }
1046
1047         if (kcd->prev.vert) {
1048                 glColor3ubv(kcd->colors.point);
1049                 glPointSize(11);
1050
1051                 glBegin(GL_POINTS);
1052                 glVertex3fv(kcd->prev.cage);
1053                 glEnd();
1054         }
1055
1056         if (kcd->prev.bmface) {
1057                 glColor3ubv(kcd->colors.curpoint);
1058                 glPointSize(9);
1059
1060                 glBegin(GL_POINTS);
1061                 glVertex3fv(kcd->prev.cage);
1062                 glEnd();
1063         }
1064
1065         if (kcd->curr.edge) {
1066                 glColor3ubv(kcd->colors.edge);
1067                 glLineWidth(2.0);
1068
1069                 glBegin(GL_LINES);
1070                 glVertex3fv(kcd->curr.edge->v1->cageco);
1071                 glVertex3fv(kcd->curr.edge->v2->cageco);
1072                 glEnd();
1073
1074                 glLineWidth(1.0);
1075         }
1076         else if (kcd->curr.vert) {
1077                 glColor3ubv(kcd->colors.point);
1078                 glPointSize(11);
1079
1080                 glBegin(GL_POINTS);
1081                 glVertex3fv(kcd->curr.cage);
1082                 glEnd();
1083         }
1084
1085         if (kcd->curr.bmface) {
1086                 glColor3ubv(kcd->colors.curpoint);
1087                 glPointSize(9);
1088
1089                 glBegin(GL_POINTS);
1090                 glVertex3fv(kcd->curr.cage);
1091                 glEnd();
1092         }
1093
1094         if (kcd->totlinehit > 0) {
1095                 KnifeLineHit *lh;
1096                 int i;
1097
1098                 glEnable(GL_BLEND);
1099                 glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
1100
1101                 /* draw any snapped verts first */
1102                 glColor4ubv(kcd->colors.point_a);
1103                 glPointSize(11);
1104                 glBegin(GL_POINTS);
1105                 lh = kcd->linehits;
1106                 for (i = 0; i < kcd->totlinehit; i++, lh++) {
1107                         if (lh->v)
1108                                 glVertex3fv(lh->cagehit);
1109                 }
1110                 glEnd();
1111
1112                 /* now draw the rest */
1113                 glColor4ubv(kcd->colors.curpoint_a);
1114                 glPointSize(7);
1115                 glBegin(GL_POINTS);
1116                 lh = kcd->linehits;
1117                 for (i = 0; i < kcd->totlinehit; i++, lh++) {
1118                         if (!lh->v)
1119                                 glVertex3fv(lh->cagehit);
1120                 }
1121                 glEnd();
1122                 glDisable(GL_BLEND);
1123         }
1124
1125         if (kcd->totkedge > 0) {
1126                 BLI_mempool_iter iter;
1127                 KnifeEdge *kfe;
1128
1129                 glLineWidth(1.0);
1130                 glBegin(GL_LINES);
1131
1132                 BLI_mempool_iternew(kcd->kedges, &iter);
1133                 for (kfe = BLI_mempool_iterstep(&iter); kfe; kfe = BLI_mempool_iterstep(&iter)) {
1134                         if (!kfe->is_cut)
1135                                 continue;
1136
1137                         glColor3ubv(kcd->colors.line);
1138
1139                         glVertex3fv(kfe->v1->cageco);
1140                         glVertex3fv(kfe->v2->cageco);
1141                 }
1142
1143                 glEnd();
1144                 glLineWidth(1.0);
1145         }
1146
1147         if (kcd->totkvert > 0) {
1148                 BLI_mempool_iter iter;
1149                 KnifeVert *kfv;
1150
1151                 glPointSize(5.0);
1152
1153                 glBegin(GL_POINTS);
1154                 BLI_mempool_iternew(kcd->kverts, &iter);
1155                 for (kfv = BLI_mempool_iterstep(&iter); kfv; kfv = BLI_mempool_iterstep(&iter)) {
1156                         if (!kfv->is_cut)
1157                                 continue;
1158
1159                         glColor3ubv(kcd->colors.point);
1160
1161                         glVertex3fv(kfv->cageco);
1162                 }
1163
1164                 glEnd();
1165         }
1166
1167         glPopMatrix();
1168
1169         if (v3d->zbuf) glEnable(GL_DEPTH_TEST);
1170 }
1171
1172 /**
1173  * Find intersection of v1-v2 with face f.
1174  * Only take intersections that are at least \a face_tol_sq (in screen space) away
1175  * from other intersection elements.
1176  * If v1-v2 is coplanar with f, call that "no intersection though
1177  * it really means "infinite number of intersections".
1178  * In such a case we should have gotten hits on edges or verts of the face.
1179  */
1180 static bool knife_ray_intersect_face(
1181         KnifeTool_OpData *kcd,
1182         const float s[2], const float v1[3], const float v2[3],
1183         BMFace *f, const float face_tol_sq,
1184         float hit_co[3], float hit_cageco[3])
1185 {
1186         int tottri, tri_i;
1187         float raydir[3];
1188         float tri_norm[3], tri_plane[4];
1189         float se1[2], se2[2];
1190         float d, lambda;
1191         BMLoop **tri;
1192         ListBase *lst;
1193         Ref *ref;
1194         KnifeEdge *kfe;
1195
1196         sub_v3_v3v3(raydir, v2, v1);
1197         normalize_v3(raydir);
1198         tri_i = get_lowest_face_tri(kcd, f);
1199         tottri = kcd->em->tottri;
1200         BLI_assert(tri_i >= 0 && tri_i < tottri);
1201
1202         for (; tri_i < tottri; tri_i++) {
1203                 const float *lv1, *lv2, *lv3;
1204
1205                 tri = kcd->em->looptris[tri_i];
1206                 if (tri[0]->f != f)
1207                         break;
1208                 lv1 = kcd->cagecos[BM_elem_index_get(tri[0]->v)];
1209                 lv2 = kcd->cagecos[BM_elem_index_get(tri[1]->v)];
1210                 lv3 = kcd->cagecos[BM_elem_index_get(tri[2]->v)];
1211                 /* using epsilon test in case ray is directly through an internal
1212                  * tesselation edge and might not hit either tesselation tri with
1213                  * an exact test;
1214                  * we will exclude hits near real edges by a later test */
1215                 if (isect_ray_tri_epsilon_v3(v1, raydir, lv1, lv2, lv3, &lambda, NULL, KNIFE_FLT_EPS)) {
1216                         /* check if line coplanar with tri */
1217                         normal_tri_v3(tri_norm, lv1, lv2, lv3);
1218                         plane_from_point_normal_v3(tri_plane, lv1, tri_norm);
1219                         if ((dist_squared_to_plane_v3(v1, tri_plane) < KNIFE_FLT_EPS) &&
1220                             (dist_squared_to_plane_v3(v2, tri_plane) < KNIFE_FLT_EPS))
1221                         {
1222                                 return false;
1223                         }
1224                         copy_v3_v3(hit_cageco, v1);
1225                         madd_v3_v3fl(hit_cageco, raydir, lambda);
1226                         /* Now check that far enough away from verts and edges */
1227                         lst = knife_get_face_kedges(kcd, f);
1228                         for (ref = lst->first; ref; ref = ref->next) {
1229                                 kfe = ref->ref;
1230                                 knife_project_v2(kcd, kfe->v1->cageco, se1);
1231                                 knife_project_v2(kcd, kfe->v2->cageco, se2);
1232                                 d = dist_squared_to_line_segment_v2(s, se1, se2);
1233                                 if (d < face_tol_sq) {
1234                                         return false;
1235                                 }
1236                         }
1237
1238                         transform_point_by_tri_v3(
1239                                 hit_co, hit_cageco,
1240                                 tri[0]->v->co, tri[1]->v->co, tri[2]->v->co,
1241                                 lv1, lv2, lv3);
1242                         return true;
1243                 }
1244         }
1245         return false;
1246 }
1247
1248 /**
1249  * Calculate the center and maximum excursion of mesh.
1250  */
1251 static void calc_ortho_extent(KnifeTool_OpData *kcd)
1252 {
1253         BMIter iter;
1254         BMVert *v;
1255         BMesh *bm = kcd->em->bm;
1256         float min[3], max[3];
1257
1258         INIT_MINMAX(min, max);
1259
1260         if (kcd->cagecos) {
1261                 minmax_v3v3_v3_array(min, max, kcd->cagecos, bm->totvert);
1262         }
1263         else {
1264                 BM_ITER_MESH (v, &iter, bm, BM_VERTS_OF_MESH) {
1265                         minmax_v3v3_v3(min, max, v->co);
1266                 }
1267         }
1268
1269         kcd->ortho_extent = len_v3v3(min, max) / 2;
1270         mid_v3_v3v3(kcd->ortho_extent_center, min, max);
1271 }
1272
1273 static BMElem *bm_elem_from_knife_vert(KnifeVert *kfv, KnifeEdge **r_kfe)
1274 {
1275         BMElem *ele_test;
1276         KnifeEdge *kfe = NULL;
1277
1278         /* vert? */
1279         ele_test = (BMElem *)kfv->v;
1280
1281         if (r_kfe || ele_test == NULL) {
1282                 if (kfv->v == NULL) {
1283                         Ref *ref;
1284                         for (ref = kfv->edges.first; ref; ref = ref->next) {
1285                                 kfe = ref->ref;
1286                                 if (kfe->e) {
1287                                         if (r_kfe) {
1288                                                 *r_kfe = kfe;
1289                                         }
1290                                         break;
1291                                 }
1292                         }
1293                 }
1294         }
1295
1296         /* edge? */
1297         if (ele_test == NULL) {
1298                 if (kfe) {
1299                         ele_test = (BMElem *)kfe->e;
1300                 }
1301         }
1302
1303         /* face? */
1304         if (ele_test == NULL) {
1305                 if (BLI_listbase_is_single(&kfe->faces)) {
1306                         ele_test = ((Ref *)kfe->faces.first)->ref;
1307                 }
1308         }
1309
1310         return ele_test;
1311 }
1312
1313 static BMElem *bm_elem_from_knife_edge(KnifeEdge *kfe)
1314 {
1315         BMElem *ele_test;
1316
1317         ele_test = (BMElem *)kfe->e;
1318
1319         if (ele_test == NULL) {
1320                 ele_test = (BMElem *)kfe->basef;
1321         }
1322
1323         return ele_test;
1324 }
1325
1326 /* Do edges e1 and e2 go between exactly the same coordinates? */
1327 static bool coinciding_edges(BMEdge *e1, BMEdge *e2)
1328 {
1329         const float *co11, *co12, *co21, *co22;
1330
1331         co11 = e1->v1->co;
1332         co12 = e1->v2->co;
1333         co21 = e2->v1->co;
1334         co22 = e2->v2->co;
1335         if ((equals_v3v3(co11, co21) && equals_v3v3(co12, co22)) ||
1336             (equals_v3v3(co11, co22) && equals_v3v3(co12, co21)))
1337         {
1338                 return true;
1339         }
1340         else {
1341                 return false;
1342         }
1343 }
1344
1345 /* Callback used in point_is_visible to exclude hits on the faces that are the same
1346  * as or contain the hitting element (which is in user_data).
1347  * Also (see T44492) want to exclude hits on faces that butt up to the hitting element
1348  * (e.g., when you double an edge by an edge split).
1349  */
1350 static bool bm_ray_cast_cb_elem_not_in_face_check(BMFace *f, void *user_data)
1351 {
1352         bool ans;
1353         BMEdge *e, *e2;
1354         BMIter iter;
1355
1356         switch (((BMElem *)user_data)->head.htype) {
1357                 case BM_FACE:
1358                         ans = (BMFace *)user_data != f;
1359                         break;
1360                 case BM_EDGE:
1361                         e = (BMEdge *)user_data;
1362                         ans = !BM_edge_in_face(e, f);
1363                         if (ans) {
1364                                 /* Is it a boundary edge, coincident with a split edge? */
1365                                 if (BM_edge_is_boundary(e)) {
1366                                         BM_ITER_ELEM(e2, &iter, f, BM_EDGES_OF_FACE) {
1367                                                 if (coinciding_edges(e, e2)) {
1368                                                         ans = false;
1369                                                         break;
1370                                                 }
1371                                         }
1372                                 }
1373                         }
1374                         break;
1375                 case BM_VERT:
1376                         ans = !BM_vert_in_face((BMVert *)user_data, f);
1377                         break;
1378                 default:
1379                         ans = true;
1380                         break;
1381         }
1382         return ans;
1383 }
1384
1385
1386 /**
1387  * Check if \a p is visible (not clipped, not occluded by another face).
1388  * s in screen projection of p.
1389  *
1390  * \param ele_test  Optional vert/edge/face to use when \a p is on the surface of the geometry,
1391  * intersecting faces matching this face (or connected when an vert/edge) will be ignored.
1392  */
1393 static bool point_is_visible(
1394         KnifeTool_OpData *kcd, const float p[3], const float s[2], bglMats *mats,
1395         BMElem *ele_test)
1396 {
1397         BMFace *f_hit;
1398
1399         /* If box clipping on, make sure p is not clipped */
1400         if (kcd->vc.rv3d->rflag & RV3D_CLIPPING &&
1401             ED_view3d_clipping_test(kcd->vc.rv3d, p, true))
1402         {
1403                 return false;
1404         }
1405
1406         /* If not cutting through, make sure no face is in front of p */
1407         if (!kcd->cut_through) {
1408                 float dist;
1409                 float view[3], p_ofs[3];
1410
1411                 /* TODO: I think there's a simpler way to get the required raycast ray */
1412                 ED_view3d_unproject(mats, view, s[0], s[1], 0.0f);
1413
1414                 mul_m4_v3(kcd->ob->imat, view);
1415
1416                 /* make p_ofs a little towards view, so ray doesn't hit p's face. */
1417                 sub_v3_v3(view, p);
1418                 dist = normalize_v3(view);
1419                 copy_v3_v3(p_ofs, p);
1420
1421                 /* avoid projecting behind the viewpoint */
1422                 if (kcd->is_ortho && (kcd->vc.rv3d->persp != RV3D_CAMOB)) {
1423                         dist = kcd->vc.v3d->far * 2.0f;
1424                 }
1425
1426                 if (kcd->vc.rv3d->rflag & RV3D_CLIPPING) {
1427                         float view_clip[2][3];
1428                         /* note: view_clip[0] should never get clipped */
1429                         copy_v3_v3(view_clip[0], p_ofs);
1430                         madd_v3_v3v3fl(view_clip[1], p_ofs, view, dist);
1431
1432                         if (clip_segment_v3_plane_n(view_clip[0], view_clip[1], kcd->vc.rv3d->clip_local, 6)) {
1433                                 dist = len_v3v3(p_ofs, view_clip[1]);
1434                         }
1435                 }
1436
1437                 /* see if there's a face hit between p1 and the view */
1438                 if (ele_test) {
1439                         f_hit = BKE_bmbvh_ray_cast_filter(
1440                                     kcd->bmbvh, p_ofs, view, KNIFE_FLT_EPS, &dist, NULL, NULL,
1441                                     bm_ray_cast_cb_elem_not_in_face_check, ele_test);
1442                 }
1443                 else {
1444                         f_hit = BKE_bmbvh_ray_cast(
1445                                     kcd->bmbvh, p_ofs, view, KNIFE_FLT_EPS, &dist, NULL, NULL);
1446                 }
1447
1448                 if (f_hit) {
1449                         return false;
1450                 }
1451         }
1452
1453         return true;
1454 }
1455
1456 /* Clip the line (v1, v2) to planes perpendicular to it and distances d from
1457  * the closest point on the line to the origin */
1458 static void clip_to_ortho_planes(float v1[3], float v2[3], const float center[3], const float d)
1459 {
1460         float closest[3], dir[3];
1461
1462         sub_v3_v3v3(dir, v1, v2);
1463         normalize_v3(dir);
1464
1465         /* could be v1 or v2 */
1466         sub_v3_v3(v1, center);
1467         project_plane_v3_v3v3(closest, v1, dir);
1468         add_v3_v3(closest, center);
1469
1470         madd_v3_v3v3fl(v1, closest, dir,  d);
1471         madd_v3_v3v3fl(v2, closest, dir, -d);
1472 }
1473
1474 static void set_linehit_depth(KnifeTool_OpData *kcd, KnifeLineHit *lh)
1475 {
1476         lh->m = dot_m4_v3_row_z(kcd->vc.rv3d->persmatob, lh->cagehit);
1477 }
1478
1479 /* Finds visible (or all, if cutting through) edges that intersects the current screen drag line */
1480 static void knife_find_line_hits(KnifeTool_OpData *kcd)
1481 {
1482         bglMats mats;
1483         SmallHash faces, kfes, kfvs;
1484         float v1[3], v2[3], v3[3], v4[3], s1[2], s2[2];
1485         BVHTree *planetree, *tree;
1486         BVHTreeOverlap *results, *result;
1487         BMLoop **ls;
1488         BMFace *f;
1489         KnifeEdge *kfe;
1490         KnifeVert *v;
1491         ListBase *lst;
1492         Ref *ref;
1493         KnifeLineHit *linehits = NULL;
1494         BLI_array_declare(linehits);
1495         SmallHashIter hiter;
1496         KnifeLineHit hit;
1497         void *val;
1498         void **val_p;
1499         float plane_cos[12];
1500         float s[2], se1[2], se2[2], sint[2];
1501         float r1[3], r2[3];
1502         float d, d1, d2, lambda;
1503         float vert_tol, vert_tol_sq;
1504         float line_tol, line_tol_sq;
1505         float face_tol, face_tol_sq;
1506         int isect_kind;
1507         unsigned int tot;
1508         int i;
1509         const bool use_hit_prev = true;
1510         const bool use_hit_curr = (kcd->is_drag_hold == false);
1511
1512         bgl_get_mats(&mats);
1513
1514         if (kcd->linehits) {
1515                 MEM_freeN(kcd->linehits);
1516                 kcd->linehits = NULL;
1517                 kcd->totlinehit = 0;
1518         }
1519
1520         copy_v3_v3(v1, kcd->prev.cage);
1521         copy_v3_v3(v2, kcd->curr.cage);
1522
1523         /* project screen line's 3d coordinates back into 2d */
1524         knife_project_v2(kcd, v1, s1);
1525         knife_project_v2(kcd, v2, s2);
1526
1527         if (kcd->is_interactive) {
1528                 if (len_squared_v2v2(s1, s2) < 1.0f) {
1529                         return;
1530                 }
1531         }
1532         else {
1533                 if (len_squared_v2v2(s1, s2) < KNIFE_FLT_EPS_SQUARED) {
1534                         return;
1535                 }
1536         }
1537
1538         /* unproject screen line */
1539         ED_view3d_win_to_segment(kcd->ar, kcd->vc.v3d, s1, v1, v3, true);
1540         ED_view3d_win_to_segment(kcd->ar, kcd->vc.v3d, s2, v2, v4, true);
1541
1542         mul_m4_v3(kcd->ob->imat, v1);
1543         mul_m4_v3(kcd->ob->imat, v2);
1544         mul_m4_v3(kcd->ob->imat, v3);
1545         mul_m4_v3(kcd->ob->imat, v4);
1546
1547         /* numeric error, 'v1' -> 'v2', 'v2' -> 'v4' can end up being ~2000 units apart in otho mode
1548          * (from ED_view3d_win_to_segment_clip() above)
1549          * this gives precision error; rather then solving properly
1550          * (which may involve using doubles everywhere!),
1551          * limit the distance between these points */
1552         if (kcd->is_ortho && (kcd->vc.rv3d->persp != RV3D_CAMOB)) {
1553                 if (kcd->ortho_extent == 0.0f)
1554                         calc_ortho_extent(kcd);
1555                 clip_to_ortho_planes(v1, v3, kcd->ortho_extent_center, kcd->ortho_extent + 10.0f);
1556                 clip_to_ortho_planes(v2, v4, kcd->ortho_extent_center, kcd->ortho_extent + 10.0f);
1557         }
1558
1559         /* First use bvh tree to find faces, knife edges, and knife verts that might
1560          * intersect the cut plane with rays v1-v3 and v2-v4.
1561          * This deduplicates the candidates before doing more expensive intersection tests. */
1562
1563         tree = BKE_bmbvh_tree_get(kcd->bmbvh);
1564         planetree = BLI_bvhtree_new(4, FLT_EPSILON * 4, 8, 8);
1565         copy_v3_v3(plane_cos + 0, v1);
1566         copy_v3_v3(plane_cos + 3, v2);
1567         copy_v3_v3(plane_cos + 6, v3);
1568         copy_v3_v3(plane_cos + 9, v4);
1569         BLI_bvhtree_insert(planetree, 0, plane_cos, 4);
1570         BLI_bvhtree_balance(planetree);
1571
1572         results = BLI_bvhtree_overlap(tree, planetree, &tot, NULL, NULL);
1573         if (!results) {
1574                 BLI_bvhtree_free(planetree);
1575                 return;
1576         }
1577
1578         BLI_smallhash_init(&faces);
1579         BLI_smallhash_init(&kfes);
1580         BLI_smallhash_init(&kfvs);
1581
1582         for (i = 0, result = results; i < tot; i++, result++) {
1583                 ls = (BMLoop **)kcd->em->looptris[result->indexA];
1584                 f = ls[0]->f;
1585                 set_lowest_face_tri(kcd, f, result->indexA);
1586
1587                 /* occlude but never cut unselected faces (when only_select is used) */
1588                 if (kcd->only_select && !BM_elem_flag_test(f, BM_ELEM_SELECT)) {
1589                         continue;
1590                 }
1591                 /* for faces, store index of lowest hit looptri in hash */
1592                 if (BLI_smallhash_haskey(&faces, (uintptr_t)f)) {
1593                         continue;
1594                 }
1595                 /* don't care what the value is except that it is non-NULL, for iterator */
1596                 BLI_smallhash_insert(&faces, (uintptr_t)f, f);
1597
1598                 lst = knife_get_face_kedges(kcd, f);
1599                 for (ref = lst->first; ref; ref = ref->next) {
1600                         kfe = ref->ref;
1601                         if (BLI_smallhash_haskey(&kfes, (uintptr_t)kfe))
1602                                 continue;
1603                         BLI_smallhash_insert(&kfes, (uintptr_t)kfe, kfe);
1604                         v = kfe->v1;
1605                         BLI_smallhash_reinsert(&kfvs, (uintptr_t)v, v);
1606                         v = kfe->v2;
1607                         BLI_smallhash_reinsert(&kfvs, (uintptr_t)v, v);
1608                 }
1609         }
1610
1611         /* Now go through the candidates and find intersections */
1612         /* These tolerances, in screen space, are for intermediate hits, as ends are already snapped to screen */
1613
1614         vert_tol = KNIFE_FLT_EPS_PX_VERT;
1615         line_tol = KNIFE_FLT_EPS_PX_EDGE;
1616         face_tol = KNIFE_FLT_EPS_PX_FACE;
1617
1618         vert_tol_sq = vert_tol * vert_tol;
1619         line_tol_sq = line_tol * line_tol;
1620         face_tol_sq = face_tol * face_tol;
1621
1622         /* Assume these tolerances swamp floating point rounding errors in calculations below */
1623
1624         /* first look for vertex hits */
1625         for (val_p = BLI_smallhash_iternew_p(&kfvs, &hiter, (uintptr_t *)&v); val_p;
1626              val_p = BLI_smallhash_iternext_p(&hiter, (uintptr_t *)&v))
1627         {
1628                 KnifeEdge *kfe_hit = NULL;
1629
1630                 knife_project_v2(kcd, v->cageco, s);
1631                 d = dist_squared_to_line_segment_v2(s, s1, s2);
1632                 if ((d <= vert_tol_sq) &&
1633                     (point_is_visible(kcd, v->cageco, s, &mats, bm_elem_from_knife_vert(v, &kfe_hit))))
1634                 {
1635                         memset(&hit, 0, sizeof(hit));
1636                         hit.v = v;
1637
1638                         /* If this isn't from an existing BMVert, it may have been added to a BMEdge originally.
1639                                  * knowing if the hit comes from an edge is important for edge-in-face checks later on
1640                                  * see: #knife_add_single_cut -> #knife_verts_edge_in_face, T42611 */
1641                         if (kfe_hit) {
1642                                 hit.kfe = kfe_hit;
1643                         }
1644
1645                         copy_v3_v3(hit.hit, v->co);
1646                         copy_v3_v3(hit.cagehit, v->cageco);
1647                         copy_v2_v2(hit.schit, s);
1648                         set_linehit_depth(kcd, &hit);
1649                         BLI_array_append(linehits, hit);
1650                 }
1651                 else {
1652                         /* note that these vertes aren't used */
1653                         *val_p = NULL;
1654                 }
1655         }
1656
1657         /* now edge hits; don't add if a vertex at end of edge should have hit */
1658         for (val = BLI_smallhash_iternew(&kfes, &hiter, (uintptr_t *)&kfe); val;
1659              val = BLI_smallhash_iternext(&hiter, (uintptr_t *)&kfe))
1660         {
1661                 int kfe_verts_in_cut;
1662                 /* if we intersect both verts, don't attempt to intersect the edge */
1663
1664                 kfe_verts_in_cut = (BLI_smallhash_lookup(&kfvs, (intptr_t)kfe->v1) != NULL) +
1665                                    (BLI_smallhash_lookup(&kfvs, (intptr_t)kfe->v2) != NULL);
1666
1667                 if (kfe_verts_in_cut == 2) {
1668                         continue;
1669                 }
1670
1671                 knife_project_v2(kcd, kfe->v1->cageco, se1);
1672                 knife_project_v2(kcd, kfe->v2->cageco, se2);
1673                 isect_kind = (kfe_verts_in_cut) ? -1 : isect_seg_seg_v2_point(s1, s2, se1, se2, sint);
1674                 if (isect_kind == -1) {
1675                         /* isect_seg_seg_v2 doesn't do tolerance test around ends of s1-s2 */
1676                         closest_to_line_segment_v2(sint, s1, se1, se2);
1677                         if (len_squared_v2v2(sint, s1) <= line_tol_sq)
1678                                 isect_kind = 1;
1679                         else {
1680                                 closest_to_line_segment_v2(sint, s2, se1, se2);
1681                                 if (len_squared_v2v2(sint, s2) <= line_tol_sq)
1682                                         isect_kind = 1;
1683                         }
1684                 }
1685                 if (isect_kind == 1) {
1686                         d1 = len_v2v2(sint, se1);
1687                         d2 = len_v2v2(se2, se1);
1688                         if (!(d1 <= line_tol || d2 <= line_tol || fabsf(d1 - d2) <= line_tol)) {
1689                                 float p_cage[3], p_cage_tmp[3];
1690                                 lambda = d1 / d2;
1691                                 /* Can't just interpolate between ends of kfe because
1692                                  * that doesn't work with perspective transformation.
1693                                  * Need to find 3d intersection of ray through sint */
1694                                 knife_input_ray_segment(kcd, sint, 1.0f, r1, r2);
1695                                 isect_kind = isect_line_line_v3(kfe->v1->cageco, kfe->v2->cageco, r1, r2, p_cage, p_cage_tmp);
1696                                 if (isect_kind >= 1 && point_is_visible(kcd, p_cage, sint, &mats, bm_elem_from_knife_edge(kfe))) {
1697                                         memset(&hit, 0, sizeof(hit));
1698                                         if (kcd->snap_midpoints) {
1699                                                 /* choose intermediate point snap too */
1700                                                 mid_v3_v3v3(p_cage, kfe->v1->cageco, kfe->v2->cageco);
1701                                                 mid_v2_v2v2(sint, se1, se2);
1702                                                 lambda = 0.5f;
1703                                         }
1704                                         hit.kfe = kfe;
1705                                         transform_point_by_seg_v3(
1706                                                 hit.hit, p_cage,
1707                                                 kfe->v1->co, kfe->v2->co,
1708                                                 kfe->v1->cageco, kfe->v2->cageco);
1709                                         copy_v3_v3(hit.cagehit, p_cage);
1710                                         copy_v2_v2(hit.schit, sint);
1711                                         hit.perc = lambda;
1712                                         set_linehit_depth(kcd, &hit);
1713                                         BLI_array_append(linehits, hit);
1714                                 }
1715                         }
1716                 }
1717         }
1718         /* now face hits; don't add if a vertex or edge in face should have hit */
1719         for (val = BLI_smallhash_iternew(&faces, &hiter, (uintptr_t *)&f); val;
1720              val = BLI_smallhash_iternext(&hiter, (uintptr_t *)&f))
1721         {
1722                 float p[3], p_cage[3];
1723
1724                 if (use_hit_prev && knife_ray_intersect_face(kcd, s1, v1, v3, f, face_tol_sq, p, p_cage)) {
1725                         if (point_is_visible(kcd, p_cage, s1, &mats, (BMElem *)f)) {
1726                                 memset(&hit, 0, sizeof(hit));
1727                                 hit.f = f;
1728                                 copy_v3_v3(hit.hit, p);
1729                                 copy_v3_v3(hit.cagehit, p_cage);
1730                                 copy_v2_v2(hit.schit, s1);
1731                                 set_linehit_depth(kcd, &hit);
1732                                 BLI_array_append(linehits, hit);
1733                         }
1734                 }
1735
1736                 if (use_hit_curr && knife_ray_intersect_face(kcd, s2, v2, v4, f, face_tol_sq, p, p_cage)) {
1737                         if (point_is_visible(kcd, p_cage, s2, &mats, (BMElem *)f)) {
1738                                 memset(&hit, 0, sizeof(hit));
1739                                 hit.f = f;
1740                                 copy_v3_v3(hit.hit, p);
1741                                 copy_v3_v3(hit.cagehit, p_cage);
1742                                 copy_v2_v2(hit.schit, s2);
1743                                 set_linehit_depth(kcd, &hit);
1744                                 BLI_array_append(linehits, hit);
1745                         }
1746                 }
1747         }
1748
1749         kcd->linehits = linehits;
1750         kcd->totlinehit = BLI_array_count(linehits);
1751
1752         /* find position along screen line, used for sorting */
1753         for (i = 0; i < kcd->totlinehit; i++) {
1754                 KnifeLineHit *lh = kcd->linehits + i;
1755
1756                 lh->l = len_v2v2(lh->schit, s1) / len_v2v2(s2, s1);
1757         }
1758
1759         BLI_smallhash_release(&faces);
1760         BLI_smallhash_release(&kfes);
1761         BLI_smallhash_release(&kfvs);
1762         BLI_bvhtree_free(planetree);
1763         if (results)
1764                 MEM_freeN(results);
1765 }
1766
1767 static void knife_input_ray_segment(KnifeTool_OpData *kcd, const float mval[2], const float ofs,
1768                                     float r_origin[3], float r_origin_ofs[3])
1769 {
1770         bglMats mats;
1771
1772         bgl_get_mats(&mats);
1773
1774         /* unproject to find view ray */
1775         ED_view3d_unproject(&mats, r_origin,     mval[0], mval[1], 0.0f);
1776         ED_view3d_unproject(&mats, r_origin_ofs, mval[0], mval[1], ofs);
1777
1778         /* transform into object space */
1779         invert_m4_m4(kcd->ob->imat, kcd->ob->obmat); 
1780
1781         mul_m4_v3(kcd->ob->imat, r_origin);
1782         mul_m4_v3(kcd->ob->imat, r_origin_ofs);
1783 }
1784
1785 static BMFace *knife_find_closest_face(KnifeTool_OpData *kcd, float co[3], float cageco[3], bool *is_space)
1786 {
1787         BMFace *f;
1788         float dist = KMAXDIST;
1789         float origin[3];
1790         float origin_ofs[3];
1791         float ray[3], ray_normal[3];
1792
1793         /* unproject to find view ray */
1794         knife_input_ray_segment(kcd, kcd->curr.mval, 1.0f, origin, origin_ofs);
1795         sub_v3_v3v3(ray, origin_ofs, origin);
1796         normalize_v3_v3(ray_normal, ray);
1797
1798         f = BKE_bmbvh_ray_cast(kcd->bmbvh, origin, ray_normal, 0.0f, NULL, co, cageco);
1799
1800         if (f && kcd->only_select && BM_elem_flag_test(f, BM_ELEM_SELECT) == 0) {
1801                 f = NULL;
1802         }
1803
1804         if (is_space)
1805                 *is_space = !f;
1806
1807         if (!f) {
1808                 if (kcd->is_interactive) {
1809                         /* try to use backbuffer selection method if ray casting failed */
1810                         f = EDBM_face_find_nearest(&kcd->vc, &dist);
1811
1812                         /* cheat for now; just put in the origin instead
1813                          * of a true coordinate on the face.
1814                          * This just puts a point 1.0f infront of the view. */
1815                         add_v3_v3v3(co, origin, ray);
1816                 }
1817         }
1818
1819         return f;
1820 }
1821
1822 /* find the 2d screen space density of vertices within a radius.  used to scale snapping
1823  * distance for picking edges/verts.*/
1824 static int knife_sample_screen_density(KnifeTool_OpData *kcd, const float radius)
1825 {
1826         BMFace *f;
1827         bool is_space;
1828         float co[3], cageco[3], sco[2];
1829
1830         BLI_assert(kcd->is_interactive == true);
1831
1832         f = knife_find_closest_face(kcd, co, cageco, &is_space);
1833
1834         if (f && !is_space) {
1835                 const float radius_sq = radius * radius;
1836                 ListBase *lst;
1837                 Ref *ref;
1838                 float dis_sq;
1839                 int c = 0;
1840
1841                 knife_project_v2(kcd, cageco, sco);
1842
1843                 lst = knife_get_face_kedges(kcd, f);
1844                 for (ref = lst->first; ref; ref = ref->next) {
1845                         KnifeEdge *kfe = ref->ref;
1846                         int i;
1847
1848                         for (i = 0; i < 2; i++) {
1849                                 KnifeVert *kfv = i ? kfe->v2 : kfe->v1;
1850
1851                                 knife_project_v2(kcd, kfv->cageco, kfv->sco);
1852
1853                                 dis_sq = len_squared_v2v2(kfv->sco, sco);
1854                                 if (dis_sq < radius_sq) {
1855                                         if (kcd->vc.rv3d->rflag & RV3D_CLIPPING) {
1856                                                 if (ED_view3d_clipping_test(kcd->vc.rv3d, kfv->cageco, true) == 0) {
1857                                                         c++;
1858                                                 }
1859                                         }
1860                                         else {
1861                                                 c++;
1862                                         }
1863                                 }
1864                         }
1865                 }
1866
1867                 return c;
1868         }
1869
1870         return 0;
1871 }
1872
1873 /* returns snapping distance for edges/verts, scaled by the density of the
1874  * surrounding mesh (in screen space)*/
1875 static float knife_snap_size(KnifeTool_OpData *kcd, float maxsize)
1876 {
1877         float density;
1878
1879         if (kcd->is_interactive) {
1880                 density = (float)knife_sample_screen_density(kcd, maxsize * 2.0f);
1881         }
1882         else {
1883                 density = 1.0f;
1884         }
1885
1886         if (density < 1.0f)
1887                 density = 1.0f;
1888
1889         return min_ff(maxsize / (density * 0.5f), maxsize);
1890 }
1891
1892 /* p is closest point on edge to the mouse cursor */
1893 static KnifeEdge *knife_find_closest_edge(KnifeTool_OpData *kcd, float p[3], float cagep[3],
1894                                           BMFace **fptr, bool *is_space)
1895 {
1896         BMFace *f;
1897         float co[3], cageco[3], sco[2];
1898         float maxdist = knife_snap_size(kcd, kcd->ethresh);
1899
1900         if (kcd->ignore_vert_snapping)
1901                 maxdist *= 0.5f;
1902
1903         f = knife_find_closest_face(kcd, co, cageco, NULL);
1904         *is_space = !f;
1905
1906         kcd->curr.bmface = f;
1907
1908         if (f) {
1909                 const float maxdist_sq = maxdist * maxdist;
1910                 KnifeEdge *cure = NULL;
1911                 float cur_cagep[3];
1912                 ListBase *lst;
1913                 Ref *ref;
1914                 float dis_sq, curdis_sq = FLT_MAX;
1915
1916                 /* set p to co, in case we don't find anything, means a face cut */
1917                 copy_v3_v3(p, co);
1918                 copy_v3_v3(cagep, cageco);
1919
1920                 knife_project_v2(kcd, cageco, sco);
1921
1922                 /* look through all edges associated with this face */
1923                 lst = knife_get_face_kedges(kcd, f);
1924                 for (ref = lst->first; ref; ref = ref->next) {
1925                         KnifeEdge *kfe = ref->ref;
1926                         float test_cagep[3];
1927                         float lambda;
1928
1929                         /* project edge vertices into screen space */
1930                         knife_project_v2(kcd, kfe->v1->cageco, kfe->v1->sco);
1931                         knife_project_v2(kcd, kfe->v2->cageco, kfe->v2->sco);
1932
1933                         /* check if we're close enough and calculate 'lambda' */
1934                         if (kcd->is_angle_snapping) {
1935                         /* if snapping, check we're in bounds */
1936                                 float sco_snap[2];
1937                                 isect_line_line_v2_point(kfe->v1->sco, kfe->v2->sco, kcd->prev.mval, kcd->curr.mval, sco_snap);
1938                                 lambda = line_point_factor_v2(sco_snap, kfe->v1->sco, kfe->v2->sco);
1939
1940                                 /* be strict about angle-snapping within edge */
1941                                 if ((lambda < 0.0f - KNIFE_FLT_EPSBIG) || (lambda > 1.0f + KNIFE_FLT_EPSBIG)) {
1942                                         continue;
1943                                 }
1944
1945                                 dis_sq = len_squared_v2v2(sco, sco_snap);
1946                                 if (dis_sq < curdis_sq && dis_sq < maxdist_sq) {
1947                                         /* we already have 'lambda' */
1948                                 }
1949                                 else {
1950                                         continue;
1951                                 }
1952                         }
1953                         else {
1954                                 dis_sq = dist_squared_to_line_segment_v2(sco, kfe->v1->sco, kfe->v2->sco);
1955                                 if (dis_sq < curdis_sq && dis_sq < maxdist_sq) {
1956                                         lambda = line_point_factor_v2(sco, kfe->v1->sco, kfe->v2->sco);
1957                                 }
1958                                 else {
1959                                         continue;
1960                                 }
1961                         }
1962
1963                         /* now we have 'lambda' calculated (in screen-space) */
1964                         knife_interp_v3_v3v3(kcd, test_cagep, kfe->v1->cageco, kfe->v2->cageco, lambda);
1965
1966                         if (kcd->vc.rv3d->rflag & RV3D_CLIPPING) {
1967                                 /* check we're in the view */
1968                                 if (ED_view3d_clipping_test(kcd->vc.rv3d, test_cagep, true)) {
1969                                         continue;
1970                                 }
1971                         }
1972
1973                         cure = kfe;
1974                         curdis_sq = dis_sq;
1975                         copy_v3_v3(cur_cagep, test_cagep);
1976                 }
1977
1978                 if (fptr)
1979                         *fptr = f;
1980
1981                 if (cure) {
1982                         if (!kcd->ignore_edge_snapping || !(cure->e)) {
1983                                 KnifeVert *edgesnap = NULL;
1984
1985                                 if (kcd->snap_midpoints) {
1986                                         mid_v3_v3v3(p, cure->v1->co, cure->v2->co);
1987                                         mid_v3_v3v3(cagep, cure->v1->cageco, cure->v2->cageco);
1988                                 }
1989                                 else {
1990                                         float lambda = line_point_factor_v3(cur_cagep, cure->v1->cageco, cure->v2->cageco);
1991                                         copy_v3_v3(cagep, cur_cagep);
1992                                         interp_v3_v3v3(p, cure->v1->co, cure->v2->co, lambda);
1993                                 }
1994
1995                                 /* update mouse coordinates to the snapped-to edge's screen coordinates
1996                                  * this is important for angle snap, which uses the previous mouse position */
1997                                 edgesnap = new_knife_vert(kcd, p, cagep);
1998                                 kcd->curr.mval[0] = edgesnap->sco[0];
1999                                 kcd->curr.mval[1] = edgesnap->sco[1];
2000
2001                         }
2002                         else {
2003                                 return NULL;
2004                         }
2005                 }
2006
2007                 return cure;
2008         }
2009
2010         if (fptr)
2011                 *fptr = NULL;
2012
2013         return NULL;
2014 }
2015
2016 /* find a vertex near the mouse cursor, if it exists */
2017 static KnifeVert *knife_find_closest_vert(KnifeTool_OpData *kcd, float p[3], float cagep[3], BMFace **fptr,
2018                                           bool *is_space)
2019 {
2020         BMFace *f;
2021         float co[3], cageco[3], sco[2], maxdist = knife_snap_size(kcd, kcd->vthresh);
2022
2023         if (kcd->ignore_vert_snapping)
2024                 maxdist *= 0.5f;
2025
2026         f = knife_find_closest_face(kcd, co, cageco, is_space);
2027
2028         kcd->curr.bmface = f;
2029
2030         if (f) {
2031                 const float maxdist_sq = maxdist * maxdist;
2032                 ListBase *lst;
2033                 Ref *ref;
2034                 KnifeVert *curv = NULL;
2035                 float dis_sq, curdis_sq = FLT_MAX;
2036
2037                 /* set p to co, in case we don't find anything, means a face cut */
2038                 copy_v3_v3(p, co);
2039                 copy_v3_v3(cagep, cageco);
2040
2041                 knife_project_v2(kcd, cageco, sco);
2042
2043                 lst = knife_get_face_kedges(kcd, f);
2044                 for (ref = lst->first; ref; ref = ref->next) {
2045                         KnifeEdge *kfe = ref->ref;
2046                         int i;
2047
2048                         for (i = 0; i < 2; i++) {
2049                                 KnifeVert *kfv = i ? kfe->v2 : kfe->v1;
2050
2051                                 knife_project_v2(kcd, kfv->cageco, kfv->sco);
2052
2053                                 /* be strict about angle snapping, the vertex needs to be very close to the angle, or we ignore */
2054                                 if (kcd->is_angle_snapping) {
2055                                         if (dist_squared_to_line_segment_v2(kfv->sco, kcd->prev.mval, kcd->curr.mval) > KNIFE_FLT_EPSBIG) {
2056                                                 continue;
2057                                         }
2058                                 }
2059
2060                                 dis_sq = len_squared_v2v2(kfv->sco, sco);
2061                                 if (dis_sq < curdis_sq && dis_sq < maxdist_sq) {
2062                                         if (kcd->vc.rv3d->rflag & RV3D_CLIPPING) {
2063                                                 if (ED_view3d_clipping_test(kcd->vc.rv3d, kfv->cageco, true) == 0) {
2064                                                         curv = kfv;
2065                                                         curdis_sq = dis_sq;
2066                                                 }
2067                                         }
2068                                         else {
2069                                                 curv = kfv;
2070                                                 curdis_sq = dis_sq;
2071                                         }
2072                                 }
2073                         }
2074                 }
2075
2076                 if (!kcd->ignore_vert_snapping || !(curv && curv->v)) {
2077                         if (fptr)
2078                                 *fptr = f;
2079
2080                         if (curv) {
2081                                 copy_v3_v3(p, curv->co);
2082                                 copy_v3_v3(cagep, curv->cageco);
2083
2084                                 /* update mouse coordinates to the snapped-to vertex's screen coordinates
2085                                  * this is important for angle snap, which uses the previous mouse position */
2086                                 kcd->curr.mval[0] = curv->sco[0];
2087                                 kcd->curr.mval[1] = curv->sco[1];
2088                         }
2089
2090                         return curv;
2091                 }
2092                 else {
2093                         if (fptr)
2094                                 *fptr = f;
2095
2096                         return NULL;
2097                 }
2098         }
2099
2100         if (fptr)
2101                 *fptr = NULL;
2102
2103         return NULL;
2104 }
2105
2106 /**
2107  * Snaps a 2d vector to an angle, relative to \a v_ref.
2108  */
2109 static float snap_v2_angle(float r[2], const float v[2], const float v_ref[2], float angle_snap)
2110 {
2111         float m2[2][2];
2112         float v_unit[2];
2113         float angle, angle_delta;
2114
2115         BLI_ASSERT_UNIT_V2(v_ref);
2116
2117         normalize_v2_v2(v_unit, v);
2118         angle = angle_signed_v2v2(v_unit, v_ref);
2119         angle_delta = (round(angle / angle_snap) * angle_snap) - angle;
2120         rotate_m2(m2, angle_delta);
2121
2122         mul_v2_m2v2(r, m2, v);
2123         return angle + angle_delta;
2124 }
2125
2126 /* update both kcd->curr.mval and kcd->mval to snap to required angle */
2127 static bool knife_snap_angle(KnifeTool_OpData *kcd)
2128 {
2129         const float dvec_ref[2] = {0.0f, 1.0f};
2130         float dvec[2], dvec_snap[2];
2131         float snap_step = DEG2RADF(5);
2132
2133         sub_v2_v2v2(dvec, kcd->curr.mval, kcd->prev.mval);
2134         if (is_zero_v2(dvec)) {
2135                 return false;
2136         }
2137
2138         kcd->angle = snap_v2_angle(dvec_snap, dvec, dvec_ref, snap_step);
2139
2140         add_v2_v2v2(kcd->curr.mval, kcd->prev.mval, dvec_snap);
2141
2142         copy_v2_v2(kcd->mval, kcd->curr.mval);
2143
2144         return true;
2145 }
2146
2147 /* update active knife edge/vert pointers */
2148 static int knife_update_active(KnifeTool_OpData *kcd)
2149 {
2150         knife_pos_data_clear(&kcd->curr);
2151         copy_v2_v2(kcd->curr.mval, kcd->mval);
2152
2153         /* view matrix may have changed, reproject */
2154         knife_project_v2(kcd, kcd->prev.cage, kcd->prev.mval);
2155
2156         if (kcd->angle_snapping && (kcd->mode == MODE_DRAGGING)) {
2157                 kcd->is_angle_snapping = knife_snap_angle(kcd);
2158         }
2159         else {
2160                 kcd->is_angle_snapping = false;
2161         }
2162
2163         kcd->curr.vert = knife_find_closest_vert(kcd, kcd->curr.co, kcd->curr.cage, &kcd->curr.bmface, &kcd->curr.is_space);
2164
2165         if (!kcd->curr.vert &&
2166             /* no edge snapping while dragging (edges are too sticky when cuts are immediate) */
2167             !kcd->is_drag_hold)
2168         {
2169                 kcd->curr.edge = knife_find_closest_edge(kcd, kcd->curr.co, kcd->curr.cage,
2170                                                          &kcd->curr.bmface, &kcd->curr.is_space);
2171         }
2172
2173         /* if no hits are found this would normally default to (0, 0, 0) so instead
2174          * get a point at the mouse ray closest to the previous point.
2175          * Note that drawing lines in `free-space` isn't properly supported
2176          * but theres no guarantee (0, 0, 0) has any geometry either - campbell */
2177         if (kcd->curr.vert == NULL && kcd->curr.edge == NULL && kcd->curr.bmface == NULL) {
2178                 float origin[3];
2179                 float origin_ofs[3];
2180
2181                 knife_input_ray_segment(kcd, kcd->curr.mval, 1.0f, origin, origin_ofs);
2182
2183                 if (!isect_line_plane_v3(kcd->curr.cage, origin, origin_ofs, kcd->prev.cage, kcd->proj_zaxis)) {
2184                         copy_v3_v3(kcd->curr.cage, kcd->prev.cage);
2185
2186                         /* should never fail! */
2187                         BLI_assert(0);
2188                 }
2189         }
2190
2191         if (kcd->mode == MODE_DRAGGING) {
2192                 knife_find_line_hits(kcd);
2193         }
2194         return 1;
2195 }
2196
2197 static int sort_verts_by_dist_cb(void *co_p, const void *cur_a_p, const void *cur_b_p)
2198 {
2199         const KnifeVert *cur_a = ((const Ref *)cur_a_p)->ref;
2200         const KnifeVert *cur_b = ((const Ref *)cur_b_p)->ref;
2201         const float *co = co_p;
2202         const float a_sq = len_squared_v3v3(co, cur_a->co);
2203         const float b_sq = len_squared_v3v3(co, cur_b->co);
2204
2205         if      (a_sq < b_sq) return -1;
2206         else if (a_sq > b_sq) return  1;
2207         else                  return  0;
2208 }
2209
2210 /* The chain so far goes from an instantiated vertex to kfv (some may be reversed).
2211  * If possible, complete the chain to another instantiated vertex and return 1, else return 0.
2212  * The visited hash says which KnifeVert's have already been tried, not including kfv. */
2213 static bool find_chain_search(KnifeTool_OpData *kcd, KnifeVert *kfv, ListBase *fedges, SmallHash *visited,
2214                               ListBase *chain)
2215 {
2216         Ref *r;
2217         KnifeEdge *kfe;
2218         KnifeVert *kfv_other;
2219
2220         if (kfv->v)
2221                 return true;
2222
2223         BLI_smallhash_insert(visited, (uintptr_t)kfv, NULL);
2224         /* Try all possible next edges. Could either go through fedges
2225          * (all the KnifeEdges for the face being cut) or could go through
2226          * kve->edges and restrict to cutting face and uninstantiated edges.
2227          * Not clear which is better. Let's do the first. */
2228         for (r = fedges->first; r; r = r->next) {
2229                 kfe = r->ref;
2230                 kfv_other = NULL;
2231                 if (kfe->v1 == kfv)
2232                         kfv_other = kfe->v2;
2233                 else if (kfe->v2 == kfv)
2234                         kfv_other = kfe->v1;
2235                 if (kfv_other && !BLI_smallhash_haskey(visited, (uintptr_t)kfv_other)) {
2236                         knife_append_list(kcd, chain, kfe);
2237                         if (find_chain_search(kcd, kfv_other, fedges, visited, chain))
2238                                 return true;
2239                         BLI_remlink(chain, chain->last);
2240                 }
2241         }
2242         return false;
2243 }
2244
2245 static ListBase *find_chain_from_vertex(KnifeTool_OpData *kcd, KnifeEdge *kfe, BMVert *v, ListBase *fedges)
2246 {
2247         SmallHash visited_, *visited = &visited_;
2248         ListBase *ans;
2249         bool found;
2250
2251         ans = knife_empty_list(kcd);
2252         knife_append_list(kcd, ans, kfe);
2253         found = false;
2254         BLI_smallhash_init(visited);
2255         if (kfe->v1->v == v) {
2256                 BLI_smallhash_insert(visited, (uintptr_t)(kfe->v1), NULL);
2257                 found = find_chain_search(kcd, kfe->v2, fedges, visited, ans);
2258         }
2259         else {
2260                 BLI_assert(kfe->v2->v == v);
2261                 BLI_smallhash_insert(visited, (uintptr_t)(kfe->v2), NULL);
2262                 found = find_chain_search(kcd, kfe->v1, fedges, visited, ans);
2263         }
2264
2265         BLI_smallhash_release(visited);
2266
2267         if (found)
2268                 return ans;
2269         else
2270                 return NULL;
2271 }
2272
2273 /* Find a chain in fedges from one instantiated vertex to another.
2274  * Remove the edges in the chain from fedges and return a separate list of the chain. */
2275 static ListBase *find_chain(KnifeTool_OpData *kcd, ListBase *fedges)
2276 {
2277         Ref *r, *ref;
2278         KnifeEdge *kfe;
2279         BMVert *v1, *v2;
2280         ListBase *ans;
2281
2282         ans = NULL;
2283
2284         for (r = fedges->first; r; r = r->next) {
2285                 kfe = r->ref;
2286                 v1 = kfe->v1->v;
2287                 v2 = kfe->v2->v;
2288                 if (v1 && v2) {
2289                         ans = knife_empty_list(kcd);
2290                         knife_append_list(kcd, ans, kfe);
2291                         break;
2292                 }
2293                 if (v1)
2294                         ans = find_chain_from_vertex(kcd, kfe, v1, fedges);
2295                 else if (v2)
2296                         ans = find_chain_from_vertex(kcd, kfe, v2, fedges);
2297                 if (ans)
2298                         break;
2299         }
2300         if (ans) {
2301                 BLI_assert(!BLI_listbase_is_empty(ans));
2302                 for (r = ans->first; r; r = r->next) {
2303                         ref = find_ref(fedges, r->ref);
2304                         BLI_assert(ref != NULL);
2305                         BLI_remlink(fedges, ref);
2306                 }
2307         }
2308         return ans;
2309 }
2310
2311 /* The hole so far goes from kfvfirst to kfv (some may be reversed).
2312  * If possible, complete the hole back to kfvfirst and return 1, else return 0.
2313  * The visited hash says which KnifeVert's have already been tried, not including kfv or kfvfirst. */
2314 static bool find_hole_search(KnifeTool_OpData *kcd, KnifeVert *kfvfirst, KnifeVert *kfv, ListBase *fedges,
2315                              SmallHash *visited, ListBase *hole)
2316 {
2317         Ref *r;
2318         KnifeEdge *kfe, *kfelast;
2319         KnifeVert *kfv_other;
2320
2321         if (kfv == kfvfirst)
2322                 return true;
2323
2324         BLI_smallhash_insert(visited, (uintptr_t)kfv, NULL);
2325         kfelast = ((Ref *)hole->last)->ref;
2326         for (r = fedges->first; r; r = r->next) {
2327                 kfe = r->ref;
2328                 if (kfe == kfelast)
2329                         continue;
2330                 if (kfe->v1->v || kfe->v2->v)
2331                         continue;
2332                 kfv_other = NULL;
2333                 if (kfe->v1 == kfv)
2334                         kfv_other = kfe->v2;
2335                 else if (kfe->v2 == kfv)
2336                         kfv_other = kfe->v1;
2337                 if (kfv_other && !BLI_smallhash_haskey(visited, (uintptr_t)kfv_other)) {
2338                         knife_append_list(kcd, hole, kfe);
2339                         if (find_hole_search(kcd, kfvfirst, kfv_other, fedges, visited, hole))
2340                                 return true;
2341                         BLI_remlink(hole, hole->last);
2342                 }
2343         }
2344         return false;
2345 }
2346
2347 /* Find a hole (simple cycle with no instantiated vertices).
2348  * Remove the edges in the cycle from fedges and return a separate list of the cycle */
2349 static ListBase *find_hole(KnifeTool_OpData *kcd, ListBase *fedges)
2350 {
2351         ListBase *ans;
2352         Ref *r, *ref;
2353         KnifeEdge *kfe;
2354         SmallHash visited_, *visited = &visited_;
2355         bool found;
2356
2357         ans = NULL;
2358         found = false;
2359
2360         for (r = fedges->first; r && !found; r = r->next) {
2361                 kfe = r->ref;
2362                 if (kfe->v1->v || kfe->v2->v || kfe->v1 == kfe->v2)
2363                         continue;
2364
2365                 BLI_smallhash_init(visited);
2366                 ans = knife_empty_list(kcd);
2367                 knife_append_list(kcd, ans, kfe);
2368
2369                 found = find_hole_search(kcd, kfe->v1, kfe->v2, fedges, visited, ans);
2370
2371                 BLI_smallhash_release(visited);
2372         }
2373
2374         if (found) {
2375                 for (r = ans->first; r; r = r->next) {
2376                         kfe = r->ref;
2377                         ref = find_ref(fedges, r->ref);
2378                         if (ref)
2379                                 BLI_remlink(fedges, ref);
2380                 }
2381                 return ans;
2382         }
2383         else {
2384                 return NULL;
2385         }
2386 }
2387
2388 /* Try to find "nice" diagonals - short, and far apart from each other.
2389  * If found, return true and make a 'main chain' going across f which uses
2390  * the two diagonals and one part of the hole, and a 'side chain' that
2391  * completes the hole. */
2392 static bool find_hole_chains(KnifeTool_OpData *kcd, ListBase *hole, BMFace *f, ListBase **mainchain,
2393                              ListBase **sidechain)
2394 {
2395         float (*fco)[2], (*hco)[2];
2396         BMVert **fv;
2397         KnifeVert **hv;
2398         KnifeEdge **he;
2399         Ref *r;
2400         KnifeVert *kfv, *kfvother;
2401         KnifeEdge *kfe;
2402         ListBase *chain;
2403         BMVert *v;
2404         BMIter iter;
2405         int nh, nf, i, j, k, m, ax, ay, sep = 0 /* Quite warnings */, bestsep;
2406         int besti[2], bestj[2];
2407         float dist_sq, dist_best_sq;
2408
2409         nh = BLI_listbase_count(hole);
2410         nf = f->len;
2411         if (nh < 2 || nf < 3)
2412                 return false;
2413
2414         /* Gather 2d projections of hole and face vertex coordinates.
2415          * Use best-axis projection - not completely accurate, maybe revisit */
2416         axis_dominant_v3(&ax, &ay, f->no);
2417         hco = BLI_memarena_alloc(kcd->arena, nh * sizeof(float[2]));
2418         fco = BLI_memarena_alloc(kcd->arena, nf * sizeof(float[2]));
2419         hv = BLI_memarena_alloc(kcd->arena, nh * sizeof(KnifeVert *));
2420         fv = BLI_memarena_alloc(kcd->arena, nf * sizeof(BMVert *));
2421         he = BLI_memarena_alloc(kcd->arena, nh * sizeof(KnifeEdge *));
2422
2423         i = 0;
2424         kfv = NULL;
2425         kfvother = NULL;
2426         for (r = hole->first; r; r = r->next) {
2427                 kfe = r->ref;
2428                 he[i] = kfe;
2429                 if (kfvother == NULL) {
2430                         kfv = kfe->v1;
2431                 }
2432                 else {
2433                         kfv = kfvother;
2434                         BLI_assert(kfv == kfe->v1 || kfv == kfe->v2);
2435                 }
2436                 hco[i][0] = kfv->co[ax];
2437                 hco[i][1] = kfv->co[ay];
2438                 hv[i] = kfv;
2439                 kfvother = (kfe->v1 == kfv) ? kfe->v2 : kfe->v1;
2440                 i++;
2441         }
2442
2443         j = 0;
2444         BM_ITER_ELEM (v, &iter, f, BM_VERTS_OF_FACE) {
2445                 fco[j][0] = v->co[ax];
2446                 fco[j][1] = v->co[ay];
2447                 fv[j] = v;
2448                 j++;
2449         }
2450
2451         /* For first diagonal  (m == 0), want shortest length.
2452          * For second diagonal (m == 1), want max separation of index of hole
2453          * vertex from the hole vertex used in the first diagonal, and from there
2454          * want the one with shortest length not to the same vertex as the first diagonal. */
2455         for (m = 0; m < 2; m++) {
2456                 besti[m] = -1;
2457                 bestj[m] = -1;
2458                 dist_best_sq = FLT_MAX;
2459                 bestsep = 0;
2460                 for (i = 0; i < nh; i++) {
2461                         if (m == 1) {
2462                                 if (i == besti[0])
2463                                         continue;
2464                                 sep = (i + nh - besti[0]) % nh;
2465                                 sep = MIN2(sep, nh - sep);
2466                                 if (sep < bestsep)
2467                                         continue;
2468                                 dist_best_sq = FLT_MAX;
2469                         }
2470                         for (j = 0; j < nf; j++) {
2471                                 bool ok;
2472
2473                                 if (m == 1 && j == bestj[0])
2474                                         continue;
2475                                 dist_sq = len_squared_v2v2(hco[i], fco[j]);
2476                                 if (dist_sq > dist_best_sq)
2477                                         continue;
2478
2479                                 ok = true;
2480                                 for (k = 0; k < nh && ok; k++) {
2481                                         if (k == i || (k + 1) % nh == i)
2482                                                 continue;
2483                                         if (isect_line_line_v2(hco[i], fco[j], hco[k], hco[(k + 1) % nh]))
2484                                                 ok = false;
2485                                 }
2486                                 if (!ok)
2487                                         continue;
2488                                 for (k = 0; k < nf && ok; k++) {
2489                                         if (k == j || (k + 1) % nf == j)
2490                                                 continue;
2491                                         if (isect_line_line_v2(hco[i], fco[j], fco[k], fco[(k + 1) % nf]))
2492                                                 ok = false;
2493                                 }
2494                                 if (ok) {
2495                                         besti[m] = i;
2496                                         bestj[m] = j;
2497                                         if (m == 1)
2498                                                 bestsep = sep;
2499                                         dist_best_sq = dist_sq;
2500                                 }
2501                         }
2502                 }
2503         }
2504
2505         if (besti[0] != -1 && besti[1] != -1) {
2506                 BLI_assert(besti[0] != besti[1] && bestj[0] != bestj[1]);
2507                 kfe = new_knife_edge(kcd);
2508                 kfe->v1 = get_bm_knife_vert(kcd, fv[bestj[0]]);
2509                 kfe->v2 = hv[besti[0]];
2510                 chain = knife_empty_list(kcd);
2511                 knife_append_list(kcd, chain, kfe);
2512                 for (i = besti[0]; i != besti[1]; i = (i + 1) % nh) {
2513                         knife_append_list(kcd, chain, he[i]);
2514                 }
2515                 kfe = new_knife_edge(kcd);
2516                 kfe->v1 = hv[besti[1]];
2517                 kfe->v2 = get_bm_knife_vert(kcd, fv[bestj[1]]);
2518                 knife_append_list(kcd, chain, kfe);
2519                 *mainchain = chain;
2520
2521                 chain = knife_empty_list(kcd);
2522                 for (i = besti[1]; i != besti[0]; i = (i + 1) % nh) {
2523                         knife_append_list(kcd, chain, he[i]);
2524                 }
2525                 *sidechain = chain;
2526
2527                 return true;
2528         }
2529         else {
2530                 return false;
2531         }
2532 }
2533
2534 static bool knife_verts_edge_in_face(KnifeVert *v1, KnifeVert *v2, BMFace *f)
2535 {
2536         bool v1_inside, v2_inside;
2537         bool v1_inface, v2_inface;
2538         BMLoop *l1, *l2;
2539
2540         if (!f || !v1 || !v2)
2541                 return false;
2542
2543         l1 = v1->v ? BM_face_vert_share_loop(f, v1->v) : NULL;
2544         l2 = v2->v ? BM_face_vert_share_loop(f, v2->v) : NULL;
2545
2546         if ((l1 && l2) && BM_loop_is_adjacent(l1, l2)) {
2547                 /* boundary-case, always false to avoid edge-in-face checks below */
2548                 return false;
2549         }
2550
2551         /* find out if v1 and v2, if set, are part of the face */
2552         v1_inface = (l1 != NULL);
2553         v2_inface = (l2 != NULL);
2554
2555         /* BM_face_point_inside_test uses best-axis projection so this isn't most accurate test... */
2556         v1_inside = v1_inface ? false : BM_face_point_inside_test(f, v1->co);
2557         v2_inside = v2_inface ? false : BM_face_point_inside_test(f, v2->co);
2558         if ((v1_inface && v2_inside) ||
2559             (v2_inface && v1_inside) ||
2560             (v1_inside && v2_inside))
2561         {
2562                 return true;
2563         }
2564
2565         if (v1_inface && v2_inface) {
2566                 float mid[3];
2567                 /* Can have case where v1 and v2 are on shared chain between two faces.
2568                  * BM_face_splits_check_legal does visibility and self-intersection tests,
2569                  * but it is expensive and maybe a bit buggy, so use a simple
2570                  * "is the midpoint in the face" test */
2571                 mid_v3_v3v3(mid, v1->co, v2->co);
2572                 return BM_face_point_inside_test(f, mid);
2573         }
2574         return false;
2575 }
2576
2577 static bool knife_edge_in_face(KnifeEdge *kfe, BMFace *f)
2578 {
2579         return knife_verts_edge_in_face(kfe->v1, kfe->v2, f);
2580 }
2581
2582 /* Split face f with KnifeEdges on chain.  f remains as one side, the face formed is put in *newface.
2583  * The new face will be on the left side of the chain as viewed from the normal-out side of f. */
2584 static void knife_make_chain_cut(KnifeTool_OpData *kcd, BMFace *f, ListBase *chain, BMFace **r_f_new)
2585 {
2586         BMesh *bm = kcd->em->bm;
2587         KnifeEdge *kfe, *kfelast;
2588         BMVert *v1, *v2;
2589         BMLoop *l_v1, *l_v2;
2590         BMFace *f_new;
2591         Ref *ref;
2592         KnifeVert *kfv, *kfvprev;
2593         BMLoop *l_new, *l_iter;
2594         int i;
2595         int nco = BLI_listbase_count(chain) - 1;
2596         float (*cos)[3] = BLI_array_alloca(cos, nco);
2597         KnifeVert **kverts = BLI_array_alloca(kverts, nco);
2598
2599         kfe = ((Ref *)chain->first)->ref;
2600         v1 = kfe->v1->v ? kfe->v1->v : kfe->v2->v;
2601         kfelast = ((Ref *)chain->last)->ref;
2602         v2 = kfelast->v2->v ? kfelast->v2->v : kfelast->v1->v;
2603         BLI_assert(v1 != NULL && v2 != NULL);
2604         kfvprev = kfe->v1->v == v1 ? kfe->v1 : kfe->v2;
2605         for (ref = chain->first, i = 0; i < nco && ref != chain->last; ref = ref->next, i++) {
2606                 kfe = ref->ref;
2607                 BLI_assert(kfvprev == kfe->v1 || kfvprev == kfe->v2);
2608                 kfv = kfe->v1 == kfvprev ? kfe->v2 : kfe->v1;
2609                 copy_v3_v3(cos[i], kfv->co);
2610                 kverts[i] = kfv;
2611                 kfvprev = kfv;
2612         }
2613         BLI_assert(i == nco);
2614         l_new = NULL;
2615
2616         if ((l_v1 = BM_face_vert_share_loop(f, v1)) &&
2617             (l_v2 = BM_face_vert_share_loop(f, v2)))
2618         {
2619                 if (nco == 0) {
2620                         /* Want to prevent creating two-sided polygons */
2621                         if (v1 == v2 || BM_edge_exists(v1, v2)) {
2622                                 f_new = NULL;
2623                         }
2624                         else {
2625                                 f_new = BM_face_split(bm, f, l_v1, l_v2, &l_new, NULL, true);
2626                         }
2627                 }
2628                 else {
2629                         f_new = BM_face_split_n(bm, f, l_v1, l_v2, cos, nco, &l_new, NULL);
2630                         if (f_new) {
2631                                 /* Now go through lnew chain matching up chain kv's and assign real v's to them */
2632                                 for (l_iter = l_new->next, i = 0; i < nco; l_iter = l_iter->next, i++) {
2633                                         BLI_assert(equals_v3v3(cos[i], l_iter->v->co));
2634                                         if (kcd->select_result) {
2635                                                 BM_edge_select_set(bm, l_iter->e, true);
2636                                         }
2637                                         kverts[i]->v = l_iter->v;
2638                                 }
2639                         }
2640                 }
2641         }
2642         else {
2643                 f_new = NULL;
2644         }
2645
2646         /* the select chain above doesnt account for the first loop */
2647         if (kcd->select_result) {
2648                 if (l_new) {
2649                         BM_edge_select_set(bm, l_new->e, true);
2650                 }
2651         }
2652         else if (f_new) {
2653                 BM_elem_select_copy(bm, bm, f_new, f);
2654         }
2655
2656         *r_f_new = f_new;
2657 }
2658
2659 static void knife_make_face_cuts(KnifeTool_OpData *kcd, BMFace *f, ListBase *kfedges)
2660 {
2661         BMesh *bm = kcd->em->bm;
2662         KnifeEdge *kfe;
2663         BMFace *fnew, *fnew2, *fhole;
2664         ListBase *chain, *hole, *sidechain;
2665         Ref *ref, *refnext;
2666         int count, oldcount;
2667
2668         oldcount = BLI_listbase_count(kfedges);
2669         while ((chain = find_chain(kcd, kfedges)) != NULL) {
2670                 ListBase fnew_kfedges;
2671                 knife_make_chain_cut(kcd, f, chain, &fnew);
2672                 if (!fnew) {
2673                         return;
2674                 }
2675
2676                 /* Move kfedges to fnew_kfedges if they are now in fnew.
2677                  * The chain edges were removed already */
2678                 BLI_listbase_clear(&fnew_kfedges);
2679                 for (ref = kfedges->first; ref; ref = refnext) {
2680                         kfe = ref->ref;
2681                         refnext = ref->next;
2682                         if (knife_edge_in_face(kfe, fnew)) {
2683                                 BLI_remlink(kfedges, ref);
2684                                 kfe->basef = fnew;
2685                                 BLI_addtail(&fnew_kfedges, ref);
2686                         }
2687                         else if (!knife_edge_in_face(kfe, f)) {
2688                                 /* Concave ngon's - this edge might not be in either faces, T41730 */
2689                                 BLI_remlink(kfedges, ref);
2690                         }
2691                 }
2692                 if (fnew_kfedges.first)
2693                         knife_make_face_cuts(kcd, fnew, &fnew_kfedges);
2694
2695                 /* find_chain should always remove edges if it returns true,
2696                  * but guard against infinite loop anyway */
2697                 count = BLI_listbase_count(kfedges);
2698                 if (count >= oldcount) {
2699                         BLI_assert(!"knife find_chain infinite loop");
2700                         return;
2701                 }
2702                 oldcount = count;
2703         }
2704
2705         while ((hole = find_hole(kcd, kfedges)) != NULL) {
2706                 if (find_hole_chains(kcd, hole, f, &chain, &sidechain)) {
2707                         ListBase fnew_kfedges, fnew2_kfedges;
2708
2709                         /* chain goes across f and sidechain comes back
2710                          * from the second last vertex to the second vertex.
2711                          */
2712                         knife_make_chain_cut(kcd, f, chain, &fnew);
2713                         if (!fnew) {
2714                                 BLI_assert(!"knife failed hole cut");
2715                                 return;
2716                         }
2717                         kfe = ((Ref *)sidechain->first)->ref;
2718                         if (knife_edge_in_face(kfe, f)) {
2719                                 knife_make_chain_cut(kcd, f, sidechain, &fnew2);
2720                                 if (fnew2 == NULL) {
2721                                         return;
2722                                 }
2723                                 fhole = f;
2724                         }
2725                         else if (knife_edge_in_face(kfe, fnew)) {
2726                                 knife_make_chain_cut(kcd, fnew, sidechain, &fnew2);
2727                                 if (fnew2 == NULL) {
2728                                         return;
2729                                 }
2730                                 fhole = fnew2;
2731                         }
2732                         else {
2733                                 /* shouldn't happen except in funny edge cases */
2734                                 return;
2735                         }
2736                         BM_face_kill(bm, fhole);
2737                         /* Move kfedges to either fnew or fnew2 if appropriate.
2738                          * The hole edges were removed already */
2739                         BLI_listbase_clear(&fnew_kfedges);
2740                         BLI_listbase_clear(&fnew2_kfedges);
2741                         for (ref = kfedges->first; ref; ref = refnext) {
2742                                 kfe = ref->ref;
2743                                 refnext = ref->next;
2744                                 if (knife_edge_in_face(kfe, fnew)) {
2745                                         BLI_remlink(kfedges, ref);
2746                                         kfe->basef = fnew;
2747                                         BLI_addtail(&fnew_kfedges, ref);
2748                                 }
2749                                 else if (knife_edge_in_face(kfe, fnew2)) {
2750                                         BLI_remlink(kfedges, ref);
2751                                         kfe->basef = fnew2;
2752                                         BLI_addtail(&fnew2_kfedges, ref);
2753                                 }
2754                         }
2755                         /* We'll skip knife edges that are in the newly formed hole.
2756                          * (Maybe we shouldn't have made a hole in the first place?) */
2757                         if (fnew != fhole && fnew_kfedges.first)
2758                                 knife_make_face_cuts(kcd, fnew, &fnew_kfedges);
2759                         if (fnew2 != fhole && fnew2_kfedges.first)
2760                                 knife_make_face_cuts(kcd, fnew2, &fnew2_kfedges);
2761                         if (f == fhole)
2762                                 break;
2763                         /* find_hole should always remove edges if it returns true,
2764                          * but guard against infinite loop anyway */
2765                         count = BLI_listbase_count(kfedges);
2766                         if (count >= oldcount) {
2767                                 BLI_assert(!"knife find_hole infinite loop");
2768                                 return;
2769                         }
2770                         oldcount = count;
2771                 }
2772         }
2773 }
2774
2775 /* Use the network of KnifeEdges and KnifeVerts accumulated to make real BMVerts and BMEdedges */
2776 static void knife_make_cuts(KnifeTool_OpData *kcd)
2777 {
2778         BMesh *bm = kcd->em->bm;
2779         KnifeEdge *kfe;
2780         KnifeVert *kfv;
2781         BMFace *f;
2782         BMEdge *e, *enew;
2783         ListBase *lst;
2784         Ref *ref;
2785         float pct;
2786         SmallHashIter hiter;
2787         BLI_mempool_iter iter;
2788         SmallHash fhash_, *fhash = &fhash_;
2789         SmallHash ehash_, *ehash = &ehash_;
2790
2791         BLI_smallhash_init(fhash);
2792         BLI_smallhash_init(ehash);
2793
2794         /* put list of cutting edges for a face into fhash, keyed by face */
2795         BLI_mempool_iternew(kcd->kedges, &iter);
2796         for (kfe = BLI_mempool_iterstep(&iter); kfe; kfe = BLI_mempool_iterstep(&iter)) {
2797
2798                 /* select edges that lie directly on the cut */
2799                 if (kcd->select_result) {
2800                         if (kfe->e && kfe->is_cut) {
2801                                 BM_edge_select_set(bm, kfe->e, true);
2802                         }
2803                 }
2804
2805                 f = kfe->basef;
2806                 if (!f || kfe->e)
2807                         continue;
2808                 lst = BLI_smallhash_lookup(fhash, (uintptr_t)f);
2809                 if (!lst) {
2810                         lst = knife_empty_list(kcd);
2811                         BLI_smallhash_insert(fhash, (uintptr_t)f, lst);
2812                 }
2813                 knife_append_list(kcd, lst, kfe);
2814         }
2815
2816         /* put list of splitting vertices for an edge into ehash, keyed by edge */
2817         BLI_mempool_iternew(kcd->kverts, &iter);
2818         for (kfv = BLI_mempool_iterstep(&iter); kfv; kfv = BLI_mempool_iterstep(&iter)) {
2819                 if (kfv->v)
2820                         continue;  /* already have a BMVert */
2821                 for (ref = kfv->edges.first; ref; ref = ref->next) {
2822                         kfe = ref->ref;
2823                         e = kfe->e;
2824                         if (!e)
2825                                 continue;
2826                         lst = BLI_smallhash_lookup(ehash, (uintptr_t)e);
2827                         if (!lst) {
2828                                 lst = knife_empty_list(kcd);
2829                                 BLI_smallhash_insert(ehash, (uintptr_t)e, lst);
2830                         }
2831                         /* there can be more than one kfe in kfv's list with same e */
2832                         if (!find_ref(lst, kfv))
2833                                 knife_append_list(kcd, lst, kfv);
2834                 }
2835         }
2836
2837         /* split bmesh edges where needed */
2838         for (lst = BLI_smallhash_iternew(ehash, &hiter, (uintptr_t *)&e); lst;
2839              lst = BLI_smallhash_iternext(&hiter, (uintptr_t *)&e))
2840         {
2841                 BLI_listbase_sort_r(lst, sort_verts_by_dist_cb, e->v1->co);
2842
2843                 for (ref = lst->first; ref; ref = ref->next) {
2844                         kfv = ref->ref;
2845                         pct = line_point_factor_v3(kfv->co, e->v1->co, e->v2->co);
2846                         kfv->v = BM_edge_split(bm, e, e->v1, &enew, pct);
2847                 }
2848         }
2849
2850         if (kcd->only_select) {
2851                 EDBM_flag_disable_all(kcd->em, BM_ELEM_SELECT);
2852         }
2853
2854         /* do cuts for each face */
2855         for (lst = BLI_smallhash_iternew(fhash, &hiter, (uintptr_t *)&f); lst;
2856              lst = BLI_smallhash_iternext(&hiter, (uintptr_t *)&f))
2857         {
2858                 knife_make_face_cuts(kcd, f, lst);
2859         }
2860
2861         BLI_smallhash_release(fhash);
2862         BLI_smallhash_release(ehash);
2863 }
2864
2865 /* called on tool confirmation */
2866 static void knifetool_finish_ex(KnifeTool_OpData *kcd)
2867 {
2868         knife_make_cuts(kcd);
2869
2870         EDBM_selectmode_flush(kcd->em);
2871         EDBM_mesh_normals_update(kcd->em);
2872         EDBM_update_generic(kcd->em, true, true);
2873
2874         /* re-tessellating makes this invalid, dont use again by accident */
2875         knifetool_free_bmbvh(kcd);
2876 }
2877 static void knifetool_finish(wmOperator *op)
2878 {
2879         KnifeTool_OpData *kcd = op->customdata;
2880         knifetool_finish_ex(kcd);
2881 }
2882
2883 static void knife_recalc_projmat(KnifeTool_OpData *kcd)
2884 {
2885         invert_m4_m4(kcd->ob->imat, kcd->ob->obmat);
2886         ED_view3d_ob_project_mat_get(kcd->ar->regiondata, kcd->ob, kcd->projmat);
2887         invert_m4_m4(kcd->projmat_inv, kcd->projmat);
2888
2889         mul_v3_mat3_m4v3(kcd->proj_zaxis, kcd->ob->imat, kcd->vc.rv3d->viewinv[2]);
2890         normalize_v3(kcd->proj_zaxis);
2891
2892         kcd->is_ortho = ED_view3d_clip_range_get(kcd->vc.v3d, kcd->vc.rv3d,
2893                                                  &kcd->clipsta, &kcd->clipend, true);
2894 }
2895
2896 /* called when modal loop selection is done... */
2897 static void knifetool_exit_ex(bContext *C, KnifeTool_OpData *kcd)
2898 {
2899         if (!kcd)
2900                 return;
2901
2902         if (kcd->is_interactive) {
2903                 WM_cursor_modal_restore(CTX_wm_window(C));
2904
2905                 /* deactivate the extra drawing stuff in 3D-View */
2906                 ED_region_draw_cb_exit(kcd->ar->type, kcd->draw_handle);
2907         }
2908
2909         /* free the custom data */
2910         BLI_mempool_destroy(kcd->refs);
2911         BLI_mempool_destroy(kcd->kverts);
2912         BLI_mempool_destroy(kcd->kedges);
2913
2914         BLI_ghash_free(kcd->origedgemap, NULL, NULL);
2915         BLI_ghash_free(kcd->origvertmap, NULL, NULL);
2916         BLI_ghash_free(kcd->kedgefacemap, NULL, NULL);
2917         BLI_ghash_free(kcd->facetrimap, NULL, NULL);
2918
2919         BLI_memarena_free(kcd->arena);
2920
2921         /* tag for redraw */
2922         ED_region_tag_redraw(kcd->ar);
2923
2924         knifetool_free_bmbvh(kcd);
2925
2926         if (kcd->linehits)
2927                 MEM_freeN(kcd->linehits);
2928
2929         /* destroy kcd itself */
2930         MEM_freeN(kcd);
2931 }
2932 static void knifetool_exit(bContext *C, wmOperator *op)
2933 {
2934         KnifeTool_OpData *kcd = op->customdata;
2935         knifetool_exit_ex(C, kcd);
2936         op->customdata = NULL;
2937 }
2938
2939 static void knifetool_update_mval(KnifeTool_OpData *kcd, const float mval[2])
2940 {
2941         knife_recalc_projmat(kcd);
2942         copy_v2_v2(kcd->mval, mval);
2943
2944         if (knife_update_active(kcd)) {
2945                 ED_region_tag_redraw(kcd->ar);
2946         }
2947 }
2948
2949 static void knifetool_update_mval_i(KnifeTool_OpData *kcd, const int mval_i[2])
2950 {
2951         float mval[2] = {UNPACK2(mval_i)};
2952         knifetool_update_mval(kcd, mval);
2953 }
2954
2955 static void knifetool_init_bmbvh(KnifeTool_OpData *kcd)
2956 {
2957         BM_mesh_elem_index_ensure(kcd->em->bm, BM_VERT);
2958
2959         kcd->cagecos = (const float (*)[3])BKE_editmesh_vertexCos_get(kcd->em, kcd->scene, NULL);
2960
2961         kcd->bmbvh = BKE_bmbvh_new_from_editmesh(
2962                 kcd->em,
2963                 BMBVH_RETURN_ORIG |
2964                 ((kcd->only_select && kcd->cut_through) ? BMBVH_RESPECT_SELECT : BMBVH_RESPECT_HIDDEN),
2965                 kcd->cagecos, false);
2966 }
2967
2968 static void knifetool_free_bmbvh(KnifeTool_OpData *kcd)
2969 {
2970         if (kcd->bmbvh) {
2971                 BKE_bmbvh_free(kcd->bmbvh);
2972                 kcd->bmbvh = NULL;
2973         }
2974
2975         if (kcd->cagecos) {
2976                 MEM_freeN((void *)kcd->cagecos);
2977                 kcd->cagecos = NULL;
2978         }
2979 }
2980
2981 /* called when modal loop selection gets set up... */
2982 static void knifetool_init(bContext *C, KnifeTool_OpData *kcd,
2983                            const bool only_select, const bool cut_through, const bool is_interactive)
2984 {
2985         Scene *scene = CTX_data_scene(C);
2986         Object *obedit = CTX_data_edit_object(C);
2987
2988         /* assign the drawing handle for drawing preview line... */
2989         kcd->scene = scene;
2990         kcd->ob = obedit;
2991         kcd->ar = CTX_wm_region(C);
2992
2993         em_setup_viewcontext(C, &kcd->vc);
2994
2995         kcd->em = BKE_editmesh_from_object(kcd->ob);
2996
2997         /* cut all the way through the mesh if use_occlude_geometry button not pushed */
2998         kcd->is_interactive = is_interactive;
2999         kcd->cut_through = cut_through;
3000         kcd->only_select = only_select;
3001
3002         knifetool_init_bmbvh(kcd);
3003
3004         kcd->arena = BLI_memarena_new(MEM_SIZE_OPTIMAL(1 << 15), "knife");
3005         kcd->vthresh = KMAXDIST - 1;
3006         kcd->ethresh = KMAXDIST;
3007
3008         knife_recalc_projmat(kcd);
3009
3010         ED_region_tag_redraw(kcd->ar);
3011
3012         kcd->refs = BLI_mempool_create(sizeof(Ref), 0, 2048, 0);
3013         kcd->kverts = BLI_mempool_create(sizeof(KnifeVert), 0, 512, BLI_MEMPOOL_ALLOW_ITER);
3014         kcd->kedges = BLI_mempool_create(sizeof(KnifeEdge), 0, 512, BLI_MEMPOOL_ALLOW_ITER);
3015
3016         kcd->origedgemap = BLI_ghash_ptr_new("knife origedgemap");
3017         kcd->origvertmap = BLI_ghash_ptr_new("knife origvertmap");
3018         kcd->kedgefacemap = BLI_ghash_ptr_new("knife kedgefacemap");
3019         kcd->facetrimap = BLI_ghash_ptr_new("knife facetrimap");
3020
3021         /* can't usefully select resulting edges in face mode */
3022         kcd->select_result = (kcd->em->selectmode != SCE_SELECT_FACE);
3023
3024         knife_pos_data_clear(&kcd->curr);
3025         knife_pos_data_clear(&kcd->prev);
3026
3027         if (is_interactive) {
3028                 kcd->draw_handle = ED_region_draw_cb_activate(kcd->ar->type, knifetool_draw, kcd, REGION_DRAW_POST_VIEW);
3029
3030                 knife_init_colors(&kcd->colors);
3031         }
3032 }
3033
3034 static void knifetool_cancel(bContext *C, wmOperator *op)
3035 {
3036         /* this is just a wrapper around exit() */
3037         knifetool_exit(C, op);
3038 }
3039
3040 static int knifetool_invoke(bContext *C, wmOperator *op, const wmEvent *event)
3041 {
3042         const bool only_select = RNA_boolean_get(op->ptr, "only_selected");
3043         const bool cut_through = !RNA_boolean_get(op->ptr, "use_occlude_geometry");
3044
3045         KnifeTool_OpData *kcd;
3046
3047         if (only_select) {
3048                 Object *obedit = CTX_data_edit_object(C);
3049                 BMEditMesh *em = BKE_editmesh_from_object(obedit);
3050                 if (em->bm->totfacesel == 0) {
3051                         BKE_report(op->reports, RPT_ERROR, "Selected faces required");
3052                         return OPERATOR_CANCELLED;
3053                 }
3054         }
3055
3056         view3d_operator_needs_opengl(C);
3057
3058         /* alloc new customdata */
3059         kcd = op->customdata = MEM_callocN(sizeof(KnifeTool_OpData), __func__);
3060
3061         knifetool_init(C, kcd, only_select, cut_through, true);
3062
3063         op->flag |= OP_IS_MODAL_CURSOR_REGION;
3064
3065         /* add a modal handler for this operator - handles loop selection */
3066         WM_cursor_modal_set(CTX_wm_window(C), BC_KNIFECURSOR);
3067         WM_event_add_modal_handler(C, op);
3068
3069         knifetool_update_mval_i(kcd, event->mval);
3070
3071         knife_update_header(C, op, kcd);
3072
3073         return OPERATOR_RUNNING_MODAL;
3074 }
3075
3076 wmKeyMap *knifetool_modal_keymap(wmKeyConfig *keyconf)
3077 {
3078         static EnumPropertyItem modal_items[] = {
3079                 {KNF_MODAL_CANCEL, "CANCEL", 0, "Cancel", ""},
3080                 {KNF_MODAL_CONFIRM, "CONFIRM", 0, "Confirm", ""},
3081                 {KNF_MODAL_MIDPOINT_ON, "SNAP_MIDPOINTS_ON", 0, "Snap To Midpoints On", ""},
3082                 {KNF_MODAL_MIDPOINT_OFF, "SNAP_MIDPOINTS_OFF", 0, "Snap To Midpoints Off", ""},
3083                 {KNF_MODEL_IGNORE_SNAP_ON, "IGNORE_SNAP_ON", 0, "Ignore Snapping On", ""},
3084                 {KNF_MODEL_IGNORE_SNAP_OFF, "IGNORE_SNAP_OFF", 0, "Ignore Snapping Off", ""},
3085                 {KNF_MODAL_ANGLE_SNAP_TOGGLE, "ANGLE_SNAP_TOGGLE", 0, "Toggle Angle Snapping", ""},
3086                 {KNF_MODAL_CUT_THROUGH_TOGGLE, "CUT_THROUGH_TOGGLE", 0, "Toggle Cut Through", ""},
3087                 {KNF_MODAL_NEW_CUT, "NEW_CUT", 0, "End Current Cut", ""},
3088                 {KNF_MODAL_ADD_CUT, "ADD_CUT", 0, "Add Cut", ""},
3089                 {KNF_MODAL_PANNING, "PANNING", 0, "Panning", ""},
3090                 {0, NULL, 0, NULL, NULL}
3091         };
3092
3093         wmKeyMap *keymap = WM_modalkeymap_get(keyconf, "Knife Tool Modal Map");
3094
3095         /* this function is called for each spacetype, only needs to add map once */
3096         if (keymap && keymap->modal_items)
3097                 return NULL;
3098
3099         keymap = WM_modalkeymap_add(keyconf, "Knife Tool Modal Map", modal_items);
3100
3101  &