Function function_use_stack

tangle/src/ir.rs:618–673

()

    #[test]
    pub fn function_use_stack() {
        let mut ir = IR::new();
        let mut f = Node::function::<1, 1>(&mut ir);
        let input = f.add_argument(&mut ir);
        let output = f.add_return(&mut ir);
        // We create a stack slot
        let mut ss1 = f.add_stack_slot(&mut ir);

        // ...and push, which should resolve to a use of ss1
        let mut push = Node::simple(Operation::StoreStack);
        ss1 = f.add_body(push, &mut ir, |i, r| {
            i.connect_input(0, ss1, r);
            i.connect_input(1, input, r);
            i.sinks()[0]
        }).1;

        // get the value
        let mut pop = Node::simple(Operation::LoadStack);
        let ss_val = f.add_body(pop, &mut ir, |i, r| {
            i.connect_input(0, ss1, r);
            i.sinks()[0]
        }).1;

        // then add to it via ss1
        let mut two = Node::constant(2);
        let mut add = Node::simple(Operation::Add);
        let mut two_const = f.add_body(two, &mut ir, |two, r| {
            two.sinks()[0]
        }).1;
        let added_val = f.add_body(add, &mut ir, |add, r| {
            println!("add closure {}", r.ports[ss1].storage);
            add.connect_input(0, ss_val, r);
            add.connect_input(1, two_const, r);
            add.sinks()[0]
        }).1;

        // and store it again
        let mut push = Node::simple(Operation::StoreStack);
        f.add_body(push, &mut ir, |i, r| {
            i.connect_input(0, ss1, r);
            i.connect_input(1, added_val, r);
            ss1 = i.sinks()[0];
        });

        // ...then pop it into the result register
        let mut pop = Node::simple(Operation::LoadStack);
        f.add_body(pop, &mut ir, |i, r| {
            i.connect_input(0, ss1, r);
            i.connect_output(0, output, r);
        });

        ir.set_body(f);
        let hello_fn: extern "C" fn(usize) -> usize = ir.compile_fn().unwrap().0;
        assert_eq!(hello_fn(0), 2);
        assert_eq!(hello_fn(1), 3);
    }

    #[test]