from pyutils import *
from .environment import *

import time
import random

def tc1():
    gen_file_def = "regtest_tc01"
    sm_idx = 0
    tc_des = "测试寄存器的读写状态，将系统改为8B10B进行通信\\n" 

    genPkt = GenPkt()
    genPkt.clean()

    dev_num = get_slv_dev_num()

    genPkt.addBeforeExc(genTcStartFrameCode(gen_file_def, tc_des))

    change_8b10b_code = ""
    for i in range(dev_num + 1):
        change_8b10b_code = change_8b10b_code + genIFRwDataCode(i, 0x004c, True, 2, [0x1, 0x59])\
            +genIFRwDataCode(i, 0x004c, False, 2)


    genPkt.addAfterExc(#genWriteTransLenCode(gen_file_def, sm_idx)
                       #+genWriteTransPktCode(gen_file_def, sm_idx)
                       #+genTrgTransPktCode(gen_file_def, sm_idx)
                       #+genReadAckPktCode(gen_file_def, sm_idx)
                       change_8b10b_code
                       +genTcEndFrameCode(gen_file_def))
    genPkt.gen("./generated/shin/%s.vh"%(gen_file_def, ), 1)
    return gen_file_def


def tc2():
    gen_file_def = "regtest_tc02"
    sm_idx = 0
    tc_des = "测试寄存器的读写状态，将系统改为4B5B进行通信\\n" 

    genPkt = GenPkt()
    genPkt.clean()

    dev_num = get_slv_dev_num()

    genPkt.addBeforeExc(genTcStartFrameCode(gen_file_def, tc_des))

    change_4b5b_code = ""
    for i in range(dev_num + 1):
        change_4b5b_code = change_4b5b_code + genIFRwDataCode(i, 0x004c, True, 2, [0x0, 0x59])\
            +genIFRwDataCode(i, 0x004c, False, 2)


    genPkt.addAfterExc(#genWriteTransLenCode(gen_file_def, sm_idx)
                       #+genWriteTransPktCode(gen_file_def, sm_idx)
                       #+genTrgTransPktCode(gen_file_def, sm_idx)
                       #+genReadAckPktCode(gen_file_def, sm_idx)
                       change_4b5b_code
                       +genTcEndFrameCode(gen_file_def))
    genPkt.gen("./generated/shin/%s.vh"%(gen_file_def, ), 1)
    return gen_file_def