mirror of https://github.com/VLSIDA/OpenRAM.git
93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
# See LICENSE for licensing information.
|
|
#
|
|
# Copyright (c) 2016-2023 Regents of the University of California and The Board
|
|
# of Regents for the Oklahoma Agricultural and Mechanical College
|
|
# (acting for and on behalf of Oklahoma State University)
|
|
# All rights reserved.
|
|
#
|
|
"""
|
|
Some baseline functions to run scripts.
|
|
"""
|
|
|
|
import os
|
|
import subprocess
|
|
import time
|
|
from openram import debug
|
|
from openram import OPTS
|
|
|
|
|
|
def run_script(cell_name, script="lvs"):
|
|
""" Run script and create output files. """
|
|
|
|
echo_cmd_output = OPTS.verbose_level > 1
|
|
|
|
cwd = os.getcwd()
|
|
os.chdir(OPTS.openram_temp)
|
|
errfile = "{0}{1}.{2}.err".format(OPTS.openram_temp, cell_name, script)
|
|
outfile = "{0}{1}.{2}.out".format(OPTS.openram_temp, cell_name, script)
|
|
resultsfile = "{0}{1}.{2}.report".format(OPTS.openram_temp, cell_name, script)
|
|
|
|
scriptpath = '{0}run_{1}.sh'.format(OPTS.openram_temp, script)
|
|
|
|
# Wrap with conda activate & conda deactivate
|
|
if OPTS.use_conda:
|
|
from openram import CONDA_HOME
|
|
with open(scriptpath, "r") as f:
|
|
script_content = f.readlines()
|
|
with open(scriptpath, "w") as f:
|
|
# First line is shebang
|
|
f.write(script_content[0])
|
|
# Activate conda using the activate script
|
|
f.write("source {}/bin/activate\n".format(CONDA_HOME))
|
|
for line in script_content[1:]:
|
|
f.write(line)
|
|
# Deactivate conda at the end
|
|
f.write("conda deactivate\n")
|
|
|
|
debug.info(2, "Starting {}".format(scriptpath))
|
|
start = time.time()
|
|
with open(outfile, 'wb') as fo, open(errfile, 'wb') as fe:
|
|
p = subprocess.Popen(
|
|
[scriptpath], stdout=fo, stderr=fe, cwd=OPTS.openram_temp)
|
|
|
|
if echo_cmd_output:
|
|
tailo = subprocess.Popen([
|
|
'tail',
|
|
'-f', # Follow the output
|
|
'--pid', str(p.pid), # Close when this pid exits
|
|
outfile,
|
|
])
|
|
taile = subprocess.Popen([
|
|
'tail',
|
|
'-f', # Follow the output
|
|
'--pid', str(p.pid), # Close when this pid exits
|
|
errfile,
|
|
])
|
|
|
|
lastoutput = start
|
|
while p.poll() == None:
|
|
runningfor = time.time() - start
|
|
outputdelta = time.time() - lastoutput
|
|
if outputdelta > 30:
|
|
lastoutput = time.time()
|
|
debug.info(1, "Still running {} ({:.0f} seconds)".format(scriptpath, runningfor))
|
|
time.sleep(1)
|
|
assert p.poll() != None, (p.poll(), p)
|
|
p.wait()
|
|
|
|
# Kill the tail commands if they haven't finished.
|
|
if echo_cmd_output:
|
|
if tailo.poll() != None:
|
|
tailo.kill()
|
|
tailo.wait()
|
|
if taile.poll() != None:
|
|
taile.kill()
|
|
taile.wait()
|
|
|
|
debug.info(2, "Finished {} with {}".format(scriptpath, p.returncode))
|
|
|
|
os.chdir(cwd)
|
|
|
|
return (outfile, errfile, resultsfile)
|
|
|