litex/migen/sim/icarus.py

# Copyright (C) 2012 Vermeer Manufacturing Co.
# License: GPLv3 with additional permissions (see README).

import subprocess
import os
import time

def _str2file(filename, contents):
	f = open(filename, "w")
	f.write(contents)
	f.close()

class Runner:
	def __init__(self, options=None, extra_files=None, top_file="migensim_top.v", dut_file="migensim_dut.v", vvp_file=None, keep_files=False):
		if extra_files is None: extra_files = []
		if vvp_file is None: vvp_file = dut_file + "vp"
		if options is None: options = []
		self.options = options
		self.extra_files = extra_files
		self.top_file = top_file
		self.dut_file = dut_file
		self.vvp_file = vvp_file
		self.keep_files = keep_files
	
	def start(self, c_top, c_dut):
		_str2file(self.top_file, c_top)
		_str2file(self.dut_file, c_dut)
		subprocess.check_call(["iverilog", "-o", self.vvp_file] + self.options + [self.top_file, self.dut_file] + self.extra_files)
		self.process = subprocess.Popen(["vvp", "-mmigensim", "-Mvpi", self.vvp_file])

	def close(self):
		if hasattr(self, "process"):
			self.process.terminate()
			if self.process.poll() is None:
				time.sleep(.1)
				self.process.kill()
			self.process.wait()
		if not self.keep_files:
			for f in [self.top_file, self.dut_file, self.vvp_file]:
				try:
					os.remove(f)
				except OSError:
					pass