IMP logo
IMP Reference Guide  2.22.0
The Integrative Modeling Platform
fit_restraint.py
1 ## \example em/fit_restraint.py
2 # A simple example showing how to set up a fit restraint. The number of
3 # spheres and resolution are randomly chosen and so should not be
4 # considered significant.
5 
6 import IMP.em
7 import IMP.core
8 import IMP.atom
9 import sys
10 
11 IMP.setup_from_argv(sys.argv, "fit restraint")
12 
13 IMP.set_log_level(IMP.SILENT)
14 m = IMP.Model()
15 # 1. setup the input protein
16 # 1.1 select a selector.
18 # 1.2 read the protein
19 mh = IMP.atom.read_pdb(IMP.em.get_example_path("input.pdb"), m, sel)
20 ps = IMP.core.get_leaves(mh)
22 # 2. read the density map
23 resolution = 8.
24 voxel_size = 1.5
25 dmap = IMP.em.read_map(
27 dmap.get_header_writable().set_resolution(resolution)
28 # 3. calculate the cross correlation between the density and the map
29 print("The cross-correlation score is:",
30  1. - IMP.em.compute_fitting_score(ps, dmap))
31 # 4. add a fitting restraint
32 r = IMP.em.FitRestraint(ps, dmap)
33 print("The fit of the particles in the density is:", r.evaluate(False))
Strings setup_from_argv(const Strings &argv, std::string description, std::string positional_description, int num_positional)
void add_radii(Hierarchy d, const ForceFieldParameters *ffp=get_all_atom_CHARMM_parameters(), FloatKey radius_key=FloatKey("radius"))
Add vdW radius from given force field.
GenericHierarchies get_leaves(Hierarchy mhd)
Get all the leaves of the bit of hierarchy.
void read_pdb(TextInput input, int model, Hierarchy h)
Class for storing model, its restraints, constraints, and particles.
Definition: Model.h:86
Float compute_fitting_score(const ParticlesTemp &ps, DensityMap *em_map, FloatKey wei_key=atom::Mass::get_mass_key())
Compute fitting scores for a given set of rigid transformations.
Basic utilities for handling cryo-electron microscopy 3D density maps.
std::string get_example_path(std::string file_name)
Return the full path to one of this module's example files.
void set_log_level(LogLevel l)
Set the current global log level.
Basic functionality that is expected to be used by a wide variety of IMP users.
Calculate score based on fit to EM map.
Definition: FitRestraint.h:39
Select all non-water non-alternative ATOM and HETATM records.
Definition: pdb.h:356
Functionality for loading, creating, manipulating and scoring atomic structures.