diff options
Diffstat (limited to 'dimension')
-rw-r--r-- | dimension/Makefile.am | 3 | ||||
-rw-r--r-- | dimension/client.py.in | 26 | ||||
-rw-r--r-- | dimension/preview.py | 66 |
3 files changed, 93 insertions, 2 deletions
diff --git a/dimension/Makefile.am b/dimension/Makefile.am index c12a6b7..a892601 100644 --- a/dimension/Makefile.am +++ b/dimension/Makefile.am @@ -26,7 +26,8 @@ dist_bin_SCRIPTS = dimension # since it's not really installed, so disable the --help and --version checks AM_INSTALLCHECK_STD_OPTIONS_EXEMPT = dimension -pkgpython_PYTHON = __init__.py +pkgpython_PYTHON = __init__.py \ + preview.py nodist_pkgpython_PYTHON = client.py clean-local: diff --git a/dimension/client.py.in b/dimension/client.py.in index 4881f62..127b96c 100644 --- a/dimension/client.py.in +++ b/dimension/client.py.in @@ -23,9 +23,16 @@ import argparse import re import os import sys +import threading from contextlib import contextmanager from dimension import * +have_preview = True +try: + from dimension import preview +except ImportError: + have_preview = False + def main(): """Invoke the client from the command line.""" @@ -68,6 +75,9 @@ def main(): parser.add_argument("input", action = "store", type = str, help = "the input scene description file") + parser.add_argument("-p", "--preview", action = "store_true", + help = "display a preview while the image renders") + # Debugging/testing options parser.add_argument("--strict", action = "store_true", help = argparse.SUPPRESS) @@ -118,6 +128,8 @@ def main(): # Make the canvas canvas = Canvas(width = args.region_width, height = args.region_height) canvas.optimize_PNG() + if args.preview: + canvas.optimize_GL() # Make the scene object scene = Scene(canvas = canvas, @@ -146,12 +158,19 @@ def main(): # 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 - progress_bar(render_message, future) + 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() # Write the output file @@ -238,3 +257,8 @@ def progress_bar(str, future): # Swallow the failure exception pass raise + +def progress_bar_async(str, future): + thread = threading.Thread(target = progress_bar, args = (str, future)) + thread.start() + return thread diff --git a/dimension/preview.py b/dimension/preview.py new file mode 100644 index 0000000..868c16b --- /dev/null +++ b/dimension/preview.py @@ -0,0 +1,66 @@ +#!/usr/bin/env python3 + +######################################################################### +# Copyright (C) 2011 Tavian Barnes <tavianator@tavianator.com> # +# # +# This file is part of Dimension. # +# # +# Dimension is free software; you can redistribute it and/or modify it # +# under the terms of the GNU General Public License as published by the # +# Free Software Foundation; either version 3 of the License, or (at # +# your option) any later version. # +# # +# Dimension is distributed in the hope that it will be useful, but # +# WITHOUT ANY WARRANTY; without even the implied warranty of # +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU # +# General Public License for more details. # +# # +# You should have received a copy of the GNU General Public License # +# along with this program. If not, see <http://www.gnu.org/licenses/>. # +######################################################################### + +from PyQt4 import QtCore, QtGui, QtOpenGL + +class Preview(QtOpenGL.QGLWidget): + """Surface that the scene is rendered to.""" + def __init__(self, parent, canvas): + QtOpenGL.QGLWidget.__init__(self, parent) + self.canvas = canvas + + def paintGL(self): + self.canvas.draw_GL() + +class PreviewWindow(QtGui.QMainWindow): + """Main window for a rendering preview.""" + def __init__(self, canvas, future): + QtGui.QMainWindow.__init__(self) + self.canvas = canvas + self.future = future + + self.setMinimumSize(canvas.width, canvas.height) + self.setMaximumSize(canvas.width, canvas.height) + self.widget = Preview(self, canvas) + self.setCentralWidget(self.widget) + + self.render_timer = QtCore.QTimer(self) + self.render_timer.timeout.connect(self.update_preview) + self.render_timer.start(0) + + @QtCore.pyqtSlot() + def update_preview(self): + self.widget.updateGL() + if self.future.progress() == 1: + self.render_timer.stop() + self.close_timer = QtCore.QTimer(self) + self.close_timer.singleShot(1000, self.close) + + @QtCore.pyqtSlot() + def close(self): + QtCore.QCoreApplication.instance().quit() + +def show_preview(canvas, future): + app = QtGui.QApplication(["Dimension Preview"]) + window = PreviewWindow(canvas, future) + window.show() + app.exec() + del window |