python/retrace: Support gziped traces.

diff --git a/src/gallium/state_trackers/python/retrace/interpreter.py b/src/gallium/state_trackers/python/retrace/interpreter.py
index 7502c096b66d54c10aeca0c6f409671373a26823..f596a54d10242a0dc9bb833b19e406b87cbe948b 100755 (executable)
--- a/src/gallium/state_trackers/python/retrace/interpreter.py
+++ b/src/gallium/state_trackers/python/retrace/interpreter.py
@@ -471,7 +471,12 @@ class Interpreter:
 
 def main():
     for arg in sys.argv[1:]:
-        parser = TraceParser(open(arg, 'rt'))
+        if arg.endswith('.gz'):
+            import gzip
+            stream = gzip.GzipFile(arg, 'rt')
+        else:
+            stream = open(arg, 'rt')
+        parser = TraceParser(stream)
         trace = parser.parse()
         interpreter = Interpreter()
         interpreter.interpret(trace)

diff --git a/src/gallium/state_trackers/python/retrace/parser.py b/src/gallium/state_trackers/python/retrace/parser.py
index d02fd7f741f3cc4919b87bbabca912afd4010451..2ee4d3068db7247665c5826b24368713385a79bf 100755 (executable)
--- a/src/gallium/state_trackers/python/retrace/parser.py
+++ b/src/gallium/state_trackers/python/retrace/parser.py
@@ -322,9 +322,15 @@ class TraceParser(XmlParser):
 
 def main():
     for arg in sys.argv[1:]:
-        parser = TraceParser(open(arg, 'rt'))
+        if arg.endswith('.gz'):
+            import gzip
+            stream = gzip.GzipFile(arg, 'rt')
+        else:
+            stream = open(arg, 'rt')
+        parser = TraceParser(stream)
         trace = parser.parse()
-        print trace
+        for call in trace.calls:
+            print call
 
 
 if __name__ == '__main__':