kmc7468/cs420
1
0
Fork 0
mirror of https://github.com/kmc7468/cs420.git synced 2025-12-16 15:38:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update IR visualizer

Browse Source
This commit is contained in:
Minseong Jang
2022-01-27 15:21:56 +09:00
parent 99d0ff1311
commit fd1043e09b
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/ir/visualize.rs
View File

@@ -48,9 +48,8 @@ impl Translate<TranslationUnit> for Visualizer {
for (bid, block) in &definition.blocks { for (bid, block) in &definition.blocks {
for (iid, instruction) in block.instructions.iter().enumerate() { for (iid, instruction) in block.instructions.iter().enumerate() {
if let Instruction::Call { callee, .. } = &instruction.inner { if let Instruction::Call { callee, .. } = &instruction.inner {
let callee = self.translate_callee(callee)?;
let from = self.translate_instruction_node(name, *bid, iid); let from = self.translate_instruction_node(name, *bid, iid);
let to = self.get_function_first_instruction(&callee); let to = self.translate_callee(name, callee)?;
edges.push(format!("{} -> {};", from, to)); edges.push(format!("{} -> {};", from, to));
} }
@@ -93,11 +92,15 @@ impl Visualizer {
} }
#[inline] #[inline]
fn translate_callee(&mut self, callee: &Operand) -> Result<String, ()> { fn translate_callee(&mut self, name: &str, callee: &Operand) -> Result<String, ()> {
match callee { match callee {
Operand::Constant(_constant @ Constant::GlobalVariable { name, .. }) => { Operand::Constant(_constant @ Constant::GlobalVariable { name, .. }) => {
Ok(name.clone()) Ok(self.get_function_first_instruction(name))
} }
Operand::Register {
rid: _rid @ RegisterId::Temp { bid, iid },
..
} => Ok(self.translate_instruction_node(name, *bid, *iid)),
_ => todo!("translate_callee: operand {:?}", callee), _ => todo!("translate_callee: operand {:?}", callee),
} }
} }