src/vg/vg_lines.h

   1 /* Copyright (C) 2021-2022 Harry Godden (hgn) - All Rights Reserved */
   2
   3 #ifndef VG_LINES_H
   4 #define VG_LINES_H
   5
   6 #include "vg/vg.h"
   7
   8 static int debug_lines_enable = 1;
   9
  10 #ifdef VG_3D
  11  typedef v3f line_co;
  12 #else
  13  typedef v2f line_co;
  14 #endif
  15
  16 static struct vg_shader _shader_lines =
  17 {
  18    .name = "[vg] lines",
  19    .link = NULL,
  20    .vs =
  21    {
  22       .orig_file = NULL,
  23       .static_src =
  24
  25 #ifdef VG_3D
  26         "uniform mat4 uPv;"
  27         "layout (location=0) in vec3 a_co;"
  28 #else
  29         "uniform mat3 uPv;"
  30         "layout (location=0) in vec2 a_co;"
  31 #endif
  32         "layout (location=1) in vec4 a_colour;"
  33         ""
  34         "out vec4 s_colour;"
  35         ""
  36         "void main()"
  37         "{"
  38 #ifdef VG_3D
  39         "       vec4 vert_pos = uPv * vec4( a_co, 1.0 );"
  40 #else
  41         "       vec4 vert_pos = vec4( uPv * vec3( a_co, 1.0 ), 1.0 );"
  42 #endif
  43         "  s_colour = a_colour;"
  44         "       gl_Position = vert_pos;"
  45         "}"
  46    },
  47    .fs =
  48    {
  49       .orig_file = NULL,
  50       .static_src =
  51
  52         "out vec4 FragColor;"
  53         ""
  54         "in vec4 s_colour;"
  55         ""
  56         "void main()"
  57         "{"
  58         "       FragColor = s_colour;"
  59         "}"
  60    }
  61 };
  62
  63 struct
  64 {
  65         struct vg_lines_vert
  66         {
  67 #ifdef VG_3D
  68       v3f co;
  69 #else
  70                 v2f co;
  71 #endif
  72
  73                 u32 colour;
  74         }
  75         *buffer;
  76
  77         GLuint vao, vbo;
  78         u32 draw_idx, cap, buffer_size;
  79 }
  80 vg_lines;
  81
  82 static int vg_lines_init(void)
  83 {
  84    vg_info( "vg_lines_init\n" );
  85
  86    vg_convar_push( (struct vg_convar){
  87       .name = "debug_lines",
  88       .data = &debug_lines_enable,
  89       .data_type = k_convar_dtype_i32,
  90       .opt_i32 = { .min=0, .max=1, .clamp=1 },
  91       .persistent = 1
  92    });
  93
  94    vg_shader_register( &_shader_lines );
  95
  96
  97    if( vg_acquire_thread_sync(1) )
  98    {
  99       glGenVertexArrays( 1, &vg_lines.vao );
 100       glGenBuffers( 1, &vg_lines.vbo );
 101       glBindVertexArray( vg_lines.vao );
 102       glBindBuffer( GL_ARRAY_BUFFER, vg_lines.vbo );
 103
 104       vg_lines.cap = 50000;
 105       vg_lines.buffer_size = vg_lines.cap * sizeof( struct vg_lines_vert );
 106
 107       glBufferData( GL_ARRAY_BUFFER, vg_lines.buffer_size,
 108                     NULL, GL_DYNAMIC_DRAW );
 109       glBindVertexArray( vg_lines.vao );
 110
 111       if( VG_CHECK_GL_ERR() )
 112       {
 113          vg_release_thread_sync(1);
 114          vg_error( "Failed\n" );
 115       }
 116
 117       /* Pointers */
 118       glVertexAttribPointer(
 119          0,
 120          sizeof( vg_lines.buffer[0].co ) / sizeof(float),
 121          GL_FLOAT,
 122          GL_FALSE,
 123          sizeof( struct vg_lines_vert ),
 124          (void *)0
 125       );
 126       glEnableVertexAttribArray( 0 );
 127
 128       glVertexAttribPointer(
 129          1,
 130          4,
 131          GL_UNSIGNED_BYTE,
 132          GL_TRUE,
 133          sizeof( struct vg_lines_vert ),
 134          (void*)(offsetof( struct vg_lines_vert, colour ))
 135       );
 136       glEnableVertexAttribArray( 1 );
 137
 138       if( VG_CHECK_GL_ERR() )
 139          goto il_delete_buffers_and_fail;
 140
 141       /* Alloc RAM */
 142       vg_lines.buffer = malloc( vg_lines.buffer_size );
 143       if( vg_lines.buffer )
 144       {
 145          vg_success( "done\n" );
 146          vg_release_thread_sync(1);
 147          return 1;
 148       }
 149
 150 il_delete_buffers_and_fail:
 151       glDeleteBuffers( 1, &vg_lines.vbo );
 152       glDeleteVertexArrays( 1, &vg_lines.vao );
 153       vg_release_thread_sync(1);
 154       return 0;
 155    }
 156
 157    return 0;
 158 }
 159
 160 static void vg_lines_free(void *nothing)
 161 {
 162         glDeleteVertexArrays( 1, &vg_lines.vao );
 163         glDeleteBuffers( 1, &vg_lines.vbo );
 164         free( vg_lines.buffer );
 165 }
 166
 167 static void vg_lines_drawall( float* projection )
 168 {
 169         glUseProgram( _shader_lines.id );
 170
 171 #ifdef VG_3D
 172    glUniformMatrix4fv
 173 #else
 174         glUniformMatrix3fv
 175 #endif
 176    ( glGetUniformLocation( _shader_lines.id, "uPv" ), 1, GL_FALSE, projection );
 177
 178         glBindVertexArray( vg_lines.vao );
 179         glBindBuffer( GL_ARRAY_BUFFER, vg_lines.vbo );
 180
 181         glBufferSubData( GL_ARRAY_BUFFER, 0, vg_lines.draw_idx *
 182          sizeof(struct vg_lines_vert), vg_lines.buffer );
 183
 184         glEnable( GL_BLEND );
 185         glBlendFunc( GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA );
 186         glBlendEquation( GL_FUNC_ADD );
 187
 188    if( debug_lines_enable )
 189       glDrawArrays( GL_LINES, 0, vg_lines.draw_idx );
 190
 191         glDisable( GL_BLEND );
 192         vg_lines.draw_idx = 0;
 193 }
 194
 195 static void vg_line2( line_co from, line_co to, u32 fc, u32 tc )
 196 {
 197         struct vg_lines_vert *v = &vg_lines.buffer[vg_lines.draw_idx];
 198
 199 #ifdef VG_3D
 200         v3_copy( from, v[0].co );
 201         v3_copy( to, v[1].co );
 202 #else
 203         v2_copy( from, v[0].co );
 204         v2_copy( to, v[1].co );
 205 #endif
 206
 207         v[0].colour = fc;
 208         v[1].colour = tc;
 209
 210         vg_lines.draw_idx += 2;
 211 }
 212
 213 static void vg_line( line_co from, line_co to, u32 colour )
 214 {
 215         vg_line2( from, to, colour, colour );
 216 }
 217
 218 static void vg_line_box( line_co min, line_co max, u32 colour )
 219 {
 220 #ifdef VG_3D
 221    /* TODO... */
 222 #else
 223         vg_line( min, (v2f){min[0],max[1]}, colour );
 224         vg_line( (v2f){min[0],max[1]}, max, colour );
 225         vg_line( max, (v2f){max[0],min[1]}, colour );
 226         vg_line( (v2f){max[0],min[1]}, min, colour );
 227 #endif
 228 }
 229
 230 static void vg_line_boxf( boxf box, u32 colour )
 231 {
 232    v3f p000, p001, p010, p011, p100, p101, p110, p111;
 233
 234    p000[0]=box[0][0];p000[1]=box[0][1];p000[2]=box[0][2];
 235    p001[0]=box[0][0];p001[1]=box[0][1];p001[2]=box[1][2];
 236    p010[0]=box[0][0];p010[1]=box[1][1];p010[2]=box[0][2];
 237    p011[0]=box[0][0];p011[1]=box[1][1];p011[2]=box[1][2];
 238
 239    p100[0]=box[1][0];p100[1]=box[0][1];p100[2]=box[0][2];
 240    p101[0]=box[1][0];p101[1]=box[0][1];p101[2]=box[1][2];
 241    p110[0]=box[1][0];p110[1]=box[1][1];p110[2]=box[0][2];
 242    p111[0]=box[1][0];p111[1]=box[1][1];p111[2]=box[1][2];
 243
 244    vg_line( p000, p001, colour );
 245    vg_line( p001, p011, colour );
 246    vg_line( p011, p010, colour );
 247    vg_line( p010, p000, colour );
 248
 249    vg_line( p100, p101, colour );
 250    vg_line( p101, p111, colour );
 251    vg_line( p111, p110, colour );
 252    vg_line( p110, p100, colour );
 253
 254    vg_line( p100, p000, colour );
 255    vg_line( p101, p001, colour );
 256    vg_line( p110, p010, colour );
 257    vg_line( p111, p011, colour );
 258 }
 259
 260 static void vg_line_boxf_transformed( m4x3f m, boxf box, u32 colour )
 261 {
 262    v3f p000, p001, p010, p011, p100, p101, p110, p111;
 263
 264    p000[0]=box[0][0];p000[1]=box[0][1];p000[2]=box[0][2];
 265    p001[0]=box[0][0];p001[1]=box[0][1];p001[2]=box[1][2];
 266    p010[0]=box[0][0];p010[1]=box[1][1];p010[2]=box[0][2];
 267    p011[0]=box[0][0];p011[1]=box[1][1];p011[2]=box[1][2];
 268
 269    p100[0]=box[1][0];p100[1]=box[0][1];p100[2]=box[0][2];
 270    p101[0]=box[1][0];p101[1]=box[0][1];p101[2]=box[1][2];
 271    p110[0]=box[1][0];p110[1]=box[1][1];p110[2]=box[0][2];
 272    p111[0]=box[1][0];p111[1]=box[1][1];p111[2]=box[1][2];
 273
 274    m4x3_mulv( m, p000, p000 );
 275    m4x3_mulv( m, p001, p001 );
 276    m4x3_mulv( m, p010, p010 );
 277    m4x3_mulv( m, p011, p011 );
 278    m4x3_mulv( m, p100, p100 );
 279    m4x3_mulv( m, p101, p101 );
 280    m4x3_mulv( m, p110, p110 );
 281    m4x3_mulv( m, p111, p111 );
 282
 283    vg_line( p000, p001, colour );
 284    vg_line( p001, p011, colour );
 285    vg_line( p011, p010, colour );
 286    vg_line( p010, p000, colour );
 287
 288    vg_line( p100, p101, colour );
 289    vg_line( p101, p111, colour );
 290    vg_line( p111, p110, colour );
 291    vg_line( p110, p100, colour );
 292
 293    vg_line( p100, p000, colour );
 294    vg_line( p101, p001, colour );
 295    vg_line( p110, p010, colour );
 296    vg_line( p111, p011, colour );
 297
 298    vg_line( p000, p110, colour );
 299    vg_line( p100, p010, colour );
 300 }
 301
 302 static void vg_line_cross(v3f pos,u32 colour, float scale)
 303 {
 304    v3f p0, p1;
 305    v3_add( (v3f){ scale,0.0f,0.0f}, pos, p0 );
 306    v3_add( (v3f){-scale,0.0f,0.0f}, pos, p1 );
 307    vg_line( p0, p1, colour );
 308    v3_add( (v3f){0.0f, scale,0.0f}, pos, p0 );
 309    v3_add( (v3f){0.0f,-scale,0.0f}, pos, p1 );
 310    vg_line( p0, p1, colour );
 311    v3_add( (v3f){0.0f,0.0f, scale}, pos, p0 );
 312    v3_add( (v3f){0.0f,0.0f,-scale}, pos, p1 );
 313    vg_line( p0, p1, colour );
 314 }
 315
 316 static void vg_line_pt3( v3f pt, float size, u32 colour )
 317 {
 318    boxf box =
 319    {
 320       { pt[0]-size, pt[1]-size, pt[2]-size },
 321       { pt[0]+size, pt[1]+size, pt[2]+size }
 322    };
 323
 324    vg_line_boxf( box, colour );
 325 }
 326
 327 #endif /* VG_LINES_H */