huia/huia-compiler/src/block.rs

use crate::ir::IR;
use crate::stable::StringIdx;
use crate::ty::TyIdx;
use std::slice::IterMut;

#[derive(Debug, Default, PartialEq, Clone)]
pub struct Block {
    arguments: Vec<(StringIdx, TyIdx)>,
    ir: Vec<IR>,
}

impl Block {
    /// Consumes the block and returns the IR data.
    pub fn ir(self) -> Vec<IR> {
        self.ir
    }

    pub fn is_empty(&self) -> bool {
        self.ir.is_empty()
    }

    pub fn iter_mut(&mut self) -> IterMut<IR> {
        self.ir.iter_mut()
    }

    pub fn new(arguments: Vec<(StringIdx, TyIdx)>) -> Block {
        Block {
            arguments,
            ir: Vec::default(),
        }
    }

    pub fn pop(&mut self) -> Option<IR> {
        self.ir.pop()
    }

    pub fn push(&mut self, ir: IR) {
        self.ir.push(ir);
    }

    pub fn set_arguments(&mut self, arguments: Vec<(StringIdx, TyIdx)>) {
        self.arguments = arguments;
    }
}

#[derive(Debug, Clone, PartialEq)]
pub struct BlockIdx(usize);

impl From<usize> for BlockIdx {
    fn from(i: usize) -> BlockIdx {
        BlockIdx(i)
    }
}

impl From<BlockIdx> for usize {
    fn from(i: BlockIdx) -> usize {
        i.0
    }
}

impl From<&BlockIdx> for usize {
    fn from(i: &BlockIdx) -> usize {
        i.0
    }
}