GP: Refactor drawing engine to single VBO
[blender.git] / source / blender / gpencil_modifiers / intern / MOD_gpenciloffset.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_gpenciloffset.c
28  *  \ingroup modifiers
29  */
30
31 #include <stdio.h>
32
33 #include "DNA_meshdata_types.h"
34 #include "DNA_scene_types.h"
35 #include "DNA_object_types.h"
36 #include "DNA_gpencil_types.h"
37 #include "DNA_gpencil_modifier_types.h"
38
39 #include "BLI_utildefines.h"
40 #include "BLI_math.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         OffsetGpencilModifierData *gpmd = (OffsetGpencilModifierData *)md;
56         gpmd->pass_index = 0;
57         gpmd->layername[0] = '\0';
58         gpmd->vgname[0] = '\0';
59         ARRAY_SET_ITEMS(gpmd->loc, 0.0f, 0.0f, 0.0f);
60         ARRAY_SET_ITEMS(gpmd->rot, 0.0f, 0.0f, 0.0f);
61         ARRAY_SET_ITEMS(gpmd->scale, 0.0f, 0.0f, 0.0f);
62 }
63
64 static void copyData(const GpencilModifierData *md, GpencilModifierData *target)
65 {
66         BKE_gpencil_modifier_copyData_generic(md, target);
67 }
68
69 /* change stroke offsetness */
70 static void deformStroke(
71         GpencilModifierData *md, Depsgraph *UNUSED(depsgraph),
72         Object *ob, bGPDlayer *gpl, bGPDstroke *gps)
73 {
74         OffsetGpencilModifierData *mmd = (OffsetGpencilModifierData *)md;
75         const int def_nr = defgroup_name_index(ob, mmd->vgname);
76
77         float mat[4][4];
78         float loc[3], rot[3], scale[3];
79
80         if (!is_stroke_affected_by_modifier(
81                     ob,
82                     mmd->layername, mmd->pass_index, mmd->layer_pass, 1, gpl, gps,
83                     mmd->flag & GP_OFFSET_INVERT_LAYER, mmd->flag & GP_OFFSET_INVERT_PASS,
84                     mmd->flag & GP_OFFSET_INVERT_LAYERPASS))
85         {
86                 return;
87         }
88
89         for (int i = 0; i < gps->totpoints; i++) {
90                 bGPDspoint *pt = &gps->points[i];
91                 MDeformVert *dvert = gps->dvert != NULL ? &gps->dvert[i] : NULL;
92
93                 /* verify vertex group */
94                 const float weight = get_modifier_point_weight(dvert, (mmd->flag & GP_OFFSET_INVERT_VGROUP) != 0, def_nr);
95                 if (weight < 0.0f) {
96                         continue;
97                 }
98                 /* calculate matrix */
99                 mul_v3_v3fl(loc, mmd->loc, weight);
100                 mul_v3_v3fl(rot, mmd->rot, weight);
101                 mul_v3_v3fl(scale, mmd->scale, weight);
102                 add_v3_fl(scale, 1.0);
103                 loc_eul_size_to_mat4(mat, loc, rot, scale);
104
105                 mul_m4_v3(mat, &pt->x);
106         }
107 }
108
109 static void bakeModifier(
110         struct Main *UNUSED(bmain), Depsgraph *depsgraph,
111         GpencilModifierData *md, Object *ob)
112 {
113         bGPdata *gpd = ob->data;
114
115         for (bGPDlayer *gpl = gpd->layers.first; gpl; gpl = gpl->next) {
116                 for (bGPDframe *gpf = gpl->frames.first; gpf; gpf = gpf->next) {
117                         for (bGPDstroke *gps = gpf->strokes.first; gps; gps = gps->next) {
118                                 deformStroke(md, depsgraph, ob, gpl, gps);
119                         }
120                 }
121         }
122 }
123
124 GpencilModifierTypeInfo modifierType_Gpencil_Offset = {
125         /* name */              "Offset",
126         /* structName */        "OffsetGpencilModifierData",
127         /* structSize */        sizeof(OffsetGpencilModifierData),
128         /* type */              eGpencilModifierTypeType_Gpencil,
129         /* flags */             eGpencilModifierTypeFlag_SupportsEditmode,
130
131         /* copyData */          copyData,
132
133         /* deformStroke */      deformStroke,
134         /* generateStrokes */   NULL,
135         /* bakeModifier */      bakeModifier,
136         /* remapTime */         NULL,
137
138         /* initData */          initData,
139         /* freeData */          NULL,
140         /* isDisabled */        NULL,
141         /* updateDepsgraph */   NULL,
142         /* dependsOnTime */     NULL,
143         /* foreachObjectLink */ NULL,
144         /* foreachIDLink */     NULL,
145         /* foreachTexLink */    NULL,
146         /* getDuplicationFactor */ NULL,
147 };