Blender  V3.3
MOD_gpencilmirror.c
Go to the documentation of this file.
1 /* SPDX-License-Identifier: GPL-2.0-or-later
2  * Copyright 2018 Blender Foundation. */
3 
8 #include <stdio.h>
9 
10 #include "BLI_utildefines.h"
11 
12 #include "BLI_listbase.h"
13 #include "BLI_math_vector.h"
14 
15 #include "BLT_translation.h"
16 
17 #include "DNA_defaults.h"
19 #include "DNA_gpencil_types.h"
20 #include "DNA_object_types.h"
21 #include "DNA_scene_types.h"
22 #include "DNA_screen_types.h"
23 
24 #include "BKE_context.h"
25 #include "BKE_gpencil.h"
26 #include "BKE_gpencil_geom.h"
27 #include "BKE_gpencil_modifier.h"
28 #include "BKE_lib_query.h"
29 #include "BKE_main.h"
30 #include "BKE_modifier.h"
31 #include "BKE_scene.h"
32 #include "BKE_screen.h"
33 
34 #include "UI_interface.h"
35 #include "UI_resources.h"
36 
38 #include "MOD_gpencil_ui_common.h"
39 #include "MOD_gpencil_util.h"
40 
41 #include "DEG_depsgraph.h"
42 #include "DEG_depsgraph_build.h"
43 #include "DEG_depsgraph_query.h"
44 
45 static void initData(GpencilModifierData *md)
46 {
48 
49  BLI_assert(MEMCMP_STRUCT_AFTER_IS_ZERO(gpmd, modifier));
50 
52 }
53 
54 static void copyData(const GpencilModifierData *md, GpencilModifierData *target)
55 {
57 }
58 
59 /* Mirror is using current object as origin. */
60 static void update_mirror_local(bGPDstroke *gps, int axis)
61 {
62  int i;
63  bGPDspoint *pt;
64  float factor[3] = {1.0f, 1.0f, 1.0f};
65  factor[axis] = -1.0f;
66 
67  for (i = 0, pt = gps->points; i < gps->totpoints; i++, pt++) {
68  mul_v3_v3(&pt->x, factor);
69  }
70 }
71 
72 /* Mirror is using other object as origin. */
73 static void update_mirror_object(Object *ob,
75  bGPDstroke *gps,
76  int axis)
77 {
78  float mtx[4][4];
79  unit_m4(mtx);
80  mtx[axis][axis] = -1.0f;
81 
82  float tmp[4][4];
83  float itmp[4][4];
84  invert_m4_m4(tmp, mmd->object->obmat);
85  mul_m4_m4m4(tmp, tmp, ob->obmat);
86  invert_m4_m4(itmp, tmp);
87  mul_m4_series(mtx, itmp, mtx, tmp);
88 
89  for (int i = 0; i < gps->totpoints; i++) {
90  mul_m4_v3(mtx, &gps->points[i].x);
91  }
92 }
93 
94 static void update_position(Object *ob, MirrorGpencilModifierData *mmd, bGPDstroke *gps, int axis)
95 {
96  if (mmd->object == NULL) {
97  update_mirror_local(gps, axis);
98  }
99  else {
100  update_mirror_object(ob, mmd, gps, axis);
101  }
102 }
103 
104 static void generate_geometry(
105  GpencilModifierData *md, Object *ob, bGPDlayer *gpl, bGPDframe *gpf, const bool update)
106 {
108  bGPdata *gpd = ob->data;
109  bGPDstroke *gps, *gps_new = NULL;
110  int tot_strokes;
111  int i;
112 
113  /* check each axis for mirroring */
114  for (int xi = 0; xi < 3; xi++) {
115  if (mmd->flag & (GP_MIRROR_AXIS_X << xi)) {
116 
117  /* count strokes to avoid infinite loop after adding new strokes to tail of listbase */
118  tot_strokes = BLI_listbase_count(&gpf->strokes);
119 
120  for (i = 0, gps = gpf->strokes.first; i < tot_strokes; i++, gps = gps->next) {
122  mmd->layername,
123  mmd->material,
124  mmd->pass_index,
125  mmd->layer_pass,
126  1,
127  gpl,
128  gps,
133  gps_new = BKE_gpencil_stroke_duplicate(gps, true, true);
134  update_position(ob, mmd, gps_new, xi);
135  if (update) {
137  }
138  BLI_addtail(&gpf->strokes, gps_new);
139  }
140  }
141  }
142  }
143 }
144 
145 /* Generic "generateStrokes" callback */
147 {
149  bGPdata *gpd = (bGPdata *)ob->data;
150 
151  LISTBASE_FOREACH (bGPDlayer *, gpl, &gpd->layers) {
153  if (gpf == NULL) {
154  continue;
155  }
156  generate_geometry(md, ob, gpl, gpf, false);
157  }
158 }
159 
160 static void bakeModifier(Main *UNUSED(bmain),
163  Object *ob)
164 {
166  bGPdata *gpd = ob->data;
167  int oldframe = (int)DEG_get_ctime(depsgraph);
168 
169  LISTBASE_FOREACH (bGPDlayer *, gpl, &gpd->layers) {
170  LISTBASE_FOREACH (bGPDframe *, gpf, &gpl->frames) {
171  /* apply mirror effects on this frame */
172  scene->r.cfra = gpf->framenum;
174 
175  /* compute mirror effects on this frame */
176  generate_geometry(md, ob, gpl, gpf, true);
177  }
178  }
179 
180  /* return frame state and DB to original state */
181  scene->r.cfra = oldframe;
183 }
184 
185 static bool isDisabled(GpencilModifierData *UNUSED(md), int UNUSED(userRenderParams))
186 {
187  // MirrorGpencilModifierData *mmd = (MirrorGpencilModifierData *)md;
188 
189  return false;
190 }
191 
194  const int UNUSED(mode))
195 {
197  if (lmd->object != NULL) {
198  DEG_add_object_relation(ctx->node, lmd->object, DEG_OB_COMP_GEOMETRY, "Mirror Modifier");
199  DEG_add_object_relation(ctx->node, lmd->object, DEG_OB_COMP_TRANSFORM, "Mirror Modifier");
200  }
201  DEG_add_object_relation(ctx->node, ctx->object, DEG_OB_COMP_TRANSFORM, "Mirror Modifier");
202 }
203 
204 static void foreachIDLink(GpencilModifierData *md, Object *ob, IDWalkFunc walk, void *userData)
205 {
207 
208  walk(userData, ob, (ID **)&mmd->material, IDWALK_CB_USER);
209  walk(userData, ob, (ID **)&mmd->object, IDWALK_CB_NOP);
210 }
211 
212 static void panel_draw(const bContext *UNUSED(C), Panel *panel)
213 {
214  uiLayout *row;
215  uiLayout *layout = panel->layout;
217 
219 
220  uiLayoutSetPropSep(layout, true);
221 
222  row = uiLayoutRowWithHeading(layout, true, IFACE_("Axis"));
223  uiItemR(row, ptr, "use_axis_x", toggles_flag, NULL, ICON_NONE);
224  uiItemR(row, ptr, "use_axis_y", toggles_flag, NULL, ICON_NONE);
225  uiItemR(row, ptr, "use_axis_z", toggles_flag, NULL, ICON_NONE);
226 
227  uiItemR(layout, ptr, "object", 0, NULL, ICON_NONE);
228 
230 }
231 
232 static void mask_panel_draw(const bContext *UNUSED(C), Panel *panel)
233 {
234  gpencil_modifier_masking_panel_draw(panel, true, false);
235 }
236 
237 static void panelRegister(ARegionType *region_type)
238 {
242  region_type, "mask", "Influence", NULL, mask_panel_draw, panel_type);
243 }
244 
246  /* name */ N_("Mirror"),
247  /* structName */ "MirrorGpencilModifierData",
248  /* structSize */ sizeof(MirrorGpencilModifierData),
251 
252  /* copyData */ copyData,
253 
254  /* deformStroke */ NULL,
255  /* generateStrokes */ generateStrokes,
256  /* bakeModifier */ bakeModifier,
257  /* remapTime */ NULL,
258 
259  /* initData */ initData,
260  /* freeData */ NULL,
261  /* isDisabled */ isDisabled,
262  /* updateDepsgraph */ updateDepsgraph,
263  /* dependsOnTime */ NULL,
264  /* foreachIDLink */ foreachIDLink,
265  /* foreachTexLink */ NULL,
266  /* panelRegister */ panelRegister,
267 };
struct bGPDstroke * BKE_gpencil_stroke_duplicate(struct bGPDstroke *gps_src, bool dup_points, bool dup_curve)
Definition: gpencil.c:855
void BKE_gpencil_stroke_geometry_update(struct bGPdata *gpd, struct bGPDstroke *gps)
void BKE_gpencil_modifier_copydata_generic(const struct GpencilModifierData *md_src, struct GpencilModifierData *md_dst)
@ eGpencilModifierTypeFlag_SupportsEditmode
struct bGPDframe * BKE_gpencil_frame_retime_get(struct Depsgraph *depsgraph, struct Scene *scene, struct Object *ob, struct bGPDlayer *gpl)
@ eGpencilModifierTypeType_Gpencil
@ IDWALK_CB_USER
Definition: BKE_lib_query.h:73
@ IDWALK_CB_NOP
Definition: BKE_lib_query.h:33
void(* IDWalkFunc)(void *userData, struct Object *ob, struct ID **idpoin, int cb_flag)
Definition: BKE_modifier.h:107
void BKE_scene_graph_update_for_newframe(struct Depsgraph *depsgraph)
Definition: scene.cc:2728
#define BLI_assert(a)
Definition: BLI_assert.h:46
#define LISTBASE_FOREACH(type, var, list)
Definition: BLI_listbase.h:336
void BLI_addtail(struct ListBase *listbase, void *vlink) ATTR_NONNULL(1)
Definition: listbase.c:80
int BLI_listbase_count(const struct ListBase *listbase) ATTR_WARN_UNUSED_RESULT ATTR_NONNULL(1)
void mul_m4_m4m4(float R[4][4], const float A[4][4], const float B[4][4])
Definition: math_matrix.c:259
void unit_m4(float m[4][4])
Definition: rct.c:1090
bool invert_m4_m4(float R[4][4], const float A[4][4])
Definition: math_matrix.c:1287
void mul_m4_v3(const float M[4][4], float r[3])
Definition: math_matrix.c:729
#define mul_m4_series(...)
MINLINE void mul_v3_v3(float r[3], const float a[3])
#define UNUSED(x)
#define MEMCMP_STRUCT_AFTER_IS_ZERO(struct_var, member)
#define MEMCPY_STRUCT_AFTER(struct_dst, struct_src, member)
#define IFACE_(msgid)
struct Depsgraph Depsgraph
Definition: DEG_depsgraph.h:35
void DEG_add_object_relation(struct DepsNodeHandle *node_handle, struct Object *object, eDepsObjectComponentType component, const char *description)
@ DEG_OB_COMP_GEOMETRY
@ DEG_OB_COMP_TRANSFORM
float DEG_get_ctime(const Depsgraph *graph)
struct Scene * DEG_get_evaluated_scene(const struct Depsgraph *graph)
#define DNA_struct_default_get(struct_name)
Definition: DNA_defaults.h:29
struct MirrorGpencilModifierData MirrorGpencilModifierData
@ GP_MIRROR_INVERT_MATERIAL
@ GP_MIRROR_INVERT_LAYER
@ GP_MIRROR_INVERT_LAYERPASS
@ eGpencilModifierType_Mirror
Object is a sort of wrapper for general info.
PointerRNA * gpencil_modifier_panel_get_property_pointers(Panel *panel, PointerRNA *r_ob_ptr)
void gpencil_modifier_masking_panel_draw(Panel *panel, bool use_material, bool use_vertex)
void gpencil_modifier_panel_end(uiLayout *layout, PointerRNA *ptr)
PanelType * gpencil_modifier_subpanel_register(ARegionType *region_type, const char *name, const char *label, PanelDrawFn draw_header, PanelDrawFn draw, PanelType *parent)
PanelType * gpencil_modifier_panel_register(ARegionType *region_type, GpencilModifierType type, PanelDrawFn draw)
bool is_stroke_affected_by_modifier(Object *ob, char *mlayername, Material *material, const int mpassindex, const int gpl_passindex, const int minpoints, bGPDlayer *gpl, bGPDstroke *gps, const bool inv1, const bool inv2, const bool inv3, const bool inv4)
static void update_position(Object *ob, MirrorGpencilModifierData *mmd, bGPDstroke *gps, int axis)
static void updateDepsgraph(GpencilModifierData *md, const ModifierUpdateDepsgraphContext *ctx, const int UNUSED(mode))
static bool isDisabled(GpencilModifierData *UNUSED(md), int UNUSED(userRenderParams))
static void update_mirror_object(Object *ob, MirrorGpencilModifierData *mmd, bGPDstroke *gps, int axis)
static void mask_panel_draw(const bContext *UNUSED(C), Panel *panel)
static void foreachIDLink(GpencilModifierData *md, Object *ob, IDWalkFunc walk, void *userData)
static void bakeModifier(Main *UNUSED(bmain), Depsgraph *depsgraph, GpencilModifierData *md, Object *ob)
static void copyData(const GpencilModifierData *md, GpencilModifierData *target)
static void update_mirror_local(bGPDstroke *gps, int axis)
static void panel_draw(const bContext *UNUSED(C), Panel *panel)
GpencilModifierTypeInfo modifierType_Gpencil_Mirror
static void panelRegister(ARegionType *region_type)
static void generate_geometry(GpencilModifierData *md, Object *ob, bGPDlayer *gpl, bGPDframe *gpf, const bool update)
static void initData(GpencilModifierData *md)
static void generateStrokes(GpencilModifierData *md, Depsgraph *depsgraph, Object *ob)
#define C
Definition: RandGen.cpp:25
uiLayout * uiLayoutRowWithHeading(uiLayout *layout, bool align, const char *heading)
void uiLayoutSetPropSep(uiLayout *layout, bool is_sep)
@ UI_ITEM_R_TOGGLE
@ UI_ITEM_R_FORCE_BLANK_DECORATE
void uiItemR(uiLayout *layout, struct PointerRNA *ptr, const char *propname, int flag, const char *name, int icon)
Scene scene
const Depsgraph * depsgraph
static ulong * next
static void update(bNodeTree *ntree)
Definition: DNA_ID.h:368
void * first
Definition: DNA_listBase.h:31
Definition: BKE_main.h:121
struct DepsNodeHandle * node
Definition: BKE_modifier.h:134
float obmat[4][4]
void * data
struct uiLayout * layout
struct RenderData r
ListBase strokes
bGPDspoint * points
ListBase layers
#define N_(msgid)
PointerRNA * ptr
Definition: wm_files.c:3480