##########################################################################/
-import base
-import glapi
-
-
-
-class ConstRemover(base.Rebuilder):
+"""GL retracer generator."""
- def visit_const(self, const):
- return const.type
- def visit_opaque(self, opaque):
- expr = opaque.expr
- if expr.startswith('const '):
- expr = expr[6:]
- return base.Opaque(expr)
-
-
-class ValueExtractor(base.Visitor):
+import stdapi
+import glapi
+from retrace import Retracer
+
+
+class GlRetracer(Retracer):
+
+ 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 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 ' }'
- def visit_literal(self, literal, lvalue, rvalue):
- print ' %s = %s;' % (lvalue, rvalue)
+ 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 ' }'
- def visit_alias(self, alias, lvalue, rvalue):
- self.visit(alias.type, lvalue, rvalue)
-
- def visit_enum(self, enum, lvalue, rvalue):
- print ' %s = %s;' % (lvalue, rvalue)
-
- def visit_bitmask(self, bitmask, lvalue, rvalue):
- self.visit(bitmask.type, lvalue, rvalue)
-
- def visit_array(self, array, lvalue, rvalue):
- print ' const Trace::Array *__a%s = dynamic_cast<const Trace::Array *>(&%s);' % (array.id, rvalue)
- print ' if (__a%s) {' % (array.id)
- length = '__a%s->values.size()' % array.id
- print ' %s = new %s[%s];' % (lvalue, array.type, length)
- index = '__i' + array.id
- print ' for(size_t {i} = 0; {i} < {length}; ++{i}) {{'.format(i = index, length = length)
- try:
- self.visit(array.type, '%s[%s]' % (lvalue, index), '*__a%s->values[%s]' % (array.id, index))
- finally:
- print ' }'
- print ' } else {'
- print ' %s = NULL;' % lvalue
+ 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 ' }'
-
- def visit_pointer(self, pointer, lvalue, rvalue):
- # FIXME
- raise NotImplementedError
- def visit_handle(self, handle, lvalue, rvalue):
- self.visit(handle.type, lvalue, "__%s_map[%s]" %(handle.name, rvalue));
- print ' std::cout << "%s " << static_cast<%s>(%s) << " <- " << %s << "\\n";' % (handle.name, handle.type, rvalue, lvalue)
-
- def visit_blob(self, blob, lvalue, rvalue):
- print ' %s = static_cast<%s>((%s).blob());' % (lvalue, blob, rvalue)
-
- def visit_string(self, string, lvalue, rvalue):
- print ' %s = (%s).string();' % (lvalue, rvalue)
+ 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 self.pointer_function_names and arg.name == 'pointer' or
+ function.name in self.draw_elements_function_names and arg.name == 'indices'):
+ self.extract_pointer(function, arg, arg_type, lvalue, rvalue)
+ else:
+ Retracer.extract_arg(self, function, arg, arg_type, lvalue, rvalue)
+
+ 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 ' }'
+if __name__ == '__main__':
+ print r'''
+#include <string.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 __screenshots = 0;
+
+
+static void
+checkGlError(void) {
+ if (insideGlBeginEnd) {
+ return;
+ }
+
+ GLenum error = glGetError();
+ if (error == GL_NO_ERROR) {
+ return;
+ }
+
+ std::cerr << "warning: glGetError() = ";
+ switch (error) {
+ case GL_INVALID_ENUM:
+ std::cerr << "GL_INVALID_ENUM";
+ break;
+ case GL_INVALID_VALUE:
+ std::cerr << "GL_INVALID_VALUE";
+ break;
+ case GL_INVALID_OPERATION:
+ std::cerr << "GL_INVALID_OPERATION";
+ break;
+ case GL_STACK_OVERFLOW:
+ std::cerr << "GL_STACK_OVERFLOW";
+ break;
+ case GL_STACK_UNDERFLOW:
+ std::cerr << "GL_STACK_UNDERFLOW";
+ break;
+ case GL_OUT_OF_MEMORY:
+ std::cerr << "GL_OUT_OF_MEMORY";
+ break;
+ case GL_INVALID_FRAMEBUFFER_OPERATION:
+ std::cerr << "GL_INVALID_FRAMEBUFFER_OPERATION";
+ break;
+ case GL_TABLE_TOO_LARGE:
+ std::cerr << "GL_TABLE_TOO_LARGE";
+ break;
+ default:
+ std::cerr << error;
+ break;
+ }
+ std::cerr << "\n";
+}
+'''
+ api = glapi.glapi
+ retracer = GlRetracer()
+ retracer.retrace_api(glapi.glapi)
+ print r'''
-class ValueWrapper(base.Visitor):
+static Trace::Parser parser;
- def visit_literal(self, literal, lvalue, rvalue):
- pass
+static void display_noop(void) {
+}
- def visit_alias(self, alias, lvalue, rvalue):
- self.visit(alias.type, lvalue, rvalue)
-
- def visit_enum(self, enum, lvalue, rvalue):
- pass
-
- def visit_bitmask(self, bitmask, lvalue, rvalue):
- pass
-
- def visit_array(self, array, lvalue, rvalue):
- print ' const Trace::Array *__a%s = dynamic_cast<const Trace::Array *>(&%s);' % (array.id, rvalue)
- print ' if (__a%s) {' % (array.id)
- length = '__a%s->values.size()' % array.id
- index = '__i' + array.id
- print ' for(size_t {i} = 0; {i} < {length}; ++{i}) {{'.format(i = index, length = length)
- try:
- self.visit(array.type, '%s[%s]' % (lvalue, index), '*__a%s->values[%s]' % (array.id, index))
- finally:
- print ' }'
- print ' }'
-
- def visit_pointer(self, pointer, lvalue, rvalue):
- # FIXME
- raise NotImplementedError
+#include "bmp.hpp"
- def visit_handle(self, handle, lvalue, rvalue):
- print " __%s_map[static_cast<%s>(%s)] = %s;" % (handle.name, handle.type, rvalue, lvalue)
- print ' std::cout << "%s " << static_cast<%s>(%s) << " -> " << %s << "\\n";' % (handle.name, handle.type, rvalue, lvalue)
+static void frame_complete(void) {
+ ++__frame;
- def visit_blob(self, blob, lvalue, rvalue):
- pass
-
- def visit_string(self, string, lvalue, rvalue):
- pass
-
-
-
-def retrace_function(function):
- print 'static void retrace_%s(Trace::Call &call) {' % function.name
- success = True
- for arg in function.args:
- arg.type = ConstRemover().visit(arg.type)
- print ' %s %s;' % (arg.type, arg.name)
- rvalue = 'call.arg("%s")' % (arg.name,)
- lvalue = arg.name
- try:
- ValueExtractor().visit(arg.type, lvalue, rvalue)
- except NotImplementedError:
- success = False
- print ' %s = 0; // FIXME' % arg.name
- if not success:
- print ' std::cerr << "warning: unsupported call %s\\n";' % function.name
- print ' return;'
- arg_names = ", ".join([arg.name for arg in function.args])
- print ' %s(%s);' % (function.name, arg_names)
- for arg in function.args:
- if arg.output:
- arg.type = ConstRemover().visit(arg.type)
- rvalue = 'call.arg("%s")' % (arg.name,)
- lvalue = arg.name
- try:
- ValueWrapper().visit(arg.type, lvalue, rvalue)
- except NotImplementedError:
- print ' // FIXME: %s' % arg.name
- print '}'
- print
-
-
-def retrace_functions(functions):
- for function in functions:
- if function.sideeffects:
- retrace_function(function)
-
- print 'static bool retrace_call(Trace::Call &call) {'
- for function in functions:
- if not function.sideeffects:
- print ' if (call.name == "%s") {' % function.name
- print ' return true;'
- print ' }'
- print
- print ' std::cout << call;'
- print ' std::cout.flush();'
- print
- for function in functions:
- if function.sideeffects:
- print ' if (call.name == "%s") {' % function.name
- print ' retrace_%s(call);' % function.name
- print ' return true;'
- print ' }'
- print ' std::cerr << "warning: unsupported call " << call.name << "\\n";'
- print ' return false;'
- print '}'
- print
-
+ if (__screenshots && !__reshape_window) {
+ char filename[PATH_MAX];
+ snprintf(filename, sizeof filename, "screenshot_%04u.bmp", __frame);
+ unsigned char *pixels = new unsigned char[__window_height*__window_width*4];
+ glReadPixels(0, 0, __window_width, __window_height, GL_RGBA, GL_UNSIGNED_BYTE, pixels);
+ BMP::write(filename, pixels, __window_width, __window_height, __window_width*4);
+ }
-if __name__ == '__main__':
- print
- print '#include <stdlib.h>'
- print '#include <string.h>'
- print '#include <GL/glew.h>'
- print '#include <GL/glut.h>'
- print
- print '#include "trace_parser.hpp"'
- print
- print 'static std::map<GLuint, GLuint> __texture_map;'
- print
+}
- retrace_functions(glapi.glapi.functions)
+static void display(void) {
+ Trace::Call *call;
- print '''
+ while ((call = parser.parse_call())) {
+ if (call->name() == "glFlush") {
+ glFlush();
+ if (!double_buffer) {
+ frame_complete();
+ }
+ }
+
+ if (!retrace_call(*call)) {
+ if (call->name() == "glXSwapBuffers" ||
+ call->name() == "wglSwapBuffers") {
+ if (double_buffer)
+ glutSwapBuffers();
+ else
+ glFlush();
+ frame_complete();
+ return;
+ }
+ }
+ }
-Trace::Parser parser;
+ // Reached the end of trace
+ glFlush();
-static bool insideGlBeginEnd;
+ long long endTime = OS::GetTime();
+ float timeInterval = (endTime - __startTime) * 1.0E-6;
-static void display(void) {
- Trace::Call *call;
-
- while ((call = parser.parse_call())) {
- if (call->name == "glFlush" ||
- call->name == "glXSwapBuffers" ||
- call->name == "wglSwapBuffers") {
- glFlush();
- return;
- }
-
- retrace_call(*call);
-
- if (call->name == "glBegin") {
- insideGlBeginEnd = true;
- }
-
- if (call->name == "glEnd") {
- insideGlBeginEnd = false;
- }
-
- if (!insideGlBeginEnd) {
- GLenum error = glGetError();
- if (error != GL_NO_ERROR) {
- std::cerr << "warning: glGetError() = ";
- switch (error) {
- case GL_INVALID_ENUM:
- std::cerr << "GL_INVALID_ENUM";
- break;
- case GL_INVALID_VALUE:
- std::cerr << "GL_INVALID_VALUE";
- break;
- case GL_INVALID_OPERATION:
- std::cerr << "GL_INVALID_OPERATION";
- break;
- case GL_STACK_OVERFLOW:
- std::cerr << "GL_STACK_OVERFLOW";
- break;
- case GL_STACK_UNDERFLOW:
- std::cerr << "GL_STACK_UNDERFLOW";
- break;
- case GL_OUT_OF_MEMORY:
- std::cerr << "GL_OUT_OF_MEMORY";
- break;
- case GL_INVALID_FRAMEBUFFER_OPERATION:
- std::cerr << "GL_INVALID_FRAMEBUFFER_OPERATION";
- break;
- case GL_TABLE_TOO_LARGE:
- std::cerr << "GL_TABLE_TOO_LARGE";
- break;
- default:
- std::cerr << error;
- break;
- }
- std::cerr << "\\n";
- }
- }
- }
+ std::cout <<
+ "Rendered " << __frame << " frames"
+ " in " << timeInterval << " secs,"
+ " average of " << (__frame/timeInterval) << " fps\n";
- glFlush();
- glutIdleFunc(NULL);
+ glutDisplayFunc(&display_noop);
+ glutIdleFunc(NULL);
}
static void idle(void) {
- glutPostRedisplay();
+ if (__reshape_window) {
+ // XXX: doesn't quite work
+ glutReshapeWindow(__window_width, __window_height);
+ __reshape_window = false;
+ }
+ glutPostRedisplay();
}
int main(int argc, char **argv)
{
- glutInit(&argc, argv);
- glutInitWindowPosition(0, 0);
- glutInitWindowSize(800, 600);
- glutInitDisplayMode(GLUT_DEPTH | GLUT_RGB | GLUT_SINGLE);
- glutCreateWindow(argv[0]);
- glewInit();
-
- glutDisplayFunc(&display);
- glutIdleFunc(&idle);
-
- for (int i = 1; i < argc; ++i) {
- if (parser.open(argv[i])) {
- glutMainLoop();
- parser.close();
- }
- }
-
- return 0;
+
+ int i;
+ for (i = 1; i < argc; ++i) {
+ const char *arg = argv[i];
+
+ if (arg[0] != '-') {
+ break;
+ }
+
+ if (!strcmp(arg, "--")) {
+ break;
+ } else if (!strcmp(arg, "-db")) {
+ double_buffer = true;
+ } else if (!strcmp(arg, "-s")) {
+ __screenshots = true;
+ } else if (!strcmp(arg, "-v")) {
+ ++verbosity;
+ } else {
+ std::cerr << "error: unknown option " << arg << "\n";
+ return 1;
+ }
+ }
+
+ glutInit(&argc, argv);
+ glutInitWindowPosition(0, 0);
+ glutInitWindowSize(__window_width, __window_height);
+ glutInitDisplayMode(GLUT_DEPTH | GLUT_RGB | (double_buffer ? GLUT_DOUBLE : GLUT_SINGLE));
+ glutCreateWindow(argv[0]);
+
+ glutDisplayFunc(&display);
+ glutIdleFunc(&idle);
+
+ for (GLuint h = 0; h < 1024; ++h) {
+ __list_map[h] = h;
+ }
+
+ for ( ; i < argc; ++i) {
+ if (parser.open(argv[i])) {
+ __startTime = OS::GetTime();
+ glutMainLoop();
+ parser.close();
+ }
+ }
+
+ return 0;
}
-'''
+'''