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