2 * Copyright (C) 2021-2022 Mt.ZERO Software, Harry Godden - All Rights Reserved
8 #include "vg/vg_store.h"
9 #include "vg/vg_stdint.h"
10 #include "world_info.h"
13 * Designed to be used across client and server,
14 * for the client its only storing the local users records, for server its
18 typedef struct highscore highscore
;
19 typedef struct highscore_record highscore_record
;
20 typedef struct highscore_track_table highscore_track_table
;
21 typedef struct highscore_database highscore_database
;
22 typedef struct highscore_playerinfo highscore_playerinfo
;
26 struct highscore_playerinfo
33 aatree_pool_node aapn
;
34 aatree_node aa_playerid
;
38 struct highscore_record
40 u16 trackid
, points
, time
, reserved0
;
56 aatree_pool_node pool
;
60 struct highscore_track_table
62 aatree_ptr root_points
,
70 struct highscore_database
72 highscore_track_table tracks
[ 128 ];
74 aatree_ptr pool_head
, playerinfo_head
;
76 playerinfo_capacity
, playerinfo_root
;
83 static struct highscore_system
85 highscore_database dbheader
;
91 aainfo_playerinfo_playerid
,
94 void *data
, *playerinfo_data
;
96 u32 pool_size
, playerinfo_pool_size
;
100 static int highscore_cmp_points( void *a
, void *b
)
102 highscore_record
*pa
= a
, *pb
= b
;
103 return (int)pa
->points
- (int)pb
->points
;
106 static int highscore_cmp_datetime( void *a
, void *b
)
108 highscore_record
*pa
= a
, *pb
= b
;
110 if( pa
->datetime
== pb
->datetime
) return 0;
111 return pa
->datetime
< pb
->datetime
? 1: -1;
114 static int highscore_cmp_time( void *a
, void *b
)
116 highscore_record
*pa
= a
, *pb
= b
;
117 return (int)pb
->time
- (int)pa
->time
;
120 static int highscore_cmp_playerid( void *a
, void *b
)
122 highscore_record
*pa
= a
, *pb
= b
;
123 if( pa
->playerid
== pb
->playerid
) return 0;
124 return pa
->playerid
< pb
->playerid
? -1: 1;
127 static int highscore_cmp_playerinfo_playerid( void *a
, void *b
)
129 highscore_playerinfo
*pa
= a
, *pb
= b
;
130 if( pa
->playerid
== pb
->playerid
) return 0;
131 return pa
->playerid
< pb
->playerid
? -1: 1;
134 static void *highscore_malloc( u32 count
, u32 size
)
136 size_t requested_mem
= size
* count
;
137 void *data
= vg_alloc( requested_mem
);
139 requested_mem
/= 1024;
140 requested_mem
/= 1024;
144 vg_error( "Could not allocated %dmb of memory\n", requested_mem
);
148 vg_success( "Allocated %dmb for %u records\n", requested_mem
, count
);
153 static void highscores_free(void)
155 vg_free( highscore_system
.data
);
156 vg_free( highscore_system
.playerinfo_data
);
159 static void highscores_create_db(void)
161 struct highscore_system
*sys
= &highscore_system
;
163 vg_info( "Initializing database nodes\n" );
164 memset( &sys
->dbheader
, 0, sizeof(highscore_database
) );
166 sys
->dbheader
.pool_head
= aatree_init_pool( &sys
->aainfo
, sys
->pool_size
);
167 sys
->dbheader
.entry_capacity
= sys
->pool_size
;
169 for( int i
=0; i
<vg_list_size(sys
->dbheader
.tracks
); i
++ )
171 highscore_track_table
*table
= &sys
->dbheader
.tracks
[i
];
172 table
->root_points
= AATREE_PTR_NIL
;
173 table
->root_playerid
= AATREE_PTR_NIL
;
174 table
->root_time
= AATREE_PTR_NIL
;
175 table
->root_datetime
= AATREE_PTR_NIL
;
178 /* Initialize secondary db */
179 sys
->dbheader
.playerinfo_head
= aatree_init_pool(
180 &sys
->aainfo_playerinfo
,
181 sys
->playerinfo_pool_size
);
182 sys
->dbheader
.playerinfo_capacity
= sys
->playerinfo_pool_size
;
183 sys
->dbheader
.playerinfo_root
= AATREE_PTR_NIL
;
186 static int highscores_read(void)
188 struct highscore_system
*sys
= &highscore_system
;
190 FILE *fp
= fopen( ".aadb", "rb" );
193 vg_info( "Loading existing database\n" );
195 u64 count
= fread( &sys
->dbheader
, sizeof(highscore_database
), 1, fp
);
199 vg_error( "Unexpected EOF reading database header\n" );
203 count
= fread( sys
->data
, sizeof(highscore_record
), sys
->pool_size
, fp
);
204 if( count
!= sys
->pool_size
)
206 vg_error( "Unexpected EOF reading database contents;"
207 " %lu records of %u were read\n", count
, sys
->pool_size
);
211 count
= fread( sys
->playerinfo_data
, sizeof(highscore_playerinfo
),
212 sys
->playerinfo_pool_size
, fp
);
214 if( count
!= sys
->playerinfo_pool_size
)
216 vg_error( "Unexpected EOF reading playerinfo contents;"
217 " %lu records of %u were read\n", count
,
218 sys
->playerinfo_pool_size
);
227 vg_low( "No existing database found (.aadb)\n" );
232 static void highscores_init( u32 pool_size
, u32 playerinfo_pool_size
)
234 struct highscore_system
*sys
= &highscore_system
;
236 sys
->data
= highscore_malloc( pool_size
, sizeof(highscore_record
) );
237 sys
->playerinfo_data
= highscore_malloc( playerinfo_pool_size
,
238 sizeof(highscore_playerinfo
));
240 /* This is ugly.. too bad! */
241 sys
->aainfo
.base
= highscore_system
.data
;
242 sys
->aainfo
.stride
= sizeof(highscore_record
);
243 sys
->aainfo
.offset
= offsetof(highscore_record
,pool
);
244 sys
->aainfo
.p_cmp
= NULL
;
246 sys
->aainfo_datetime
.base
= highscore_system
.data
;
247 sys
->aainfo_datetime
.stride
= sizeof(highscore_record
);
248 sys
->aainfo_datetime
.offset
= offsetof(highscore_record
,aa
.datetime
);
249 sys
->aainfo_datetime
.p_cmp
= highscore_cmp_datetime
;
251 sys
->aainfo_points
.base
= highscore_system
.data
;
252 sys
->aainfo_points
.stride
= sizeof(highscore_record
);
253 sys
->aainfo_points
.offset
= offsetof(highscore_record
,aa
.points
);
254 sys
->aainfo_points
.p_cmp
= highscore_cmp_points
;
256 sys
->aainfo_time
.base
= highscore_system
.data
;
257 sys
->aainfo_time
.stride
= sizeof(highscore_record
);
258 sys
->aainfo_time
.offset
= offsetof(highscore_record
,aa
.time
);
259 sys
->aainfo_time
.p_cmp
= highscore_cmp_time
;
261 sys
->aainfo_playerid
.base
= highscore_system
.data
;
262 sys
->aainfo_playerid
.stride
= sizeof(highscore_record
);
263 sys
->aainfo_playerid
.offset
= offsetof(highscore_record
,aa
.playerid
);
264 sys
->aainfo_playerid
.p_cmp
= highscore_cmp_playerid
;
266 sys
->aainfo_playerinfo_playerid
.base
= highscore_system
.playerinfo_data
;
267 sys
->aainfo_playerinfo_playerid
.stride
= sizeof(highscore_playerinfo
);
268 sys
->aainfo_playerinfo_playerid
.offset
=
269 offsetof(highscore_playerinfo
,aa_playerid
);
270 sys
->aainfo_playerinfo_playerid
.p_cmp
= highscore_cmp_playerinfo_playerid
;
272 sys
->aainfo_playerinfo
.base
= highscore_system
.playerinfo_data
;
273 sys
->aainfo_playerinfo
.stride
= sizeof(highscore_playerinfo
);
274 sys
->aainfo_playerinfo
.offset
= offsetof(highscore_playerinfo
,aapn
);
275 sys
->aainfo_playerinfo
.p_cmp
= NULL
;
277 sys
->playerinfo_pool_size
= playerinfo_pool_size
;
278 sys
->pool_size
= pool_size
;
281 static int highscores_serialize_all(void)
283 struct highscore_system
*sys
= &highscore_system
;
284 vg_info( "Serializing database\n" );
286 FILE *fp
= fopen( ".aadb", "wb" );
290 vg_error( "Could not open .aadb\n" );
294 fwrite( &sys
->dbheader
, sizeof(highscore_database
), 1, fp
);
295 fwrite( sys
->data
, sizeof(highscore_record
),
296 sys
->dbheader
.entry_capacity
, fp
);
297 fwrite( sys
->playerinfo_data
, sizeof(highscore_playerinfo
),
298 sys
->dbheader
.playerinfo_capacity
, fp
);
304 static highscore_record
*highscore_find_user_record( u64 playerid
, u32 trackid
)
306 struct highscore_system
*sys
= &highscore_system
;
308 highscore_track_table
*table
= &sys
->dbheader
.tracks
[trackid
];
309 highscore_record temp
;
310 temp
.playerid
= playerid
;
313 aatree_find( &sys
->aainfo_playerid
, table
->root_playerid
, &temp
);
315 if( find
== AATREE_PTR_NIL
)
318 return aatree_get_data( &sys
->aainfo_playerid
, find
);
321 static aatree_ptr
highscores_push_record( highscore_record
*record
)
323 struct highscore_system
*sys
= &highscore_system
;
325 vg_low( "Inserting record into database for track %hu\n",record
->trackid
);
327 if( record
->trackid
>= vg_list_size(sys
->dbheader
.tracks
) )
329 vg_error( "TrackID out of range (%hu>=%d)\n", record
->trackid
,
330 vg_list_size(sys
->dbheader
.tracks
) );
332 return AATREE_PTR_NIL
;
335 /* Search for existing record on this track */
336 highscore_track_table
*table
= &sys
->dbheader
.tracks
[record
->trackid
];
337 aatree_ptr existing
= aatree_find( &sys
->aainfo_playerid
,
338 table
->root_playerid
,
341 if( existing
!= AATREE_PTR_NIL
)
343 highscore_record
*crecord
= aatree_get_data( &sys
->aainfo_playerid
,
346 if( crecord
->time
< record
->time
||
347 (crecord
->time
== record
->time
&& crecord
->points
> record
->points
))
349 vg_low( "Not overwriting better score\n" );
353 vg_low( "Freeing existing record for player %lu\n", record
->playerid
);
354 table
->root_playerid
= aatree_del( &sys
->aainfo_playerid
, existing
);
355 table
->root_datetime
= aatree_del( &sys
->aainfo_datetime
, existing
);
356 table
->root_points
= aatree_del( &sys
->aainfo_points
, existing
);
357 table
->root_time
= aatree_del( &sys
->aainfo_time
, existing
);
359 aatree_pool_free( &sys
->aainfo
, existing
, &sys
->dbheader
.pool_head
);
363 aatree_pool_alloc( &sys
->aainfo
, &sys
->dbheader
.pool_head
);
365 if( index
== AATREE_PTR_NIL
)
367 vg_error( "Database records are over capacity!\n" );
371 highscore_record
*dst
= aatree_get_data( &sys
->aainfo
, index
);
372 memset( dst
, 0, sizeof(highscore_record
) );
374 dst
->trackid
= record
->trackid
;
375 dst
->datetime
= record
->datetime
;
376 dst
->playerid
= record
->playerid
;
377 dst
->points
= record
->points
;
378 dst
->time
= record
->time
;
381 aatree_insert( &sys
->aainfo_time
, table
->root_time
, index
);
382 table
->root_datetime
=
383 aatree_insert( &sys
->aainfo_datetime
, table
->root_datetime
, index
);
384 table
->root_playerid
=
385 aatree_insert( &sys
->aainfo_playerid
, table
->root_playerid
, index
);
387 aatree_insert( &sys
->aainfo_points
, table
->root_points
, index
);
392 static aatree_ptr
highscore_set_user_nickname( u64 steamid
, char nick
[16] )
395 for( int i
=0; i
<16; i
++ )
399 vg_low( "Updating %lu's nickname -> %s\n", steamid
, name
);
401 struct highscore_system
*sys
= &highscore_system
;
403 highscore_playerinfo temp
;
404 temp
.playerid
= steamid
;
406 aatree_ptr record
= aatree_find( &sys
->aainfo_playerinfo_playerid
,
407 sys
->dbheader
.playerinfo_root
,
409 highscore_playerinfo
*info
;
411 if( record
!= AATREE_PTR_NIL
)
413 info
= aatree_get_data( &sys
->aainfo_playerinfo
, record
);
417 record
= aatree_pool_alloc( &sys
->aainfo_playerinfo
,
418 &sys
->dbheader
.playerinfo_head
);
420 if( record
== AATREE_PTR_NIL
)
422 vg_error( "Player info database is over capacity!\n" );
423 return AATREE_PTR_NIL
;
426 info
= aatree_get_data( &sys
->aainfo_playerinfo
, record
);
427 memset( info
, 0, sizeof(highscore_playerinfo
) );
429 info
->playerid
= steamid
;
430 sys
->dbheader
.playerinfo_root
= aatree_insert(
431 &sys
->aainfo_playerinfo_playerid
,
432 sys
->dbheader
.playerinfo_root
,
436 for( int i
=0; i
<16; i
++ )
437 info
->nickname
[i
] = nick
[i
];
439 return AATREE_PTR_NIL
;
442 /* Get the length of a string, bounded by '\0' or len, whichever is first */
443 static int highscore_strlen( const char *str
, int len
)
446 for( str_length
=0; str_length
<len
; str_length
++ )
447 if( !str
[str_length
] )
453 /* Print the string(max length:len) centered into buf (has width:width) */
454 static void highscore_strc( char *buf
, const char *str
, int len
, int width
)
456 int str_length
= highscore_strlen( str
, len
),
457 offs
= (width
-str_length
)/2;
459 for( int i
=0; i
<str_length
; i
++ )
470 /* Print the string(max length:len) left aligned into buf */
471 static void highscore_strl( char *buf
, const char *str
, int len
)
473 for( int i
=0; i
<len
; i
++ )
482 /* Print the string (max length:len) right aligned into buf (has width:width) */
483 static void highscore_strr( char *buf
, const char *str
, int len
, int width
)
485 int str_length
= highscore_strlen( str
, len
);
487 for( int i
=0; i
<len
; i
++ )
492 buf
[width
-str_length
+i
] = str
[i
];
496 /* Print integer (padded with: alt), right aligned into buf(width: len) */
497 static void highscore_intr( char *buf
, int value
, int len
, char alt
)
505 buf
[ len
-1 - (i
++) ] = '0' + (value
% 10);
510 buf
[ len
-1 - i
] = alt
;
513 /* Print integer into buffer with max length len */
514 static void highscore_intl( char *buf
, int value
, int len
)
524 temp
[ i
++ ] = '0' + (value
% 10);
531 for( int j
=0; j
<i
; j
++ )
533 buf
[j
] = temp
[ i
-1-j
];
537 /* Clear buffer with length using clr character */
538 static void highscore_clear( char *buf
, char clr
, int length
)
540 for( int i
=0; i
<length
; i
++ )
546 --------------------------
547 #| Player | Time | Pts
548 1|aaaabbbbcc 5:23.32 30000
550 3|aaabbbcccl 2:30.45 20000
560 /* Generate a highscores board in text form, the width is always 27. Buffer
561 * must be (count+3)*27 in size. */
562 static void highscores_board_generate( char *buf
, u32 id
, u32 count
)
565 highscore_clear( buf
, ' ', (count
+3)*w
);
567 struct track_info
*inf
= &track_infos
[id
];
568 struct highscore_system
*sys
= &highscore_system
;
570 highscore_track_table
*table
= &sys
->dbheader
.tracks
[ id
];
571 aatree_ptr it
= aatree_kth( &sys
->aainfo_time
, table
->root_time
, 0 );
573 highscore_strc ( buf
+w
*0, inf
->name
, w
,w
);
574 highscore_clear( buf
+w
*1, '-', w
);
575 highscore_strl ( buf
+w
*2, " #| Player | Time ", 27 );
577 for( int i
=0; i
<count
; i
++ )
579 char *line
= buf
+w
*(3+i
);
580 highscore_intr( line
, i
+1, 2, ' ' );
583 if( it
== AATREE_PTR_NIL
)
586 highscore_record
*record
= aatree_get_data( &sys
->aainfo_time
, it
);
587 highscore_playerinfo temp
;
588 temp
.playerid
= record
->playerid
;
590 aatree_ptr info_ptr
= aatree_find( &sys
->aainfo_playerinfo_playerid
,
591 sys
->dbheader
.playerinfo_root
,
595 if( info_ptr
== AATREE_PTR_NIL
)
596 highscore_strl( line
+3, "unknown", 16 );
599 highscore_playerinfo
*inf
= aatree_get_data(
600 &sys
->aainfo_playerinfo_playerid
, info_ptr
);
602 highscore_strl( line
+3, inf
->nickname
, 16 );
605 u16 centiseconds
= record
->time
,
606 seconds
= centiseconds
/ 100,
607 minutes
= seconds
/ 60;
613 if( minutes
> 9 ) minutes
= 9;
616 highscore_intr( line
+20, minutes
, 1, '0' );
618 highscore_intr( line
+22, seconds
, 2, '0' );
620 highscore_intr( line
+25, centiseconds
, 2, '0' );
624 highscore_intl( line
+22, record
->points
, 5 );
626 it
= aatree_next( &sys
->aainfo_time
, it
);
630 /* Print string out to file using newlines. Count is number of records
631 * ( this requires a buffer of (count+3)*27 size */
632 static void highscores_board_printf( FILE *fp
, const char *buf
, u32 count
)
636 for( int i
=0; i
<count
+3; i
++ )
637 fprintf( fp
, "%.27s\n", &buf
[i
*w
] );
640 #endif /* HIGHSCORES_H */