diff --git a/scripts/codegen/__init__.py b/scripts/codegen/__init__.py index 820392b..4bc6c07 100644 --- a/scripts/codegen/__init__.py +++ b/scripts/codegen/__init__.py @@ -5,10 +5,15 @@ supported_architectures = { } def arch_includes(header, root=""): + from pathlib import Path + root = Path(root) + header = Path(header) + prefix = "if" for arch, define in supported_architectures.items(): + path = root / "arch" / arch / header cog.outl(f"#{prefix} defined({define})") - cog.outl(f"#include <{root}arch/{arch}/{header}>") + cog.outl(f"#include <{path}>") prefix = "elif" cog.outl("#else") cog.outl('#error "Unsupported platform"')