From: José Fonseca <jose.r.fonseca@gmail.com>
Date: Wed, 9 May 2012 18:33:33 +0000 (+0100)
Subject: Cleanup return value wrapping.
X-Git-Url: https://git.cworth.org/git?a=commitdiff_plain;h=3a259b8ee79fcd00c14b379354757e40e04db893;p=apitrace

Cleanup return value wrapping.
---

diff --git a/wrappers/trace.py b/wrappers/trace.py
index e82bd2c..d209051 100644
--- a/wrappers/trace.py
+++ b/wrappers/trace.py
@@ -416,7 +416,6 @@ class Tracer:
             print '    %s _result;' % function.type
         self.traceFunctionImplBody(function)
         if function.type is not stdapi.Void:
-            self.wrapRet(function, "_result")
             print '    return _result;'
         print '}'
         print
@@ -437,6 +436,8 @@ class Tracer:
         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:
@@ -589,11 +590,11 @@ class Tracer:
                 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)'