#!/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')