git.cworth.org Git - apitrace/blob - wgltrace.py

   1 ##########################################################################
   2 #
   3 # Copyright 2008-2009 VMware, Inc.
   4 # All Rights Reserved.
   5 #
   6 # Permission is hereby granted, free of charge, to any person obtaining a copy
   7 # of this software and associated documentation files (the "Software"), to deal
   8 # in the Software without restriction, including without limitation the rights
   9 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10 # copies of the Software, and to permit persons to whom the Software is
  11 # furnished to do so, subject to the following conditions:
  12 #
  13 # The above copyright notice and this permission notice shall be included in
  14 # all copies or substantial portions of the Software.
  15 #
  16 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22 # THE SOFTWARE.
  23 #
  24 ##########################################################################/
  25
  26
  27 """WGL tracing code generator."""
  28
  29
  30 from specs.stdapi import API
  31 from specs.glapi import glapi
  32 from specs.wglapi import wglapi
  33 from dispatch import function_pointer_type, function_pointer_value
  34 from gltrace import GlTracer
  35 from codegen import *
  36
  37
  38 class WglTracer(GlTracer):
  39
  40     def wrap_ret(self, function, instance):
  41         GlTracer.wrap_ret(self, function, instance)
  42
  43         if function.name == "wglGetProcAddress":
  44             print '    if (%s) {' % instance
  45
  46             func_dict = dict([(f.name, f) for f in glapi.functions + wglapi.functions])
  47
  48             def handle_case(function_name):
  49                 f = func_dict[function_name]
  50                 ptype = function_pointer_type(f)
  51                 pvalue = function_pointer_value(f)
  52                 print '    %s = (%s)%s;' % (pvalue, ptype, instance)
  53                 print '    %s = (%s)&%s;' % (instance, function.type, f.name);
  54
  55             def handle_default():
  56                 print '    os::log("apitrace: warning: unknown function \\"%s\\"\\n", lpszProc);'
  57
  58             string_switch('lpszProc', func_dict.keys(), handle_case, handle_default)
  59             print '    }'
  60
  61
  62 if __name__ == '__main__':
  63     print
  64     print '#define _GDI32_'
  65     print
  66     print '#include <string.h>'
  67     print '#include <windows.h>'
  68     print
  69     print '#include "trace_writer.hpp"'
  70     print '#include "os.hpp"'
  71     print
  72     print '''
  73 static HINSTANCE g_hDll = NULL;
  74
  75 PROC
  76 __getPublicProcAddress(LPCSTR lpProcName)
  77 {
  78     if (!g_hDll) {
  79         char szDll[MAX_PATH] = {0};
  80
  81         if (!GetSystemDirectoryA(szDll, MAX_PATH)) {
  82             return NULL;
  83         }
  84
  85         strcat(szDll, "\\\\opengl32.dll");
  86
  87         g_hDll = LoadLibraryA(szDll);
  88         if (!g_hDll) {
  89             return NULL;
  90         }
  91     }
  92
  93     return GetProcAddress(g_hDll, lpProcName);
  94 }
  95
  96     '''
  97     print '// To validate our prototypes'
  98     print '#define GL_GLEXT_PROTOTYPES'
  99     print '#define WGL_GLXEXT_PROTOTYPES'
 100     print
 101     print '#include "glproc.hpp"'
 102     print '#include "glsize.hpp"'
 103     print
 104     api = API()
 105     api.add_api(glapi)
 106     api.add_api(wglapi)
 107     tracer = WglTracer()
 108     tracer.trace_api(api)