more db stuff
[carveJwlIkooP6JGAAIwe30JlM.git] / highscores.h
index 2ffaf6acdd19f90bea341182801835b31c181900..84ab9a4164a5c9e0a8ab14e9d61179d4ca60437e 100644 (file)
@@ -14,15 +14,28 @@ typedef struct highscore highscore;
 typedef struct highscore_record highscore_record;
 typedef struct highscore_track_table highscore_track_table;
 typedef struct highscore_database highscore_database;
+typedef struct highscore_playerinfo highscore_playerinfo;
 
 #pragma pack(push,1)
+
+struct highscore_playerinfo
+{
+   char nickname[16];
+   u64 playerid;
+
+   union
+   {
+      aatree_pool_node aapn;
+      aatree_node aa_playerid;
+   };
+};
+
 struct highscore_record
 {
    u16 trackid, points, time, reserved0;
    u64 playerid;
    u32 datetime;
-
-   u32 reserved[7];
+   u32 reserved1;
    
    union
    {
@@ -35,12 +48,7 @@ struct highscore_record
       }
       aa;
 
-      struct
-      {
-         /* TODO pool allocator */
-         u32 next, prev;
-      }
-      pool;
+      aatree_pool_node pool;
    };
 };
 
@@ -58,8 +66,11 @@ struct highscore_database
 {
    highscore_track_table tracks[ 128 ];
 
-   aatree_ptr pool_head;
-   u32 reserved[63];
+   aatree_ptr pool_head, playerinfo_head;
+   u32 entry_capacity, 
+       playerinfo_capacity, playerinfo_root;
+
+   u32 reserved[59];
 };
 
 #pragma pack(pop)
@@ -71,9 +82,11 @@ static struct highscore_system
           aainfo_points,
           aainfo_time,
           aainfo_playerid,
-          aainfo_datetime;
+          aainfo_datetime,
+          aainfo_playerinfo_playerid,
+          aainfo_playerinfo;
 
-   void *data;
+   void *data, *playerinfo_data;
 }
 highscore_system;
 
@@ -104,24 +117,52 @@ static int highscore_cmp_playerid( void *a, void *b )
    return pa->playerid < pb->playerid? -1: 1;
 }
 
-static int highscores_init( u32 pool_size )
+static int highscore_cmp_playerinfo_playerid( void *a, void *b )
 {
-   struct highscore_system *sys = &highscore_system;
+   highscore_playerinfo *pa = a, *pb = b;
+   if( pa->playerid == pb->playerid ) return 0;
+   return pa->playerid < pb->playerid? -1: 1;
+}
 
