Weight Paint Overlay: Refactor wire drawing
[blender.git] / source / blender / draw / intern / draw_cache_impl.h
1 /*
2  * Copyright 2016, Blender Foundation.
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  * Contributor(s): Blender Institute
19  *
20  */
21
22 /** \file draw_cache_impl.h
23  *  \ingroup draw
24  */
25
26 #ifndef __DRAW_CACHE_IMPL_H__
27 #define __DRAW_CACHE_IMPL_H__
28
29 struct CurveCache;
30 struct GPUMaterial;
31 struct GPUTexture;
32 struct GPUBatch;
33 struct GPUIndexBuf;
34 struct GPUVertBuf;
35 struct ListBase;
36 struct ModifierData;
37 struct ParticleSystem;
38 struct PTCacheEdit;
39 struct SpaceImage;
40
41 struct Curve;
42 struct Lattice;
43 struct Mesh;
44 struct MetaBall;
45 struct bGPdata;
46
47 /* Expose via BKE callbacks */
48 void DRW_mball_batch_cache_dirty_tag(struct MetaBall *mb, int mode);
49 void DRW_mball_batch_cache_free(struct MetaBall *mb);
50
51 void DRW_curve_batch_cache_dirty_tag(struct Curve *cu, int mode);
52 void DRW_curve_batch_cache_free(struct Curve *cu);
53
54 void DRW_mesh_batch_cache_dirty_tag(struct Mesh *me, int mode);
55 void DRW_mesh_batch_cache_free(struct Mesh *me);
56
57 void DRW_lattice_batch_cache_dirty_tag(struct Lattice *lt, int mode);
58 void DRW_lattice_batch_cache_free(struct Lattice *lt);
59
60 void DRW_particle_batch_cache_dirty_tag(struct ParticleSystem *psys, int mode);
61 void DRW_particle_batch_cache_free(struct ParticleSystem *psys);
62
63 void DRW_gpencil_batch_cache_dirty_tag(struct bGPdata *gpd);
64 void DRW_gpencil_batch_cache_free(struct bGPdata *gpd);
65
66 /* Curve */
67 void DRW_curve_batch_cache_create_requested(struct Object *ob);
68
69 struct GPUBatch *DRW_curve_batch_cache_get_wire_edge(struct Curve *cu);
70 struct GPUBatch *DRW_curve_batch_cache_get_normal_edge(struct Curve *cu);
71 struct GPUBatch *DRW_curve_batch_cache_get_edit_edges(struct Curve *cu);
72 struct GPUBatch *DRW_curve_batch_cache_get_edit_verts(struct Curve *cu, bool handles);
73
74 struct GPUBatch *DRW_curve_batch_cache_get_triangles_with_normals(struct Curve *cu);
75 struct GPUBatch **DRW_curve_batch_cache_get_surface_shaded(
76         struct Curve *cu, struct GPUMaterial **gpumat_array, uint gpumat_array_len);
77 struct GPUBatch *DRW_curve_batch_cache_get_wireframes_face(struct Curve *cu);
78
79 /* Metaball */
80 struct GPUBatch *DRW_metaball_batch_cache_get_triangles_with_normals(struct Object *ob);
81 struct GPUBatch **DRW_metaball_batch_cache_get_surface_shaded(struct Object *ob, struct MetaBall *mb, struct GPUMaterial **gpumat_array, uint gpumat_array_len);
82 struct GPUBatch *DRW_metaball_batch_cache_get_wireframes_face(struct Object *ob);
83
84 /* DispList */
85 void DRW_displist_vertbuf_create_pos_and_nor(struct ListBase *lb, struct GPUVertBuf *vbo);
86 void DRW_displist_vertbuf_create_pos_and_nor_and_uv_tess(
87         struct ListBase *lb, struct GPUVertBuf *vbo_pos_nor, struct GPUVertBuf *vbo_uv);
88 void DRW_displist_vertbuf_create_wireframe_data_tess(struct ListBase *lb, struct GPUVertBuf *vbo);
89 void DRW_displist_indexbuf_create_triangles_in_order(struct ListBase *lb, struct GPUIndexBuf *vbo);
90 void DRW_displist_indexbuf_create_triangles_tess_split_by_material(
91         struct ListBase *lb, struct GPUIndexBuf **ibo_mat, uint mat_len);
92
93 /* Lattice */
94 struct GPUBatch *DRW_lattice_batch_cache_get_all_edges(struct Lattice *lt, bool use_weight, const int actdef);
95 struct GPUBatch *DRW_lattice_batch_cache_get_all_verts(struct Lattice *lt);
96 struct GPUBatch *DRW_lattice_batch_cache_get_edit_verts(struct Lattice *lt);
97
98 /* Vertex Group Selection and display options */
99 struct DRW_MeshWeightState {
100         int defgroup_active;
101         int defgroup_len;
102
103         short flags;
104         char alert_mode;
105
106         /* Set of all selected bones for Multipaint. */
107         bool *defgroup_sel; /* [defgroup_len] */
108         int   defgroup_sel_count;
109 };
110
111 /* DRW_MeshWeightState.flags */
112 enum {
113         DRW_MESH_WEIGHT_STATE_MULTIPAINT          = (1 << 0),
114         DRW_MESH_WEIGHT_STATE_AUTO_NORMALIZE      = (1 << 1),
115 };
116
117 void DRW_mesh_weight_state_clear(struct DRW_MeshWeightState *wstate);
118 void DRW_mesh_weight_state_copy(struct DRW_MeshWeightState *wstate_dst, const struct DRW_MeshWeightState *wstate_src);
119 bool DRW_mesh_weight_state_compare(const struct DRW_MeshWeightState *a, const struct DRW_MeshWeightState *b);
120
121 /* Mesh */
122 void DRW_mesh_batch_cache_create_requested(struct Object *ob, struct Mesh *me);
123
124 struct GPUBatch **DRW_mesh_batch_cache_get_surface_shaded(
125         struct Mesh *me, struct GPUMaterial **gpumat_array, uint gpumat_array_len,
126         char **auto_layer_names, int **auto_layer_is_srgb, int *auto_layer_count);
127 struct GPUBatch **DRW_mesh_batch_cache_get_surface_texpaint(struct Mesh *me);
128 struct GPUBatch *DRW_mesh_batch_cache_get_surface_texpaint_single(struct Mesh *me);
129 struct GPUBatch *DRW_mesh_batch_cache_get_wire_loops(struct Mesh *me);
130 struct GPUBatch *DRW_mesh_batch_cache_get_weight_overlay_faces(struct Mesh *me);
131 struct GPUBatch *DRW_mesh_batch_cache_get_weight_overlay_verts(struct Mesh *me);
132 struct GPUBatch *DRW_mesh_batch_cache_get_all_edges(struct Mesh *me);
133 struct GPUBatch *DRW_mesh_batch_cache_get_triangles_with_normals(struct Mesh *me);
134 struct GPUBatch *DRW_mesh_batch_cache_get_triangles_with_normals_and_weights(struct Mesh *me, const struct DRW_MeshWeightState *wstate);
135 struct GPUBatch *DRW_mesh_batch_cache_get_triangles_with_normals_and_vert_colors(struct Mesh *me);
136 struct GPUBatch *DRW_mesh_batch_cache_get_triangles_with_select_id(struct Mesh *me, bool use_hide, uint select_id_offset);
137 struct GPUBatch *DRW_mesh_batch_cache_get_triangles_with_select_mask(struct Mesh *me, bool use_hide);
138 struct GPUBatch *DRW_mesh_batch_cache_get_loose_edges_with_normals(struct Mesh *me);
139 struct GPUBatch *DRW_mesh_batch_cache_get_points_with_normals(struct Mesh *me);
140 struct GPUBatch *DRW_mesh_batch_cache_get_all_verts(struct Mesh *me);
141 struct GPUBatch *DRW_mesh_batch_cache_get_fancy_edges(struct Mesh *me);
142 struct GPUBatch *DRW_mesh_batch_cache_get_edge_detection(struct Mesh *me, bool *r_is_manifold);
143 struct GPUBatch *DRW_mesh_batch_cache_get_edit_triangles(struct Mesh *me);
144 struct GPUBatch *DRW_mesh_batch_cache_get_edit_triangles_nor(struct Mesh *me);
145 struct GPUBatch *DRW_mesh_batch_cache_get_edit_triangles_lnor(struct Mesh *me);
146 struct GPUBatch *DRW_mesh_batch_cache_get_edit_vertices(struct Mesh *me);
147 struct GPUBatch *DRW_mesh_batch_cache_get_edit_loose_edges(struct Mesh *me);
148 struct GPUBatch *DRW_mesh_batch_cache_get_edit_loose_edges_nor(struct Mesh *me);
149 struct GPUBatch *DRW_mesh_batch_cache_get_edit_loose_verts(struct Mesh *me);
150 struct GPUBatch *DRW_mesh_batch_cache_get_edit_facedots(struct Mesh *me);
151 /* edit-mesh selection (use generic function for faces) */
152 struct GPUBatch *DRW_mesh_batch_cache_get_facedots_with_select_id(struct Mesh *me, uint select_id_offset);
153 struct GPUBatch *DRW_mesh_batch_cache_get_edges_with_select_id(struct Mesh *me, uint select_id_offset);
154 struct GPUBatch *DRW_mesh_batch_cache_get_verts_with_select_id(struct Mesh *me, uint select_id_offset);
155 /* Object mode Wireframe overlays */
156 struct GPUBatch *DRW_mesh_batch_cache_get_wireframes_face(struct Mesh *me);
157
158 void DRW_mesh_cache_sculpt_coords_ensure(struct Mesh *me);
159
160 enum {
161         UVEDIT_EDGES          = (1 << 0),
162         UVEDIT_DATA           = (1 << 1),
163         UVEDIT_FACEDOTS       = (1 << 2),
164         UVEDIT_FACES          = (1 << 3),
165         UVEDIT_STRETCH_ANGLE  = (1 << 4),
166         UVEDIT_STRETCH_AREA   = (1 << 5),
167         UVEDIT_SYNC_SEL       = (1 << 6),
168 };
169
170 /* For Image UV editor. */
171 struct GPUBatch *DRW_mesh_batch_cache_get_texpaint_loop_wire(struct Mesh *me);
172 void DRW_mesh_cache_uvedit(
173         struct Object *me, struct SpaceImage *sima, struct Scene *scene, uchar state,
174         struct GPUBatch **faces, struct GPUBatch **edges, struct GPUBatch **verts, struct GPUBatch **facedots);
175
176 /* Edit mesh bitflags (is this the right place?) */
177
178 enum {
179         VFLAG_VERTEX_ACTIVE   = 1 << 0,
180         VFLAG_VERTEX_SELECTED = 1 << 1,
181         VFLAG_VERTEX_EXISTS   = 1 << 2,
182         VFLAG_FACE_ACTIVE     = 1 << 3,
183         VFLAG_FACE_SELECTED   = 1 << 4,
184         VFLAG_FACE_FREESTYLE  = 1 << 5,
185         /* Beware to not go over 1 << 7 (it's a byte flag)
186          * (see gpu_shader_edit_mesh_overlay_geom.glsl) */
187 };
188
189 enum {
190         VFLAG_EDGE_EXISTS   = 1 << 0,
191         VFLAG_EDGE_ACTIVE   = 1 << 1,
192         VFLAG_EDGE_SELECTED = 1 << 2,
193         VFLAG_EDGE_SEAM     = 1 << 3,
194         VFLAG_EDGE_SHARP    = 1 << 4,
195         VFLAG_EDGE_FREESTYLE = 1 << 5,
196         /* Beware to not go over 1 << 7 (it's a byte flag)
197          * (see gpu_shader_edit_mesh_overlay_geom.glsl) */
198 };
199
200 /* Particles */
201 struct GPUBatch *DRW_particles_batch_cache_get_hair(
202         struct Object *object, struct ParticleSystem *psys, struct ModifierData *md);
203 struct GPUBatch *DRW_particles_batch_cache_get_dots(
204         struct Object *object, struct ParticleSystem *psys);
205 struct GPUBatch *DRW_particles_batch_cache_get_edit_strands(
206         struct Object *object, struct ParticleSystem *psys, struct PTCacheEdit *edit, bool use_weight);
207 struct GPUBatch *DRW_particles_batch_cache_get_edit_inner_points(
208         struct Object *object, struct ParticleSystem *psys, struct PTCacheEdit *edit);
209 struct GPUBatch *DRW_particles_batch_cache_get_edit_tip_points(
210         struct Object *object, struct ParticleSystem *psys, struct PTCacheEdit *edit);
211
212 /* Common */
213 #define DRW_ADD_FLAG_FROM_VBO_REQUEST(flag, vbo, value) (flag |= DRW_vbo_requested(vbo) ? value : 0)
214 #define DRW_ADD_FLAG_FROM_IBO_REQUEST(flag, ibo, value) (flag |= DRW_ibo_requested(ibo) ? value : 0)
215
216 /* Test and assign NULL if test fails */
217 #define DRW_TEST_ASSIGN_VBO(v) (v = (DRW_vbo_requested(v) ? v : NULL))
218 #define DRW_TEST_ASSIGN_IBO(v) (v = (DRW_ibo_requested(v) ? v : NULL))
219
220 struct GPUBatch *DRW_batch_request(struct GPUBatch **batch);
221 bool DRW_batch_requested(struct GPUBatch *batch, int prim_type);
222 void DRW_ibo_request(struct GPUBatch *batch, struct GPUIndexBuf **ibo);
223 bool DRW_ibo_requested(struct GPUIndexBuf *ibo);
224 void DRW_vbo_request(struct GPUBatch *batch, struct GPUVertBuf **vbo);
225 bool DRW_vbo_requested(struct GPUVertBuf *vbo);
226
227 #endif /* __DRAW_CACHE_IMPL_H__ */