1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
|
/* SPDX-License-Identifier: GPL-2.0-or-later */
/** \file
* \ingroup wm
*/
#include <stdio.h>
#include "BLI_ghash.h"
#include "BLI_listbase.h"
#include "BLI_utildefines.h"
#include "BKE_context.h"
#include "BKE_main.h"
#include "DNA_screen_types.h"
#include "DNA_space_types.h"
#include "MEM_guardedalloc.h"
#include "RNA_access.h"
#include "RNA_define.h"
#include "WM_api.h"
#include "WM_types.h"
#include "ED_screen.h"
/* only for own init/exit calls (wm_gizmotype_init/wm_gizmotype_free) */
#include "wm.h"
/* own includes */
#include "wm_gizmo_intern.h"
#include "wm_gizmo_wmapi.h"
/* -------------------------------------------------------------------- */
/** \name Gizmo Type Append
*
* \note This follows conventions from #WM_operatortype_find #WM_operatortype_append & friends.
* \{ */
static GHash *global_gizmotype_hash = NULL;
const wmGizmoType *WM_gizmotype_find(const char *idname, bool quiet)
{
if (idname[0]) {
wmGizmoType *gzt;
gzt = BLI_ghash_lookup(global_gizmotype_hash, idname);
if (gzt) {
return gzt;
}
if (!quiet) {
printf("search for unknown gizmo '%s'\n", idname);
}
}
else {
if (!quiet) {
printf("search for empty gizmo\n");
}
}
return NULL;
}
void WM_gizmotype_iter(GHashIterator *ghi)
{
BLI_ghashIterator_init(ghi, global_gizmotype_hash);
}
static wmGizmoType *wm_gizmotype_append__begin(void)
{
wmGizmoType *gzt = MEM_callocN(sizeof(wmGizmoType), "gizmotype");
gzt->srna = RNA_def_struct_ptr(&BLENDER_RNA, "", &RNA_GizmoProperties);
#if 0
/* Set the default i18n context now, so that opfunc can redefine it if needed! */
RNA_def_struct_translation_context(ot->srna, BLT_I18NCONTEXT_OPERATOR_DEFAULT);
ot->translation_context = BLT_I18NCONTEXT_OPERATOR_DEFAULT;
#endif
return gzt;
}
static void wm_gizmotype_append__end(wmGizmoType *gzt)
{
BLI_assert(gzt->struct_size >= sizeof(wmGizmo));
RNA_def_struct_identifier(&BLENDER_RNA, gzt->srna, gzt->idname);
BLI_ghash_insert(global_gizmotype_hash, (void *)gzt->idname, gzt);
}
void WM_gizmotype_append(void (*gtfunc)(struct wmGizmoType *))
{
wmGizmoType *gzt = wm_gizmotype_append__begin();
gtfunc(gzt);
wm_gizmotype_append__end(gzt);
}
void WM_gizmotype_append_ptr(void (*gtfunc)(struct wmGizmoType *, void *), void *userdata)
{
wmGizmoType *mt = wm_gizmotype_append__begin();
gtfunc(mt, userdata);
wm_gizmotype_append__end(mt);
}
void WM_gizmotype_free_ptr(wmGizmoType *gzt)
{
if (gzt->rna_ext.srna) { /* python gizmo, allocs own string */
MEM_freeN((void *)gzt->idname);
}
BLI_freelistN(&gzt->target_property_defs);
MEM_freeN(gzt);
}
/**
* \param C: May be NULL.
*/
static void gizmotype_unlink(bContext *C, Main *bmain, wmGizmoType *gzt)
{
/* Free instances. */
for (bScreen *screen = bmain->screens.first; screen; screen = screen->id.next) {
LISTBASE_FOREACH (ScrArea *, area, &screen->areabase) {
LISTBASE_FOREACH (SpaceLink *, sl, &area->spacedata) {
ListBase *lb = (sl == area->spacedata.first) ? &area->regionbase : &sl->regionbase;
LISTBASE_FOREACH (ARegion *, region, lb) {
wmGizmoMap *gzmap = region->gizmo_map;
if (gzmap) {
wmGizmoGroup *gzgroup;
for (gzgroup = gzmap->groups.first; gzgroup; gzgroup = gzgroup->next) {
for (wmGizmo *gz = gzgroup->gizmos.first, *gz_next; gz; gz = gz_next) {
gz_next = gz->next;
BLI_assert(gzgroup->parent_gzmap == gzmap);
if (gz->type == gzt) {
WM_gizmo_unlink(&gzgroup->gizmos, gzgroup->parent_gzmap, gz, C);
ED_region_tag_redraw_editor_overlays(region);
}
}
}
}
}
}
}
}
}
void WM_gizmotype_remove_ptr(bContext *C, Main *bmain, wmGizmoType *gzt)
{
BLI_assert(gzt == WM_gizmotype_find(gzt->idname, false));
BLI_ghash_remove(global_gizmotype_hash, gzt->idname, NULL, NULL);
gizmotype_unlink(C, bmain, gzt);
}
bool WM_gizmotype_remove(bContext *C, Main *bmain, const char *idname)
{
wmGizmoType *gzt = BLI_ghash_lookup(global_gizmotype_hash, idname);
if (gzt == NULL) {
return false;
}
WM_gizmotype_remove_ptr(C, bmain, gzt);
return true;
}
static void wm_gizmotype_ghash_free_cb(wmGizmoType *gzt)
{
WM_gizmotype_free_ptr(gzt);
}
void wm_gizmotype_free(void)
{
BLI_ghash_free(global_gizmotype_hash, NULL, (GHashValFreeFP)wm_gizmotype_ghash_free_cb);
global_gizmotype_hash = NULL;
}
void wm_gizmotype_init(void)
{
/* reserve size is set based on blender default setup */
global_gizmotype_hash = BLI_ghash_str_new_ex("wm_gizmotype_init gh", 128);
}
/** \} */
|
