power_play/src/meta/meta.c
2025-12-30 00:06:00 -06:00

1027 lines
36 KiB
C

#include "meta.h"
BuildCtx Build = Zi;
////////////////////////////////////////////////////////////
//~ Helpers
i32 GetBuildStatus(void)
{
return Atomic32Fetch(&Build.status);
}
void SetBuildStatus(i32 code)
{
Atomic32FetchTestSet(&Build.status, 0, code);
}
void EchoLine(String msg)
{
TempArena scratch = BeginScratchNoConflict();
{
String msg_w_newline = StringF(scratch.arena, "%F\n", FmtString(msg));
Echo(msg_w_newline);
}
EndScratch(scratch);
}
void EchoLineOrNothing(String msg)
{
String trimmed_msg = TrimWhitespace(msg);
if (trimmed_msg.len > 0)
{
EchoLine(trimmed_msg);
}
}
LineCol LineColFromPos(String data, i64 pos)
{
TempArena scratch = BeginScratchNoConflict();
LineCol result = Zi;
for (u64 cur = 0; cur < data.len && cur <= (u64)pos; ++cur)
{
u8 c = data.text[cur];
if (c == '\n')
{
++result.line;
result.col = 0;
}
else if (c != '\r')
{
++result.col;
}
}
result.line += 1;
EndScratch(scratch);
return result;
}
String StringFromMetaErrors(Arena *arena, M_ErrorList errors)
{
TempArena scratch = BeginScratch(arena);
StringList error_strings = Zi;
for (M_Error *e = errors.first; e; e = e->next)
{
M_Token *token = e->token;
String token_file = token->file->name;
String token_file_data = token->file->data;
if (token_file.len > 0)
{
i64 token_pos = -1;
if (
token->s.len > 0
&& token_file_data.len > 0
&& token->s.text > token_file_data.text
&& token->s.text < (token_file_data.text + token_file_data.len)
)
{
token_pos = token->s.text - token_file_data.text;
}
LineCol line_col = Zi;
if (token_pos >= 0)
{
line_col = LineColFromPos(token_file_data, token_pos);
}
String formatted = StringF(
scratch.arena,
"%F:%F:%F: error: %F",
FmtString(token_file),
FmtSint(line_col.line),
FmtSint(line_col.col),
FmtString(e->msg)
);
PushStringToList(scratch.arena, &error_strings, formatted);
}
else
{
PushStringToList(scratch.arena, &error_strings, StringF(scratch.arena, "error: %F", FmtString(e->msg)));
}
}
String result = StringFromList(arena, error_strings, Lit("\n"));
EndScratch(scratch);
return result;
}
EmbedObj Embed(String store_name, String dir_path)
{
Arena *perm = PermArena();
EmbedObj result = Zi;
// Generate resource archive contents
String arc_contents = Zi;
{
StringList files = Zi;
F_FilesFromDir(perm, &files, dir_path, F_IterFlag_Recurse);
Struct(EntryNode)
{
EntryNode *next;
String entry_name;
String file_name;
};
EntryNode *first_entry = 0;
EntryNode *last_entry = 0;
u64 entries_count = 0;
for (StringListNode *file_node = files.first; file_node; file_node = file_node->next)
{
String file_name = file_node->s;
if (F_IsFile(file_name))
{
String entry_name = file_name;
if (entry_name.len > (dir_path.len + 1))
{
entry_name.len -= dir_path.len + 1;
entry_name.text += dir_path.len + 1;
}
entry_name = StringF(perm, "%F/%F", FmtString(store_name), FmtString(entry_name));
for (u64 i = 0; i < entry_name.len; ++i)
{
if (entry_name.text[i] == '\\')
{
entry_name.text[i] = '/';
}
}
EntryNode *en = PushStruct(perm, EntryNode);
en->entry_name = entry_name;
en->file_name = file_name;
SllQueuePush(first_entry, last_entry, en);
++entries_count;
}
}
// TODO: Cache dynamic bitbuffs?
BB_Buff bb = BB_AcquireDynamicBuff(Gibi(2));
BB_Writer bw = BB_WriterFromBuff(&bb);
// Write magic
BB_WriteUBits(&bw, ResourceEmbeddedMagic, 64);
// Write header
BB_WriteUBits(&bw, entries_count, 64);
// Reserve entries space
u64 entry_size = 0
+ 8 // Name start
+ 8 // Name end
+ 8 // Data start
+ 8; // Data end
u8 *entries_start = BB_GetWrittenRaw(&bw) + BB_GetNumBytesWritten(&bw);
u64 entries_size = entry_size * entries_count;
String entries_str = STRING(entries_size, entries_start);
BB_WriteSeekBytes(&bw, entries_size);
BB_Buff entries_bb = BB_BuffFromString(entries_str);
BB_Writer entries_bw = BB_WriterFromBuff(&entries_bb);
// Write entries
for (EntryNode *en = first_entry; en; en = en->next)
{
// TODO: Copy file data directly into archive file
String file_data = F_DataFromFile(perm, en->file_name);
// Write name
BB_WriteAlignBytes(&bw, 64);
u64 name_start = BB_GetNumBytesWritten(&bw) + 1;
BB_WriteString(&bw, en->entry_name);
u64 name_len = BB_GetNumBytesWritten(&bw) - name_start;
// Write data
BB_WriteAlignBytes(&bw, 64);
// FIXME: Why no +1 here?
u64 data_start = BB_GetNumBytesWritten(&bw);
BB_WriteBytes(&bw, file_data);
u64 data_len = BB_GetNumBytesWritten(&bw) - data_start;
// Write entry
BB_WriteUBits(&entries_bw, name_start, 64);
BB_WriteUBits(&entries_bw, name_len, 64);
BB_WriteUBits(&entries_bw, data_start, 64);
BB_WriteUBits(&entries_bw, data_len, 64);
}
arc_contents.len = BB_GetNumBytesWritten(&bw);
arc_contents.text = BB_GetWrittenRaw(&bw);
}
// Write archive to file
String arc_path = StringF(perm, "%F.arc", FmtString(store_name));
F_ClearWrite(arc_path, arc_contents);
// Generate object file
if (IsPlatformWindows)
{
// Generate RC file
String rc_out_file = StringF(perm, "%F.rc", FmtString(store_name));
{
RandState rs = Zi;
StringList rc_out_lines = Zi;
String arc_file_cp = F_GetFullCrossPlatform(perm, arc_path);
String line = StringF(perm, "%F_%F RCDATA \"%F\"", FmtString(Lit(Stringize(W32_EmbeddedDataPrefix))), FmtHex(RandU64FromState(&rs)), FmtString(arc_file_cp));
PushStringToList(perm, &rc_out_lines, line);
// Write to file
String rc_out = StringFromList(perm, rc_out_lines, Lit("\n"));
F_ClearWrite(rc_out_file, rc_out);
}
// Compile RC file
result.obj_file = StringF(perm, "%F.res", FmtString(store_name));
{
result.obj_file, FmtString(F_GetFull(perm, rc_out_file));
String cmd = StringF(perm, "rc.exe -nologo -fo %F %F", FmtString(result.obj_file), FmtString(F_GetFull(perm, rc_out_file)));
OS_CommandResult cmd_result = OS_RunCommand(perm, cmd);
String cmd_output = TrimWhitespace(cmd_result.output);
result.output = cmd_output;
result.return_code = cmd_result.code;
}
SetBuildStatus(result.return_code);
}
else
{
// TODO: Generate object files using .incbin on non-windows platforms
Panic(Lit("Resource embedding not implemented for this platform"));
}
return result;
}
////////////////////////////////////////////////////////////
//~ Build
void BuildEntryPoint(WaveLaneCtx *lane)
{
Arena *perm = PermArena();
//////////////////////////////
//- Dirty check
// Return rebuild code if metaprogram is dirty
if (lane->idx == 0)
{
// Read old metahash
u64 old_metahash = 0;
if (F_IsFile(Lit("metahash.dat")))
{
String hashes_str = F_DataFromFile(perm, Lit("metahash.dat"));
if (hashes_str.len == sizeof(old_metahash))
{
CopyBytes(&old_metahash, hashes_str.text, sizeof(old_metahash));
}
OS_Rm(Lit("metahash.dat"));
}
// Compute new metahash
u64 new_metahash = 0;
{
StringList check_files = Zi;
F_FilesFromDir(perm, &check_files, Lit("../src/base"), F_IterFlag_Recurse);
F_FilesFromDir(perm, &check_files, Lit("../src/meta"), F_IterFlag_Recurse);
PushStringToList(perm, &check_files, Lit("../src/config.h"));
for (StringListNode *n = check_files.first; n; n = n->next)
{
String file = n->s;
new_metahash = MixU64s(HashFnv64(new_metahash, file), OS_LastWriteTimestampFromPath(file));
}
}
// Exit if metaprogram needs recompilation
if (old_metahash == 0 || old_metahash == new_metahash)
{
F_ClearWrite(Lit("metahash.dat"), StringFromStruct(&new_metahash));
}
else
{
EchoLine(Lit("Metaprogram is dirty"));
ExitNow(MetaRebuildCode);
}
}
//////////////////////////////
//- Command line
Struct(CmdLineArgs)
{
String leaf_layer_name;
};
CmdLineArgs cmdline = Zi;
{
CommandlineArg layer_arg = CommandlineArgFromName(Lit("layer"));
String layer_name = Zi;
if (layer_arg.name.len != 0)
{
layer_name = layer_arg.value;
}
else
{
layer_name = StringFromCommandlineIdx(1);
}
if (layer_name.len == 0)
{
layer_name = Lit("pp");
if (lane->idx == 0)
{
EchoLine(Lit("No layer supplied, assuming \"pp\" build"));
}
}
cmdline.leaf_layer_name = layer_name;
}
if (lane->idx == 0)
{
EchoLine(StringF(perm, "Building layer \"%F\"", FmtString(cmdline.leaf_layer_name)));
}
//////////////////////////////
//- Generate compiler params
CompilerParams cp = Zi;
{
//- Common
{
PushStringToList(perm, &cp.defs, Lit("-DIsConsoleApp=0"));
PushStringToList(perm, &cp.defs, Lit("-DIsRtcEnabled=1"));
PushStringToList(perm, &cp.defs, Lit("-DIsAsanEnabled=0"));
PushStringToList(perm, &cp.defs, Lit("-DIsCrtlibEnabled=1"));
PushStringToList(perm, &cp.defs, Lit("-DIsDebinfoEnabled=1"));
PushStringToList(perm, &cp.defs, Lit("-DIsDeveloperModeEnabled=1"));
PushStringToList(perm, &cp.defs, Lit("-DIsUnoptimized=1"));
PushStringToList(perm, &cp.defs, Lit("-DIsTestingEnabled=0"));
PushStringToList(perm, &cp.defs, Lit("-DIsHotSwappingEnabled=1"));
}
//- Msvc
{
PushStringToList(perm, &cp.compiler_only_flags_msvc, Lit("-diagnostics:column"));
PushStringToList(perm, &cp.flags_msvc, Lit("-INCREMENTAL:NO"));
PushStringToList(perm, &cp.flags_msvc, Lit("-nologo"));
// Optimization
PushStringToList(perm, &cp.compiler_only_flags_msvc, Lit("-Od"));
// Debug info
PushStringToList(perm, &cp.flags_msvc, Lit("-DEBUG:FULL"));
PushStringToList(perm, &cp.compiler_only_flags_msvc, Lit("-Z7"));
// Enable warnings
PushStringToList(perm, &cp.warnings_msvc, Lit("-W4"));
PushStringToList(perm, &cp.warnings_msvc, Lit("-WX"));
// PushStringToList(perm, &cp.warnings_msvc, Lit("-we4013")); // function undefined; assuming extern returning int
PushStringToList(perm, &cp.warnings_msvc, Lit("-we4668")); // 'X' is not defined as a preprocessor macro, replacing with '0' for '#if/#elif'
// Disable warnings
PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4244")); // 'function': conversion from 'int' to 'f32', possible loss of data
PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4201")); // nonstandard extension used: nameless struct/union
PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4324")); // structure was padded due to alignment specifier
PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4100")); // unreferenced parameter
PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4101")); // unreferenced local variable
PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4189")); // local variable is initialized but not referenced
PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4200")); // nonstandard extension used: zero-sized array in struct/union
PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4702")); // unreachable code
PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4305")); // 'initializing': truncation from 'double' to 'f32'
// PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4127")); // conditional expression is constant
// PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4820")); // bytes padding added after data member
// PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4464")); // relative include path contains '..'
// PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4061")); // enumerator is not explicitly handled by a case label
// PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4242")); // conversion from X to Y, possible loss of data
// PushStringToList(perm, &cp.warnings_msvc, Lit("-wd4388")); // signed/unsigned mismatch
// PushStringToList(perm, &cp.warnings_msvc, Lit("-wd5045")); // Compiler will insert Spectre mitigation for memory load if /Qspectre switch specified
}
//- Clang
{
PushStringToList(perm, &cp.flags_clang, Lit("-std=c99"));
PushStringToList(perm, &cp.flags_clang, Lit("-fno-finite-loops"));
PushStringToList(perm, &cp.flags_clang, Lit("-fno-strict-aliasing"));
PushStringToList(perm, &cp.flags_clang, Lit("-g -gcodeview"));
PushStringToList(perm, &cp.flags_clang, Lit("-O0"));
PushStringToList(perm, &cp.flags_clang, Lit("-msse4.2"));
// Enable warnings
PushStringToList(perm, &cp.warnings_clang, Lit("-Wall"));
PushStringToList(perm, &cp.warnings_clang, Lit("-Werror"));
PushStringToList(perm, &cp.warnings_clang, Lit("-Wframe-larger-than=65536"));
PushStringToList(perm, &cp.warnings_clang, Lit("-Wmissing-prototypes"));
PushStringToList(perm, &cp.warnings_clang, Lit("-Wmissing-declarations"));
PushStringToList(perm, &cp.warnings_clang, Lit("-Wunused-variable"));
PushStringToList(perm, &cp.warnings_clang, Lit("-Wunused-but-set-variable"));
PushStringToList(perm, &cp.warnings_clang, Lit("-Wunused-parameter"));
PushStringToList(perm, &cp.warnings_clang, Lit("-Wimplicit-fallthrough"));
PushStringToList(perm, &cp.warnings_clang, Lit("-Wswitch"));
// Disable warnings
PushStringToList(perm, &cp.warnings_clang, Lit("-Wno-initializer-overrides"));
PushStringToList(perm, &cp.warnings_clang, Lit("-Wno-microsoft-enum-forward-reference"));
PushStringToList(perm, &cp.warnings_clang, Lit("-Wno-unused-variable"));
PushStringToList(perm, &cp.warnings_clang, Lit("-Wno-unused-parameter"));
PushStringToList(perm, &cp.warnings_clang, Lit("-Wno-incompatible-function-pointer-types"));
}
//- Dxc
{
// PushStringToList(perm, &cp.flags_dxc, Lit("-Od"));
PushStringToList(perm, &cp.flags_dxc, Lit("-O3"));
PushStringToList(perm, &cp.flags_dxc, Lit("-Zi -Qembed_debug"));
}
}
//////////////////////////////
//- Phase 1/3: Prep
// Phase 1/3: Prep (narrow)
// - Parse layers
// - Generate final C file
// - Generate final HLSL file
// - Generate resource dirs info
//
// Phase 2/3: Compile (wide)
// - Compile C
// - Compile & embed shaders
// - Embed resource dirs
//
// Phase 3/3: Link (narrow)
// - Link
// TODO: Dispatch OS commands asynchronously
String shader_store_name = Lit("ShadersStore");
String c_out_file = F_GetFull(perm, StringF(perm, "%F_gen.c", FmtString(cmdline.leaf_layer_name)));
String gpu_out_file = F_GetFull(perm, StringF(perm, "%F_gen.g", FmtString(cmdline.leaf_layer_name)));
if (lane->idx == 0 && GetBuildStatus() == 0)
{
//- Parse layers
{
// Lex
StringList src_dirs = Zi;
PushStringToList(perm, &src_dirs, Lit("../src"));
M_TokenFileList lexed = M_TokensFromSrcDirs(perm, src_dirs);
// Parse
M_LayerList parsed = M_LayersFromTokenFiles(perm, lexed);;
// Flatten
StringList starting_layer_names = Zi;
PushStringToList(perm, &starting_layer_names, cmdline.leaf_layer_name);
Build.layers_parse = M_FlattenEntries(perm, parsed, starting_layer_names);
SetBuildStatus(Build.layers_parse.errors.count > 0);
}
//- Generate C file
{
StringList c_store_lines = Zi;
StringList c_shader_lines = Zi;
StringList c_include_lines = Zi;
StringList c_bootstrap_lines = Zi;
{
for (M_Entry *entry = Build.layers_parse.first; entry->valid; entry = entry->next)
{
M_EntryKind kind = entry->kind;
M_Token *entry_tok = entry->name_token;
M_Token *arg0_tok = entry->arg_tokens[0];
M_Token *arg1_tok = entry->arg_tokens[1];
switch (kind)
{
default: break;
case M_EntryKind_EmbedDir:
{
if (arg0_tok->valid && arg1_tok->valid)
{
String store_name = arg0_tok->s;
String token_file = arg1_tok->file->name;
String token_parent_dir = F_GetParentDir(token_file);
String arg_dir = arg1_tok->s;
String full = F_GetFullCrossPlatform(perm, StringF(perm, "%F/%F", FmtString(token_parent_dir), FmtString(arg_dir)));
if (F_IsDir(full))
{
u64 hash = HashFnv64(Fnv64Basis, StringF(perm, "%F/", FmtString(store_name)));
String line = StringF(perm, "ResourceStore %F = { 0x%F };", FmtString(store_name), FmtHex(hash));
PushStringToList(perm, &c_store_lines, line);
}
else
{
String err = StringF(perm, "Directory '%F' not found", FmtString(full));
M_PushError(perm, &Build.c_parse.errors, arg1_tok, err);
}
}
else
{
M_PushError(perm, &Build.c_parse.errors, entry_tok, Lit("Expected resource store & directory name"));
}
} break;
case M_EntryKind_VertexShader:
case M_EntryKind_PixelShader:
case M_EntryKind_ComputeShader:
{
if (arg0_tok->valid)
{
String shader_type = kind == M_EntryKind_VertexShader ? Lit("VertexShader")
: kind == M_EntryKind_PixelShader ? Lit("PixelShader")
: kind == M_EntryKind_ComputeShader ? Lit("ComputeShader")
: Lit("");
String shader_name = arg0_tok->s;
u64 hash = HashFnv64(Fnv64Basis, StringF(perm, "%F/%F", FmtString(shader_store_name), FmtString(shader_name)));
String line = StringF(perm, "%F %F = { 0x%F };", FmtString(shader_type), FmtString(shader_name), FmtHex(hash));
PushStringToList(perm, &c_shader_lines, line);
}
else
{
M_PushError(perm, &Build.c_parse.errors, entry_tok, Lit("Expected shader name"));
}
} break;
case M_EntryKind_IncludeC:
{
if (arg0_tok->valid)
{
String token_file = arg0_tok->file->name;
String token_parent_dir = F_GetParentDir(token_file);
String arg_file = arg0_tok->s;
String full = F_GetFull(perm, StringF(perm, "%F/%F", FmtString(token_parent_dir), FmtString(arg_file)));
if (F_IsFile(full))
{
String line = StringF(perm, "#include \"%F\"", FmtString(full));
PushStringToList(perm, &c_include_lines, line);
}
else
{
String err = StringF(perm, "File '%F' not found", FmtString(full));
M_PushError(perm, &Build.c_parse.errors, arg0_tok, err);
}
}
else
{
M_PushError(perm, &Build.c_parse.errors, entry_tok, Lit("Expected file name"));
}
} break;
case M_EntryKind_Bootstrap:
{
if (arg0_tok->valid)
{
String bootstrap = arg0_tok->s;
String line = StringF(perm, " %F();", FmtString(bootstrap));
PushStringToList(perm, &c_bootstrap_lines, line);
}
else
{
M_PushError(perm, &Build.c_parse.errors, entry_tok, Lit("Expected bootstrap function name"));
}
} break;
}
}
}
if (Build.c_parse.errors.count == 0)
{
StringList c_out_lines = Zi;
PushStringToList(perm, &c_out_lines, Lit("// Auto generated file"));
// Include base layer
{
String base_inc_path = F_GetFull(perm, Lit("../src/base/base_inc.h"));
PushStringToList(perm, &c_out_lines, Lit(""));
PushStringToList(perm, &c_out_lines, Lit("//- Base layer includes"));
PushStringToList(perm, &c_out_lines, StringF(perm, "#include \"%F\"", FmtString(base_inc_path)));
}
// Define resource stores
if (c_store_lines.count > 0)
{
PushStringToList(perm, &c_out_lines, Lit(""));
PushStringToList(perm, &c_out_lines, Lit("//- Resource stores"));
for (StringListNode *n = c_store_lines.first; n; n = n->next)
{
PushStringToList(perm, &c_out_lines, n->s);
}
}
// Define shaders
if (c_shader_lines.count > 0)
{
PushStringToList(perm, &c_out_lines, Lit(""));
PushStringToList(perm, &c_out_lines, Lit("//- Shaders"));
for (StringListNode *n = c_shader_lines.first; n; n = n->next)
{
PushStringToList(perm, &c_out_lines, n->s);
}
}
// Include dependency layers
if (c_include_lines.count > 0)
{
PushStringToList(perm, &c_out_lines, Lit(""));
PushStringToList(perm, &c_out_lines, Lit("//- Dependency graph includes"));
for (StringListNode *n = c_include_lines.first; n; n = n->next)
{
PushStringToList(perm, &c_out_lines, n->s);
}
}
// Define BootstrapLayers
{
PushStringToList(perm, &c_out_lines, Lit(""));
PushStringToList(perm, &c_out_lines, Lit("//- Bootstrap"));
PushStringToList(perm, &c_out_lines, Lit("void BootstrapLayers(void)"));
PushStringToList(perm, &c_out_lines, Lit("{"));
for (StringListNode *n = c_bootstrap_lines.first; n; n = n->next)
{
PushStringToList(perm, &c_out_lines, n->s);
}
PushStringToList(perm, &c_out_lines, Lit("}"));
}
// Write to file
PushStringToList(perm, &c_out_lines, Lit(""));
String c_out = StringFromList(perm, c_out_lines, Lit("\n"));
F_ClearWrite(c_out_file, c_out);
}
SetBuildStatus(Build.c_parse.errors.count > 0);
}
//- Generate HLSL file
{
// Clear shader store
// TODO: Move to separate artifacts dir that gets cleared, including archive files
OS_Mkdir(shader_store_name);
{
// Remove all old shaders
StringList files = Zi;
F_FilesFromDir(perm, &files, shader_store_name, F_IterFlag_None);
for (StringListNode *n = files.first; n; n = n->next)
{
String file = n->s;
// Safety check to prevent non-shader files from being removed
if (StringEndsWith(file, Lit("VS")) || StringEndsWith(file, Lit("CS")) || StringEndsWith(file, Lit("PS")))
{
OS_Rm(n->s);
}
else
{
// Unexpected file in shader store
Assert(0);
}
}
}
// Generate GPU file & shader entries
{
StringList gpu_include_lines = Zi;
{
for (M_Entry *entry = Build.layers_parse.first; entry->valid; entry = entry->next)
{
M_EntryKind kind = entry->kind;
M_Token *entry_tok = entry->name_token;
M_Token *arg0_tok = entry->arg_tokens[0];
M_Token *arg1_tok = entry->arg_tokens[1];
switch (kind)
{
default: break;
case M_EntryKind_IncludeG:
{
if (arg0_tok->valid)
{
String token_file = arg0_tok->file->name;
String token_parent_dir = F_GetParentDir(token_file);
String arg_file = arg0_tok->s;
String full = F_GetFull(perm, StringF(perm, "%F/%F", FmtString(token_parent_dir), FmtString(arg_file)));
if (F_IsFile(full))
{
String line = StringF(perm, "#include \"%F\"", FmtString(full));
PushStringToList(perm, &gpu_include_lines, line);
}
else
{
String err = StringF(perm, "File '%F' not found", FmtString(full));
M_PushError(perm, &Build.gpu_parse.errors, arg0_tok, err);
}
}
else
{
M_PushError(perm, &Build.gpu_parse.errors, entry_tok, Lit("Expected file name"));
}
} break;
case M_EntryKind_VertexShader:
case M_EntryKind_PixelShader:
case M_EntryKind_ComputeShader:
{
if (arg0_tok->valid)
{
ShaderEntryKind shader_kind = kind == M_EntryKind_VertexShader ? ShaderEntryKind_VS
: kind == M_EntryKind_PixelShader ? ShaderEntryKind_PS
: kind == M_EntryKind_ComputeShader ? ShaderEntryKind_CS
: ShaderEntryKind_VS;
String shader_name = arg0_tok->s;
ShaderEntry *e = PushStruct(perm, ShaderEntry);
e->kind = shader_kind;
e->name = shader_name;
SllQueuePush(Build.gpu_parse.first_shader_entry, Build.gpu_parse.last_shader_entry, e);
++Build.gpu_parse.shader_entries_count;
}
else
{
M_PushError(perm, &Build.gpu_parse.errors, entry_tok, Lit("Expected shader name"));
}
} break;
}
}
}
if (Build.gpu_parse.errors.count == 0)
{
StringList gpu_out_lines = Zi;
PushStringToList(perm, &gpu_out_lines, Lit("// Auto generated file"));
// Include base layer
{
String base_inc_path = F_GetFull(perm, Lit("../src/base/base_inc.h"));
PushStringToList(perm, &gpu_out_lines, Lit(""));
PushStringToList(perm, &gpu_out_lines, Lit("//- Base layer includes"));
PushStringToList(perm, &gpu_out_lines, StringF(perm, "#include \"%F\"", FmtString(base_inc_path)));
}
// Include dependency layers
if (gpu_out_lines.count > 0)
{
PushStringToList(perm, &gpu_out_lines, Lit(""));
PushStringToList(perm, &gpu_out_lines, Lit("//- Dependency graph includes"));
for (StringListNode *n = gpu_include_lines.first; n; n = n->next)
{
PushStringToList(perm, &gpu_out_lines, n->s);
}
}
// Write to file
PushStringToList(perm, &gpu_out_lines, Lit(""));
String c_out = StringFromList(perm, gpu_out_lines, Lit("\n"));
F_ClearWrite(gpu_out_file, c_out);
}
}
SetBuildStatus(Build.gpu_parse.errors.count > 0);
}
//- Generate archive info
{
// Push embedded archive dirs
for (M_Entry *entry = Build.layers_parse.first; entry->valid; entry = entry->next)
{
M_EntryKind kind = entry->kind;
M_Token *entry_tok = entry->name_token;
M_Token *arg0_tok = entry->arg_tokens[0];
M_Token *arg1_tok = entry->arg_tokens[1];
switch (kind)
{
default: break;
case M_EntryKind_EmbedDir:
{
if (arg0_tok->valid && arg1_tok->valid)
{
String store_name = arg0_tok->s;
String token_file = arg1_tok->file->name;
String token_parent_dir = F_GetParentDir(token_file);
String arg_dir = arg1_tok->s;
String full = F_GetFullCrossPlatform(perm, StringF(perm, "%F/%F", FmtString(token_parent_dir), FmtString(arg_dir)));
if (F_IsDir(full))
{
ResDir *rd = PushStruct(perm, ResDir);
rd->store_name = store_name;
rd->dir_path = full;
SllQueuePush(Build.res_dir_parse.first_res_dir, Build.res_dir_parse.last_res_dir, rd);
++Build.res_dir_parse.res_dirs_count;
}
else
{
String err = StringF(perm, "Directory '%F' not found", FmtString(full));
M_PushError(perm, &Build.res_dir_parse.errors, arg1_tok, err);
}
}
else
{
M_PushError(perm, &Build.res_dir_parse.errors, entry_tok, Lit("Expected resource store & directory name"));
}
} break;
}
}
SetBuildStatus(Build.res_dir_parse.errors.count > 0);
}
//- Prep obj arrays
{
// Gpu objs
Build.gpu_objs.count = Build.gpu_parse.shader_entries_count;
Build.gpu_objs.array = PushStructs(perm, GpuObj, Build.gpu_objs.count);
// Embed objs
Build.embed_objs.count += Build.res_dir_parse.res_dirs_count;
if (Build.gpu_parse.shader_entries_count > 0)
{
Build.embed_objs.count += 1;
}
Build.embed_objs.array = PushStructs(perm, EmbedObj, Build.embed_objs.count);
}
}
WaveSync(lane);
//////////////////////////////
//- Phase 2/3: Compile
{
u64 task_idx = 0;
u32 embed_idx = 0;
//- Compile C
{
if (lane->idx == WaveLaneIdxFromTaskIdx(lane, task_idx++) && GetBuildStatus() == 0)
{
Build.c_obj.obj_file = StringF(perm, "%F_gen_c.obj", FmtString(cmdline.leaf_layer_name));
String cmd = StringF(
perm,
"cl.exe /c %F -Fo:%F %F %F %F %F",
FmtString(c_out_file),
FmtString(Build.c_obj.obj_file),
FmtString(StringFromList(perm, cp.flags_msvc, Lit(" "))),
FmtString(StringFromList(perm, cp.compiler_only_flags_msvc, Lit(" "))),
FmtString(StringFromList(perm, cp.warnings_msvc, Lit(" "))),
FmtString(StringFromList(perm, cp.defs, Lit(" ")))
);
OS_CommandResult cmd_result = OS_RunCommand(perm, cmd);
String cmd_output = TrimWhitespace(cmd_result.output);
Build.c_obj.output = cmd_output;
Build.c_obj.return_code = cmd_result.code;
// Ignore MSVC file-name echo
if (MatchString(TrimWhitespace(Build.c_obj.output), F_GetFileName(c_out_file)))
{
Build.c_obj.output = Zstr;
}
SetBuildStatus(Build.c_obj.return_code);
}
}
//- Compile shaders
{
u32 gpu_obj_idx = 0;
for (ShaderEntry *e = Build.gpu_parse.first_shader_entry; e; e = e->next)
{
if (lane->idx == WaveLaneIdxFromTaskIdx(lane, task_idx++) && GetBuildStatus() == 0)
{
String shader_name = e->name;
GpuObj *gpu_obj = &Build.gpu_objs.array[gpu_obj_idx];
String out_file = StringF(perm, "%F/%F", FmtString(shader_store_name), FmtString(e->name));
String target = e->kind == ShaderEntryKind_VS ? Lit("vs_6_6")
: e->kind == ShaderEntryKind_PS ? Lit("ps_6_6")
: e->kind == ShaderEntryKind_CS ? Lit("cs_6_6")
: Lit("vs_6_6");
String compile_cmd = StringF(
perm,
"dxc.exe -T %F -E %F -Fo %F %F %F %F",
FmtString(target),
FmtString(e->name),
FmtString(out_file),
FmtString(gpu_out_file),
FmtString(StringFromList(perm, cp.defs, Lit(" "))),
FmtString(StringFromList(perm, cp.flags_dxc, Lit(" ")))
);
OS_CommandResult cmd_result = OS_RunCommand(perm, compile_cmd);
gpu_obj->output = cmd_result.output;
gpu_obj->return_code = cmd_result.code;
SetBuildStatus(gpu_obj->return_code);
// Final shader compilation lane embeds shader archive
u32 finished_count = Atomic32FetchAdd(&Build.gpu_objs.finished_count, 1) + 1;
if (finished_count == Build.gpu_objs.count && GetBuildStatus() == 0)
{
EmbedObj *embed = &Build.embed_objs.array[embed_idx];
*embed = Embed(shader_store_name, shader_store_name);
}
}
++gpu_obj_idx;
}
if (Build.gpu_parse.shader_entries_count > 0)
{
++embed_idx;
}
}
//- Embed resource dirs
{
for (ResDir *rd = Build.res_dir_parse.first_res_dir; rd; rd = rd->next)
{
if (lane->idx == WaveLaneIdxFromTaskIdx(lane, task_idx++) && GetBuildStatus() == 0)
{
String dir_path = rd->dir_path;
String store_name = rd->store_name;
EmbedObj *embed = &Build.embed_objs.array[embed_idx];
*embed = Embed(store_name, dir_path);
}
++embed_idx;
}
}
}
WaveSync(lane);
//////////////////////////////
//- Phase 3/3: Link
if (lane->idx == 0 && GetBuildStatus() == 0)
{
String exe_file = StringF(perm, "%F.exe", FmtString(cmdline.leaf_layer_name));
{
// Wait for exe to become writable (wait for program to close)
f32 timeout = 1.0;
OS_File file = OS_OpenFile(exe_file, OS_FileFlag_Write, NsFromSeconds(timeout));
OS_CloseFile(file);
}
i64 start_ns = TimeNs();
String obj_files_str = Zi;
{
StringList obj_files = Zi;
PushStringToList(perm, &obj_files, Build.c_obj.obj_file);
for (u64 embed_obj_idx = 0; embed_obj_idx < Build.embed_objs.count; ++embed_obj_idx)
{
EmbedObj *embed_obj = &Build.embed_objs.array[embed_obj_idx];
PushStringToList(perm, &obj_files, embed_obj->obj_file);
}
obj_files_str = StringFromList(perm, obj_files, Lit(" "));
}
String cmd = StringF(
perm,
"link.exe %F /OUT:%F %F %F",
FmtString(obj_files_str),
FmtString(exe_file),
FmtString(StringFromList(perm, cp.flags_msvc, Lit(" "))),
FmtString(StringFromList(perm, cp.linker_only_flags_msvc, Lit(" ")))
);
OS_CommandResult result = OS_RunCommand(perm, cmd);
Build.link.output = TrimWhitespace(result.output);
Build.link.return_code = result.code;
i64 link_elapsed_ns = TimeNs() - start_ns;
// EchoLine(StringF(perm, ">>>>> Linked in %Fs", FmtFloat(SecondsFromNs(link_elapsed_ns))));
SetBuildStatus(Build.link.return_code);
}
//////////////////////////////
//- Display build results
if (lane->idx == 0)
{
String gpu_obj_output = Zi;
{
GpuObj *disp_obj = 0;
for (u32 gpu_obj_idx = 0; gpu_obj_idx < Build.gpu_objs.count; ++gpu_obj_idx)
{
GpuObj *gpu_obj = &Build.gpu_objs.array[gpu_obj_idx];
if (!disp_obj || TrimWhitespace(disp_obj->output).len == 0 || gpu_obj->return_code != 0)
{
disp_obj = gpu_obj;
}
}
if (disp_obj)
{
gpu_obj_output = TrimWhitespace(disp_obj->output);
}
}
String embed_obj_output = Zi;
{
StringList embed_obj_outputs = Zi;
for (u32 embed_obj_idx = 0; embed_obj_idx < Build.embed_objs.count; ++embed_obj_idx)
{
EmbedObj *embed_obj = &Build.embed_objs.array[embed_obj_idx];
PushStringToList(perm, &embed_obj_outputs, embed_obj->output);
}
embed_obj_output = StringFromList(perm, embed_obj_outputs, Lit("\n"));
}
EchoLineOrNothing(StringFromMetaErrors(perm, Build.layers_parse.errors));
EchoLineOrNothing(StringFromMetaErrors(perm, Build.c_parse.errors));
EchoLineOrNothing(StringFromMetaErrors(perm, Build.gpu_parse.errors));
EchoLineOrNothing(Build.c_obj.output);
EchoLineOrNothing(gpu_obj_output);
EchoLineOrNothing(StringFromMetaErrors(perm, Build.res_dir_parse.errors));
EchoLineOrNothing(embed_obj_output);
EchoLineOrNothing(Build.link.output);
}
//////////////////////////////
//- Exit
if (lane->idx == 0)
{
EchoLine(StringF(perm, "Runtime: %Fs", FmtFloat(SecondsFromNs(TimeNs()))));
ExitNow(GetBuildStatus());
}
}
////////////////////////////////////////////////////////////
//~ @hookimpl Bootstrap layers
void BootstrapLayers(void)
{
OS_Bootstrap();
CpuTopologyInfo cpu_info = GetCpuTopologyInfo();
i32 meta_lanes_count = cpu_info.num_logical_cores - 1;
DispatchWave(Lit("Meta"), MaxI32(meta_lanes_count, 1), BuildEntryPoint, 0);
}