-   size_t requested_mem = pool_size * sizeof(highscore_record);
-   sys->data = malloc( requested_mem );
+static void *highscore_malloc( u32 count, u32 size )
+{
+   size_t requested_mem = size * count;
+   void *data = malloc( requested_mem );
 
    requested_mem /= 1024;
    requested_mem /= 1024;
 
-   if( !highscore_system.data )
+   if( !data )
    {
-      vg_error( "Could not allocated %dmb of memory for database\n",
-                  requested_mem );
-      return 0;
+      vg_error( "Could not allocated %dmb of memory\n", requested_mem );
+      return NULL;
    }
    else
-      vg_success( "Allocated %dmb for database\n", requested_mem );
+      vg_success( "Allocated %dmb for %u records\n", requested_mem, count );
+
+   return data;
+}
+
+static void highscores_free(void)
+{
+   free( highscore_system.data );
+   free( highscore_system.playerinfo_data );
+}
+
+static int highscores_init( u32 pool_size, u32 playerinfo_pool_size )
+{
+   struct highscore_system *sys = &highscore_system;
+
+   sys->data = highscore_malloc( pool_size, sizeof(highscore_record) );
+   if( !sys->data ) return 0;
+
+   sys->playerinfo_data = 
+      highscore_malloc( playerinfo_pool_size, sizeof(highscore_playerinfo));
+   if( !sys->playerinfo_data ) 
+   {
+      free( sys->data );
+      return 0;
+   }
 
    /* This is ugly.. too bad! */
    sys->aainfo.base = highscore_system.data;
@@ -149,16 +190,64 @@ static int highscores_init( u32 pool_size )
    sys->aainfo_playerid.offset = offsetof(highscore_record,aa.playerid);
    sys->aainfo_playerid.p_cmp = highscore_cmp_playerid;
 
+   sys->aainfo_playerinfo_playerid.base = highscore_system.playerinfo_data;
+   sys->aainfo_playerinfo_playerid.stride = sizeof(highscore_playerinfo);
+   sys->aainfo_playerinfo_playerid.offset = 
+      offsetof(highscore_playerinfo,aa_playerid);
+   sys->aainfo_playerinfo_playerid.p_cmp = highscore_cmp_playerinfo_playerid;
+
+   sys->aainfo_playerinfo.base = highscore_system.playerinfo_data;
+   sys->aainfo_playerinfo.stride = sizeof(highscore_playerinfo);
+   sys->aainfo_playerinfo.offset = offsetof(highscore_playerinfo,aapn);
+   sys->aainfo_playerinfo.p_cmp = NULL;
 
-   /* TODO: Load from disk if avalible */
-   if( 0 )
+   FILE *fp = fopen( ".aadb", "rb" );
+   if( fp )
    {
+      vg_info( "Loading existing database\n" );
+      
+      u64 count = fread( &sys->dbheader, sizeof(highscore_database), 1, fp );
+
+      if( count != 1 )
+      {
+         vg_error( "Unexpected EOF reading database header\n" );
+         
+         highscores_free();
+         return 0;
+      }
+
+      count = fread( sys->data, sizeof(highscore_record), pool_size, fp );
+      if( count != pool_size )
+      {
+         vg_error( "Unexpected EOF reading database contents;"
+                   " %lu records of %u were read\n", count, pool_size );
+
+         highscores_free();
+         return 0;
+      }
 
+      count = fread( sys->playerinfo_data, sizeof(highscore_playerinfo),
+                     playerinfo_pool_size, fp );
+      if( count != playerinfo_pool_size )
+      {
+         vg_error( "Unexpected EOF reading playerinfo contents;"
+                   " %lu records of %u were read\n", count, 
+                   playerinfo_pool_size );
+         
+         highscores_free();
+         return 0;
+      }
+
+      fclose( fp );
    }
    else
    {
+      vg_log( "No existing database found (.aadb)\n" );
       vg_info( "Initializing database nodes\n" );
+      memset( &sys->dbheader, 0, sizeof(highscore_database) );
+
       sys->dbheader.pool_head = aatree_init_pool( &sys->aainfo, pool_size );
+      sys->dbheader.entry_capacity = pool_size;
 
       for( int i=0; i<vg_list_size(sys->dbheader.tracks); i++ )
       {
@@ -168,14 +257,39 @@ static int highscores_init( u32 pool_size )
          table->root_time = AATREE_PTR_NIL;
          table->root_datetime = AATREE_PTR_NIL;
       }
+
+      /* Initialize secondary db */
+      sys->dbheader.playerinfo_head = aatree_init_pool( 
+            &sys->aainfo_playerinfo,
+            playerinfo_pool_size );
+      sys->dbheader.playerinfo_capacity = playerinfo_pool_size;
+      sys->dbheader.playerinfo_root = AATREE_PTR_NIL;
    }
 
    return 1;
 }
 
-static void highscores_free(void)
+static int highscores_serialize_all(void)
 {
-   free( highscore_system.data );
+   struct highscore_system *sys = &highscore_system;
+   vg_info( "Serializing database\n" );
+
+   FILE *fp = fopen( ".aadb", "wb" );
+
+   if( !fp )
+   {
+      vg_error( "Could not open .aadb\n" );
+      return 0;
+   }
+
+   fwrite( &sys->dbheader, sizeof(highscore_database), 1, fp );
+   fwrite( sys->data, sizeof(highscore_record), 
+           sys->dbheader.entry_capacity, fp );
+   fwrite( sys->playerinfo_data, sizeof(highscore_playerinfo),
+            sys->dbheader.playerinfo_capacity, fp );
+
+   fclose( fp );
+   return 1;
 }
 
 static aatree_ptr highscores_push_record( highscore_record *record )
@@ -201,6 +315,16 @@ static aatree_ptr highscores_push_record( highscore_record *record )
 
    if( existing != AATREE_PTR_NIL )
    {
+      highscore_record *crecord = aatree_get_data( &sys->aainfo_playerid, 
+                                                   existing );
+
+      if( crecord->time < record->time || 
+            (crecord->time == record->time && crecord->points > record->points))
+      {
+         vg_log( "Not overwriting better score\n" );
+         return existing;
+      }
+
       vg_log( "Freeing existing record for player %lu\n", record->playerid );
       table->root_playerid = aatree_del( &sys->aainfo_playerid, existing );
       table->root_datetime = aatree_del( &sys->aainfo_datetime, existing );
@@ -214,7 +338,10 @@ static aatree_ptr highscores_push_record( highscore_record *record )
       aatree_pool_alloc( &sys->aainfo, &sys->dbheader.pool_head );
 
    if( index == AATREE_PTR_NIL )
+   {
+      vg_error( "Database records are over capacity!\n" );
       return index;
+   }
 
    highscore_record *dst = aatree_get_data( &sys->aainfo, index );
    memset( dst, 0, sizeof(highscore_record) );
@@ -237,12 +364,69 @@ static aatree_ptr highscores_push_record( highscore_record *record )
    return index;
 }
 
