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 stdapi import API
  31 from glapi import glapi
  32 from 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 get_function_address(self, function):
  41         return '__%s' % (function.name,)
  42
  43     def wrap_ret(self, function, instance):
  44         if function.name == "wglGetProcAddress":
  45             print '    if (%s) {' % instance
  46
  47             func_dict = dict([(f.name, f) for f in glapi.functions + wglapi.functions])
  48
  49             def handle_case(function_name):
  50                 f = func_dict[function_name]
  51                 ptype = function_pointer_type(f)
  52                 pvalue = function_pointer_value(f)
  53                 print '    %s = (%s)%s;' % (pvalue, ptype, instance)
  54                 print '    %s = (%s)&%s;' % (instance, function.type, f.name);
  55
  56             def handle_default():
  57                 print '    OS::DebugMessage("apitrace: unknown function \\"%s\\"\\n", lpszProc);'
  58
  59             string_switch('lpszProc', func_dict.keys(), handle_case, handle_default)
  60             print '    }'
  61
  62
  63 if __name__ == '__main__':
  64     print
  65     print '#define _GDI32_'
  66     print
  67     print '#include <string.h>'
  68     print '#include <windows.h>'
  69     print
  70     print '#include "trace_write.hpp"'
  71     print '#include "os.hpp"'
  72     print
  73     print '''
  74 static HINSTANCE g_hDll = NULL;
  75
  76 static PROC
  77 __getPublicProcAddress(LPCSTR lpProcName)
  78 {
  79     if (!g_hDll) {
  80         char szDll[MAX_PATH] = {0};
  81
  82         if (!GetSystemDirectoryA(szDll, MAX_PATH)) {
  83             return NULL;
  84         }
  85
  86         strcat(szDll, "\\\\opengl32.dll");
  87
  88         g_hDll = LoadLibraryA(szDll);
  89         if (!g_hDll) {
  90             return NULL;
  91         }
  92     }
  93
  94     return GetProcAddress(g_hDll, lpProcName);
  95 }
  96
  97     '''
  98     print '#include "glproc.hpp"'
  99     print '#include "glsize.hpp"'
 100     print
 101     print 'extern "C" {'
 102     print
 103     api = API()
 104     api.add_api(glapi)
 105     api.add_api(wglapi)
 106     tracer = WglTracer()
 107     tracer.trace_api(api)
 108     print
 109     print '} /* extern "C" */'