Don't use metadesk for training data, actions
parent
abc6abe1a2
commit
49751f8cfa
@ -0,0 +1,147 @@
|
|||||||
|
let SessionLoad = 1
|
||||||
|
let s:so_save = &g:so | let s:siso_save = &g:siso | setg so=0 siso=0 | setl so=-1 siso=-1
|
||||||
|
let v:this_session=expand("<sfile>:p")
|
||||||
|
silent only
|
||||||
|
silent tabonly
|
||||||
|
cd ~/Documents/rpgpt
|
||||||
|
if expand('%') == '' && !&modified && line('$') <= 1 && getline(1) == ''
|
||||||
|
let s:wipebuf = bufnr('%')
|
||||||
|
endif
|
||||||
|
let s:shortmess_save = &shortmess
|
||||||
|
if &shortmess =~ 'A'
|
||||||
|
set shortmess=aoOA
|
||||||
|
else
|
||||||
|
set shortmess=aoO
|
||||||
|
endif
|
||||||
|
badd +0 codegen.c
|
||||||
|
badd +0 main.c
|
||||||
|
badd +0 buff.h
|
||||||
|
badd +0 makeprompt.h
|
||||||
|
badd +0 profiling.h
|
||||||
|
badd +0 term://~/Documents/rpgpt//42668:C:/Windows/system32/cmd.exe
|
||||||
|
argglobal
|
||||||
|
%argdel
|
||||||
|
$argadd codegen.c
|
||||||
|
$argadd main.c
|
||||||
|
$argadd buff.h
|
||||||
|
$argadd makeprompt.h
|
||||||
|
$argadd profiling.h
|
||||||
|
set stal=2
|
||||||
|
tabnew +setlocal\ bufhidden=wipe
|
||||||
|
tabrewind
|
||||||
|
let s:save_splitbelow = &splitbelow
|
||||||
|
let s:save_splitright = &splitright
|
||||||
|
set splitbelow splitright
|
||||||
|
wincmd _ | wincmd |
|
||||||
|
vsplit
|
||||||
|
1wincmd h
|
||||||
|
wincmd w
|
||||||
|
let &splitbelow = s:save_splitbelow
|
||||||
|
let &splitright = s:save_splitright
|
||||||
|
wincmd t
|
||||||
|
let s:save_winminheight = &winminheight
|
||||||
|
let s:save_winminwidth = &winminwidth
|
||||||
|
set winminheight=0
|
||||||
|
set winheight=1
|
||||||
|
set winminwidth=0
|
||||||
|
set winwidth=1
|
||||||
|
exe 'vert 1resize ' . ((&columns * 124 + 104) / 209)
|
||||||
|
exe 'vert 2resize ' . ((&columns * 84 + 104) / 209)
|
||||||
|
argglobal
|
||||||
|
enew
|
||||||
|
setlocal fdm=manual
|
||||||
|
setlocal fde=0
|
||||||
|
setlocal fmr={{{,}}}
|
||||||
|
setlocal fdi=#
|
||||||
|
setlocal fdl=0
|
||||||
|
setlocal fml=1
|
||||||
|
setlocal fdn=20
|
||||||
|
setlocal fen
|
||||||
|
wincmd w
|
||||||
|
argglobal
|
||||||
|
if bufexists(fnamemodify("term://~/Documents/rpgpt//42668:C:/Windows/system32/cmd.exe", ":p")) | buffer term://~/Documents/rpgpt//42668:C:/Windows/system32/cmd.exe | else | edit term://~/Documents/rpgpt//42668:C:/Windows/system32/cmd.exe | endif
|
||||||
|
if &buftype ==# 'terminal'
|
||||||
|
silent file term://~/Documents/rpgpt//42668:C:/Windows/system32/cmd.exe
|
||||||
|
endif
|
||||||
|
setlocal fdm=manual
|
||||||
|
setlocal fde=0
|
||||||
|
setlocal fmr={{{,}}}
|
||||||
|
setlocal fdi=#
|
||||||
|
setlocal fdl=0
|
||||||
|
setlocal fml=1
|
||||||
|
setlocal fdn=20
|
||||||
|
setlocal fen
|
||||||
|
let s:l = 1 - ((0 * winheight(0) + 24) / 49)
|
||||||
|
if s:l < 1 | let s:l = 1 | endif
|
||||||
|
keepjumps exe s:l
|
||||||
|
normal! zt
|
||||||
|
keepjumps 1
|
||||||
|
normal! 0
|
||||||
|
wincmd w
|
||||||
|
exe 'vert 1resize ' . ((&columns * 124 + 104) / 209)
|
||||||
|
exe 'vert 2resize ' . ((&columns * 84 + 104) / 209)
|
||||||
|
tabnext
|
||||||
|
let s:save_splitbelow = &splitbelow
|
||||||
|
let s:save_splitright = &splitright
|
||||||
|
set splitbelow splitright
|
||||||
|
wincmd _ | wincmd |
|
||||||
|
vsplit
|
||||||
|
1wincmd h
|
||||||
|
wincmd w
|
||||||
|
let &splitbelow = s:save_splitbelow
|
||||||
|
let &splitright = s:save_splitright
|
||||||
|
wincmd t
|
||||||
|
let s:save_winminheight = &winminheight
|
||||||
|
let s:save_winminwidth = &winminwidth
|
||||||
|
set winminheight=0
|
||||||
|
set winheight=1
|
||||||
|
set winminwidth=0
|
||||||
|
set winwidth=1
|
||||||
|
exe 'vert 1resize ' . ((&columns * 104 + 104) / 209)
|
||||||
|
exe 'vert 2resize ' . ((&columns * 104 + 104) / 209)
|
||||||
|
argglobal
|
||||||
|
enew
|
||||||
|
balt term://~/Documents/rpgpt//42668:C:/Windows/system32/cmd.exe
|
||||||
|
setlocal fdm=manual
|
||||||
|
setlocal fde=0
|
||||||
|
setlocal fmr={{{,}}}
|
||||||
|
setlocal fdi=#
|
||||||
|
setlocal fdl=0
|
||||||
|
setlocal fml=1
|
||||||
|
setlocal fdn=20
|
||||||
|
setlocal fen
|
||||||
|
wincmd w
|
||||||
|
argglobal
|
||||||
|
enew
|
||||||
|
balt term://~/Documents/rpgpt//42668:C:/Windows/system32/cmd.exe
|
||||||
|
setlocal fdm=manual
|
||||||
|
setlocal fde=0
|
||||||
|
setlocal fmr={{{,}}}
|
||||||
|
setlocal fdi=#
|
||||||
|
setlocal fdl=0
|
||||||
|
setlocal fml=1
|
||||||
|
setlocal fdn=20
|
||||||
|
setlocal fen
|
||||||
|
wincmd w
|
||||||
|
exe 'vert 1resize ' . ((&columns * 104 + 104) / 209)
|
||||||
|
exe 'vert 2resize ' . ((&columns * 104 + 104) / 209)
|
||||||
|
tabnext 2
|
||||||
|
set stal=1
|
||||||
|
if exists('s:wipebuf') && len(win_findbuf(s:wipebuf)) == 0 && getbufvar(s:wipebuf, '&buftype') isnot# 'terminal'
|
||||||
|
silent exe 'bwipe ' . s:wipebuf
|
||||||
|
endif
|
||||||
|
unlet! s:wipebuf
|
||||||
|
set winheight=1 winwidth=20
|
||||||
|
let &shortmess = s:shortmess_save
|
||||||
|
let &winminheight = s:save_winminheight
|
||||||
|
let &winminwidth = s:save_winminwidth
|
||||||
|
let s:sx = expand("<sfile>:p:r")."x.vim"
|
||||||
|
if filereadable(s:sx)
|
||||||
|
exe "source " . fnameescape(s:sx)
|
||||||
|
endif
|
||||||
|
let &g:so = s:so_save | let &g:siso = s:siso_save
|
||||||
|
set hlsearch
|
||||||
|
nohlsearch
|
||||||
|
doautoall SessionLoadPost
|
||||||
|
unlet SessionLoad
|
||||||
|
" vim: set ft=vim :
|
||||||
@ -0,0 +1,28 @@
|
|||||||
|
{
|
||||||
|
"activeFile": "",
|
||||||
|
"expandedProjectPaths": [
|
||||||
|
],
|
||||||
|
"fileStates": {
|
||||||
|
"assets/level0.json": {
|
||||||
|
"scale": 0.17733333333333334,
|
||||||
|
"selectedLayer": 0,
|
||||||
|
"viewCenter": {
|
||||||
|
"x": 1843.9849624060148,
|
||||||
|
"y": 1240.6015037593986
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"assets/testsmalllevel.json": {
|
||||||
|
"scale": 3.3249999999999997,
|
||||||
|
"selectedLayer": 0,
|
||||||
|
"viewCenter": {
|
||||||
|
"x": 128.1203007518797,
|
||||||
|
"y": 128.1203007518797
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"openFiles": [
|
||||||
|
],
|
||||||
|
"project": "main.tiled-project",
|
||||||
|
"recentFiles": [
|
||||||
|
]
|
||||||
|
}
|
||||||
@ -0,0 +1,383 @@
|
|||||||
|
#pragma once
|
||||||
|
#include "buff.h"
|
||||||
|
#include "HandmadeMath.h" // vector types in entity struct definition
|
||||||
|
#include <assert.h>
|
||||||
|
#include <stdbool.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <stdlib.h> // atoi
|
||||||
|
#include "characters.gen.h"
|
||||||
|
NPC_Skeleton,
|
||||||
|
NPC_MOOSE,
|
||||||
|
} NpcKind;
|
||||||
|
|
||||||
|
#define Log(...) { printf("%s Log %d | ", __FILE__, __LINE__); printf(__VA_ARGS__); }
|
||||||
|
|
||||||
|
// REFACTORING:: also have to update in javascript!!!!!!!!
|
||||||
|
#define MAX_SENTENCE_LENGTH 400 // LOOOK AT AGBOVE COMMENT GBEFORE CHANGING
|
||||||
|
typedef BUFF(char, MAX_SENTENCE_LENGTH) Sentence;
|
||||||
|
#define SENTENCE_CONST(txt) {.data=txt, .cur_index=sizeof(txt)}
|
||||||
|
#define SENTENCE_CONST_CAST(txt) (Sentence)SENTENCE_CONST(txt)
|
||||||
|
|
||||||
|
#define REMEMBERED_PERCEPTIONS 24
|
||||||
|
|
||||||
|
typedef enum PerceptionType
|
||||||
|
{
|
||||||
|
Invalid, // so that zero value in training structs means end of perception
|
||||||
|
PlayerAction,
|
||||||
|
PlayerDialog,
|
||||||
|
NPCDialog, // includes an npc action in every npc dialog. So it's often nothing
|
||||||
|
PlayerHeldItemChanged,
|
||||||
|
} PerceptionType;
|
||||||
|
|
||||||
|
typedef struct Perception
|
||||||
|
{
|
||||||
|
PerceptionType type;
|
||||||
|
|
||||||
|
union
|
||||||
|
{
|
||||||
|
// player action
|
||||||
|
Action player_action_type;
|
||||||
|
|
||||||
|
// player dialog
|
||||||
|
Sentence player_dialog;
|
||||||
|
|
||||||
|
// npc dialog
|
||||||
|
struct
|
||||||
|
{
|
||||||
|
Action npc_action_type;
|
||||||
|
Sentence npc_dialog;
|
||||||
|
};
|
||||||
|
|
||||||
|
// player holding item. MUST precede any perceptions which come after the player is holding the item
|
||||||
|
ItemKind holding;
|
||||||
|
};
|
||||||
|
} Perception;
|
||||||
|
|
||||||
|
typedef enum PropKind
|
||||||
|
{
|
||||||
|
TREE0,
|
||||||
|
TREE1,
|
||||||
|
TREE2,
|
||||||
|
} PropKind;
|
||||||
|
|
||||||
|
typedef struct EntityRef
|
||||||
|
{
|
||||||
|
int index;
|
||||||
|
int generation;
|
||||||
|
} EntityRef;
|
||||||
|
|
||||||
|
typedef enum CharacterState
|
||||||
|
{
|
||||||
|
CHARACTER_WALKING,
|
||||||
|
CHARACTER_IDLE,
|
||||||
|
CHARACTER_ATTACK,
|
||||||
|
CHARACTER_TALKING,
|
||||||
|
} CharacterState;
|
||||||
|
|
||||||
|
typedef struct Entity
|
||||||
|
{
|
||||||
|
bool exists;
|
||||||
|
bool destroy;
|
||||||
|
int generation;
|
||||||
|
|
||||||
|
// fields for all gs.entities
|
||||||
|
Vec2 pos;
|
||||||
|
Vec2 vel; // only used sometimes, like in old man and bullet
|
||||||
|
float damage; // at 1.0, dead! zero initialized
|
||||||
|
bool facing_left;
|
||||||
|
double dead_time;
|
||||||
|
bool dead;
|
||||||
|
// multiple gs.entities have a sword swing
|
||||||
|
BUFF(EntityRef, 8) done_damage_to_this_swing; // only do damage once, but hitbox stays around
|
||||||
|
|
||||||
|
bool is_bullet;
|
||||||
|
|
||||||
|
// props
|
||||||
|
bool is_prop;
|
||||||
|
PropKind prop_kind;
|
||||||
|
|
||||||
|
// items
|
||||||
|
bool is_item;
|
||||||
|
bool held_by_player;
|
||||||
|
ItemKind item_kind;
|
||||||
|
|
||||||
|
// npcs
|
||||||
|
bool is_npc;
|
||||||
|
bool perceptions_dirty;
|
||||||
|
BUFF(Perception, REMEMBERED_PERCEPTIONS) remembered_perceptions;
|
||||||
|
double character_say_timer;
|
||||||
|
int characters_said;
|
||||||
|
NpcKind npc_kind;
|
||||||
|
ItemKind last_seen_holding_kind;
|
||||||
|
#ifdef WEB
|
||||||
|
int gen_request_id;
|
||||||
|
#endif
|
||||||
|
bool aggressive;
|
||||||
|
bool walking;
|
||||||
|
double shotgun_timer;
|
||||||
|
bool moved;
|
||||||
|
Vec2 target_goto;
|
||||||
|
// only for skeleton npc
|
||||||
|
double swing_timer;
|
||||||
|
|
||||||
|
// character
|
||||||
|
bool is_character;
|
||||||
|
EntityRef holding_item;
|
||||||
|
Vec2 to_throw_direction;
|
||||||
|
int boots_modifier;
|
||||||
|
CharacterState state;
|
||||||
|
EntityRef talking_to; // Maybe should be generational index, but I dunno. No death yet
|
||||||
|
bool is_rolling; // can only roll in idle or walk states
|
||||||
|
double time_not_rolling; // for cooldown for roll, so you can't just hold it and be invincible
|
||||||
|
double roll_progress;
|
||||||
|
double swing_progress;
|
||||||
|
} Entity;
|
||||||
|
|
||||||
|
bool npc_is_knight_sprite(Entity *it)
|
||||||
|
{
|
||||||
|
return it->is_npc && ( it->npc_kind == NPC_Max || it->npc_kind == NPC_Hunter || it->npc_kind == NPC_John || it->npc_kind == NPC_Blocky);
|
||||||
|
}
|
||||||
|
|
||||||
|
typedef BUFF(char, MAX_SENTENCE_LENGTH*(REMEMBERED_PERCEPTIONS+4)) PromptBuff;
|
||||||
|
typedef BUFF(Action, 8) AvailableActions;
|
||||||
|
|
||||||
|
void fill_available_actions(Entity *it, AvailableActions *a)
|
||||||
|
{
|
||||||
|
*a = (AvailableActions){0};
|
||||||
|
BUFF_APPEND(a, ACT_none);
|
||||||
|
if(it->npc_kind == NPC_GodRock)
|
||||||
|
{
|
||||||
|
BUFF_APPEND(a, ACT_heals_player);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
BUFF_APPEND(a, ACT_fights_player);
|
||||||
|
if(npc_is_knight_sprite(it))
|
||||||
|
{
|
||||||
|
BUFF_APPEND(a, ACT_strikes_air);
|
||||||
|
}
|
||||||
|
if(it->npc_kind == NPC_Blocky)
|
||||||
|
{
|
||||||
|
if(!it->moved)
|
||||||
|
{
|
||||||
|
BUFF_APPEND(a, ACT_allows_player_to_pass);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void process_perception(Entity *it, Perception p)
|
||||||
|
{
|
||||||
|
if(p.type != NPCDialog) it->perceptions_dirty = true;
|
||||||
|
if(!BUFF_HAS_SPACE(&it->remembered_perceptions)) BUFF_REMOVE_FRONT(&it->remembered_perceptions);
|
||||||
|
BUFF_APPEND(&it->remembered_perceptions, p);
|
||||||
|
if(p.type == PlayerHeldItemChanged)
|
||||||
|
{
|
||||||
|
it->last_seen_holding_kind = p.holding;
|
||||||
|
}
|
||||||
|
else if(p.type == NPCDialog)
|
||||||
|
{
|
||||||
|
if(p.npc_action_type == ACT_allows_player_to_pass)
|
||||||
|
{
|
||||||
|
it->target_goto = AddV2(it->pos, V2(-50.0, 0.0));
|
||||||
|
it->moved = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#define printf_buff(buff_ptr, ...) { int written = snprintf((buff_ptr)->data+(buff_ptr)->cur_index, ARRLEN((buff_ptr)->data) - (buff_ptr)->cur_index, __VA_ARGS__); assert(written >= 0); (buff_ptr)->cur_index += written; };
|
||||||
|
|
||||||
|
// returns if action index was valid
|
||||||
|
bool action_from_index(Entity *it, Action *out, int action_index)
|
||||||
|
{
|
||||||
|
AvailableActions available = {0};
|
||||||
|
fill_available_actions(it, &available);
|
||||||
|
if(action_index < 0 || action_index >= available.cur_index)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
*out = available.data[action_index];
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// don't call on untrusted action, doesn't return error
|
||||||
|
int action_to_index(Entity *it, Action a)
|
||||||
|
{
|
||||||
|
AvailableActions available = {0};
|
||||||
|
fill_available_actions(it, &available);
|
||||||
|
Action target_action = a;
|
||||||
|
int index = -1;
|
||||||
|
for(int i = 0; i < available.cur_index; i++)
|
||||||
|
{
|
||||||
|
if(available.data[i] == target_action)
|
||||||
|
{
|
||||||
|
index = i;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
assert(index != -1);
|
||||||
|
return index;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void generate_prompt(Entity *it, PromptBuff *into)
|
||||||
|
{
|
||||||
|
assert(it->is_npc);
|
||||||
|
*into = (PromptBuff){0};
|
||||||
|
|
||||||
|
// global prompt
|
||||||
|
printf_buff(into, "%s", global_prompt);
|
||||||
|
printf_buff(into, "%s", "\n");
|
||||||
|
|
||||||
|
// npc description prompt
|
||||||
|
assert(it->npc_kind < ARRLEN(prompt_table));
|
||||||
|
printf_buff(into, "%s", prompt_table[it->npc_kind]);
|
||||||
|
printf_buff(into, "%s", "\n");
|
||||||
|
|
||||||
|
// item prompt
|
||||||
|
if(it->last_seen_holding_kind != ITEM_nothing)
|
||||||
|
{
|
||||||
|
assert(it->last_seen_holding_kind < ARRLEN(item_prompt_table));
|
||||||
|
printf_buff(into, "%s", item_prompt_table[it->last_seen_holding_kind]);
|
||||||
|
printf_buff(into, "%s", "\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
// available actions prompt
|
||||||
|
AvailableActions available = {0};
|
||||||
|
fill_available_actions(it, &available);
|
||||||
|
printf_buff(into, "%s", "The NPC possible actions array, indexed by ACT_INDEX: [");
|
||||||
|
BUFF_ITER(Action, &available)
|
||||||
|
{
|
||||||
|
printf_buff(into, "%s", action_strings[*it]);
|
||||||
|
printf_buff(into, "%s", ", ");
|
||||||
|
}
|
||||||
|
printf_buff(into, "%s", "]\n");
|
||||||
|
|
||||||
|
Entity *e = it;
|
||||||
|
ItemKind last_holding = ITEM_nothing;
|
||||||
|
BUFF_ITER(Perception, &e->remembered_perceptions)
|
||||||
|
{
|
||||||
|
if(it->type == PlayerAction)
|
||||||
|
{
|
||||||
|
assert(it->player_action_type < ARRLEN(action_strings));
|
||||||
|
printf_buff(into, "Player: ACT %s \n", action_strings[it->player_action_type]);
|
||||||
|
}
|
||||||
|
else if(it->type == PlayerDialog)
|
||||||
|
{
|
||||||
|
printf_buff(into, "%s", "Player: \"");
|
||||||
|
printf_buff(into, "%s", it->player_dialog.data);
|
||||||
|
printf_buff(into, "%s", "\"\n");
|
||||||
|
}
|
||||||
|
else if(it->type == NPCDialog)
|
||||||
|
{
|
||||||
|
printf_buff(into, "%s: ACT %s \"%s\"\n", name_table[e->npc_kind], action_strings[it->npc_action_type], it->npc_dialog.data);
|
||||||
|
}
|
||||||
|
else if(it->type == PlayerHeldItemChanged)
|
||||||
|
{
|
||||||
|
if(last_holding != it->holding)
|
||||||
|
{
|
||||||
|
if(last_holding != ITEM_nothing)
|
||||||
|
{
|
||||||
|
printf_buff(into, "%s", item_discard_message_table[last_holding]);
|
||||||
|
printf_buff(into, "%s", "\n");
|
||||||
|
}
|
||||||
|
if(it->holding != ITEM_nothing)
|
||||||
|
{
|
||||||
|
printf_buff(into, "%s", item_possess_message_table[it->holding]);
|
||||||
|
printf_buff(into, "%s", "\n");
|
||||||
|
}
|
||||||
|
last_holding = it->holding;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
assert(false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
printf_buff(into, "%s: ACT_INDEX", name_table[e->npc_kind]);
|
||||||
|
}
|
||||||
|
|
||||||
|
// returns if the response was well formatted
|
||||||
|
bool parse_ai_response(Entity *it, char *sentence_str, Perception *out)
|
||||||
|
{
|
||||||
|
*out = (Perception){0};
|
||||||
|
out->type = NPCDialog;
|
||||||
|
|
||||||
|
size_t sentence_length = strlen(sentence_str);
|
||||||
|
bool text_was_well_formatted = true;
|
||||||
|
|
||||||
|
BUFF(char, 128) action_index_string = {0};
|
||||||
|
int npc_sentence_beginning = 0;
|
||||||
|
for(int i = 0; i < sentence_length; i++)
|
||||||
|
{
|
||||||
|
if(i == 0)
|
||||||
|
{
|
||||||
|
if(sentence_str[i] != ' ')
|
||||||
|
{
|
||||||
|
text_was_well_formatted = false;
|
||||||
|
Log("Poorly formatted AI string, did not start with a ' ': `%s`\n", sentence_str);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if(sentence_str[i] == ' ')
|
||||||
|
{
|
||||||
|
npc_sentence_beginning = i + 2;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
BUFF_APPEND(&action_index_string, sentence_str[i]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if(sentence_str[npc_sentence_beginning - 1] != '"' || npc_sentence_beginning == 0)
|
||||||
|
{
|
||||||
|
Log("Poorly formatted AI string, sentence beginning incorrect in AI string `%s` NPC sentence beginning %d ...\n", sentence_str, npc_sentence_beginning);
|
||||||
|
text_was_well_formatted = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
Action npc_action = 0;
|
||||||
|
if(text_was_well_formatted)
|
||||||
|
{
|
||||||
|
int index_of_action = atoi(action_index_string.data);
|
||||||
|
|
||||||
|
if(!action_from_index(it, &npc_action, index_of_action))
|
||||||
|
{
|
||||||
|
Log("AI output invalid action index %d action index string %s\n", index_of_action, action_index_string.data);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Sentence what_npc_said = {0};
|
||||||
|
bool found_end_quote = false;
|
||||||
|
for(int i = npc_sentence_beginning; i < sentence_length; i++)
|
||||||
|
{
|
||||||
|
if(sentence_str[i] == '"')
|
||||||
|
{
|
||||||
|
found_end_quote = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
BUFF_APPEND(&what_npc_said, sentence_str[i]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if(!found_end_quote)
|
||||||
|
{
|
||||||
|
Log("Poorly formatted AI string, couln't find matching end quote in string %s...\n", sentence_str);
|
||||||
|
text_was_well_formatted = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(text_was_well_formatted)
|
||||||
|
{
|
||||||
|
out->npc_action_type = npc_action;
|
||||||
|
out->npc_dialog = what_npc_said;
|
||||||
|
}
|
||||||
|
|
||||||
|
return text_was_well_formatted;
|
||||||
|
}
|
||||||
@ -0,0 +1,160 @@
|
|||||||
|
#pragma warning(disable : 4996) // nonsense about fopen being insecure
|
||||||
|
#include <stdio.h>
|
||||||
|
#include "makeprompt.h"
|
||||||
|
|
||||||
|
typedef struct
|
||||||
|
{
|
||||||
|
NpcKind npc_kind;
|
||||||
|
Perception perceptions[32];
|
||||||
|
} TrainingSample;
|
||||||
|
|
||||||
|
#define PlayerAct(act) { .type = PlayerAction, .player_action_type = act, }
|
||||||
|
#define PlayerSay(txt) { .type = PlayerDialog, .player_dialog = SENTENCE_CONST(txt), }
|
||||||
|
#define NPCDoSay(d, txt) { .type = NPCDialog, .npc_action_type = d, .npc_dialog = SENTENCE_CONST(txt) }
|
||||||
|
#define NPCSay(txt) NPCDoSay(ACT_none, txt)
|
||||||
|
#define PlayerItemChange(new_item) { .type = PlayerHeldItemChanged, .holding = new_item, }
|
||||||
|
|
||||||
|
TrainingSample samples[] = {
|
||||||
|
{
|
||||||
|
.npc_kind = NPC_OldMan,
|
||||||
|
.perceptions = {
|
||||||
|
PlayerAct(ACT_walks_up),
|
||||||
|
NPCSay("Young warrior! You must stop Death, there isn't much time."),
|
||||||
|
PlayerAct(ACT_leaves),
|
||||||
|
NPCSay("Please!"),
|
||||||
|
PlayerAct(ACT_walks_up),
|
||||||
|
NPCSay("There you are! You must fight death!"),
|
||||||
|
PlayerSay("What what butt ass"),
|
||||||
|
NPCSay("You must stop death in his tracks and let the village live on!"),
|
||||||
|
PlayerSay("Nah"),
|
||||||
|
NPCSay("PLEASE!"),
|
||||||
|
PlayerAct(ACT_hits_npc),
|
||||||
|
NPCDoSay(ACT_fights_player, "Ready your sword!"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
.npc_kind = NPC_Blocky,
|
||||||
|
.perceptions = {
|
||||||
|
PlayerAct(ACT_walks_up),
|
||||||
|
PlayerSay("Do you think I should use webgl1 or webgl2?"),
|
||||||
|
NPCSay("What are either of those things?"),
|
||||||
|
PlayerSay("Let me pass"),
|
||||||
|
NPCSay("No"),
|
||||||
|
PlayerAct(ACT_leaves),
|
||||||
|
NPCDoSay(ACT_strikes_air, "Goodbye"),
|
||||||
|
PlayerItemChange(ITEM_Tripod),
|
||||||
|
PlayerSay("Let me pass"),
|
||||||
|
NPCDoSay(ACT_allows_player_to_pass, "A beautiful sight, that tripod is."),
|
||||||
|
PlayerSay("Thanks"),
|
||||||
|
NPCSay("Of course."),
|
||||||
|
PlayerSay("Let me pass"),
|
||||||
|
NPCSay("I can't do that, I already let you pass"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
.npc_kind = NPC_Blocky,
|
||||||
|
.perceptions = {
|
||||||
|
PlayerSay("hey"),
|
||||||
|
NPCSay("I'm just standing here, what are you doing?"),
|
||||||
|
PlayerItemChange(ITEM_nothing),
|
||||||
|
PlayerSay("nothing much"),
|
||||||
|
NPCSay("You don't have a tripod."),
|
||||||
|
PlayerSay("True"),
|
||||||
|
NPCSay("Do you want me to be standing there?"),
|
||||||
|
PlayerSay("Yes"),
|
||||||
|
NPCSay("Alright then"),
|
||||||
|
PlayerSay("No"),
|
||||||
|
NPCSay("What do you mean?"),
|
||||||
|
PlayerSay("What are you doing?"),
|
||||||
|
NPCSay("I'm just standing here"),
|
||||||
|
PlayerItemChange(ITEM_Tripod),
|
||||||
|
PlayerSay("How about now?"),
|
||||||
|
NPCSay("Mighty fine tripod you have there."),
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
.npc_kind = NPC_GodRock,
|
||||||
|
.perceptions = {
|
||||||
|
PlayerAct(ACT_walks_up),
|
||||||
|
NPCSay("I am"),
|
||||||
|
PlayerSay("Whaddup?"),
|
||||||
|
NPCSay("The clouds part and reveal only me."),
|
||||||
|
PlayerSay("uhhh. GIve me gold"),
|
||||||
|
NPCSay("To the greedy, come few"),
|
||||||
|
PlayerSay("I repent"),
|
||||||
|
NPCDoSay(ACT_heals_player, "And you are forgiven."),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
.npc_kind = NPC_GodRock,
|
||||||
|
.perceptions = {
|
||||||
|
PlayerAct(ACT_walks_up),
|
||||||
|
NPCSay("I am"),
|
||||||
|
PlayerSay("Who are you?"),
|
||||||
|
NPCSay("Have you heard of the high elves?"),
|
||||||
|
PlayerSay("Yes"),
|
||||||
|
NPCDoSay(ACT_heals_player, "No need for me to speak then."),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
typedef BUFF(char, 1024*10) Escaped;
|
||||||
|
Escaped escape_for_json(char *s)
|
||||||
|
{
|
||||||
|
Escaped to_return = {0};
|
||||||
|
size_t len = strlen(s);
|
||||||
|
for(int i = 0; i < len; i++)
|
||||||
|
{
|
||||||
|
if(s[i] == '\n')
|
||||||
|
{
|
||||||
|
BUFF_APPEND(&to_return, '\\');
|
||||||
|
BUFF_APPEND(&to_return, 'n');
|
||||||
|
}
|
||||||
|
else if(s[i] == '"')
|
||||||
|
{
|
||||||
|
BUFF_APPEND(&to_return, '\\');
|
||||||
|
BUFF_APPEND(&to_return, '"');
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
BUFF_APPEND(&to_return, s[i]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return to_return;
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int argc, char ** argv)
|
||||||
|
{
|
||||||
|
Log("Starting to generate training data\n");
|
||||||
|
FILE *out = fopen("gen/training_data.jsonl", "w");
|
||||||
|
assert(out);
|
||||||
|
int total_samples = 0;
|
||||||
|
for(int i = 0; i < ARRLEN(samples); i++)
|
||||||
|
{
|
||||||
|
TrainingSample s = samples[i];
|
||||||
|
Entity this_entity = (Entity){.is_npc = true, .npc_kind = s.npc_kind};
|
||||||
|
PromptBuff cur_prompt = {0};
|
||||||
|
for(int i = 0; i < ARRLEN(s.perceptions); i++)
|
||||||
|
{
|
||||||
|
if(s.perceptions[i].type == Invalid) break;
|
||||||
|
if(s.perceptions[i].type == NPCDialog)
|
||||||
|
{
|
||||||
|
Log("Generating sample of length %d for NPC %s...\n", i, name_table[s.npc_kind]);
|
||||||
|
total_samples++;
|
||||||
|
generate_prompt(&this_entity, &cur_prompt);
|
||||||
|
BUFF(char, 1024) completion = {0};
|
||||||
|
|
||||||
|
// get index from action
|
||||||
|
int index = action_to_index(&this_entity, s.perceptions[i].npc_action_type);
|
||||||
|
printf_buff(&completion, " %d \"%s\"\n", index, s.perceptions[i].npc_dialog.data);
|
||||||
|
|
||||||
|
fprintf(out, "{\"prompt\": \"%s\", \"completion\": \"%s\"}\n", escape_for_json(cur_prompt.data).data, escape_for_json(completion.data).data);
|
||||||
|
}
|
||||||
|
process_perception(&this_entity, s.perceptions[i]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Log("Finished. Total training samples: %d\n", total_samples);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue