def testRMSDVarEpsilon(self):
output_path = "tests/data/1f5k_adaptive_rmsd_vareps"
controlFile = "tests/data/templetized_controlFile_1f5k_rmsd_vareps.conf"
adaptiveSampling.main(controlFile)
self.check_succesful_simulation(output_path, 2)
# cleanup
shutil.rmtree(output_path)
def run(self, hook=False):
with helpers.cd(os.path.dirname(self.adaptive_file)):
if hook:
ad.main(self.adaptive_file,
clusteringHook=self.interactive_clustering)
else:
ad.main(self.adaptive_file)
def testReport_xtc(self):
output_path = "tests/data/1f5k_adaptive_rmsd_inv_report_xtc"
controlFile = "tests/data/templetized_controlFile_1f5k_rmsd_inv_report_xtc.conf"
adaptiveSampling.main(controlFile)
self.check_succesful_simulation(output_path, 2)
# cleanup
shutil.rmtree(output_path)
def testOpenMM1ab1(self):
output_path = "tests/data/openmm_1ab1"
controlFile = "tests/data/templetized_controlFile_1ab1_md.conf"
adaptiveSampling.main(controlFile)
self.check_succesful_simulation(output_path, 2, 4)
# cleanup
shutil.rmtree(output_path)
def testIntegration3(self):
"""
Simulations are actually run
"""
controlFile = "tests/data/3ptb_data/integrationTest3.conf"
goldenPath = "tests/data/3ptb_data/originTest3"
outputPath = "tests/data/3ptb_data/Test3"
elements = [22, 14, 16, 19, 1]
goldenClusters = []
metrics = [[1.00000e+00, 0.00000e+00, 0.00000e+00, -5.28675e+02, 6.41969e-03],
[1.00000e+00, 0.00000e+00, 0.00000e+00, -5.28657e+02, 1.88599e-02],
[1.00000e+00, 0.00000e+00, 0.00000e+00, -5.28675e+02, 6.41969e-03],
[1.00000e+00, 0.00000e+00, 0.00000e+00, -5.28665e+02, 1.24213e-02],
[1.00000e+00, 5.00000e+00, 5.00000e+00, -5.28660e+02, 1.75149e-02]]
for i in range(5):
pdb = atomset.PDB()
pdb.initialise(goldenPath+"/2/clustering/cluster_%d.pdb" % i, resname="AEN")
cluster = clustering.Cluster(pdb, 4, metrics=metrics[i])
cluster.elements = elements[i]
cluster.contacts = 0
goldenClusters.append(cluster)
# name = socket.gethostname()
# if "bsccv" not in name and "login" not in name:
# print("Some integration can't be run due to not having PELE installed")
# return True
# self.integrationTest(controlFile, goldenPath, outputPath, goldenClusters)
tmpFolder = "tmp_" + outputPath.replace("/", "_")
adaptiveSampling.main(controlFile)
self.check_succesful_simulation(outputPath, 3)
# cleanup
shutil.rmtree(outputPath)
shutil.rmtree(tmpFolder)
def testNullIndependentMetric_xtc(self):
output_path = "tests/data/1f5k_adaptive_null_independent_metric_xtc/"
controlFile = "tests/data/templetized_controlFile_1f5k_null_independent_metric.conf"
adaptiveSampling.main(controlFile)
self.check_succesful_simulation(output_path, 2)
# cleanup
shutil.rmtree(output_path)
def testCMEpsilon_xtc(self):
output_path = "tests/data/1f5k_adaptive_cm_eps_xtc"
controlFile = "tests/data/templetized_controlFile_1f5k_cm_epsilon_xtc.conf"
adaptiveSampling.main(controlFile)
self.check_succesful_simulation(output_path, 2)
# cleanup
shutil.rmtree(output_path)
def testOpenMM3ptb_noligand(self):
output_path = "tests/data/openmm_3ptb_no_ligand"
controlFile = "tests/data/templetized_controlFile_3ptb_no_ligand_md.conf"
adaptiveSampling.main(controlFile)
self.check_succesful_simulation(output_path, 1, 4)
# cleanup
shutil.rmtree(output_path)
def test_simulation_cofactors_ligand(self):
output_path = "tests/data/cofactors_ligand"
controlFile = "tests/data/cofactors_ligand.conf"
if os.path.exists(output_path):
shutil.rmtree(output_path)
adaptiveSampling.main(controlFile)
self.check_succesful_simulation(output_path, 1, 4)
# cleanup
shutil.rmtree(output_path)
def testRestartXTC(self):
output_path = "tests/data/openmm_xtc_restart"
controlFile = "tests/data/controlFile_restart_xtc.conf"
if os.path.exists(output_path):
shutil.rmtree(output_path)
shutil.copytree("tests/data/restart_xtc", output_path)
adaptiveSampling.main(controlFile)
self.check_succesful_simulation(output_path, 2, 1)
# cleanup
utilities.cleanup(output_path)
def testOpenMM1ab1_CUDA(self):
output_path = "tests/data/openmm_1ab1_CUDA"
controlFile = "tests/data/templetized_controlFile_1ab1_md_CUDA.conf"
adaptiveSampling.main(controlFile)
self.check_succesful_simulation(output_path, 2, 4)
# ensure all replicas have enough time to check the output before
# deleting it
time.sleep(10)
# cleanup
utilities.cleanup(output_path)
def testRestartAt1(self):
output_path = "tests/data/openmm_restart_1"
controlFile = "tests/data/templetized_controlFile_restart_1_md.conf"
if os.path.exists(output_path):
shutil.rmtree(output_path)
shutil.copytree("tests/data/restart_1", output_path)
adaptiveSampling.main(controlFile)
self.check_succesful_simulation(output_path, 2, 4)
# cleanup
shutil.rmtree(output_path)
def integrationTest(self, controlFile, goldenPath, outputPath, goldenClusters):
# Function to test --> integration test
adaptiveSampling.main(controlFile)
# Assertions
tmpFolder = "tmp_" + outputPath.replace("/", "_")
self.checkClusteringObjects(goldenClusters, outputPath)
self.checkStartingConformations(goldenPath, tmpFolder)
self.checkTrajectories(goldenPath, outputPath)
# cleanup
shutil.rmtree(outputPath)
shutil.rmtree(tmpFolder)
def testRestartEmptyClustering(self):
controlFile = "tests/data/3ptb_data/restartTest.conf"
outputPath = "tests/data/3ptb_data/RestartTest"
tmpFolder = "tmp_" + outputPath.replace("/", "_")
clusteringObjectPath = os.path.join(outputPath, "2", "clustering", "object.pkl")
if not os.path.exists(os.path.join(outputPath, "1", "clustering")):
os.makedirs(os.path.join(outputPath, "1", "clustering"))
# name = socket.gethostname()
# if "bsccv" not in name and "login" not in name:
# print("Some integration can't be run due to not having PELE installed")
# return True
# Function to test --> integration test
shutil.copy("tests/data/3ptb_data/object_test_bk.pkl", os.path.join(outputPath, "1", "clustering", "object.pkl"))
adaptiveSampling.main(controlFile)
# Assertions
self.assertTrue(adaptiveSampling.checkIntegrityClusteringObject(clusteringObjectPath))
# Remove clustering object from the simulation and create and empty one
open(clusteringObjectPath, "w").close()
# cleanup
shutil.rmtree(tmpFolder)
def automateSimulation(args):
"""
Run multiple AdaptivePELE simulation with the same parameters, changing
only the seed
:param args: Object containing the command line arguments
:type args: object
"""
controlFile = args.controlFile
numSimulations = args.numSimulations
nProcessors = args.nProcessors
nSteps = args.nSteps
simulationName = args.simulationName
simulationParameters = simulationrunner.SimulationParameters()
simulationParameters.templetizedControlFile = controlFile
simulationRunner = simulationrunner.SimulationRunner(simulationParameters)
epochs = args.epochs
if epochs:
rangeOfEpochs = epochs
else:
rangeOfEpochs = list(range(1, numSimulations + 1))
print("rangeOfEpochs", rangeOfEpochs)
for i in rangeOfEpochs:
controlFileDictionary = {"SEED": "%d%d%d", "OUTPUTPATH": "%s_%d"}
SEED_i = int(controlFileDictionary["SEED"] % (i, nProcessors, nSteps))
controlFileDictionary["SEED"] = SEED_i
outputPath_i = controlFileDictionary["OUTPUTPATH"] % (simulationName,
i)
controlFileDictionary["OUTPUTPATH"] = outputPath_i
controlFileName = "tmp_%s_controlfile_%s_%d.conf" % (
os.path.splitext(controlFile)[0], simulationName, i)
controlFileName = controlFileName.replace("/", "_")
simulationRunner.makeWorkingControlFile(controlFileName,
controlFileDictionary)
print("Starting simulation %d" % i)
adaptiveSampling.main(controlFileName)
def run_adaptive(args):
"""
Main function to prepare and launch the simulation.
1) Create EnviroBuilder - variables, folders, logger...
2) Prepare ligand and receptor
3) Launch simulation
4) Analyse simulation
"""
builder = ParametersBuilder()
parameters = builder.build_adaptive_variables(args)
parameters.create_files_and_folders()
shutil.copy(args.yamlfile, parameters.pele_dir)
if parameters.adaptive_restart and not parameters.only_analysis:
with helpers.cd(parameters.pele_dir):
adaptiveSampling.main(parameters.ad_ex_temp)
parameters.logger.info("Simulation run successfully (:\n\n")
elif not parameters.only_analysis:
parameters.logger.info(
"System: {}; Platform Functionality: {}\n\n".format(
parameters.residue, parameters.software))
# Create inputs directory
if not os.path.exists(parameters.inputs_dir):
os.mkdir(parameters.inputs_dir)
if parameters.perturbation:
syst = sp.SystemBuilder.build_system(
parameters.system,
args.mae_lig,
args.residue,
parameters.pele_dir,
inputs_dir=parameters.inputs_dir,
)
else:
syst = sp.SystemBuilder(
parameters.system,
None,
None,
parameters.pele_dir,
inputs_dir=parameters.inputs_dir,
)
parameters.logger.info("Prepare complex {}".format(syst.system))
# If user overrides 'system' with 'input'
if parameters.input:
parameters.inputs_simulation = []
for input in parameters.input:
input_path = os.path.join(parameters.inputs_dir,
os.path.basename(input))
shutil.copy(input, input_path)
if parameters.no_ppp:
input_proc = input_path
else:
input_proc = os.path.basename(
ppp.main(
input_path,
parameters.
inputs_dir, # to ensure it goes to pele_dir/inputs, not pele_dir
output_pdb=[
"",
],
charge_terminals=args.charge_ter,
no_gaps_ter=args.gaps_ter,
constrain_smiles=None,
ligand_pdb=parameters.ligand_ref,
)[0])
input_proc = os.path.join(parameters.inputs_dir, input_proc)
parameters.inputs_simulation.append(input_proc)
parameters.adap_ex_input = ", ".join([
'"' + input + '"' for input in parameters.inputs_simulation
]).strip('"')
# If randomization in necessary (PPI, site_finder, global exploration)...
elif args.full or args.randomize or args.ppi:
ligand_positions, box_radius, box_center = rd.randomize_starting_position(
parameters.ligand_ref,
parameters.receptor,
outputfolder=parameters.inputs_dir,
nposes=parameters.poses,
test=parameters.test,
user_center=parameters.center_of_interface,
logger=parameters.logger,
)
if not args.gpcr_orth:
parameters.box_center = box_center
parameters.box_radius = box_radius
if parameters.no_ppp:
receptor = syst.system
else:
receptor = ppp.main(
syst.system,
parameters.
inputs_dir, # to ensure it goes to pele_dir/input, not pele_dir
output_pdb=[
"",
],
charge_terminals=args.charge_ter,
no_gaps_ter=args.gaps_ter,
constrain_smiles=None,
ligand_pdb=parameters.ligand_ref,
)[0]
inputs = rd.join(
receptor,
ligand_positions,
parameters.residue,
output_folder=parameters.inputs_dir,
)
inputs = [
os.path.join(parameters.inputs_dir, inp) for inp in inputs
]
parameters.adap_ex_input = ", ".join(
['"' + input + '"' for input in inputs]).strip('"')
hp.silentremove(ligand_positions)
# Prepare System
if (
parameters.no_ppp or parameters.input
): # No need to run system through PPP, if we already preprocessed parameters.input
missing_residues = []
if parameters.input:
# If we have more than one input
for input in parameters.input:
shutil.copy(input, parameters.inputs_dir)
else:
shutil.copy(parameters.system, parameters.inputs_dir)
else:
parameters.nonstandard.extend(hp.find_nonstd_residue(syst.system))
parameters.system, missing_residues, _, _, _ = ppp.main(
syst.system,
parameters.inputs_dir,
output_pdb=[
"",
],
charge_terminals=args.charge_ter,
no_gaps_ter=args.gaps_ter,
mid_chain_nonstd_residue=parameters.nonstandard,
skip=parameters.skip_prep,
back_constr=parameters.ca_constr,
constrain_smiles=None,
ligand_pdb=parameters.ligand_ref,
ca_interval=parameters.ca_interval,
)
parameters.constraints = alpha_constraints.retrieve_constraints(
parameters.system,
interval=parameters.ca_interval,
back_constr=parameters.ca_constr,
ter_constr=parameters.terminal_constr,
)
# Metal constraints
if not args.no_metal_constraints:
metal_constraints, parameters.external_constraints = mc.main(
args.external_constraints,
os.path.join(
parameters.inputs_dir,
parameters.adap_ex_input.split(",")[0].strip().strip('"'),
),
syst.system,
permissive=parameters.permissive_metal_constr,
all_metals=args.constrain_all_metals,
external=parameters.external_constraints,
logger=parameters.logger,
)
parameters.external_constraints = hp.retrieve_constraints_for_pele(
parameters.external_constraints, parameters.system)
metal_constraints_json = hp.retrieve_constraints_for_pele(
metal_constraints,
os.path.join(
parameters.inputs_dir,
parameters.adap_ex_input.split(",")[0].strip().strip('"'),
),
)
parameters.external_constraints.extend(metal_constraints_json)
else:
parameters.external_constraints = hp.retrieve_constraints_for_pele(
parameters.external_constraints, parameters.system)
# Keep JSON ordered by having first title and then constraints
if parameters.external_constraints:
parameters.constraints = (parameters.constraints[0:1] +
parameters.external_constraints +
parameters.constraints[1:])
if parameters.remove_constraints:
parameters.constraints = ""
parameters.logger.info("Complex {} prepared\n\n".format(
parameters.system))
# Ligand parameters and simulation box
if parameters.perturbation:
ligand_params = lg.LigandParametrization(parameters)
ligand_params.generate()
box = bx.BoxSetter(
parameters.box_center,
parameters.box_radius,
parameters.ligand_ref,
parameters.logger,
)
parameters.box = box.generate_json()
else:
parameters.box = ""
# Parametrize missing residues
for res, __, _ in missing_residues:
if res != args.residue and res not in parameters.skip_ligand_prep:
parameters.logger.info(
"Creating template for residue {}".format(res))
with hp.cd(parameters.pele_dir):
mr.create_template(parameters, res)
parameters.logger.info("Template {}z created\n\n".format(res))
# Solvent parameters
solvent = sv.ImplicitSolvent(
parameters.solvent,
parameters.obc_tmp,
parameters.template_folder,
parameters.obc_file,
parameters.logger,
)
solvent.generate()
# Build PCA
if parameters.pca_traj:
pca_obj = pca.PCA(parameters.pca_traj, parameters.pele_dir)
parameters.pca = pca_obj.generate(parameters.logger)
# Core constraints based on SMILES string
if parameters.constrain_core:
smiles_input_pdb = os.path.join(
parameters.inputs_dir,
parameters.adap_ex_input.split(",")[0])
smiles = smiles_constraints.SmilesConstraints(
smiles_input_pdb,
parameters.constrain_core,
parameters.residue,
parameters.chain,
parameters.constrain_core_spring,
)
smi_constraint = smiles.run()
parameters.constraints = (parameters.constraints[0:1] +
smi_constraint +
parameters.constraints[1:])
# Waters
input_waters = [
input.strip().strip('"')
for input in parameters.adap_ex_input.split(",")
]
input_waters = [
os.path.join(parameters.inputs_dir, inp) for inp in input_waters
]
water_obj = wt.WaterIncluder(
input_waters,
parameters.n_waters,
user_waters=parameters.waters,
ligand_perturbation_params=parameters.parameters,
water_center=args.water_center,
water_radius=parameters.water_radius,
allow_empty_selectors=parameters.allow_empty_selectors,
water_temp=parameters.water_temp,
water_trials=parameters.water_trials,
water_overlap=parameters.water_overlap,
water_constr=parameters.water_constr,
test=parameters.test,
water_freq=parameters.water_freq,
ligand_residue=parameters.residue,
)
water_obj.run()
parameters.parameters = water_obj.ligand_perturbation_params
# Check if atoms need mapping due to preprocessing
args = AtomMapper(args, parameters, syst.system).run()
# Metrics builder - builds JSON strings for PELE to be able to track atom distances, RMSD, etc.
metrics = mt.MetricBuilder()
parameters.metrics = (metrics.distance_to_atom_json(
os.path.join(
parameters.inputs_dir,
parameters.adap_ex_input.split(",")[0].strip().strip('"'),
),
args.atom_dist,
) if args.atom_dist else "")
parameters.native = (metrics.rsmd_to_json(
args.native, parameters.chain) if args.native else "")
# interaction restrictions
# TODO this is not the place to initialize parameters for the interaction restrictions
if args.interaction_restrictions:
interaction_restrictions = ir.InteractionRestrictionsBuilder()
interaction_restrictions.parse_interaction_restrictions(
parameters.system, args.interaction_restrictions)
parameters.met_interaction_restrictions = (
interaction_restrictions.metrics_to_json())
parameters.interaction_restrictions = (
interaction_restrictions.conditions_to_json())
else:
parameters.met_interaction_restrictions = ""
parameters.interaction_restrictions = ""
# metal polarisation
if parameters.polarize_metals:
mp.change_metal_charges(
parameters.template_folder,
parameters.forcefield,
parameters.polarization_factor,
parameters.system,
)
# Point adaptive.conf to input dir
parameters.adap_ex_input = os.path.join(parameters.inputs_dir,
parameters.adap_ex_input)
# Fill in simulation templates
adaptive = ad.SimulationBuilder(parameters.ad_ex_temp,
parameters.pele_exit_temp,
parameters.topology)
adaptive.generate_inputs(parameters, water_obj)
# Run simulation only if we are not in debug mode
if not parameters.debug:
parameters.logger.info("Running Simulation")
adaptive.run()
parameters.logger.info("Simulation run successfully (:\n\n")
# Run analysis
if parameters.analyse and not parameters.debug:
from pele_platform.analysis import Analysis
analysis_folder = os.path.join(parameters.pele_dir, "results")
analysis = Analysis.from_parameters(parameters)
analysis.generate(analysis_folder,
clustering_type=parameters.clustering_method.lower(),
bandwidth=parameters.bandwidth,
analysis_nclust=parameters.analysis_nclust,
max_top_clusters=parameters.max_top_clusters,
min_population=parameters.min_population)
return parameters
def run(self, hook=False) -> None:
# Launch montecarlo simulation
with helpers.cd(os.path.dirname(self.adaptive_file)):
ad.main(self.adaptive_file)
def run_adaptive(args: YamlParser):
"""
Main function to prepare and launch the simulation.
1) Create Parameters - variables, folders, logger...
2) Prepare ligand and receptor
3) Launch simulation
4) Analyse simulation
"""
builder = ParametersBuilder()
parameters = builder.build_adaptive_variables(args)
parameters.create_files_and_folders()
shutil.copy(args.yamlfile, parameters.pele_dir)
missing_residues = []
if parameters.adaptive_restart and not parameters.only_analysis:
with helpers.cd(parameters.pele_dir):
adaptiveSampling.main(parameters.ad_ex_temp)
parameters.logger.info("Simulation run successfully (:\n\n")
args.adaptive_restart = False
elif not parameters.only_analysis and not parameters.restart:
parameters.logger.info(
"System: {}; Platform Functionality: {}\n\n".format(
parameters.residue, parameters.software))
pdb_checker.PDBChecker(parameters.system).check_negative_residues()
# Create inputs directory
if not os.path.exists(parameters.inputs_dir):
os.mkdir(parameters.inputs_dir)
if parameters.perturbation:
syst = sp.SystemBuilder.build_system(
parameters.system,
args.mae_lig,
args.residue,
parameters.pele_dir,
inputs_dir=parameters.inputs_dir)
else:
syst = sp.SystemBuilder(parameters.system,
None,
None,
parameters.pele_dir,
inputs_dir=parameters.inputs_dir)
parameters.logger.info("Prepare complex {}".format(syst.system))
# If user overrides 'system' with 'input'
if parameters.input:
parameters.inputs_simulation = []
for input in parameters.input:
input_path = os.path.join(parameters.inputs_dir,
os.path.basename(input))
shutil.copy(input, input_path)
if parameters.no_ppp:
input_proc = input_path
else:
input_proc, missing_residues, _, _, _ = ppp.main(
input_path,
parameters.
inputs_dir, # to ensure it goes to pele_dir/inputs, not pele_dir
output_pdb=[
"",
],
charge_terminals=args.charge_ter,
no_gaps_ter=args.gaps_ter,
constrain_smiles=None,
ligand_pdb=parameters.ligand_ref)
input_proc = os.path.basename(input_proc)
input_proc = os.path.join(parameters.inputs_dir, input_proc)
parameters.inputs_simulation.append(input_proc)
parameters.adap_ex_input = ", ".join([
'"' + input + '"' for input in parameters.inputs_simulation
]).strip('"')
# If randomization in necessary (PPI, site_finder, global exploration)...
elif args.full or args.randomize or args.ppi or args.site_finder:
ligand_positions, box_radius, box_center = rd.randomize_starting_position(
parameters, args.box_center, args.box_radius)
if parameters.no_ppp:
receptor = syst.system
else:
receptor, missing_residues, _, _, _ = ppp.main(
syst.system,
parameters.
inputs_dir, # to ensure it goes to pele_dir/input, not pele_dir
output_pdb=[
"",
],
charge_terminals=args.charge_ter,
no_gaps_ter=args.gaps_ter,
constrain_smiles=None,
ligand_pdb=parameters.ligand_ref)
inputs = rd.join(receptor,
ligand_positions,
parameters.residue,
output_folder=parameters.inputs_dir)
parameters.input = [
os.path.join(parameters.inputs_dir, inp) for inp in inputs
]
if args.gpcr_orth or args.out_in:
randomization_box_radius, randomization_box_center = rd.set_box(
args.final_site if args.final_site else
args.orthosteric_site, ligand_positions, parameters.system)
if not args.box_center:
parameters.box_center = randomization_box_center
if not args.box_radius:
parameters.box_radius = randomization_box_radius
else:
parameters.box_center = box_center
parameters.box_radius = box_radius
parameters.adap_ex_input = ", ".join(
['"' + input + '"' for input in parameters.input]).strip('"')
hp.silentremove(ligand_positions)
# Prepare System
if parameters.no_ppp or parameters.input: # No need to run system through PPP, if we already preprocessed
# parameters.input
if parameters.input:
# If we have more than one input
for input in parameters.input:
try:
shutil.copy(input, parameters.inputs_dir)
except shutil.SameFileError: # systems that go through randomization are already moved
pass
else:
shutil.copy(parameters.system, parameters.inputs_dir)
else:
parameters.nonstandard.extend(hp.find_nonstd_residue(syst.system))
parameters.system, missing_residues, _, _, _ = ppp.main(
syst.system,
parameters.inputs_dir,
output_pdb=[
"",
],
charge_terminals=args.charge_ter,
no_gaps_ter=args.gaps_ter,
mid_chain_nonstd_residue=parameters.nonstandard,
skip=parameters.skip_prep,
back_constr=parameters.ca_constr,
constrain_smiles=None,
ligand_pdb=parameters.ligand_ref,
ca_interval=parameters.ca_interval)
parameters.constraints = alpha_constraints.retrieve_constraints(
parameters.system,
interval=parameters.ca_interval,
back_constr=parameters.ca_constr,
ter_constr=parameters.terminal_constr)
# Metal constraints
if not args.no_metal_constraints:
metal_constraints, parameters.external_constraints = mc.main(
args.external_constraints,
os.path.join(
parameters.inputs_dir,
parameters.adap_ex_input.split(",")[0].strip().strip('"')),
syst.system,
permissive=parameters.permissive_metal_constr,
all_metals=args.constrain_all_metals,
external=parameters.external_constraints,
logger=parameters.logger)
parameters.external_constraints = hp.retrieve_constraints_for_pele(
parameters.external_constraints, parameters.system)
metal_constraints_json = hp.retrieve_constraints_for_pele(
metal_constraints,
os.path.join(
parameters.inputs_dir,
parameters.adap_ex_input.split(",")[0].strip().strip('"')))
parameters.external_constraints.extend(metal_constraints_json)
else:
parameters.external_constraints = hp.retrieve_constraints_for_pele(
parameters.external_constraints, parameters.system)
# Keep JSON ordered by having first title and then constraints
if parameters.external_constraints:
parameters.constraints = (parameters.constraints[0:1] +
parameters.external_constraints +
parameters.constraints[1:])
if parameters.remove_constraints:
parameters.constraints = ""
parameters.logger.info(f"Complex {parameters.system} prepared\n\n")
# Ligand/metal and solvent parameters
if (parameters.perturbation or
parameters.sidechain_perturbation) and parameters.use_peleffy:
ligand_parametrizer = parametrizer.Parametrizer.from_parameters(
parameters)
ligand_parametrizer.parametrize_ligands_from(
pdb_file=syst.system, ppp_file=parameters.system)
elif (parameters.perturbation or parameters.sidechain_perturbation
) and not parameters.use_peleffy:
# Parametrize the ligand
ligand_params = lg.LigandParametrization(parameters)
ligand_params.generate()
# Parametrize missing residues identified by PPP
for res, __, _ in missing_residues:
if res != args.residue and res not in parameters.skip_ligand_prep:
parameters.logger.info(
"Creating template for residue {}".format(res))
with hp.cd(parameters.pele_dir):
mr.create_template(parameters, res)
parameters.logger.info(
"Template {}z created\n\n".format(res))
# Covalent residue parametrization should not run in refinement simulation
if parameters.covalent_residue and os.path.basename(
parameters.pele_dir) != "2_refinement":
parametrizer.parametrize_covalent_residue(
parameters.pele_data,
parameters.pele_dir,
parameters.gridres,
parameters.residue_type,
parameters.residue,
ppp_system=parameters.system)
if parameters.ligand_conformations:
ligand_conformations.LigandConformations(
path=parameters.ligand_conformations,
system=parameters.system,
resname=parameters.residue,
forcefield=parameters.forcefield,
pele_dir=parameters.pele_dir).generate()
# Create simulation box, if performing perturbation
if parameters.perturbation and parameters.box:
box = bx.BoxSetter(parameters.box_center, parameters.box_radius,
parameters.ligand_ref, parameters.logger)
parameters.box = box.generate_json()
else:
parameters.box = ""
# Solvent parameters
solvent = sv.ImplicitSolvent(
parameters.solvent,
parameters.obc_tmp,
parameters.template_folder,
parameters.obc_file,
parameters.logger,
)
solvent.generate()
# Build PCA
if parameters.pca_traj:
pca_obj = pca.PCA(parameters.pca_traj, parameters.pele_dir)
parameters.pca = pca_obj.generate(parameters.logger)
# Core constraints based on SMILES string
if parameters.constrain_core:
smiles_input_pdb = os.path.join(
parameters.inputs_dir,
parameters.adap_ex_input.split(",")[0])
smiles = smiles_constraints.SmilesConstraints(
smiles_input_pdb, parameters.constrain_core,
parameters.residue, parameters.chain,
parameters.constrain_core_spring)
smi_constraint = smiles.run()
parameters.constraints = (parameters.constraints[0:1] +
smi_constraint +
parameters.constraints[1:])
# Waters
input_waters = [
input.strip().strip('"')
for input in parameters.adap_ex_input.split(",")
]
input_waters = [
os.path.join(parameters.inputs_dir, inp) for inp in input_waters
]
water_obj = wt.WaterIncluder(
input_waters,
parameters.n_waters,
user_waters=parameters.waters,
ligand_perturbation_params=parameters.parameters,
water_center=args.water_center,
water_radius=parameters.water_radius,
allow_empty_selectors=parameters.allow_empty_selectors,
water_temp=parameters.water_temp,
water_trials=parameters.water_trials,
water_overlap=parameters.water_overlap,
water_constr=parameters.water_constr,
test=parameters.test,
water_freq=parameters.water_freq,
ligand_residue=parameters.residue)
water_obj.run()
parameters.parameters = water_obj.ligand_perturbation_params
parameters.water_ids_to_track = water_obj.water_ids_to_track
# Check if atoms need mapping due to preprocessing
args = AtomMapper(args, parameters, syst.system).run()
# Metrics builder - builds JSON strings for PELE to be able to
# track atom distances, RMSD, etc.
metrics = mt.MetricBuilder()
parameters.metrics = (metrics.distance_to_atom_json(
os.path.join(
parameters.inputs_dir,
parameters.adap_ex_input.split(",")[0].strip().strip('"'),
),
args.atom_dist,
) if args.atom_dist else "")
parameters.native = (metrics.rmsd_to_json(
args.native, parameters.chain) if args.native else "")
parameters.local_nonbonding_energy = metrics.local_nonbonding_energy_json(
parameters.covalent_residue, parameters.nonbonding_radius)
# metal polarisation
if parameters.polarize_metals:
mp.change_metal_charges(
parameters.template_folder,
parameters.forcefield,
parameters.polarization_factor,
parameters.system,
)
# Predict cluster values in a short equilibration
if parameters.cluster_conditions == "auto":
preequilibration = PreEquilibrator(args, parameters)
parameters.cluster_conditions = preequilibration.run()
# Point adaptive.conf to input dir
parameters.adap_ex_input = os.path.join(parameters.inputs_dir,
parameters.adap_ex_input)
# Fill in simulation templates
adaptive = ad.SimulationBuilder(parameters.ad_ex_temp,
parameters.pele_exit_temp,
parameters.topology)
adaptive.generate_inputs(parameters, water_obj)
# Run simulation only if we are not in debug mode
if not parameters.debug:
parameters.logger.info("Running Simulation")
adaptive.run()
parameters.logger.info("Simulation run successfully (:\n\n")
elif parameters.restart and not parameters.only_analysis:
# Start simulation from scratch (unlike adaptive_restart) but use files created in debug mode
parameters.logger.info(
f"Launching simulation from {parameters.pele_dir}")
adaptive = ad.SimulationBuilder(parameters.ad_ex_temp,
parameters.pele_exit_temp,
parameters.topology)
adaptive.run()
parameters.logger.info("Simulation run successfully (:\n\n")
# Run analysis
if parameters.analyse and not parameters.debug:
from pele_platform.analysis import Analysis
# Retrieve water IDs to track from existing pele.conf, if running analysis only
if parameters.only_analysis:
parameters.water_ids_to_track = wt.water_ids_from_conf(
parameters.pele_temp)
analysis_folder = os.path.join(parameters.pele_dir, "results")
analysis = Analysis.from_parameters(parameters)
analysis.generate(
analysis_folder,
clustering_type=parameters.clustering_method.lower(),
bandwidth=parameters.bandwidth,
analysis_nclust=parameters.analysis_nclust,
max_top_clusters=parameters.max_top_clusters,
min_population=parameters.min_population,
max_top_poses=parameters.max_top_poses,
top_clusters_criterion=parameters.top_clusters_criterion,
representatives_criterion=parameters.
cluster_representatives_criterion)
return parameters
def main(jsonBlock):
# Parameters
firstControlFile = jsonBlock["firstControlFile"]
ntrajs = jsonBlock["ntrajs"]
resname = jsonBlock["resname"]
symmetries = jsonBlock["symmetries"]
trajFolder = jsonBlock["trajFolder"]
clusteringObject = jsonBlock["clusteringObject"]
clusteringThreshold = jsonBlock["clusteringThreshold"]
pathwayFilename = jsonBlock["pathwayFilename"]
templetizedInitialName = jsonBlock["templetizedInitialName"].encode()
secondControlFileTemp = jsonBlock["secondControlFileTemp"]
secondControlFile = jsonBlock["secondControlFile"]
distanceThreshold = jsonBlock["distanceThreshold"]
if firstControlFile:
# Run first adaptive
adaptiveSampling.main(firstControlFile)
# Cluster trajectories
ClOrd, thresholdCalculator = clusterTrajectories(resname,
trajFolder,
clusteringObject,
clusteringThreshold,
symmetries=symmetries)
# use graph algorithm to establish a path
initial_cluster = 0
final_cluster = ClOrd.getOptimalMetric()
pathway = createPathway(initial_cluster, final_cluster, ClOrd,
distanceThreshold)
# write pathway into a single trajectory
writePathwayTrajectory(ClOrd, pathway, pathwayFilename)
# create clustering object with only the pathway clusters
ClPath = OrderedContactsClustering(thresholdCalculator,
resname=resname,
reportBaseFilename="report",
columnOfReportFile=4,
symmetries=symmetries)
ClPath.clusters.clusters = map(lambda x: ClOrd.clusters.clusters[x],
pathway)
# spawning along the trajectory
spawningParams = spawning.SpawningParams()
densityCalculatorBuilder = densitycalculator.DensityCalculatorBuilder()
densityCalculator = densityCalculatorBuilder.build({})
spawningPathway = spawning.InverselyProportionalToPopulationCalculator(
densityCalculator)
# Set a least 1 processors from the extrems of the path
degeneracies = spawningPathway.calculate(ClPath.clusters.clusters,
ntrajs - 2, spawningParams)
degeneracies[0] += 1
degeneracies[-1] += 1
print "degeneracies over pathway:"
print degeneracies
print ""
plots = False
if plots:
plotMetricsDegeneracy(ClPath, resname, degeneracies)
# Prepare second adaptive
makeNewControlFile(degeneracies, ClPath, templetizedInitialName,
secondControlFileTemp, secondControlFile)
adaptiveSampling.main(secondControlFile)
def run_adaptive(args):
# Build Folders and Logging and env variable that will containt
#all main attributes of the simulation
env = pele.EnviroBuilder.build_env(args)
shutil.copy(args.yamlfile, env.pele_dir)
if env.adaptive_restart:
with helpers.cd(env.pele_dir):
adt.main(env.ad_ex_temp)
env.logger.info("Simulation run succesfully (:\n\n")
else:
##PREPWIZARD##
if args.prepwizard:
args.system = pp.run_prepwizard(args.system)
env.logger.info("System: {}; Platform Functionality: {}\n\n".format(env.residue, env.software))
if env.perturbation:
syst = sp.SystemBuilder.build_system(args.system, args.mae_lig, args.residue, env.pele_dir)
else:
syst = sp.SystemBuilder(args.system, None, None, env.pele_dir)
env.logger.info("Prepare complex {}".format(syst.system))
########Choose your own input####################
if args.input:
env.inputs_simulation = []
for input in env.input:
input_path = os.path.join(env.pele_dir, os.path.basename(input))
shutil.copy(input, input_path)
input_proc = os.path.basename(ppp.main(input_path, env.pele_dir, output_pdb=["" , ],
charge_terminals=args.charge_ter, no_gaps_ter=args.gaps_ter)[0])
env.inputs_simulation.append(input_proc)
hp.silentremove([input_path])
env.adap_ex_input = ", ".join(['"' + input + '"' for input in env.inputs_simulation])
elif args.full or args.randomize:
ligand_positions, box_center, box_radius = rd.randomize_starting_position(env.ligand_ref, "input_ligand.pdb", env.residue, env.receptor, None, None, env, poses=env.poses)
# Use choice stays as first priority
env.box_center = box_center if not env.box_center else env.box_center
env.box_radius = box_radius if not env.box_radius else env.box_radius
receptor = ppp.main(syst.system, env.pele_dir, output_pdb=["" , ],
charge_terminals=args.charge_ter, no_gaps_ter=args.gaps_ter)[0]
inputs = rd.join(receptor, ligand_positions, env)
env.adap_ex_input = ", ".join(['"' + os.path.basename(input) + '"' for input in inputs]).strip('"')
hp.silentremove(ligand_positions)
#Parsing input for errors and saving them as inputs
##########Prepare System################
if env.no_ppp:
env.adap_ex_input = system_fix = syst.system
missing_residues = []
gaps = {}
metals = {}
env.constraints = ct.retrieve_constraints(system_fix, gaps, metals, back_constr=env.ca_constr)
shutil.copy(env.adap_ex_input, env.pele_dir)
else:
system_fix, missing_residues, gaps, metals, env.constraints = ppp.main(syst.system, env.pele_dir, output_pdb=["" , ], charge_terminals=args.charge_ter, no_gaps_ter=args.gaps_ter, mid_chain_nonstd_residue=env.nonstandard, skip=env.skip_prep, back_constr=env.ca_constr)
if env.remove_constraints:
env.constraints = ""
env.logger.info(cs.SYSTEM.format(missing_residues, gaps, metals))
env.logger.info("Complex {} prepared\n\n".format(system_fix))
############Build metrics##################
env.logger.info("Setting metrics")
metrics = mt.Metrics_Builder(syst.system)
if env.atom_dist:
metrics.distance_to_atom(args.atom_dist)
env.metrics = "\n".join(metrics.get_metrics()) if metrics.get_metrics() else None
env.logger.info("Metrics set\n\n")
############Parametrize Ligand###############
if env.perturbation:
env.logger.info("Creating template for residue {}".format(args.residue))
with hp.cd(env.pele_dir):
plop.parametrize_miss_residues(args, env, syst)
env.logger.info("Template {}z created\n\n".format(args.residue.lower()))
if env.external_template:
for template_file in env.external_template:
cmd_to_move_template = "cp {} {}".format(template_file, env.template_folder)
subprocess.call(cmd_to_move_template.split())
if env.external_rotamers:
for rotamer_file in env.external_rotamers:
cmd_to_move_rotamer_file = "cp {} {}".format(rotamer_file, env.rotamers_folder)
subprocess.call(cmd_to_move_rotamer_file.split())
###########Parametrize missing residues#################
for res, __, _ in missing_residues:
if res != args.residue:
env.logger.info("Creating template for residue {}".format(res))
with hp.cd(env.pele_dir):
mr.create_template(args, env, res)
env.logger.info("Template {}z created\n\n".format(res))
#########Parametrize solvent parameters if need it##############
env.logger.info("Setting implicit solvent: {}".format(env.solvent))
if env.solvent == "OBC":
shutil.copy(env.obc_tmp, env.obc_file)
for template in glob.glob(os.path.join(env.template_folder, "*")):
obc.main(template, env.obc_file)
env.logger.info("Implicit solvent set\n\n".format(env.solvent))
#################Set Box###################
env.logger.info("Generating exploration box")
if not env.box_center:
env.box_center = cm.center_of_mass(env.ligand_ref)
env.logger.info("Box {} generated\n\n".format(env.box_center))
else:
env.logger.info("Box {} generated\n\n".format(env.box_center))
env.box = cs.BOX.format(env.box_radius, env.box_center) if env.box_radius else ""
else:
env.box=""
#####Build PCA#######
if env.pca_traj:
if isinstance(env.pca_traj, str):
pdbs = glob.glob(env.pca_traj)
elif isinstance(env.pca_traj, list):
pdbs = env.pca_traj
pdbs_full_path = [os.path.abspath(pdb) for pdb in pdbs]
output = os.path.basename(pdbs[0])[:-4] + "_ca_pca_modes.nmd"
pca_script = os.path.join(cs.DIR, "Utilities/Helpers/calculatePCA4PELE.py")
command = 'python {} --pdb "{}"'.format(pca_script, " ".join(pdbs_full_path))
with helpers.cd(env.pele_dir):
os.system(command)
env.pca = cs.PCA.format(output)
############Fill in Simulation Templates############
env.logger.info("Running Simulation")
if env.adaptive:
ext.external_adaptive_file(env)
if env.pele:
ext.external_pele_file(env)
adaptive = ad.SimulationBuilder(env.ad_ex_temp, env.pele_exit_temp, env)
if not env.debug:
adaptive.run()
env.logger.info("Simulation run succesfully (:\n\n")
return env
def run_adaptive(args: pv.EnviroBuilder) -> pv.EnviroBuilder:
"""
Main function to prepare and launch simulation
1) Crate working folders and inputs
2) Prepare ligand and receptor
3) Launch simulation
4) Analyse simulation
"""
# Build Folders and Logging and env variable that will containt
#all main attributes of the simulation
env = pele.EnviroBuilder()
env.software = "Adaptive"
env.build_adaptive_variables(args)
env.create_files_and_folders()
shutil.copy(args.yamlfile, env.pele_dir)
#######
if env.adaptive_restart and not env.only_analysis:
with helpers.cd(env.pele_dir):
adt.main(env.ad_ex_temp)
env.logger.info("Simulation run succesfully (:\n\n")
elif not env.only_analysis:
env.logger.info("System: {}; Platform Functionality: {}\n\n".format(env.residue, env.software))
if env.perturbation:
syst = sp.SystemBuilder.build_system(env.system, args.mae_lig, args.residue, env.pele_dir)
else:
syst = sp.SystemBuilder(env.system, None, None, env.pele_dir)
env.logger.info("Prepare complex {}".format(syst.system))
########Choose your own input####################
# User specifies more than one input
if env.input:
env.inputs_simulation = []
for input in env.input:
input_path = os.path.join(env.pele_dir, os.path.basename(input))
shutil.copy(input, input_path)
if env.no_ppp:
input_proc = input
else:
input_proc = os.path.basename(ppp.main(input_path, env.pele_dir, output_pdb=["" , ],
charge_terminals=args.charge_ter, no_gaps_ter=args.gaps_ter,
constrain_smiles=env.constrain_smiles, ligand_pdb=env.ligand_ref)[0])
env.inputs_simulation.append(input_proc)
hp.silentremove([input_path])
env.adap_ex_input = ", ".join(['"' + input + '"' for input in env.inputs_simulation]).strip('"')
# Global exploration mode: Create inputs around protein
elif args.full or args.randomize or args.ppi:
ligand_positions, box_radius, box_center = rd.randomize_starting_position(env.ligand_ref, env.receptor,
outputfolder=env.pele_dir, nposes=env.poses, test=env.test, user_center=env.center_of_interface)
if not args.gpcr_orth:
env.box_center = box_center
env.box_radius = box_radius
if env.no_ppp:
receptor = syst.system
else:
receptor = ppp.main(syst.system, env.pele_dir, output_pdb=["" , ],
charge_terminals=args.charge_ter, no_gaps_ter=args.gaps_ter,
constrain_smiles=env.constrain_smiles, ligand_pdb=env.ligand_ref)[0]
inputs = rd.join(receptor, ligand_positions, env.residue, output_folder=env.pele_dir)
env.adap_ex_input = ", ".join(['"' + os.path.basename(input) + '"' for input in inputs]).strip('"')
hp.silentremove(ligand_positions)
#Parsing input for errors and saving them as inputs
##########Prepare System################
if env.no_ppp:
missing_residues = []
gaps = {}
metals = {}
env.constraints = ct.retrieve_constraints(env.system, gaps, metals, back_constr=env.ca_constr)
if env.input:
# If we have more than one input
for input in env.input: shutil.copy(input, env.pele_dir)
else:
shutil.copy(env.system, env.pele_dir)
else:
env.system, missing_residues, gaps, metals, env.constraints = ppp.main(syst.system, env.pele_dir, output_pdb=["" , ], charge_terminals=args.charge_ter, no_gaps_ter=args.gaps_ter, mid_chain_nonstd_residue=env.nonstandard, skip=env.skip_prep, back_constr=env.ca_constr, constrain_smiles=env.constrain_smiles, ligand_pdb=env.ligand_ref)
################METAL CONSTRAINTS##################
if not args.no_metal_constraints:
metal_constraints, env.external_constraints = mc.main(args.external_constraints, os.path.join(env.pele_dir, env.adap_ex_input.split(",")[0].strip().strip('"')), syst.system, permissive=env.permissive_metal_constr, all_metals=args.constrain_all_metals, external=env.external_constraints)
metal_constraints_json = hp.retrieve_constraints_for_pele(metal_constraints, env.system)
env.external_constraints = hp.retrieve_constraints_for_pele(env.external_constraints, env.system)
metal_constraints_json = hp.retrieve_constraints_for_pele(metal_constraints, env.system)
env.external_constraints.extend(metal_constraints_json)
else:
env.external_constraints = hp.retrieve_constraints_for_pele(env.external_constraints, env.system)
# Keep Json ordered by having first title and then constraints
if env.external_constraints:
env.constraints = env.constraints[0:1] + env.external_constraints + env.constraints[1:]
if env.remove_constraints:
env.constraints = ""
env.logger.info(cs.SYSTEM.format(missing_residues, gaps, metals))
env.logger.info("Complex {} prepared\n\n".format(env.system))
####Ligand parameters and simulation box
if env.perturbation:
ligand_params = lg.LigandParametrization(env)
ligand_params.generate()
box = bx.BoxSetter(env.box_center, env.box_radius,
env.ligand_ref, env.logger)
env.box = box.generate_json()
else:
env.box=""
###########Parametrize missing residues
for res, __, _ in missing_residues:
if res != args.residue and res not in env.skip_ligand_prep:
env.logger.info("Creating template for residue {}".format(res))
with hp.cd(env.pele_dir):
mr.create_template(env, res)
env.logger.info("Template {}z created\n\n".format(res))
#########Solvent parameters
solvent = sv.ImplicitSolvent(env.solvent, env.obc_tmp,
env.template_folder, env.obc_file, env.logger)
solvent.generate()
#####Build PCA#######
if env.pca_traj:
pca_obj = pca.PCA(env.pca_traj, env.pele_dir)
env.pca = pca_obj.generate()
####### Add waters, if needed
if args.n_waters:
# Add n water molecules to minimisation inputs
input_waters = [input.strip().strip('"') for input in env.adap_ex_input.split(",")]
input_waters = [os.path.join(env.pele_dir, inp) for inp in input_waters]
wt.water_checker(args)
wt.add_water(input_waters, args.residue, args.n_waters, test=env.test)
wt.set_water_control_file(env)
elif args.waters:
wt.set_water_control_file(env)
############Fill in Simulation Templates############
adaptive = ad.SimulationBuilder(env.ad_ex_temp,
env.pele_exit_temp, env.topology)
adaptive.generate_inputs(env)
if env.analyse and not env.debug:
report = pt.analyse_simulation(env.report_name, env.traj_name[:-4]+"_",
os.path.join(env.pele_dir, env.output), env.residue, cpus=env.cpus,
output_folder=env.pele_dir, clustering=env.perturbation, mae=env.mae,
nclusts=env.analysis_nclust, overwrite=env.overwrite, topology=env.topology,
be_column=env.be_column, limit_column=env.limit_column, te_column=env.te_column)
print("Pdf summary report successfully writen to: {}".format(report))
return env
