Skip to main content

kasl_ir/block/
block_data.rs

1//
2//  Copyright 2026 Shuntaro Kasatani
3//
4//  Licensed under the Apache License, Version 2.0 (the "License");
5//  you may not use this file except in compliance with the License.
6//  You may obtain a copy of the License at
7//
8//      http://www.apache.org/licenses/LICENSE-2.0
9//
10//  Unless required by applicable law or agreed to in writing, software
11//  distributed under the License is distributed on an "AS IS" BASIS,
12//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13//  See the License for the specific language governing permissions and
14//  limitations under the License.
15//
16
17use crate::{Block, Inst, Value, value::fmt_vals};
18use std::fmt::Display;
19
20#[derive(Default, Clone)]
21pub struct BlockData {
22    pub(crate) params: Vec<Value>,
23    pub(crate) insts: Vec<Inst>,
24}
25
26impl BlockData {
27    /// Returns the parameters of the block.
28    pub fn get_params(&self) -> &[Value] {
29        &self.params
30    }
31
32    /// Returns the instructions in the block.
33    pub fn get_insts(&self) -> &[Inst] {
34        &self.insts
35    }
36
37    /// Returns all successors of the block.
38    pub fn get_successors(&self) -> Vec<Block> {
39        self.insts
40            .last()
41            .map(|last| last.successors())
42            .unwrap_or_default()
43    }
44}
45
46impl Display for BlockData {
47    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
48        writeln!(f, "({}):", fmt_vals(&self.params))?;
49        for inst in &self.insts {
50            writeln!(f, "    {}", inst)?;
51        }
52        writeln!(f)
53    }
54}