IMP  2.2.0
The Integrative Modeling Platform
rotamer_pdb2.py
1 ## \example rotamer/rotamer_pdb2.py
2 # rotamer_pdb.py is a script demonstrating the usage of
3 # RotamerCalculator and RotamerLibrary. It reads a PDB file and a
4 # rotamer library file, and tries to rotate the atoms based on the
5 # most probable chi angles from the rotamer library. Then it saves
6 # the rotated atoms to a specified output PDB file.
7 #
8 # Usage:
9 #
10 # `python rotamer_pdb.py -i <input>.pdb -l <rotamer_library>.lib -o <output>.pdb`
11 #
12 # Example (the result will be saved into transformed_1z5s_A.pdb):
13 #
14 # `../../../tools/imppy.sh python rotamer_pdb.py -i ../../atom/test/input/1z5s_A.pdb \
15 # -l /path/to/ALL.bbdep.rotamers.lib -o transformed_1z5s_A.pdb`
16 #
17 
18 #!/usr/bin/env python
19 
20 import IMP
21 import IMP.base
22 import IMP.core
23 import IMP.atom
24 import IMP.algebra
25 import IMP.rotamer
26 
27 
28 def transform(input_pdb, input_lib, output_pdb):
29  # read rotamer library
31  rl.read_library_file(input_lib)
33 
34  # read the original PDB
35  m = IMP.kernel.Model()
36  orig_h = IMP.atom.read_pdb(input_pdb, m)
37  mh = IMP.atom.get_by_type(orig_h, IMP.atom.RESIDUE_TYPE)
38 
39  # transform...
41  rc.transform(orig_h, hps, 0.9, 1e-6, 6)
42 
43  # save the rotated atoms to output PDB
44  IMP.atom.write_pdb(orig_h, output_pdb)
45 
46 
47 def quick_test():
50  rc.set_was_used(True)
51 
52 
53 if __name__ == '__main__':
54 
55  import sys
56 
57  P = IMP.OptionParser()
58  P.add_option('--input_pdb', '-i', action='store', type='string',
59  help='input PDB file (required)')
60  P.add_option('--input_lib', '-l', action='store', type='string',
61  help='input rotamer library file (required)')
62  P.add_option('--output_pdb', '-o', action='store', type='string',
63  help='output PDB file (required)')
64  P.add_option('--verbose', '-v', action='store_true',
65  help='show more messages')
66  opts, args = P.parse_args()
67  if IMP.base.get_bool_flag('run_quick_test') or \
68  not (opts.input_pdb or opts.input_lib or opts.output_pdb):
69  quick_test()
70  sys.exit(0)
71  if not opts.input_pdb:
72  print '--input_pdb is required'
73  sys.exit(1)
74  if not opts.output_pdb:
75  print '--output_pdb is required'
76  sys.exit(1)
77  if not opts.input_lib:
78  print '--input_lib is required'
79  sys.exit(1)
80  if opts.verbose:
81  IMP.base.set_log_level(IMP.base.VERBOSE)
82  else:
83  IMP.base.set_log_level(IMP.base.SILENT)
84  transform(opts.input_pdb, opts.input_lib, opts.output_pdb)
void write_pdb(const Selection &mhd, base::TextOutput out, unsigned int model=1)
IMP-specific subclass of optparse.OptionParser.
void set_log_level(LogLevel l)
Set the current global log level.
A class storing a whole rotamer library read from a file.
bool get_bool_flag(std::string name)
See IMP.base for more information.
Definition: base/Array.h:20
void transform(Hierarchy h, const algebra::Transformation3D &tr)
A class performing the rotations of atoms in the residues.
Hierarchies get_by_type(Hierarchy mhd, GetByType t)
See IMP.rotamer for more information.
See IMP.core for more information.
See IMP.algebra for more information.
See IMP.atom for more information.
void read_pdb(base::TextInput input, int model, Hierarchy h)
Class for storing model, its restraints, constraints, and particles.
Definition: kernel/Model.h:72