ext_csr_substance.c

   1 // Copyright (C) 2021 Harry Godden (hgn)
   2
   3 // Basic buffers-only version of csRadar
   4 //=======================================================================================================================
   5
   6 #include "csRadar.h"
   7
   8 float min_height = 0.f;
   9 v4f remapping_range;
  10
  11 // GBuffer shader
  12 void frag_gbuffer( void *dest, vmf_vert tri[3], float bca, float bcb, float bcc );
  13 void frag_gbuffer_clear( void *dest );
  14
  15 csr_shader shader_gbuffer =
  16 {
  17         .stride = sizeof(float)*8, // (origin) x,y, (actual height) z
  18         .frag = frag_gbuffer,
  19         .clear = frag_gbuffer_clear
  20 };
  21
  22 // Main drawing function
  23 void draw_buffers( csr_api *api, int bounds_only );
  24
  25 // Extension implementation
  26 // =========================================================================================================
  27
  28 // API ENTRY
  29 void csr_ext_main( csr_api *api )
  30 {
  31         if( !csr_init( api ) )
  32                 return;
  33
  34         csr_create_target( &api->target, api->resolution, api->resolution, api->sampling_mode, &shader_gbuffer );
  35         csr_rt_clear( &api->target );
  36
  37         // Compute bounds, collect models
  38         draw_buffers( api, 1 );
  39         csr_auto_fit( &api->target, api->padding );
  40         vmf_load_models( api->map );
  41
  42         min_height = api->target.bounds[0][2];
  43         v2_copy( api->target.bounds[0], remapping_range );
  44         v2_sub( api->target.bounds[1], api->target.bounds[0], &remapping_range[2] );
  45
  46         api->target.bounds[0][2] = api->min_z;
  47         api->target.bounds[1][2] = api->max_z;
  48
  49         // Draw everything
  50         draw_buffers( api, 0 );
  51 }
  52
  53 void csr_ext_exit( csr_api *api )
  54 {
  55         csr_rt_free( &api->target );
  56 }
  57
  58 void draw_buffers( csr_api *api, int bounds_only )
  59 {
  60         csr_filter filter = { .compute_bounds_only = bounds_only };
  61         vmf_map *map = api->map;
  62
  63         if( api->num_strings == 1 )
  64         {
  65                 // Draw everything
  66                 csr_vmf_render( &api->target, map, map->root, &filter, NULL, NULL );
  67
  68                 if( !bounds_only )
  69                 {
  70                         csr_rt_save_buffers( &api->target, api->output_path, "all" );
  71                         csr_rt_save_tga( &api->target, "test_test_test.tga", 3*sizeof(float), 4 );
  72                 }
  73         }
  74         else
  75         {
  76                 // Draw groups
  77                 for( int i = 1; i < api->num_strings; i ++ )
  78                 {
  79                         csr_filter_update_from_strings( &filter, &api->strings[i] );
  80
  81                         csr_vmf_render( &api->target, map, map->root, &filter, NULL, NULL );
  82
  83                         if( !bounds_only )
  84                         {
  85                                 csr_rt_save_buffers( &api->target, api->output_path, api->strings[i].str );
  86                                 //csr_rt_save_c32f( ... );
  87                                 //csr_rt_save_tga( ... );
  88
  89                                 // tar_write_dds( ... );
  90                                 csr_rt_save_tga( &api->target, "test_test_test.tga", 3*sizeof(float), 4 );
  91
  92                                 csr_rt_clear( &api->target );
  93                         }
  94                 }
  95         }
  96 }
  97
  98 void frag_gbuffer( void *dest, vmf_vert tri[3], float bca, float bcb, float bcc )
  99 {
 100         float *dest_colour = (float *)dest;
 101
 102         // Position
 103         v2_sub( tri[0].origin, remapping_range, dest_colour );
 104         v2_div( dest_colour, &remapping_range[2], dest_colour );
 105
 106         dest_colour[2] = tri[0].co[2]*bca + tri[1].co[2]*bcb + tri[2].co[2]*bcc;
 107
 108         // Normals
 109         v3_muls( tri[0].nrm, bca, dest_colour+3 );
 110         v3_muladds( dest_colour+3, tri[1].nrm, bcb, dest_colour+3 );
 111         v3_muladds( dest_colour+3, tri[2].nrm, bcc, dest_colour+3 );
 112
 113         v3_muls( dest_colour+3, 0.5f, dest_colour+3 );
 114         v3_add( (v3f){0.5f,0.5f,0.5f}, dest_colour+3, dest_colour+3 );
 115
 116         // Mask
 117         dest_colour[6] = 1.f;
 118 }
 119
 120 void frag_gbuffer_clear( void *dest )
 121 {
 122         float *dest_colour = (float *)dest;
 123         dest_colour[0] = 0.f;
 124         dest_colour[1] = 0.f;
 125         dest_colour[2] = min_height;
 126
 127         dest_colour[3] = 0.5f;
 128         dest_colour[4] = 0.5f;
 129         dest_colour[5] = 1.0f;
 130
 131         dest_colour[6] = 0.f;
 132 }