def generate_dax(self):
"Generate a workflow (DAX, config files, and replica catalog)"
ts = datetime.utcnow().strftime('%Y%m%dT%H%M%SZ')
dax = ADAG("refinement-%s" % ts)
# These are all the global input files for the workflow
coordinates = File(self.coordinates)
parameters = File(self.parameters)
extended_system = File(self.extended_system)
topfile = File(self.topfile)
sassena_db = File(self.sassena_db)
incoherent_db = File(self.incoherent_db)
coherent_db = File(self.coherent_db)
# This job untars the sassena db and makes it available to the other
# jobs in the workflow
untarjob = Job("tar", node_label="untar")
if self.is_synthetic_workflow:
untarjob.addArguments("-p", "-xzvf", sassena_db.name)
untarjob.addArguments("-a", "tar")
for output_file in [ "incoherent_db", "coherent_db" ]:
untarjob.addArguments(self.keg_params.output_file("tar", output_file, eval(output_file).name))
self.keg_params.add_keg_params(untarjob)
else:
untarjob.addArguments("-xzvf", sassena_db)
untarjob.uses(sassena_db, link=Link.INPUT)
untarjob.uses(incoherent_db, link=Link.OUTPUT, transfer=False)
untarjob.uses(coherent_db, link=Link.OUTPUT, transfer=False)
untarjob.profile("globus", "jobtype", "single")
untarjob.profile("globus", "maxwalltime", "1")
untarjob.profile("globus", "count", "1")
dax.addJob(untarjob)
# For each charge that was listed in the config file
for charge in self.charges:
structure = "Q%s.psf" % charge
# Equilibrate files
eq_conf = File("equilibrate_%s.conf" % charge)
eq_coord = File("equilibrate_%s.restart.coord" % charge)
eq_xsc = File("equilibrate_%s.restart.xsc" % charge)
eq_vel = File("equilibrate_%s.restart.vel" % charge)
# Production files
prod_conf = File("production_%s.conf" % charge)
prod_dcd = File("production_%s.dcd" % charge)
# Ptraj files
ptraj_conf = File("ptraj_%s.conf" % charge)
ptraj_fit = File("ptraj_%s.fit" % charge)
ptraj_dcd = File("ptraj_%s.dcd" % charge)
# Sassena incoherent files
incoherent_conf = File("sassenaInc_%s.xml" % charge)
fqt_incoherent = File("fqt_inc_%s.hd5" % charge)
# Sassena coherent files
coherent_conf = File("sassenaCoh_%s.xml" % charge)
fqt_coherent = File("fqt_coh_%s.hd5" % charge)
# Generate psf and configuration files for this charge pipeline
self.generate_psf(charge)
self.generate_eq_conf(charge, structure)
self.generate_prod_conf(charge, structure)
self.generate_ptraj_conf(charge)
self.generate_incoherent_conf(charge)
self.generate_coherent_conf(charge)
# Equilibrate job
eqjob = Job("namd", node_label="namd_eq_%s" % charge)
if self.is_synthetic_workflow:
eqjob.addArguments("-p", eq_conf)
eqjob.addArguments("-a", "namd_eq_%s" % charge)
eqjob.addArguments("-i", eq_conf.name, structure, coordinates.name,
parameters.name, extended_system.name)
task_label = "namd-eq"
for output_file in [ "eq_coord", "eq_xsc", "eq_vel" ]:
eqjob.addArguments(self.keg_params.output_file(task_label, output_file, eval(output_file).name))
self.keg_params.add_keg_params(eqjob, task_label)
else:
eqjob.addArguments(eq_conf)
eqjob.uses(eq_conf, link=Link.INPUT)
eqjob.uses(structure, link=Link.INPUT)
eqjob.uses(coordinates, link=Link.INPUT)
eqjob.uses(parameters, link=Link.INPUT)
eqjob.uses(extended_system, link=Link.INPUT)
eqjob.uses(eq_coord, link=Link.OUTPUT, transfer=False)
eqjob.uses(eq_xsc, link=Link.OUTPUT, transfer=False)
eqjob.uses(eq_vel, link=Link.OUTPUT, transfer=False)
if self.is_synthetic_workflow:
eqjob.profile("globus", "jobtype", "mpi")
eqjob.profile("globus", "maxwalltime", "1")
eqjob.profile("globus", "count", "8")
else:
eqjob.profile("globus", "jobtype", "mpi")
eqjob.profile("globus", "maxwalltime", self.getconf("equilibrate_maxwalltime"))
eqjob.profile("globus", "count", self.getconf("equilibrate_cores"))
dax.addJob(eqjob)
# Production job
prodjob = Job("namd", node_label="namd_prod_%s" % charge)
if self.is_synthetic_workflow:
prodjob.addArguments("-p", prod_conf)
prodjob.addArguments("-a", "namd_prod_%s" % charge)
prodjob.addArguments("-i", prod_conf.name, structure, coordinates.name,
parameters.name, eq_coord.name, eq_xsc.name, eq_vel.name)
task_label = "namd-prod"
prodjob.addArguments(self.keg_params.output_file(task_label, "prod_dcd", prod_dcd.name))
self.keg_params.add_keg_params(prodjob, task_label)
else:
prodjob.addArguments(prod_conf)
prodjob.uses(prod_conf, link=Link.INPUT)
prodjob.uses(structure, link=Link.INPUT)
prodjob.uses(coordinates, link=Link.INPUT)
prodjob.uses(parameters, link=Link.INPUT)
prodjob.uses(eq_coord, link=Link.INPUT)
prodjob.uses(eq_xsc, link=Link.INPUT)
prodjob.uses(eq_vel, link=Link.INPUT)
prodjob.uses(prod_dcd, link=Link.OUTPUT, transfer=True)
if self.is_synthetic_workflow:
prodjob.profile("globus", "jobtype", "mpi")
prodjob.profile("globus", "maxwalltime", "6")
prodjob.profile("globus", "count", "8")
else:
prodjob.profile("globus", "jobtype", "mpi")
prodjob.profile("globus", "maxwalltime", self.getconf("production_maxwalltime"))
prodjob.profile("globus", "count", self.getconf("production_cores"))
dax.addJob(prodjob)
dax.depends(prodjob, eqjob)
# ptraj job
ptrajjob = Job(namespace="amber", name="ptraj", node_label="amber_ptraj_%s" % charge)
if self.is_synthetic_workflow:
ptrajjob.addArguments("-p", topfile)
ptrajjob.addArguments("-a", "amber_ptraj_%s" % charge)
ptrajjob.addArguments("-i", topfile.name, ptraj_conf.name, prod_dcd.name)
task_label = "amber-ptraj"
for output_file in [ "ptraj_fit", "ptraj_dcd" ]:
ptrajjob.addArguments(self.keg_params.output_file(task_label, output_file, eval(output_file).name))
self.keg_params.add_keg_params(ptrajjob, task_label)
else:
ptrajjob.addArguments(topfile)
ptrajjob.setStdin(ptraj_conf)
ptrajjob.uses(topfile, link=Link.INPUT)
ptrajjob.uses(ptraj_conf, link=Link.INPUT)
ptrajjob.uses(prod_dcd, link=Link.INPUT)
ptrajjob.uses(ptraj_fit, link=Link.OUTPUT, transfer=True)
ptrajjob.uses(ptraj_dcd, link=Link.OUTPUT, transfer=True)
ptrajjob.profile("globus", "jobtype", "single")
ptrajjob.profile("globus", "maxwalltime", self.getconf("ptraj_maxwalltime"))
ptrajjob.profile("globus", "count", self.getconf("ptraj_cores"))
dax.addJob(ptrajjob)
dax.depends(ptrajjob, prodjob)
# sassena incoherent job
incojob = Job("sassena", node_label="sassena_inc_%s" % charge)
if self.is_synthetic_workflow:
incojob.addArguments("-p", "--config", incoherent_conf)
incojob.addArguments("-a", "sassena_inc_%s" % charge)
incojob.addArguments("-i", incoherent_conf.name, ptraj_dcd.name, incoherent_db.name, coordinates.name)
task_label = "sassena-inc"
incojob.addArguments(self.keg_params.output_file(task_label, "fqt_incoherent", fqt_incoherent.name))
self.keg_params.add_keg_params(incojob, task_label)
else:
incojob.addArguments("--config", incoherent_conf)
incojob.uses(incoherent_conf, link=Link.INPUT)
incojob.uses(ptraj_dcd, link=Link.INPUT)
incojob.uses(incoherent_db, link=Link.INPUT)
incojob.uses(coordinates, link=Link.INPUT)
incojob.uses(fqt_incoherent, link=Link.OUTPUT, transfer=True)
if self.is_synthetic_workflow:
incojob.profile("globus", "jobtype", "mpi")
incojob.profile("globus", "maxwalltime", "6")
incojob.profile("globus", "count", "8")
else:
incojob.profile("globus", "jobtype", "mpi")
incojob.profile("globus", "maxwalltime", self.getconf("sassena_maxwalltime"))
incojob.profile("globus", "count", self.getconf("sassena_cores"))
dax.addJob(incojob)
dax.depends(incojob, ptrajjob)
dax.depends(incojob, untarjob)
# sassena coherent job
cojob = Job("sassena", node_label="sassena_coh_%s" % charge)
if self.is_synthetic_workflow:
cojob.addArguments("-p", "--config", coherent_conf)
cojob.addArguments("-a", "sassena_coh_%s" % charge)
cojob.addArguments("-i", coherent_conf.name, ptraj_dcd.name, coherent_db.name, coordinates.name)
task_label = "sassena-coh"
cojob.addArguments(self.keg_params.output_file(task_label, "fqt_coherent", fqt_coherent.name))
self.keg_params.add_keg_params(cojob, task_label)
else:
cojob.addArguments("--config", coherent_conf)
cojob.uses(coherent_conf, link=Link.INPUT)
cojob.uses(ptraj_dcd, link=Link.INPUT)
cojob.uses(coherent_db, link=Link.INPUT)
cojob.uses(coordinates, link=Link.INPUT)
cojob.uses(fqt_coherent, link=Link.OUTPUT, transfer=True)
if self.is_synthetic_workflow:
cojob.profile("globus", "jobtype", "mpi")
cojob.profile("globus", "maxwalltime", "6")
cojob.profile("globus", "count", "8")
else:
cojob.profile("globus", "jobtype", "mpi")
cojob.profile("globus", "maxwalltime", self.getconf("sassena_maxwalltime"))
cojob.profile("globus", "count", self.getconf("sassena_cores"))
dax.addJob(cojob)
dax.depends(cojob, prodjob)
dax.depends(cojob, untarjob)
# Write the DAX file
dax.writeXMLFile(self.daxfile)
