origen_metal 1.4.0

Bare metal APIs for the Origen SDK
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

//! A simple example processor which removes all flow logic and redundant content from resources
//! blocks, basically leaving just the test definitions
use super::super::nodes::PGM;
use crate::ast::*;
use crate::Result;

pub struct CleanResources {
    in_resources: bool,
}

pub fn run(node: &Node<PGM>) -> Result<Node<PGM>> {
    let mut p = CleanResources { in_resources: false };
    Ok(node.process(&mut p)?.unwrap())
}

impl Processor<PGM> for CleanResources {
    fn on_node(&mut self, node: &Node<PGM>) -> Result<Return<PGM>> {
        match &node.attrs {
            PGM::Resources => {
                let orig = self.in_resources;
                self.in_resources = true;
                let nodes = node.process_children(self)?;
                self.in_resources = orig;
                Ok(Return::ReplaceChildren(nodes))
            }
            PGM::Test(_, _) => {
                Ok(Return::ProcessChildren)
            }
            // For all other nodes
            _ => {
                if self.in_resources {
                    // Throw away everything except those with a handler defined above
                    Ok(Return::UnwrapWithProcessedChildren)
                } else {
                    Ok(Return::ProcessChildren)
                }
            }
        }
    }
}