Mercurial > repos > bgruening > openbabel_remove_protonation_state
annotate distance_finder.py @ 12:e946a539ddf7 draft
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 944ea4bb8a9cd4244152a4a4fecd0485fabc2ad0"
| author | bgruening | 
|---|---|
| date | Tue, 28 Jul 2020 12:03:54 +0000 | 
| parents | 35a7d2dbe673 | 
| children | ad98596cf471 | 
| rev | line source | 
|---|---|
| 10 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 1 # Reports distances of ligands to reference points. An example input for the points is: | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 2 # | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 3 # 5.655 1.497 18.223 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 4 # 1.494 -8.367 18.574 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 5 # 13.034 6.306 25.232 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 6 # | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 7 # Data can be space or tab separated but must contain 3 and only 3 numbers for the x, y and z coordinates | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 8 # | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 9 # That would encode 3 points. | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 10 # Each record in the SDF input is read and the closest heavy atom to each of the reference points is recorded as | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 11 # a property named distance1 where the numeric part is the index (starting from 1) of the points (in that example | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 12 # there would be properties for distance1, distance2 and distance3. | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 13 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 14 import argparse, os, sys, math | 
| 12 
e946a539ddf7
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 944ea4bb8a9cd4244152a4a4fecd0485fabc2ad0"
 bgruening parents: 
10diff
changeset | 15 | 
| 10 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 16 from openbabel import pybel | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 17 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 18 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 19 def log(*args, **kwargs): | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 20 """Log output to STDERR | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 21 """ | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 22 print(*args, file=sys.stderr, ** kwargs) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 23 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 24 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 25 def execute(ligands_sdf, points_file, outfile): | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 26 """ | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 27 :param ligands_sdf: A SDF with the 3D molecules to test | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 28 :param points_file: A file with the points to consider. | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 29 :param outfile: The name of the file for the SDF output | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 30 :return: | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 31 """ | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 32 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 33 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 34 points = [] | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 35 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 36 # read the points | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 37 with open(points_file, 'r') as f: | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 38 for line in f.readlines(): | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 39 line.strip() | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 40 if line: | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 41 p = line.split() | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 42 if len(p) == 3: | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 43 points.append((float(p[0]), float(p[1]), float(p[2]))) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 44 log("Read points",p) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 45 continue | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 46 log("Failed to read line:", line) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 47 log('Found', len(points), 'atom points') | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 48 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 49 sdf_writer = pybel.Outputfile("sdf", outfile, overwrite=True) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 50 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 51 count = 0 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 52 for mol in pybel.readfile("sdf", ligands_sdf): | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 53 count += 1 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 54 if count % 50000 == 0: | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 55 log('Processed', count) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 56 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 57 try: | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 58 # print("Processing mol", mol.title) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 59 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 60 clone = pybel.Molecule(mol) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 61 clone.removeh() | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 62 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 63 coords = [] | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 64 for atom in clone.atoms: | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 65 coords.append(atom.coords) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 66 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 67 p = 0 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 68 for point in points: | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 69 p += 1 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 70 distances = [] | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 71 for i in coords: | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 72 # calculates distance based on cartesian coordinates | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 73 distance = math.sqrt((point[0] - i[0])**2 + (point[1] - i[1])**2 + (point[2] - i[2])**2) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 74 distances.append(distance) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 75 # log("distance:", distance) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 76 min_distance = min(distances) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 77 # log('Min:', min_distance) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 78 # log(count, p, min_distance) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 79 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 80 mol.data['distance' + str(p)] = min_distance | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 81 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 82 sdf_writer.write(mol) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 83 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 84 except Exception as e: | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 85 log('Failed to handle molecule: '+ str(e)) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 86 continue | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 87 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 88 sdf_writer.close() | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 89 log('Wrote', count, 'molecules') | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 90 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 91 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 92 def main(): | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 93 global work_dir | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 94 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 95 parser = argparse.ArgumentParser(description='XChem distances - measure distances to particular points') | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 96 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 97 parser.add_argument('-i', '--input', help="SDF containing the 3D molecules to score)") | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 98 parser.add_argument('-p', '--points', help="PDB format file with atoms") | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 99 parser.add_argument('-o', '--outfile', default='output.sdf', help="File name for results") | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 100 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 101 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 102 args = parser.parse_args() | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 103 log("XChem distances args: ", args) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 104 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 105 execute(args.input, args.points, args.outfile) | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 106 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 107 | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 108 if __name__ == "__main__": | 
| 
35a7d2dbe673
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 6c84abdd07f292048bf2194073e2e938e94158c4"
 bgruening parents: diff
changeset | 109 main() | 
