git.cworth.org Git - apitrace/blob - d3dshader.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 from stdapi import Type
  28
  29
  30 class D3DShader(Type):
  31
  32     def __init__(self, version):
  33         self.version = version
  34         Type.__init__(self, "const DWORD *")
  35
  36     def decl(self):
  37         print '#include <d3dx9.h>'
  38         print '#include <stdio.h>'
  39         print
  40         print 'void Dump%s(const DWORD *tokens);' % (self.id)
  41
  42     def impl(self):
  43         print '''
  44 typedef HRESULT
  45 (WINAPI *PD3DXDISASSEMBLESHADER)(
  46     CONST DWORD *pShader,
  47     BOOL EnableColorCode,
  48     LPCSTR pComments,
  49     LPD3DXBUFFER *ppDisassembly
  50 );
  51         '''
  52         print 'void Dump%s(const DWORD *tokens)' % (self.id)
  53         print '''{
  54     static BOOL firsttime = TRUE;
  55     static HMODULE hD3DXModule = NULL;
  56     static PD3DXDISASSEMBLESHADER pfnD3DXDisassembleShader = NULL;
  57
  58     if(firsttime) {
  59         if(!hD3DXModule) {
  60             unsigned release;
  61             int version;
  62             for(release = 0; release <= 1; ++release) {
  63                 /* Version 41 corresponds to Mar 2009 version of DirectX Runtime / SDK */
  64                 for(version = 41; version >= 0; --version) {
  65                     char filename[256];
  66                     _snprintf(filename, sizeof(filename),
  67                               "d3dx9%s%s%u.dll", release ? "" : "d", version ? "_" : "", version);
  68                     hD3DXModule = LoadLibraryA(filename);
  69                     if(hD3DXModule)
  70                         goto found;
  71                 }
  72             }
  73 found:
  74             ;
  75         }
  76
  77         if (hD3DXModule)
  78             if(!pfnD3DXDisassembleShader)
  79                 pfnD3DXDisassembleShader = (PD3DXDISASSEMBLESHADER)GetProcAddress(hD3DXModule, "D3DXDisassembleShader");
  80
  81         firsttime = FALSE;
  82     }
  83
  84     if(pfnD3DXDisassembleShader) {
  85         LPD3DXBUFFER pDisassembly = NULL;
  86
  87         if (pfnD3DXDisassembleShader( (DWORD *)tokens, FALSE, NULL, &pDisassembly) == D3D_OK)
  88             Trace::LiteralString((char *)pDisassembly->GetBufferPointer());
  89
  90         if(pDisassembly)
  91             pDisassembly->Release();
  92     }
  93 }
  94 '''
  95
  96     def dump(self, instance):
  97         print '    Dump%s(%s);' % (self.id, instance)