Graphite/node-graph/interpreted-executor/benches/benchmark_util.rs

use criterion::BenchmarkGroup;
use criterion::measurement::Measurement;
use futures::executor::block_on;
use graph_craft::proto::ProtoNetwork;
use graph_craft::util::{DEMO_ART, compile, load_from_name};
use graphene_std::application_io::EditorApi;
use interpreted_executor::dynamic_executor::DynamicExecutor;
use interpreted_executor::util::wrap_network_in_scope;

pub fn setup_network(name: &str) -> (DynamicExecutor, ProtoNetwork) {
	let mut network = load_from_name(name);
	let editor_api = std::sync::Arc::new(EditorApi::default());
	println!("generating substitutions");
	let substitutions = preprocessor::generate_node_substitutions();
	println!("expanding network");
	preprocessor::expand_network(&mut network, &substitutions);
	let network = wrap_network_in_scope(network, editor_api);
	let proto_network = compile(network);
	let executor = block_on(DynamicExecutor::new(proto_network.clone())).unwrap();
	(executor, proto_network)
}

pub fn bench_for_each_demo<M: Measurement, F>(group: &mut BenchmarkGroup<M>, f: F)
where
	F: Fn(&str, &mut BenchmarkGroup<M>),
{
	for name in DEMO_ART {
		f(name, group);
	}
}