#!/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, iface=5): spi_api="IF_IQSPI" if iface == 0: spi_api = "IF_IQSPI" elif iface == 1: spi_api = "IF_OSPI" elif iface == 2: spi_api = "IF_OQSPI" else: spi_api = "IF_IDX" 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), (`%s(%d)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", addr, data_len, spi_api, dev) return code else: code = '' code = code + '''%s( (16'h%04x), (`SPI_READ), (%d), (`%s(%d)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", addr, data_len, spi_api, 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''' def genAddNewDev(devIndex): dev_code = ("ShinTop s_ShinTop_slv3(\n" " .clock(clock),\n" " .reset(reset | isSoftReset[4]),\n" "\n" " .io_multiCLK_0(multiCLK[0]),\n" " .io_multiCLK_1(multiCLK[1]),\n" " .io_multiCLK_2(multiCLK[2]),\n" " .io_multiCLK_3(multiCLK[3]),\n" " .io_multiCLK_4(multiCLK[4]),\n" " .io_multiCLK_5(multiCLK[5]),\n" " .io_multiCLK_6(multiCLK[6]),\n" " .io_multiCLK_7(multiCLK[7]),\n" " .io_multiCLK_8(multiCLK[8]),\n" " .io_multiCLK_9(multiCLK[9]),\n" " .io_multiCLK_10(multiCLK[10]),\n" " .io_multiCLK_11(multiCLK[11]),\n" " .io_multiCLK_12(multiCLK[12]),\n" " .io_multiCLK_13(multiCLK[13]),\n" " .io_multiCLK_14(multiCLK[14]),\n" " .io_multiCLK_15(multiCLK[15]),\n" "\n" " .io_dDatIn(dDat_w[3]),\n" " .io_dDatOut(),\n" " .io_uDatIn(1'b0),\n" " .io_uDatOut(uDat_w[3]),\n" "\n" " .io_iqspi_sck(qspi_sck),\n" " .io_iqspi_mosi(qspi_mosi),\n" " .io_iqspi_miso(qspi_miso[4*4+3:4*4]),\n" " .io_iqspi_isDirIn(),\n" " .io_iqspi_csn(spi_csn[4]),\n" "\n" " .io_ospi_sck (spi_sck),\n" " .io_ospi_mosi(spi_mosi),\n" " .io_ospi_miso(spi_miso[4]),\n" " .io_ospi_csn (spi_csn[9]),\n" "\n" " .io_oqspi_sck(qspi_sck),\n" " .io_oqspi_mosi(qspi_mosi),\n" " .io_oqspi_miso(qspi_miso[4*9+3:4*9]),\n" " .io_oqspi_isDirIn(),\n" " .io_oqspi_csn(spi_csn[14]),\n" "\n" " .io_isSoftReset(isSoftReset[4]),\n" " .io_interrupt(io_int_sig[4]),\n" " .io_latchPin_0(latch0),\n" " .io_latchPin_1(latch1),\n" " .io_hwSerial(0),\n" " .io_uniCode(0)\n" ");\n" )