ba3b2d84861a411212e67c8ce33aef045f4fbfb0
[blender.git] / source / blender / gpencil_modifiers / intern / MOD_gpenciltint.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_gpenciltint.c
28  *  \ingroup modifiers
29  */
30
31 #include <stdio.h>
32
33 #include "DNA_scene_types.h"
34 #include "DNA_object_types.h"
35 #include "DNA_gpencil_types.h"
36 #include "DNA_gpencil_modifier_types.h"
37
38 #include "BLI_blenlib.h"
39 #include "BLI_ghash.h"
40 #include "BLI_math_vector.h"
41 #include "BLI_utildefines.h"
42
43 #include "BKE_context.h"
44 #include "BKE_global.h"
45 #include "BKE_gpencil.h"
46 #include "BKE_gpencil_modifier.h"
47 #include "BKE_material.h"
48 #include "BKE_main.h"
49
50 #include "DEG_depsgraph.h"
51
52 #include "MOD_gpencil_util.h"
53 #include "MOD_gpencil_modifiertypes.h"
54
55 static void initData(GpencilModifierData *md)
56 {
57         TintGpencilModifierData *gpmd = (TintGpencilModifierData *)md;
58         gpmd->pass_index = 0;
59         gpmd->factor = 0.5f;
60         gpmd->layername[0] = '\0';
61         ARRAY_SET_ITEMS(gpmd->rgb, 1.0f, 1.0f, 1.0f);
62         gpmd->flag |= GP_TINT_CREATE_COLORS;
63         gpmd->modify_color = GP_MODIFY_COLOR_BOTH;
64 }
65
66 static void copyData(const GpencilModifierData *md, GpencilModifierData *target)
67 {
68         BKE_gpencil_modifier_copyData_generic(md, target);
69 }
70
71 /* tint strokes */
72 static void deformStroke(
73         GpencilModifierData *md, Depsgraph *UNUSED(depsgraph),
74         Object *ob, bGPDlayer *gpl, bGPDstroke *gps)
75 {
76         TintGpencilModifierData *mmd = (TintGpencilModifierData *)md;
77
78         if (!is_stroke_affected_by_modifier(
79                     ob,
80                     mmd->layername, mmd->pass_index, mmd->layer_pass, 1, gpl, gps,
81                     mmd->flag & GP_TINT_INVERT_LAYER, mmd->flag & GP_TINT_INVERT_PASS,
82                     mmd->flag & GP_TINT_INVERT_LAYERPASS))
83         {
84                 return;
85         }
86
87         if (mmd->modify_color != GP_MODIFY_COLOR_FILL) {
88                 interp_v3_v3v3(gps->runtime.tmp_stroke_rgba, gps->runtime.tmp_stroke_rgba, mmd->rgb, mmd->factor);
89                 /* if factor is > 1, the alpha must be changed to get full tint */
90                 if (mmd->factor > 1.0f) {
91                         gps->runtime.tmp_stroke_rgba[3] += mmd->factor - 1.0f;
92                 }
93                 CLAMP4(gps->runtime.tmp_stroke_rgba, 0.0f, 1.0f);
94         }
95
96         if (mmd->modify_color != GP_MODIFY_COLOR_STROKE) {
97                 interp_v3_v3v3(gps->runtime.tmp_fill_rgba, gps->runtime.tmp_fill_rgba, mmd->rgb, mmd->factor);
98                 /* if factor is > 1, the alpha must be changed to get full tint */
99                 if (mmd->factor > 1.0f && gps->runtime.tmp_fill_rgba[3] > 1e-5) {
100                         gps->runtime.tmp_fill_rgba[3] += mmd->factor - 1.0f;
101                 }
102                 CLAMP4(gps->runtime.tmp_fill_rgba, 0.0f, 1.0f);
103         }
104
105         /* if factor > 1.0, affect the strength of the stroke */
106         if (mmd->factor > 1.0f) {
107                 for (int i = 0; i < gps->totpoints; i++) {
108                         bGPDspoint *pt = &gps->points[i];
109                         pt->strength += mmd->factor - 1.0f;
110                         CLAMP(pt->strength, 0.0f, 1.0f);
111                 }
112         }
113 }
114
115 static void bakeModifier(
116         Main *bmain, Depsgraph *depsgraph,
117         GpencilModifierData *md, Object *ob)
118 {
119         TintGpencilModifierData *mmd = (TintGpencilModifierData *)md;
120         bGPdata *gpd = ob->data;
121
122         GHash *gh_color = BLI_ghash_str_new("GP_Tint modifier");
123         for (bGPDlayer *gpl = gpd->layers.first; gpl; gpl = gpl->next) {
124                 for (bGPDframe *gpf = gpl->frames.first; gpf; gpf = gpf->next) {
125                         for (bGPDstroke *gps = gpf->strokes.first; gps; gps = gps->next) {
126
127                                 Material *mat = give_current_material(ob, gps->mat_nr + 1);
128                                 if (mat == NULL)
129                                         continue;
130                                 MaterialGPencilStyle *gp_style = mat->gp_style;
131                                 /* skip stroke if it doesn't have color info */
132                                 if (ELEM(NULL, gp_style))
133                                         continue;
134
135                                 copy_v4_v4(gps->runtime.tmp_stroke_rgba, gp_style->stroke_rgba);
136                                 copy_v4_v4(gps->runtime.tmp_fill_rgba, gp_style->fill_rgba);
137
138                                 deformStroke(md, depsgraph, ob, gpl, gps);
139
140                                 gpencil_apply_modifier_material(
141                                         bmain, ob, mat, gh_color, gps,
142                                         (bool)(mmd->flag & GP_TINT_CREATE_COLORS));
143                         }
144                 }
145         }
146         /* free hash buffers */
147         if (gh_color) {
148                 BLI_ghash_free(gh_color, NULL, NULL);
149                 gh_color = NULL;
150         }
151 }
152
153 GpencilModifierTypeInfo modifierType_Gpencil_Tint = {
154         /* name */              "Tint",
155         /* structName */        "TintGpencilModifierData",
156         /* structSize */        sizeof(TintGpencilModifierData),
157         /* type */              eGpencilModifierTypeType_Gpencil,
158         /* flags */             eGpencilModifierTypeFlag_SupportsEditmode,
159
160         /* copyData */          copyData,
161
162         /* deformStroke */      deformStroke,
163         /* generateStrokes */   NULL,
164         /* bakeModifier */      bakeModifier,
165         /* remapTime */         NULL,
166
167         /* initData */          initData,
168         /* freeData */          NULL,
169         /* isDisabled */        NULL,
170         /* updateDepsgraph */   NULL,
171         /* dependsOnTime */     NULL,
172         /* foreachObjectLink */ NULL,
173         /* foreachIDLink */     NULL,
174         /* foreachTexLink */    NULL,
175 };