From 9a2ff1bac0adb498fb56453b650877bd0aa70bfe Mon Sep 17 00:00:00 2001
From: Darren Ranalli <github@dranalli.com>
Date: Thu, 24 Jun 2004 22:45:40 +0000
Subject: [PATCH] added caller/callee printout to printProfile

---
 direct/src/showbase/PythonUtil.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/direct/src/showbase/PythonUtil.py b/direct/src/showbase/PythonUtil.py
index a3a144f546..8c7a577ee6 100644
--- a/direct/src/showbase/PythonUtil.py
+++ b/direct/src/showbase/PythonUtil.py
@@ -589,29 +589,34 @@ PyUtilProfileDefaultSorts = ['cumulative', 'time', 'calls']
 #            self.load()
 #        import __builtin__
 #        __builtin__.func = func
-#        PythonUtil.startProfile(cmd='func()', filename='loadProfile')
+#        PythonUtil.startProfile(cmd='func()', filename='profileData')
 #        del __builtin__.func
 #
 def startProfile(filename=PyUtilProfileDefaultFilename,
                  lines=PyUtilProfileDefaultLines,
                  sorts=PyUtilProfileDefaultSorts,
                  silent=0,
+                 callInfo=1,
                  cmd='run()'):
     import profile
     profile.run(cmd, filename)
     if not silent:
-        printProfile(filename, lines, sorts)
+        printProfile(filename, lines, sorts, callInfo)
 
 # call this to see the results again
 def printProfile(filename=PyUtilProfileDefaultFilename,
                  lines=PyUtilProfileDefaultLines,
-                 sorts=PyUtilProfileDefaultSorts,):
+                 sorts=PyUtilProfileDefaultSorts,
+                 callInfo=1):
     import pstats
     s = pstats.Stats(filename)
     s.strip_dirs()
     for sort in sorts:
         s.sort_stats(sort)
         s.print_stats(lines)
+        if callInfo:
+            s.print_callees(lines)
+            s.print_callers(lines)
 
 class Functor:
     def __init__(self, function, *args, **kargs):