You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
125 lines
4.2 KiB
125 lines
4.2 KiB
use crate::parser::{tokenize, parse, Token, Ast};
|
|
|
|
#[test]
|
|
fn test_tokenize() {
|
|
let input = "(+ (* 1:u8 2) 2)";
|
|
let expected = vec![
|
|
Token::LeftParen,
|
|
Token::Symbol("+".to_string()),
|
|
Token::LeftParen,
|
|
Token::Symbol("*".to_string()),
|
|
Token::Symbol("1:u8".to_string()),
|
|
Token::Symbol("2".to_string()),
|
|
Token::RightParen,
|
|
Token::Symbol("2".to_string()),
|
|
Token::RightParen,
|
|
];
|
|
assert_eq!(tokenize(input), expected);
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse() {
|
|
let src = "
|
|
(module Shader Logical GLSL450)
|
|
(import :std GLSL.std.450)
|
|
(bind (frag-coord:*vec4fi) (Builtin FragCoord))
|
|
(bind (out-color:*vec4fo) (Location 0))
|
|
(dec frag-coord:*vec4fi Input)
|
|
(dec out-color:*vec4fo Output)
|
|
(entry main Fragment OriginUpperLeft (:frag-coord :out-color))
|
|
(fun (main)
|
|
(store-ptr (out-color)
|
|
(vec4fi (/ (.xy (load-ptr frag-coord))
|
|
(vec2f 1920.0 1080.0))
|
|
1.0
|
|
1.0)))
|
|
";
|
|
let ast = parse(tokenize(src));
|
|
println!("{:?}", ast);
|
|
let test_ast: Ast = Ast::Root(vec![
|
|
Ast::List(vec![
|
|
Ast::Symbol("module".to_string()),
|
|
Ast::Symbol("Shader".to_string()),
|
|
Ast::Symbol("Logical".to_string()),
|
|
Ast::Symbol("GLSL450".to_string()),
|
|
]),
|
|
Ast::List(vec![
|
|
Ast::Symbol("import".to_string()),
|
|
Ast::Symbol(":std".to_string()),
|
|
Ast::Symbol("GLSL.std.450".to_string()),
|
|
]),
|
|
Ast::List(vec![
|
|
Ast::Symbol("bind".to_string()),
|
|
Ast::List(vec![
|
|
Ast::Symbol("frag-coord:*vec4fi".to_string()),
|
|
]),
|
|
Ast::List(vec![
|
|
Ast::Symbol("Builtin".to_string()),
|
|
Ast::Symbol("FragCoord".to_string()),
|
|
]),
|
|
]),
|
|
Ast::List(vec![
|
|
Ast::Symbol("bind".to_string()),
|
|
Ast::List(vec![
|
|
Ast::Symbol("out-color:*vec4fo".to_string()),
|
|
]),
|
|
Ast::List(vec![
|
|
Ast::Symbol("Location".to_string()),
|
|
Ast::Symbol("0".to_string()),
|
|
]),
|
|
]),
|
|
Ast::List(vec![
|
|
Ast::Symbol("dec".to_string()),
|
|
Ast::Symbol("frag-coord:*vec4fi".to_string()),
|
|
Ast::Symbol("Input".to_string()),
|
|
]),
|
|
Ast::List(vec![
|
|
Ast::Symbol("dec".to_string()),
|
|
Ast::Symbol("out-color:*vec4fo".to_string()),
|
|
Ast::Symbol("Output".to_string()),
|
|
]),
|
|
Ast::List(vec![
|
|
Ast::Symbol("entry".to_string()),
|
|
Ast::Symbol("main".to_string()),
|
|
Ast::Symbol("Fragment".to_string()),
|
|
Ast::Symbol("OriginUpperLeft".to_string()),
|
|
Ast::List(vec![
|
|
Ast::Symbol(":frag-coord".to_string()),
|
|
Ast::Symbol(":out-color".to_string()),
|
|
]),
|
|
]),
|
|
Ast::List(vec![
|
|
Ast::Symbol("fun".to_string()),
|
|
Ast::List(vec![
|
|
Ast::Symbol("main".to_string()),
|
|
]),
|
|
Ast::List(vec![
|
|
Ast::Symbol("store-ptr".to_string()),
|
|
Ast::List(vec![
|
|
Ast::Symbol("out-color".to_string()),
|
|
]),
|
|
Ast::List(vec![
|
|
Ast::Symbol("vec4fi".to_string()),
|
|
Ast::List(vec![
|
|
Ast::Symbol("/".to_string()),
|
|
Ast::List(vec![
|
|
Ast::Symbol(".xy".to_string()),
|
|
Ast::List(vec![
|
|
Ast::Symbol("load-ptr".to_string()),
|
|
Ast::Symbol("frag-coord".to_string()),
|
|
]),
|
|
]),
|
|
Ast::List(vec![
|
|
Ast::Symbol("vec2f".to_string()),
|
|
Ast::Symbol("1920.0".to_string()),
|
|
Ast::Symbol("1080.0".to_string()),
|
|
]),
|
|
]),
|
|
Ast::Symbol("1.0".to_string()),
|
|
Ast::Symbol("1.0".to_string()),
|
|
]),
|
|
]),
|
|
]),
|
|
]);
|
|
assert_eq!(ast, test_ast);
|
|
} |