132 lines
5.3 KiB
Python
132 lines
5.3 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# encoding=utf-8
|
|||
|
|
# ============================================================================
|
|||
|
|
# Copyright (c) HiSilicon (Shanghai) Technologies Co., Ltd. 2022-2022. All rights reserved.
|
|||
|
|
# Description: Provides dma driver source \n
|
|||
|
|
#
|
|||
|
|
# History: \n
|
|||
|
|
# 2023-01-16， Create file. \n
|
|||
|
|
# ============================================================================
|
|||
|
|
import os
|
|||
|
|
import shutil
|
|||
|
|
import re
|
|||
|
|
import sys
|
|||
|
|
root_dir = os.path.split(os.path.realpath(__file__))[0]
|
|||
|
|
sys.path.append(os.path.join(root_dir, '..', 'config'))
|
|||
|
|
sys.path.append(root_dir)
|
|||
|
|
from enviroment import TargetEnvironment
|
|||
|
|
from utils.build_utils import output_root
|
|||
|
|
|
|||
|
|
class statistics(object):
|
|||
|
|
def __init__(self, file):
|
|||
|
|
self.name = []
|
|||
|
|
self.text_size = []
|
|||
|
|
self.data_size = []
|
|||
|
|
self.bss_size = []
|
|||
|
|
self.group = [[0 for i in range(6)] for j in range(100)]
|
|||
|
|
with open(file, encoding="utf-8") as f:
|
|||
|
|
cluster = f.read()
|
|||
|
|
component_name = cluster[cluster.find("[")+1:cluster.find("]")]
|
|||
|
|
component_name = component_name.replace("\n", '')
|
|||
|
|
component_name = component_name.replace(" ", '')
|
|||
|
|
component_name = component_name.replace("'", '')
|
|||
|
|
self.name = component_name.split(",")
|
|||
|
|
for i in range(0, len(self.name)):
|
|||
|
|
self.text_size.append(0)
|
|||
|
|
self.data_size.append(0)
|
|||
|
|
self.bss_size.append(0)
|
|||
|
|
group_name = cluster[cluster.find("group_details = {")+20:cluster.find("}")]
|
|||
|
|
group_name = group_name.split("\n")
|
|||
|
|
for i in range(0, len(group_name)):
|
|||
|
|
name_g = group_name[i].replace("[", '')
|
|||
|
|
name_g = name_g.replace("]", '')
|
|||
|
|
name_g = name_g.replace(" ", '')
|
|||
|
|
name_g = name_g.replace("'", '')
|
|||
|
|
name_g = re.split(':|,', name_g)
|
|||
|
|
if (len(name_g) > 0):
|
|||
|
|
self.group[i] = name_g
|
|||
|
|
|
|||
|
|
def data_set(self, file_name, index):
|
|||
|
|
cmds = 'size ' + file_name
|
|||
|
|
string = os.popen(cmds).read()
|
|||
|
|
for line in string.splitlines():
|
|||
|
|
number = re.findall("\d+",line)
|
|||
|
|
if len(number) > 0:
|
|||
|
|
self.text_size[index] = int(self.text_size[index]) + int(number[0])
|
|||
|
|
self.data_size[index] = int(self.data_size[index]) + int(number[1])
|
|||
|
|
self.bss_size[index] = int(self.bss_size[index]) + int(number[2])
|
|||
|
|
|
|||
|
|
def data_find(self, file_name):
|
|||
|
|
file_name = file_name.replace(".a", '')
|
|||
|
|
file_name = file_name.replace("lib", '')
|
|||
|
|
for i in range(0, len(self.group)):
|
|||
|
|
if len(self.group[i]) == 0:
|
|||
|
|
break
|
|||
|
|
for j in range(1, len(self.group[i])):
|
|||
|
|
if (file_name == self.group[i][j]):
|
|||
|
|
file_name = self.group[i][0]
|
|||
|
|
for i in range(0, len(self.name)):
|
|||
|
|
if self.name[i] == file_name:
|
|||
|
|
return i
|
|||
|
|
return -1
|
|||
|
|
|
|||
|
|
def creat(self, path):
|
|||
|
|
with open(path + '/codesize_statistic.html', "w") as f:
|
|||
|
|
f.write('<h1>Codesize Statistics</h1>')
|
|||
|
|
f.write('<table border="1" width = "40%" cellspacing="0" cellpadding="0" align="left">')
|
|||
|
|
f.write('<tr><th>drivers</th><th>text</th><th>data</th><th>bss</th><tr>\n')
|
|||
|
|
|
|||
|
|
total_text_size = 0
|
|||
|
|
total_data_size = 0
|
|||
|
|
total_bss_size = 0
|
|||
|
|
|
|||
|
|
for i in range(0, len(self.name)):
|
|||
|
|
|
|||
|
|
total = self.text_size[i] + self.data_size[i] + self.bss_size[i]
|
|||
|
|
if not self.name[i] == 0 and total > 0:
|
|||
|
|
total_text_size += self.text_size[i]
|
|||
|
|
total_data_size += self.data_size[i]
|
|||
|
|
total_bss_size += self.bss_size[i]
|
|||
|
|
|
|||
|
|
result = '<tr><td>%s</td><td>%s</td><td>%s</td><td>%s</td></tr>\n' % \
|
|||
|
|
(str(self.name[i]), str(self.text_size[i]), str(self.data_size[i]), str(self.bss_size[i]))
|
|||
|
|
f.write(result)
|
|||
|
|
result = '<tr><td>Total</td><td>%s</td><td>%s</td><td>%s</td></tr>\n' % \
|
|||
|
|
(str(total_text_size), str(total_data_size), str(total_bss_size))
|
|||
|
|
f.write(result)
|
|||
|
|
f.write('</table>')
|
|||
|
|
|
|||
|
|
def find_listdir(stat, path):
|
|||
|
|
if os.path.exists(path):
|
|||
|
|
file_list = os.listdir(path)
|
|||
|
|
for file in file_list:
|
|||
|
|
fulldir = os.path.join(path, file)
|
|||
|
|
if os.path.isdir(fulldir):
|
|||
|
|
find_listdir(stat, fulldir)
|
|||
|
|
if os.path.isfile(fulldir):
|
|||
|
|
bool = file.endswith(".a")
|
|||
|
|
if bool:
|
|||
|
|
i = stat.data_find(file)
|
|||
|
|
if i >= 0:
|
|||
|
|
org_work_path = os.getcwd()
|
|||
|
|
os.chdir(path)
|
|||
|
|
stat.data_set(file, i)
|
|||
|
|
os.chdir(org_work_path)
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
root_dir = os.path.split(os.path.realpath(__file__))[0]
|
|||
|
|
root_dir = os.path.join(root_dir, '..', '..')
|
|||
|
|
os.chdir(root_dir)
|
|||
|
|
sys.path.append(os.path.join(root_dir, 'build', 'config'))
|
|||
|
|
sys.path.append(os.path.join(root_dir, 'build', 'script'))
|
|||
|
|
|
|||
|
|
env = TargetEnvironment(sys.argv[1])
|
|||
|
|
chip = env.get('chip')
|
|||
|
|
core = env.get('core')
|
|||
|
|
stat = statistics("./build/config/target_config/%s/codesize_statistic_config.json" % (chip))
|
|||
|
|
path = os.path.join(output_root, chip, core, sys.argv[1])
|
|||
|
|
find_listdir(stat, path)
|
|||
|
|
stat.creat(path)
|
|||
|
|
|
|||
|
|
print('codesize statistics done')
|