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}