mirror of
https://github.com/storytold/materialcraft.git
synced 2026-11-08 07:34:24 +00:00
chasse20
68 lines
3.6 KiB
Rust
68 lines
3.6 KiB
Rust
use materialcraft_engine::{
|
|
core::{FunctionGraph, GraphKind, Value},
|
|
execute,
|
|
formats::sbs::{read_sbs, write_sbs},
|
|
new_session,
|
|
nodes::function::{EvaluationContext, EvaluationState, compile},
|
|
};
|
|
use serde_json::json;
|
|
|
|
#[test]
|
|
fn incomplete_function_round_trip_preserves_nodes_without_choosing_an_output() {
|
|
let mut session = new_session();
|
|
execute(&mut session, "file.new", &json!({"kind": "function", "name": "Unfinished"})).unwrap();
|
|
for operation in ["const_float1", "mul"] {
|
|
execute(&mut session, "graph.function.edit", &json!({"edit": {"action": "add", "operation": operation, "at": [0, 0]}})).unwrap();
|
|
}
|
|
assert_eq!(session.graph().unwrap().function.as_ref().unwrap().output, None);
|
|
|
|
let bytes = write_sbs(&session.pkg).unwrap();
|
|
let read = read_sbs(&bytes, "unfinished.sbs", &|_| None).unwrap();
|
|
assert!(read.warnings.is_empty(), "{:?}", read.warnings);
|
|
let (_, graph) = read.pkg.graph_by_name("Unfinished").unwrap();
|
|
let function = graph.function.as_ref().unwrap();
|
|
assert_eq!(function.output, None);
|
|
assert_eq!(function.nodes.iter().map(|node| node.op.as_str()).collect::<Vec<_>>(), vec!["const_float1", "mul"]);
|
|
assert_eq!(function.nodes[1].inputs, vec![None, None]);
|
|
}
|
|
|
|
#[test]
|
|
fn standalone_function_creation_wiring_frame_movement_and_native_round_trip() {
|
|
let mut session = new_session();
|
|
execute(&mut session, "file.new", &json!({"kind": "function", "name": "Multiply"})).unwrap();
|
|
assert_eq!(session.graph().unwrap().kind(), GraphKind::Function);
|
|
assert!(session.graph().unwrap().nodes.is_empty());
|
|
assert!(execute(&mut session, "graph.node.add", &json!({"def": "uniform"})).is_err());
|
|
|
|
let mut function = FunctionGraph::default();
|
|
let first = function.add_constant(Value::Float(2.0));
|
|
let second = function.add_constant(Value::Float(3.0));
|
|
function.output = Some(function.add("mul", &[first, second], None));
|
|
execute(&mut session, "graph.function.set", &json!({"function": function})).unwrap();
|
|
let original = session.pkg.clone();
|
|
let mut cycle = function.clone();
|
|
cycle.nodes[2].inputs[0] = cycle.output;
|
|
assert!(execute(&mut session, "graph.function.set", &json!({"function": cycle})).is_err());
|
|
assert_eq!(session.pkg, original);
|
|
|
|
let frame = execute(&mut session, "graph.frame.add", &json!({"rect": [-20, -20, 500, 240]})).unwrap()["id"].as_u64().unwrap();
|
|
execute(&mut session, "graph.comment.add", &json!({"at": [0, 120], "text": "Two times three"})).unwrap();
|
|
execute(&mut session, "graph.frame.move", &json!({"id": frame, "by": [32, 64]})).unwrap();
|
|
assert_eq!(session.graph().unwrap().function.as_ref().unwrap().nodes[0].pos, [32.0, 64.0]);
|
|
|
|
let bytes = write_sbs(&session.pkg).unwrap();
|
|
let xml = String::from_utf8(bytes.clone()).unwrap();
|
|
assert!(xml.contains("<function><identifier v=\"Multiply\"/>"));
|
|
let read = read_sbs(&bytes, "functions.sbs", &|_| None).unwrap();
|
|
assert!(read.warnings.is_empty(), "{:?}", read.warnings);
|
|
let graph = read.pkg.graphs.iter().find(|graph| graph.name == "Multiply").unwrap();
|
|
assert_eq!(graph.kind(), GraphKind::Function);
|
|
assert_eq!(graph.frames.len(), 1);
|
|
assert_eq!(graph.comments[0].text, "Two times three");
|
|
let function = graph.function.as_ref().unwrap();
|
|
assert_eq!(function.nodes[0].pos, [32.0, 64.0]);
|
|
let variables = Default::default();
|
|
let context = EvaluationContext { position: [0.0; 2], size: [1.0; 2], images: &[], tile: true, variables: &variables };
|
|
assert_eq!(compile(function, 0).unwrap().evaluate(&context, &mut EvaluationState::default()).unwrap(), Value::Float(6.0));
|
|
}
|