pass
def visitEnum(self, enum):
- print 'static const trace::EnumValue __enum%s_values[] = {' % (enum.tag)
+ print 'static const trace::EnumValue _enum%s_values[] = {' % (enum.tag)
for value in enum.values:
print ' {"%s", %s},' % (value, value)
print '};'
print
- print 'static const trace::EnumSig __enum%s_sig = {' % (enum.tag)
- print ' %u, %u, __enum%s_values' % (enum.id, len(enum.values), enum.tag)
+ print 'static const trace::EnumSig _enum%s_sig = {' % (enum.tag)
+ print ' %u, %u, _enum%s_values' % (enum.id, len(enum.values), enum.tag)
print '};'
print
def visitBitmask(self, bitmask):
- print 'static const trace::BitmaskFlag __bitmask%s_flags[] = {' % (bitmask.tag)
+ print 'static const trace::BitmaskFlag _bitmask%s_flags[] = {' % (bitmask.tag)
for value in bitmask.values:
print ' {"%s", %s},' % (value, value)
print '};'
print
- print 'static const trace::BitmaskSig __bitmask%s_sig = {' % (bitmask.tag)
- print ' %u, %u, __bitmask%s_flags' % (bitmask.id, len(bitmask.values), bitmask.tag)
+ print 'static const trace::BitmaskSig _bitmask%s_sig = {' % (bitmask.tag)
+ print ' %u, %u, _bitmask%s_flags' % (bitmask.id, len(bitmask.values), bitmask.tag)
print '};'
print
print ' _write__%s(%s);' % (struct.tag, instance)
def visitArray(self, array, instance):
- length = '__c' + array.type.tag
- index = '__i' + array.type.tag
+ length = '_c' + array.type.tag
+ index = '_i' + array.type.tag
print ' if (%s) {' % instance
- print ' size_t %s = %s;' % (length, array.length)
+ print ' size_t %s = %s > 0 ? %s : 0;' % (length, array.length, array.length)
print ' trace::localWriter.beginArray(%s);' % length
print ' for (size_t %s = 0; %s < %s; ++%s) {' % (index, index, length, index)
print ' trace::localWriter.beginElement();'
print ' trace::localWriter.writeBlob(%s, %s);' % (instance, blob.size)
def visitEnum(self, enum, instance):
- print ' trace::localWriter.writeEnum(&__enum%s_sig, %s);' % (enum.tag, instance)
+ print ' trace::localWriter.writeEnum(&_enum%s_sig, %s);' % (enum.tag, instance)
def visitBitmask(self, bitmask, instance):
- print ' trace::localWriter.writeBitmask(&__bitmask%s_sig, %s);' % (bitmask.tag, instance)
+ print ' trace::localWriter.writeBitmask(&_bitmask%s_sig, %s);' % (bitmask.tag, instance)
def visitPointer(self, pointer, instance):
print ' if (%s) {' % instance
if isinstance(elem_type, stdapi.Interface):
self.visitInterfacePointer(elem_type, instance)
else:
- self.visitPointer(self, pointer, instance)
+ self.visitPointer(pointer, instance)
def visitInterface(self, interface, instance):
raise NotImplementedError
self.footer(api)
def header(self, api):
- pass
+ print '#ifdef _WIN32'
+ print '# include <malloc.h> // alloca'
+ print '# ifndef alloca'
+ print '# define alloca _alloca'
+ print '# endif'
+ print '#else'
+ print '# include <alloca.h> // alloca'
+ print '#endif'
+ print
+ print '#include "trace.hpp"'
+ print
def footer(self, api):
pass
def traceFunctionDecl(self, function):
# Per-function declarations
- if function.args:
- print 'static const char * __%s_args[%u] = {%s};' % (function.name, len(function.args), ', '.join(['"%s"' % arg.name for arg in function.args]))
- else:
- print 'static const char ** __%s_args = NULL;' % (function.name,)
- print 'static const trace::FunctionSig __%s_sig = {%u, "%s", %u, __%s_args};' % (function.name, function.id, function.name, len(function.args), function.name)
- print
+ if not function.internal:
+ if function.args:
+ print 'static const char * _%s_args[%u] = {%s};' % (function.name, len(function.args), ', '.join(['"%s"' % arg.name for arg in function.args]))
+ else:
+ print 'static const char ** _%s_args = NULL;' % (function.name,)
+ print 'static const trace::FunctionSig _%s_sig = {%u, "%s", %u, _%s_args};' % (function.name, function.id, function.name, len(function.args), function.name)
+ print
def isFunctionPublic(self, function):
return True
print 'extern "C" PRIVATE'
print function.prototype() + ' {'
if function.type is not stdapi.Void:
- print ' %s __result;' % function.type
+ print ' %s _result;' % function.type
+
+ # No-op if tracing is disabled
+ print ' if (!trace::isTracingEnabled()) {'
+ Tracer.invokeFunction(self, function)
+ if function.type is not stdapi.Void:
+ print ' return _result;'
+ else:
+ print ' return;'
+ print ' }'
+
self.traceFunctionImplBody(function)
if function.type is not stdapi.Void:
- self.wrapRet(function, "__result")
- print ' return __result;'
+ print ' return _result;'
print '}'
print
def traceFunctionImplBody(self, function):
- print ' unsigned __call = trace::localWriter.beginEnter(&__%s_sig);' % (function.name,)
- for arg in function.args:
- if not arg.output:
- self.unwrapArg(function, arg)
- self.serializeArg(function, arg)
- print ' trace::localWriter.endEnter();'
+ if not function.internal:
+ print ' unsigned _call = trace::localWriter.beginEnter(&_%s_sig);' % (function.name,)
+ for arg in function.args:
+ if not arg.output:
+ self.unwrapArg(function, arg)
+ self.serializeArg(function, arg)
+ print ' trace::localWriter.endEnter();'
self.invokeFunction(function)
- print ' trace::localWriter.beginLeave(__call);'
- for arg in function.args:
- if arg.output:
- self.serializeArg(function, arg)
- self.wrapArg(function, arg)
- if function.type is not stdapi.Void:
- self.serializeRet(function, "__result")
- print ' trace::localWriter.endLeave();'
-
- def invokeFunction(self, function, prefix='__', suffix=''):
+ if not function.internal:
+ print ' trace::localWriter.beginLeave(_call);'
+ for arg in function.args:
+ if arg.output:
+ self.serializeArg(function, arg)
+ self.wrapArg(function, arg)
+ if function.type is not stdapi.Void:
+ self.serializeRet(function, "_result")
+ print ' trace::localWriter.endLeave();'
+ if function.type is not stdapi.Void:
+ self.wrapRet(function, "_result")
+
+ def invokeFunction(self, function, prefix='_', suffix=''):
if function.type is stdapi.Void:
result = ''
else:
- result = '__result = '
+ result = '_result = '
dispatch = prefix + function.name + suffix
print ' %s%s(%s);' % (result, dispatch, ', '.join([str(arg.name) for arg in function.args]))
for other_arg in function.args:
if not other_arg.output and other_arg.type is REFIID:
riid = other_arg
- if riid is not None and isinstance(arg.type, stdapi.Pointer):
- assert isinstance(arg.type.type, stdapi.ObjPointer)
- obj_type = arg.type.type.type
- assert obj_type is stdapi.Void
+ if riid is not None \
+ and isinstance(arg.type, stdapi.Pointer) \
+ and isinstance(arg.type.type, stdapi.ObjPointer):
self.wrapIid(function, riid, arg)
return
for method in interface.iterMethods():
print " " + method.prototype() + ";"
print
- self.declareWrapperInterfaceVariables(interface)
+ #print "private:"
+ for type, name, value in self.enumWrapperInterfaceVariables(interface):
+ print ' %s %s;' % (type, name)
print "};"
print
- def declareWrapperInterfaceVariables(self, interface):
- #print "private:"
- print " DWORD m_dwMagic;"
- print " %s * m_pInstance;" % (interface.name,)
+ def enumWrapperInterfaceVariables(self, interface):
+ return [
+ ("DWORD", "m_dwMagic", "0xd8365d6c"),
+ ("%s *" % interface.name, "m_pInstance", "pInstance"),
+ ]
def implementWrapperInterface(self, interface):
self.interface = interface
print '%s::%s(%s * pInstance) {' % (getWrapperInterfaceName(interface), getWrapperInterfaceName(interface), interface.name)
- print ' m_dwMagic = 0xd8365d6c;'
- print ' m_pInstance = pInstance;'
+ for type, name, value in self.enumWrapperInterfaceVariables(interface):
+ print ' %s = %s;' % (name, value)
print '}'
print
print '%s::~%s() {' % (getWrapperInterfaceName(interface), getWrapperInterfaceName(interface))
def implementWrapperInterfaceMethod(self, interface, base, method):
print method.prototype(getWrapperInterfaceName(interface) + '::' + method.name) + ' {'
if method.type is not stdapi.Void:
- print ' %s __result;' % method.type
+ print ' %s _result;' % method.type
self.implementWrapperInterfaceMethodBody(interface, base, method)
if method.type is not stdapi.Void:
- print ' return __result;'
+ print ' return _result;'
print '}'
print
def implementWrapperInterfaceMethodBody(self, interface, base, method):
- print ' static const char * __args[%u] = {%s};' % (len(method.args) + 1, ', '.join(['"this"'] + ['"%s"' % arg.name for arg in method.args]))
- print ' static const trace::FunctionSig __sig = {%u, "%s", %u, __args};' % (method.id, interface.name + '::' + method.name, len(method.args) + 1)
- print ' unsigned __call = trace::localWriter.beginEnter(&__sig);'
+ assert not method.internal
+
+ print ' static const char * _args[%u] = {%s};' % (len(method.args) + 1, ', '.join(['"this"'] + ['"%s"' % arg.name for arg in method.args]))
+ print ' static const trace::FunctionSig _sig = {%u, "%s", %u, _args};' % (method.id, interface.name + '::' + method.name, len(method.args) + 1)
+
+ print ' %s *_this = static_cast<%s *>(m_pInstance);' % (base, base)
+
+ print ' unsigned _call = trace::localWriter.beginEnter(&_sig);'
print ' trace::localWriter.beginArg(0);'
print ' trace::localWriter.writePointer((uintptr_t)m_pInstance);'
print ' trace::localWriter.endArg();'
self.invokeMethod(interface, base, method)
- print ' trace::localWriter.beginLeave(__call);'
+ print ' trace::localWriter.beginLeave(_call);'
for arg in method.args:
if arg.output:
self.serializeArg(method, arg)
self.wrapArg(method, arg)
if method.type is not stdapi.Void:
- print ' trace::localWriter.beginReturn();'
- self.serializeValue(method.type, "__result")
- print ' trace::localWriter.endReturn();'
- self.wrapValue(method.type, '__result')
+ self.serializeRet(method, '_result')
print ' trace::localWriter.endLeave();'
+ if method.type is not stdapi.Void:
+ self.wrapRet(method, '_result')
+
if method.name == 'Release':
assert method.type is not stdapi.Void
- print ' if (!__result)'
+ print ' if (!_result)'
print ' delete this;'
def implementIidWrapper(self, api):
print
def wrapIid(self, function, riid, out):
+ # Cast output arg to `void **` if necessary
+ out_name = out.name
+ obj_type = out.type.type.type
+ if not obj_type is stdapi.Void:
+ assert isinstance(obj_type, stdapi.Interface)
+ out_name = 'reinterpret_cast<void * *>(%s)' % out_name
+
print r' if (%s && *%s) {' % (out.name, out.name)
functionName = function.name
else_ = ''
if self.interface is not None:
functionName = self.interface.name + '::' + functionName
- print r' %sif (*%s == m_pInstance) {' % (else_, out.name,)
- print r' *%s = this;' % (out.name,)
- print r' if (%s) {' % ' && '.join('%s != IID_%s' % (riid.name, iface.name) for iface in self.interface.iterBases())
- print r' warnIID("%s", %s, "unexpected");' % (functionName, riid.name)
- print r' }'
+ print r' if (*%s == m_pInstance &&' % (out_name,)
+ print r' (%s)) {' % ' || '.join('%s == IID_%s' % (riid.name, iface.name) for iface in self.interface.iterBases())
+ print r' *%s = this;' % (out_name,)
print r' }'
else_ = 'else '
print r' %s{' % else_
- print r' wrapIID("%s", %s, %s);' % (functionName, riid.name, out.name)
+ print r' wrapIID("%s", %s, %s);' % (functionName, riid.name, out_name)
print r' }'
print r' }'
if method.type is stdapi.Void:
result = ''
else:
- result = '__result = '
- print ' %sstatic_cast<%s *>(m_pInstance)->%s(%s);' % (result, base, method.name, ', '.join([str(arg.name) for arg in method.args]))
+ result = '_result = '
+ print ' %s_this->%s(%s);' % (result, method.name, ', '.join([str(arg.name) for arg in method.args]))
def emit_memcpy(self, dest, src, length):
- print ' unsigned __call = trace::localWriter.beginEnter(&trace::memcpy_sig);'
+ print ' unsigned _call = trace::localWriter.beginEnter(&trace::memcpy_sig);'
print ' trace::localWriter.beginArg(0);'
print ' trace::localWriter.writePointer((uintptr_t)%s);' % dest
print ' trace::localWriter.endArg();'
print ' trace::localWriter.writeUInt(%s);' % length
print ' trace::localWriter.endArg();'
print ' trace::localWriter.endEnter();'
- print ' trace::localWriter.beginLeave(__call);'
+ print ' trace::localWriter.beginLeave(_call);'
print ' trace::localWriter.endLeave();'
projects / apitrace / blobdiff