38aa4159632da37e78efbc1b4f8d593c87d7db58
[blender.git] / source / blender / gpencil_modifiers / intern / MOD_gpencilsubdiv.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) 2017, 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_gpencilsubdiv.c
28   *  \ingroup modifiers
29   */
30
31 #include <stdio.h>
32
33 #include "MEM_guardedalloc.h"
34
35 #include "DNA_meshdata_types.h"
36 #include "DNA_scene_types.h"
37 #include "DNA_object_types.h"
38 #include "DNA_gpencil_types.h"
39 #include "DNA_gpencil_modifier_types.h"
40
41 #include "BLI_math.h"
42 #include "BLI_utildefines.h"
43
44 #include "BKE_context.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         SubdivGpencilModifierData *gpmd = (SubdivGpencilModifierData *)md;
56         gpmd->pass_index = 0;
57         gpmd->level = 1;
58         gpmd->layername[0] = '\0';
59 }
60
61 static void copyData(const GpencilModifierData *md, GpencilModifierData *target)
62 {
63         BKE_gpencil_modifier_copyData_generic(md, target);
64 }
65
66 /* subdivide stroke to get more control points */
67 static void deformStroke(
68         GpencilModifierData *md, Depsgraph *UNUSED(depsgraph),
69         Object *ob, bGPDlayer *gpl, bGPDstroke *gps)
70 {
71         SubdivGpencilModifierData *mmd = (SubdivGpencilModifierData *)md;
72
73         if (!is_stroke_affected_by_modifier(
74                     ob,
75                     mmd->layername, mmd->pass_index, mmd->layer_pass, 3, gpl, gps,
76                     mmd->flag & GP_SUBDIV_INVERT_LAYER, mmd->flag & GP_SUBDIV_INVERT_PASS,
77                     mmd->flag & GP_SUBDIV_INVERT_LAYERPASS))
78         {
79                 return;
80         }
81
82         BKE_gpencil_subdivide(gps, mmd->level, mmd->flag);
83 }
84
85 static void bakeModifier(
86         struct Main *UNUSED(bmain), Depsgraph *depsgraph,
87         GpencilModifierData *md, Object *ob)
88 {
89         bGPdata *gpd = ob->data;
90
91         for (bGPDlayer *gpl = gpd->layers.first; gpl; gpl = gpl->next) {
92                 for (bGPDframe *gpf = gpl->frames.first; gpf; gpf = gpf->next) {
93                         for (bGPDstroke *gps = gpf->strokes.first; gps; gps = gps->next) {
94                                 deformStroke(md, depsgraph, ob, gpl, gps);
95                         }
96                 }
97         }
98 }
99
100 GpencilModifierTypeInfo modifierType_Gpencil_Subdiv = {
101         /* name */              "Subdivision",
102         /* structName */        "SubdivGpencilModifierData",
103         /* structSize */        sizeof(SubdivGpencilModifierData),
104         /* type */              eGpencilModifierTypeType_Gpencil,
105         /* flags */             eGpencilModifierTypeFlag_SupportsEditmode,
106
107         /* copyData */          copyData,
108
109         /* deformStroke */      deformStroke,
110         /* generateStrokes */   NULL,
111         /* bakeModifier */      bakeModifier,
112         /* remapTime */         NULL,
113
114         /* initData */          initData,
115         /* freeData */          NULL,
116         /* isDisabled */        NULL,
117         /* updateDepsgraph */   NULL,
118         /* dependsOnTime */     NULL,
119         /* foreachObjectLink */ NULL,
120         /* foreachIDLink */     NULL,
121         /* foreachTexLink */    NULL,
122 };