vcd2latex.py
· 3.9 KiB · Python
Исходник
#!/usr/bin/env python3
import vcdvcd, math
from vcdvcd import VCDVCD
vcdfile = "digital_port_tb.vcd"
vcd = VCDVCD(vcdfile, only_sigs=True)
all_signals = vcd.signals
# print(all_signals)
dataW = [0 for i in range(len(all_signals))]
data = [[] for i in range(len(all_signals))]
###########
def nibble_to_hex(s):
for c in s:
if not c in '01':
return c
return hex(int(s, 2))[2:].upper()
def binary_string_to_hex(s, w):
"""
Convert a binary string to hexadecimal.
If any non 0/1 values are present such as 'x', return that single character
as a representation.
:param s: the string to be converted
:type s: str
"""
if len(s) == 1:
if s != '0' and s != '1':
return s
s = s.zfill(w)
n = 4
groups = [s[i:i+n] for i in range(0, len(s), n)]
hexgroups = [nibble_to_hex(x) for x in groups]
return "".join(hexgroups)
return s
# for c in s:
# if not c in '01':
# return c
# return hex(int(s, 2))[2:]
class ParserCallbacks(vcdvcd.StreamParserCallbacks):
def __init__(self, deltas=True):
self._deltas = deltas
self._references_to_widths = {}
def enddefinitions(
self,
vcd,
signals,
cur_sig_vals
):
if signals:
self._print_dumps_refs = signals
else:
self._print_dumps_refs = sorted(vcd.data[i].references[0] for i in cur_sig_vals.keys())
for i, ref in enumerate(self._print_dumps_refs, 1):
if i == 0:
i = 1
identifier_code = vcd.references_to_ids[ref]
size = int(vcd.data[identifier_code].size)
width = max(((size // 4)), int(math.floor(math.log10(i))) + 1)
self._references_to_widths[ref] = width
# print(ref, i, size)
dataW[i-1] = size
def time(
self,
vcd,
time,
cur_sig_vals
):
if (not self._deltas or vcd.signal_changed):
ss = []
ss.append('{}'.format(time))
for i, ref in enumerate(self._print_dumps_refs):
identifier_code = vcd.references_to_ids[ref]
value = cur_sig_vals[identifier_code]
data[i].append(binary_string_to_hex(value, dataW[i]))
###########
callbacks = ParserCallbacks()
VCDVCD(
vcdfile,
signals=all_signals,
store_tvs=False,
callbacks=callbacks,
)
ltx = '''
\\documentclass{standalone}
\\usepackage{tikz-timing}
\\begin{document}
\\begin{tikztimingtable}[timing/xunit=35,timing/yunit=10]
'''
def latexEscape(val):
return val.\
replace("_", "\\_").\
replace("{", "\\{").\
replace("}", "\\}").\
replace("&", "\\&").\
replace("%", "\\%").\
replace("$", "\\$").\
replace("#", "\\#").\
replace("$", "\\$")
for sig in range(len(data)):
sigName = latexEscape(all_signals[sig])
sigSeries = data[sig]
sigWidth = dataW[sig]
# print(f"SigName: {sigName} - SigWidth: {sigWidth}")
ltx += format(f" {sigName} &")
for value in sigSeries:
prefix = ""
token = format(f" D{{{value}}} ")
if value == "x":
token = "X"
elif value == "z":
token = "Z"
elif sigWidth == 1:
token = "H" if value == "1" else "L"
if "x" in token:
prefix = " [red] "
elif "z" in token:
prefix = " [blue] "
ltx += format(f" {prefix}{token} ;")
ltx += " \\\\\n"
# ltx += "\\vertlines[help lines,opacity=0.3]{}\n"
ltx += '''\\extracode
\\vertlines[help lines,opacity=0.3]{}
\\end{tikztimingtable}
\\end{document}
'''
# print(ltx)
with open("tmp.tex", "w") as f:
f.write(ltx)
'''
\\begin{tikztimingtable}[timing/wscale=0.8]
M-cycle & X 8D{M1} 8D{M2/M1} X \\\\
Instruction & ; [opacity=0.4] 9D{Previous} ; [opacity=1.0] 8D{LD r, r'} ; [opacity=0.4] X \\\\
Mem R/W & X 8D{R: opcode}; [opacity=0.4] 8D{R: next op} X \\\\
Mem addr & X 8D{PC} ; [opacity=0.4] 8D{PC+1} X \\\\
\\end{tikztimingtable}
'''
| 1 | #!/usr/bin/env python3 |
| 2 | |
| 3 | import vcdvcd, math |
| 4 | from vcdvcd import VCDVCD |
| 5 | |
| 6 | vcdfile = "digital_port_tb.vcd" |
| 7 | |
| 8 | vcd = VCDVCD(vcdfile, only_sigs=True) |
| 9 | |
| 10 | all_signals = vcd.signals |
| 11 | # print(all_signals) |
| 12 | dataW = [0 for i in range(len(all_signals))] |
| 13 | data = [[] for i in range(len(all_signals))] |
| 14 | |
| 15 | ########### |
| 16 | |
| 17 | def nibble_to_hex(s): |
| 18 | for c in s: |
| 19 | if not c in '01': |
| 20 | return c |
| 21 | return hex(int(s, 2))[2:].upper() |
| 22 | |
| 23 | def binary_string_to_hex(s, w): |
| 24 | """ |
| 25 | Convert a binary string to hexadecimal. |
| 26 | If any non 0/1 values are present such as 'x', return that single character |
| 27 | as a representation. |
| 28 | :param s: the string to be converted |
| 29 | :type s: str |
| 30 | """ |
| 31 | if len(s) == 1: |
| 32 | if s != '0' and s != '1': |
| 33 | return s |
| 34 | |
| 35 | s = s.zfill(w) |
| 36 | n = 4 |
| 37 | groups = [s[i:i+n] for i in range(0, len(s), n)] |
| 38 | hexgroups = [nibble_to_hex(x) for x in groups] |
| 39 | return "".join(hexgroups) |
| 40 | return s |
| 41 | # for c in s: |
| 42 | # if not c in '01': |
| 43 | # return c |
| 44 | # return hex(int(s, 2))[2:] |
| 45 | |
| 46 | |
| 47 | class ParserCallbacks(vcdvcd.StreamParserCallbacks): |
| 48 | def __init__(self, deltas=True): |
| 49 | self._deltas = deltas |
| 50 | self._references_to_widths = {} |
| 51 | |
| 52 | def enddefinitions( |
| 53 | self, |
| 54 | vcd, |
| 55 | signals, |
| 56 | cur_sig_vals |
| 57 | ): |
| 58 | if signals: |
| 59 | self._print_dumps_refs = signals |
| 60 | else: |
| 61 | self._print_dumps_refs = sorted(vcd.data[i].references[0] for i in cur_sig_vals.keys()) |
| 62 | for i, ref in enumerate(self._print_dumps_refs, 1): |
| 63 | if i == 0: |
| 64 | i = 1 |
| 65 | identifier_code = vcd.references_to_ids[ref] |
| 66 | size = int(vcd.data[identifier_code].size) |
| 67 | width = max(((size // 4)), int(math.floor(math.log10(i))) + 1) |
| 68 | self._references_to_widths[ref] = width |
| 69 | # print(ref, i, size) |
| 70 | dataW[i-1] = size |
| 71 | |
| 72 | def time( |
| 73 | self, |
| 74 | vcd, |
| 75 | time, |
| 76 | cur_sig_vals |
| 77 | ): |
| 78 | if (not self._deltas or vcd.signal_changed): |
| 79 | ss = [] |
| 80 | ss.append('{}'.format(time)) |
| 81 | for i, ref in enumerate(self._print_dumps_refs): |
| 82 | identifier_code = vcd.references_to_ids[ref] |
| 83 | value = cur_sig_vals[identifier_code] |
| 84 | data[i].append(binary_string_to_hex(value, dataW[i])) |
| 85 | |
| 86 | |
| 87 | ########### |
| 88 | |
| 89 | |
| 90 | callbacks = ParserCallbacks() |
| 91 | VCDVCD( |
| 92 | vcdfile, |
| 93 | signals=all_signals, |
| 94 | store_tvs=False, |
| 95 | callbacks=callbacks, |
| 96 | ) |
| 97 | |
| 98 | ltx = ''' |
| 99 | \\documentclass{standalone} |
| 100 | \\usepackage{tikz-timing} |
| 101 | |
| 102 | \\begin{document} |
| 103 | \\begin{tikztimingtable}[timing/xunit=35,timing/yunit=10] |
| 104 | ''' |
| 105 | |
| 106 | def latexEscape(val): |
| 107 | return val.\ |
| 108 | replace("_", "\\_").\ |
| 109 | replace("{", "\\{").\ |
| 110 | replace("}", "\\}").\ |
| 111 | replace("&", "\\&").\ |
| 112 | replace("%", "\\%").\ |
| 113 | replace("$", "\\$").\ |
| 114 | replace("#", "\\#").\ |
| 115 | replace("$", "\\$") |
| 116 | |
| 117 | |
| 118 | for sig in range(len(data)): |
| 119 | sigName = latexEscape(all_signals[sig]) |
| 120 | sigSeries = data[sig] |
| 121 | sigWidth = dataW[sig] |
| 122 | # print(f"SigName: {sigName} - SigWidth: {sigWidth}") |
| 123 | ltx += format(f" {sigName} &") |
| 124 | for value in sigSeries: |
| 125 | prefix = "" |
| 126 | token = format(f" D{{{value}}} ") |
| 127 | if value == "x": |
| 128 | token = "X" |
| 129 | elif value == "z": |
| 130 | token = "Z" |
| 131 | elif sigWidth == 1: |
| 132 | token = "H" if value == "1" else "L" |
| 133 | if "x" in token: |
| 134 | prefix = " [red] " |
| 135 | elif "z" in token: |
| 136 | prefix = " [blue] " |
| 137 | |
| 138 | ltx += format(f" {prefix}{token} ;") |
| 139 | ltx += " \\\\\n" |
| 140 | |
| 141 | # ltx += "\\vertlines[help lines,opacity=0.3]{}\n" |
| 142 | ltx += '''\\extracode |
| 143 | \\vertlines[help lines,opacity=0.3]{} |
| 144 | \\end{tikztimingtable} |
| 145 | \\end{document} |
| 146 | |
| 147 | ''' |
| 148 | |
| 149 | # print(ltx) |
| 150 | with open("tmp.tex", "w") as f: |
| 151 | f.write(ltx) |
| 152 | |
| 153 | |
| 154 | ''' |
| 155 | \\begin{tikztimingtable}[timing/wscale=0.8] |
| 156 | M-cycle & X 8D{M1} 8D{M2/M1} X \\\\ |
| 157 | Instruction & ; [opacity=0.4] 9D{Previous} ; [opacity=1.0] 8D{LD r, r'} ; [opacity=0.4] X \\\\ |
| 158 | Mem R/W & X 8D{R: opcode}; [opacity=0.4] 8D{R: next op} X \\\\ |
| 159 | Mem addr & X 8D{PC} ; [opacity=0.4] 8D{PC+1} X \\\\ |
| 160 | \\end{tikztimingtable} |
| 161 | ''' |
| 162 |