RM-autogen-data.py: Generate a simple CSV file
[glsdk/host-tools.git] / respart / RM-autogen-data.py
1 #!/usr/bin/python
2 import subprocess, time, os, sys
3 import argparse, string, re
4 import xlrd, xlwt
5 from xlwt.Utils import rowcol_to_cell
7 COL_COMMENT = 0
8 COL_RES_TYPE = 1
9 COL_SUB_TYPE = 2
10 COL_RES_FIELD = 3
11 COL_RES_COUNT = 5
12 COL_RES_START = 6
13 COL_HOST_START = 7
15 ROW_HOST_ID = 0
16 ROW_RES_START = 2
18 dict_dev = {}
19 dict_subtype = {}
20 dict_host = {}
21 utypes = []
23 def evalcmd(cmd):
24         out = subprocess.check_output(cmd, shell=True).decode("utf-8")
25         return out
27 def gen_soc_py_data(soc):
28         output = '''%s
30 RESASG_TYPE_SHIFT = 6
31 RESASG_SUBTYPE_SHIFT = 0
33 const_values  = {
34 %s
35 %s
36 %s}'''
37         header = "# %s.py - Auto generated SoC data\n" % soc
38         devs = ""
39         for i in sorted(dict_dev.items(), key=lambda x: x[1]):
40                 (dev, dev_id) = i
41                 devs += '"%s" : %d,\n' % (dev, dev_id)
43         subtypes = ""
44         for i in sorted(dict_subtype.items(), key=lambda x: x[1]):
45                 (subtype, subtype_id) = i
46                 subtypes += '"%s" : %d,\n' % (subtype, subtype_id)
48         hosts = ""
49         for i in sorted(dict_host.items(), key=lambda x: x[1]):
50                 (host, host_id) = i
51                 hosts += '"%s" : %d,\n' % (host, host_id)
53         return output % (header, devs, subtypes, hosts)
55 def gen_rm_resasg_csv():
56         output = ""
57         for entry in utypes:
58                 (dev, subtype, start, count) = entry
59                 output += "%s,%s,%d,%d\n" % (dev, subtype, start, count)
60         return output
62 def gen_rm_resasg_sheet(sheet):
64         fmt_header = xlwt.easyxf('font: color-index blue, bold on')
65         fmt_grayed = xlwt.easyxf('font: bold on; align: horiz center; pattern: pattern solid, fore-colour gray25')
67         sheet.write(0, COL_RES_TYPE, "Type", fmt_header)
68         sheet.write(0, COL_SUB_TYPE, "Subtype", fmt_header)
69         sheet.write(0, COL_RES_COUNT, "Availalbe count", fmt_header)
70         sheet.write(0, COL_RES_START, "start", fmt_header)
71         col = COL_HOST_START
72         for item in sorted(dict_host.items(), key=lambda x: x[1]):
73                 (host, host_id) = item
74                 sheet.write(0, col, host, fmt_header)
75                 if (host == "HOST_ID_ALL"):
76                         COL_BALANCE = col
77                 col = col + 1
78         row = 1
79         for entry in utypes:
80                 (dev, subtype, start, count) = entry
81                 sheet.write(row, COL_RES_TYPE, dev)
82                 sheet.write(row, COL_SUB_TYPE, subtype)
83                 sheet.write(row, COL_RES_START, start)
84                 sheet.write(row, COL_RES_COUNT, count)
85                 formula = xlwt.Formula('%s - SUM(%s:%s)' % (
86                         rowcol_to_cell(row, COL_RES_COUNT),
87                         rowcol_to_cell(row, COL_HOST_START),
88                         rowcol_to_cell(row, COL_BALANCE - 1)
89                 ))
90                 sheet.write(row, COL_BALANCE, formula, fmt_grayed)
91                 row = row + 1
94 ################################################################################
95 ##                          Main program starts here                          ##
96 ################################################################################
98 parser = argparse.ArgumentParser(prog='RM-autogen.py', formatter_class=argparse.RawTextHelpFormatter,
99         description='RM-autogen.py - Auto generate the Resource Management data')
101 parser.add_argument('-s', '--soc', required=True, dest='soc',
102         action='store', choices=['j721e', 'am65x'],
103         help='SoC name')
105 parser.add_argument('-o', '--output', required=True, dest='output',
106         action='store',
107         help='output file name')
109 parser.add_argument('--sysfw_path', required=True, dest='prefix',
110         action='store',
111         help='Path to system firmware repo')
113 args = parser.parse_args()
114 print(args)
116 # Parse docuemntation and extract host_id defines
117 output = evalcmd('cat %s/output/%s/sec_proxy/hosts.h | grep "#define HOST_ID" | awk -F"[ ()]*" \'{print $2" " $3}\'' % (args.prefix, args.soc))
118 for line in output.split('\n'):
119         if (line == ''):
120                 continue
121         (host, host_id) = line.split(' ')
122         host_id = int(host_id.strip(string.ascii_letters), 0)
123         dict_host[host] = host_id
125 # Parse docuemntation and extract dev_id and subtype defines
126 output = evalcmd('cat %s/output/%s/rm/resasg_types.rst | grep  -v "\------" | grep -A100000 "+======" | tail -n +2' % (args.prefix, args.soc))
127 dev = dev_id = None
128 for line in output.split('\n'):
129         array = line.replace(' ', '').split('|')
130         if(len(array) == 1):
131                 continue
133         if (array[1] != ''):
134                 (x, dev, dev_id, subtype, subtype_id, utype_id, start, count, x) = array
135         elif (array[3] != ''):
136                 (x, x, x, subtype, subtype_id, utype_id, start, count, x) = array
137         else:
138                 (x, x, x, x, x, x, start, count, x) = array
139         start=int(start, 0)
140         count=int(count, 0)
142         #print (dev, dev_id, subtype, subtype_id, utype_id, start, count)
143         dict_dev[dev] = int(dev_id, 0)
144         dict_subtype[subtype] = int(subtype_id, 0)
145         utypes.append((dev, subtype, start, count))
147 '''
148 #Parse bordconfig to extract the dev, subtype and range of resources
149 output = evalcmd('cat %s/output/%s/rm/boardcfg_rm_data.c | grep -A100000 "resasg_entries" | awk \'BEGIN { FS="\\n"; RS="{" } { print $2 " " $3 " " $4 }\' | awk -F" " \'{print $3 " " $4 " " $7 " " $10}\'' % (args.prefix, args.soc))
150 for line in output.split('\n'):
151         if (line == ''):
152                 continue
153         match = re.match("RESASG_UTYPE.(.*), (.*)., (.*)U, (.*)U,.*", line)
154         if (match == None):
155                 continue
156         (dev, subtype, start, count) = match.groups(0)
157         start = int(start)
158         count = int(count)
159         #print((dev, subtype, start, count))
160         utypes.append((dev, subtype, start, count))
161 '''
163 #print(dict_dev)
164 #print(dict_subtype)
165 #print(utypes)
167 #Generate the soc data defines
168 data = gen_soc_py_data(args.soc)
169 ofile = open("%s.py" % args.soc, "w")
170 ofile.write(data)
171 ofile.close()
173 #Generate the csv file with default values
174 data = gen_rm_resasg_csv()
175 ofile = open("%s.csv" % args.soc, "w")
176 ofile.write(data)
177 ofile.close()
179 #Generate the excel sheet with default values
180 workbook = xlwt.Workbook()
181 sheet = workbook.add_sheet(args.soc)
182 gen_rm_resasg_sheet(sheet)
183 workbook.save(args.output)