32d3d69050c74fc79c982059226b1e419a161cf9
[blender.git] / source / blender / gpencil_modifiers / intern / MOD_gpenciltime.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) 2018, Blender Foundation
19  * This is a new part of Blender
20  *
21  * Contributor(s): Antonio Vazquez
22  *
23  * ***** END GPL LICENSE BLOCK *****
24  *
25  */
26
27 /** \file blender/gpencil_modifiers/intern/MOD_gpenciltime.c
28  *  \ingroup modifiers
29  */
30
31 #include <stdio.h>
32 #include <string.h>
33
34 #include "BLI_utildefines.h"
35
36 #include "DNA_meshdata_types.h"
37 #include "DNA_scene_types.h"
38 #include "DNA_object_types.h"
39 #include "DNA_gpencil_types.h"
40 #include "DNA_gpencil_modifier_types.h"
41
42 #include "BKE_colortools.h"
43 #include "BKE_context.h"
44 #include "BKE_deform.h"
45 #include "BKE_gpencil.h"
46 #include "BKE_gpencil_modifier.h"
47
48 #include "DEG_depsgraph.h"
49
50 #include "MOD_gpencil_util.h"
51 #include "MOD_gpencil_modifiertypes.h"
52
53 static void initData(GpencilModifierData *md)
54 {
55         TimeGpencilModifierData *gpmd = (TimeGpencilModifierData *)md;
56         gpmd->layername[0] = '\0';
57         gpmd->offset = 1;
58         gpmd->flag |= GP_TIME_KEEP_LOOP;
59 }
60
61 static void copyData(const GpencilModifierData *md, GpencilModifierData *target)
62 {
63         BKE_gpencil_modifier_copyData_generic(md, target);
64 }
65
66 static int remapTime(struct GpencilModifierData *md, struct Depsgraph *UNUSED(depsgraph),
67         struct Scene *scene, struct Object *ob, struct bGPDlayer *gpl, int cfra)
68 {
69         TimeGpencilModifierData *mmd = (TimeGpencilModifierData *)md;
70         const int sfra = scene->r.sfra;
71         const int efra = scene->r.efra;
72         const int nfra = cfra + mmd->offset;
73
74         const bool invgpl = mmd->flag & GP_SIMPLIFY_INVERT_LAYER;
75
76         /* omit if filter by layer */
77         if (mmd->layername[0] != '\0') {
78                 if (invgpl == false) {
79                         if (!STREQ(mmd->layername, gpl->info)) {
80                                 return cfra;
81                         }
82                 }
83                 else {
84                         if (STREQ(mmd->layername, gpl->info)) {
85                                 return cfra;
86                         }
87                 }
88         }
89
90         if (mmd->flag & GP_TIME_KEEP_LOOP) {
91                 /* if the sum of the cfra is out scene frame range, recalc */
92                 if (cfra + mmd->offset < sfra) {
93                         const int delta = abs(sfra - nfra);
94                         return efra - delta + 1;
95                 }
96                 else if (cfra + mmd->offset > efra) {
97                         return nfra - efra + sfra - 1;
98                 }
99         }
100
101         return cfra + mmd->offset;
102 }
103
104 GpencilModifierTypeInfo modifierType_Gpencil_Time = {
105         /* name */              "Time",
106         /* structName */        "TimeGpencilModifierData",
107         /* structSize */        sizeof(TimeGpencilModifierData),
108         /* type */              eGpencilModifierTypeType_Gpencil,
109         /* flags */             eGpencilModifierTypeFlag_Single |
110                                                         eGpencilModifierTypeFlag_NoApply,
111
112         /* copyData */          copyData,
113
114         /* deformStroke */      NULL,
115         /* generateStrokes */   NULL,
116         /* bakeModifier */      NULL,
117         /* remapTime */         remapTime,
118
119         /* initData */          initData,
120         /* freeData */          NULL,
121         /* isDisabled */        NULL,
122         /* updateDepsgraph */   NULL,
123         /* dependsOnTime */     NULL,
124         /* foreachObjectLink */ NULL,
125         /* foreachIDLink */     NULL,
126         /* foreachTexLink */    NULL,
127 };