]> git.cworth.org Git - apitrace/blobdiff - glretrace.py
Fix thinko.
[apitrace] / glretrace.py
index ff22e996acc4e6fc09fe4070a58c3b266b606beb..f809b9e9502bb0b518f1d9f2da1154e14462dad3 100644 (file)
@@ -24,6 +24,9 @@
 ##########################################################################/
 
 
+"""GL retracer generator."""
+
+
 import stdapi
 import glapi
 from retrace import Retracer
@@ -31,76 +34,140 @@ from retrace import Retracer
 
 class GlRetracer(Retracer):
 
-    def filter_function(self, function):
-        return function.name not in [
-            "glNewBufferRegion",
-            "glDeleteBufferRegion",
-            "glReadBufferRegion",
-            "glDrawBufferRegion",
-            "glBufferRegionEnabled",
-        ]
-
     def retrace_function(self, function):
         Retracer.retrace_function(self, function)
 
+    draw_array_function_names = set([
+        "glDrawArrays",
+        "glDrawArraysEXT",
+        "glDrawArraysIndirect",
+        "glDrawArraysInstanced",
+        "glDrawArraysInstancedARB",
+        "glDrawArraysInstancedEXT",
+        "glDrawMeshArraysSUN",
+        "glMultiDrawArrays",
+        "glMultiDrawArraysEXT",
+        "glMultiModeDrawArraysIBM",
+    ])
+
+    draw_elements_function_names = set([
+        "glDrawElements",
+        "glDrawElementsBaseVertex",
+        "glDrawElementsIndirect",
+        "glDrawElementsInstanced",
+        "glDrawElementsInstancedARB",
+        "glDrawElementsInstancedBaseVertex",
+        "glDrawElementsInstancedEXT",
+        "glDrawRangeElements",
+        "glDrawRangeElementsBaseVertex",
+        "glDrawRangeElementsEXT",
+        "glMultiDrawElements",
+        "glMultiDrawElementsBaseVertex",
+        "glMultiDrawElementsEXT",
+        "glMultiModeDrawElementsIBM",
+    ])
+
     def call_function(self, function):
-        if function.name in ("glDrawArrays", "glDrawElements", "glDrawRangeElements", "glMultiDrawElements"):
+        if (function.name in self.draw_array_function_names or 
+            function.name in self.draw_elements_function_names):
             print '    GLint __array_buffer = 0;'
             print '    glGetIntegerv(GL_ARRAY_BUFFER_BINDING, &__array_buffer);'
             print '    if (!__array_buffer) {'
             self.fail_function(function)
             print '    }'
 
+        if function.name in self.draw_elements_function_names:
+            print '    GLint __element_array_buffer = 0;'
+            print '    glGetIntegerv(GL_ELEMENT_ARRAY_BUFFER_BINDING, &__element_array_buffer);'
+            print '    if (!__element_array_buffer) {'
+            self.fail_function(function)
+            print '    }'
+
+        if function.name == "glViewport":
+            print '    if (x + width > __window_width) {'
+            print '        __window_width = x + width;'
+            print '        __reshape_window = true;'
+            print '    }'
+            print '    if (y + height > __window_height) {'
+            print '        __window_height = y + height;'
+            print '        __reshape_window = true;'
+            print '    }'
+
         if function.name == "glEnd":
             print '    insideGlBeginEnd = false;'
+        
         Retracer.call_function(self, function)
+
         if function.name == "glBegin":
             print '    insideGlBeginEnd = true;'
         else:
             # glGetError is not allowed inside glBegin/glEnd
             print '    checkGlError();'
 
+    pointer_function_names = set([
+        "glColorPointer",
+        "glColorPointerEXT",
+        "glEdgeFlagPointer",
+        "glEdgeFlagPointerEXT",
+        "glFogCoordPointer",
+        "glFogCoordPointerEXT",
+        "glIndexPointer",
+        "glIndexPointerEXT",
+        "glMatrixIndexPointerARB",
+        "glNormalPointer",
+        "glNormalPointerEXT",
+        "glSecondaryColorPointer",
+        "glSecondaryColorPointerEXT",
+        "glTexCoordPointer",
+        "glTexCoordPointerEXT",
+        "glVertexAttribLPointer",
+        "glVertexAttribPointer",
+        "glVertexAttribPointerARB",
+        "glVertexAttribPointerNV",
+        "glVertexPointer",
+        "glVertexPointerEXT",
+    ])
 
     def extract_arg(self, function, arg, arg_type, lvalue, rvalue):
