Android学习：dmtracedump 的替代品


使用 traceview 分析代码性能， google文档上说有一个工具 dmtracedump，可以生成函数调用图。 试验了一下，失败！

google 一下，原来此工具一直处于 broken 状态。 以前玩过 KFI, 一同好还自己开发了工具去生成jpg的函数调用图。 于是，花几个小时写了一个脚本，放在这里，方便同好。

  1. #!/usr/bin/env python   
  2. """ 
  3. turn the traceview data into a jpg pic, showing methods call relationship 
  4. """  
  5. import sys  
  6. import os  
  7. import struct  
  8. import re  
  9. ################################################################################   
  10. ########################  Global Variable  #####################################   
  11. ################################################################################   
  12. target_thread=1 #the thread that we want to track, filt out other threads   
  13. #all_actions = ["enter","exit","exception","reserved"]   
  14. all_threads = {}  
  15. all_methods = {}  
  16. all_records = []  
  17. parent_methods = {}  
  18. child_methods = {}  
  19. method_calls = {}  
  20. ################################################################################   
  21. ##############################   Methods   #####################################   
  22. ################################################################################   
  23. def add_one_thread(line):  
  24.     fields = line.split("/t")  
  25.     all_threads[int(fields[0],10)]=fields  
  26. def add_one_method(line):  
  27.     fields = line.split("/t")  
  28.     all_methods[int(fields[0],16)]=fields  
  29. def add_one_record(one):  
  30.     thread_id,=struct.unpack("B",one[:1])  
  31.     if (thread_id == target_thread):  
  32.         tmp,=struct.unpack("L",one[1:5])  
  33.         method_id= (tmp / 4) * 4;  
  34.         method_action= tmp % 4;  
  35.         time_offset,=struct.unpack("L",one[5:])  
  36.         all_records.append([thread_id, method_id, method_action, time_offset])  
  37. def handle_one_call(parent_method_id,method_id):  
  38.     if not (parent_methods.has_key(parent_method_id)):  
  39.         parent_methods[parent_method_id]=1  
  40.     if not (child_methods.has_key(method_id)):  
  41.         child_methods[method_id]=1  
  42.     if method_calls.has_key(parent_method_id):  
  43.         if method_calls[parent_method_id].has_key(method_id):  
  44.             method_calls[parent_method_id][method_id]+=1  
  45.         else:  
  46.             method_calls[parent_method_id][method_id]=1  
  47.     else:  
  48.         method_calls[parent_method_id]={}  
  49.         method_calls[parent_method_id][method_id]=1  
  50. def gen_funcname(method_id):  
  51.     r1=re.compile(r'[/$<>]')  
  52.     str1=r1.sub("_",all_methods[method_id][1])  
  53.     str2=r1.sub("_",all_methods[method_id][2])  
  54.     return str1+"_"+str2  
  55. def gen_dot_script_file():  
  56.     myoutfile = open("graph.dot", "w")  
  57.     myoutfile.write("digraph vanzo {/n/n");  
  58.     for one in all_methods.keys():  
  59.         if parent_methods.has_key(one):  
  60.             myoutfile.write(gen_funcname(one)+"  [shape=rectangle];/n")  
  61.         else:  
  62.             if child_methods.has_key(one):  
  63.                 myoutfile.write(gen_funcname(one)+"  [shape=ellipse];/n")  
  64.     for one in method_calls.keys():  
  65.         for two in method_calls[one]:  
  66.                 myoutfile.write(gen_funcname(one) + ' -> ' + gen_funcname(two) +  ' [label="' + str(method_calls[one][two]) + '"  fontsize="10"];/n')  
  67.     myoutfile.write("/n}/n");  
  68.     myoutfile.close  
  69.       
  70. ################################################################################   
  71. ########################## Script starts from here #############################   
  72. ################################################################################   
  73. if len(sys.argv) < 2:    
  74.     print 'No input file specified.'    
  75.     sys.exit()  
  76. if not (os.path.exists(sys.argv[1])):  
  77.     print "input file not exists"  
  78.     sys.exit()  
  79. #Now handle the text part   
  80. current_section=0  
  81. for line in open(sys.argv[1]):  
  82.     line2 = line.strip()  
  83.     if (line2.startswith("*")):  
  84.         if (line2.startswith("*version")):  
  85.             current_section=1  
  86.         else:  
  87.             if (line2.startswith("*threads")):  
  88.                 current_section=2  
  89.             else:  
  90.                 if (line2.startswith("*methods")):  
  91.                     current_section=3  
  92.                 else:   
  93.                     if (line2.startswith("*end")):  
  94.                         current_section=4  
  95.                         break  
  96.         continue  
  97.     if current_section==2:  
  98.         add_one_thread(line2)  
  99.     if current_section==3:  
  100.         add_one_method(line2)  
  101.       
  102. #Now handle the binary part   
  103. mybinfile = open(sys.argv[1], "rb")   
  104. alldata = mybinfile.read()  
  105. mybinfile.close()  
  106. pos=alldata.find("SLOW")  
  107. offset,=struct.unpack("H",alldata[pos+6:pos+8])  
  108. pos2=pos+offset #pos2 is where the record begin   
  109. numofrecords = len(alldata) - pos2  
  110. numofrecords = numofrecords / 9  
  111. for i in xrange(numofrecords):  
  112.     add_one_record(alldata[pos2 + i * 9:pos2 + i * 9 + 9])  
  113. my_stack=[0]  
  114. for onerecord in all_records:  
  115.     thread_id=onerecord[0];      
  116.     method_id=onerecord[1];  
  117.     action=onerecord[2];  
  118.     time=onerecord[3];  
  119.     if(action==0):  
  120.         if(len(my_stack) > 1):  
  121.             parent_method_id=my_stack[-1]  
  122.             handle_one_call(parent_method_id,method_id)  
  123.         my_stack.append(method_id)  
  124.     else:  
  125.         if(action==1):  
  126.             if(len(my_stack) > 1):  
  127.                 my_stack.pop()  
  128. gen_dot_script_file()  
  129. os.system("dot -Tjpg graph.dot -o output.jpg;rm -f graph.dot");  

相关内容