azur/bobbylisp
1
1
Fork 0
mirror of https://github.com/azur1s/bobbylisp.git synced 2024-10-16 02:37:40 -05:00
Code Issues Projects Releases Wiki Activity

refactor: buffered io i think

Browse source
This commit is contained in:
Natapat Samutpong 2022-02-06 14:24:08 +07:00
parent d0555cda52
commit 4ce5e43c78
1 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
blspc/src/main.rs
View file

@ -1,4 +1,4 @@
use std::{fs::{read_to_string, File}, path::{Path, PathBuf}, io::{Write, Seek}, time::Instant, process::exit}; use std::{fs::{read_to_string, File}, path::{Path, PathBuf}, io::{Write, BufWriter}, time::Instant, process::exit};
use structopt::StructOpt; use structopt::StructOpt;
@ -15,13 +15,12 @@ mod vm;
use vm::{vm::VM, parser::parse_instr}; use vm::{vm::VM, parser::parse_instr};
fn main() { fn main() {
let start = Instant::now();
let args = Opts::from_args(); let args = Opts::from_args();
match args.commands { match args.commands {
args::Args::Compile(args) => { args::Args::Compile(args) => {
let src = read_to_string(&args.file).unwrap(); let src = read_to_string(&args.file).unwrap();
let debug = args.debug; let debug = args.debug;
compile_src(src, args.output, args.file, debug, start); compile_src(src, args.output, args.file, debug);
}, },
args::Args::Run(args) => { args::Args::Run(args) => {
let src = read_to_string(&args.file).unwrap(); let src = read_to_string(&args.file).unwrap();
@ -31,12 +30,13 @@ fn main() {
} }
} }
fn compile_src(src: String, path: Option<PathBuf>, file: PathBuf, debug: bool, start: Instant) { fn compile_src(src: String, path: Option<PathBuf>, file: PathBuf, debug: bool) {
let file_name = match path { let file_name = match path {
Some(path) => path, Some(path) => path,
None => Path::new(&file).to_path_buf(), None => Path::new(&file).to_path_buf(),
}.file_stem().unwrap().to_str().unwrap().to_string(); }.file_stem().unwrap().to_str().unwrap().to_string();
let start = Instant::now();
let tokens = tokenize(&cover_paren(src)); let tokens = tokenize(&cover_paren(src));
let mut parser = Parser::new(tokens.clone()); let mut parser = Parser::new(tokens.clone());
let result = parser.parse(); let result = parser.parse();
@ -48,12 +48,12 @@ fn compile_src(src: String, path: Option<PathBuf>, file: PathBuf, debug: bool, s
let code = compiler.compile(ast); let code = compiler.compile(ast);
match code { match code {
Ok(code) => { Ok(code) => {
let mut file = File::create(format!("{}.bsm", file_name)).unwrap(); let file = File::create(format!("{}.bsm", file_name)).unwrap();
let mut buffer = BufWriter::new(file);
for line in code { for line in code {
if line.to_string().starts_with(";") { continue; } writeln!(buffer, "{}", line).unwrap();
write!(file, "{}\n", line).unwrap();
} }
file.seek(std::io::SeekFrom::End(-1)).unwrap(); // Trim last newline buffer.flush().unwrap();
let elapsed = start.elapsed(); let elapsed = start.elapsed();
println!("Compiled in {}.{}s", elapsed.as_secs(), elapsed.subsec_millis()); println!("Compiled in {}.{}s", elapsed.as_secs(), elapsed.subsec_millis());