allow replays to pass through rifts
[carveJwlIkooP6JGAAIwe30JlM.git] / player_replay.c
index 83b9b835544671c094d3172f0e9fb4feb4784272..9fbbfcc6e7dd07c5ae9b2433071dd8b3154e061a 100644 (file)
 #define PLAYER_REPLAY_C
 
 #include "player_replay.h"
+#include "input.h"
+#include "gui.h"
+#include "freecam.h"
+
+static void replay_clear( replay_buffer *replay ){
+   replay->head = NULL;
+   replay->tail = NULL;
+   replay->cursor_frame = NULL;
+   replay->statehead = NULL;
+   replay->cursor = -99999.9;
+}
 
-VG_STATIC void local_replay_init( u32 bytes ){
-   localplayer.replay.data = vg_linear_alloc( vg_mem.rtmemory, bytes );
-   localplayer.replay.size = bytes;
-   localplayer.replay.cursor = -9999.9;
-   vg_console_reg_var( "k_replay_test", &k_replay_test, k_var_dtype_i32, 0 );
+static 
+void * replay_frame_data( replay_frame *frame, enum replay_framedata type ){
+   void *baseptr = frame;
+   return baseptr + frame->data_table[type][0];
 }
 
-u32 replay_frame_size( replay_frame *frame ){
-   /* eventually it will be a dynamic size */
-   return vg_align8( sizeof( replay_frame ) );
+static u16
+replay_frame_calculate_data_offsets( u16 data_table[4][2] ){
+   u32 total = vg_align8( sizeof(replay_frame) );
+   for( u32 i=0; i<4; i++ ){
+      data_table[i][0] = total;
+      total += vg_align8(data_table[i][1]);
+
+      if( total > 0xffff )
+         vg_fatal_error( "Exceeded frame storage capacity\n" );
+   }
+   return total;
 }
 
-VG_STATIC replay_frame *replay_newframe( replay_buffer *replay ){
-   replay_frame *frame = NULL;
-   if( replay->head ){
-      assert( replay->head );
+static void replay_tailpop( replay_buffer *replay ){
+   if( replay->cursor_frame == replay->tail )
+      replay->cursor_frame = NULL;
+   if( replay->statehead == replay->tail )
+      replay->statehead = NULL;
 
-      u32 headsize = replay_frame_size( replay->head ),
-          nextpos  = ((void *)replay->head - replay->data) + headsize,
-          nextsize = vg_align8( sizeof(replay_frame) );
+   replay->tail = replay->tail->r;
 
-      if( nextsize > replay->size ){
-         vg_error( "Keyframe too big\n" );
-         return NULL;
-      }
+   if( replay->tail )
+      replay->tail->l = NULL;
+   else
+      replay->head = NULL;
+}
+
+static replay_frame *replay_newframe( replay_buffer *replay, 
+                                         u16 animator_size,
+                                         u16 gamestate_size,
+                                         u16 sfx_count ){
+   u16 data_table[4][2];
+   data_table[ k_replay_framedata_animator ][1]  = animator_size;
+   data_table[ k_replay_framedata_gamestate ][1] = gamestate_size;
+   data_table[ k_replay_framedata_sfx ][1]       = 0;
+   data_table[ k_replay_framedata_internal_gamestate ][1] = 0;
+   if( gamestate_size ){
+      data_table[ k_replay_framedata_internal_gamestate ][1] = 
+         sizeof( replay_gamestate );
+   }
 
-      if( nextpos + nextsize > replay->size )
+   u32 nextsize = replay_frame_calculate_data_offsets( data_table );
+
+   replay_frame *frame = NULL;
+   if( replay->head ){
+      u32 headsize = replay->head->total_size,
+          nextpos  = ((void *)replay->head - replay->data) + headsize;
+
+      if( nextpos + nextsize > replay->size ){
          nextpos = 0;
+         
+         /* maintain contiguity */
+         while( replay->tail ){
+            if( (void *)replay->tail - replay->data )
+               replay_tailpop( replay );
+            else break;
+         }
+      }
 
 check_again:;
-
       u32 tailpos = (void *)replay->tail - replay->data;
 
       if( tailpos >= nextpos ){
          if( nextpos + nextsize > tailpos ){
-            if( replay->cursor_frame == replay->tail ){
-               replay->cursor_frame = NULL;
-            }
-
-            replay->tail = replay->tail->r;
+            replay_tailpop( replay );
 
-            if( replay->tail ) {
-               replay->tail->l = NULL;
+            if( replay->tail )
                goto check_again;
-            }
-            else{
-               replay->head = NULL;
-            }
          }
       }
 
       frame = replay->data + nextpos;
 
-      if( replay->head ){
+      if( replay->head )
          replay->head->r = frame;
-      }
    }
-   else {
+   else
       frame = replay->data;
+
+   for( u32 i=0; i<4; i++ ){
+      frame->data_table[i][0] = data_table[i][0];
+      frame->data_table[i][1] = data_table[i][1];
    }
 
+   frame->total_size = nextsize;
    frame->l = replay->head;
    frame->r = NULL;
    replay->head = frame;
    if( !replay->tail ) replay->tail = frame;
+   if( gamestate_size ) replay->statehead = frame;
 
    return frame;
 }
 
-VG_STATIC void replay_seek( replay_buffer *replay, f64 t ){
+static void replay_seek( replay_buffer *replay, f64 t ){
    if( !replay->head ) return;
    assert( replay->tail );
 
@@ -93,8 +135,8 @@ VG_STATIC void replay_seek( replay_buffer *replay, f64 t ){
    if( dir == 0.0 ) return;
    dir = vg_signf( dir );
    
-   u32 i=0;
-   for( ; i<100; i++ ){
+   u32 i=4096;
+   while( i --> 0 ){
       if( dir < 0.0 )
          if( t > replay->cursor_frame->time ) break;
 
@@ -108,15 +150,390 @@ VG_STATIC void replay_seek( replay_buffer *replay, f64 t ){
          if( t < next->time ) break;
 
       replay->cursor_frame = next;
+      replay->cursor = next->time;
+
+      if( !i ) return;
    }
 
    replay->cursor = t;
 }
 
-VG_STATIC void replay_debug_info( player_instance *player ){
+static replay_frame *replay_find_recent_stateframe( replay_buffer *replay ){
+   replay_frame *frame = replay->cursor_frame;
+   u32 i=4096;
+   while( i --> 0 ){
+      if( !frame ) return frame;
+      if( frame->data_table[ k_replay_framedata_gamestate ][1] ) return frame;
+      frame = frame->l;
+   }
+
+   return NULL;
+}
+
+static f32 replay_subframe_time( replay_buffer *replay ){
+   replay_frame *frame = replay->cursor_frame;
+   if( !frame ) return 0.0f;
+   replay_frame *next = frame->r;
+   if( next ){
+      f64 l = next->time - frame->time,
+          t = (l <= (1.0/128.0))? 0.0: (replay->cursor - frame->time) / l;
+      return vg_clampf( t, 0.0f, 1.0f );
+   }
+   else 
+      return 0.0f;
+}
+
+static void replay_get_frame_camera( replay_frame *frame, camera *cam ){
+   cam->fov = frame->cam_fov;
+   v3_copy( frame->cam_pos, cam->pos );
+   v3_copy( frame->cam_angles, cam->angles );
+}
+
+static void replay_get_camera( replay_buffer *replay, camera *cam ){
+   cam->nearz = 0.1f;
+   cam->farz = 100.0f;
+   if( replay->cursor_frame ){
+      replay_frame *next = replay->cursor_frame->r;
+
+      if( next ){
+         camera temp;
+         
+         replay_get_frame_camera( replay->cursor_frame, cam );
+         replay_get_frame_camera( next, &temp );
+         camera_lerp( cam, &temp, replay_subframe_time( replay ), cam );
+      }
+      else {
+         replay_get_frame_camera( replay->cursor_frame, cam );
+      }
+   }
+   else {
+      v3_zero( cam->pos );
+      v3_zero( cam->angles );
+      cam->fov = 90.0f;
+   }
+}
+
+struct replay_rb{
+   v3f co, v, w;
+   v4f q;
+};
+
+static void skaterift_record_frame( replay_buffer *replay, 
+                                    int force_gamestate ){
+   f64 delta      = 9999999.9,
+       statedelta = 9999999.9;
+
+   if( replay->head )
+      delta = vg.time - replay->head->time;
+
+   if( replay->statehead )
+      statedelta = vg.time - replay->statehead->time;
+
+   const f64 k_replay_rate = 1.0/30.0,
+             k_gamestate_rate = 0.5;
+
+   int save_frame = 0,
+       save_state = 0;
+
+   if( force_gamestate ) save_state = 1;
+   if( statedelta > k_gamestate_rate ) save_state = 1;
+   if( delta > k_replay_rate ) save_frame = 1;
+   if( save_state ) save_frame = 1;
+
+   if( !save_frame ) return;
+
+   u16 gamestate_size = 0;
+   if( save_state ){
+      gamestate_size = (u32 []){ 
+         [k_player_subsystem_walk ] = sizeof(struct player_walk_state),
+         [k_player_subsystem_drive] = 0,
+         [k_player_subsystem_skate] = sizeof(struct player_skate_state),
+         [k_player_subsystem_dead ] = localplayer.ragdoll.part_count * 
+                                       sizeof(struct replay_rb)
+      }[ localplayer.subsystem ];
+   }
+
+   u16 animator_size = (u16 []){
+      [k_player_subsystem_walk ] = sizeof(struct player_walk_animator),
+      [k_player_subsystem_drive] = 0,
+      [k_player_subsystem_skate] = sizeof(struct player_skate_animator),
+      [k_player_subsystem_dead ] = sizeof(struct player_dead_animator)
+   }[ localplayer.subsystem ];
+   
+   replay_frame *frame = replay_newframe( replay,
+                                          animator_size, gamestate_size, 0 );
+   frame->system = localplayer.subsystem;
+
+   if( save_state ){
+      replay_gamestate *gs = 
+         replay_frame_data( frame, k_replay_framedata_internal_gamestate );
+
+      /* permanent block */
+      memcpy( &gs->rb, &localplayer.rb, sizeof(rigidbody) );
+      memcpy( &gs->cam_control, &localplayer.cam_control, 
+               sizeof(struct player_cam_controller) );
+      v3_copy( localplayer.angles, gs->angles );
+
+      void *dst = replay_frame_data( frame, k_replay_framedata_gamestate );
+
+      /* subsytem/dynamic block */
+      if( localplayer.subsystem == k_player_subsystem_walk )
+         memcpy( dst, &player_walk.state, gamestate_size );
+      else if( localplayer.subsystem == k_player_subsystem_skate )
+         memcpy( dst, &player_skate.state, gamestate_size );
+      else if( localplayer.subsystem == k_player_subsystem_dead ){
+         struct replay_rb *arr = dst;
+         for( u32 i=0; i<localplayer.ragdoll.part_count; i ++ ){
+            rigidbody *rb = &localplayer.ragdoll.parts[i].obj.rb;
+            v3_copy( rb->co, arr[i].co );
+            v3_copy( rb->w, arr[i].w );
+            v3_copy( rb->v, arr[i].v );
+            v4_copy( rb->q, arr[i].q );
+         }
+      }
+   }
+
+   replay->cursor = vg.time;
+   replay->cursor_frame = frame;
+   frame->time = vg.time;
+
+   /* camera */
+   v3_copy( localplayer.cam.pos, frame->cam_pos );
+   if( localplayer.gate_waiting ){
+      m4x3_mulv( localplayer.gate_waiting->transport, 
+                  frame->cam_pos, frame->cam_pos );
+
+      v3f v0;
+      v3_angles_vector( localplayer.cam.angles, v0 );
+      m3x3_mulv( localplayer.gate_waiting->transport, v0, v0 );
+      v3_angles( v0, frame->cam_angles );
+   }
+   else 
+      v3_copy( localplayer.cam.angles, frame->cam_angles );
+
+   frame->cam_fov = localplayer.cam.fov;
+
+   /* animator */
+   void *dst = replay_frame_data( frame, k_replay_framedata_animator );
+
+   if( localplayer.subsystem == k_player_subsystem_walk )
+      memcpy( dst, &player_walk.animator, animator_size );
+   else if( localplayer.subsystem == k_player_subsystem_skate )
+      memcpy( dst, &player_skate.animator, animator_size );
+   else if( localplayer.subsystem == k_player_subsystem_dead ){
+      memcpy( dst, &player_dead.animator, animator_size );
+   }
+}
+
+static
+void skaterift_restore_frame( replay_frame *frame ){
+   replay_gamestate *gs = 
+      replay_frame_data( frame, k_replay_framedata_internal_gamestate );
+   void *src = replay_frame_data( frame, k_replay_framedata_gamestate );
+   u16 src_size = frame->data_table[ k_replay_framedata_gamestate ][1];
+   assert( src_size );
+
+   if(frame->system == k_player_subsystem_walk ){
+      memcpy( &player_walk.state, src, src_size );
+   }
+   else if( frame->system == k_player_subsystem_skate ){
+      memcpy( &player_skate.state, src, src_size );
+   }
+   else if( frame->system == k_player_subsystem_dead ){
+      player__dead_transition(0);
+      struct replay_rb *arr = src;
+
+      for( u32 i=0; i<localplayer.ragdoll.part_count; i ++ ){
+         struct ragdoll_part *part = &localplayer.ragdoll.parts[i];
+         rigidbody *rb = &part->obj.rb;
+
+         v3_copy( arr[i].co, rb->co );
+         v3_copy( arr[i].w, rb->w );
+         v3_copy( arr[i].v, rb->v );
+         v4_copy( arr[i].q, rb->q );
+
+         v3_copy( arr[i].co, part->prev_co );
+         v4_copy( arr[i].q, part->prev_q );
+      }
+   }
+
+   localplayer.subsystem = frame->system;
+
+   memcpy( &localplayer.rb, &gs->rb, sizeof(rigidbody) );
+   v3_copy( gs->angles, localplayer.angles );
+
+   v3_copy( frame->cam_pos, localplayer.cam.pos );
+   v3_copy( frame->cam_angles, localplayer.cam.angles );
+   localplayer.cam.fov = frame->cam_fov;
+
+   memcpy( &localplayer.cam_control, &gs->cam_control, 
+            sizeof(struct player_cam_controller) );
+
+   /* chop end off replay */
+   frame->r = NULL;
+   skaterift.replay.statehead = frame;
+   skaterift.replay.head = frame;
+   skaterift.replay.cursor_frame = frame;
+   skaterift.replay.cursor = frame->time;
+   skaterift.replay_control = k_replay_control_scrub;
+   skaterift.activity = k_skaterift_default;
+   vg.time = frame->time;
+}
+
+static void skaterift_replay_resume(void){
+   replay_frame *prev = replay_find_recent_stateframe(&skaterift.replay);
+
+   if( prev ){
+      skaterift.replay_control = k_replay_control_resume;
+      skaterift.resume_target = prev;
+      skaterift.resume_begin = skaterift.replay.cursor;
+      skaterift.resume_transition = 0.0f;
+   }
+
+   gui_helper_clear();
+}
+
+static void skaterift_replay_update_helpers(void);
+
+static void skaterift_replay_pre_update(void){
+   if( skaterift.activity != k_skaterift_replay ) return;
+
+   if( skaterift.replay_control == k_replay_control_resume ){
+      if( skaterift.replay.cursor_frame == skaterift.resume_target ||
+          skaterift.replay.cursor_frame == NULL ){
+         skaterift_restore_frame( skaterift.resume_target );
+      }
+      else {
+         vg_slewf( &skaterift.resume_transition, 1.0f, 
+                   vg.time_frame_delta * (1.0f/1.0f) );
+
+         if( skaterift.resume_transition >= 1.0f )
+            skaterift_restore_frame( skaterift.resume_target );
+         else {
+            f64 target = vg_lerp( skaterift.resume_begin, 
+                              skaterift.resume_target->time, 
+                              vg_smoothstepf( skaterift.resume_transition ) );
+            replay_seek( &skaterift.replay, target );
+         }
+      }
+   }
+   else {
+      if( button_down( k_srbind_replay_play ) )
+         skaterift.replay_control = k_replay_control_play;
+      if( button_down( k_srbind_replay_freecam ) ){
+         skaterift.freecam = skaterift.freecam ^ 0x1;
+
+         if( skaterift.freecam )
+            replay_get_camera( &skaterift.replay, &skaterift.replay_freecam );
+
+         skaterift_replay_update_helpers();
+      }
+
+      f32 target_speed = axis_state( k_sraxis_replay_h ) * 5.0;
+      if( button_press( k_srbind_reset ) )
+         target_speed += -2.0;
+
+      if( fabsf(target_speed) > 0.01f )
+         skaterift.replay_control = k_replay_control_scrub;
+
+      if( skaterift.replay_control == k_replay_control_play )
+         target_speed = 1.0;
+
+      vg_slewf( &skaterift.track_velocity, target_speed, 
+                18.0f*vg.time_frame_delta );
+
+      if( fabsf( skaterift.track_velocity ) > 0.0001f ){
+         f64 target = skaterift.replay.cursor;
+         target += skaterift.track_velocity * vg.time_frame_delta;
+
+         replay_seek( &skaterift.replay, target );
+      }
+
+      if( button_down( k_srbind_mback ) ){
+         if( skaterift.replay.statehead )
+            skaterift_restore_frame( skaterift.replay.statehead );
+         else
+            skaterift.activity = k_skaterift_default;
+         srinput.state = k_input_state_resume;
+         gui_helper_clear();
+      }
+
+      if( skaterift.freecam ){
+         //freecam_preupdate();
+      }
+      else {
+         if( button_down( k_srbind_replay_resume ) ){
+            skaterift_replay_resume();
+         }
+      }
+   }
+}
+
+static void skaterift_replay_update_helpers(void){
+   skaterift.helper_resume->greyed = skaterift.freecam;
+
+   vg_str freecam_text;
+   vg_strnull( &freecam_text, skaterift.helper_freecam->text, 
+               GUI_HELPER_TEXT_LENGTH );
+   vg_strcat( &freecam_text, skaterift.freecam? "exit freecam": "freecam" );
+}
+
+static void skaterift_replay_post_render(void){
+   if( world_static.active_instance != k_world_purpose_client )
+      return;
+
+   /* capture the current resume frame at the very last point */
+   if( button_down( k_srbind_reset ) ){
+      if( skaterift.activity == k_skaterift_default ){
+         localplayer.rewinded_since_last_gate = 1;
+         skaterift.activity = k_skaterift_replay;
+         skaterift_record_frame( &skaterift.replay, 1 );
+         if( skaterift.replay.head ){
+            skaterift.replay.cursor = skaterift.replay.head->time;
+            skaterift.replay.cursor_frame = skaterift.replay.head;
+         }
+         skaterift.replay_control = k_replay_control_scrub;
+
+         gui_helper_clear();
+         vg_str text;
+
+         if( gui_new_helper( input_axis_list[k_sraxis_replay_h], &text ) )
+            vg_strcat( &text, "scrub" );
+
+         if( (skaterift.helper_resume = gui_new_helper( 
+                     input_button_list[k_srbind_replay_resume], &text )) )
+            vg_strcat( &text, "resume" );
+         
+         if( gui_new_helper( input_button_list[k_srbind_replay_play], &text ))
+            vg_strcat( &text, "playback" );
+
+         skaterift.helper_freecam = gui_new_helper(
+               input_button_list[k_srbind_replay_freecam], &text );
+
+         skaterift_replay_update_helpers();
+      }
+   }
+}
+
+#if 0
+static void skaterift_get_replay_camera( camera *cam ){
+   if( skaterift.freecam ){
+      cam->nearz = 0.1f;
+      cam->farz = 100.0f;
+      v3_copy( skaterift.replay_freecam.pos, cam->pos );
+      v3_copy( skaterift.replay_freecam.angles, cam->angles );
+      cam->fov = skaterift.replay_freecam.fov;
+   }
+   else{
+      replay_get_camera( &skaterift.replay, &skaterift.cam );
+   }
+}
+#endif
+
+static void skaterift_replay_debug_info(void){
    player__debugtext( 2, "replay info" );
 
-   replay_buffer *replay = &player->replay;
+   replay_buffer *replay = &skaterift.replay;
 
    u32 head = 0,
        tail = 0;
@@ -125,6 +542,20 @@ VG_STATIC void replay_debug_info( player_instance *player ){
 
    player__debugtext( 1, "head @%u | tail @%u\n", head, tail );
 
+   if( replay->statehead ){
+      for( u32 i=0; i<4; i++ ){
+         player__debugtext( 1, "[%u]: [%hu, %hu]\n", i,
+              replay->statehead->data_table[i][0],
+              replay->statehead->data_table[i][1] );
+      }
+      u32 state = (void *)replay->statehead - replay->data;
+      player__debugtext( 1, "gs @%u\n", state );
+      player__debugtext( 1, "gamestate_size: %hu\n", 
+           replay->statehead->data_table[k_replay_framedata_gamestate][1] );
+   }
+   else
+      player__debugtext( 1, "gs @NULL\n" );
+
    f64 start = replay->cursor,
        end   = replay->cursor;
    if( replay->tail ) start = replay->tail->time;
@@ -136,10 +567,10 @@ VG_STATIC void replay_debug_info( player_instance *player ){
    player__debugtext( 1, "cursor: %.2fs / %.2fs\n", cur, len );
 }
 
-VG_STATIC void replay_imgui( player_instance *player ){
-   if( !k_replay_test ) return;
+static void skaterift_replay_imgui(void){
+   if( skaterift.activity != k_skaterift_replay ) return;
 
-   replay_buffer *replay = &player->replay;
+   replay_buffer *replay = &skaterift.replay;
    f64 start = replay->cursor,
        end   = replay->cursor;
    if( replay->tail ) start = replay->tail->time;
@@ -149,14 +580,42 @@ VG_STATIC void replay_imgui( player_instance *player ){
 
    char buffer[ 128 ];
 
-   ui_px height = 20,
-         cwidth = 8;
-   ui_rect bar = { 0, vg.window_y - height, vg.window_x, height };
+   /* mainbar */
+   ui_px height = 32,
+         cwidth = 2;
+   ui_rect bar = { 0, 0, vg.window_x, height };
    ui_fill( bar, ui_colour( k_ui_bg ) );
 
-   ui_rect cusor = { cur * (f64)vg.window_x - (cwidth/2), bar[1], 
+   /* cursor frame block */
+   if( replay->cursor_frame ){
+      if( replay->cursor_frame->r ){
+         f64 l = (replay->cursor_frame->r->time-replay->cursor_frame->time)/len,
+             s = (replay->cursor_frame->time - start) / len;
+         ui_rect box = { s*(f64)vg.window_x, 0, 
+                         VG_MAX(4,(ui_px)(l*vg.window_x)), bar[3]+2 };
+         ui_fill( box, ui_colour( k_ui_bg+4 ) );
+      }
+   }
+
+   /* cursor */
+   ui_rect cusor = { cur * (f64)vg.window_x - (cwidth/2), 0, 
                      cwidth, bar[3] };
-   ui_fill( cusor, ui_colour( k_ui_bg+4 ) );
+   ui_fill( cusor, ui_colour( k_ui_bg+7 ) );
+
+   /* latest state marker */
+   if( replay->statehead ){
+      f64 t = (replay->statehead->time - start) / len;
+      ui_rect tag = { t*(f64)vg.window_x, 0, 2, bar[3]+8 };
+      ui_fill( tag, ui_colour( k_ui_green+k_ui_brighter ) );
+   }
+
+   /* previous state marker */
+   replay_frame *prev = replay_find_recent_stateframe( replay );
+   if( prev ){
+      f64 t = (prev->time - start) / len;
+      ui_rect tag = { t*(f64)vg.window_x, 0, 2, bar[3]+8 };
+      ui_fill( tag, ui_colour( k_ui_yellow+k_ui_brighter ) );
+   }
 
    cusor[1] -= height;
    cusor[2] = 200;