animation board tweaks

[carveJwlIkooP6JGAAIwe30JlM.git] / player_render.c

#ifndef PLAYER_RENDER_C
#define PLAYER_RENDER_C

#include "player.h"
#include "player_render.h"
#include "camera.h"
#include "player_model.h"
#include "ent_skateshop.h"
#include "audio.h"

#include "shaders/model_character_view.h"
#include "shaders/model_board_view.h"
#include "shaders/model_entity.h"

VG_STATIC void player_avatar_load( struct player_avatar *av, const char *path )
{
   mdl_open( &av->meta, path, vg_mem.rtmemory );
   mdl_load_metadata_block( &av->meta, vg_mem.rtmemory );
   mdl_load_animation_block( &av->meta, vg_mem.rtmemory );
   mdl_close( &av->meta );

   struct skeleton *sk = &av->sk;
   skeleton_setup( sk, vg_mem.rtmemory, &av->meta );

   av->id_hip        = skeleton_bone_id( sk, "hips" );
   av->id_ik_hand_l  = skeleton_bone_id( sk, "hand.IK.L" );
   av->id_ik_hand_r  = skeleton_bone_id( sk, "hand.IK.R" );
   av->id_ik_elbow_l = skeleton_bone_id( sk, "elbow.L" );
   av->id_ik_elbow_r = skeleton_bone_id( sk, "elbow.R" );
   av->id_head       = skeleton_bone_id( sk, "head" );
   av->id_ik_foot_l  = skeleton_bone_id( sk, "foot.IK.L" );
   av->id_ik_foot_r  = skeleton_bone_id( sk, "foot.IK.R" );
   av->id_board      = skeleton_bone_id( sk, "board" );
   av->id_wheel_l    = skeleton_bone_id( sk, "wheel.L" );
   av->id_wheel_r    = skeleton_bone_id( sk, "wheel.R" );
   av->id_ik_knee_l  = skeleton_bone_id( sk, "knee.L" );
   av->id_ik_knee_r  = skeleton_bone_id( sk, "knee.R" );
}

/* TODO: Standard model load */

VG_STATIC void dynamic_model_load( mdl_context *ctx,
                                   struct dynamic_model_1texture *mdl, 
                                   const char *path )
{
   if( !mdl_arrcount( &ctx->textures ) )
      vg_fatal_error( "No texture in model" );

   mdl_texture *tex0 = mdl_arritm( &ctx->textures, 0 );
   void *data = vg_linear_alloc( vg_mem.scratch, tex0->file.pack_size );
   mdl_fread_pack_file( ctx, &tex0->file, data );

   vg_tex2d_load_qoi_async( data, tex0->file.pack_size,
                            VG_TEX2D_NEAREST|VG_TEX2D_CLAMP,
                            &mdl->texture );

   mdl_async_load_glmesh( ctx, &mdl->mesh );
}

VG_STATIC void dynamic_model_unload( struct dynamic_model_1texture *mdl ){
   mesh_free( &mdl->mesh );
   glDeleteTextures( 1, &mdl->texture );
}

/* TODO: allow error handling */
VG_STATIC void player_board_load( struct player_board *board, 
                                  const char *path ){

   vg_linear_clear( vg_mem.scratch );

   mdl_context ctx;
   mdl_open( &ctx, path, vg_mem.scratch );
   mdl_load_metadata_block( &ctx, vg_mem.scratch );

   dynamic_model_load( &ctx, &board->mdl, path );

   mdl_array_ptr markers;
   mdl_load_array( &ctx, &markers, "ent_marker", vg_mem.scratch );

   /* TODO: you get put into a new section, the above is standard mdl loads. */
   for( int i=0; i<4; i++ )
      board->wheels[i].indice_count = 0;
   for( int i=0; i<2; i++ )
      board->trucks[i].indice_count = 0;
   board->board.indice_count = 0;

   for( u32 i=0; i<mdl_arrcount(&ctx.meshs); i++ ){
      mdl_mesh *mesh = mdl_arritm( &ctx.meshs, i );

      if( mdl_entity_id_type( mesh->entity_id ) != k_ent_marker )
         continue;

      u32 index = mdl_entity_id_id( mesh->entity_id );
      ent_marker *marker = mdl_arritm( &markers, index );

      mdl_submesh *sm0 = mdl_arritm( &ctx.submeshs, mesh->submesh_start );
      
      const char *alias = mdl_pstr( &ctx, marker->pstr_alias );
      u32 lr = marker->transform.co[0] > 0.0f? 1: 0,
          fb = marker->transform.co[2] > 0.0f? 0: 1;

      if( !strcmp( alias, "wheel" ) ){
         u32 id = fb<<1 | lr;
         board->wheels[ id ] = *sm0;
         v3_copy( marker->transform.co, board->wheel_positions[ id ] );
      }
      else if( !strcmp( alias, "board" ) ){
         board->board = *sm0;
         v3_copy( marker->transform.co, board->board_position );
      }
      else if( !strcmp( alias, "truck" ) ){
         board->trucks[ fb ] = *sm0;
         v3_copy( marker->transform.co, board->truck_positions[ fb ] );
      }
   }

   mdl_close( &ctx );
}

