devraza/tinypc
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: remove ~duplicate function branch

Browse source
This commit is contained in:
Muhammad Nauman Raza 2025-01-29 19:55:50 +00:00
parent f7fdda37f4
commit f5eb252266
Signed by: devraza
GPG key ID: 91EAD6081011574B
1 changed files with 9 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
src/main.rs
View file

@ -2,19 +2,11 @@ use std::collections::HashMap;
use std::fs::File;
use std::io::{self, BufRead, BufReader};
fn retrieve(memory: &HashMap<String, i64>, ops: &[String]) -> i64 {
fn retrieve(labels: &HashMap<String, usize>, ops: &[String]) -> i64 {
if ops.len() >= 3 {
*memory.get(&ops[2]).unwrap_or(&0)
*labels.get(&ops[2]).unwrap_or(&0) as i64
} else {
*memory.get(&ops[1]).unwrap_or(&0)
}
}
fn branch(labels: &HashMap<String, usize>, ops: &[String]) -> usize {
if ops.len() >= 3 {
*labels.get(&ops[2]).unwrap_or(&0)
} else {
*labels.get(&ops[1]).unwrap_or(&0)
*labels.get(&ops[1]).unwrap_or(&0) as i64
}
}
@ -39,31 +31,31 @@ fn process(
memory.insert(key.clone(), *accumulator);
}
Some("LDA") => {
*accumulator = retrieve(memory, &ops);
*accumulator = retrieve(labels, &ops);
}
Some("ADD") => {
if let Ok(value) = ops[1].parse::<i64>() {
*accumulator += value;
} else {
*accumulator += retrieve(memory, &ops);
*accumulator += retrieve(labels, &ops);
}
}
Some("SUB") => {
if let Ok(value) = ops[1].parse::<i64>() {
*accumulator -= value;
} else {
*accumulator -= retrieve(memory, &ops);
*accumulator -= retrieve(labels, &ops);
}
}
Some("BRA") => return branch(labels, &ops),
Some("BRA") => return retrieve(labels, &ops) as usize,
Some("BRP") => {
if *accumulator >= 0 {
return branch(labels, &ops);
return retrieve(labels, &ops) as usize;
}
}
Some("BRZ") => {
if *accumulator == 0 {
return branch(labels, &ops);
return retrieve(labels, &ops) as usize;
}
}
Some("HLT") => std::process::exit(0),