Blender  V2.93
gpencil_add_stroke.c
Go to the documentation of this file.
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 
24 #include "BLI_math.h"
25 #include "BLI_utildefines.h"
26 
27 #include "DNA_gpencil_types.h"
28 #include "DNA_material_types.h"
29 #include "DNA_object_types.h"
30 #include "DNA_scene_types.h"
31 
32 #include "BKE_context.h"
33 #include "BKE_gpencil.h"
34 #include "BKE_gpencil_geom.h"
35 #include "BKE_main.h"
36 #include "BKE_material.h"
37 
38 #include "DEG_depsgraph.h"
39 
40 #include "ED_gpencil.h"
41 
42 /* Definition of the most important info from a color */
43 typedef struct ColorTemplate {
44  const char *name;
45  float line[4];
46  float fill[4];
47 } ColorTemplate;
48 
49 /* Add color an ensure duplications (matched by name) */
50 static int gpencil_stroke_material(Main *bmain,
51  Object *ob,
52  const ColorTemplate *pct,
53  const bool fill)
54 {
55  int index;
56  Material *ma = BKE_gpencil_object_material_ensure_by_name(bmain, ob, pct->name, &index);
57 
58  copy_v4_v4(ma->gp_style->stroke_rgba, pct->line);
59  srgb_to_linearrgb_v4(ma->gp_style->stroke_rgba, ma->gp_style->stroke_rgba);
60 
61  copy_v4_v4(ma->gp_style->fill_rgba, pct->fill);
62  srgb_to_linearrgb_v4(ma->gp_style->fill_rgba, ma->gp_style->fill_rgba);
63 
64  if (fill) {
65  ma->gp_style->flag |= GP_MATERIAL_FILL_SHOW;
66  }
67 
68  return index;
69 }
70 
71 /* ***************************************************************** */
72 /* Stroke Geometry */
73 
74 static const float data0[175 * GP_PRIM_DATABUF_SIZE] = {
75  -1.281f, 0.0f, -0.315f, 0.038f, 1.0f, -1.269f, 0.0f, -0.302f, 0.069f, 1.0f,
76  -1.261f, 0.0f, -0.293f, 0.089f, 1.0f, -1.251f, 0.0f, -0.282f, 0.112f, 1.0f,
77  -1.241f, 0.0f, -0.271f, 0.134f, 1.0f, -1.23f, 0.0f, -0.259f, 0.155f, 1.0f,
78  -1.219f, 0.0f, -0.247f, 0.175f, 1.0f, -1.208f, 0.0f, -0.234f, 0.194f, 1.0f,
79  -1.196f, 0.0f, -0.221f, 0.211f, 1.0f, -1.184f, 0.0f, -0.208f, 0.227f, 1.0f,
80  -1.172f, 0.0f, -0.194f, 0.242f, 1.0f, -1.159f, 0.0f, -0.18f, 0.256f, 1.0f,
81  -1.147f, 0.0f, -0.165f, 0.268f, 1.0f, -1.134f, 0.0f, -0.151f, 0.28f, 1.0f,
82  -1.121f, 0.0f, -0.136f, 0.29f, 1.0f, -1.108f, 0.0f, -0.121f, 0.299f, 1.0f,
83  -1.094f, 0.0f, -0.106f, 0.307f, 1.0f, -1.08f, 0.0f, -0.091f, 0.315f, 1.0f,
84  -1.066f, 0.0f, -0.076f, 0.322f, 1.0f, -1.052f, 0.0f, -0.061f, 0.329f, 1.0f,
85  -1.037f, 0.0f, -0.047f, 0.335f, 1.0f, -1.022f, 0.0f, -0.032f, 0.341f, 1.0f,
86  -1.007f, 0.0f, -0.017f, 0.346f, 1.0f, -0.991f, 0.0f, -0.003f, 0.351f, 1.0f,
87  -0.975f, 0.0f, 0.012f, 0.355f, 1.0f, -0.959f, 0.0f, 0.027f, 0.36f, 1.0f,
88  -0.942f, 0.0f, 0.041f, 0.364f, 1.0f, -0.926f, 0.0f, 0.056f, 0.368f, 1.0f,
89  -0.909f, 0.0f, 0.071f, 0.371f, 1.0f, -0.893f, 0.0f, 0.086f, 0.373f, 1.0f,
90  -0.876f, 0.0f, 0.1f, 0.376f, 1.0f, -0.859f, 0.0f, 0.115f, 0.377f, 1.0f,
91  -0.842f, 0.0f, 0.129f, 0.378f, 1.0f, -0.824f, 0.0f, 0.144f, 0.379f, 1.0f,
92  -0.807f, 0.0f, 0.158f, 0.379f, 1.0f, -0.79f, 0.0f, 0.172f, 0.379f, 1.0f,
93  -0.773f, 0.0f, 0.186f, 0.38f, 1.0f, -0.755f, 0.0f, 0.199f, 0.38f, 1.0f,
94  -0.738f, 0.0f, 0.212f, 0.381f, 1.0f, -0.721f, 0.0f, 0.224f, 0.382f, 1.0f,
95  -0.703f, 0.0f, 0.236f, 0.384f, 1.0f, -0.686f, 0.0f, 0.248f, 0.386f, 1.0f,
96  -0.67f, 0.0f, 0.26f, 0.388f, 1.0f, -0.653f, 0.0f, 0.27f, 0.39f, 1.0f,
97  -0.637f, 0.0f, 0.28f, 0.393f, 1.0f, -0.621f, 0.0f, 0.29f, 0.396f, 1.0f,
98  -0.605f, 0.0f, 0.298f, 0.399f, 1.0f, -0.589f, 0.0f, 0.306f, 0.403f, 1.0f,
99  -0.574f, 0.0f, 0.313f, 0.407f, 1.0f, -0.559f, 0.0f, 0.319f, 0.411f, 1.0f,
100  -0.544f, 0.0f, 0.325f, 0.415f, 1.0f, -0.53f, 0.0f, 0.331f, 0.42f, 1.0f,
101  -0.516f, 0.0f, 0.336f, 0.425f, 1.0f, -0.503f, 0.0f, 0.34f, 0.431f, 1.0f,
102  -0.489f, 0.0f, 0.344f, 0.437f, 1.0f, -0.477f, 0.0f, 0.347f, 0.443f, 1.0f,
103  -0.464f, 0.0f, 0.35f, 0.45f, 1.0f, -0.452f, 0.0f, 0.352f, 0.457f, 1.0f,
104  -0.44f, 0.0f, 0.354f, 0.464f, 1.0f, -0.429f, 0.0f, 0.355f, 0.471f, 1.0f,
105  -0.418f, 0.0f, 0.355f, 0.479f, 1.0f, -0.407f, 0.0f, 0.355f, 0.487f, 1.0f,
106  -0.397f, 0.0f, 0.354f, 0.495f, 1.0f, -0.387f, 0.0f, 0.353f, 0.503f, 1.0f,
107  -0.378f, 0.0f, 0.351f, 0.512f, 1.0f, -0.368f, 0.0f, 0.348f, 0.52f, 1.0f,
108  -0.36f, 0.0f, 0.344f, 0.528f, 1.0f, -0.351f, 0.0f, 0.34f, 0.537f, 1.0f,
109  -0.344f, 0.0f, 0.336f, 0.545f, 1.0f, -0.336f, 0.0f, 0.33f, 0.553f, 1.0f,
110  -0.329f, 0.0f, 0.324f, 0.562f, 1.0f, -0.322f, 0.0f, 0.318f, 0.57f, 1.0f,
111  -0.316f, 0.0f, 0.31f, 0.579f, 1.0f, -0.311f, 0.0f, 0.303f, 0.588f, 1.0f,
112  -0.306f, 0.0f, 0.294f, 0.597f, 1.0f, -0.301f, 0.0f, 0.285f, 0.606f, 1.0f,
113  -0.297f, 0.0f, 0.275f, 0.615f, 1.0f, -0.293f, 0.0f, 0.264f, 0.625f, 1.0f,
114  -0.29f, 0.0f, 0.253f, 0.635f, 1.0f, -0.288f, 0.0f, 0.241f, 0.644f, 1.0f,
115  -0.286f, 0.0f, 0.229f, 0.654f, 1.0f, -0.285f, 0.0f, 0.216f, 0.664f, 1.0f,
116  -0.284f, 0.0f, 0.202f, 0.675f, 1.0f, -0.283f, 0.0f, 0.188f, 0.685f, 1.0f,
117  -0.283f, 0.0f, 0.173f, 0.696f, 1.0f, -0.284f, 0.0f, 0.158f, 0.707f, 1.0f,
118  -0.285f, 0.0f, 0.142f, 0.718f, 1.0f, -0.286f, 0.0f, 0.125f, 0.729f, 1.0f,
119  -0.288f, 0.0f, 0.108f, 0.74f, 1.0f, -0.29f, 0.0f, 0.091f, 0.751f, 1.0f,
120  -0.293f, 0.0f, 0.073f, 0.761f, 1.0f, -0.295f, 0.0f, 0.054f, 0.772f, 1.0f,
121  -0.298f, 0.0f, 0.035f, 0.782f, 1.0f, -0.302f, 0.0f, 0.016f, 0.793f, 1.0f,
122  -0.305f, 0.0f, -0.004f, 0.804f, 1.0f, -0.309f, 0.0f, -0.024f, 0.815f, 1.0f,
123  -0.313f, 0.0f, -0.044f, 0.828f, 1.0f, -0.317f, 0.0f, -0.065f, 0.843f, 1.0f,
124  -0.321f, 0.0f, -0.085f, 0.86f, 1.0f, -0.326f, 0.0f, -0.106f, 0.879f, 1.0f,
125  -0.33f, 0.0f, -0.127f, 0.897f, 1.0f, -0.335f, 0.0f, -0.148f, 0.915f, 1.0f,
126  -0.339f, 0.0f, -0.168f, 0.932f, 1.0f, -0.344f, 0.0f, -0.189f, 0.947f, 1.0f,
127  -0.348f, 0.0f, -0.21f, 0.962f, 1.0f, -0.353f, 0.0f, -0.23f, 0.974f, 1.0f,
128  -0.357f, 0.0f, -0.25f, 0.985f, 1.0f, -0.361f, 0.0f, -0.27f, 0.995f, 1.0f,
129  -0.365f, 0.0f, -0.29f, 1.004f, 1.0f, -0.369f, 0.0f, -0.309f, 1.011f, 1.0f,
130  -0.372f, 0.0f, -0.328f, 1.018f, 1.0f, -0.375f, 0.0f, -0.347f, 1.024f, 1.0f,
131  -0.377f, 0.0f, -0.365f, 1.029f, 1.0f, -0.379f, 0.0f, -0.383f, 1.033f, 1.0f,
132  -0.38f, 0.0f, -0.4f, 1.036f, 1.0f, -0.38f, 0.0f, -0.417f, 1.037f, 1.0f,
133  -0.38f, 0.0f, -0.434f, 1.037f, 1.0f, -0.379f, 0.0f, -0.449f, 1.035f, 1.0f,
134  -0.377f, 0.0f, -0.464f, 1.032f, 1.0f, -0.374f, 0.0f, -0.478f, 1.029f, 1.0f,
135  -0.371f, 0.0f, -0.491f, 1.026f, 1.0f, -0.366f, 0.0f, -0.503f, 1.023f, 1.0f,
136  -0.361f, 0.0f, -0.513f, 1.021f, 1.0f, -0.354f, 0.0f, -0.523f, 1.019f, 1.0f,
137  -0.347f, 0.0f, -0.531f, 1.017f, 1.0f, -0.339f, 0.0f, -0.538f, 1.016f, 1.0f,
138  -0.33f, 0.0f, -0.543f, 1.016f, 1.0f, -0.32f, 0.0f, -0.547f, 1.016f, 1.0f,
139  -0.31f, 0.0f, -0.549f, 1.016f, 1.0f, -0.298f, 0.0f, -0.55f, 1.017f, 1.0f,
140  -0.286f, 0.0f, -0.55f, 1.017f, 1.0f, -0.274f, 0.0f, -0.548f, 1.018f, 1.0f,
141  -0.261f, 0.0f, -0.544f, 1.017f, 1.0f, -0.247f, 0.0f, -0.539f, 1.017f, 1.0f,
142  -0.232f, 0.0f, -0.533f, 1.016f, 1.0f, -0.218f, 0.0f, -0.525f, 1.015f, 1.0f,
143  -0.202f, 0.0f, -0.515f, 1.013f, 1.0f, -0.186f, 0.0f, -0.503f, 1.009f, 1.0f,
144  -0.169f, 0.0f, -0.49f, 1.005f, 1.0f, -0.151f, 0.0f, -0.475f, 0.998f, 1.0f,
145  -0.132f, 0.0f, -0.458f, 0.99f, 1.0f, -0.112f, 0.0f, -0.44f, 0.98f, 1.0f,
146  -0.091f, 0.0f, -0.42f, 0.968f, 1.0f, -0.069f, 0.0f, -0.398f, 0.955f, 1.0f,
147  -0.045f, 0.0f, -0.375f, 0.939f, 1.0f, -0.021f, 0.0f, -0.35f, 0.923f, 1.0f,
148  0.005f, 0.0f, -0.324f, 0.908f, 1.0f, 0.031f, 0.0f, -0.297f, 0.895f, 1.0f,
149  0.06f, 0.0f, -0.268f, 0.882f, 1.0f, 0.089f, 0.0f, -0.238f, 0.87f, 1.0f,
150  0.12f, 0.0f, -0.207f, 0.858f, 1.0f, 0.153f, 0.0f, -0.175f, 0.844f, 1.0f,
151  0.187f, 0.0f, -0.14f, 0.828f, 1.0f, 0.224f, 0.0f, -0.104f, 0.81f, 1.0f,
152  0.262f, 0.0f, -0.067f, 0.79f, 1.0f, 0.302f, 0.0f, -0.027f, 0.769f, 1.0f,
153  0.344f, 0.0f, 0.014f, 0.747f, 1.0f, 0.388f, 0.0f, 0.056f, 0.724f, 1.0f,
154  0.434f, 0.0f, 0.1f, 0.7f, 1.0f, 0.483f, 0.0f, 0.145f, 0.676f, 1.0f,
155  0.533f, 0.0f, 0.191f, 0.651f, 1.0f, 0.585f, 0.0f, 0.238f, 0.625f, 1.0f,
156  0.637f, 0.0f, 0.284f, 0.599f, 1.0f, 0.69f, 0.0f, 0.33f, 0.573f, 1.0f,
157  0.746f, 0.0f, 0.376f, 0.546f, 1.0f, 0.802f, 0.0f, 0.421f, 0.516f, 1.0f,
158  0.859f, 0.0f, 0.464f, 0.483f, 1.0f, 0.915f, 0.0f, 0.506f, 0.446f, 1.0f,
159  0.97f, 0.0f, 0.545f, 0.407f, 1.0f, 1.023f, 0.0f, 0.581f, 0.365f, 1.0f,
160  1.075f, 0.0f, 0.614f, 0.322f, 1.0f, 1.122f, 0.0f, 0.643f, 0.28f, 1.0f,
161  1.169f, 0.0f, 0.671f, 0.236f, 1.0f, 1.207f, 0.0f, 0.693f, 0.202f, 1.0f,
162  1.264f, 0.0f, 0.725f, 0.155f, 1.0f,
163 };
164 
165 /* ***************************************************************** */
166 /* Color Data */
167 
168 static const ColorTemplate gp_stroke_material_black = {
169  "Black",
170  {0.0f, 0.0f, 0.0f, 1.0f},
171  {0.0f, 0.0f, 0.0f, 0.0f},
172 };
173 
174 static const ColorTemplate gp_stroke_material_white = {
175  "White",
176  {1.0f, 1.0f, 1.0f, 1.0f},
177  {0.0f, 0.0f, 0.0f, 0.0f},
178 };
179 
180 static const ColorTemplate gp_stroke_material_red = {
181  "Red",
182  {1.0f, 0.0f, 0.0f, 1.0f},
183  {0.0f, 0.0f, 0.0f, 0.0f},
184 };
185 
186 static const ColorTemplate gp_stroke_material_green = {
187  "Green",
188  {0.0f, 1.0f, 0.0f, 1.0f},
189  {0.0f, 0.0f, 0.0f, 0.0f},
190 };
191 
192 static const ColorTemplate gp_stroke_material_blue = {
193  "Blue",
194  {0.0f, 0.0f, 1.0f, 1.0f},
195  {0.0f, 0.0f, 0.0f, 0.0f},
196 };
197 
198 static const ColorTemplate gp_stroke_material_grey = {
199  "Grey",
200  {0.358f, 0.358f, 0.358f, 1.0f},
201  {0.5f, 0.5f, 0.5f, 1.0f},
202 };
203 
204 /* ***************************************************************** */
205 /* Stroke API */
206 
207 /* Add a Simple stroke with colors
208  * (original design created by Daniel M. Lara and Matias Mendiola). */
209 void ED_gpencil_create_stroke(bContext *C, Object *ob, float mat[4][4])
210 {
211  Main *bmain = CTX_data_main(C);
212  Scene *scene = CTX_data_scene(C);
213  bGPdata *gpd = (bGPdata *)ob->data;
214  bGPDstroke *gps;
215 
216  /* create colors */
217  int color_black = gpencil_stroke_material(bmain, ob, &gp_stroke_material_black, false);
218  gpencil_stroke_material(bmain, ob, &gp_stroke_material_white, false);
219  gpencil_stroke_material(bmain, ob, &gp_stroke_material_red, false);
220  gpencil_stroke_material(bmain, ob, &gp_stroke_material_green, false);
221  gpencil_stroke_material(bmain, ob, &gp_stroke_material_blue, false);
222  gpencil_stroke_material(bmain, ob, &gp_stroke_material_grey, true);
223 
224  /* set first color as active and in brushes */
225  ob->actcol = color_black + 1;
226 
227  /* layers */
228  bGPDlayer *colors = BKE_gpencil_layer_addnew(gpd, "Colors", false);
229  bGPDlayer *lines = BKE_gpencil_layer_addnew(gpd, "Lines", true);
230 
231  /* frames */
232  bGPDframe *frame_color = BKE_gpencil_frame_addnew(colors, CFRA);
233  bGPDframe *frame_lines = BKE_gpencil_frame_addnew(lines, CFRA);
234  UNUSED_VARS(frame_color);
235 
236  /* generate stroke */
237  gps = BKE_gpencil_stroke_add(frame_lines, color_black, 175, 75, false);
238  BKE_gpencil_stroke_add_points(gps, data0, 175, mat);
239  BKE_gpencil_stroke_geometry_update(gpd, gps);
240 
241  /* update depsgraph */
242  DEG_id_tag_update(&gpd->id, ID_RECALC_TRANSFORM | ID_RECALC_GEOMETRY);
243  gpd->flag |= GP_DATA_CACHE_IS_DIRTY;
244 }
BKE_context.h
CTX_data_scene
struct Scene * CTX_data_scene(const bContext *C)
Definition: context.c:1034
CTX_data_main
struct Main * CTX_data_main(const bContext *C)
Definition: context.c:1018
BKE_gpencil.h
BKE_gpencil_frame_addnew
struct bGPDframe * BKE_gpencil_frame_addnew(struct bGPDlayer *gpl, int cframe)
Definition: gpencil.c:539
BKE_gpencil_stroke_add
struct bGPDstroke * BKE_gpencil_stroke_add(struct bGPDframe *gpf, int mat_idx, int totpoints, short thickness, const bool insert_at_head)
Definition: gpencil.c:871
BKE_gpencil_object_material_ensure_by_name
struct Material * BKE_gpencil_object_material_ensure_by_name(struct Main *bmain, struct Object *ob, const char *name, int *r_index)
Definition: gpencil.c:2466
GP_PRIM_DATABUF_SIZE
#define GP_PRIM_DATABUF_SIZE
Definition: BKE_gpencil.h:157
BKE_gpencil_stroke_add_points
void BKE_gpencil_stroke_add_points(struct bGPDstroke *gps, const float *array, const int totpoints, const float mat[4][4])
Definition: gpencil.c:800
BKE_gpencil_layer_addnew
struct bGPDlayer * BKE_gpencil_layer_addnew(struct bGPdata *gpd, const char *name, bool setactive)
Definition: gpencil.c:659
BKE_gpencil_geom.h
BKE_gpencil_stroke_geometry_update
void BKE_gpencil_stroke_geometry_update(struct bGPdata *gpd, struct bGPDstroke *gps)
Definition: gpencil_geom.c:1307
BKE_main.h
BKE_material.h
General operations, lookup, etc. for materials.
BLI_math.h
srgb_to_linearrgb_v4
MINLINE void srgb_to_linearrgb_v4(float linear[4], const float srgb[4])
Definition: math_color_inline.c:95
copy_v4_v4
MINLINE void copy_v4_v4(float r[4], const float a[4])
Definition: math_vector_inline.c:74
BLI_utildefines.h
UNUSED_VARS
#define UNUSED_VARS(...)
Definition: BLI_utildefines.h:734
DEG_depsgraph.h
DEG_id_tag_update
void DEG_id_tag_update(struct ID *id, int flag)
Definition: depsgraph_tag.cc:745
ID_RECALC_TRANSFORM
@ ID_RECALC_TRANSFORM
Definition: DNA_ID.h:599
ID_RECALC_GEOMETRY
@ ID_RECALC_GEOMETRY
Definition: DNA_ID.h:611
DNA_gpencil_types.h
GP_DATA_CACHE_IS_DIRTY
@ GP_DATA_CACHE_IS_DIRTY
Definition: DNA_gpencil_types.h:765
DNA_material_types.h
GP_MATERIAL_FILL_SHOW
@ GP_MATERIAL_FILL_SHOW
Definition: DNA_material_types.h:131
DNA_object_types.h
Object is a sort of wrapper for general info.
DNA_scene_types.h
CFRA
#define CFRA
Definition: DNA_scene_types.h:1999
ED_gpencil.h
C
#define C
Definition: RandGen.cpp:39
scene
Scene scene
Definition: deg_eval_copy_on_write.cc:120
gp_stroke_material_blue
static const ColorTemplate gp_stroke_material_blue
Definition: gpencil_add_stroke.c:192
gp_stroke_material_white
static const ColorTemplate gp_stroke_material_white
Definition: gpencil_add_stroke.c:174
gp_stroke_material_green
static const ColorTemplate gp_stroke_material_green
Definition: gpencil_add_stroke.c:186
gpencil_stroke_material
static int gpencil_stroke_material(Main *bmain, Object *ob, const ColorTemplate *pct, const bool fill)
Definition: gpencil_add_stroke.c:50
gp_stroke_material_grey
static const ColorTemplate gp_stroke_material_grey
Definition: gpencil_add_stroke.c:198
gp_stroke_material_black
static const ColorTemplate gp_stroke_material_black
Definition: gpencil_add_stroke.c:168
ED_gpencil_create_stroke
void ED_gpencil_create_stroke(bContext *C, Object *ob, float mat[4][4])
Definition: gpencil_add_stroke.c:209
data0
static const float data0[175 *GP_PRIM_DATABUF_SIZE]
Definition: gpencil_add_stroke.c:74
gp_stroke_material_red
static const ColorTemplate gp_stroke_material_red
Definition: gpencil_add_stroke.c:180
ColorTemplate
struct ColorTemplate ColorTemplate
ColorTemplate::name
const char * name
Definition: gpencil_add_lineart.c:47
Main
Definition: BKE_main.h:116
Material::gp_style
struct MaterialGPencilStyle * gp_style
Definition: DNA_material_types.h:222
Object::data
void * data
Definition: DNA_object_types.h:265
Generated on Tue Jan 31 2023 14:37:24 for Blender by  doxygen 1.9.1