VG_STATIC void player_board_unload( struct player_board *board ){
   dynamic_model_unload( &board->mdl );
}

VG_STATIC void player_model_load( struct player_model *board, const char *path){
   vg_linear_clear( vg_mem.scratch );

   mdl_context ctx;
   mdl_open( &ctx, path, vg_mem.scratch );
   mdl_load_metadata_block( &ctx, vg_mem.scratch );

   dynamic_model_load( &ctx, &board->mdl, path );

   mdl_close( &ctx );
}

VG_STATIC void player_model_unload( struct player_model *board ){
   dynamic_model_unload( &board->mdl );
}

VG_STATIC void player__pre_render( player_instance *player )
{
   if( _player_animate[ player->subsystem ] ){
      player_animation res;
      _player_animate[ player->subsystem ]( player, &res );

      m4x3f transform;
      q_m3x3( res.root_q, transform );
      v3_copy( res.root_co, transform[3] );

      struct skeleton *sk = &player->playeravatar->sk;

      if( player->holdout_time > 0.0f ){
         skeleton_lerp_pose( sk, res.pose, player->holdout_pose, 
                                           player->holdout_time, res.pose );
         player->holdout_time -= vg.time_frame_delta * 2.0f;
      }

      skeleton_apply_pose( sk, res.pose, k_anim_apply_defer_ik );
      skeleton_apply_ik_pass( sk );
      skeleton_apply_pose( sk, res.pose, k_anim_apply_deffered_only );
      skeleton_apply_inverses( sk );
      skeleton_apply_transform( sk, transform );

      skeleton_debug( sk );
   }

   if( _player_post_animate[ player->subsystem ] )
      _player_post_animate[ player->subsystem ]( player );

   struct player_avatar *av = player->playeravatar;
   struct player_board *board = 
      addon_cache_item_if_loaded( k_addon_type_board,
                                  player->board_view_slot );

   v3f vp0, vp1;

   if( board ){
      v3_copy((v3f){0.0f,0.1f, board->truck_positions[0][2]}, vp0 );
      v3_copy((v3f){0.0f,0.1f, board->truck_positions[1][2]}, vp1 );
   }
   else{
      v3_zero( vp0 );
      v3_zero( vp1 );
   }

   struct ub_world_lighting *ubo = &world_current_instance()->ub_lighting;
   m4x3_mulv( av->sk.final_mtx[ av->id_board ], vp0, ubo->g_board_0 );
   m4x3_mulv( av->sk.final_mtx[ av->id_board ], vp1, ubo->g_board_1 );

   if( player->immobile ){
      player__cam_iterate( player );
      return;
   }

   if( player->rewinding ){
      if( player->rewind_time <= 0.0f ){
         double taken = vg.time - player->rewind_start;
         vg_success( "Rewind took (rt, pt, tl): %f, %f, %f\n", 
                           taken, player->rewind_predicted_time,
                           player->rewind_total_length );

         player->rewinding = 0;
         player->rewind_length = 1;
         player->rewind_total_length = 0.0f;
         player->rewind_accum = 0.0f;
         world_render.sky_target_rate = 1.0;
         world_static.time = world_static.last_use;
      }
      else{
         world_render.sky_target_rate = -100.0;

         float budget         = vg.time_delta,
               overall_length = player->rewind_length;

         for( int i=0; (i<10)&&(player->rewind_time>0.0f)&&(budget>0.0f); i++ ){
            /* Interpolate frames */
            int i0 = floorf( player->rewind_time ),
                i1 = VG_MIN( i0+1, player->rewind_length-1 );
            
            struct rewind_frame *fr =  &player->rewind_buffer[i0],
                                *fr1 = &player->rewind_buffer[i1];

            float dist = vg_maxf( v3_dist( fr->pos, fr1->pos ), 0.001f ),
                  subl = vg_fractf( player->rewind_time ) + 0.001f,

                  sramp = 3.0f-(1.0f/(0.4f+0.4f*player->rewind_time)),
                  speed = sramp*28.0f + 0.5f*player->rewind_time,
                  mod  = speed * (budget / dist),

                  advl = vg_minf( mod, subl ),
                  advt = (advl / mod) * budget;
            
            player->dist_accum += speed * advt;
            player->rewind_time -= advl;
            budget -= advt;
         }

         player->rewind_time = vg_maxf( 0.0f, player->rewind_time );

         float current_time = vg.time - player->rewind_start,
               remaining    = player->rewind_predicted_time - current_time;

         if( player->rewind_sound_wait ){
            if( player->rewind_predicted_time >= 6.5f ){
               if( remaining <= 6.5f ){
                  audio_lock();
                  audio_oneshot( &audio_rewind[3], 1.0f, 0.0f );
                  audio_unlock();
                  player->rewind_sound_wait = 0;
               }
            }
            else if( player->rewind_predicted_time >= 2.5f ){
               if( remaining <= 2.5f ){
                  audio_lock();
                  audio_oneshot( &audio_rewind[2], 1.0f, 0.0f );
                  audio_unlock();
                  player->rewind_sound_wait = 0;
               }
            }
            else if( player->rewind_predicted_time >= 1.5f ){
               if( remaining <= 1.5f ){
                  audio_lock();
                  audio_oneshot( &audio_rewind[1], 1.0f, 0.0f );
                  audio_unlock();
                  player->rewind_sound_wait = 0;
               }
            }
         }
         
         int i0 = floorf( player->rewind_time ),
             i1 = VG_MIN( i0+1, player->rewind_length-1 );
         
         struct rewind_frame *fr =  &player->rewind_buffer[i0],
                             *fr1 = &player->rewind_buffer[i1];
         
         float sub = vg_fractf(player->rewind_time);

         v3_lerp( fr->pos, fr1->pos, sub, player->cam_override_pos );
         player->cam_override_angles[0] = 
            vg_alerpf( fr->ang[0], fr1->ang[0], sub );
         player->cam_override_angles[1] = 
            vg_lerpf ( fr->ang[1], fr1->ang[1], sub );
         player->cam_override_fov = player->cam.fov;

         float blend = player->rewind_time * 0.25f;
         player->cam_override_strength = vg_clampf( blend, 0.0f, 1.0f );
      }
   }
   else player->cam_override_strength = 0.0f;

   player__cam_iterate( player );
}

