forked from AbleOS/ableos
96 lines
2.9 KiB
Rust
96 lines
2.9 KiB
Rust
#![feature(iter_next_chunk)]
|
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
println!("cargo:rerun-if-changed=build.rs");
|
|
println!("cargo:rerun-if-changed=../hbbytecode/instructions.in");
|
|
|
|
let instructions = include_str!("../hbbytecode/instructions.in");
|
|
|
|
let mut generated = String::new();
|
|
use std::fmt::Write;
|
|
|
|
writeln!(&mut generated, "impl crate::codegen::Func {{")?;
|
|
|
|
for line in instructions.lines() {
|
|
let line = line.strip_suffix(";").unwrap();
|
|
let [opcode, name, ty, doc] = line.splitn(4, ',').map(str::trim).next_chunk().unwrap();
|
|
|
|
writeln!(&mut generated, "/// {}", doc.trim_matches('"'))?;
|
|
write!(&mut generated, "pub fn {}(&mut self", name.to_lowercase())?;
|
|
for (i, c) in ty.chars().enumerate() {
|
|
let (name, ty) = match c {
|
|
'N' => continue,
|
|
'R' => ("reg", "u8"),
|
|
'B' => ("imm", "u8"),
|
|
'H' => ("imm", "u16"),
|
|
'W' => ("imm", "u32"),
|
|
'D' => ("imm", "u64"),
|
|
'P' => ("offset", "u32"),
|
|
'O' => ("offset", "u32"),
|
|
'A' => ("addr", "u64"),
|
|
_ => panic!("unknown type: {}", c),
|
|
};
|
|
write!(&mut generated, ", {name}{i}: {ty}")?;
|
|
}
|
|
writeln!(&mut generated, ") {{")?;
|
|
|
|
let mut offset = 1;
|
|
for (i, c) in ty.chars().enumerate() {
|
|
let width = match c {
|
|
'N' => 0,
|
|
'R' => 1,
|
|
'B' => 1,
|
|
'H' => 2,
|
|
'W' => 4,
|
|
'D' => 8,
|
|
'A' => 8,
|
|
'P' => 2,
|
|
'O' => 4,
|
|
_ => panic!("unknown type: {}", c),
|
|
};
|
|
|
|
if matches!(c, 'P' | 'O') {
|
|
writeln!(
|
|
&mut generated,
|
|
" self.offset(offset{i}, {offset}, {width});",
|
|
)?;
|
|
}
|
|
|
|
offset += width;
|
|
}
|
|
|
|
write!(
|
|
&mut generated,
|
|
" self.extend(crate::as_bytes(&crate::Args({opcode}"
|
|
)?;
|
|
for (i, c) in ty.chars().enumerate() {
|
|
let name = match c {
|
|
'N' => continue,
|
|
'R' => "reg",
|
|
'B' | 'H' | 'W' | 'D' => "imm",
|
|
'P' => "0u16",
|
|
'O' => "0u32",
|
|
'A' => "addr",
|
|
_ => panic!("unknown type: {}", c),
|
|
};
|
|
|
|
if matches!(c, 'P' | 'O') {
|
|
write!(&mut generated, ", {name}")?;
|
|
} else {
|
|
write!(&mut generated, ", {name}{i}")?;
|
|
}
|
|
}
|
|
for _ in ty.len() - (ty == "N") as usize..4 {
|
|
write!(&mut generated, ", ()")?;
|
|
}
|
|
writeln!(&mut generated, ")));")?;
|
|
|
|
writeln!(&mut generated, "}}")?;
|
|
}
|
|
|
|
writeln!(&mut generated, "}}")?;
|
|
|
|
std::fs::write("src/instrs.rs", generated)?;
|
|
|
|
Ok(())
|
|
}
|