-        if function.name in [
-            "glColorPointer",
-            "glEdgeFlagPointer",
-            "glIndexPointer",
-            "glNormalPointer",
-            "glTexCoordPointer",
-            "glVertexPointer",
-            "glFogCoordPointer",
-            "glSecondaryColorPointer",
-            "glVertexAttribPointer",
-        ] and arg.name == 'pointer':
-            self.extract_pointer(function, arg, arg_type, lvalue, rvalue)
-        else:
-            Retracer.extract_arg(self, function, arg, arg_type, lvalue, rvalue)
+        if (function.name in self.pointer_function_names and arg.name == 'pointer' or
+            function.name in self.draw_elements_function_names and arg.name == 'indices'):
+            print '    if (dynamic_cast<Trace::Null *>(&%s)) {' % rvalue
+            print '        %s = 0;' % (lvalue)
+            print '    } else {'
+            print '        %s = (%s)(uintptr_t)(%s);' % (lvalue, arg_type, rvalue)
+            print '    }'
+            return
+
+        if function.name.startswith('glUniform') and function.args[0].name == arg.name == 'location':
+            print '    GLint program = -1;'
+            print '    glGetIntegerv(GL_CURRENT_PROGRAM, &program);'
 
-    def extract_pointer(self, function, arg, arg_type, lvalue, rvalue):
-        print '    if (dynamic_cast<Trace::Null *>(&%s)) {' % rvalue
-        print '        %s = 0;' % (lvalue)
-        print '    } else {'
-        print '        %s = (%s)(uintptr_t)(%s);' % (lvalue, arg_type, rvalue)
-        print '    }'
+        Retracer.extract_arg(self, function, arg, arg_type, lvalue, rvalue)
 
 
 if __name__ == '__main__':
