diff options
author | Tavian Barnes <tavianator@gmail.com> | 2011-11-04 19:02:27 -0400 |
---|---|---|
committer | Tavian Barnes <tavianator@gmail.com> | 2011-11-04 19:02:27 -0400 |
commit | 3e40d9c7b75dec008beb6b4ecf71b60e8bae088d (patch) | |
tree | c6961bc743fc1a1ee909973e216313a45febcdb2 /dimension/client.py.in | |
parent | bdcce150b4154f84ef24b52b44801c3e735b3eb0 (diff) | |
download | dimension-3e40d9c7b75dec008beb6b4ecf71b60e8bae088d.tar.xz |
Let Future objects be used as context managers.
Diffstat (limited to 'dimension/client.py.in')
-rw-r--r-- | dimension/client.py.in | 69 |
1 files changed, 28 insertions, 41 deletions
diff --git a/dimension/client.py.in b/dimension/client.py.in index 127b96c..2eaf3c3 100644 --- a/dimension/client.py.in +++ b/dimension/client.py.in @@ -157,28 +157,27 @@ def main(): scene.adc_bailout = float(args.adc_bailout) # Ray-trace the scene - future = scene.ray_trace_async() - bar = None - if not args.quiet: - if scene.nthreads == 1: - render_message = "Rendering scene" - else: - render_message = "Rendering scene (using %d threads)" % scene.nthreads - bar = progress_bar_async(render_message, future) - if args.preview: - if have_preview: - preview.show_preview(canvas, future) - else: - print("Couldn't display preview window", file = sys.stderr) - bar.join() - future.join() + with scene.ray_trace_async() as future: + bar = None + if not args.quiet: + if scene.nthreads == 1: + render_message = "Rendering scene" + else: + render_message = "Rendering scene (using %d threads)" % scene.nthreads + bar = progress_bar_async(render_message, future) + if args.preview: + if have_preview: + preview.show_preview(canvas, future) + else: + print("Couldn't display preview window", file = sys.stderr) + if bar is not None: + bar.join() # Write the output file export_timer = Timer() - future = canvas.write_PNG_async(args.output) - if not args.quiet: - progress_bar("Writing %s" % args.output, future) - future.join() + with canvas.write_PNG_async(args.output) as future: + if not args.quiet: + progress_bar("Writing %s" % args.output, future) export_timer.stop() # Print execution times @@ -233,30 +232,18 @@ def working_directory(newwd): def progress_bar(str, future): """Display a progress bar while a Future completes.""" - try: - print(str, end = " ") - sys.stdout.flush() - - term_width = terminal_width() - width = term_width - (len(str) + 1)%term_width - for i in range(width): - future.wait((i + 1)/width) - print(".", end = "") - sys.stdout.flush() - - print() - sys.stdout.flush() - except KeyboardInterrupt: - print() + print(str, end = " ") + sys.stdout.flush() + + term_width = terminal_width() + width = term_width - (len(str) + 1)%term_width + for i in range(width): + future.wait((i + 1)/width) + print(".", end = "") sys.stdout.flush() - future.cancel() - try: - future.join() - except RuntimeError: - # Swallow the failure exception - pass - raise + print() + sys.stdout.flush() def progress_bar_async(str, future): thread = threading.Thread(target = progress_bar, args = (str, future)) |