mirrors/RustPython-Parser
1
0
Fork 0
mirror of https://github.com/RustPython/Parser.git synced 2025-08-27 22:04:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add an ast optimizer

Browse source
This commit is contained in:
Noah 2021-04-04 18:03:19 -05:00
parent 9941eef853
commit a9330765fa
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
porcelain/src/lib.rs
View file

@ -1,6 +1,7 @@
use rustpython_bytecode::CodeObject; use rustpython_bytecode::CodeObject;
use rustpython_compiler_core::{compile, symboltable}; use rustpython_compiler_core::{compile, symboltable};
use rustpython_parser::{ast::Location, parser}; use rustpython_parser::ast::{fold::Fold, ConstantOptimizer, Location};
use rustpython_parser::parser;
use std::fmt; use std::fmt;
pub use compile::{CompileOpts, Mode}; pub use compile::{CompileOpts, Mode};
@ -80,10 +81,15 @@ pub fn compile(
compile::Mode::Eval => parser::Mode::Expression, compile::Mode::Eval => parser::Mode::Expression,
compile::Mode::Single => parser::Mode::Interactive, compile::Mode::Single => parser::Mode::Interactive,
}; };
let ast = match parser::parse(source, mode) { let mut ast = match parser::parse(source, mode) {
Ok(x) => x, Ok(x) => x,
Err(e) => return Err(CompileError::from_parse(e, source, source_path)), Err(e) => return Err(CompileError::from_parse(e, source, source_path)),
}; };
if opts.optimize > 0 {
ast = ConstantOptimizer::new()
.fold_mod(ast)
.unwrap_or_else(|e| match e {});
}
compile::compile_top(&ast, source_path, opts).map_err(|e| CompileError::from_compile(e, source)) compile::compile_top(&ast, source_path, opts).map_err(|e| CompileError::from_compile(e, source))
} }