sway-core/src/asm_lang/virtual_immediate.rs - master - githubmirror/sway

sway-core/src/asm_lang/virtual_immediate.rs
209 строк5 KB
Daniel Frederico Lins Leite
Improve `asm` const propagation (#7550)
12 мар 2026, 15:39
12 мар 2026, 15:39645d811
use crate::asm_generation::fuel::compiler_constants;

use fuel_vm::fuel_asm::Imm18;
use sway_error::error::CompileError;
use sway_types::span::Span;

use std::convert::TryInto;
use std::fmt;

#[repr(u8)]
pub enum WideOperations {
    Add = 0,
    Sub = 1,
    Not = 2,
    Or = 3,
    Xor = 4,
    And = 5,
    Lsh = 6,
    Rsh = 7,
}

#[repr(u8)]
pub enum WideCmp {
    Equality = 0,
    LessThan = 2,
    GreaterThan = 3,
}

/// 6-bit immediate value type
#[derive(Clone, Debug)]
pub struct VirtualImmediate06 {
    value: u8,
}

impl VirtualImmediate06 {
    pub(crate) fn try_new(raw: u64, err_msg_span: Span) -> Result<Self, CompileError> {
        if raw > compiler_constants::SIX_BITS {
            Err(CompileError::Immediate06TooLarge {
                val: raw,
                span: err_msg_span,
            })
        } else {
            Ok(Self {
                value: raw.try_into().unwrap(),
            })
        }
    }

    pub(crate) fn new(raw: u64) -> Self {
        Self::try_new(raw, Span::dummy()).unwrap_or_else(|_| panic!("{} cannot fit in 6 bits", raw))
    }

    pub fn wide_op(op: WideOperations, rhs_indirect: bool) -> VirtualImmediate06 {
        VirtualImmediate06 {
            value: (op as u8) | if rhs_indirect { 32u8 } else { 0 },
        }
    }

    pub fn wide_cmp(op: WideCmp, rhs_indirect: bool) -> VirtualImmediate06 {
        VirtualImmediate06 {
            value: (op as u8) | if rhs_indirect { 32u8 } else { 0 },
        }
    }

    pub fn wide_mul(lhs_indirect: bool, rhs_indirect: bool) -> VirtualImmediate06 {
        let lhs = if lhs_indirect { 16u8 } else { 0 };
        let rhs = if rhs_indirect { 32u8 } else { 0 };
        VirtualImmediate06 { value: lhs | rhs }
    }

    pub fn wide_div(rhs_indirect: bool) -> VirtualImmediate06 {
        let rhs = if rhs_indirect { 32u8 } else { 0 };
        VirtualImmediate06 { value: rhs }
    }

    pub fn value(&self) -> u8 {
        self.value
    }
}

impl fmt::Display for VirtualImmediate06 {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "i{}", self.value)
    }
}

/// 12-bits immediate value type
#[derive(Clone, Debug)]
pub struct VirtualImmediate12 {
    value: u16,
}

impl VirtualImmediate12 {
    pub(crate) fn try_new(raw: u64, err_msg_span: Span) -> Result<Self, CompileError> {
        if raw > compiler_constants::TWELVE_BITS {
            Err(CompileError::Immediate12TooLarge {
                val: raw,
                span: err_msg_span,
            })
        } else {
            Ok(Self {
                value: raw.try_into().unwrap(),
            })
        }
    }

    pub(crate) fn new(raw: u64) -> Self {
        Self::try_new(raw, Span::dummy())
            .unwrap_or_else(|_| panic!("{} cannot fit in 12 bits", raw))
    }

    pub fn value(&self) -> u16 {
        self.value
    }
}

impl fmt::Display for VirtualImmediate12 {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "i{}", self.value)
    }
}

impl TryInto<VirtualImmediate12> for u64 {
    type Error = CompileError;

    fn try_into(self) -> Result<VirtualImmediate12, Self::Error> {
        VirtualImmediate12::try_new(self, Span::dummy())
    }
}

/// 18-bits immediate value type
#[derive(Clone, Debug)]
pub struct VirtualImmediate18 {
    value: u32,
}

impl From<u8> for VirtualImmediate18 {
    fn from(value: u8) -> Self {
        VirtualImmediate18::new(value as u64)
    }
}

impl VirtualImmediate18 {
    pub(crate) fn try_new(raw: u64, err_msg_span: Span) -> Result<Self, CompileError> {
        if raw > compiler_constants::EIGHTEEN_BITS {
            Err(CompileError::Immediate18TooLarge {
                val: raw,
                span: err_msg_span,
            })
        } else {
            Ok(Self {
                value: raw.try_into().unwrap(),
            })
        }
    }

    pub(crate) fn new(raw: u64) -> Self {
        Self::try_new(raw, Span::dummy())
            .unwrap_or_else(|_| panic!("{} cannot fit in 18 bits", raw))
    }

    pub fn value(&self) -> u32 {
        self.value
    }

    pub fn as_imm18(&self) -> Option<Imm18> {
        Imm18::new_checked(self.value)
    }
}

impl fmt::Display for VirtualImmediate18 {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "i{}", self.value)
    }
}

/// 24-bits immediate value type
#[derive(Clone, Debug)]
pub struct VirtualImmediate24 {
    value: u32,
}
impl VirtualImmediate24 {
    pub(crate) fn try_new(raw: u64, err_msg_span: Span) -> Result<Self, CompileError> {
        if raw > compiler_constants::TWENTY_FOUR_BITS {
            Err(CompileError::Immediate24TooLarge {
                val: raw,
                span: err_msg_span,
            })
        } else {
            Ok(Self {
                value: raw.try_into().unwrap(),
            })
        }
    }

    pub(crate) fn new(raw: u64) -> Self {
        Self::try_new(raw, Span::dummy())
            .unwrap_or_else(|_| panic!("{} cannot fit in 24 bits", raw))
    }

    pub fn value(&self) -> u32 {
        self.value
    }
}
impl fmt::Display for VirtualImmediate24 {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "i{}", self.value)
    }
}