VG_STATIC void render_board( camera *cam, world_instance *world,
                             struct player_board *board, m4x3f root,
                             struct board_pose *pose,
                             enum board_shader shader )
{
   if( !board ) return;

   v3f inverse;

   glActiveTexture( GL_TEXTURE0 );
   glBindTexture( GL_TEXTURE_2D, board->mdl.texture );

   if( shader == k_board_shader_player ){
      shader_model_board_view_use();
      shader_model_board_view_uTexMain( 0 );
      shader_model_board_view_uCamera( cam->transform[3] );
      shader_model_board_view_uPv( cam->mtx.pv );
      shader_model_board_view_uTexSceneDepth( 1 );
      render_fb_bind_texture( gpipeline.fb_main, 2, 1 );

      render_fb_inverse_ratio( gpipeline.fb_main, inverse );

      inverse[2] = main_camera.farz-main_camera.nearz;

      shader_model_board_view_uInverseRatioDepth( inverse );
      render_fb_inverse_ratio( NULL, inverse );
      inverse[2] = cam->farz-cam->nearz;
      shader_model_board_view_uInverseRatioMain( inverse );

      world_link_lighting_ub( world, _shader_model_board_view.id );
      world_bind_position_texture( world, _shader_model_board_view.id,
                                 _uniform_model_board_view_g_world_depth, 2 );
      world_bind_light_array( world, _shader_model_board_view.id,
                                 _uniform_model_board_view_uLightsArray, 3 );
      world_bind_light_index( world, _shader_model_board_view.id,
                                 _uniform_model_board_view_uLightsIndex, 4 );
   }
   else if( shader == k_board_shader_entity ){
      shader_model_entity_use();
      shader_model_entity_uTexMain( 0 );
      shader_model_entity_uCamera( cam->transform[3] );
      shader_model_entity_uPv( cam->mtx.pv );
      
      world_link_lighting_ub( world, _shader_model_entity.id );
      world_bind_position_texture( world, _shader_model_entity.id,
                                 _uniform_model_entity_g_world_depth, 2 );
      world_bind_light_array( world, _shader_model_entity.id,
                                 _uniform_model_entity_uLightsArray, 3 );
      world_bind_light_index( world, _shader_model_entity.id,
                                 _uniform_model_entity_uLightsIndex, 4 );
   }

   mesh_bind( &board->mdl.mesh );

   m4x4f m4mdl;

