RileyCapture

python/IECoreRenderManTest/RileyCapture.py

@@ -0,0 +1,63 @@
+##########################################################################
+#
+#  Copyright (c) 2024, Cinesite VFX Ltd. All rights reserved.
+#
+#  Redistribution and use in source and binary forms, with or without
+#  modification, are permitted provided that the following conditions are
+#  met:
+#
+#      * Redistributions of source code must retain the above
+#        copyright notice, this list of conditions and the following
+#        disclaimer.
+#
+#      * Redistributions in binary form must reproduce the above
+#        copyright notice, this list of conditions and the following
+#        disclaimer in the documentation and/or other materials provided with
+#        the distribution.
+#
+#      * Neither the name of John Haddon nor the names of
+#        any other contributors to this software may be used to endorse or
+#        promote products derived from this software without specific prior
+#        written permission.
+#
+#  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
+#  IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
+#  THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+#  PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
+#  CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+#  EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+#  PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+#  PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+#  LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+#  NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+#  SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+#
+##########################################################################
+
+import json
+import os
+import pathlib
+import shutil
+import tempfile
+
+class RileyCapture :
+
+	def __init__( self ) :
+
+		self.__dir = pathlib.Path( tempfile.mkdtemp( prefix = "rileyCapture" ) )
+		self.__fileName = self.__dir / "capture.json"
+		self.json = {}
+
+	def __enter__( self ) :
+
+		os.environ["RILEY_CAPTURE"] = str( self.__fileName )
+		os.environ["RILEY_CAPTURE_FORMAT"] = "json"
+
+		return self
+
+	def __exit__( self, type, value, traceBack ) :
+
+		del os.environ["RILEY_CAPTURE"]
+		del os.environ["RILEY_CAPTURE_FORMAT"]
+		self.json = json.load( open( self.__fileName ) )
+		shutil.rmtree( self.__dir )

python/IECoreRenderManTest/__init__.py

@@ -34,6 +34,7 @@
 #
 ##########################################################################
 
+from .RileyCapture import RileyCapture
 from .RendererTest import RendererTest
 
 if __name__ == "__main__":