summaryrefslogtreecommitdiffstats
path: root/dimension/client.py.in
diff options
context:
space:
mode:
authorTavian Barnes <tavianator@gmail.com>2011-11-04 19:02:27 -0400
committerTavian Barnes <tavianator@gmail.com>2011-11-04 19:02:27 -0400
commit3e40d9c7b75dec008beb6b4ecf71b60e8bae088d (patch)
treec6961bc743fc1a1ee909973e216313a45febcdb2 /dimension/client.py.in
parentbdcce150b4154f84ef24b52b44801c3e735b3eb0 (diff)
downloaddimension-3e40d9c7b75dec008beb6b4ecf71b60e8bae088d.tar.xz
Let Future objects be used as context managers.
Diffstat (limited to 'dimension/client.py.in')
-rw-r--r--dimension/client.py.in69
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))