Merge branch 'trim-auto'

[apitrace-tests] / trim_stress_driver.py

diff --git a/trim_stress_driver.py b/trim_stress_driver.py
new file mode 100644 (file)
index 0000000..64863f8
--- /dev/null
+++ b/trim_stress_driver.py
@@ -0,0 +1,98 @@
+#!/usr/bin/env python
+
+# Copyright 2012 Intel Corporation
+#
+# Permission is hereby granted, free of charge, to any person
+# obtaining a copy of this software and associated documentation files
+# (the "Software"), to deal in the Software without restriction,
+# including without limitation the rights to use, copy, modify, merge,
+# publish, distribute, sublicense, and/or sell copies of the Software,
+# and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
+# BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
+# ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+'''Stress test driver for apitrace trim.'''
+
+import os, errno, shutil, subprocess
+
+from base_driver import *
+
+def rm_and_mkdir(dir):
+
+    # Operate only on local directories
+    dir = './' + dir
+
+    # Failing to delete a directory that doesn't exist is no failure
+    def rmtree_onerror(function, path, excinfo):
+        if excinfo[0] == OSError and excinfo[1].errno != errno.ENOENT:
+            raise
+
+    shutil.rmtree(dir, onerror = rmtree_onerror)
+    os.makedirs(dir)
+
+class TrimStressDriver(Driver):
+
+    def trim_stress(self, trace_file):
+        "Execute an apitrace-trim stress test for the given trace."
+
+        (dir, file_name) = os.path.split(trace_file)
+        (name, extension) = file_name.rsplit('.', 1)
+
+        ref_dir = name + '-ref/'
+        out_dir = name + '-out/'
+        trim_file = name + '.trace.trim'
+
+        rm_and_mkdir(ref_dir)
+        rm_and_mkdir(out_dir)
+
+        subprocess.check_call([self.options.apitrace, "dump-images", "--call-nos=no", "--output=" + ref_dir, trace_file]);
+
+        # Count the number of frame snapshots generated
+        frames = 0
+        for dirpath, dirs, files in os.walk(ref_dir):
+            # Don't descend into any sub-directories (not that there
+            # should be any)
+            del dirs[:]
+            frames = len(files)
+
+        for frame in range(frames):
+            try:
+                subprocess.check_call([self.options.apitrace, "trim", "--auto", "--frame=%d" % (frame), "--output=" + trim_file, trace_file])
+            except:
+                print "An error occurred while trimming frame %d from %s" % (frame, trace_file)
+                fail()
+            try:
+                subprocess.check_call([self.options.apitrace, "dump-images", "--call-nos=no", "--output=" + out_dir + "frame", trim_file])
+            except:
+                print "An error occurred replaying %s to generate a frame snapshot" % (trim_file)
+                fail()
+            os.rename("%s/frame0000000000.png" % (out_dir), "%s/%010d.png" % (out_dir, frame))
+
+        try:
+            subprocess.check_call([self.options.apitrace, "diff-images", "-v", ref_dir, out_dir])
+        except:
+            print "Trimmed frames did not match reference images. See " + name + "-index.html"
+            fail()
+        finally:
+            os.rename("index.html", name + "-index.html")
+
+    def run(self):
+        self.parseOptions()
+
+       self.trim_stress(self.args[0])
+
+        pass_()
+
+if __name__ == '__main__':
+    TrimStressDriver().run()