luaur-code-gen 0.1.0

use crate::functions::vm_reg_op::vm_reg_op;
use crate::records::const_prop_state::ConstPropState;
use crate::records::ir_op::IrOp;
use crate::records::register_info::RegisterInfo;

impl ConstPropState {
    pub fn try_get_register_info(&mut self, op: IrOp) -> Option<*mut RegisterInfo> {
        if op.kind() == crate::enums::ir_op_kind::IrOpKind::VmReg {
            let vm_reg = vm_reg_op(op);
            if vm_reg > self.max_reg {
                self.max_reg = vm_reg;
            }
            return Some(&mut self.regs[vm_reg as usize] as *mut RegisterInfo);
        }

        if let Some(link) = self.try_get_reg_link(op) {
            unsafe {
                let reg = (*link).reg;
                let reg_i32 = reg as i32;
                if reg_i32 > self.max_reg {
                    self.max_reg = reg_i32;
                }
                return Some(&mut self.regs[reg as usize] as *mut RegisterInfo);
            }
        }

        None
    }
}