diff options
Diffstat (limited to 'source/blender/draw/intern/draw_cache.c')
-rw-r--r-- | source/blender/draw/intern/draw_cache.c | 163 |
1 files changed, 162 insertions, 1 deletions
diff --git a/source/blender/draw/intern/draw_cache.c b/source/blender/draw/intern/draw_cache.c index 864a95101cf..79ca4209894 100644 --- a/source/blender/draw/intern/draw_cache.c +++ b/source/blender/draw/intern/draw_cache.c @@ -66,6 +66,9 @@ static struct DRWShapeCache { Batch *drw_speaker; Batch *drw_bone_octahedral; Batch *drw_bone_octahedral_wire; + Batch *drw_bone_box; + Batch *drw_bone_box_wire; + Batch *drw_bone_wire_wire; Batch *drw_bone_point; Batch *drw_bone_point_wire; Batch *drw_bone_arrows; @@ -103,6 +106,9 @@ void DRW_shape_cache_free(void) BATCH_DISCARD_ALL_SAFE(SHC.drw_speaker); BATCH_DISCARD_ALL_SAFE(SHC.drw_bone_octahedral); BATCH_DISCARD_ALL_SAFE(SHC.drw_bone_octahedral_wire); + BATCH_DISCARD_ALL_SAFE(SHC.drw_bone_box); + BATCH_DISCARD_ALL_SAFE(SHC.drw_bone_box_wire); + BATCH_DISCARD_ALL_SAFE(SHC.drw_bone_wire_wire); BATCH_DISCARD_ALL_SAFE(SHC.drw_bone_point); BATCH_DISCARD_ALL_SAFE(SHC.drw_bone_point_wire); BATCH_DISCARD_ALL_SAFE(SHC.drw_bone_arrows); @@ -1360,6 +1366,161 @@ Batch *DRW_cache_bone_octahedral_wire_outline_get(void) return SHC.drw_bone_octahedral_wire; } + +/* XXX TODO move that 1 unit cube to more common/generic place? */ +static const float bone_box_verts[8][3] = { + { 1.0f, 0.0f, 1.0f}, + { 1.0f, 0.0f, -1.0f}, + {-1.0f, 0.0f, -1.0f}, + {-1.0f, 0.0f, 1.0f}, + { 1.0f, 1.0f, 1.0f}, + { 1.0f, 1.0f, -1.0f}, + {-1.0f, 1.0f, -1.0f}, + {-1.0f, 1.0f, 1.0f} +}; + +static const unsigned int bone_box_wire[24] = { + 0, 1, 1, 2, 2, 3, 3, 0, + 4, 5, 5, 6, 6, 7, 7, 4, + 0, 4, 1, 5, 2, 6, 3, 7, +}; + +/* aligned with bone_octahedral_wire + * Contains adjacent normal index */ +static const unsigned int bone_box_wire_adjacent_face[24] = { + 0, 2, 0, 4, 1, 6, 1, 8, + 3, 10, 5, 10, 7, 11, 9, 11, + 3, 8, 2, 5, 4, 7, 6, 9, +}; + +static const unsigned int bone_box_solid_tris[12][3] = { + {0, 1, 2}, /* bottom */ + {0, 2, 3}, + + {0, 1, 5}, /* sides */ + {0, 5, 4}, + + {1, 2, 6}, + {1, 6, 5}, + + {2, 3, 7}, + {2, 7, 6}, + + {3, 0, 4}, + {3, 4, 7}, + + {4, 5, 6}, /* top */ + {4, 6, 7}, +}; + +/* aligned with bone_octahedral_solid_tris */ +static const float bone_box_solid_normals[12][3] = { + { 0.0f, -1.0f, 0.0f}, + { 0.0f, -1.0f, 0.0f}, + + { 1.0f, 0.0f, 0.0f}, + { 1.0f, 0.0f, 0.0f}, + + { 0.0f, 0.0f, -1.0f}, + { 0.0f, 0.0f, -1.0f}, + + {-1.0f, 0.0f, 0.0f}, + {-1.0f, 0.0f, 0.0f}, + + { 0.0f, 0.0f, 1.0f}, + { 0.0f, 0.0f, 1.0f}, + + { 0.0f, 1.0f, 0.0f}, + { 0.0f, 1.0f, 0.0f}, +}; + +Batch *DRW_cache_bone_box_get(void) +{ + if (!SHC.drw_bone_box) { + unsigned int v_idx = 0; + + static VertexFormat format = { 0 }; + static unsigned int pos_id, nor_id; + if (format.attrib_ct == 0) { + pos_id = VertexFormat_add_attrib(&format, "pos", COMP_F32, 3, KEEP_FLOAT); + nor_id = VertexFormat_add_attrib(&format, "nor", COMP_F32, 3, KEEP_FLOAT); + } + + /* Vertices */ + VertexBuffer *vbo = VertexBuffer_create_with_format(&format); + VertexBuffer_allocate_data(vbo, 36); + + for (int i = 0; i < 12; i++) { + for (int j = 0; j < 3; j++) { + VertexBuffer_set_attrib(vbo, nor_id, v_idx, bone_box_solid_normals[i]); + VertexBuffer_set_attrib(vbo, pos_id, v_idx++, bone_box_verts[bone_box_solid_tris[i][j]]); + } + } + + SHC.drw_bone_box = Batch_create(PRIM_TRIANGLES, vbo, NULL); + } + return SHC.drw_bone_box; +} + +Batch *DRW_cache_bone_box_wire_outline_get(void) +{ + if (!SHC.drw_bone_box_wire) { + unsigned int v_idx = 0; + + static VertexFormat format = { 0 }; + static unsigned int pos_id, n1_id, n2_id; + if (format.attrib_ct == 0) { + pos_id = VertexFormat_add_attrib(&format, "pos", COMP_F32, 3, KEEP_FLOAT); + n1_id = VertexFormat_add_attrib(&format, "N1", COMP_F32, 3, KEEP_FLOAT); + n2_id = VertexFormat_add_attrib(&format, "N2", COMP_F32, 3, KEEP_FLOAT); + } + + /* Vertices */ + VertexBuffer *vbo = VertexBuffer_create_with_format(&format); + VertexBuffer_allocate_data(vbo, 12 * 2); + + for (int i = 0; i < 12; i++) { + const float *co1 = bone_box_verts[bone_box_wire[i * 2]]; + const float *co2 = bone_box_verts[bone_box_wire[i * 2 + 1]]; + const float *n1 = bone_box_solid_normals[bone_box_wire_adjacent_face[i * 2]]; + const float *n2 = bone_box_solid_normals[bone_box_wire_adjacent_face[i * 2 + 1]]; + add_fancy_edge(vbo, pos_id, n1_id, n2_id, &v_idx, co1, co2, n1, n2); + } + + SHC.drw_bone_box_wire = Batch_create(PRIM_LINES, vbo, NULL); + } + return SHC.drw_bone_box_wire; +} + + +Batch *DRW_cache_bone_wire_wire_outline_get(void) +{ + if (!SHC.drw_bone_wire_wire) { + unsigned int v_idx = 0; + + static VertexFormat format = { 0 }; + static unsigned int pos_id, n1_id, n2_id; + if (format.attrib_ct == 0) { + pos_id = VertexFormat_add_attrib(&format, "pos", COMP_F32, 3, KEEP_FLOAT); + n1_id = VertexFormat_add_attrib(&format, "N1", COMP_F32, 3, KEEP_FLOAT); + n2_id = VertexFormat_add_attrib(&format, "N2", COMP_F32, 3, KEEP_FLOAT); + } + + /* Vertices */ + VertexBuffer *vbo = VertexBuffer_create_with_format(&format); + VertexBuffer_allocate_data(vbo, 2); + + const float co1[3] = {0.0f, 0.0f, 0.0f}; + const float co2[3] = {0.0f, 1.0f, 0.0f}; + const float n[3] = {1.0f, 0.0f, 0.0f}; + add_fancy_edge(vbo, pos_id, n1_id, n2_id, &v_idx, co1, co2, n, n); + + SHC.drw_bone_wire_wire = Batch_create(PRIM_LINES, vbo, NULL); + } + return SHC.drw_bone_wire_wire; +} + + Batch *DRW_cache_bone_point_get(void) { if (!SHC.drw_bone_point) { @@ -1773,4 +1934,4 @@ Batch *DRW_cache_lattice_vert_overlay_get(Object *ob) return DRW_lattice_batch_cache_get_overlay_verts(lt); } -/** \} */
\ No newline at end of file +/** \} */ |