-    print
-    print '#include <string.h>'
-    print '#include <iostream>'
-    print
-    print '#ifdef WIN32'
-    print '#include <windows.h>'
-    print '#endif'
-    print
-    print '#include <GL/glew.h>'
-    print '#include <GL/glut.h>'
-    print
-    print 'static bool double_buffer = false;'
-    print 'static bool insideGlBeginEnd = false;'
-    print
-    print '''
+    print r'''
+#include <string.h>
+#include <stdio.h>
+#include <iostream>
+
+#include "glproc.hpp"
+#include <GL/glut.h>
+
+static bool double_buffer = false;
+static bool insideGlBeginEnd = false;
+
+static int __window_width = 256, __window_height = 256;
+bool __reshape_window = false;
+
+unsigned __frame = 0;
+long long __startTime = 0;
+bool __wait = false;
+
+const char *__compare_prefix = NULL;
+const char *__snapshot_prefix = NULL;
+
+
 static void
 checkGlError(void) {
     if (insideGlBeginEnd) {
@@ -142,45 +209,127 @@ checkGlError(void) {
         std::cerr << error;
         break;
     }
-    std::cerr << "\\n";
+    std::cerr << "\n";
 }
 '''
     api = glapi.glapi
     retracer = GlRetracer()
     retracer.retrace_api(glapi.glapi)
-    print '''
+    print r'''
 
 static Trace::Parser parser;
 
+static void display_noop(void) {
+}
+
+#include "image.hpp"
+
+static void frame_complete(void) {
+    ++__frame;
+    
+    if (!__reshape_window && (__snapshot_prefix || __compare_prefix)) {
+        Image::Image *ref = NULL;
+        if (__compare_prefix) {
+            char filename[PATH_MAX];
+            snprintf(filename, sizeof filename, "%s%04u.png", __compare_prefix, __frame);
+            ref = Image::readPNG(filename);
+            if (!ref) {
+                return;
+            }
+            if (verbosity)
+                std::cout << "Read " << filename << "\n";
+        }
+        
+        Image::Image src(__window_width, __window_height, true);
+        glReadPixels(0, 0, __window_width, __window_height, GL_RGBA, GL_UNSIGNED_BYTE, src.pixels);
+
+        if (__snapshot_prefix) {
+            char filename[PATH_MAX];
+            snprintf(filename, sizeof filename, "%s%04u.png", __snapshot_prefix, __frame);
+            if (src.writePNG(filename) && verbosity) {
+                std::cout << "Wrote " << filename << "\n";
+            }
+        }
+
+        if (ref) {
+            std::cout << "Frame " << __frame << " average precision of " << src.compare(*ref) << " bits\n";
+            delete ref;
+        }
+    }
+
+}
+
 static void display(void) {
     Trace::Call *call;
 
     while ((call = parser.parse_call())) {
-        if (call->name() == "glFlush") {
-            glFlush();
-            return;
-        }
-        
-        if (!retrace_call(*call)) {
-            if (call->name() == "glXSwapBuffers" ||
-                call->name() == "wglSwapBuffers") {
+        const std::string &name = call->name();
+
+        if ((name[0] == 'w' && name[1] == 'g' && name[2] == 'l') ||
+            (name[0] == 'g' && name[1] == 'l' && name[2] == 'X')) {
+            // XXX: We ignore the majority of the OS-specific calls for now
+            if (name == "glXSwapBuffers" ||
+                name == "wglSwapBuffers") {
                 if (double_buffer)
                     glutSwapBuffers();
                 else
                     glFlush();
+                frame_complete();
                 return;
+            } else {
+                continue;
             }
         }
+
+        if (name == "glFlush") {
+            glFlush();
+            if (!double_buffer) {
+                frame_complete();
+            }
+        }
+        
+        retrace_call(*call);
     }
 
+    // Reached the end of trace
     glFlush();
-    glutIdleFunc(NULL);
+
+    long long endTime = OS::GetTime();
+    float timeInterval = (endTime - __startTime) * 1.0E-6;
+
+    std::cout << 
+        "Rendered " << __frame << " frames"
+        " in " <<  timeInterval << " secs,"
+        " average of " << (__frame/timeInterval) << " fps\n";
+
+    if (__wait) {
+        glutDisplayFunc(&display_noop);
+        glutIdleFunc(NULL);
+    } else {
+        exit(0);
+    }
 }
 
 static void idle(void) {
+    if (__reshape_window) {
+        // XXX: doesn't quite work
+        glutReshapeWindow(__window_width, __window_height);
+        __reshape_window = false;
+    }
     glutPostRedisplay();
 }
 
+static void usage(void) {
+    std::cout << 
+        "Usage: glretrace [OPTION] TRACE\n"
+        "Replay TRACE.\n"
+        "\n"
+        "  -c PREFIX    compare against snapshots\n"
+        "  -db          use a double buffer visual\n"
+        "  -s PREFIX    take snapshots\n"
+        "  -v           verbose output\n";
+}
+
 int main(int argc, char **argv)
 {
 
@@ -194,23 +343,31 @@ int main(int argc, char **argv)
 
         if (!strcmp(arg, "--")) {
             break;
-        }
-        else if (!strcmp(arg, "-db")) {
+        } else if (!strcmp(arg, "-c")) {
+            __compare_prefix = argv[++i];
+        } else if (!strcmp(arg, "-db")) {
             double_buffer = true;
+        } else if (!strcmp(arg, "--help")) {
+            usage();
+            return 0;
+        } else if (!strcmp(arg, "-s")) {
+            __snapshot_prefix = argv[++i];
         } else if (!strcmp(arg, "-v")) {
             ++verbosity;
+        } else if (!strcmp(arg, "-w")) {
+            __wait = true;
         } else {
-            std::cerr << "error: unknown option " << arg << "\\n";
+            std::cerr << "error: unknown option " << arg << "\n";
+            usage();
             return 1;
         }
     }
 
     glutInit(&argc, argv);
     glutInitWindowPosition(0, 0);
-    glutInitWindowSize(800, 600);
+    glutInitWindowSize(__window_width, __window_height);
     glutInitDisplayMode(GLUT_DEPTH | GLUT_RGB | (double_buffer ? GLUT_DOUBLE : GLUT_SINGLE));
     glutCreateWindow(argv[0]);
-    glewInit();
 
     glutDisplayFunc(&display);
     glutIdleFunc(&idle);
@@ -221,6 +378,7 @@ int main(int argc, char **argv)
 
     for ( ; i < argc; ++i) {
         if (parser.open(argv[i])) {
+            __startTime = OS::GetTime();
             glutMainLoop();
             parser.close();
         }