   if( board->board.indice_count ){
      m4x3f mlocal;
      m3x3_identity( mlocal );

      mdl_keyframe kf;
      v3_zero( kf.co );
      q_identity( kf.q );
      v3_zero( kf.s );

      v4f qroll;
      q_axis_angle( qroll, (v3f){0.0f,0.0f,1.0f}, pose->lean * 0.6f );
      keyframe_rotate_around( &kf, (v3f){0.0f,0.11f,0.0f}, 
                              (v3f){0.0f,0.0f,0.0f}, qroll );

      v3_add( board->board_position, kf.co, mlocal[3] );
      q_m3x3( kf.q, mlocal );

      m4x3_mul( root, mlocal, mlocal );

      if( shader == k_board_shader_entity ){
         /* TODO: provide a way to supply previous mdl mtx? */
         m4x3_expand( mlocal, m4mdl );
         m4x4_mul( cam->mtx_prev.pv, m4mdl, m4mdl );
         shader_model_entity_uPvmPrev( m4mdl );
         shader_model_entity_uMdl( mlocal );
      }
      else
         shader_model_board_view_uMdl( mlocal );

      mdl_draw_submesh( &board->board );
   }

   for( int i=0; i<2; i++ ){
      if( !board->trucks[i].indice_count )
         continue;

      m4x3f mlocal;
      m3x3_identity( mlocal );
      v3_copy( board->truck_positions[i], mlocal[3] );
      m4x3_mul( root, mlocal, mlocal );

      if( shader == k_board_shader_entity ){
         m4x3_expand( mlocal, m4mdl );
         m4x4_mul( cam->mtx_prev.pv, m4mdl, m4mdl );
         shader_model_entity_uPvmPrev( m4mdl );
         shader_model_entity_uMdl( mlocal );
      }
      else
         shader_model_board_view_uMdl( mlocal );

      mdl_draw_submesh( &board->trucks[i] );
   }

   for( int i=0; i<4; i++ ){
      if( !board->wheels[i].indice_count )
         continue;

      m4x3f mlocal;
      m3x3_identity( mlocal );
      v3_copy( board->wheel_positions[i], mlocal[3] );
      m4x3_mul( root, mlocal, mlocal );

      if( shader == k_board_shader_entity ){
         m4x3_expand( mlocal, m4mdl );
         m4x4_mul( cam->mtx_prev.pv, m4mdl, m4mdl );
         shader_model_entity_uPvmPrev( m4mdl );
         shader_model_entity_uMdl( mlocal );
      }
      else
         shader_model_board_view_uMdl( mlocal );

      mdl_draw_submesh( &board->wheels[i] );
   }
}

VG_STATIC void render_playermodel( camera *cam, world_instance *world,
                                   struct player_model *model,
                                   struct skeleton *skeleton ){
   if( !model ) return;
   
   shader_model_character_view_use();

        glActiveTexture( GL_TEXTURE0 );
        glBindTexture( GL_TEXTURE_2D, model->mdl.texture );
   shader_model_character_view_uTexMain( 0 );
   shader_model_character_view_uCamera( cam->transform[3] );
   shader_model_character_view_uPv( cam->mtx.pv );
   shader_model_character_view_uTexSceneDepth( 1 );
   render_fb_bind_texture( gpipeline.fb_main, 2, 1 );
   v3f inverse;
   render_fb_inverse_ratio( gpipeline.fb_main, inverse );
   inverse[2] = main_camera.farz-main_camera.nearz;

   shader_model_character_view_uInverseRatioDepth( inverse );
   render_fb_inverse_ratio( NULL, inverse );
   inverse[2] = cam->farz-cam->nearz;
   shader_model_character_view_uInverseRatioMain( inverse );

   world_link_lighting_ub( world, _shader_model_character_view.id );
   world_bind_position_texture( world, _shader_model_character_view.id,
                              _uniform_model_character_view_g_world_depth, 2 );
   world_bind_light_array( world, _shader_model_character_view.id,
                              _uniform_model_character_view_uLightsArray, 3 );
   world_bind_light_index( world, _shader_model_character_view.id,
                              _uniform_model_character_view_uLightsIndex, 4 );

   glUniformMatrix4x3fv( _uniform_model_character_view_uTransforms,
                         skeleton->bone_count,
                         0,
                         (float *)skeleton->final_mtx );
   
   mesh_bind( &model->mdl.mesh );
   mesh_draw( &model->mdl.mesh );
}

PLAYER_API void player__render( camera *cam, player_instance *player )
{
   world_instance *world = world_current_instance();
   SDL_AtomicLock( &addon_system.sl_cache_using_resources );

   struct player_model *model = 
      addon_cache_item_if_loaded( k_addon_type_player, 
                                  player->playermodel_view_slot );

   if( !model ) model = &player->fallback_model;
   render_playermodel( cam, world, model, &player->playeravatar->sk );

   struct player_board *board = 
      addon_cache_item_if_loaded( k_addon_type_board,
                                  player->board_view_slot );

   render_board( cam, world, board, player->playeravatar->sk.final_mtx[
                                    player->playeravatar->id_board],
                                    &player->board_pose,
                                    k_board_shader_player );

   SDL_AtomicUnlock( &addon_system.sl_cache_using_resources );
}

#endif /* PLAYER_RENDER_C */