133344489410661289de3216c72019137aa4b6bf
[blender.git] / source / blender / editors / gpencil / gpencil_edit.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) 2008, Blender Foundation, Joshua Leung
19  * This is a new part of Blender
20  *
21  * Contributor(s): Joshua Leung
22  *
23  * ***** END GPL LICENSE BLOCK *****
24  */
25
26 /** \file blender/editors/gpencil/gpencil_edit.c
27  *  \ingroup edgpencil
28  */
29
30 #include <stdio.h>
31 #include <string.h>
32 #include <stdlib.h>
33 #include <stddef.h>
34 #include <math.h>
35
36 #include "MEM_guardedalloc.h"
37
38 #include "BLI_math.h"
39 #include "BLI_blenlib.h"
40 #include "BLI_rand.h"
41 #include "BLI_utildefines.h"
42
43 #include "BLF_translation.h"
44
45 #include "DNA_anim_types.h"
46 #include "DNA_curve_types.h"
47 #include "DNA_object_types.h"
48 #include "DNA_node_types.h"
49 #include "DNA_scene_types.h"
50 #include "DNA_screen_types.h"
51 #include "DNA_space_types.h"
52 #include "DNA_view3d_types.h"
53 #include "DNA_gpencil_types.h"
54
55 #include "BKE_context.h"
56 #include "BKE_curve.h"
57 #include "BKE_depsgraph.h"
58 #include "BKE_fcurve.h"
59 #include "BKE_global.h"
60 #include "BKE_gpencil.h"
61 #include "BKE_library.h"
62 #include "BKE_object.h"
63 #include "BKE_report.h"
64 #include "BKE_scene.h"
65 #include "BKE_screen.h"
66 #include "BKE_tracking.h"
67
68 #include "UI_interface.h"
69
70 #include "WM_api.h"
71 #include "WM_types.h"
72
73 #include "RNA_access.h"
74 #include "RNA_define.h"
75
76 #include "UI_view2d.h"
77
78 #include "ED_gpencil.h"
79 #include "ED_view3d.h"
80 #include "ED_clip.h"
81 #include "ED_keyframing.h"
82
83 #include "gpencil_intern.h"
84
85
86 /* ************************************************ */
87 /* Context Wrangling... */
88
89 /* Get pointer to active Grease Pencil datablock, and an RNA-pointer to trace back to whatever owns it */
90 bGPdata **ED_gpencil_data_get_pointers(const bContext *C, PointerRNA *ptr)
91 {
92         ID *screen_id = (ID *)CTX_wm_screen(C);
93         Scene *scene = CTX_data_scene(C);
94         ScrArea *sa = CTX_wm_area(C);
95
96         /* if there's an active area, check if the particular editor may
97          * have defined any special Grease Pencil context for editing...
98          */
99         if (sa) {
100                 switch (sa->spacetype) {
101                         case SPACE_VIEW3D: /* 3D-View */
102                         {
103                                 Object *ob = CTX_data_active_object(C);
104
105                                 /* TODO: we can include other data-types such as bones later if need be... */
106
107                                 /* just in case no active/selected object */
108                                 if (ob && (ob->flag & SELECT)) {
109                                         /* for now, as long as there's an object, default to using that in 3D-View */
110                                         if (ptr) RNA_id_pointer_create(&ob->id, ptr);
111                                         return &ob->gpd;
112                                 }
113                                 break;
114                         }
115                         case SPACE_NODE: /* Nodes Editor */
116                         {
117                                 SpaceNode *snode = (SpaceNode *)CTX_wm_space_data(C);
118
119                                 /* return the GP data for the active node block/node */
120                                 if (snode && snode->nodetree) {
121                                         /* for now, as long as there's an active node tree, default to using that in the Nodes Editor */
122                                         if (ptr) RNA_id_pointer_create(&snode->nodetree->id, ptr);
123                                         return &snode->nodetree->gpd;
124                                 }
125
126                                 /* even when there is no node-tree, don't allow this to flow to scene */
127                                 return NULL;
128                         }
129                         case SPACE_SEQ: /* Sequencer */
130                         {
131                                 SpaceSeq *sseq = (SpaceSeq *)CTX_wm_space_data(C);
132
133                                 /* for now, Grease Pencil data is associated with the space (actually preview region only) */
134                                 /* XXX our convention for everything else is to link to data though... */
135                                 if (ptr) RNA_pointer_create(screen_id, &RNA_SpaceSequenceEditor, sseq, ptr);
136                                 return &sseq->gpd;
137                         }
138                         case SPACE_IMAGE: /* Image/UV Editor */
139                         {
140                                 SpaceImage *sima = (SpaceImage *)CTX_wm_space_data(C);
141
142                                 /* for now, Grease Pencil data is associated with the space... */
143                                 /* XXX our convention for everything else is to link to data though... */
144                                 if (ptr) RNA_pointer_create(screen_id, &RNA_SpaceImageEditor, sima, ptr);
145                                 return &sima->gpd;
146                         }
147                         case SPACE_CLIP: /* Nodes Editor */
148                         {
149                                 SpaceClip *sc = (SpaceClip *)CTX_wm_space_data(C);
150                                 MovieClip *clip = ED_space_clip_get_clip(sc);
151
152                                 if (clip) {
153                                         if (sc->gpencil_src == SC_GPENCIL_SRC_TRACK) {
154                                                 MovieTrackingTrack *track = BKE_tracking_track_get_active(&clip->tracking);
155
156                                                 if (!track)
157                                                         return NULL;
158
159                                                 if (ptr)
160                                                         RNA_pointer_create(&clip->id, &RNA_MovieTrackingTrack, track, ptr);
161
162                                                 return &track->gpd;
163                                         }
164                                         else {
165                                                 if (ptr)
166                                                         RNA_id_pointer_create(&clip->id, ptr);
167
168                                                 return &clip->gpd;
169                                         }
170                                 }
171                                 break;
172                         }
173                         default: /* unsupported space */
174                                 return NULL;
175                 }
176         }
177
178         /* just fall back on the scene's GP data */
179         if (ptr) RNA_id_pointer_create((ID *)scene, ptr);
180         return (scene) ? &scene->gpd : NULL;
181 }
182
183 /* Get the active Grease Pencil datablock */
184 bGPdata *ED_gpencil_data_get_active(const bContext *C)
185 {
186         bGPdata **gpd_ptr = ED_gpencil_data_get_pointers(C, NULL);
187         return (gpd_ptr) ? *(gpd_ptr) : NULL;
188 }
189
190 bGPdata *ED_gpencil_data_get_active_v3d(Scene *scene, View3D *v3d)
191 {
192         Base *base = scene->basact;
193         bGPdata *gpd = NULL;
194         /* We have to make sure active object is actually visible and selected, else we must use default scene gpd,
195          * to be consistent with ED_gpencil_data_get_active's behavior.
196          */
197
198         if (base && TESTBASE(v3d, base)) {
199                 gpd = base->object->gpd;
200         }
201         return gpd ? gpd : scene->gpd;
202 }
203
204 /* ************************************************ */
205 /* Panel Operators */
206
207 /* poll callback for adding data/layers - special */
208 static int gp_add_poll(bContext *C)
209 {
210         /* the base line we have is that we have somewhere to add Grease Pencil data */
211         return ED_gpencil_data_get_pointers(C, NULL) != NULL;
212 }
213
214 /* ******************* Add New Data ************************ */
215
216 /* add new datablock - wrapper around API */
217 static int gp_data_add_exec(bContext *C, wmOperator *op)
218 {
219         bGPdata **gpd_ptr = ED_gpencil_data_get_pointers(C, NULL);
220
221         if (gpd_ptr == NULL) {
222                 BKE_report(op->reports, RPT_ERROR, "Nowhere for grease pencil data to go");
223                 return OPERATOR_CANCELLED;
224         }
225         else {
226                 /* decrement user count and add new datablock */
227                 bGPdata *gpd = (*gpd_ptr);
228
229                 id_us_min(&gpd->id);
230                 *gpd_ptr = gpencil_data_addnew(DATA_("GPencil"));
231         }
232
233         /* notifiers */
234         WM_event_add_notifier(C, NC_GPENCIL | ND_DATA | NA_EDITED, NULL);
235
236         return OPERATOR_FINISHED;
237 }
238
239 void GPENCIL_OT_data_add(wmOperatorType *ot)
240 {
241         /* identifiers */
242         ot->name = "Grease Pencil Add New";
243         ot->idname = "GPENCIL_OT_data_add";
244         ot->description = "Add new Grease Pencil datablock";
245         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
246
247         /* callbacks */
248         ot->exec = gp_data_add_exec;
249         ot->poll = gp_add_poll;
250 }
251
252 /* ******************* Unlink Data ************************ */
253
254 /* poll callback for adding data/layers - special */
255 static int gp_data_unlink_poll(bContext *C)
256 {
257         bGPdata **gpd_ptr = ED_gpencil_data_get_pointers(C, NULL);
258
259         /* if we have access to some active data, make sure there's a datablock before enabling this */
260         return (gpd_ptr && *gpd_ptr);
261 }
262
263
264 /* unlink datablock - wrapper around API */
265 static int gp_data_unlink_exec(bContext *C, wmOperator *op)
266 {
267         bGPdata **gpd_ptr = ED_gpencil_data_get_pointers(C, NULL);
268
269         if (gpd_ptr == NULL) {
270                 BKE_report(op->reports, RPT_ERROR, "Nowhere for grease pencil data to go");
271                 return OPERATOR_CANCELLED;
272         }
273         else {
274                 /* just unlink datablock now, decreasing its user count */
275                 bGPdata *gpd = (*gpd_ptr);
276
277                 id_us_min(&gpd->id);
278                 *gpd_ptr = NULL;
279         }
280
281         /* notifiers */
282         WM_event_add_notifier(C, NC_GPENCIL | ND_DATA | NA_EDITED, NULL); 
283
284         return OPERATOR_FINISHED;
285 }
286
287 void GPENCIL_OT_data_unlink(wmOperatorType *ot)
288 {
289         /* identifiers */
290         ot->name = "Grease Pencil Unlink";
291         ot->idname = "GPENCIL_OT_data_unlink";
292         ot->description = "Unlink active Grease Pencil datablock";
293         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
294
295         /* callbacks */
296         ot->exec = gp_data_unlink_exec;
297         ot->poll = gp_data_unlink_poll;
298 }
299
300 /* ******************* Add New Layer ************************ */
301
302 /* add new layer - wrapper around API */
303 static int gp_layer_add_exec(bContext *C, wmOperator *op)
304 {
305         bGPdata **gpd_ptr = ED_gpencil_data_get_pointers(C, NULL);
306
307         /* if there's no existing Grease-Pencil data there, add some */
308         if (gpd_ptr == NULL) {
309                 BKE_report(op->reports, RPT_ERROR, "Nowhere for grease pencil data to go");
310                 return OPERATOR_CANCELLED;
311         }
312         if (*gpd_ptr == NULL)
313                 *gpd_ptr = gpencil_data_addnew(DATA_("GPencil"));
314
315         /* add new layer now */
316         gpencil_layer_addnew(*gpd_ptr, DATA_("GP_Layer"), 1);
317
318         /* notifiers */
319         WM_event_add_notifier(C, NC_GPENCIL | ND_DATA | NA_EDITED, NULL);
320
321         return OPERATOR_FINISHED;
322 }
323
324 void GPENCIL_OT_layer_add(wmOperatorType *ot)
325 {
326         /* identifiers */
327         ot->name = "Add New Layer";
328         ot->idname = "GPENCIL_OT_layer_add";
329         ot->description = "Add new Grease Pencil layer for the active Grease Pencil datablock";
330         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
331
332         /* callbacks */
333         ot->exec = gp_layer_add_exec;
334         ot->poll = gp_add_poll;
335 }
336
337 /* ******************* Delete Active Frame ************************ */
338
339 static int gp_actframe_delete_poll(bContext *C)
340 {
341         bGPdata *gpd = ED_gpencil_data_get_active(C);
342         bGPDlayer *gpl = gpencil_layer_getactive(gpd);
343
344         /* only if there's an active layer with an active frame */
345         return (gpl && gpl->actframe);
346 }
347
348 /* delete active frame - wrapper around API calls */
349 static int gp_actframe_delete_exec(bContext *C, wmOperator *op)
350 {
351         Scene *scene = CTX_data_scene(C);
352         bGPdata *gpd = ED_gpencil_data_get_active(C);
353         bGPDlayer *gpl = gpencil_layer_getactive(gpd);
354         bGPDframe *gpf = gpencil_layer_getframe(gpl, CFRA, 0);
355
356         /* if there's no existing Grease-Pencil data there, add some */
357         if (gpd == NULL) {
358                 BKE_report(op->reports, RPT_ERROR, "No grease pencil data");
359                 return OPERATOR_CANCELLED;
360         }
361         if (ELEM(NULL, gpl, gpf)) {
362                 BKE_report(op->reports, RPT_ERROR, "No active frame to delete");
363                 return OPERATOR_CANCELLED;
364         }
365
366         /* delete it... */
367         gpencil_layer_delframe(gpl, gpf);
368
369         /* notifiers */
370         WM_event_add_notifier(C, NC_GPENCIL | ND_DATA | NA_EDITED, NULL);
371
372         return OPERATOR_FINISHED;
373 }
374
375 void GPENCIL_OT_active_frame_delete(wmOperatorType *ot)
376 {
377         /* identifiers */
378         ot->name = "Delete Active Frame";
379         ot->idname = "GPENCIL_OT_active_frame_delete";
380         ot->description = "Delete the active frame for the active Grease Pencil datablock";
381         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
382
383         /* callbacks */
384         ot->exec = gp_actframe_delete_exec;
385         ot->poll = gp_actframe_delete_poll;
386 }
387
388 /* ************************************************ */
389 /* Grease Pencil to Data Operator */
390
391 /* defines for possible modes */
392 enum {
393         GP_STROKECONVERT_PATH = 1,
394         GP_STROKECONVERT_CURVE,
395         GP_STROKECONVERT_POLY,
396 };
397
398 /* Defines for possible timing modes */
399 enum {
400         GP_STROKECONVERT_TIMING_NONE = 1,
401         GP_STROKECONVERT_TIMING_LINEAR = 2,
402         GP_STROKECONVERT_TIMING_FULL = 3,
403         GP_STROKECONVERT_TIMING_CUSTOMGAP = 4,
404 };
405
406 /* RNA enum define */
407 static EnumPropertyItem prop_gpencil_convertmodes[] = {
408         {GP_STROKECONVERT_PATH, "PATH", 0, "Path", ""},
409         {GP_STROKECONVERT_CURVE, "CURVE", 0, "Bezier Curve", ""},
410         {GP_STROKECONVERT_POLY, "POLY", 0, "Polygon Curve", ""},
411         {0, NULL, 0, NULL, NULL}
412 };
413
414 static EnumPropertyItem prop_gpencil_convert_timingmodes_restricted[] = {
415         {GP_STROKECONVERT_TIMING_NONE, "NONE", 0, "No Timing", "Ignore timing"},
416         {GP_STROKECONVERT_TIMING_LINEAR, "LINEAR", 0, "Linear", "Simple linear timing"},
417         {0, NULL, 0, NULL, NULL},
418 };
419
420 static EnumPropertyItem prop_gpencil_convert_timingmodes[] = {
421         {GP_STROKECONVERT_TIMING_NONE, "NONE", 0, "No Timing", "Ignore timing"},
422         {GP_STROKECONVERT_TIMING_LINEAR, "LINEAR", 0, "Linear", "Simple linear timing"},
423         {GP_STROKECONVERT_TIMING_FULL, "FULL", 0, "Original", "Use the original timing, gaps included"},
424         {GP_STROKECONVERT_TIMING_CUSTOMGAP, "CUSTOMGAP", 0, "Custom Gaps",
425                                             "Use the original timing, but with custom gap lengths (in frames)"},
426         {0, NULL, 0, NULL, NULL},
427 };
428
429 static EnumPropertyItem *rna_GPConvert_mode_items(bContext *UNUSED(C), PointerRNA *ptr, PropertyRNA *UNUSED(prop),
430                                                   bool *UNUSED(r_free))
431 {
432         if (RNA_boolean_get(ptr, "use_timing_data")) {
433                 return prop_gpencil_convert_timingmodes;
434         }
435         return prop_gpencil_convert_timingmodes_restricted;
436 }
437
438 /* --- */
439
440 /* convert the coordinates from the given stroke point into 3d-coordinates 
441  *      - assumes that the active space is the 3D-View
442  */
443 static void gp_strokepoint_convertcoords(bContext *C, bGPDstroke *gps, bGPDspoint *pt, float p3d[3], rctf *subrect)
444 {
445         Scene *scene = CTX_data_scene(C);
446         View3D *v3d = CTX_wm_view3d(C);
447         ARegion *ar = CTX_wm_region(C);
448
449         if (gps->flag & GP_STROKE_3DSPACE) {
450                 /* directly use 3d-coordinates */
451                 copy_v3_v3(p3d, &pt->x);
452         }
453         else {
454                 const float *fp = ED_view3d_cursor3d_get(scene, v3d);
455                 float mvalf[2];
456
457                 /* get screen coordinate */
458                 if (gps->flag & GP_STROKE_2DSPACE) {
459                         View2D *v2d = &ar->v2d;
460                         UI_view2d_view_to_region_fl(v2d, pt->x, pt->y, &mvalf[0], &mvalf[1]);
461                 }
462                 else {
463                         if (subrect) {
464                                 mvalf[0] = (((float)pt->x / 100.0f) * BLI_rctf_size_x(subrect)) + subrect->xmin;
465                                 mvalf[1] = (((float)pt->y / 100.0f) * BLI_rctf_size_y(subrect)) + subrect->ymin;
466                         }
467                         else {
468                                 mvalf[0] = (float)pt->x / 100.0f * ar->winx;
469                                 mvalf[1] = (float)pt->y / 100.0f * ar->winy;
470                         }
471                 }
472
473                 ED_view3d_win_to_3d(ar, fp, mvalf, p3d);
474         }
475 }
476
477 /* --- */
478
479 /* temp struct for gp_stroke_path_animation() */
480 typedef struct tGpTimingData {
481         /* Data set from operator settings */
482         int mode;
483         int frame_range; /* Number of frames evaluated for path animation */
484         int start_frame, end_frame;
485         bool realtime; /* Will overwrite end_frame in case of Original or CustomGap timing... */
486         float gap_duration, gap_randomness; /* To be used with CustomGap mode*/
487         int seed;
488
489         /* Data set from points, used to compute final timing FCurve */
490         int num_points, cur_point;
491
492         /* Distances */
493         float *dists;
494         float tot_dist;
495
496         /* Times */
497         float *times; /* Note: Gap times will be negative! */
498         float tot_time, gap_tot_time;
499         double inittime;
500
501         /* Only used during creation of dists & times lists. */
502         float offset_time;
503 } tGpTimingData;
504
505 /* Init point buffers for timing data.
506  * Note this assumes we only grow those arrays!
507  */
508 static void gp_timing_data_set_nbr(tGpTimingData *gtd, const int nbr)
509 {
510         float *tmp;
511
512         BLI_assert(nbr > gtd->num_points);
513
514         /* distances */
515         tmp = gtd->dists;
516         gtd->dists = MEM_callocN(sizeof(float) * nbr, __func__);
517         if (tmp) {
518                 memcpy(gtd->dists, tmp, sizeof(float) * gtd->num_points);
519                 MEM_freeN(tmp);
520         }
521
522         /* times */
523         tmp = gtd->times;
524         gtd->times = MEM_callocN(sizeof(float) * nbr, __func__);
525         if (tmp) {
526                 memcpy(gtd->times, tmp, sizeof(float) * gtd->num_points);
527                 MEM_freeN(tmp);
528         }
529
530         gtd->num_points = nbr;
531 }
532
533 /* add stroke point to timing buffers */
534 static void gp_timing_data_add_point(tGpTimingData *gtd, const double stroke_inittime, const float time,
535                                      const float delta_dist)
536 {
537         float delta_time = 0.0f;
538         const int cur_point = gtd->cur_point;
539
540         if (!cur_point) {
541                 /* Special case, first point, if time is not 0.0f we have to compensate! */
542                 gtd->offset_time = -time;
543                 gtd->times[cur_point] = 0.0f;
544         }
545         else if (time < 0.0f) {
546                 /* This is a gap, negative value! */
547                 gtd->times[cur_point] = -(((float)(stroke_inittime - gtd->inittime)) + time + gtd->offset_time);
548                 delta_time = -gtd->times[cur_point] - gtd->times[cur_point - 1];
549
550                 gtd->gap_tot_time += delta_time;
551         }
552         else {
553                 gtd->times[cur_point] = (((float)(stroke_inittime - gtd->inittime)) + time + gtd->offset_time);
554                 delta_time = gtd->times[cur_point] - fabsf(gtd->times[cur_point - 1]);
555         }
556
557         gtd->tot_time += delta_time;
558         gtd->tot_dist += delta_dist;
559         gtd->dists[cur_point] = gtd->tot_dist;
560
561         gtd->cur_point++;
562 }
563
564 /* In frames! Binary search for FCurve keys have a threshold of 0.01, so we can't set
565  * arbitrarily close points - this is esp. important with NoGaps mode!
566  */
567 #define MIN_TIME_DELTA 0.02f
568
569 /* Loop over next points to find the end of the stroke, and compute */
570 static int gp_find_end_of_stroke_idx(tGpTimingData *gtd, RNG *rng, const int idx, const int nbr_gaps,
571                                      int *nbr_done_gaps, const float tot_gaps_time, const float delta_time,
572                                      float *next_delta_time)
573 {
574         int j;
575
576         for (j = idx + 1; j < gtd->num_points; j++) {
577                 if (gtd->times[j] < 0) {
578                         gtd->times[j] = -gtd->times[j];
579                         if (gtd->mode == GP_STROKECONVERT_TIMING_CUSTOMGAP) {
580                                 /* In this mode, gap time between this stroke and the next should be 0 currently...
581                                  * So we have to compute its final duration!
582                                  */
583                                 if (gtd->gap_randomness > 0.0f) {
584                                         /* We want gaps that are in gtd->gap_duration +/- gtd->gap_randomness range,
585                                          * and which sum to exactly tot_gaps_time...
586                                          */
587                                         int rem_gaps = nbr_gaps - (*nbr_done_gaps);
588                                         if (rem_gaps < 2) {
589                                                 /* Last gap, just give remaining time! */
590                                                 *next_delta_time = tot_gaps_time;
591                                         }
592                                         else {
593                                                 float delta, min, max;
594
595                                                 /* This code ensures that if the first gaps have been shorter than average gap_duration,
596                                                  * next gaps will tend to be longer (i.e. try to recover the lateness), and vice-versa!
597                                                  */
598                                                 delta = delta_time - (gtd->gap_duration * (*nbr_done_gaps));
599
600                                                 /* Clamp min between [-gap_randomness, 0.0], with lower delta giving higher min */
601                                                 min = -gtd->gap_randomness - delta;
602                                                 CLAMP(min, -gtd->gap_randomness, 0.0f);
603
604                                                 /* Clamp max between [0.0, gap_randomness], with lower delta giving higher max */
605                                                 max = gtd->gap_randomness - delta;
606                                                 CLAMP(max, 0.0f, gtd->gap_randomness);
607                                                 *next_delta_time += gtd->gap_duration + (BLI_rng_get_float(rng) * (max - min)) + min;
608                                         }
609                                 }
610                                 else {
611                                         *next_delta_time += gtd->gap_duration;
612                                 }
613                         }
614                         (*nbr_done_gaps)++;
615                         break;
616                 }
617         }
618
619         return j - 1;
620 }
621
622 static void gp_stroke_path_animation_preprocess_gaps(tGpTimingData *gtd, RNG *rng, int *nbr_gaps, float *tot_gaps_time)
623 {
624         int i;
625         float delta_time = 0.0f;
626
627         for (i = 0; i < gtd->num_points; i++) {
628                 if (gtd->times[i] < 0 && i) {
629                         (*nbr_gaps)++;
630                         gtd->times[i] = -gtd->times[i] - delta_time;
631                         delta_time += gtd->times[i] - gtd->times[i - 1];
632                         gtd->times[i] = -gtd->times[i - 1]; /* Temp marker, values *have* to be different! */
633                 }
634                 else {
635                         gtd->times[i] -= delta_time;
636                 }
637         }
638         gtd->tot_time -= delta_time;
639
640         *tot_gaps_time = (float)(*nbr_gaps) * gtd->gap_duration;
641         gtd->tot_time += *tot_gaps_time;
642         if (G.debug & G_DEBUG) {
643                 printf("%f, %f, %f, %d\n", gtd->tot_time, delta_time, *tot_gaps_time, *nbr_gaps);
644         }
645         if (gtd->gap_randomness > 0.0f) {
646                 BLI_rng_srandom(rng, gtd->seed);
647         }
648 }
649
650 static void gp_stroke_path_animation_add_keyframes(ReportList *reports, PointerRNA ptr, PropertyRNA *prop, FCurve *fcu,
651                                                    Curve *cu, tGpTimingData *gtd, RNG *rng, const float time_range,
652                                                    const int nbr_gaps, const float tot_gaps_time)
653 {
654         /* Use actual recorded timing! */
655         const float time_start = (float)gtd->start_frame;
656
657         float last_valid_time = 0.0f;
658         int end_stroke_idx = -1, start_stroke_idx = 0;
659         float end_stroke_time = 0.0f;
660
661         /* CustomGaps specific */
662         float delta_time = 0.0f, next_delta_time = 0.0f;
663         int nbr_done_gaps = 0;
664
665         int i;
666         float cfra;
667
668         /* This is a bit tricky, as:
669          * - We can't add arbitrarily close points on FCurve (in time).
670          * - We *must* have all "caps" points of all strokes in FCurve, as much as possible!
671          */
672         for (i = 0; i < gtd->num_points; i++) {
673                 /* If new stroke... */
674                 if (i > end_stroke_idx) {
675                         start_stroke_idx = i;
676                         delta_time = next_delta_time;
677                         /* find end of that new stroke */
678                         end_stroke_idx = gp_find_end_of_stroke_idx(gtd, rng, i, nbr_gaps, &nbr_done_gaps,
679                                                                    tot_gaps_time, delta_time, &next_delta_time);
680                         /* This one should *never* be negative! */
681                         end_stroke_time = time_start + ((gtd->times[end_stroke_idx] + delta_time) / gtd->tot_time * time_range);
682                 }
683
684                 /* Simple proportional stuff... */
685                 cu->ctime = gtd->dists[i] / gtd->tot_dist * cu->pathlen;
686                 cfra = time_start + ((gtd->times[i] + delta_time) / gtd->tot_time * time_range);
687
688                 /* And now, the checks about timing... */
689                 if (i == start_stroke_idx) {
690                         /* If first point of a stroke, be sure it's enough ahead of last valid keyframe, and
691                          * that the end point of the stroke is far enough!
692                          * In case it is not, we keep the end point...
693                          * Note that with CustomGaps mode, this is here we set the actual gap timing!
694                          */
695                         if ((end_stroke_time - last_valid_time) > MIN_TIME_DELTA * 2) {
696                                 if ((cfra - last_valid_time) < MIN_TIME_DELTA) {
697                                         cfra = last_valid_time + MIN_TIME_DELTA;
698                                 }
699                                 insert_keyframe_direct(reports, ptr, prop, fcu, cfra, INSERTKEY_FAST);
700                                 last_valid_time = cfra;
701                         }
702                         else if (G.debug & G_DEBUG) {
703                                 printf("\t Skipping start point %d, too close from end point %d\n", i, end_stroke_idx);
704                         }
705                 }
706                 else if (i == end_stroke_idx) {
707                         /* Always try to insert end point of a curve (should be safe enough, anyway...) */
708                         if ((cfra - last_valid_time) < MIN_TIME_DELTA) {
709                                 cfra = last_valid_time + MIN_TIME_DELTA;
710                         }
711                         insert_keyframe_direct(reports, ptr, prop, fcu, cfra, INSERTKEY_FAST);
712                         last_valid_time = cfra;
713                 }
714                 else {
715                         /* Else ("middle" point), we only insert it if it's far enough from last keyframe,
716                          * and also far enough from (not yet added!) end_stroke keyframe!
717                          */
718                         if ((cfra - last_valid_time) > MIN_TIME_DELTA && (end_stroke_time - cfra) > MIN_TIME_DELTA) {
719                                 insert_keyframe_direct(reports, ptr, prop, fcu, cfra, INSERTKEY_FAST);
720                                 last_valid_time = cfra;
721                         }
722                         else if (G.debug & G_DEBUG) {
723                                 printf("\t Skipping \"middle\" point %d, too close from last added point or end point %d\n",
724                                        i, end_stroke_idx);
725                         }
726                 }
727         }
728 }
729
730 static void gp_stroke_path_animation(bContext *C, ReportList *reports, Curve *cu, tGpTimingData *gtd)
731 {
732         Scene *scene = CTX_data_scene(C);
733         bAction *act;
734         FCurve *fcu;
735         PointerRNA ptr;
736         PropertyRNA *prop = NULL;
737         int nbr_gaps = 0, i;
738
739         if (gtd->mode == GP_STROKECONVERT_TIMING_NONE)
740                 return;
741
742         /* gap_duration and gap_randomness are in frames, but we need seconds!!! */
743         gtd->gap_duration = FRA2TIME(gtd->gap_duration);
744         gtd->gap_randomness = FRA2TIME(gtd->gap_randomness);
745
746         /* Enable path! */
747         cu->flag |= CU_PATH;
748         cu->pathlen = gtd->frame_range;
749
750         /* Get RNA pointer to read/write path time values */
751         RNA_id_pointer_create((ID *)cu, &ptr);
752         prop = RNA_struct_find_property(&ptr, "eval_time");
753
754         /* Ensure we have an F-Curve to add keyframes to */
755         act = verify_adt_action((ID *)cu, true);
756         fcu = verify_fcurve(act, NULL, &ptr, "eval_time", 0, true);
757
758         if (G.debug & G_DEBUG) {
759                 printf("%s: tot len: %f\t\ttot time: %f\n", __func__, gtd->tot_dist, gtd->tot_time);
760                 for (i = 0; i < gtd->num_points; i++) {
761                         printf("\tpoint %d:\t\tlen: %f\t\ttime: %f\n", i, gtd->dists[i], gtd->times[i]);
762                 }
763         }
764
765         if (gtd->mode == GP_STROKECONVERT_TIMING_LINEAR) {
766                 float cfra;
767
768                 /* Linear extrapolation! */
769                 fcu->extend = FCURVE_EXTRAPOLATE_LINEAR;
770
771                 cu->ctime = 0.0f;
772                 cfra = (float)gtd->start_frame;
773                 insert_keyframe_direct(reports, ptr, prop, fcu, cfra, INSERTKEY_FAST);
774
775                 cu->ctime = cu->pathlen;
776                 if (gtd->realtime) {
777                         cfra += (float)TIME2FRA(gtd->tot_time); /* Seconds to frames */
778                 }
779                 else {
780                         cfra = (float)gtd->end_frame;
781                 }
782                 insert_keyframe_direct(reports, ptr, prop, fcu, cfra, INSERTKEY_FAST);
783         }
784         else {
785                 /* Use actual recorded timing! */
786                 RNG *rng = BLI_rng_new(0);
787                 float time_range;
788
789                 /* CustomGaps specific */
790                 float tot_gaps_time = 0.0f;
791
792                 /* Pre-process gaps, in case we don't want to keep their original timing */
793                 if (gtd->mode == GP_STROKECONVERT_TIMING_CUSTOMGAP) {
794                         gp_stroke_path_animation_preprocess_gaps(gtd, rng, &nbr_gaps, &tot_gaps_time);
795                 }
796
797                 if (gtd->realtime) {
798                         time_range = (float)TIME2FRA(gtd->tot_time); /* Seconds to frames */
799                 }
800                 else {
801                         time_range = (float)(gtd->end_frame - gtd->start_frame);
802                 }
803
804                 if (G.debug & G_DEBUG) {
805                         printf("GP Stroke Path Conversion: Starting keying!\n");
806                 }
807
808                 gp_stroke_path_animation_add_keyframes(reports, ptr, prop, fcu, cu, gtd, rng, time_range,
809                                                        nbr_gaps, tot_gaps_time);
810
811                 BLI_rng_free(rng);
812         }
813
814         /* As we used INSERTKEY_FAST mode, we need to recompute all curve's handles now */
815         calchandles_fcurve(fcu);
816
817         if (G.debug & G_DEBUG) {
818                 printf("%s: \ntot len: %f\t\ttot time: %f\n", __func__, gtd->tot_dist, gtd->tot_time);
819                 for (i = 0; i < gtd->num_points; i++) {
820                         printf("\tpoint %d:\t\tlen: %f\t\ttime: %f\n", i, gtd->dists[i], gtd->times[i]);
821                 }
822                 printf("\n\n");
823         }
824
825         WM_event_add_notifier(C, NC_ANIMATION | ND_KEYFRAME | NA_EDITED, NULL);
826
827         /* send updates */
828         DAG_id_tag_update(&cu->id, 0);
829 }
830
831 #undef MIN_TIME_DELTA
832
833 #define GAP_DFAC 0.01f
834 #define WIDTH_CORR_FAC 0.1f
835 #define BEZT_HANDLE_FAC 0.3f
836
837 /* convert stroke to 3d path */
838
839 /* helper */
840 static void gp_stroke_to_path_add_point(tGpTimingData *gtd, BPoint *bp, const float p[3], const float prev_p[3],
841                                         const bool do_gtd, const double inittime, const float time,
842                                         const float width, const float rad_fac, float minmax_weights[2])
843 {
844         copy_v3_v3(bp->vec, p);
845         bp->vec[3] = 1.0f;
846
847         /* set settings */
848         bp->f1 = SELECT;
849         bp->radius = width * rad_fac;
850         bp->weight = width;
851         CLAMP(bp->weight, 0.0f, 1.0f);
852         if (bp->weight < minmax_weights[0]) {
853                 minmax_weights[0] = bp->weight;
854         }
855         else if (bp->weight > minmax_weights[1]) {
856                 minmax_weights[1] = bp->weight;
857         }
858
859         /* Update timing data */
860         if (do_gtd) {
861                 gp_timing_data_add_point(gtd, inittime, time, len_v3v3(prev_p, p));
862         }
863 }
864
865 static void gp_stroke_to_path(bContext *C, bGPDlayer *gpl, bGPDstroke *gps, Curve *cu, rctf *subrect, Nurb **curnu,
866                               float minmax_weights[2], const float rad_fac, bool stitch, const bool add_start_point,
867                               const bool add_end_point, tGpTimingData *gtd)
868 {
869         bGPDspoint *pt;
870         Nurb *nu = (curnu) ? *curnu : NULL;
871         BPoint *bp, *prev_bp = NULL;
872         const bool do_gtd = (gtd->mode != GP_STROKECONVERT_TIMING_NONE);
873         const int add_start_end_points = (add_start_point ? 1 : 0) + (add_end_point ? 1 : 0);
874         int i, old_nbp = 0;
875
876         /* create new 'nurb' or extend current one within the curve */
877         if (nu) {
878                 old_nbp = nu->pntsu;
879
880                 /* If stitch, the first point of this stroke is already present in current nu.
881                  * Else, we have to add two additional points to make the zero-radius link between strokes.
882                  */
883                 BKE_nurb_points_add(nu, gps->totpoints + (stitch ? -1 : 2) + add_start_end_points);
884         }
885         else {
886                 nu = (Nurb *)MEM_callocN(sizeof(Nurb), "gpstroke_to_path(nurb)");
887
888                 nu->pntsu = gps->totpoints + add_start_end_points;
889                 nu->pntsv = 1;
890                 nu->orderu = 2; /* point-to-point! */
891                 nu->type = CU_NURBS;
892                 nu->flagu = CU_NURB_ENDPOINT;
893                 nu->resolu = cu->resolu;
894                 nu->resolv = cu->resolv;
895                 nu->knotsu = NULL;
896
897                 nu->bp = (BPoint *)MEM_callocN(sizeof(BPoint) * nu->pntsu, "bpoints");
898
899                 stitch = false; /* Security! */
900         }
901
902         if (do_gtd) {
903                 gp_timing_data_set_nbr(gtd, nu->pntsu);
904         }
905
906         /* If needed, make the link between both strokes with two zero-radius additional points */
907         /* About "zero-radius" point interpolations:
908          * - If we have at least two points in current curve (most common case), we linearly extrapolate
909          *   the last segment to get the first point (p1) position and timing.
910          * - If we do not have those (quite odd, but may happen), we linearly interpolate the last point
911          *   with the first point of the current stroke.
912          * The same goes for the second point, first segment of the current stroke is "negatively" extrapolated
913          * if it exists, else (if the stroke is a single point), linear interpolation with last curve point...
914          */
915         if (curnu && !stitch && old_nbp) {
916                 float p1[3], p2[3], p[3], next_p[3];
917                 float dt1 = 0.0f, dt2 = 0.0f;
918
919                 BLI_assert(gps->prev != NULL);
920
921                 prev_bp = NULL;
922                 if ((old_nbp > 1) && (gps->prev->totpoints > 1)) {
923                         /* Only use last curve segment if previous stroke was not a single-point one! */
924                         prev_bp = &nu->bp[old_nbp - 2];
925                 }
926                 bp = &nu->bp[old_nbp - 1];
927
928                 /* First point */
929                 gp_strokepoint_convertcoords(C, gps, gps->points, p, subrect);
930                 if (prev_bp) {
931                         interp_v3_v3v3(p1, bp->vec, prev_bp->vec, -GAP_DFAC);
932                         if (do_gtd) {
933                                 const int idx = gps->prev->totpoints - 1;
934                                 dt1 = interpf(gps->prev->points[idx - 1].time, gps->prev->points[idx].time, -GAP_DFAC);
935                         }
936                 }
937                 else {
938                         interp_v3_v3v3(p1, bp->vec, p, GAP_DFAC);
939                         if (do_gtd) {
940                                 dt1 = interpf(gps->inittime - gps->prev->inittime, 0.0f, GAP_DFAC);
941                         }
942                 }
943                 bp++;
944                 gp_stroke_to_path_add_point(gtd, bp, p1, (bp - 1)->vec, do_gtd, gps->prev->inittime, dt1,
945                                             0.0f, rad_fac, minmax_weights);
946
947                 /* Second point */
948                 /* Note dt2 is always negative, which marks the gap. */
949                 if (gps->totpoints > 1) {
950                         gp_strokepoint_convertcoords(C, gps, gps->points + 1, next_p, subrect);
951                         interp_v3_v3v3(p2, p, next_p, -GAP_DFAC);
952                         if (do_gtd) {
953                                 dt2 = interpf(gps->points[1].time, gps->points[0].time, -GAP_DFAC);
954                         }
955                 }
956                 else {
957                         interp_v3_v3v3(p2, p, bp->vec, GAP_DFAC);
958                         if (do_gtd) {
959                                 dt2 = interpf(gps->prev->inittime - gps->inittime, 0.0f, GAP_DFAC);
960                         }
961                 }
962                 bp++;
963                 gp_stroke_to_path_add_point(gtd, bp, p2, p1, do_gtd, gps->inittime, dt2, 0.0f, rad_fac, minmax_weights);
964
965                 old_nbp += 2;
966         }
967         else if (add_start_point) {
968                 float p[3], next_p[3];
969                 float dt = 0.0f;
970
971                 gp_strokepoint_convertcoords(C, gps, gps->points, p, subrect);
972                 if (gps->totpoints > 1) {
973                         gp_strokepoint_convertcoords(C, gps, gps->points + 1, next_p, subrect);
974                         interp_v3_v3v3(p, p, next_p, -GAP_DFAC);
975                         if (do_gtd) {
976                                 dt = interpf(gps->points[1].time, gps->points[0].time, -GAP_DFAC);
977                         }
978                 }
979                 else {
980                         p[0] -= GAP_DFAC;  /* Rather arbitrary... */
981                         dt = -GAP_DFAC;  /* Rather arbitrary too! */
982                 }
983                 bp = &nu->bp[old_nbp];
984                 /* Note we can't give anything else than 0.0 as time here, since a negative one (which would be expected value)
985                  * would not work (it would be *before* gtd->inittime, which is not supported currently).
986                  */
987                 gp_stroke_to_path_add_point(gtd, bp, p, p, do_gtd, gps->inittime, dt, 0.0f, rad_fac, minmax_weights);
988
989                 old_nbp++;
990         }
991
992         if (old_nbp) {
993                 prev_bp = &nu->bp[old_nbp - 1];
994         }
995
996         /* add points */
997         for (i = (stitch) ? 1 : 0, pt = &gps->points[(stitch) ? 1 : 0], bp = &nu->bp[old_nbp];
998              i < gps->totpoints;
999              i++, pt++, bp++)
1000         {
1001                 float p[3];
1002                 float width = pt->pressure * gpl->thickness * WIDTH_CORR_FAC;
1003
1004                 /* get coordinates to add at */
1005                 gp_strokepoint_convertcoords(C, gps, pt, p, subrect);
1006
1007                 gp_stroke_to_path_add_point(gtd, bp, p, (prev_bp) ? prev_bp->vec : p, do_gtd, gps->inittime, pt->time,
1008                                             width, rad_fac, minmax_weights);
1009
1010                 prev_bp = bp;
1011         }
1012
1013         if (add_end_point) {
1014                 float p[3];
1015                 float dt = 0.0f;
1016
1017                 if (gps->totpoints > 1) {
1018                         interp_v3_v3v3(p, prev_bp->vec, (prev_bp - 1)->vec, -GAP_DFAC);
1019                         if (do_gtd) {
1020                                 const int idx = gps->totpoints - 1;
1021                                 dt = interpf(gps->points[idx - 1].time, gps->points[idx].time, -GAP_DFAC);
1022                         }
1023                 }
1024                 else {
1025                         copy_v3_v3(p, prev_bp->vec);
1026                         p[0] += GAP_DFAC;  /* Rather arbitrary... */
1027                         dt = GAP_DFAC;  /* Rather arbitrary too! */
1028                 }
1029                 /* Note bp has already been incremented in main loop above, so it points to the right place. */
1030                 gp_stroke_to_path_add_point(gtd, bp, p, prev_bp->vec, do_gtd, gps->inittime, dt, 0.0f, rad_fac, minmax_weights);
1031         }
1032
1033         /* add nurb to curve */
1034         if (!curnu || !*curnu) {
1035                 BLI_addtail(&cu->nurb, nu);
1036         }
1037         if (curnu) {
1038                 *curnu = nu;
1039         }
1040
1041         BKE_nurb_knot_calc_u(nu);
1042 }
1043
1044 /* convert stroke to 3d bezier */
1045
1046 /* helper */
1047 static void gp_stroke_to_bezier_add_point(tGpTimingData *gtd, BezTriple *bezt,
1048                                           const float p[3], const float h1[3], const float h2[3], const float prev_p[3],
1049                                           const bool do_gtd, const double inittime, const float time,
1050                                           const float width, const float rad_fac, float minmax_weights[2])
1051 {
1052         copy_v3_v3(bezt->vec[0], h1);
1053         copy_v3_v3(bezt->vec[1], p);
1054         copy_v3_v3(bezt->vec[2], h2);
1055
1056         /* set settings */
1057         bezt->h1 = bezt->h2 = HD_FREE;
1058         bezt->f1 = bezt->f2 = bezt->f3 = SELECT;
1059         bezt->radius = width * rad_fac;
1060         bezt->weight = width;
1061         CLAMP(bezt->weight, 0.0f, 1.0f);
1062         if (bezt->weight < minmax_weights[0]) {
1063                 minmax_weights[0] = bezt->weight;
1064         }
1065         else if (bezt->weight > minmax_weights[1]) {
1066                 minmax_weights[1] = bezt->weight;
1067         }
1068
1069         /* Update timing data */
1070         if (do_gtd) {
1071                 gp_timing_data_add_point(gtd, inittime, time, len_v3v3(prev_p, p));
1072         }
1073 }
1074
1075 static void gp_stroke_to_bezier(bContext *C, bGPDlayer *gpl, bGPDstroke *gps, Curve *cu, rctf *subrect, Nurb **curnu,
1076                               float minmax_weights[2], const float rad_fac, bool stitch, const bool add_start_point,
1077                               const bool add_end_point, tGpTimingData *gtd)
1078 {
1079         bGPDspoint *pt;
1080         Nurb *nu = (curnu) ? *curnu : NULL;
1081         BezTriple *bezt, *prev_bezt = NULL;
1082         int i, tot, old_nbezt = 0;
1083         const int add_start_end_points = (add_start_point ? 1 : 0) + (add_end_point ? 1 : 0);
1084         float p3d_cur[3], p3d_prev[3], p3d_next[3], h1[3], h2[3];
1085         const bool do_gtd = (gtd->mode != GP_STROKECONVERT_TIMING_NONE);
1086
1087         /* create new 'nurb' or extend current one within the curve */
1088         if (nu) {
1089                 old_nbezt = nu->pntsu;
1090                 /* If we do stitch, first point of current stroke is assumed the same as last point of previous stroke,
1091                  * so no need to add it.
1092                  * If no stitch, we want to add two additional points to make a "zero-radius" link between both strokes.
1093                  */
1094                 BKE_nurb_bezierPoints_add(nu, gps->totpoints + ((stitch) ? -1 : 2) + add_start_end_points);
1095         }
1096         else {
1097                 nu = (Nurb *)MEM_callocN(sizeof(Nurb), "gpstroke_to_bezier(nurb)");
1098
1099                 nu->pntsu = gps->totpoints + add_start_end_points;
1100                 nu->resolu = 12;
1101                 nu->resolv = 12;
1102                 nu->type = CU_BEZIER;
1103                 nu->bezt = (BezTriple *)MEM_callocN(sizeof(BezTriple) * nu->pntsu, "bezts");
1104
1105                 stitch = false; /* Security! */
1106         }
1107
1108         if (do_gtd) {
1109                 gp_timing_data_set_nbr(gtd, nu->pntsu);
1110         }
1111
1112         tot = gps->totpoints;
1113
1114         /* get initial coordinates */
1115         pt = gps->points;
1116         if (tot) {
1117                 gp_strokepoint_convertcoords(C, gps, pt, (stitch) ? p3d_prev : p3d_cur, subrect);
1118                 if (tot > 1) {
1119                         gp_strokepoint_convertcoords(C, gps, pt + 1, (stitch) ? p3d_cur : p3d_next, subrect);
1120                 }
1121                 if (stitch && tot > 2) {
1122                         gp_strokepoint_convertcoords(C, gps, pt + 2, p3d_next, subrect);
1123                 }
1124         }
1125
1126         /* If needed, make the link between both strokes with two zero-radius additional points */
1127         if (curnu && old_nbezt) {
1128                 BLI_assert(gps->prev != NULL);
1129
1130                 /* Update last point's second handle */
1131                 if (stitch) {
1132                         bezt = &nu->bezt[old_nbezt - 1];
1133                         interp_v3_v3v3(h2, bezt->vec[1], p3d_cur, BEZT_HANDLE_FAC);
1134                         copy_v3_v3(bezt->vec[2], h2);
1135                         pt++;
1136                 }
1137
1138                 /* Create "link points" */
1139                 /* About "zero-radius" point interpolations:
1140                  * - If we have at least two points in current curve (most common case), we linearly extrapolate
1141                  *   the last segment to get the first point (p1) position and timing.
1142                  * - If we do not have those (quite odd, but may happen), we linearly interpolate the last point
1143                  *   with the first point of the current stroke.
1144                  * The same goes for the second point, first segment of the current stroke is "negatively" extrapolated
1145                  * if it exists, else (if the stroke is a single point), linear interpolation with last curve point...
1146                  */
1147                 else {
1148                         float p1[3], p2[3];
1149                         float dt1 = 0.0f, dt2 = 0.0f;
1150
1151                         prev_bezt = NULL;
1152                         if ((old_nbezt > 1) && (gps->prev->totpoints > 1)) {
1153                                 /* Only use last curve segment if previous stroke was not a single-point one! */
1154                                 prev_bezt = &nu->bezt[old_nbezt - 2];
1155                         }
1156                         bezt = &nu->bezt[old_nbezt - 1];
1157
1158                         /* First point */
1159                         if (prev_bezt) {
1160                                 interp_v3_v3v3(p1, prev_bezt->vec[1], bezt->vec[1], 1.0f + GAP_DFAC);
1161                                 if (do_gtd) {
1162                                         const int idx = gps->prev->totpoints - 1;
1163                                         dt1 = interpf(gps->prev->points[idx - 1].time, gps->prev->points[idx].time, -GAP_DFAC);
1164                                 }
1165                         }
1166                         else {
1167                                 interp_v3_v3v3(p1, bezt->vec[1], p3d_cur, GAP_DFAC);
1168                                 if (do_gtd) {
1169                                         dt1 = interpf(gps->inittime - gps->prev->inittime, 0.0f, GAP_DFAC);
1170                                 }
1171                         }
1172
1173                         /* Second point */
1174                         /* Note dt2 is always negative, which marks the gap. */
1175                         if (tot > 1) {
1176                                 interp_v3_v3v3(p2, p3d_cur, p3d_next, -GAP_DFAC);
1177                                 if (do_gtd) {
1178                                         dt2 = interpf(gps->points[1].time, gps->points[0].time, -GAP_DFAC);
1179                                 }
1180                         }
1181                         else {
1182                                 interp_v3_v3v3(p2, p3d_cur, bezt->vec[1], GAP_DFAC);
1183                                 if (do_gtd) {
1184                                         dt2 = interpf(gps->prev->inittime - gps->inittime, 0.0f, GAP_DFAC);
1185                                 }
1186                         }
1187
1188                         /* Second handle of last point of previous stroke. */
1189                         interp_v3_v3v3(h2, bezt->vec[1], p1, BEZT_HANDLE_FAC);
1190                         copy_v3_v3(bezt->vec[2], h2);
1191
1192                         /* First point */
1193                         interp_v3_v3v3(h1, p1, bezt->vec[1], BEZT_HANDLE_FAC);
1194                         interp_v3_v3v3(h2, p1, p2, BEZT_HANDLE_FAC);
1195                         bezt++;
1196                         gp_stroke_to_bezier_add_point(gtd, bezt, p1, h1, h2, (bezt - 1)->vec[1], do_gtd, gps->prev->inittime, dt1,
1197                                                       0.0f, rad_fac, minmax_weights);
1198
1199                         /* Second point */
1200                         interp_v3_v3v3(h1, p2, p1, BEZT_HANDLE_FAC);
1201                         interp_v3_v3v3(h2, p2, p3d_cur, BEZT_HANDLE_FAC);
1202                         bezt++;
1203                         gp_stroke_to_bezier_add_point(gtd, bezt, p2, h1, h2, p1, do_gtd, gps->inittime, dt2,
1204                                                       0.0f, rad_fac, minmax_weights);
1205
1206                         old_nbezt += 2;
1207                         copy_v3_v3(p3d_prev, p2);
1208                 }
1209         }
1210         else if (add_start_point) {
1211                 float p[3];
1212                 float dt = 0.0f;
1213
1214                 if (gps->totpoints > 1) {
1215                         interp_v3_v3v3(p, p3d_cur, p3d_next, -GAP_DFAC);
1216                         if (do_gtd) {
1217                                 dt = interpf(gps->points[1].time, gps->points[0].time, -GAP_DFAC);
1218                         }
1219                 }
1220                 else {
1221                         copy_v3_v3(p, p3d_cur);
1222                         p[0] -= GAP_DFAC;  /* Rather arbitrary... */
1223                         dt = -GAP_DFAC;  /* Rather arbitrary too! */
1224                 }
1225                 interp_v3_v3v3(h1, p, p3d_cur, -BEZT_HANDLE_FAC);
1226                 interp_v3_v3v3(h2, p, p3d_cur, BEZT_HANDLE_FAC);
1227                 bezt = &nu->bezt[old_nbezt];
1228                 gp_stroke_to_bezier_add_point(gtd, bezt, p, h1, h2, p, do_gtd, gps->inittime, dt,
1229                                               0.0f, rad_fac, minmax_weights);
1230
1231                 old_nbezt++;
1232                 copy_v3_v3(p3d_prev, p);
1233         }
1234
1235         if (old_nbezt) {
1236                 prev_bezt = &nu->bezt[old_nbezt - 1];
1237         }
1238
1239         /* add points */
1240         for (i = stitch ? 1 : 0, bezt = &nu->bezt[old_nbezt]; i < tot; i++, pt++, bezt++) {
1241                 float width = pt->pressure * gpl->thickness * WIDTH_CORR_FAC;
1242
1243                 if (i || old_nbezt) {
1244                         interp_v3_v3v3(h1, p3d_cur, p3d_prev, BEZT_HANDLE_FAC);
1245                 }
1246                 else {
1247                         interp_v3_v3v3(h1, p3d_cur, p3d_next, -BEZT_HANDLE_FAC);
1248                 }
1249
1250                 if (i < tot - 1) {
1251                         interp_v3_v3v3(h2, p3d_cur, p3d_next, BEZT_HANDLE_FAC);
1252                 }
1253                 else {
1254                         interp_v3_v3v3(h2, p3d_cur, p3d_prev, -BEZT_HANDLE_FAC);
1255                 }
1256
1257                 gp_stroke_to_bezier_add_point(gtd, bezt, p3d_cur, h1, h2, prev_bezt ? prev_bezt->vec[1] : p3d_cur,
1258                                               do_gtd, gps->inittime, pt->time, width, rad_fac, minmax_weights);
1259
1260                 /* shift coord vects */
1261                 copy_v3_v3(p3d_prev, p3d_cur);
1262                 copy_v3_v3(p3d_cur, p3d_next);
1263
1264                 if (i + 2 < tot) {
1265                         gp_strokepoint_convertcoords(C, gps, pt + 2, p3d_next, subrect);
1266                 }
1267
1268                 prev_bezt = bezt;
1269         }
1270
1271         if (add_end_point) {
1272                 float p[3];
1273                 float dt = 0.0f;
1274
1275                 if (gps->totpoints > 1) {
1276                         interp_v3_v3v3(p, prev_bezt->vec[1], (prev_bezt - 1)->vec[1], -GAP_DFAC);
1277                         if (do_gtd) {
1278                                 const int idx = gps->totpoints - 1;
1279                                 dt = interpf(gps->points[idx - 1].time, gps->points[idx].time, -GAP_DFAC);
1280                         }
1281                 }
1282                 else {
1283                         copy_v3_v3(p, prev_bezt->vec[1]);
1284                         p[0] += GAP_DFAC;  /* Rather arbitrary... */
1285                         dt = GAP_DFAC;  /* Rather arbitrary too! */
1286                 }
1287
1288                 /* Second handle of last point of this stroke. */
1289                 interp_v3_v3v3(h2, prev_bezt->vec[1], p, BEZT_HANDLE_FAC);
1290                 copy_v3_v3(prev_bezt->vec[2], h2);
1291
1292                 /* The end point */
1293                 interp_v3_v3v3(h1, p, prev_bezt->vec[1], BEZT_HANDLE_FAC);
1294                 interp_v3_v3v3(h2, p, prev_bezt->vec[1], -BEZT_HANDLE_FAC);
1295                 /* Note bezt has already been incremented in main loop above, so it points to the right place. */
1296                 gp_stroke_to_bezier_add_point(gtd, bezt, p, h1, h2, prev_bezt->vec[1], do_gtd, gps->inittime, dt,
1297                                               0.0f, rad_fac, minmax_weights);
1298         }
1299
1300         /* must calculate handles or else we crash */
1301         BKE_nurb_handles_calc(nu);
1302
1303         if (!curnu || !*curnu) {
1304                 BLI_addtail(&cu->nurb, nu);
1305         }
1306         if (curnu) {
1307                 *curnu = nu;
1308         }
1309 }
1310
1311 #undef GAP_DFAC
1312 #undef WIDTH_CORR_FAC
1313 #undef BEZT_HANDLE_FAC
1314
1315 static void gp_stroke_finalize_curve_endpoints(Curve *cu)
1316 {
1317         /* start */
1318         Nurb *nu = cu->nurb.first;
1319         int i = 0;
1320         if (nu->bezt) {
1321                 BezTriple *bezt = nu->bezt;
1322                 if (bezt) {
1323                         bezt[i].weight = bezt[i].radius = 0.0f;
1324                 }
1325         }
1326         else if (nu->bp) {
1327                 BPoint *bp = nu->bp;
1328                 if (bp) {
1329                         bp[i].weight = bp[i].radius = 0.0f;
1330                 }
1331         }
1332
1333         /* end */
1334         nu = cu->nurb.last;
1335         i = nu->pntsu - 1;
1336         if (nu->bezt) {
1337                 BezTriple *bezt = nu->bezt;
1338                 if (bezt) {
1339                         bezt[i].weight = bezt[i].radius = 0.0f;
1340                 }
1341         }
1342         else if (nu->bp) {
1343                 BPoint *bp = nu->bp;
1344                 if (bp) {
1345                         bp[i].weight = bp[i].radius = 0.0f;
1346                 }
1347         }
1348 }
1349
1350 static void gp_stroke_norm_curve_weights(Curve *cu, const float minmax_weights[2])
1351 {
1352         Nurb *nu;
1353         const float delta = minmax_weights[0];
1354         float fac;
1355         int i;
1356
1357         /* when delta == minmax_weights[0] == minmax_weights[1], we get div by zero [#35686] */
1358         if (IS_EQF(delta, minmax_weights[1]))
1359                 fac = 1.0f;
1360         else
1361                 fac = 1.0f / (minmax_weights[1] - delta);
1362
1363         for (nu = cu->nurb.first; nu; nu = nu->next) {
1364                 if (nu->bezt) {
1365                         BezTriple *bezt = nu->bezt;
1366                         for (i = 0; i < nu->pntsu; i++, bezt++) {
1367                                 bezt->weight = (bezt->weight - delta) * fac;
1368                         }
1369                 }
1370                 else if (nu->bp) {
1371                         BPoint *bp = nu->bp;
1372                         for (i = 0; i < nu->pntsu; i++, bp++) {
1373                                 bp->weight = (bp->weight - delta) * fac;
1374                         }
1375                 }
1376         }
1377 }
1378
1379 static int gp_camera_view_subrect(bContext *C, rctf *subrect)
1380 {
1381         View3D *v3d = CTX_wm_view3d(C);
1382         ARegion *ar = CTX_wm_region(C);
1383
1384         if (v3d) {
1385                 RegionView3D *rv3d = ar->regiondata;
1386
1387                 /* for camera view set the subrect */
1388                 if (rv3d->persp == RV3D_CAMOB) {
1389                         Scene *scene = CTX_data_scene(C);
1390                         ED_view3d_calc_camera_border(scene, ar, v3d, rv3d, subrect, true); /* no shift */
1391                         return 1;
1392                 }
1393         }
1394
1395         return 0;
1396 }
1397
1398 /* convert a given grease-pencil layer to a 3d-curve representation (using current view if appropriate) */
1399 static void gp_layer_to_curve(bContext *C, ReportList *reports, bGPdata *gpd, bGPDlayer *gpl, const int mode,
1400                               const bool norm_weights, const float rad_fac, const bool link_strokes, tGpTimingData *gtd)
1401 {
1402         struct Main *bmain = CTX_data_main(C);
1403         View3D *v3d = CTX_wm_view3d(C);  /* may be NULL */
1404         Scene *scene = CTX_data_scene(C);
1405         bGPDframe *gpf = gpencil_layer_getframe(gpl, CFRA, 0);
1406         bGPDstroke *gps, *prev_gps = NULL;
1407         Object *ob;
1408         Curve *cu;
1409         Nurb *nu = NULL;
1410         Base *base_orig = BASACT, *base_new = NULL;
1411         float minmax_weights[2] = {1.0f, 0.0f};
1412
1413         /* camera framing */
1414         rctf subrect, *subrect_ptr = NULL;
1415
1416         /* error checking */
1417         if (ELEM(NULL, gpd, gpl, gpf))
1418                 return;
1419
1420         /* only convert if there are any strokes on this layer's frame to convert */
1421         if (BLI_listbase_is_empty(&gpf->strokes))
1422                 return;
1423
1424         /* initialize camera framing */
1425         if (gp_camera_view_subrect(C, &subrect)) {
1426                 subrect_ptr = &subrect;
1427         }
1428
1429         /* init the curve object (remove rotation and get curve data from it)
1430          *      - must clear transforms set on object, as those skew our results
1431          */
1432         ob = BKE_object_add_only_object(bmain, OB_CURVE, gpl->info);
1433         cu = ob->data = BKE_curve_add(bmain, gpl->info, OB_CURVE);
1434         base_new = BKE_scene_base_add(scene, ob);
1435
1436         cu->flag |= CU_3D;
1437
1438         gtd->inittime = ((bGPDstroke *)gpf->strokes.first)->inittime;
1439
1440         /* add points to curve */
1441         for (gps = gpf->strokes.first; gps; gps = gps->next) {
1442                 const bool add_start_point = (link_strokes && !(prev_gps));
1443                 const bool add_end_point = (link_strokes && !(gps->next));
1444
1445                 /* Detect new strokes created because of GP_STROKE_BUFFER_MAX reached, and stitch them to previous one. */
1446                 bool stitch = false;
1447                 if (prev_gps) {
1448                         bGPDspoint *pt1 = &prev_gps->points[prev_gps->totpoints - 1];
1449                         bGPDspoint *pt2 = &gps->points[0];
1450
1451                         if ((pt1->x == pt2->x) && (pt1->y == pt2->y)) {
1452                                 stitch = true;
1453                         }
1454                 }
1455
1456                 /* Decide whether we connect this stroke to previous one */
1457                 if (!(stitch || link_strokes)) {
1458                         nu = NULL;
1459                 }
1460
1461                 switch (mode) {
1462                         case GP_STROKECONVERT_PATH: 
1463                                 gp_stroke_to_path(C, gpl, gps, cu, subrect_ptr, &nu, minmax_weights, rad_fac, stitch,
1464                                                   add_start_point, add_end_point, gtd);
1465                                 break;
1466                         case GP_STROKECONVERT_CURVE:
1467                         case GP_STROKECONVERT_POLY:  /* convert after */
1468                                 gp_stroke_to_bezier(C, gpl, gps, cu, subrect_ptr, &nu, minmax_weights, rad_fac, stitch,
1469                                                     add_start_point, add_end_point, gtd);
1470                                 break;
1471                         default:
1472                                 BLI_assert(!"invalid mode");
1473                                 break;
1474                 }
1475                 prev_gps = gps;
1476         }
1477
1478         /* If link_strokes, be sure first and last points have a zero weight/size! */
1479         if (link_strokes) {
1480                 gp_stroke_finalize_curve_endpoints(cu);
1481         }
1482
1483         /* Update curve's weights, if needed */
1484         if (norm_weights && ((minmax_weights[0] > 0.0f) || (minmax_weights[1] < 1.0f))) {
1485                 gp_stroke_norm_curve_weights(cu, minmax_weights);
1486         }
1487
1488         /* Create the path animation, if needed */
1489         gp_stroke_path_animation(C, reports, cu, gtd);
1490
1491         if (mode == GP_STROKECONVERT_POLY) {
1492                 for (nu = cu->nurb.first; nu; nu = nu->next) {
1493                         BKE_nurb_type_convert(nu, CU_POLY, false);
1494                 }
1495         }
1496
1497         /* set the layer and select */
1498         base_new->lay  = ob->lay  = base_orig ? base_orig->lay : BKE_screen_view3d_layer_active(v3d, scene);
1499         base_new->flag = ob->flag = base_new->flag | SELECT;
1500 }
1501
1502 /* --- */
1503
1504 /* Check a GP layer has valid timing data! Else, most timing options are hidden in the operator.
1505  * op may be NULL.
1506  */
1507 static bool gp_convert_check_has_valid_timing(bContext *C, bGPDlayer *gpl, wmOperator *op)
1508 {
1509         Scene *scene = CTX_data_scene(C);
1510         bGPDframe *gpf = NULL;
1511         bGPDstroke *gps = NULL;
1512         bGPDspoint *pt;
1513         double base_time, cur_time, prev_time = -1.0;
1514         int i;
1515         bool valid = true;
1516
1517         if (!gpl || !(gpf = gpencil_layer_getframe(gpl, CFRA, 0)) || !(gps = gpf->strokes.first))
1518                 return false;
1519
1520         do {
1521                 base_time = cur_time = gps->inittime;
1522                 if (cur_time <= prev_time) {
1523                         valid = false;
1524                         break;
1525                 }
1526
1527                 prev_time = cur_time;
1528                 for (i = 0, pt = gps->points; i < gps->totpoints; i++, pt++) {
1529                         cur_time = base_time + (double)pt->time;
1530                         /* First point of a stroke should have the same time as stroke's inittime,
1531                          * so it's the only case where equality is allowed!
1532                          */
1533                         if ((i && cur_time <= prev_time) || (cur_time < prev_time)) {
1534                                 valid = false;
1535                                 break;
1536                         }
1537                         prev_time = cur_time;
1538                 }
1539
1540                 if (!valid) {
1541                         break;
1542                 }
1543         } while ((gps = gps->next));
1544
1545         if (op) {
1546                 RNA_boolean_set(op->ptr, "use_timing_data", valid);
1547         }
1548         return valid;
1549 }
1550
1551 /* Check end_frame is always > start frame! */
1552 static void gp_convert_set_end_frame(struct Main *UNUSED(main), struct Scene *UNUSED(scene), struct PointerRNA *ptr)
1553 {
1554         int start_frame = RNA_int_get(ptr, "start_frame");
1555         int end_frame = RNA_int_get(ptr, "end_frame");
1556
1557         if (end_frame <= start_frame) {
1558                 RNA_int_set(ptr, "end_frame", start_frame + 1);
1559         }
1560 }
1561
1562 static int gp_convert_poll(bContext *C)
1563 {
1564         bGPdata *gpd = ED_gpencil_data_get_active(C);
1565         bGPDlayer *gpl = NULL;
1566         bGPDframe *gpf = NULL;
1567         ScrArea *sa = CTX_wm_area(C);
1568         Scene *scene = CTX_data_scene(C);
1569
1570         /* only if the current view is 3D View, if there's valid data (i.e. at least one stroke!),
1571          * and if we are not in edit mode!
1572          */
1573         return ((sa && sa->spacetype == SPACE_VIEW3D) &&
1574                 (gpl = gpencil_layer_getactive(gpd)) &&
1575                 (gpf = gpencil_layer_getframe(gpl, CFRA, 0)) &&
1576                 (gpf->strokes.first) &&
1577                 (scene->obedit == NULL));
1578 }
1579
1580 static int gp_convert_layer_exec(bContext *C, wmOperator *op)
1581 {
1582         PropertyRNA *prop = RNA_struct_find_property(op->ptr, "use_timing_data");
1583         bGPdata *gpd = ED_gpencil_data_get_active(C);
1584         bGPDlayer *gpl = gpencil_layer_getactive(gpd);
1585         Scene *scene = CTX_data_scene(C);
1586         const int mode = RNA_enum_get(op->ptr, "type");
1587         const bool norm_weights = RNA_boolean_get(op->ptr, "use_normalize_weights");
1588         const float rad_fac = RNA_float_get(op->ptr, "radius_multiplier");
1589         const bool link_strokes = RNA_boolean_get(op->ptr, "use_link_strokes");
1590         bool valid_timing;
1591         tGpTimingData gtd;
1592
1593         /* check if there's data to work with */
1594         if (gpd == NULL) {
1595                 BKE_report(op->reports, RPT_ERROR, "No Grease Pencil data to work on");
1596                 return OPERATOR_CANCELLED;
1597         }
1598
1599         if (!RNA_property_is_set(op->ptr, prop) && !gp_convert_check_has_valid_timing(C, gpl, op)) {
1600                 BKE_report(op->reports, RPT_WARNING,
1601                            "Current Grease Pencil strokes have no valid timing data, most timing options will be hidden!");
1602         }
1603         valid_timing = RNA_property_boolean_get(op->ptr, prop);
1604
1605         gtd.mode = RNA_enum_get(op->ptr, "timing_mode");
1606         /* Check for illegal timing mode! */
1607         if (!valid_timing && !ELEM(gtd.mode, GP_STROKECONVERT_TIMING_NONE, GP_STROKECONVERT_TIMING_LINEAR)) {
1608                 gtd.mode = GP_STROKECONVERT_TIMING_LINEAR;
1609                 RNA_enum_set(op->ptr, "timing_mode", gtd.mode);
1610         }
1611         if (!link_strokes) {
1612                 gtd.mode = GP_STROKECONVERT_TIMING_NONE;
1613         }
1614
1615         /* grab all relevant settings */
1616         gtd.frame_range = RNA_int_get(op->ptr, "frame_range");
1617         gtd.start_frame = RNA_int_get(op->ptr, "start_frame");
1618         gtd.realtime = valid_timing ? RNA_boolean_get(op->ptr, "use_realtime") : false;
1619         gtd.end_frame = RNA_int_get(op->ptr, "end_frame");
1620         gtd.gap_duration = RNA_float_get(op->ptr, "gap_duration");
1621         gtd.gap_randomness = RNA_float_get(op->ptr, "gap_randomness");
1622         gtd.gap_randomness = min_ff(gtd.gap_randomness, gtd.gap_duration);
1623         gtd.seed = RNA_int_get(op->ptr, "seed");
1624         gtd.num_points = gtd.cur_point = 0;
1625         gtd.dists = gtd.times = NULL;
1626         gtd.tot_dist = gtd.tot_time = gtd.gap_tot_time = 0.0f;
1627         gtd.inittime = 0.0;
1628         gtd.offset_time = 0.0f;
1629
1630         /* perform conversion */
1631         gp_layer_to_curve(C, op->reports, gpd, gpl, mode, norm_weights, rad_fac, link_strokes, &gtd);
1632
1633         /* free temp memory */
1634         if (gtd.dists) {
1635                 MEM_freeN(gtd.dists);
1636                 gtd.dists = NULL;
1637         }
1638         if (gtd.times) {
1639                 MEM_freeN(gtd.times);
1640                 gtd.times = NULL;
1641         }
1642
1643         /* notifiers */
1644         WM_event_add_notifier(C, NC_OBJECT | NA_ADDED, NULL);
1645         WM_event_add_notifier(C, NC_SCENE | ND_OB_ACTIVE, scene);
1646
1647         /* done */
1648         return OPERATOR_FINISHED;
1649 }
1650
1651 static bool gp_convert_draw_check_prop(PointerRNA *ptr, PropertyRNA *prop)
1652 {
1653         const char *prop_id = RNA_property_identifier(prop);
1654         const bool link_strokes = RNA_boolean_get(ptr, "use_link_strokes");
1655         int timing_mode = RNA_enum_get(ptr, "timing_mode");
1656         bool realtime = RNA_boolean_get(ptr, "use_realtime");
1657         float gap_duration = RNA_float_get(ptr, "gap_duration");
1658         float gap_randomness = RNA_float_get(ptr, "gap_randomness");
1659         const bool valid_timing = RNA_boolean_get(ptr, "use_timing_data");
1660
1661         /* Always show those props */
1662         if (strcmp(prop_id, "type") == 0 ||
1663             strcmp(prop_id, "use_normalize_weights") == 0 ||
1664             strcmp(prop_id, "radius_multiplier") == 0 ||
1665             strcmp(prop_id, "use_link_strokes") == 0)
1666         {
1667                 return true;
1668         }
1669
1670         /* Never show this prop */
1671         if (strcmp(prop_id, "use_timing_data") == 0)
1672                 return false;
1673
1674         if (link_strokes) {
1675                 /* Only show when link_stroke is true */
1676                 if (strcmp(prop_id, "timing_mode") == 0)
1677                         return true;
1678
1679                 if (timing_mode != GP_STROKECONVERT_TIMING_NONE) {
1680                         /* Only show when link_stroke is true and stroke timing is enabled */
1681                         if (strcmp(prop_id, "frame_range") == 0 ||
1682                             strcmp(prop_id, "start_frame") == 0)
1683                         {
1684                                 return true;
1685                         }
1686
1687                         /* Only show if we have valid timing data! */
1688                         if (valid_timing && strcmp(prop_id, "use_realtime") == 0)
1689                                 return true;
1690
1691                         /* Only show if realtime or valid_timing is false! */
1692                         if ((!realtime || !valid_timing) && strcmp(prop_id, "end_frame") == 0)
1693                                 return true;
1694
1695                         if (valid_timing && timing_mode == GP_STROKECONVERT_TIMING_CUSTOMGAP) {
1696                                 /* Only show for custom gaps! */
1697                                 if (strcmp(prop_id, "gap_duration") == 0)
1698                                         return true;
1699
1700                                 /* Only show randomness for non-null custom gaps! */
1701                                 if (strcmp(prop_id, "gap_randomness") == 0 && (gap_duration > 0.0f))
1702                                         return true;
1703
1704                                 /* Only show seed for randomize action! */
1705                                 if (strcmp(prop_id, "seed") == 0 && (gap_duration > 0.0f) && (gap_randomness > 0.0f))
1706                                         return true;
1707                         }
1708                 }
1709         }
1710
1711         /* Else, hidden! */
1712         return false;
1713 }
1714
1715 static void gp_convert_ui(bContext *C, wmOperator *op)
1716 {
1717         uiLayout *layout = op->layout;
1718         wmWindowManager *wm = CTX_wm_manager(C);
1719         PointerRNA ptr;
1720
1721         RNA_pointer_create(&wm->id, op->type->srna, op->properties, &ptr);
1722
1723         /* Main auto-draw call */
1724         uiDefAutoButsRNA(layout, &ptr, gp_convert_draw_check_prop, '\0');
1725 }
1726
1727 void GPENCIL_OT_convert(wmOperatorType *ot)
1728 {
1729         PropertyRNA *prop;
1730
1731         /* identifiers */
1732         ot->name = "Convert Grease Pencil";
1733         ot->idname = "GPENCIL_OT_convert";
1734         ot->description = "Convert the active Grease Pencil layer to a new Curve Object";
1735
1736         /* callbacks */
1737         ot->invoke = WM_menu_invoke;
1738         ot->exec = gp_convert_layer_exec;
1739         ot->poll = gp_convert_poll;
1740         ot->ui = gp_convert_ui;
1741
1742         /* flags */
1743         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1744
1745         /* properties */
1746         ot->prop = RNA_def_enum(ot->srna, "type", prop_gpencil_convertmodes, 0, "Type", "Which type of curve to convert to");
1747
1748         RNA_def_boolean(ot->srna, "use_normalize_weights", true, "Normalize Weight",
1749                         "Normalize weight (set from stroke width)");
1750         RNA_def_float(ot->srna, "radius_multiplier", 1.0f, 0.0f, 1000.0f, "Radius Fac",
1751                       "Multiplier for the points' radii (set from stroke width)", 0.0f, 10.0f);
1752         RNA_def_boolean(ot->srna, "use_link_strokes", true, "Link Strokes",
1753                         "Whether to link strokes with zero-radius sections of curves");
1754
1755         prop = RNA_def_enum(ot->srna, "timing_mode", prop_gpencil_convert_timingmodes, GP_STROKECONVERT_TIMING_FULL,
1756                             "Timing Mode", "How to use timing data stored in strokes");
1757         RNA_def_enum_funcs(prop, rna_GPConvert_mode_items);
1758
1759         RNA_def_int(ot->srna, "frame_range", 100, 1, 10000, "Frame Range",
1760                     "The duration of evaluation of the path control curve", 1, 1000);
1761         RNA_def_int(ot->srna, "start_frame", 1, 1, 100000, "Start Frame",
1762                     "The start frame of the path control curve", 1, 100000);
1763         RNA_def_boolean(ot->srna, "use_realtime", false, "Realtime",
1764                         "Whether the path control curve reproduces the drawing in realtime, starting from Start Frame");
1765         prop = RNA_def_int(ot->srna, "end_frame", 250, 1, 100000, "End Frame",
1766                            "The end frame of the path control curve (if Realtime is not set)", 1, 100000);
1767         RNA_def_property_update_runtime(prop, gp_convert_set_end_frame);
1768
1769         RNA_def_float(ot->srna, "gap_duration", 0.0f, 0.0f, 10000.0f, "Gap Duration",
1770                       "Custom Gap mode: (Average) length of gaps, in frames "
1771                       "(Note: Realtime value, will be scaled if Realtime is not set)", 0.0f, 1000.0f);
1772         RNA_def_float(ot->srna, "gap_randomness", 0.0f, 0.0f, 10000.0f, "Gap Randomness",
1773                       "Custom Gap mode: Number of frames that gap lengths can vary", 0.0f, 1000.0f);
1774         RNA_def_int(ot->srna, "seed", 0, 0, 1000, "Random Seed",
1775                     "Custom Gap mode: Random generator seed", 0, 100);
1776
1777         /* Note: Internal use, this one will always be hidden by UI code... */
1778         prop = RNA_def_boolean(ot->srna, "use_timing_data", false, "Has Valid Timing",
1779                                "Whether the converted Grease Pencil layer has valid timing data (internal use)");
1780         RNA_def_property_flag(prop, PROP_SKIP_SAVE);
1781 }
1782
1783 /* ************************************************ */