piske 0.1.2

The piske programming langauge for generative art
Documentation 
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
41
42
43
44
45
46
47
48
49
50

#![allow(dead_code)]

extern crate tempfile;
extern crate piske;

use std::fs::File;

use piske::value::Value;
use piske::parse::program;
use piske::visitor::{State, SymbolDefineVisitor, TypeComputationVisitor, EvaluateVisitor,
    TranspileVisitor};

pub fn expect_prog_with_state(prog: &str, val: Value, mut state: &mut State) {
    let ast = program(prog).unwrap();
    SymbolDefineVisitor::visit(&ast, &mut state).unwrap();
    TypeComputationVisitor::visit(&ast, &mut state).unwrap();
    let evaluated = EvaluateVisitor::visit(&ast, &mut state);
    assert_eq!(evaluated, Ok(val));
}

pub fn expect_prog(prog: &str, val: Value) {
    let mut state = State::default();
    expect_prog_with_state(prog, val, &mut state);
}

pub fn examine_translated_source(prog: &str) {
    let mut state = State::default();
    let ast = program(prog).unwrap();
    SymbolDefineVisitor::visit(&ast, &mut state).unwrap();
    TypeComputationVisitor::visit(&ast, &mut state).unwrap();
    let translated = TranspileVisitor::visit(&ast, &mut state).unwrap();
    println!("{}", translated.as_str());
}

pub fn test_output(mut file: &File, expected: &str) {
    use std::io::{Read, Seek, SeekFrom};

    let mut buffer = String::new();
    file.seek(SeekFrom::Start(0)).unwrap();
    file.read_to_string(&mut buffer).unwrap();
    assert_eq!(&buffer, expected);
}

pub fn new_state_with_temp_output() -> (State, File) {
    let mut state = State::default();
    let tempfile = tempfile::tempfile().unwrap();
    let temp_clone = tempfile.try_clone().unwrap();
    state.io.set_stdout(tempfile);
    (state, temp_clone)
}