1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

use std::fmt::Display;

use crate::ir::{quantity::Quantity, RegisterName};

use super::IsAction;
use crate::ir::function::statement::IsIRStatement;

#[derive(Debug, Clone)]
pub struct RenameLocal {
    pub from: RegisterName,
    pub to: Quantity,
}

impl RenameLocal {
    pub fn new(from: RegisterName, to: impl Into<Quantity>) -> Self {
        Self {
            from,
            to: to.into(),
        }
    }
}

impl Display for RenameLocal {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "rename `{}` to `{}`", self.from, self.to)
    }
}

impl IsAction for RenameLocal {
    fn perform_on_function(self, ir: &mut crate::ir::FunctionDefinition) {
        ir.iter_mut().for_each(|statement| {
            statement.on_register_change(&self.from, self.to.clone());
        });
    }
}