358 lines
8.9 KiB
Rust
358 lines
8.9 KiB
Rust
|
extern crate rand;
|
||
|
|
|
||
|
|
use rand::{distributions::Alphanumeric, Rng};
|
||
|
|
use std::collections::HashMap;
|
||
|
|
|
||
|
|
#[derive(Debug, Clone)]
|
||
|
|
pub struct Function {
|
||
|
|
pub name: String,
|
||
|
|
pub statements: Vec<Statement>,
|
||
|
|
assembly: Option<String>,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Clone)]
|
||
|
|
pub enum Statement {
|
||
|
|
FunctionCall(String, Vec<ValueNode>),
|
||
|
|
VariableDeclaration(String, ValueNode),
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Clone)]
|
||
|
|
pub enum ValueNode {
|
||
|
|
Literal(Literal),
|
||
|
|
Operator(Operator),
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Clone)]
|
||
|
|
pub enum Literal {
|
||
|
|
Int(i64),
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Clone)]
|
||
|
|
pub enum Operator {
|
||
|
|
Add(Box<ValueNode>, Box<ValueNode>),
|
||
|
|
Sub(Box<ValueNode>, Box<ValueNode>),
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Clone)]
|
||
|
|
pub struct AssemblerState {
|
||
|
|
pub functions: HashMap<String, Function>,
|
||
|
|
}
|
||
|
|
|
||
|
|
pub trait Node {
|
||
|
|
fn to_assembly(&self, state: &mut AssemblerState) -> String;
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Function {
|
||
|
|
pub fn new(name: String, statements: Vec<Statement>) -> Function {
|
||
|
|
Function {
|
||
|
|
name,
|
||
|
|
statements,
|
||
|
|
assembly: None,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn add(&self, state: &mut AssemblerState) -> Function {
|
||
|
|
state.functions.insert(self.name.clone(), self.to_owned());
|
||
|
|
self.to_owned()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl AssemblerState {
|
||
|
|
pub fn new() -> AssemblerState {
|
||
|
|
AssemblerState {
|
||
|
|
functions: get_default_functions(),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fn get_default_functions() -> HashMap<String, Function> {
|
||
|
|
let mut functions = HashMap::new();
|
||
|
|
functions.insert(
|
||
|
|
"out".to_string(),
|
||
|
|
Function {
|
||
|
|
name: "exit".to_string(),
|
||
|
|
statements: vec![],
|
||
|
|
assembly: Some(
|
||
|
|
"
|
||
|
|
out:
|
||
|
|
;
|
||
|
|
; Pop the argument of the stack
|
||
|
|
;
|
||
|
|
|
||
|
|
SE-OUT RAM-ADDR-U-IN 0xFF ; Set ram address to stack end
|
||
|
|
RAM-OUT OUTPUT-IN 0xFF ; Output top address onto output register
|
||
|
|
|
||
|
|
ROM-OUT FLAGS-IN 0x80 ; Subtract mode
|
||
|
|
SE-OUT A-IN 0xFF ; Fetch SE
|
||
|
|
ROM-OUT B-IN 0x01 ; Subtract one
|
||
|
|
ALU-OUT SE-IN 0xFF ; Store in SE
|
||
|
|
ROM-OUT FLAGS-IN 0x00 ; Add mode
|
||
|
|
|
||
|
|
;
|
||
|
|
; Pop the return address of the stack
|
||
|
|
;
|
||
|
|
|
||
|
|
SE-OUT RAM-ADDR-U-IN 0xFF ; Set ram address to stack end
|
||
|
|
RAM-OUT ROM-ADDR-U-IN 0xFF ; Store top address in ROM jump address
|
||
|
|
|
||
|
|
ROM-OUT FLAGS-IN 0x80 ; Subtract mode
|
||
|
|
SE-OUT A-IN 0xFF ; Fetch SE
|
||
|
|
ROM-OUT B-IN 0x01 ; Subtract one
|
||
|
|
ALU-OUT SE-IN 0xFF ; Store in SE
|
||
|
|
ROM-OUT FLAGS-IN 0x00 ; Add mode
|
||
|
|
|
||
|
|
F-OUT JMP 0xFF"
|
||
|
|
.to_string(),
|
||
|
|
),
|
||
|
|
},
|
||
|
|
);
|
||
|
|
functions.insert(
|
||
|
|
"exit".to_string(),
|
||
|
|
Function {
|
||
|
|
name: "exit".to_string(),
|
||
|
|
statements: vec![],
|
||
|
|
assembly: Some(
|
||
|
|
"
|
||
|
|
exit:
|
||
|
|
F-OUT JMP 0xFF"
|
||
|
|
.to_string(),
|
||
|
|
),
|
||
|
|
},
|
||
|
|
);
|
||
|
|
functions
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Node for Function {
|
||
|
|
fn to_assembly(&self, state: &mut AssemblerState) -> String {
|
||
|
|
match &self.assembly {
|
||
|
|
Some(v) => v.to_string(),
|
||
|
|
None => format!(
|
||
|
|
"{}:\n{}
|
||
|
|
;
|
||
|
|
; Pop the return address of the stack
|
||
|
|
;
|
||
|
|
|
||
|
|
SE-OUT RAM-ADDR-U-IN 0xFF ; Set ram address to stack end
|
||
|
|
RAM-OUT ROM-ADDR-U-IN 0xFF ; Store top address in ROM jump address
|
||
|
|
|
||
|
|
ROM-OUT FLAGS-IN 0x80 ; Subtract mode
|
||
|
|
SE-OUT A-IN 0xFF ; Fetch SE
|
||
|
|
ROM-OUT B-IN 0x01 ; Subtract one
|
||
|
|
ALU-OUT SE-IN 0xFF ; Store in SE
|
||
|
|
ROM-OUT FLAGS-IN 0x00 ; Add mode
|
||
|
|
|
||
|
|
F-OUT JMP 0xFF
|
||
|
|
",
|
||
|
|
self.name,
|
||
|
|
{
|
||
|
|
self.statements
|
||
|
|
.iter()
|
||
|
|
.map(|x| x.to_assembly(state))
|
||
|
|
.collect::<Vec<String>>()
|
||
|
|
.join("\n")
|
||
|
|
}
|
||
|
|
),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Node for Statement {
|
||
|
|
fn to_assembly(&self, state: &mut AssemblerState) -> String {
|
||
|
|
match self {
|
||
|
|
Statement::FunctionCall(name, arguments) => {
|
||
|
|
if !state.functions.contains_key(name) {
|
||
|
|
todo!()
|
||
|
|
}
|
||
|
|
|
||
|
|
let argument_assembly: String = arguments
|
||
|
|
.into_iter()
|
||
|
|
.map(|x| x.to_assembly(state))
|
||
|
|
.collect::<Vec<String>>()
|
||
|
|
.join("\n");
|
||
|
|
|
||
|
|
let call_id: String = rand::thread_rng()
|
||
|
|
.sample_iter(&Alphanumeric)
|
||
|
|
.take(30)
|
||
|
|
.map(char::from)
|
||
|
|
.collect();
|
||
|
|
|
||
|
|
format!(
|
||
|
|
"
|
||
|
|
;
|
||
|
|
; Stack
|
||
|
|
;
|
||
|
|
|
||
|
|
; Increment Stack Pointer
|
||
|
|
|
||
|
|
SE-OUT A-IN 0xFF ; Fetch SE
|
||
|
|
ROM-OUT B-IN 0x01 ; Add 1
|
||
|
|
ALU-OUT SE-IN 0xFF ; Store
|
||
|
|
|
||
|
|
; Store in top stack pointer
|
||
|
|
|
||
|
|
ALU-OUT RAM-ADDR-U-IN 0xFF ; Store in RAM address
|
||
|
|
ROM-OUT RAM-IN ${name}-{call_id} ; Store address after jump in ram
|
||
|
|
|
||
|
|
;
|
||
|
|
; Arguments
|
||
|
|
;
|
||
|
|
|
||
|
|
{argument_assembly}
|
||
|
|
|
||
|
|
;
|
||
|
|
; Jump
|
||
|
|
;
|
||
|
|
ROM-OUT ROM-ADDR-U-IN ${name} ; Jump to function label
|
||
|
|
F-OUT JMP 0xFF ; Jump to function
|
||
|
|
|
||
|
|
{name}-{call_id}:
|
||
|
|
"
|
||
|
|
)
|
||
|
|
}
|
||
|
|
Statement::VariableDeclaration(_name, _value) => {
|
||
|
|
todo!()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Node for ValueNode {
|
||
|
|
fn to_assembly(&self, state: &mut AssemblerState) -> String {
|
||
|
|
match self {
|
||
|
|
ValueNode::Literal(v) => v.to_assembly(state),
|
||
|
|
ValueNode::Operator(o) => o.to_assembly(state),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Node for Literal {
|
||
|
|
fn to_assembly(&self, _state: &mut AssemblerState) -> String {
|
||
|
|
match self {
|
||
|
|
Literal::Int(v) => {
|
||
|
|
format!(
|
||
|
|
"
|
||
|
|
;
|
||
|
|
; Stack
|
||
|
|
;
|
||
|
|
|
||
|
|
; Increment Stack Pointer
|
||
|
|
|
||
|
|
SE-OUT A-IN 0xFF ; Fetch SE
|
||
|
|
ROM-OUT B-IN 0x01 ; Add 1
|
||
|
|
ALU-OUT SE-IN 0xFF ; Store
|
||
|
|
|
||
|
|
; Store in top stack pointer
|
||
|
|
|
||
|
|
ALU-OUT RAM-ADDR-U-IN 0xFF ; Store in RAM address
|
||
|
|
ROM-OUT RAM-IN {v}; Store literal value in RAM"
|
||
|
|
)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Node for Operator {
|
||
|
|
fn to_assembly(&self, state: &mut AssemblerState) -> String {
|
||
|
|
match self {
|
||
|
|
Operator::Add(a, b) => {
|
||
|
|
format!(
|
||
|
|
"
|
||
|
|
{}
|
||
|
|
{}
|
||
|
|
|
||
|
|
SE-OUT RAM-ADDR-U-IN 0xFF ; Set ram address to stack end
|
||
|
|
RAM-OUT C-IN 0xFF ; Output top address onto output register
|
||
|
|
|
||
|
|
ROM-OUT FLAGS-IN 0x80 ; Subtract mode
|
||
|
|
SE-OUT A-IN 0xFF ; Fetch SE
|
||
|
|
ROM-OUT B-IN 0x01 ; Subtract one
|
||
|
|
ALU-OUT SE-IN 0xFF ; Store in SE
|
||
|
|
|
||
|
|
SE-OUT RAM-ADDR-U-IN 0xFF ; Set ram address to stack end
|
||
|
|
RAM-OUT D-IN 0xFF ; Output top address onto output register
|
||
|
|
|
||
|
|
ROM-OUT FLAGS-IN 0x80 ; Subtract mode
|
||
|
|
SE-OUT A-IN 0xFF ; Fetch SE
|
||
|
|
ROM-OUT B-IN 0x01 ; Subtract one
|
||
|
|
ALU-OUT SE-IN 0xFF ; Store in SE
|
||
|
|
ROM-OUT FLAGS-IN 0x00 ; Add mode
|
||
|
|
|
||
|
|
C-OUT A-IN 0xFF ; Move temporary values into a register for alu computation
|
||
|
|
D-OUT B-IN 0xFF ; Move temporary values into a register for alu computation
|
||
|
|
ALU-OUT C-IN 0xFF ; Move ALU output into temporary register
|
||
|
|
|
||
|
|
SE-OUT A-IN 0xFF ; Fetch SE
|
||
|
|
ROM-OUT B-IN 0x01 ; Add 1
|
||
|
|
ALU-OUT SE-IN 0xFF ; Store
|
||
|
|
|
||
|
|
ALU-OUT RAM-ADDR-U-IN 0xFF ; Store in RAM address
|
||
|
|
C-OUT RAM-IN 0xFF ; Store address after jump in ram
|
||
|
|
",
|
||
|
|
a.to_assembly(state),
|
||
|
|
b.to_assembly(state)
|
||
|
|
)
|
||
|
|
}
|
||
|
|
Operator::Sub(a, b) => {
|
||
|
|
format!(
|
||
|
|
"
|
||
|
|
{}
|
||
|
|
{}
|
||
|
|
|
||
|
|
SE-OUT RAM-ADDR-U-IN 0xFF ; Set ram address to stack end
|
||
|
|
RAM-OUT C-IN 0xFF ; Output top address onto output register
|
||
|
|
|
||
|
|
ROM-OUT FLAGS-IN 0x80 ; Subtract mode
|
||
|
|
SE-OUT A-IN 0xFF ; Fetch SE
|
||
|
|
ROM-OUT B-IN 0x01 ; Subtract one
|
||
|
|
ALU-OUT SE-IN 0xFF ; Store in SE
|
||
|
|
|
||
|
|
SE-OUT RAM-ADDR-U-IN 0xFF ; Set ram address to stack end
|
||
|
|
RAM-OUT D-IN 0xFF ; Output top address onto output register
|
||
|
|
|
||
|
|
ROM-OUT FLAGS-IN 0x80 ; Subtract mode
|
||
|
|
SE-OUT A-IN 0xFF ; Fetch SE
|
||
|
|
ROM-OUT B-IN 0x01 ; Subtract one
|
||
|
|
ALU-OUT SE-IN 0xFF ; Store in SE
|
||
|
|
ROM-OUT FLAGS-IN 0x00 ; Add mode
|
||
|
|
|
||
|
|
ROM-OUT FLAGS-IN 0x80 ; Subtract mode
|
||
|
|
C-OUT B-IN 0xFF ; Move temporary values into a register for alu computation
|
||
|
|
D-OUT A-IN 0xFF ; Move temporary values into a register for alu computation
|
||
|
|
ALU-OUT C-IN 0xFF ; Move ALU output into temporary register
|
||
|
|
ROM-OUT FLAGS-IN 0x00 ; Add mode
|
||
|
|
|
||
|
|
SE-OUT A-IN 0xFF ; Fetch SE
|
||
|
|
ROM-OUT B-IN 0x01 ; Add 1
|
||
|
|
ALU-OUT SE-IN 0xFF ; Store
|
||
|
|
|
||
|
|
ALU-OUT RAM-ADDR-U-IN 0xFF ; Store in RAM address
|
||
|
|
C-OUT RAM-IN 0xFF ; Store address after jump in ram
|
||
|
|
",
|
||
|
|
a.to_assembly(state),
|
||
|
|
b.to_assembly(state)
|
||
|
|
)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn to_assembly(functions: Vec<Function>) -> String {
|
||
|
|
let mut final_product: String = "".to_owned();
|
||
|
|
let mut std_function_assembly = "".to_owned();
|
||
|
|
let mut state = AssemblerState::new();
|
||
|
|
|
||
|
|
for std_function in state.clone().functions {
|
||
|
|
std_function_assembly += &std_function.1.to_assembly(&mut state);
|
||
|
|
}
|
||
|
|
|
||
|
|
for standard_function in &functions {
|
||
|
|
standard_function.add(&mut state);
|
||
|
|
}
|
||
|
|
|
||
|
|
for standard_function in &functions {
|
||
|
|
final_product += &standard_function.to_assembly(&mut state);
|
||
|
|
}
|
||
|
|
|
||
|
|
final_product + &std_function_assembly
|
||
|
|
}
|