102 lines
3.5 KiB
Rust
102 lines
3.5 KiB
Rust
mod builtin_cells;
|
|
mod frontend;
|
|
mod package;
|
|
mod parser;
|
|
mod rtlil;
|
|
|
|
use ariadne::Source;
|
|
use std::fs::File;
|
|
use std::io::prelude::*;
|
|
use std::path::PathBuf;
|
|
|
|
use structopt::StructOpt;
|
|
|
|
#[derive(Debug, StructOpt)]
|
|
#[structopt(name = "example", about = "An example of StructOpt usage.")]
|
|
struct Opt {
|
|
/// Input file
|
|
#[structopt(parse(from_os_str))]
|
|
input: PathBuf,
|
|
/// Debug AST
|
|
#[structopt(short)]
|
|
debug: bool,
|
|
/// Output file, stdout if not present
|
|
#[structopt(short, parse(from_os_str))]
|
|
output: Option<PathBuf>,
|
|
}
|
|
|
|
fn main() {
|
|
let opt = Opt::from_args();
|
|
let mut infile = File::open(opt.input).expect("could not open file");
|
|
// let packages = package::PackageRegistry::new();
|
|
// let mut infile = packages
|
|
// .get("builtins")
|
|
// .expect("no package")
|
|
// .open()
|
|
// .expect("could not open file");
|
|
|
|
let mut input = String::new();
|
|
infile
|
|
.read_to_string(&mut input)
|
|
.expect("error reading file");
|
|
let input: &str = input.as_str();
|
|
let input = parser::Span::new(input);
|
|
let lexed = parser::tokens::lex(input).expect("failed to lex");
|
|
let tokens = parser::tokens::TokenSpan::new(&lexed.1);
|
|
let parsed = parser::parse(tokens);
|
|
match parsed {
|
|
Err(nom::Err::Error(err) | nom::Err::Failure(err)) => {
|
|
if opt.debug {
|
|
println!("{err:#?}");
|
|
}
|
|
parser::error::convert_error(input, err)
|
|
.eprint(Source::from(input.fragment()))
|
|
.unwrap();
|
|
}
|
|
Err(_) => (unreachable!()),
|
|
Ok(res) => {
|
|
if opt.debug {
|
|
println!("{:#?}", res);
|
|
}
|
|
let mut frontendcontext = crate::frontend::Context::new();
|
|
let typed = frontendcontext.type_module(res.1);
|
|
if let Ok(module) = typed {
|
|
let mut typed_modules = vec![];
|
|
for block in module.blocks {
|
|
if opt.debug {
|
|
let mut pretty_block = String::new();
|
|
frontendcontext
|
|
.pretty_typed_block(&mut pretty_block, &block)
|
|
.unwrap();
|
|
println!("{}", &pretty_block);
|
|
}
|
|
let typed_inferred = frontendcontext.infer_types(block);
|
|
if opt.debug {
|
|
let mut pretty_block = String::new();
|
|
frontendcontext
|
|
.pretty_typed_block(&mut pretty_block, &typed_inferred)
|
|
.unwrap();
|
|
println!("{}", &pretty_block);
|
|
}
|
|
typed_modules.push(typed_inferred);
|
|
}
|
|
let typed_inferred = typed_modules.first().unwrap();
|
|
// TODO: be able to determine modules that are fully parameterized
|
|
let lowered = frontend::lowering::lower_block(&mut frontendcontext, typed_inferred);
|
|
match lowered {
|
|
Ok(res) => {
|
|
let mut file =
|
|
File::create(opt.output.unwrap_or_else(|| "out.rtlil".into()))
|
|
.expect("could not open file");
|
|
file.write_all(res.as_bytes())
|
|
.expect("failed to write output file");
|
|
}
|
|
Err(err) => eprintln!("{:#?}", err),
|
|
}
|
|
} else {
|
|
eprintln!("{:#?}", typed);
|
|
}
|
|
}
|
|
}
|
|
}
|