def set_workdir(self, workdir):
"""Set the working directory."""
self.workdir = os.path.abspath(workdir)
# Files required for the execution.
self.input_file = File(os.path.join(self.workdir, "run.abi"))
self.output_file = File(os.path.join(self.workdir, "run.abo"))
self.files_file = File(os.path.join(self.workdir, "run.files"))
self.log_file = File(os.path.join(self.workdir, "run.log"))
self.stderr_file = File(os.path.join(self.workdir, "run.err"))
# Directories with input|output|temporary data.
self.indir = Directory(os.path.join(self.workdir, "indata"))
self.outdir = Directory(os.path.join(self.workdir, "outdata"))
self.tmpdir = Directory(os.path.join(self.workdir, "tmpdata"))
def set_workdir(self, workdir):
"""Set the working directory."""
self.workdir = os.path.abspath(workdir)
# Files required for the execution.
self.input_file = File(os.path.join(self.workdir, "run.abi"))
self.output_file = File(os.path.join(self.workdir, "run.abo"))
self.files_file = File(os.path.join(self.workdir, "run.files"))
self.log_file = File(os.path.join(self.workdir, "run.log"))
self.stderr_file = File(os.path.join(self.workdir, "run.err"))
# Directories with input|output|temporary data.
self.indir = Directory(os.path.join(self.workdir, "indata"))
self.outdir = Directory(os.path.join(self.workdir, "outdata"))
self.tmpdir = Directory(os.path.join(self.workdir, "tmpdata"))
def __init__(self, path, **kwargs):
"""
Args:
path:
Path to the script file.
stdin:
The input file to feed in the executable as the standard input. Mandatory.
stdout:
The file into which the standard output is redirected. Default is 'log'.
stderr:
The file into which the standard error is redirected. Default is 'stderr'.
bindir:
The directory in which to look for binaries. Default is "".
modules:
The modules which will be loaded with 'module load'. Default is None
pre_lines:
Lines before the main execution. Default is []
post_lines:
Lines after the main execution. Default is [].
exe:
The name of the binary to be executed (located in bindir or in $PATH if not bindir). Default is abinit.
mpirun:
The mpi runner.
E.g. 'mpirun -n 6'. Default is None.
mpi_ncpus
Number of MPI processes (default 1)
omp_env
Dictionary with the OMP environment variables. Default is {}
vars:
Dictionary {varname:varvalue} with variable declaration.
"""
self.jobfile = File(path)
self.stdin = kwargs.get("stdin", "abi.files")
self.stdout = kwargs.get("stdout", "abi.log")
self.stderr = kwargs.get("stderr", "abi.err")
self.bindir = kwargs.get("bindir", "")
self.exe = os.path.join(self.bindir, kwargs.get("exe", "abinit"))
self.mpirun = kwargs.get("mpirun", "")
self.mpi_ncpus = kwargs.get("mpi_ncpus", 1)
self.omp_env = kwargs.get("omp_env", {})
if self.omp_env:
self.omp_env = OMPEnv(self.omp_env)
self.vars = kwargs.get("vars", {})
self.modules = kwargs.get("modules", [])
self.envars = kwargs.get("envars", {})
self.pre_lines = kwargs.get("pre_lines", [])
self.post_lines = kwargs.get("post_lines", [])
self.queue_params = kwargs.get("queue_params", {})
def __init__(self, strategy, workdir, runmode, task_id=1, links=None, **kwargs):
"""
Args:
strategy:
Strategy instance descring the calculation.
workdir:
Path to the working directory.
runmode:
RunMode instance or string "sequential"
links:
List of WorkLink objects specifying the dependencies of the task.
kwargs:
keyword argumens (not used here)
"""
super(AbinitTask, self).__init__()
self.strategy = strategy
self.workdir = os.path.abspath(workdir)
self.runmode = RunMode.asrunmode(runmode).copy()
self.set_id(task_id)
# Connect this task to the other tasks
# (needed if are creating a Work instance with dependencies
self._links = []
if links is not None:
self._links = links
for link in links:
print("Adding ", link.get_abivars())
self.strategy.add_extra_abivars(link.get_abivars())
# Files required for the execution.
self.input_file = File(os.path.join(self.workdir, AbinitTask.basename.input))
self.output_file = File(os.path.join(self.workdir, AbinitTask.basename.output))
self.files_file = File(os.path.join(self.workdir, AbinitTask.basename.files_file))
self.log_file = File(os.path.join(self.workdir, AbinitTask.basename.log_file))
self.stderr_file = File(os.path.join(self.workdir, AbinitTask.basename.stderr_file))
# Find number of processors ....
#runhints = self.get_runhints()
#launcher = self.runmode.make_launcher(runhints)
# Build the launcher and store it.
self.set_launcher(self.runmode.make_launcher(self))
class AbiFireTask(FireTaskBase):
# List of `AbinitEvent` subclasses that are tested in the check_status method.
# Subclasses should provide their own list if they need to check the converge status.
CRITICAL_EVENTS = []
S_INIT = Status.from_string("Initialized")
S_LOCKED = Status.from_string("Locked")
S_READY = Status.from_string("Ready")
S_SUB = Status.from_string("Submitted")
S_RUN = Status.from_string("Running")
S_DONE = Status.from_string("Done")
S_ABICRITICAL = Status.from_string("AbiCritical")
S_QCRITICAL = Status.from_string("QCritical")
S_UNCONVERGED = Status.from_string("Unconverged")
S_ERROR = Status.from_string("Error")
S_OK = Status.from_string("Completed")
ALL_STATUS = [
S_INIT,
S_LOCKED,
S_READY,
S_SUB,
S_RUN,
S_DONE,
S_ABICRITICAL,
S_QCRITICAL,
S_UNCONVERGED,
S_ERROR,
S_OK,
]
def __init__(self, abiinput):
"""
Basic __init__, subclasses are supposed to define the same input parameters, add their own and call super for
the basic ones. The input parameter should be stored as attributes of the instance for serialization and
for inspection.
"""
self.abiinput = abiinput
@serialize_fw
def to_dict(self):
d = {}
for arg in inspect.getargspec(self.__init__).args:
if arg != "self":
val = self.__getattribute__(arg)
if hasattr(val, "as_dict"):
val = val.as_dict()
elif isinstance(val, (tuple, list)):
val = [
v.as_dict() if hasattr(v, "as_dict") else v
for v in val
]
d[arg] = val
return d
@classmethod
def from_dict(cls, d):
dec = MontyDecoder()
kwargs = {
k: dec.process_decoded(v)
for k, v in d.items() if k in inspect.getargspec(cls.__init__).args
}
return cls(**kwargs)
#from Task
def set_workdir(self, workdir):
"""Set the working directory."""
self.workdir = os.path.abspath(workdir)
# Files required for the execution.
self.input_file = File(os.path.join(self.workdir, "run.abi"))
self.output_file = File(os.path.join(self.workdir, "run.abo"))
self.files_file = File(os.path.join(self.workdir, "run.files"))
self.log_file = File(os.path.join(self.workdir, "run.log"))
self.stderr_file = File(os.path.join(self.workdir, "run.err"))
# Directories with input|output|temporary data.
self.indir = Directory(os.path.join(self.workdir, "indata"))
self.outdir = Directory(os.path.join(self.workdir, "outdata"))
self.tmpdir = Directory(os.path.join(self.workdir, "tmpdata"))
# from Task
def build(self):
"""
Creates the working directory and the input files of the :class:`Task`.
It does not overwrite files if they already exist.
"""
# Create dirs for input, output and tmp data.
self.indir.makedirs()
self.outdir.makedirs()
self.tmpdir.makedirs()
# Write files file and input file.
if not self.files_file.exists:
self.files_file.write(self.filesfile_string)
self.input_file.write(str(self.abiinput))
#from Task
# Prefixes for Abinit (input, output, temporary) files.
Prefix = collections.namedtuple("Prefix", "idata odata tdata")
pj = os.path.join
prefix = Prefix(pj("indata", "in"), pj("outdata", "out"),
pj("tmpdata", "tmp"))
del Prefix, pj
#from AbintTask
@property
def filesfile_string(self):
"""String with the list of files and prefixes needed to execute ABINIT."""
lines = []
app = lines.append
pj = os.path.join
app(self.input_file.path) # Path to the input file
app(self.output_file.path) # Path to the output file
app(pj(self.workdir, self.prefix.idata)) # Prefix for input data
app(pj(self.workdir, self.prefix.odata)) # Prefix for output data
app(pj(self.workdir, self.prefix.tdata)) # Prefix for temporary data
# Paths to the pseudopotential files.
# Note that here the pseudos **must** be sorted according to znucl.
# Here we reorder the pseudos if the order is wrong.
ord_pseudos = []
znucl = self.abiinput.structure.to_abivars()["znucl"]
for z in znucl:
for p in self.abiinput.pseudos:
if p.Z == z:
ord_pseudos.append(p)
break
else:
raise ValueError(
"Cannot find pseudo with znucl %s in pseudos:\n%s" %
(z, self.pseudos))
for pseudo in ord_pseudos:
app(pseudo.path)
return "\n".join(lines)
def run_abinit(self, fw_spec):
with open(self.files_file.path, 'r') as stdin, open(self.log_file.path, 'w') as stdout, \
open(self.stderr_file.path, 'w') as stderr:
p = subprocess.Popen(['mpirun', 'abinit'],
stdin=stdin,
stdout=stdout,
stderr=stderr)
(stdoutdata, stderrdata) = p.communicate()
self.returncode = p.returncode
def get_event_report(self):
"""
Analyzes the main output file for possible Errors or Warnings.
Returns:
:class:`EventReport` instance or None if the main output file does not exist.
"""
if not self.log_file.exists:
return None
parser = events.EventsParser()
try:
report = parser.parse(self.log_file.path)
return report
except parser.Error as exc:
# Return a report with an error entry with info on the exception.
logger.critical("%s: Exception while parsing ABINIT events:\n %s" %
(self.log_file, str(exc)))
return parser.report_exception(self.log_file.path, exc)
def task_analysis(self, fw_spec):
status, msg = self.check_final_status()
if self.status != self.S_OK:
raise AbinitRuntimeError(self)
return FWAction(stored_data=dict(**self.report.as_dict()))
def run_task(self, fw_spec):
self.set_workdir(os.path.abspath('.'))
self.build()
self.run_abinit(fw_spec)
return self.task_analysis(fw_spec)
def set_status(self, status, msg=None):
self.status = status
return status, msg
def check_final_status(self):
"""
This function checks the status of the task by inspecting the output and the
error files produced by the application. Based on abipy task checkstatus().
"""
# 2) see if an error occured at starting the job
# 3) see if there is output
# 4) see if abinit reports problems
# 5) see if err file exists and is empty
# 9) the only way of landing here is if there is a output file but no err files...
# 2) Check the returncode of the process (the process of submitting the job) first.
if self.returncode != 0:
# The job was not submitted properly
return self.set_status(self.S_QCRITICAL,
msg="return code %s" % self.returncode)
# Analyze the stderr file for Fortran runtime errors.
err_msg = None
if self.stderr_file.exists:
err_msg = self.stderr_file.read()
# Start to check ABINIT status if the output file has been created.
if self.output_file.exists:
try:
self.report = self.get_event_report()
except Exception as exc:
msg = "%s exception while parsing event_report:\n%s" % (self,
exc)
logger.critical(msg)
return self.set_status(self.S_ABICRITICAL, msg=msg)
if self.report.run_completed:
# Check if the calculation converged.
not_ok = self.report.filter_types(self.CRITICAL_EVENTS)
if not_ok:
return self.set_status(self.S_UNCONVERGED)
else:
return self.set_status(self.S_OK)
# Calculation still running or errors?
if self.report.errors or self.report.bugs:
# Abinit reported problems
if self.report.errors:
logger.debug('Found errors in report')
for error in self.report.errors:
logger.debug(str(error))
try:
self.abi_errors.append(error)
except AttributeError:
self.abi_errors = [error]
# The job is unfixable due to ABINIT errors
logger.debug(
"%s: Found Errors or Bugs in ABINIT main output!" % self)
msg = "\n".join(
map(repr, self.report.errors + self.report.bugs))
return self.set_status(self.S_ABICRITICAL, msg=msg)
# 9) if we still haven't returned there is no indication of any error and the job can only still be running
# but we should actually never land here, or we have delays in the file system ....
# print('the job still seems to be running maybe it is hanging without producing output... ')
# Check time of last modification.
if self.output_file.exists and \
(time.time() - self.output_file.get_stat().st_mtime > self.manager.policy.frozen_timeout):
msg = "Task seems to be frozen, last change more than %s [s] ago" % self.manager.policy.frozen_timeout
return self.set_status(self.S_ERROR, msg)
return self.set_status(self.S_RUN)
# from GsTask
@property
def gsr_path(self):
"""Absolute path of the GSR file. Empty string if file is not present."""
# Lazy property to avoid multiple calls to has_abiext.
try:
return self._gsr_path
except AttributeError:
path = self.outdir.has_abiext("GSR")
if path: self._gsr_path = path
return path
def open_gsr(self):
"""
Open the GSR file located in the in self.outdir.
Returns :class:`GsrFile` object, None if file could not be found or file is not readable.
"""
gsr_path = self.gsr_path
if not gsr_path:
if self.status == self.S_OK:
logger.critical(
"%s reached S_OK but didn't produce a GSR file in %s" %
(self, self.outdir))
return None
# Open the GSR file.
from abipy.electrons.gsr import GsrFile
try:
return GsrFile(gsr_path)
except Exception as exc:
logger.critical("Exception while reading GSR file at %s:\n%s" %
(gsr_path, str(exc)))
return None
class AbinitTask(Task):
"""
Base class defining an abinit calculation
"""
# Prefixes for Abinit (input, output, temporary) files.
Prefix = collections.namedtuple("Prefix", "idata odata tdata")
pj = os.path.join
prefix = Prefix(pj("indata", "in"), pj("outdata","out"), pj("tmpdata","tmp"))
del Prefix, pj
# Basenames for Abinit input and output files.
Basename = collections.namedtuple("Basename", "input output files_file log_file stderr_file jobfile lockfile")
basename = Basename("run.in", "run.out", "run.files", "log", "stderr", "job.sh", "__lock__")
del Basename
def __init__(self, strategy, workdir, runmode, task_id=1, links=None, **kwargs):
"""
Args:
strategy:
Strategy instance descring the calculation.
workdir:
Path to the working directory.
runmode:
RunMode instance or string "sequential"
links:
List of WorkLink objects specifying the dependencies of the task.
kwargs:
keyword argumens (not used here)
"""
super(AbinitTask, self).__init__()
self.strategy = strategy
self.workdir = os.path.abspath(workdir)
self.runmode = RunMode.asrunmode(runmode).copy()
self.set_id(task_id)
# Connect this task to the other tasks
# (needed if are creating a Work instance with dependencies
self._links = []
if links is not None:
self._links = links
for link in links:
print("Adding ", link.get_abivars())
self.strategy.add_extra_abivars(link.get_abivars())
# Files required for the execution.
self.input_file = File(os.path.join(self.workdir, AbinitTask.basename.input))
self.output_file = File(os.path.join(self.workdir, AbinitTask.basename.output))
self.files_file = File(os.path.join(self.workdir, AbinitTask.basename.files_file))
self.log_file = File(os.path.join(self.workdir, AbinitTask.basename.log_file))
self.stderr_file = File(os.path.join(self.workdir, AbinitTask.basename.stderr_file))
# Find number of processors ....
#runhints = self.get_runhints()
#launcher = self.runmode.make_launcher(runhints)
# Build the launcher and store it.
self.set_launcher(self.runmode.make_launcher(self))
#def __str__(self):
def __repr__(self):
return "<%s at %s, workdir = %s>" % (
self.__class__.__name__, id(self), os.path.basename(self.workdir))
@property
def name(self):
return self.workdir
@property
def process(self):
try:
return self._process
except AttributeError:
# Attach a fake process so that we can poll it.
return FakeProcess()
@property
def launcher(self):
return self._launcher
def set_launcher(self, launcher):
self._launcher = launcher
@property
def jobfile_path(self):
"Absolute path of the job file (shell script)."
return os.path.join(self.workdir, self.basename.jobfile)
@property
def indata_dir(self):
head, tail = os.path.split(self.prefix.idata)
return os.path.join(self.workdir, head)
@property
def outdata_dir(self):
head, tail = os.path.split(self.prefix.odata)
return os.path.join(self.workdir, head)
@property
def tmpdata_dir(self):
head, tail = os.path.split(self.prefix.tdata)
return os.path.join(self.workdir, head)
def idata_path_from_ext(self, ext):
"Returns the path of the input file with extension ext"
return os.path.join(self.workdir, (self.prefix.idata + ext))
def odata_path_from_ext(self, ext):
"Returns the path of the output file with extension ext"
return os.path.join(self.workdir, (self.prefix.odata + ext))
@property
def pseudos(self):
"List of pseudos used in the calculation."
return self.strategy.pseudos
@property
def filesfile_string(self):
"String with the list of files and prefix needed to execute abinit."
lines = []
app = lines.append
pj = os.path.join
app(self.input_file.path) # Path to the input file
app(self.output_file.path) # Path to the output file
app(pj(self.workdir, self.prefix.idata)) # Prefix for in data
app(pj(self.workdir, self.prefix.odata)) # Prefix for out data
app(pj(self.workdir, self.prefix.tdata)) # Prefix for tmp data
# Paths to the pseudopotential files.
for pseudo in self.pseudos:
app(pseudo.path)
return "\n".join(lines)
@property
def to_dict(self):
raise NotimplementedError("")
d = {k: v.to_dict for k, v in self.items()}
d["@module"] = self.__class__.__module__
d["@class"] = self.__class__.__name__
d["strategy"] = self.strategy
d["workdir"] = workdir
return d
@staticmethod
def from_dict(d):
raise NotimplementedError("")
def isdone(self):
"True if the output file is complete."
return abinit_output_iscomplete(self.output_file.path)
@property
def isnc(self):
"True if norm-conserving calculation"
return all(p.isnc for p in self.pseudos)
@property
def ispaw(self):
"True if PAW calculation"
return all(p.ispaw for p in self.pseudos)
def make_input(self):
return self.strategy.make_input()
def outfiles(self):
"Return all the output data files produced."
files = list()
for file in os.listdir(self.outdata_dir):
if file.startswith(os.path.basename(self.prefix.odata)):
files.append(os.path.join(self.outdata_dir, file))
return files
def tmpfiles(self):
"Return all the input data files produced."
files = list()
for file in os.listdir(self.tmpdata_dir):
if file.startswith(os.path.basename(self.prefix.tdata)):
files.append(os.path.join(self.tmpdata_dir, file))
return files
def path_in_workdir(self, filename):
"Create the absolute path of filename in the top-level working directory."
return os.path.join(self.workdir, filename)
def path_in_indatadir(self, filename):
"Create the absolute path of filename in the indata directory."
return os.path.join(self.indata_dir, filename)
def path_in_outdatadir(self, filename):
"Create the absolute path of filename in the outdata directory."
return os.path.join(self.outdata_dir, filename)
def build(self, *args, **kwargs):
"""
Writes Abinit input files and directory. It does not overwrite files if they already exist.
"""
# Top-level dir
if not os.path.exists(self.workdir):
os.makedirs(self.workdir)
# Create dirs for input, output and tmp data.
if not os.path.exists(self.indata_dir):
os.makedirs(self.indata_dir)
if not os.path.exists(self.outdata_dir):
os.makedirs(self.outdata_dir)
if not os.path.exists(self.tmpdata_dir):
os.makedirs(self.tmpdata_dir)
# Write input file and files file.
if not self.input_file.exists:
self.input_file.write(self.make_input())
if not self.files_file.exists:
self.files_file.write(self.filesfile_string)
#if not os.path.exists(self.jobfile_path):
# with open(self.jobfile_path, "w") as f:
# f.write(str(self.jobfile))
def rmtree(self, *args, **kwargs):
"Remove all files and directories."
shutil.rmtree(self.workdir)
def get_runhints(self):
"""
Run abinit in sequential to obtain a set of possible
configurations for the number of processors.
RunMode is used to select the paramenters of the run.
"""
raise NotImplementedError("")
# Build a sequential launcher to run the code.
seq_launcher = self.runmode.make_seq_launcher(self)
# Launch the calculation.
retcode = seq_launcher.launch(self)
# Parse the output file to find the number of MPI processors,
# the number of OpenMP processors, memory estimate in Gb ...
runhints = RunHintsParser().parse(self.output_file.path)
self.set_status(self.S_READY)
return runhints
def setup(self, *args, **kwargs):
pass
def start(self, *args, **kwargs):
"""
Starts the calculation by performing the following steps:
- build dirs and files
- call the _setup method
- execute the job file via the shell or other means.
- return Results object
Keyword arguments:
verbose: (0)
Print message if verbose is not zero.
.. warning::
This method must be thread safe since we may want to run several indipendent
calculations with different python threads.
"""
self.build(*args, **kwargs)
self._setup(*args, **kwargs)
# Start the calculation in a subprocess and return
self._process = self.launcher.launch(self)
class AbiFireTask(FireTaskBase):
# List of `AbinitEvent` subclasses that are tested in the check_status method.
# Subclasses should provide their own list if they need to check the converge status.
CRITICAL_EVENTS = []
S_INIT = Status.from_string("Initialized")
S_LOCKED = Status.from_string("Locked")
S_READY = Status.from_string("Ready")
S_SUB = Status.from_string("Submitted")
S_RUN = Status.from_string("Running")
S_DONE = Status.from_string("Done")
S_ABICRITICAL = Status.from_string("AbiCritical")
S_QCRITICAL = Status.from_string("QCritical")
S_UNCONVERGED = Status.from_string("Unconverged")
S_ERROR = Status.from_string("Error")
S_OK = Status.from_string("Completed")
ALL_STATUS = [
S_INIT,
S_LOCKED,
S_READY,
S_SUB,
S_RUN,
S_DONE,
S_ABICRITICAL,
S_QCRITICAL,
S_UNCONVERGED,
S_ERROR,
S_OK,
]
def __init__(self, abiinput):
"""
Basic __init__, subclasses are supposed to define the same input parameters, add their own and call super for
the basic ones. The input parameter should be stored as attributes of the instance for serialization and
for inspection.
"""
self.abiinput = abiinput
@serialize_fw
def to_dict(self):
d = {}
for arg in inspect.getargspec(self.__init__).args:
if arg != "self":
val = self.__getattribute__(arg)
if hasattr(val, "as_dict"):
val = val.as_dict()
elif isinstance(val, (tuple, list)):
val = [v.as_dict() if hasattr(v, "as_dict") else v for v in val]
d[arg] = val
return d
@classmethod
def from_dict(cls, d):
dec = MontyDecoder()
kwargs = {k: dec.process_decoded(v) for k, v in d.items() if k in inspect.getargspec(cls.__init__).args}
return cls(**kwargs)
# from Task
def set_workdir(self, workdir):
"""Set the working directory."""
self.workdir = os.path.abspath(workdir)
# Files required for the execution.
self.input_file = File(os.path.join(self.workdir, "run.abi"))
self.output_file = File(os.path.join(self.workdir, "run.abo"))
self.files_file = File(os.path.join(self.workdir, "run.files"))
self.log_file = File(os.path.join(self.workdir, "run.log"))
self.stderr_file = File(os.path.join(self.workdir, "run.err"))
# Directories with input|output|temporary data.
self.indir = Directory(os.path.join(self.workdir, "indata"))
self.outdir = Directory(os.path.join(self.workdir, "outdata"))
self.tmpdir = Directory(os.path.join(self.workdir, "tmpdata"))
# from Task
def build(self):
"""
Creates the working directory and the input files of the :class:`Task`.
It does not overwrite files if they already exist.
"""
# Create dirs for input, output and tmp data.
self.indir.makedirs()
self.outdir.makedirs()
self.tmpdir.makedirs()
# Write files file and input file.
if not self.files_file.exists:
self.files_file.write(self.filesfile_string)
self.input_file.write(str(self.abiinput))
# from Task
# Prefixes for Abinit (input, output, temporary) files.
Prefix = collections.namedtuple("Prefix", "idata odata tdata")
pj = os.path.join
prefix = Prefix(pj("indata", "in"), pj("outdata", "out"), pj("tmpdata", "tmp"))
del Prefix, pj
# from AbintTask
@property
def filesfile_string(self):
"""String with the list of files and prefixes needed to execute ABINIT."""
lines = []
app = lines.append
pj = os.path.join
app(self.input_file.path) # Path to the input file
app(self.output_file.path) # Path to the output file
app(pj(self.workdir, self.prefix.idata)) # Prefix for input data
app(pj(self.workdir, self.prefix.odata)) # Prefix for output data
app(pj(self.workdir, self.prefix.tdata)) # Prefix for temporary data
# Paths to the pseudopotential files.
# Note that here the pseudos **must** be sorted according to znucl.
# Here we reorder the pseudos if the order is wrong.
ord_pseudos = []
znucl = self.abiinput.structure.to_abivars()["znucl"]
for z in znucl:
for p in self.abiinput.pseudos:
if p.Z == z:
ord_pseudos.append(p)
break
else:
raise ValueError("Cannot find pseudo with znucl %s in pseudos:\n%s" % (z, self.pseudos))
for pseudo in ord_pseudos:
app(pseudo.path)
return "\n".join(lines)
def run_abinit(self, fw_spec):
with open(self.files_file.path, "r") as stdin, open(self.log_file.path, "w") as stdout, open(
self.stderr_file.path, "w"
) as stderr:
p = subprocess.Popen(["mpirun", "abinit"], stdin=stdin, stdout=stdout, stderr=stderr)
(stdoutdata, stderrdata) = p.communicate()
self.returncode = p.returncode
def get_event_report(self):
"""
Analyzes the main output file for possible Errors or Warnings.
Returns:
:class:`EventReport` instance or None if the main output file does not exist.
"""
if not self.log_file.exists:
return None
parser = events.EventsParser()
try:
report = parser.parse(self.log_file.path)
return report
except parser.Error as exc:
# Return a report with an error entry with info on the exception.
logger.critical("%s: Exception while parsing ABINIT events:\n %s" % (self.log_file, str(exc)))
return parser.report_exception(self.log_file.path, exc)
def task_analysis(self, fw_spec):
status, msg = self.check_final_status()
if self.status != self.S_OK:
raise AbinitRuntimeError(self)
return FWAction(stored_data=dict(**self.report.as_dict()))
def run_task(self, fw_spec):
self.set_workdir(os.path.abspath("."))
self.build()
self.run_abinit(fw_spec)
return self.task_analysis(fw_spec)
def set_status(self, status, msg=None):
self.status = status
return status, msg
def check_final_status(self):
"""
This function checks the status of the task by inspecting the output and the
error files produced by the application. Based on abipy task checkstatus().
"""
# 2) see if an error occured at starting the job
# 3) see if there is output
# 4) see if abinit reports problems
# 5) see if err file exists and is empty
# 9) the only way of landing here is if there is a output file but no err files...
# 2) Check the returncode of the process (the process of submitting the job) first.
if self.returncode != 0:
# The job was not submitted properly
return self.set_status(self.S_QCRITICAL, msg="return code %s" % self.returncode)
# Analyze the stderr file for Fortran runtime errors.
err_msg = None
if self.stderr_file.exists:
err_msg = self.stderr_file.read()
# Start to check ABINIT status if the output file has been created.
if self.output_file.exists:
try:
self.report = self.get_event_report()
except Exception as exc:
msg = "%s exception while parsing event_report:\n%s" % (self, exc)
logger.critical(msg)
return self.set_status(self.S_ABICRITICAL, msg=msg)
if self.report.run_completed:
# Check if the calculation converged.
not_ok = self.report.filter_types(self.CRITICAL_EVENTS)
if not_ok:
return self.set_status(self.S_UNCONVERGED)
else:
return self.set_status(self.S_OK)
# Calculation still running or errors?
if self.report.errors or self.report.bugs:
# Abinit reported problems
if self.report.errors:
logger.debug("Found errors in report")
for error in self.report.errors:
logger.debug(str(error))
try:
self.abi_errors.append(error)
except AttributeError:
self.abi_errors = [error]
# The job is unfixable due to ABINIT errors
logger.debug("%s: Found Errors or Bugs in ABINIT main output!" % self)
msg = "\n".join(map(repr, self.report.errors + self.report.bugs))
return self.set_status(self.S_ABICRITICAL, msg=msg)
# 9) if we still haven't returned there is no indication of any error and the job can only still be running
# but we should actually never land here, or we have delays in the file system ....
# print('the job still seems to be running maybe it is hanging without producing output... ')
# Check time of last modification.
if self.output_file.exists and (
time.time() - self.output_file.get_stat().st_mtime > self.manager.policy.frozen_timeout
):
msg = "Task seems to be frozen, last change more than %s [s] ago" % self.manager.policy.frozen_timeout
return self.set_status(self.S_ERROR, msg)
return self.set_status(self.S_RUN)
# from GsTask
@property
def gsr_path(self):
"""Absolute path of the GSR file. Empty string if file is not present."""
# Lazy property to avoid multiple calls to has_abiext.
try:
return self._gsr_path
except AttributeError:
path = self.outdir.has_abiext("GSR")
if path:
self._gsr_path = path
return path
def open_gsr(self):
"""
Open the GSR file located in the in self.outdir.
Returns :class:`GsrFile` object, None if file could not be found or file is not readable.
"""
gsr_path = self.gsr_path
if not gsr_path:
if self.status == self.S_OK:
logger.critical("%s reached S_OK but didn't produce a GSR file in %s" % (self, self.outdir))
return None
# Open the GSR file.
from abipy.electrons.gsr import GsrFile
try:
return GsrFile(gsr_path)
except Exception as exc:
logger.critical("Exception while reading GSR file at %s:\n%s" % (gsr_path, str(exc)))
return None
