Initial commit

benchmark/foundation/context.py

@@ -0,0 +1,100 @@
+import os
+
+from foundation import logger
+
+
+class Context:
+    __slots__ = ('blender',
+                 'device_type',
+                 'device_name',
+                 'scenes',
+                 'scenes_dir',
+                 'configure_script',
+                 'image_output_dir')
+
+    def __init__(self):
+        # Full path to blender binary.
+        self.blender = "blender"
+        # Type of the device to run on. It must be either CPU, CUDA or OpenCL.
+        self.device_type = 'CPU'
+        # Name of the device to render on.
+        self.device_name = 'NAME'
+        # By default we use empty list, it is up to the user to fill it in.
+        self.scenes = []
+        # It is up to the user to provide proper path to scenes.
+        self.scenes_dir = ""
+        # Blender-side configuration script.
+        self.configure_script = "configure.py"
+        # Directory where render result images will be saved.
+        # Empty means no results are saved.
+        self.image_output_dir = ""
+
+    def listAllScenes(self, directory):
+        import os
+        if not os.path.exists(directory):
+            logger.ERROR("Non-exiting directory {}" . format(directory))
+            return []
+        if not os.path.isdir(directory):
+            logger.ERROR("Not a directory {}" . format(directory))
+            return []
+        all_scenes = sorted(os.listdir(directory))
+        usable_scenes = []
+        for scene in all_scenes:
+            scene_filename = self.getSceneFilename(scene, do_checks=False)
+            if os.path.exists(scene_filename) and \
+               os.path.isfile(scene_filename):
+                usable_scenes.append(scene)
+        return usable_scenes
+
+    def printSummary(self):
+        logger.INFO("Benchmark summary:")
+        logger.INFO("  Device type: {}" . format(self.device_type))
+        if self.device_name:
+            logger.INFO("  Device name: {}" . format(self.device_name))
+        logger.INFO("  Scenes: {}" . format(", ".join(self.scenes)))
+
+    def verify(self):
+        if not os.path.exists(self.blender):
+            logger.ERROR("Missing blender: {}" . format(self.blender))
+            return False
+        if not os.path.isfile(self.blender):
+            logger.ERROR("Blender is not a file: {}" . format(self.blender))
+            return False
+        if not os.path.exists(self.configure_script):
+            logger.ERROR("Missing configuration script: {}" .
+                         format(self.configure_script))
+            return False
+        if not os.path.isfile(self.configure_script):
+            logger.ERROR("Configuration script is not a file: {}" .
+                         format(self.configure_script))
+            return False
+        if self.image_output_dir:
+            if not os.path.exists(self.image_output_dir):
+                logger.ERROR("Missing image output directory: {}" .
+                             format(self.image_output_dir))
+                return False
+            if not os.path.isdir(self.image_output_dir):
+                logger.ERROR("Image out is not a directory: {}" .
+                             format(self.image_output_dir))
+                return False
+        return True
+
+    def getDeviceFileSuffix(self):
+        if self.device_type == 'CPU':
+            return '_cpu'
+        elif self.device_type in ('CUDA', 'OPENCL', 'GPU'):
+            return '_gpu'
+        else:
+            logger.FATAL("Unknown device type: {}" . format(self.device_type))
+        return ""
+
+    def getSceneFilename(self, scene, do_checks=True):
+        suffix = self.getDeviceFileSuffix()
+        blendfile = scene + suffix + ".blend"
+        filepath = os.path.join(self.scenes_dir, scene, blendfile)
+        if do_checks:
+            if not os.path.exists(filepath):
+                logger.FATAL("File not file: {}" . format(filepath))
+            if not os.path.isfile(filepath):
+                logger.FATAL("Scene is not a file: {}" . format(filepath))
+        return filepath