eb001/tb/shinTest/pyutils/GenPkt.py

#!/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):
        gen_pkt_type = 0
        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)
            gen_pkt_type = 1
        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
            gen_pkt_type = 2
            #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:
                    gen_pkt_type = 3
                    #设备索引(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
                
            if gen_pkt_type != 0:
                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, dev_idx=0):
    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(%d)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", sm_idx, dev_idx)\
        + '''%s( (`MST_RX%d_LEN_ADDR), (`SPI_WRITE), (2), (`IF_IDX(%d)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", sm_idx, dev_idx)

def genWriteTransPktCode(gen_name, sm_idx, wlog=True, dev_idx=0):
    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(%d)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", sm_idx, gen_name, dev_idx)

def genTrgTransPktCode(gen_name, sm_idx, wlog=True, dev_idx=0):
    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(%d)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", dev_idx)
        
def genReadAckPktCode(gen_name, sm_idx, wlog=True, dev_idx=0):
    return '''\n//等待回包\n'''\
        + '''wait_comm_finished(%d);\n'''%(dev_idx, )\
        + f'''$fdisplay(spidata_file, ">>读取回包，使用SM{sm_idx}：");\n'''\
        + '''%s( (`MST_RX%d_ADDR), (`SPI_READ), (`DEF_%s_len), (`IF_IDX(%d)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", sm_idx, gen_name, dev_idx)

def genEnableInsertTimeCode(pos, enableTime=True, enableDelay=True, wlog=True, dev_idx=0):
    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(%d)) );\n'''%("`SPI_TRANS" if wlog else "spi_trans", dev_idx)

def genDisableInsertTimeCode(wlog=True, dev_idx=0):
    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(%d)) ); //清除替换时间戳替换地址寄存器\n'''%("`SPI_TRANS" if wlog else "spi_trans", dev_idx)

def genCheckInsertTimeCode(wlog=True, dev_idx=0):
    return '''\n//检查替换时间戳替换地址\n'''\
        + '''$fdisplay(spidata_file, "--检查时间戳插入");\n'''\
        + '''%s( (16'h24), (`SPI_READ), (2), (`IF_IDX(%d)) ); //检查替换时间戳替换地址寄存器\n'''%("`SPI_TRANS" if wlog else "spi_trans", dev_idx)


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"
            )