1 | module main |
2 | |
3 | import benchmark |
4 | import sim |
5 | import sim.args as simargs |
6 | import sim.img |
7 | |
8 | fn main() { |
9 | args := simargs.parse_args(extra_workers: 1)! as simargs.ParallelArgs |
10 | |
11 | img_settings := img.image_settings_from_grid(args.grid) |
12 | |
13 | request_chan := chan &sim.SimRequest{cap: args.workers} |
14 | result_chan := chan &sim.SimResult{cap: args.workers} |
15 | |
16 | mut writer := img.ppm_writer_for_fname(args.filename, img_settings)! |
17 | |
18 | mut workers := []thread{cap: args.workers + 1} |
19 | mut bmark := benchmark.start() |
20 | |
21 | defer { |
22 | image_worker := workers.pop() |
23 | request_chan.close() |
24 | sim.log('Waiting for workers to finish') |
25 | workers.wait() |
26 | result_chan.close() |
27 | sim.log('Waiting for image writer to finish') |
28 | image_worker.wait() |
29 | sim.log('Workers finished!') |
30 | bmark.measure(@FN) |
31 | sim.log('Closing writer file') |
32 | writer.close() |
33 | sim.log('Done!') |
34 | } |
35 | |
36 | for id in 0 .. args.workers { |
37 | workers << spawn sim.sim_worker(id, request_chan, [result_chan]) |
38 | } |
39 | |
40 | workers << spawn img.image_worker(mut writer, result_chan, img_settings) |
41 | |
42 | handle_request := fn [request_chan] (request &sim.SimRequest) ! { |
43 | request_chan <- request |
44 | } |
45 | |
46 | sim.run(args.params, |
47 | grid: args.grid |
48 | on_request: sim.SimRequestHandler(handle_request) |
49 | ) |
50 | } |