usb-tools/amaranth
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

back.pysim: accept write_vcd(vcd_file=None).

Browse source 
Because write_vcd() is a context manager, this is useful if the VCD
file should be sometimes not written, since it avoids awkward
conditionals with duplicated code. It's not very elegant though.

Fixes #319.
This commit is contained in:
whitequark 2020-02-12 14:42:06 +00:00
parent 38aa9fb671
commit 8947096eea
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
nmigen/back/pysim.py
View file

@ -94,6 +94,9 @@ class _VCDWaveformWriter(_WaveformWriter):
trace_names[trace] = trace.name
self.traces.append(trace)
if self.vcd_writer is None:
return
for signal, names in itertools.chain(signal_names.items(), trace_names.items()):
if signal.decoder:
var_type = "string"
@ -139,7 +142,8 @@ class _VCDWaveformWriter(_WaveformWriter):
self.vcd_writer.change(vcd_var, vcd_timestamp, var_value)
def close(self, timestamp):
self.vcd_writer.close(self.timestamp_to_vcd(timestamp))
if self.vcd_writer is not None:
self.vcd_writer.close(self.timestamp_to_vcd(timestamp))
if self.gtkw_save is not None:
self.gtkw_save.dumpfile(self.vcd_file.name)
@ -153,7 +157,8 @@ class _VCDWaveformWriter(_WaveformWriter):
suffix = ""
self.gtkw_save.trace(".".join(self.gtkw_names[signal]) + suffix)
self.vcd_file.close()
if self.vcd_file is not None:
self.vcd_file.close()
if self.gtkw_file is not None:
self.gtkw_file.close()