389 lines
15 KiB
Python
389 lines
15 KiB
Python
#!/usr/bin/env python
|
||
#encoding=utf-8
|
||
import copy
|
||
import random
|
||
|
||
class GenPkt(object):
|
||
def __init__(self):
|
||
super(GenPkt, self).__init__()
|
||
|
||
self.pkt_buf = list() #bytearray(b'\0' * 2048)
|
||
self.pkt_type = 0
|
||
self.pkt_sub_pkt = 0
|
||
self.sub_pkt = list()
|
||
self.pkt_data = None
|
||
|
||
def clean(self):
|
||
self.pkt_buf = list() #bytearray(b'\0' * 2048)
|
||
self.pkt_type = 0
|
||
self.pkt_sub_pkt = 0
|
||
self.sub_pkt = list()
|
||
self.pkt_data = None
|
||
|
||
def setPktType(self, type):
|
||
if (type != 0) and (type != 1):
|
||
print('Error...set type failed:', type)
|
||
else:
|
||
self.pkt_type = type
|
||
|
||
def setPkt(self, type, data = None):
|
||
if (type != 0) and (type != 1):
|
||
if type == 2:
|
||
self.pkt_type = 2
|
||
self.pkt_data = copy.copy(data)
|
||
else:
|
||
print('Error...set type failed:', type)
|
||
else:
|
||
self.pkt_type = type
|
||
|
||
def addBeforeExc(self, str):
|
||
self.sub_pkt.append([1, str])
|
||
|
||
def addAfterExc(self, str):
|
||
self.sub_pkt.append([2, str])
|
||
|
||
def addSubPkt(self, dev_idx, ptype, addr, cmd, len, data):
|
||
if (len > 2032):
|
||
print('Error, bad subpkt len:%d', len)
|
||
return
|
||
if not (ptype in [0, 2]):
|
||
print('Error, bad subpkt type:', ptype)
|
||
return
|
||
if not (cmd in [0, 1, 2, 0xb, 0xc, 0xd, 0xe, 0xf]):
|
||
print('Error, bad cmd:', cmd)
|
||
return
|
||
|
||
self.pkt_sub_pkt = self.pkt_sub_pkt + 1
|
||
self.sub_pkt.append([0, dev_idx, ptype, addr, cmd, len, data])
|
||
|
||
def gen(self, filename="pkt_verilog_test.v", filetype=0, appendgen=False, gen_name=None):
|
||
pkt_len = 4
|
||
for sp in self.sub_pkt:
|
||
if sp[0] != 0:
|
||
pass
|
||
else:
|
||
pkt_len = pkt_len + sp[5] + 8
|
||
|
||
if pkt_len > 2044:
|
||
print('Error, bad pkt_len:', pkt_len)
|
||
return
|
||
|
||
i = 4
|
||
if (self.pkt_type == 1):
|
||
pkt_len = 4
|
||
'''
|
||
self.pkt_buf[0] = 0
|
||
self.pkt_buf[1] = (self.pkt_type & 0x3)
|
||
self.pkt_buf[2] = 0x3f
|
||
self.pkt_buf[3] = 0xff
|
||
'''
|
||
self.pkt_buf.append(0)
|
||
self.pkt_buf.append(self.pkt_type & 0x3)
|
||
self.pkt_buf.append(0x3f)
|
||
self.pkt_buf.append(0xff)
|
||
elif (self.pkt_type == 2):
|
||
pkt_len = len(self.pkt_data)
|
||
|
||
for d in self.pkt_data:
|
||
self.pkt_buf.append(d)
|
||
i = pkt_len
|
||
#print("-----------------------------", filename, self.pkt_type, pkt_len, self.pkt_buf)
|
||
else:
|
||
#for i in range(pkt_len):
|
||
'''
|
||
self.pkt_buf[0] = ((pkt_len - 4) >> 4) & 0xff
|
||
self.pkt_buf[1] = (((pkt_len - 4) << 4) & 0xf0) | (self.pkt_type & 0x3)
|
||
self.pkt_buf[2] = 0
|
||
self.pkt_buf[3] = 0
|
||
'''
|
||
self.pkt_buf.append(((pkt_len - 4) >> 4) & 0xff)
|
||
self.pkt_buf.append((((pkt_len - 4) << 4) & 0xf0) | (self.pkt_type & 0x3))
|
||
self.pkt_buf.append(0)
|
||
self.pkt_buf.append(0)
|
||
|
||
for sp in self.sub_pkt:
|
||
if sp[0] != 0:
|
||
pass
|
||
#elif sp[0] == 0:
|
||
else:
|
||
#设备索引(14bit)+类型(2bit)
|
||
temp = ((sp[1] & 0x3fff) << 2) | (sp[2] & 0x3)
|
||
|
||
#self.pkt_buf[i] = (temp >> 8) & 0xff
|
||
self.pkt_buf.append((temp >> 8) & 0xff)
|
||
i = i + 1
|
||
#self.pkt_buf[i] = temp & 0xff
|
||
self.pkt_buf.append(temp & 0xff)
|
||
i = i + 1
|
||
|
||
#地址
|
||
#self.pkt_buf[i] = (sp[3] >> 8) & 0xff
|
||
self.pkt_buf.append((sp[3] >> 8) & 0xff)
|
||
i = i + 1
|
||
#self.pkt_buf[i] = sp[3] & 0xff
|
||
self.pkt_buf.append(sp[3] & 0xff)
|
||
i = i + 1
|
||
|
||
#命令类型(4bit)+长度(12bit)
|
||
temp = (sp[4] << 12) | (sp[5] & 0xfff)
|
||
|
||
#self.pkt_buf[i] = (temp >> 8) & 0xff
|
||
self.pkt_buf.append((temp >> 8) & 0xff)
|
||
i = i + 1
|
||
#self.pkt_buf[i] = temp & 0xff
|
||
self.pkt_buf.append(temp & 0xff)
|
||
i = i + 1
|
||
|
||
#数据
|
||
if sp[6] == None:
|
||
#for _k in range(sp[5]):
|
||
# #self.pkt_buf[_k + i] = random.randint(0, 255)
|
||
# self.pkt_buf.append(random.randint(0, 255))
|
||
temp = list(range(sp[5]))
|
||
temp = [(i & 0xff) for i in temp]
|
||
self.pkt_buf = self.pkt_buf + temp
|
||
else:
|
||
for _k in range(sp[5]):
|
||
#self.pkt_buf[_k + i] = sp[6][_k]
|
||
self.pkt_buf.append(sp[6][_k])
|
||
|
||
i = i + sp[5]
|
||
|
||
#工作计数
|
||
#self.pkt_buf[i] = 0
|
||
self.pkt_buf.append(0)
|
||
i = i + 1
|
||
#self.pkt_buf[i] = 0
|
||
self.pkt_buf.append(0)
|
||
i = i + 1
|
||
|
||
#print("pkt_len:%d, data_len:%d"%(pkt_len, pkt_len - 4))
|
||
open_mode = "wt+"
|
||
if appendgen:
|
||
open_mode = "at"
|
||
f = open(filename, open_mode)
|
||
if filetype == 1:
|
||
for md in self.sub_pkt:
|
||
if md[0] == 1:
|
||
f.write("%s\n"%(md[1],))
|
||
define_str = "`define DEF_"
|
||
|
||
if appendgen and (gen_name == None):
|
||
print("Error, gen_name not define.")
|
||
raise "Error, gen_name not define."
|
||
|
||
if not appendgen:
|
||
pos1 = filename.rfind("/")
|
||
if (pos1 < -1):
|
||
pos1 = 0
|
||
else:
|
||
pos1 = pos1 + 1
|
||
|
||
pos2 = filename.rfind(".")
|
||
if (pos1 > pos2):
|
||
define_str = define_str + filename[pos1:]
|
||
else:
|
||
define_str = define_str + filename[pos1:pos2]
|
||
else:
|
||
define_str = define_str + gen_name
|
||
|
||
|
||
f.write(define_str + "_len \t%d\n"%(pkt_len))
|
||
f.write(define_str + "_sm(array) \\\n\tarray``[%d]=8'h%02X;\\\n\tarray``[%d]=8'h%02X;\n"%(
|
||
3, pkt_len & 0xff,
|
||
4, pkt_len >> 8
|
||
))
|
||
|
||
f.write(define_str + "(array) \\\n")
|
||
for _k in range(i):
|
||
if _k < (i - 1):
|
||
if type(self.pkt_buf[_k]) == int:
|
||
f.write("\tarray``[%d]=8'h%02X;\\\n"%(3 + _k, self.pkt_buf[_k]))
|
||
else:
|
||
f.write("\tarray``[%d]=%s;\\\n"%(3 + _k, self.pkt_buf[_k]))
|
||
else:
|
||
if type(self.pkt_buf[_k]) == int:
|
||
f.write("\tarray``[%d]=8'h%02X;\n"%(3 + _k, self.pkt_buf[_k]))
|
||
else:
|
||
f.write("\tarray``[%d]=%s;\n"%(3 + _k, self.pkt_buf[_k]))
|
||
|
||
for md in self.sub_pkt:
|
||
if md[0] == 2:
|
||
f.write("%s\n"%(md[1],))
|
||
else:
|
||
for _k in range(i):
|
||
f.write("txBuf[%d]=8'h%02X;\n"%(3 + _k, self.pkt_buf[_k]))
|
||
f.close()
|
||
|
||
|
||
|
||
def genWriteTransLenCode(gen_name, sm_idx, wlog=True):
|
||
return '''\n//写入发送长度\n'''\
|
||
+ '''$fdisplay(spidata_file, "--写数据交换长度");\n'''\
|
||
+ '''`DEF_%s_sm(txBuf);\n'''%(gen_name, )\
|
||
+ '''%s( (`MST_TX%d_LEN_ADDR), (`SPI_WRITE), (2), (`IF_IDX(0)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", sm_idx,)\
|
||
+ '''%s( (`MST_RX%d_LEN_ADDR), (`SPI_WRITE), (2), (`IF_IDX(0)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", sm_idx,)
|
||
|
||
def genWriteTransPktCode(gen_name, sm_idx, wlog=True):
|
||
return f'''\n$fdisplay(spidata_file, "<<准备发包，使用SM{sm_idx}：");\n'''\
|
||
+ '''`DEF_%s(txBuf);\n'''%(gen_name, )\
|
||
+ '''%s( (`MST_TX%d_ADDR), (`SPI_WRITE), (`DEF_%s_len), (`IF_IDX(0)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", sm_idx, gen_name, )
|
||
|
||
def genTrgTransPktCode(gen_name, sm_idx, wlog=True):
|
||
return '''\n//请求发送\n'''\
|
||
+ '''$fdisplay(spidata_file, "--请求发送");\n'''\
|
||
+ '''txBuf[0+3] = `MST_TX_TRIGGER(%d);\n'''%(sm_idx, )\
|
||
+ '''%s( (16'h20), (`SPI_WRITE), (1), (`IF_IDX(0)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", )
|
||
|
||
def genReadAckPktCode(gen_name, sm_idx, wlog=True):
|
||
return '''\n//等待回包\n'''\
|
||
+ '''wait_comm_finished();\n'''\
|
||
+ f'''$fdisplay(spidata_file, ">>读取回包，使用SM{sm_idx}：");\n'''\
|
||
+ '''%s( (`MST_RX%d_ADDR), (`SPI_READ), (`DEF_%s_len), (`IF_IDX(0)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", sm_idx, gen_name,)
|
||
|
||
def genEnableInsertTimeCode(pos, enableTime=True, enableDelay=True, wlog=True):
|
||
data = pos & 0x3fff
|
||
if enableTime:
|
||
data = data | 0x8000
|
||
if enableDelay:
|
||
data = data | 0x4000
|
||
return '''//写入替换时间戳替换地址\n'''\
|
||
+ '''$fdisplay(spidata_file, "--增加时间戳插入");\n'''\
|
||
+ '''txBuf[0+3] = 8'h%x;\n'''%(data & 0xff, )\
|
||
+ '''txBuf[1+3] = 8'h%x;\n'''%((data >> 8) & 0xff, )\
|
||
+ '''%s( (16'h24), (`SPI_WRITE), (2), (`IF_IDX(0)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans",)
|
||
|
||
def genDisableInsertTimeCode(wlog=True):
|
||
return '''\n//清除替换时间戳替换地址\n'''\
|
||
+ '''$fdisplay(spidata_file, "--清除时间戳插入");\n'''\
|
||
+ '''txBuf[0+3] = 8'h00;\n'''\
|
||
+ '''txBuf[1+3] = 8'h00;\n'''\
|
||
+ '''%s( (16'h24), (`SPI_WRITE), (2), (`IF_IDX(0)) ); //清除替换时间戳替换地址寄存器\n'''%("`SPI_TRANS" if wlog else "spi_trans", )
|
||
|
||
def genCheckInsertTimeCode(wlog=True):
|
||
return '''\n//检查替换时间戳替换地址\n'''\
|
||
+ '''$fdisplay(spidata_file, "--检查时间戳插入");\n'''\
|
||
+ '''%s( (16'h24), (`SPI_READ), (2), (`IF_IDX(0)) ); //检查替换时间戳替换地址寄存器\n'''%("`SPI_TRANS" if wlog else "spi_trans", )
|
||
|
||
|
||
def genIFRwDataCode(dev, addr, isWrite = False, data_len=0, data = None, wlog=True):
|
||
if isWrite == True:
|
||
if (data == None) or (len(data) == 0):
|
||
data=list()
|
||
|
||
code = ''
|
||
idx = 3
|
||
for item in data:
|
||
code = code + '''txBuf[%d]=8'h%02x;\n'''%(idx, item & 0xff)
|
||
idx = idx + 1
|
||
|
||
code = code + '''%s( (16'h%04x), (`SPI_WRITE), (%d), (`IF_IDX(%d)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", addr, data_len, dev)
|
||
return code
|
||
else:
|
||
code = ''
|
||
code = code + '''%s( (16'h%04x), (`SPI_READ), (%d), (`IF_IDX(%d)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", addr, data_len, dev)
|
||
return code
|
||
|
||
def genSlvSMWriteData(dev, sm, data_len, data = None, wlog=True):
|
||
smaddr = list()
|
||
for i in range(8):
|
||
smaddr.append(0x300 + i * 4 + 2)
|
||
if not ((sm == 1) or (sm == 3) or (sm == 5) or (sm == 7)):
|
||
#写入数据，lvds读；这个空间是lvds写，mcu读
|
||
return ''
|
||
if data_len > 2032:
|
||
return ''
|
||
|
||
if (data != None) and (len(data) != 0) and (len(data) < data_len):
|
||
return ''
|
||
|
||
if (data == None) or (len(data) == 0):
|
||
data = list()
|
||
for i in range(data_len):
|
||
data.append((i + dev) & 0xff)
|
||
|
||
code = '''txBuf[3] = 8'h%02x;\n'''%(data_len & 0xff, )\
|
||
+ '''txBuf[4] = 8'h%02x;\n'''%(int(data_len/256) & 0xff,)\
|
||
+ '''%s((16'h0%04x), (`SPI_WRITE), (2), (`IF_IDX(%d)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", smaddr[sm], dev)
|
||
|
||
for i in range(data_len):
|
||
code = code + '''txBuf[%d + 3] = 8'h%02x;\n'''%(i, data[i])
|
||
|
||
code = code + '''%s( (16'h%04x), (`SPI_WRITE), (%d), (`IF_IDX(%d)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", 0x800 + sm * 2048, data_len, dev)
|
||
return code
|
||
|
||
def genSlvSMReadData(dev, sm, data_len, wlog=True):
|
||
if not ((sm == 0) or (sm == 2) or (sm == 4) or (sm == 6)):
|
||
#写入数据，lvds读；这个空间是lvds写，mcu读
|
||
return ''
|
||
if data_len > 2032:
|
||
return ''
|
||
|
||
code = f'''$fdisplay(spidata_file, "--读取设备{dev} SM{sm} 中的数据，读取长度{data_len}B");\n'''
|
||
code = code + '''%s( (16'h%04x), (`SPI_READ), (%d), (`IF_IDX(%d)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", 0x800 + sm * 2048, data_len, dev)
|
||
|
||
return code
|
||
|
||
def setting_ebmmu_regs(regmap, index, vaddr, raddr, size, p, enable):
|
||
vaddr = int(vaddr)
|
||
raddr = int(raddr)
|
||
size = int(size)
|
||
p = int(p)
|
||
enable = int(enable)
|
||
regmap[index * 0xa + 0] = (vaddr >> 0) & 0xff
|
||
regmap[index * 0xa + 1] = (vaddr >> 8) & 0xff
|
||
regmap[index * 0xa + 2] = (vaddr >> 16) & 0xff
|
||
regmap[index * 0xa + 3] = (vaddr >> 24) & 0xff
|
||
|
||
regmap[index * 0xa + 4] = (raddr >> 0) & 0xff
|
||
regmap[index * 0xa + 5] = (raddr >> 8) & 0xff
|
||
|
||
regmap[index * 0xa + 6] = (size >> 0) & 0xff
|
||
regmap[index * 0xa + 7] = (size >> 8) & 0xff
|
||
|
||
regmap[index * 0xa + 8] = p & 0x3
|
||
regmap[index * 0xa + 9] = enable & 0x1
|
||
|
||
|
||
|
||
__run_list=list()
|
||
__pre_include_path='./generated/shin'
|
||
__pre_gen_log_path='./generated/log'
|
||
def add_tc_2_runlist(filename):
|
||
code_f = filename + '.vh'
|
||
if code_f in __run_list:
|
||
print("Error, Testcase duplicate name")
|
||
raise "DUPLICATE NAME"
|
||
__run_list.append(code_f)
|
||
|
||
def gen_tc_runlist():
|
||
runlist_file = __pre_include_path + '/tc_runlist.vh'
|
||
f = open(runlist_file, "wt+")
|
||
for tc in __run_list:
|
||
f.write(f'\t`include "{tc}"\n')
|
||
f.close()
|
||
|
||
|
||
def genTcStartFrameCode(tcname, des):
|
||
upper_name = tcname.upper()
|
||
code_str = f'''\tbegin:{upper_name}_TESTCASE\n'''\
|
||
+f'''\t\t`BEGIN_TESTCASE("{__pre_gen_log_path}/{tcname}_result.log");\n'''\
|
||
+'''\t\t_t_start = $realtime;\n'''\
|
||
+'''\t\t#1000\n'''\
|
||
+f'''\t\t$fdisplay(spidata_file,"\\n\\n测试用例：{upper_name}");\n'''
|
||
|
||
if len(des) > 0:
|
||
code_str = code_str + '''\t\t$fdisplay(spidata_file,"''' + '-' * 80 + '''");\n'''
|
||
code_str = code_str + f'''\t\t$fdisplay(spidata_file,"{des}");\n'''
|
||
code_str = code_str + '''\t\t$fdisplay(spidata_file,"''' + '-' * 80 + '''");\n'''
|
||
return code_str
|
||
|
||
def genTcEndFrameCode(tcname):
|
||
upper_name = tcname.upper()
|
||
return f'''\t\t_t_end = $realtime;\n'''\
|
||
+'''\t\t$fdisplay(spidata_file, "该测试用例运行时间：%t\\n", _t_end - _t_start);\n'''\
|
||
+f'''\t\t`END_TESTCASE("{__pre_gen_log_path}/{tcname}_result.log");\n'''\
|
||
+'''\tend\n'''
|
||
|
||
def genMessageCode(msg):
|
||
return f'''\t\t$fdisplay(spidata_file, "{msg}");\n''' |