IMP logo
IMP Reference Guide  2.7.0
The Integrative Modeling Platform
anchors.py
1 #!/usr/bin/env python
2 
3 from __future__ import print_function
4 import IMP.multifit
5 import IMP.em
6 from IMP import OptionParser
7 
8 __doc__ = "Generate anchors for a density map."
9 
10 def parse_args():
11  usage = """%prog [options] <assembly input> <output anchors prefix>
12 
13 Generate anchors for a density map."""
14  parser = OptionParser(usage)
15  parser.add_option("-s", "--size", type="int", dest="size", default=-1,
16  help="number of residues per bead")
17  options, args = parser.parse_args()
18 
19  if len(args) != 2:
20  parser.error("incorrect number of arguments")
21  return options, args
22 
23 
24 def main():
25  options, args = parse_args()
26  asmb_fn = args[0]
27  output = args[1]
28  asmb = IMP.multifit.read_settings(asmb_fn)
29  asmb.set_was_used(True)
30  dmap = IMP.em.read_map(asmb.get_assembly_header().get_dens_fn(),
32  if options.size == -1:
33  number_of_means = asmb.get_number_of_component_headers()
34  else:
35  total_num_residues = 0
36  mdl = IMP.Model()
37  for i in range(asmb.get_number_of_component_headers()):
38  total_num_residues += len(
39  IMP.atom.get_by_type(
41  asmb.get_component_header(
42  i).get_filename(
43  ),
44  mdl),
45  IMP.atom.RESIDUE_TYPE))
46  number_of_means = total_num_residues / options.size
47  print("Calculating a segmentation into", number_of_means, "regions")
48  density_threshold = asmb.get_assembly_header().get_threshold()
49 
50  IMP.multifit.get_anchors_for_density(dmap, number_of_means,
51  density_threshold,
52  output + ".pdb", output + ".cmm",
53  "", output + ".txt")
54 
55 if __name__ == "__main__":
56  main()
SettingsData * read_settings(const char *filename)
void read_pdb(TextInput input, int model, Hierarchy h)
Class for storing model, its restraints, constraints, and particles.
Definition: Model.h:72
void get_anchors_for_density(em::DensityMap *dmap, int number_of_means, float density_threshold, std::string pdb_filename, std::string cmm_filename, std::string seg_filename, std::string txt_filename)
Generate anchors in several formats for a given density map.
Fitting atomic structures into a cryo-electron microscopy density map.
Basic utilities for handling cryo-electron microscopy 3D density maps.