Cleanup: style, use braces for gpencil modifiers
[blender.git] / source / blender / gpencil_modifiers / intern / MOD_gpenciltint.c
1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software  Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  *
16  * The Original Code is Copyright (C) 2017, Blender Foundation
17  * This is a new part of Blender
18  */
19
20 /** \file
21  * \ingroup modifiers
22  */
23
24 #include <stdio.h>
25
26 #include "BLI_utildefines.h"
27
28 #include "BLI_blenlib.h"
29 #include "BLI_ghash.h"
30 #include "BLI_math_vector.h"
31
32 #include "DNA_scene_types.h"
33 #include "DNA_object_types.h"
34 #include "DNA_gpencil_types.h"
35 #include "DNA_gpencil_modifier_types.h"
36
37 #include "BKE_gpencil.h"
38 #include "BKE_gpencil_modifier.h"
39 #include "BKE_material.h"
40 #include "BKE_main.h"
41
42 #include "DEG_depsgraph.h"
43
44 #include "MOD_gpencil_util.h"
45 #include "MOD_gpencil_modifiertypes.h"
46
47 static void initData(GpencilModifierData *md)
48 {
49   TintGpencilModifierData *gpmd = (TintGpencilModifierData *)md;
50   gpmd->pass_index = 0;
51   gpmd->factor = 0.5f;
52   gpmd->layername[0] = '\0';
53   ARRAY_SET_ITEMS(gpmd->rgb, 1.0f, 1.0f, 1.0f);
54   gpmd->flag |= GP_TINT_CREATE_COLORS;
55   gpmd->modify_color = GP_MODIFY_COLOR_BOTH;
56 }
57
58 static void copyData(const GpencilModifierData *md, GpencilModifierData *target)
59 {
60   BKE_gpencil_modifier_copyData_generic(md, target);
61 }
62
63 /* tint strokes */
64 static void deformStroke(GpencilModifierData *md,
65                          Depsgraph *UNUSED(depsgraph),
66                          Object *ob,
67                          bGPDlayer *gpl,
68                          bGPDstroke *gps)
69 {
70   TintGpencilModifierData *mmd = (TintGpencilModifierData *)md;
71
72   if (!is_stroke_affected_by_modifier(ob,
73                                       mmd->layername,
74                                       mmd->pass_index,
75                                       mmd->layer_pass,
76                                       1,
77                                       gpl,
78                                       gps,
79                                       mmd->flag & GP_TINT_INVERT_LAYER,
80                                       mmd->flag & GP_TINT_INVERT_PASS,
81                                       mmd->flag & GP_TINT_INVERT_LAYERPASS)) {
82     return;
83   }
84
85   if (mmd->modify_color != GP_MODIFY_COLOR_FILL) {
86     interp_v3_v3v3(
87         gps->runtime.tmp_stroke_rgba, gps->runtime.tmp_stroke_rgba, mmd->rgb, mmd->factor);
88     /* if factor is > 1, the alpha must be changed to get full tint */
89     if (mmd->factor > 1.0f) {
90       gps->runtime.tmp_stroke_rgba[3] += mmd->factor - 1.0f;
91     }
92     CLAMP4(gps->runtime.tmp_stroke_rgba, 0.0f, 1.0f);
93   }
94
95   if (mmd->modify_color != GP_MODIFY_COLOR_STROKE) {
96     interp_v3_v3v3(gps->runtime.tmp_fill_rgba, gps->runtime.tmp_fill_rgba, mmd->rgb, mmd->factor);
97     /* if factor is > 1, the alpha must be changed to get full tint */
98     if (mmd->factor > 1.0f && gps->runtime.tmp_fill_rgba[3] > 1e-5) {
99       gps->runtime.tmp_fill_rgba[3] += mmd->factor - 1.0f;
100     }
101     CLAMP4(gps->runtime.tmp_fill_rgba, 0.0f, 1.0f);
102   }
103
104   /* if factor > 1.0, affect the strength of the stroke */
105   if (mmd->factor > 1.0f) {
106     for (int i = 0; i < gps->totpoints; i++) {
107       bGPDspoint *pt = &gps->points[i];
108       pt->strength += mmd->factor - 1.0f;
109       CLAMP(pt->strength, 0.0f, 1.0f);
110     }
111   }
112 }
113
114 static void bakeModifier(Main *bmain, Depsgraph *depsgraph, GpencilModifierData *md, Object *ob)
115 {
116   TintGpencilModifierData *mmd = (TintGpencilModifierData *)md;
117   bGPdata *gpd = ob->data;
118
119   GHash *gh_color = BLI_ghash_str_new("GP_Tint modifier");
120   for (bGPDlayer *gpl = gpd->layers.first; gpl; gpl = gpl->next) {
121     for (bGPDframe *gpf = gpl->frames.first; gpf; gpf = gpf->next) {
122       for (bGPDstroke *gps = gpf->strokes.first; gps; gps = gps->next) {
123
124         Material *mat = give_current_material(ob, gps->mat_nr + 1);
125         if (mat == NULL) {
126           continue;
127         }
128         MaterialGPencilStyle *gp_style = mat->gp_style;
129         /* skip stroke if it doesn't have color info */
130         if (ELEM(NULL, gp_style)) {
131           continue;
132         }
133
134         copy_v4_v4(gps->runtime.tmp_stroke_rgba, gp_style->stroke_rgba);
135         copy_v4_v4(gps->runtime.tmp_fill_rgba, gp_style->fill_rgba);
136
137         deformStroke(md, depsgraph, ob, gpl, gps);
138
139         gpencil_apply_modifier_material(
140             bmain, ob, mat, gh_color, gps, (bool)(mmd->flag & GP_TINT_CREATE_COLORS));
141       }
142     }
143   }
144   /* free hash buffers */
145   if (gh_color) {
146     BLI_ghash_free(gh_color, NULL, NULL);
147     gh_color = NULL;
148   }
149 }
150
151 GpencilModifierTypeInfo modifierType_Gpencil_Tint = {
152     /* name */ "Tint",
153     /* structName */ "TintGpencilModifierData",
154     /* structSize */ sizeof(TintGpencilModifierData),
155     /* type */ eGpencilModifierTypeType_Gpencil,
156     /* flags */ eGpencilModifierTypeFlag_SupportsEditmode,
157
158     /* copyData */ copyData,
159
160     /* deformStroke */ deformStroke,
161     /* generateStrokes */ NULL,
162     /* bakeModifier */ bakeModifier,
163     /* remapTime */ NULL,
164
165     /* initData */ initData,
166     /* freeData */ NULL,
167     /* isDisabled */ NULL,
168     /* updateDepsgraph */ NULL,
169     /* dependsOnTime */ NULL,
170     /* foreachObjectLink */ NULL,
171     /* foreachIDLink */ NULL,
172     /* foreachTexLink */ NULL,
173     /* getDuplicationFactor */ NULL,
174 };