python coding - Fsdb dump script

Python

python coding - Fsdb dump script

guruzoa 2012. 2. 2. 21:19

python 으로 해볼 수 있는 첫번째 example code를 작성해보았다. 필요에 의해서 만든거라 나름 잼있게 쓸 수 있는 code가 되지 않을까?

첨부파일에 code가 있음 ㅎㅎㅎ

# dump_sig_scr.py
import sys
import re

def usage():
    print """
****************************
**** Dump Signal Script ****
* ywkim@siliconworks.co.kr *
****************************
 Usage
 =====
    dump_sig_scr input_path.txt dump_sig.v fsdb/vcd
    Ex : dump_sig_scr full_path.txt dump_sig.v fsdb
"""

remove_sc = []
rm_first_slash = []
changed_dot = []

#if not sys.argv[1:] or sys.argv[3] not in ['fsdb', 'vcd']:
if len(sys.argv) <= 3:
    usage()
    sys.exit(2)
else:
    # Argument [3] check 
    if (sys.argv[3] not in ['fsdb', 'vcd']):
        usage()
        sys.exit(2)

# Try file open
    try:
        f = open(sys.argv[1], 'r')
    except IOError:
        print "%s does not exist!" % (sys.argv[1])
        sys.exit(2)

# File Read & Split space
    msg = f.read().split(' ')
    f.close()

# Remove Special Character (None or Carriage Return)
    for msg_line in msg:
        m = re.search('\r', msg_line)
        s = re.search('\n', msg_line)

if (msg_line == '' or m != None or s != None):
            pass
        else:
            remove_sc.append(msg_line)

#print remove_sc

# First / Character Remove
    p = re.compile(r'^/')

for msg_line in remove_sc:
        rm_first_slash.append(p.sub('', msg_line))

#print rm_first_slash

# Another / Character Change - Dot
    p = re.compile(r'/')

for msg_line in rm_first_slash:
        changed_dot.append(p.sub('.', msg_line))

#print changed_dot

# File Write
    f_w = open(sys.argv[2], 'w')
    f_w.write("initial\n")
    f_w.write("begin\n")

for data in changed_dot:
        if (sys.argv[3] == 'fsdb'):
            merge_msg = "  $fsdbDumpvars(1, %s);\n" % data
        else:
             merge_msg = "  $dumpvars(1, %s);\n" % data

f_w.write(merge_msg)

f_w.write("end")
    f_w.close()

P.S. 코드에 문제가 있음을 발견했다.
Vector 추출시 Sig[7:0] 이런식으로 dump하면 에러가 난다. 따라서 Sig 이렇게 바꿔줘야 한다. 수정한 버전은 아래와 같다.

UnitTest를 잘해야 하는 걸 여기서도 휴...

# dump_sig_scr.py
# Revision : 0.1
# Vector Remove : Example - Sig[7:0] => Sig

import sys
import re

remove_sc = []
rm_first_slash = []
changed_dot = []

#if not sys.argv[1:] or sys.argv[3] not in ['fsdb', 'vcd']:
if len(sys.argv) <= 3:
    usage()
    sys.exit(2)
else:
    # Argument [3] check
    if (sys.argv[3] not in ['fsdb', 'vcd']):
        usage()
        sys.exit(2)

# Try file open
    try:
        f = open(sys.argv[1], 'r')
    except IOError:
        print "%s does not exist!" % (sys.argv[1])
        sys.exit(2)

# File Read & Split space
    msg = f.read().split(' ')
    f.close()

# Remove Special Character (None or Carriage Return)
    for msg_line in msg:
        m = re.search('\r', msg_line)
        s = re.search('\n', msg_line)

if (msg_line == '' or m != None or s != None):
            pass
        else:
            remove_sc.append(msg_line)

#print remove_sc

# First / Character Remove
    p = re.compile(r'^/')

for msg_line in remove_sc:
        rm_first_slash.append(p.sub('', msg_line))

#print rm_first_slash

# Another / Character Change - Dot
    p = re.compile(r'/')

for msg_line in rm_first_slash:
        changed_dot.append(p.sub('.', msg_line))

#print changed_dot

# File Write
    f_w = open(sys.argv[2], 'w')
    f_w.write("initial\n")
    f_w.write("begin\n")

for data in changed_dot:
        #match = re.search(r'\[\d\{1,\}:\d\{1.\}\]', data)
        match = re.search(r'\[\d+:\d+\]', data)

#print match

if match:
            data = re.sub(r'\[\d+:\d+\]', '', data)

if (sys.argv[3] == 'fsdb'):
            merge_msg = "  $fsdbDumpvars(1, %s);\n" % data
        else:
             merge_msg = "  $dumpvars(1, %s);\n" % data

f_w.write(merge_msg)

f_w.write("end")
    f_w.close()