IMP  2.1.0
The Integrative Modeling Platform
score_protein_with_ligand.py
1 ## \example atom/score_protein_with_ligand.py
2 # Show how to score a number of ligand conformations loaded from a file
3 # against a protein loaded from a pdb.
4 
5 import IMP.atom
6 
8 IMP.base.set_check_level(IMP.base.NONE)
9 protein = IMP.atom.read_pdb(IMP.atom.get_example_path('1d3d-protein.pdb'), m)
10 protein_atoms = IMP.atom.get_by_type(protein, IMP.atom.ATOM_TYPE)
11 ligands = IMP.atom.read_mol2(IMP.atom.get_example_path('1d3d-ligands.mol2'), m)
12 # create the score which applies to a pair of atoms
14 ps.set_was_used(True)
15 # label each atom of the protein with the type needed for scoring
17 for l in ligands.get_children():
18  # compute the atom type for each ligand atom
20  score = 0
21  ligand_atoms = IMP.atom.get_by_type(l, IMP.atom.ATOM_TYPE)
22  for pa in protein_atoms:
23  for la in ligand_atoms:
24  # check if the atoms are close enough together
26  # score one pair of atoms
27  score += ps.evaluate((pa, la), None)
28  print "score for ", l.get_name(), "is", score
void set_check_level(CheckLevel tf)
Control runtime checks in the code.
std::string get_example_path(std::string file_name)
Return the path to installed example data for this module.
double get_distance(XYZR a, XYZR b)
Compute the sphere distance between a and b.
Definition: XYZR.h:88
Hierarchies get_by_type(Hierarchy mhd, GetByType t)
Hierarchy read_mol2(base::TextInput mol2_file, kernel::Model *model, Mol2Selector *mol2sel=nullptr)
Create a hierarchy from a Mol2 file.
A decorator for a particle with x,y,z coordinates.
Definition: XYZ.h:32
void add_protein_ligand_score_data(Hierarchy h)
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.