zhc_ir 0.6.0

Graph-based intermediate representation framework with dialect support
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

use std::rc::Rc;

use crate::{
    IR, OpMap,
    visualization::{LayoutDialect, LayoutInstructionSet, visual_annotation::VisualAnnotation},
};

pub fn annotate_layout<OpAnn: VisualAnnotation + Clone>(
    ir: &mut IR<LayoutDialect>,
    annotations: &OpMap<OpAnn>,
) {
    ir.mutate_ops_linear(|op| match op {
        LayoutInstructionSet::Operation { opid, op, .. } => {
            op.annotation = Some(Rc::new(annotations.get(opid).unwrap().clone()));
        }
        LayoutInstructionSet::Group { ir, .. } => {
            annotate_layout(ir, annotations);
        }
        _ => {}
    });
}