use timely::dataflow::operators::{Input, Probe, Enter, Leave};
use timely::dataflow::operators::vec::Map;
fn main() {
timely::execute_from_args(std::env::args(), |worker| {
let timer = std::time::Instant::now();
let mut args = std::env::args();
args.next();
let dataflows = args.next().unwrap().parse::<usize>().unwrap();
let length = args.next().unwrap().parse::<usize>().unwrap();
let record = args.next() == Some("record".to_string());
let mut inputs = Vec::new();
let mut probes = Vec::new();
for _dataflow in 0 .. dataflows {
worker.dataflow(|scope| {
let (input, stream) = scope.new_input();
let stream = scope.region(|inner| {
let mut stream = stream.enter(inner);
for _step in 0 .. length {
stream = stream.map(|x: ()| x);
}
stream.leave(scope)
});
let (probe, _stream) = stream.probe();
inputs.push(input);
probes.push(probe);
});
}
println!("{:?}\tdataflows built ({} x {})", timer.elapsed(), dataflows, length);
for round in 0 .. {
let dataflow = round % dataflows;
if record {
inputs[dataflow].send(());
}
inputs[dataflow].advance_to(round);
let mut steps = 0;
while probes[dataflow].less_than(&round) {
worker.step();
steps += 1;
}
if round % 1000 == 0 { println!("{:?}\tround {} complete in {} steps", timer.elapsed(), round, steps); }
}
}).unwrap();
}