2021-02-03 21:19:11 +00:00
|
|
|
use std::fs;
|
|
|
|
use std::env;
|
2021-02-03 21:48:11 +00:00
|
|
|
use std::fmt;
|
2021-02-03 21:54:13 +00:00
|
|
|
use std::usize;
|
2021-02-06 23:48:24 +00:00
|
|
|
use std::collections::HashMap;
|
2021-02-03 21:48:11 +00:00
|
|
|
|
2021-02-04 19:47:55 +00:00
|
|
|
mod eval;
|
|
|
|
|
2021-02-03 21:48:11 +00:00
|
|
|
struct Program {
|
|
|
|
data: Vec<String>,
|
2021-02-06 23:48:24 +00:00
|
|
|
pc: usize,
|
2021-02-07 00:10:34 +00:00
|
|
|
vars: HashMap<char, char>
|
2021-02-03 21:48:11 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
impl Program {
|
|
|
|
fn from_string(program: String) -> Program {
|
|
|
|
let mut op_list: Vec<String> = Vec::new();
|
|
|
|
|
|
|
|
for opcode in program.split(" ").collect::<Vec<&str>>() {
|
|
|
|
let mut new_op = opcode.to_owned();
|
|
|
|
new_op = new_op.replace("\n", "");
|
|
|
|
|
|
|
|
if new_op.len() != 0 {
|
|
|
|
op_list.push(new_op.to_owned());
|
|
|
|
}
|
|
|
|
}
|
2021-02-06 23:48:24 +00:00
|
|
|
|
|
|
|
return Program{ data: op_list, pc: 0, vars: HashMap::new() };
|
|
|
|
}
|
|
|
|
|
|
|
|
fn args_or_vars<'a>(&self, arguments: &'a str) -> String {
|
|
|
|
let mut deconstructed: Vec<char> = arguments.chars().collect();
|
|
|
|
|
|
|
|
for (index, char) in arguments.chars().enumerate() {
|
2021-02-07 00:10:34 +00:00
|
|
|
let value = self.vars.get(&char);
|
|
|
|
|
|
|
|
match value {
|
|
|
|
Some(content) => deconstructed[index] = *content,
|
|
|
|
None => {}
|
2021-02-06 23:48:24 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
deconstructed.into_iter().collect()
|
|
|
|
}
|
|
|
|
|
|
|
|
fn add_var(&mut self, arguments: &str) {
|
2021-02-07 00:10:34 +00:00
|
|
|
let argument_vec: Vec<char> = arguments.chars().collect();
|
2021-02-06 23:48:24 +00:00
|
|
|
|
2021-02-07 00:10:34 +00:00
|
|
|
self.vars.insert(argument_vec[0], argument_vec[1]);
|
2021-02-03 21:48:11 +00:00
|
|
|
}
|
|
|
|
|
2021-02-06 23:48:24 +00:00
|
|
|
fn eval(&mut self, instruction: &String) {
|
2021-02-04 19:47:55 +00:00
|
|
|
let opcode = instruction.chars().collect::<Vec<char>>()[0];
|
2021-02-04 17:45:53 +00:00
|
|
|
let arguments = &instruction[1..];
|
|
|
|
|
|
|
|
match opcode {
|
2021-02-06 23:48:24 +00:00
|
|
|
'p' => println!("{}", self.args_or_vars(arguments)),
|
2021-02-07 00:30:14 +00:00
|
|
|
'a' => println!("{}", eval::do_math(self.args_or_vars(arguments), '+')),
|
|
|
|
's' => println!("{}", eval::do_math(self.args_or_vars(arguments), '-')),
|
|
|
|
'm' => println!("{}", eval::do_math(self.args_or_vars(arguments), '*')),
|
|
|
|
'd' => println!("{}", eval::do_math(self.args_or_vars(arguments), '/')),
|
2021-02-06 23:48:24 +00:00
|
|
|
'l' => self.add_var(arguments),
|
2021-02-04 17:45:53 +00:00
|
|
|
_ => panic!("SyntaxError at opcode {}!", self.pc)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-02-03 21:54:13 +00:00
|
|
|
fn run(&mut self) {
|
2021-02-03 21:48:11 +00:00
|
|
|
println!("{}", self);
|
2021-02-03 21:54:13 +00:00
|
|
|
while self.pc < self.data.len() {
|
2021-02-06 23:48:24 +00:00
|
|
|
let instruction = self.data[self.pc].to_owned();
|
2021-02-03 21:54:13 +00:00
|
|
|
|
2021-02-06 23:48:24 +00:00
|
|
|
self.eval(&instruction);
|
2021-02-03 21:54:13 +00:00
|
|
|
|
|
|
|
self.pc = self.pc + 1;
|
|
|
|
}
|
2021-02-03 21:48:11 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl fmt::Display for Program {
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
|
|
write!(f, "Program ({:?})", self.data)
|
|
|
|
}
|
|
|
|
}
|
2021-02-03 21:19:11 +00:00
|
|
|
|
|
|
|
fn main() {
|
|
|
|
let args: Vec<String> = env::args().collect();
|
|
|
|
if args.len() == 1 {
|
|
|
|
panic!("You must provide an argument!");
|
|
|
|
}
|
|
|
|
|
|
|
|
let filename = &args[1];
|
2021-02-03 21:48:11 +00:00
|
|
|
|
2021-02-03 21:19:11 +00:00
|
|
|
let contents = fs::read_to_string(filename).expect("Something went wrong reading the file");
|
2021-02-03 21:54:13 +00:00
|
|
|
let mut prog = Program::from_string(contents);
|
2021-02-03 21:48:11 +00:00
|
|
|
prog.run();
|
2021-02-03 21:19:11 +00:00
|
|
|
}
|