mirror of
https://github.com/storytold/materialcraft.git
synced 2026-11-08 07:34:24 +00:00
chasse20
61 lines
2.9 KiB
Rust
61 lines
2.9 KiB
Rust
use materialcraft_engine::core::{FunctionGraph, Value};
|
|
use materialcraft_engine::{Session, execute, formats::sbs, new_session};
|
|
use serde_json::json;
|
|
|
|
fn math_function() -> FunctionGraph {
|
|
let mut graph = FunctionGraph::default();
|
|
let power = graph.add("get_float1", &[], Some(Value::Text("power".into())));
|
|
let radius = graph.add("pow2", &[power], None);
|
|
let two = graph.add_constant(Value::Float(2.0));
|
|
let angle = graph.add("log2", &[two], None);
|
|
let position = graph.add("cartesian", &[radius, angle], None);
|
|
let tail = graph.add_constant(Value::Float2([0.0, 1.0]));
|
|
graph.output = Some(graph.add("vector4", &[position, tail], None));
|
|
graph
|
|
}
|
|
|
|
fn assert_outputs(session: &mut Session, radius: f32) {
|
|
session.cook();
|
|
for node in &session.graph().unwrap().nodes {
|
|
let result = &session.cooked.nodes[&node.id];
|
|
assert!(result.error.is_none(), "{:?}", result.error);
|
|
let color = if node.def == "value_processor" {
|
|
let Some(Value::Float4(value)) = session.cooked.value(node.id, 0) else { panic!("expected typed Float4 output") };
|
|
*value
|
|
} else {
|
|
session.cooked.image(node.id, 0).unwrap().mean()
|
|
};
|
|
assert!((color[0] - radius * 1.0f32.cos()).abs() < 1e-5, "{color:?}");
|
|
assert!((color[1] - radius * 1.0f32.sin()).abs() < 1e-5, "{color:?}");
|
|
assert!(color[2].abs() < 1e-5 && (color[3] - 1.0).abs() < 1e-5);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn math_primitives_cook_in_processors_and_parameter_functions_and_round_trip() {
|
|
let mut session = new_session();
|
|
session.preview_size = Some(32);
|
|
execute(&mut session, "graph.exposed.add", &json!({"id": "power", "default": {"t": "float", "v": -2.0}})).unwrap();
|
|
let function = math_function();
|
|
for definition in ["value_processor", "pixel_processor", "uniform"] {
|
|
let id = execute(&mut session, "graph.node.add", &json!({"def": definition})).unwrap()["id"].as_u64().unwrap();
|
|
if definition == "uniform" {
|
|
execute(&mut session, "graph.node.function", &json!({"id": id, "param": "color", "function": function})).unwrap();
|
|
} else {
|
|
execute(&mut session, "graph.node.param", &json!({"id": id, "param": "function", "value": Value::Function(Box::new(function.clone()))})).unwrap();
|
|
}
|
|
}
|
|
assert_outputs(&mut session, 0.25);
|
|
execute(&mut session, "graph.exposed.set", &json!({"id": "power", "value": -1.0})).unwrap();
|
|
assert_outputs(&mut session, 0.5);
|
|
execute(&mut session, "edit.undo", &json!({})).unwrap();
|
|
assert_outputs(&mut session, 0.25);
|
|
execute(&mut session, "edit.redo", &json!({})).unwrap();
|
|
assert_outputs(&mut session, 0.5);
|
|
let report = sbs::read_sbs(&sbs::write_sbs(&session.pkg).unwrap(), "math.sbs", &|_| None).unwrap();
|
|
assert!(report.warnings.is_empty(), "{:?}", report.warnings);
|
|
let mut reopened = Session::new(report.pkg);
|
|
reopened.preview_size = Some(32);
|
|
assert_outputs(&mut reopened, 0.5);
|
|
}
|