minor stuff
[vg.git] / vg_console.h
1 /* Copyright (C) 2021-2023 Harry Godden (hgn) - All Rights Reserved */
2
3 #ifndef VG_CONSOLE_H
4 #define VG_CONSOLE_H
5
6 #ifndef VG_GAME
7 #define VG_GAME
8 #endif
9
10 #include "vg/vg_imgui.h"
11 #include "vg/vg_log.h"
12
13 #define VG_VAR_F32( NAME, ... ) \
14 { u32 flags=0x00; __VA_ARGS__ ;\
15 vg_console_reg_var( #NAME, &NAME, k_var_dtype_f32, flags ); }
16
17 #define VG_VAR_I32( NAME, ... ) \
18 { u32 flags=0x00; __VA_ARGS__ ;\
19 vg_console_reg_var( #NAME, &NAME, k_var_dtype_i32, flags ); }
20
21 #define VG_VAR_PERSISTENT 0x1
22 #define VG_VAR_CHEAT 0x2
23 #define VG_VAR_CHEATTOGGLE 0x4
24
25 typedef struct vg_var vg_var;
26 typedef struct vg_cmd vg_cmd;
27
28 struct vg_console{
29 struct vg_var{
30 void *data;
31 const char *name;
32
33 enum vg_var_dtype{
34 k_var_dtype_i32,
35 k_var_dtype_u32,
36 k_var_dtype_f32
37 }
38 data_type;
39 u32 flags;
40
41 union{ u32 _u32; f32 _f32; i32 _i32; } defaults;
42 }
43 vars[ 128 ];
44
45 struct vg_cmd{
46 int (*function)( int argc, char const *argv[] );
47 void (*poll_suggest)( int argc, char const *argv[] );
48 const char *name;
49 }
50 functions[ 32 ];
51
52 struct {
53 const char *str;
54 int len;
55
56 u32 lev_score;
57 }
58 suggestions[12];
59 u32 suggestion_count;
60 int suggestion_select,
61 suggestion_pastepos,
62 suggestion_maxlen;
63
64 u32 var_count, function_count;
65
66 char input[96],
67 input_copy[96];
68
69 char history[32][96];
70 int history_last, history_pos, history_count;
71
72 i32 enabled, cheats;
73 }
74 static vg_console;
75
76 VG_STATIC void _vg_console_draw( void );
77 void _vg_console_println( const char *str );
78 VG_STATIC int _vg_console_list( int argc, char const *argv[] );
79 VG_STATIC void _vg_console_init(void);
80 VG_STATIC void _vg_console_write_persistent(void);
81 VG_STATIC void _vg_console_free(void);
82 VG_STATIC void vg_execute_console_input( const char *cmd );
83
84 /*
85 * Console interface
86 */
87 VG_STATIC void console_history_get( char* buf, int entry_num );
88 VG_STATIC int _vg_console_enabled(void);
89 VG_STATIC void console_proc_key( SDL_Keysym ev );
90
91 /*
92 * Implementation
93 */
94 VG_STATIC int _vg_console_enabled(void){
95 return vg_console.enabled;
96 }
97
98 VG_STATIC
99 void vg_console_reg_var( const char *alias, void *ptr, enum vg_var_dtype type,
100 u32 flags )
101 {
102 if( vg_console.var_count > vg_list_size(vg_console.vars) )
103 vg_fatal_error( "Too many vars registered" );
104
105 vg_var *var = &vg_console.vars[ vg_console.var_count ++ ];
106 var->name = alias;
107 var->data = ptr;
108 var->data_type = type;
109 var->flags = flags;
110
111 if ( type == k_var_dtype_f32 ) var->defaults._f32 = *((f32 *)ptr);
112 else if( type == k_var_dtype_i32 ) var->defaults._i32 = *((i32 *)ptr);
113 else if( type == k_var_dtype_u32 ) var->defaults._u32 = *((u32 *)ptr);
114
115 vg_info( "Console variable '%s' registered\n", alias );
116 }
117
118 VG_STATIC
119 void vg_console_reg_cmd( const char *alias,
120 int (*function)(int argc, const char *argv[]),
121 void (*poll_suggest)(int argc, const char *argv[]) )
122 {
123 if( vg_console.function_count > vg_list_size(vg_console.functions) )
124 vg_fatal_error( "Too many functions registered" );
125
126 vg_cmd *cmd = &vg_console.functions[ vg_console.function_count ++ ];
127
128 cmd->function = function;
129 cmd->poll_suggest = poll_suggest;
130 cmd->name = alias;
131
132 vg_info( "Console function '%s' registered\n", alias );
133 }
134
135 VG_STATIC int _vg_console_list( int argc, char const *argv[] ){
136 for( int i=0; i<vg_console.function_count; i ++ ){
137 struct vg_cmd *cmd = &vg_console.functions[ i ];
138 vg_info( "* %s\n", cmd->name );
139 }
140
141 for( int i=0; i<vg_console.var_count; i ++ ){
142 struct vg_var *cv = &vg_console.vars[ i ];
143 vg_info( "%s\n", cv->name );
144 }
145
146 return 0;
147 }
148
149 int _test_break( int argc, const char *argv[] ){
150 vg_fatal_error( "Test crash from main, after loading (console)" );
151 return 0;
152 }
153
154 int _vg_console_exec( int argc, const char *argv[] ){
155 if( argc < 1 ) return 0;
156
157 char path[256];
158 strcpy( path, "cfg/" );
159 strncat( path, argv[0], 250 );
160
161 FILE *fp = fopen( path, "r" );
162 if( fp ){
163 char line[256];
164
165 while( fgets( line, sizeof( line ), fp ) ){
166 line[ strcspn( line, "\r\n#" ) ] = 0x00;
167
168 if( line[0] != 0x00 ){
169 vg_execute_console_input( line );
170 }
171 }
172
173 fclose( fp );
174 }
175 else{
176 vg_error( "Could not open '%s'\n", path );
177 }
178
179 return 0;
180 }
181
182 int _ccmd_vg_console_defaults( int argc, const char *argv[] ){
183 for( int i=0; i<vg_console.var_count; i ++ ){
184 struct vg_var *cv = &vg_console.vars[i];
185
186 enum vg_var_dtype type = cv->data_type;
187 void *ptr = cv->data;
188
189 if ( type == k_var_dtype_f32 ) *((f32 *)ptr) = cv->defaults._f32;
190 else if( type == k_var_dtype_i32 ) *((i32 *)ptr) = cv->defaults._i32;
191 else if( type == k_var_dtype_u32 ) *((u32 *)ptr) = cv->defaults._u32;
192 }
193
194 return 0;
195 }
196
197 VG_STATIC void _vg_console_init(void){
198 vg_console_reg_cmd( "list", _vg_console_list, NULL );
199 vg_console_reg_cmd( "crash", _test_break, NULL );
200 vg_console_reg_cmd( "exec", _vg_console_exec, NULL );
201 vg_console_reg_cmd( "defaults", _ccmd_vg_console_defaults, NULL );
202 vg_console_reg_var( "cheats", &vg_console.cheats, k_var_dtype_i32, 0 );
203 }
204
205 VG_STATIC void vg_console_load_autos(void){
206 _vg_console_exec( 1, (const char *[]){ "auto.conf" } );
207 }
208
209 VG_STATIC void _vg_console_write_persistent(void){
210 FILE *fp = fopen( "cfg/auto.conf", "w" );
211
212 for( int i=0; i<vg_console.var_count; i ++ ){
213 struct vg_var *cv = &vg_console.vars[i];
214
215 if( cv->flags & VG_VAR_PERSISTENT ){
216 switch( cv->data_type ){
217 case k_var_dtype_i32:
218 fprintf( fp, "%s %d\n", cv->name, *(i32 *)(cv->data) );
219 break;
220 case k_var_dtype_u32:
221 fprintf( fp, "%s %u\n", cv->name, *(u32 *)(cv->data) );
222 break;
223 case k_var_dtype_f32:
224 fprintf( fp, "%s %.5f\n", cv->name, *(float *)(cv->data ) );
225 break;
226 }
227 }
228 }
229
230 fclose( fp );
231 }
232
233 VG_STATIC void _vg_console_free(void)
234 {
235 _vg_console_write_persistent();
236 }
237
238 /*
239 * splits src into tokens and fills out args as pointers to those tokens
240 * returns number of tokens
241 * dst must be as long as src
242 */
243 VG_STATIC int vg_console_tokenize( const char *src, char *dst,
244 const char *args[8] )
245 {
246 int arg_count = 0,
247 in_token = 0;
248
249 for( int i=0;; i ++ ){
250 if( src[i] ){
251 if( src[i] == ' ' || src[i] == '\t' ){
252 if( in_token )
253 dst[i] = '\0';
254
255 in_token = 0;
256
257 if( arg_count == 8 )
258 break;
259 }
260 else{
261 dst[i] = src[i];
262
263 if( !in_token ){
264 args[ arg_count ++ ] = &dst[i];
265 in_token = 1;
266 }
267 }
268 }
269 else{
270 dst[i] = '\0';
271 break;
272 }
273 }
274
275 return arg_count;
276 }
277
278 VG_STATIC vg_var *vg_console_match_var( const char *kw )
279 {
280 for( int i=0; i<vg_console.var_count; i ++ ){
281 struct vg_var *cv = &vg_console.vars[ i ];
282 if( !strcmp( cv->name, kw ) ){
283 return cv;
284 }
285 }
286
287 return NULL;
288 }
289
290 VG_STATIC vg_cmd *vg_console_match_cmd( const char *kw )
291 {
292 for( int i=0; i<vg_console.function_count; i ++ ){
293 struct vg_cmd *cmd = &vg_console.functions[ i ];
294 if( !strcmp( cmd->name, kw ) ){
295 return cmd;
296 }
297 }
298
299 return NULL;
300 }
301
302 VG_STATIC void vg_execute_console_input( const char *cmd )
303 {
304 char temp[512];
305 char const *args[8];
306 int arg_count = vg_console_tokenize( cmd, temp, args );
307
308 if( arg_count == 0 )
309 return;
310
311 vg_var *cv = vg_console_match_var( args[0] );
312 vg_cmd *fn = vg_console_match_cmd( args[0] );
313
314 assert( !(cv && fn) );
315
316 if( cv ){
317 /* Cvar Matched, try get value */
318 if( arg_count >= 2 ){
319 if( cv->flags & VG_VAR_CHEAT ){
320 if( !vg_console.cheats ){
321 vg_error( "variable is cheat protected\n" );
322 return;
323 }
324 }
325
326 if( (cv->data_type == k_var_dtype_u32) ||
327 (cv->data_type == k_var_dtype_i32) )
328 {
329 int *ptr = cv->data;
330 *ptr = atoi( args[1] );
331
332 if( cv->flags & VG_VAR_CHEATTOGGLE ){
333 if( *ptr ){
334 _ccmd_vg_console_defaults( 0, NULL );
335 }
336 }
337 }
338 else if( cv->data_type == k_var_dtype_f32 ){
339 float *ptr = cv->data;
340 *ptr = atof( args[1] );
341 }
342 }
343 else{
344 if( cv->data_type == k_var_dtype_i32 )
345 vg_info( "= %d\n", *((int *)cv->data) );
346 else if( cv->data_type == k_var_dtype_u32 )
347 vg_info( "= %u\n", *((u32 *)cv->data) );
348 else if( cv->data_type == k_var_dtype_f32 )
349 vg_info( "= %.4f\n", *((float *)cv->data) );
350 }
351
352 return;
353 }
354 else if( fn ){
355 fn->function( arg_count-1, args+1 );
356 return;
357 }
358
359 vg_error( "No command/var named '%s'. Use 'list' to view all\n", args[0] );
360 }
361
362 u32 str_lev_distance( const char *s1, const char *s2 ){
363 u32 m = strlen( s1 ),
364 n = strlen( s2 );
365
366 if( m==0 ) return n;
367 if( n==0 ) return m;
368
369 assert( n+1 <= 256 );
370
371 u32 costs[ 256 ];
372
373 for( u32 k=0; k<=n; k++ )
374 costs[k] = k;
375
376 u32 i = 0;
377 for( u32 i=0; i<m; i++ ){
378 costs[0] = i+1;
379
380 u32 corner = i;
381
382 for( u32 j=0; j<n; j++ ){
383 u32 upper = costs[j+1];
384
385 if( s1[i] == s2[j] )
386 costs[ j+1 ] = corner;
387 else{
388 u32 t = (upper < corner)? upper: corner;
389 costs[j+1] = ((costs[j] < t)? costs[j]: t) + 1;
390 }
391
392 corner = upper;
393 }
394 }
395
396 return costs[n];
397 }
398
399 u32 str_lcs( const char *s1, const char *s2 ){
400 u32 m = VG_MIN( 31, strlen( s1 ) ),
401 n = VG_MIN( 31, strlen( s2 ) );
402
403 int suff[32][32],
404 result = 0;
405
406 for( int i=0; i<=m; i++ ){
407 for( int j=0; j<=n; j++ ){
408 if( i == 0 || j == 0 )
409 suff[i][j] = 0;
410 else if( s1[i-1] == s2[j-1] ){
411 suff[i][j] = suff[i-1][j-1] + 1;
412 result = VG_MAX( result, suff[i][j] );
413 }
414 else
415 suff[i][j] = 0;
416 }
417 }
418
419 return result;
420 }
421
422 /* str must not fuckoff ever! */
423 VG_STATIC void console_suggest_score_text( const char *str, const char *input,
424 int minscore )
425 {
426 /* filter duplicates */
427 for( int i=0; i<vg_console.suggestion_count; i++ )
428 if( !strcmp( vg_console.suggestions[i].str, str ) )
429 return;
430
431 /* calc score */
432 u32 score = str_lcs( str, input );
433
434 if( score < minscore )
435 return;
436
437 int best_pos = vg_console.suggestion_count;
438 for( int j=best_pos-1; j>=0; j -- )
439 if( score > vg_console.suggestions[j].lev_score )
440 best_pos = j;
441
442 /* insert if good score */
443 if( best_pos < vg_list_size( vg_console.suggestions ) ){
444 int start = VG_MIN( vg_console.suggestion_count,
445 vg_list_size( vg_console.suggestions )-1 );
446 for( int j=start; j>best_pos; j -- )
447 vg_console.suggestions[j] = vg_console.suggestions[j-1];
448
449 vg_console.suggestions[ best_pos ].str = str;
450 vg_console.suggestions[ best_pos ].len = strlen( str );
451 vg_console.suggestions[ best_pos ].lev_score = score;
452
453 if( vg_console.suggestion_count <
454 vg_list_size( vg_console.suggestions ) )
455 vg_console.suggestion_count ++;
456 }
457 }
458
459 VG_STATIC void console_update_suggestions(void)
460 {
461 if( vg_ui.focused_control_type != k_ui_control_textbox ||
462 vg_ui.textbuf != vg_console.input )
463 return;
464
465 vg_console.suggestion_count = 0;
466 vg_console.suggestion_select = -1;
467 vg_console.suggestion_maxlen = 0;
468
469 /*
470 * - must be typing something
471 * - must be at the end
472 * - prev char must not be a whitespace
473 * - cursors should match
474 */
475
476 if( vg_ui.textbox.cursor_pos == 0 ) return;
477 if( vg_ui.textbox.cursor_pos != vg_ui.textbox.cursor_user ) return;
478 if( vg_console.input[ vg_ui.textbox.cursor_pos ] != '\0' ) return;
479
480 if( (vg_console.input[ vg_ui.textbox.cursor_pos -1 ] == ' ') ||
481 (vg_console.input[ vg_ui.textbox.cursor_pos -1 ] == '\t') )
482 return;
483
484 char temp[128];
485 const char *args[8];
486
487 int token_count = vg_console_tokenize( vg_console.input, temp, args );
488 if( !token_count ) return;
489 vg_console.suggestion_pastepos = args[token_count-1]-temp;
490
491 /* Score all our commands and cvars */
492 if( token_count == 1 ){
493 for( int i=0; i<vg_console.var_count; i++ ){
494 vg_var *cvar = &vg_console.vars[i];
495 console_suggest_score_text( cvar->name, args[0], 1 );
496 }
497
498 for( int i=0; i<vg_console.function_count; i++ ){
499 vg_cmd *cmd = &vg_console.functions[i];
500 console_suggest_score_text( cmd->name, args[0], 1 );
501 }
502 }
503 else{
504 vg_cmd *cmd = vg_console_match_cmd( args[0] );
505 vg_var *var = vg_console_match_var( args[0] );
506
507 assert( !( cmd && var ) );
508
509 if( cmd )
510 if( cmd->poll_suggest )
511 cmd->poll_suggest( token_count-1, &args[1] );
512 }
513
514 /* some post processing */
515 for( int i=0; i<vg_console.suggestion_count; i++ ){
516 vg_console.suggestion_maxlen = VG_MAX( vg_console.suggestion_maxlen,
517 vg_console.suggestions[i].len );
518
519 if( vg_console.suggestions[i].lev_score <
520 vg_console.suggestions[0].lev_score/2 )
521 {
522 vg_console.suggestion_count = i;
523 return;
524 }
525 }
526 }
527
528 /*
529 * Suggestion controls
530 */
531 VG_STATIC void _console_fetch_suggestion(void)
532 {
533 char *target = &vg_console.input[ vg_console.suggestion_pastepos ];
534
535 if( vg_console.suggestion_select == -1 ){
536 strcpy( target, vg_console.input_copy );
537 _ui_textbox_move_cursor( &vg_ui.textbox.cursor_user,
538 &vg_ui.textbox.cursor_pos, 10000, 1 );
539 }
540 else{
541 strncpy( target,
542 vg_console.suggestions[ vg_console.suggestion_select ].str,
543 vg_list_size( vg_console.input )-1 );
544
545 _ui_textbox_move_cursor( &vg_ui.textbox.cursor_user,
546 &vg_ui.textbox.cursor_pos, 10000, 1 );
547 _ui_textbox_put_char( ' ' );
548 }
549 }
550
551 VG_STATIC void _console_suggest_store_normal(void)
552 {
553 if( vg_console.suggestion_select == -1 ){
554 char *target = &vg_console.input[ vg_console.suggestion_pastepos ];
555 strcpy( vg_console.input_copy, target );
556 }
557 }
558
559 VG_STATIC void _console_suggest_next(void)
560 {
561 if( vg_console.suggestion_count ){
562 _console_suggest_store_normal();
563
564 vg_console.suggestion_select ++;
565
566 if( vg_console.suggestion_select >= vg_console.suggestion_count )
567 vg_console.suggestion_select = -1;
568
569 _console_fetch_suggestion();
570 }
571 }
572
573 VG_STATIC void _console_suggest_prev(void)
574 {
575 if( vg_console.suggestion_count ){
576 _console_suggest_store_normal();
577
578 vg_console.suggestion_select --;
579
580 if( vg_console.suggestion_select < -1 )
581 vg_console.suggestion_select = vg_console.suggestion_count-1;
582
583 _console_fetch_suggestion();
584 }
585 }
586
587 VG_STATIC void _vg_console_on_update( char *buf, u32 len )
588 {
589 if( buf == vg_console.input ){
590 console_update_suggestions();
591 }
592 }
593
594 VG_STATIC void console_history_get( char* buf, int entry_num )
595 {
596 if( !vg_console.history_count )
597 return;
598
599 int offset = VG_MIN( entry_num, vg_console.history_count -1 ),
600 pick = (vg_console.history_last - offset) %
601 vg_list_size( vg_console.history );
602 strcpy( buf, vg_console.history[ pick ] );
603 }
604
605 VG_STATIC void _vg_console_on_up( char *buf, u32 len )
606 {
607 if( buf == vg_console.input ){
608 vg_console.history_pos =
609 VG_MAX
610 (
611 0,
612 VG_MIN
613 (
614 vg_console.history_pos+1,
615 VG_MIN
616 (
617 vg_list_size( vg_console.history ),
618 vg_console.history_count - 1
619 )
620 )
621 );
622
623 console_history_get( vg_console.input, vg_console.history_pos );
624 _ui_textbox_move_cursor( &vg_ui.textbox.cursor_user,
625 &vg_ui.textbox.cursor_pos,
626 vg_list_size(vg_console.input)-1, 1);
627 }
628 }
629
630 VG_STATIC void _vg_console_on_down( char *buf, u32 len )
631 {
632 if( buf == vg_console.input ){
633 vg_console.history_pos = VG_MAX( 0, vg_console.history_pos-1 );
634 console_history_get( vg_console.input, vg_console.history_pos );
635
636 _ui_textbox_move_cursor( &vg_ui.textbox.cursor_user,
637 &vg_ui.textbox.cursor_pos,
638 vg_list_size(vg_console.input)-1, 1 );
639 }
640 }
641
642 VG_STATIC void _vg_console_on_enter( char *buf, u32 len )
643 {
644 if( buf == vg_console.input ){
645 if( !strlen( vg_console.input ) )
646 return;
647
648 vg_info( "%s\n", vg_console.input );
649
650 if( strcmp( vg_console.input,
651 vg_console.history[ vg_console.history_last ]) )
652 {
653 vg_console.history_last = ( vg_console.history_last + 1) %
654 vg_list_size(vg_console.history );
655 vg_console.history_count =
656 VG_MIN( vg_list_size( vg_console.history ),
657 vg_console.history_count + 1 );
658 strcpy( vg_console.history[ vg_console.history_last ],
659 vg_console.input );
660 }
661
662 vg_console.history_pos = -1;
663 vg_execute_console_input( vg_console.input );
664 _ui_textbox_move_cursor( &vg_ui.textbox.cursor_user,
665 &vg_ui.textbox.cursor_pos, -10000, 1 );
666
667 vg_console.input[0] = '\0';
668 console_update_suggestions();
669 }
670 }
671
672 VG_STATIC void _vg_console_draw(void)
673 {
674 if( !vg_console.enabled ) return;
675
676 SDL_AtomicLock( &log_print_sl );
677
678 int ptr = vg_log.log_line_current;
679 int const fh = 14, log_lines = 32;
680 int console_lines = VG_MIN( log_lines, vg_log.log_line_count );
681
682 ui_rect rect_log = { 0, 0, vg.window_x, log_lines*fh },
683 rect_input = { 0, log_lines*fh + 1, vg.window_x, fh*2 },
684 rect_line = { 0, 0, vg.window_x, fh };
685
686 /*
687 * log
688 */
689 u32 bg_colour = (ui_colour( k_ui_bg )&0x00ffffff)|0x9f000000;
690
691 ui_fill( rect_log, bg_colour );
692 rect_line[1] = rect_log[1]+rect_log[3]-fh;
693
694 for( int i=0; i<console_lines; i ++ ){
695 ptr --;
696
697 if( ptr < 0 ) ptr = vg_list_size( vg_log.log )-1;
698
699 ui_text( rect_line, vg_log.log[ptr], 1, k_ui_align_left, 0 );
700 rect_line[1] -= fh;
701 }
702
703 /*
704 * Input area
705 */
706
707 struct ui_textbox_callbacks callbacks = {
708 .up = _vg_console_on_up,
709 .down = _vg_console_on_down,
710 .change = _vg_console_on_update,
711 .enter = _vg_console_on_enter
712 };
713 ui_textbox( rect_input, vg_console.input, vg_list_size(vg_console.input),
714 UI_TEXTBOX_AUTOFOCUS, &callbacks );
715
716 /*
717 * suggestions
718 */
719 if( vg_console.suggestion_count ){
720 ui_rect rect_suggest;
721 rect_copy( rect_input, rect_suggest );
722
723 rect_suggest[0] += 6 + UI_GLYPH_SPACING_X*vg_console.suggestion_pastepos;
724 rect_suggest[1] += rect_input[3];
725 rect_suggest[2] = UI_GLYPH_SPACING_X * vg_console.suggestion_maxlen;
726 rect_suggest[3] = vg_console.suggestion_count * fh;
727
728 ui_fill( rect_suggest, bg_colour );
729
730 rect_suggest[3] = fh;
731
732 for( int i=0; i<vg_console.suggestion_count; i ++ ){
733 u32 text_colour;
734 if( i == vg_console.suggestion_select ){
735 ui_fill( rect_suggest, ui_colour( k_ui_orange ) );
736 text_colour = ui_colourcont( k_ui_orange );
737 }
738 else text_colour = ui_colourcont( k_ui_bg );
739
740 ui_text( rect_suggest, vg_console.suggestions[i].str, 1,
741 k_ui_align_left, text_colour );
742
743 rect_suggest[1] += fh;
744 }
745 }
746
747 SDL_AtomicUnlock( &log_print_sl );
748 }
749
750
751 #endif /* VG_CONSOLE_H */