+static aatree_ptr highscore_set_user_nickname( u64 steamid, char nick[16] )
+{
+   vg_log( "Updating %lu's nickname\n", steamid );
+
+   struct highscore_system *sys = &highscore_system;
+   
+   highscore_playerinfo temp;
+   temp.playerid = steamid;
+
+   aatree_ptr record = aatree_find( &sys->aainfo_playerinfo_playerid, 
+                                     sys->dbheader.playerinfo_root,
+                                     &temp );
+   highscore_playerinfo *info;
+
+   if( record != AATREE_PTR_NIL )
+   {
+      info = aatree_get_data( &sys->aainfo_playerinfo, record );
+   }
+   else
+   {
+      record = aatree_pool_alloc( &sys->aainfo_playerinfo, 
+                                  &sys->dbheader.playerinfo_head );
+
+      if( record == AATREE_PTR_NIL )
+      {
+         vg_error( "Player info database is over capacity!\n" );
+         return AATREE_PTR_NIL;
+      }
+
+      info = aatree_get_data( &sys->aainfo_playerinfo, record );
+      memset( info, 0, sizeof(highscore_playerinfo) );
+
+      info->playerid = steamid;
+      sys->dbheader.playerinfo_root = aatree_insert( 
+            &sys->aainfo_playerinfo_playerid,
+            sys->dbheader.playerinfo_root,
+            record );
+   }
+
+   for( int i=0; i<16; i++ )
+      info->nickname[i] = nick[i];
+
+   return AATREE_PTR_NIL;
+}
+
 static void _highscore_showtime( void *data )
 {
    highscore_record *record = data;
    printf( "%hu", record->time );
 }
 
+static void _highscore_showname( void *data )
+{
+   char namebuf[17];
+   namebuf[16] = '\0';
+
+   highscore_playerinfo *info = data;
+   for( int i=0; i<16; i++ )
+      namebuf[i] = info->nickname[i];
+
+   printf( " %lu %s", info->playerid, namebuf );
+}
+
 static void highscores_print_track( u32 trackid, u32 count )
 {
    struct highscore_system *sys = &highscore_system;
@@ -250,22 +434,45 @@ static void highscores_print_track( u32 trackid, u32 count )
    highscore_track_table *table = &sys->dbheader.tracks[ trackid ];
    aatree_ptr it = aatree_kth( &sys->aainfo_time, table->root_time, 0 );
 
-   vg_info( "Highscores, top %u records for track %u\n", count, trackid );
-   vg_info( "==============================================\n" );
-
+   vg_info( "Highscores: top %u fastest records for track %u\n", count, trackid );
+   vg_info( "================================================\n" );
+   vg_info( "%3s| %16s | %5s | %5s | %s\n", "#", "Player", "Time", "Score", 
+                                                                   "TrackID" );
+   vg_info( "================================================\n" );
    int i=0;
    while( it != AATREE_PTR_NIL && i < 10 )
    {
       highscore_record *record = aatree_get_data( &sys->aainfo_time, it );
-      vg_info( "  [%d]: player(%lu), time: %hu, score: %hu, track:%hu\n",
-                  i+1, record->playerid, record->time, record->points,
+
+      highscore_playerinfo temp;
+      temp.playerid = record->playerid;
+
+      aatree_ptr info_ptr = aatree_find( &sys->aainfo_playerinfo_playerid,
+                                          sys->dbheader.playerinfo_root,
+                                          &temp );
+
+      char namebuf[17];
+      if( info_ptr == AATREE_PTR_NIL )
+         snprintf( namebuf, 16, "[%lu]", record->playerid );
+      else
+      {
+         highscore_playerinfo *inf = aatree_get_data( 
+               &sys->aainfo_playerinfo_playerid, info_ptr );
+
+         for( int i=0; i<16; i++ )
+            namebuf[i] = inf->nickname[i];
+         namebuf[16] = '\0';
+      }
+
+      vg_info( "%3d| %16s   %5hu   %5hu   %3hu\n",
+                  i+1, namebuf, record->time, record->points,
                   record->trackid );
 
       i++;
       it = aatree_next( &sys->aainfo_time, it );
    }
 
-   vg_info( "==============================================\n" );
+   vg_info( "================================================\n" );
 }
 
 #endif /* HIGHSCORES_H */