3 from __future__
import print_function
8 from IMP
import ArgumentParser
12 __doc__ =
"Fit subunits into a density map with FFT."
14 multiproc_exception =
None
16 import multiprocessing
20 if sys.platform ==
'win32' and 'WINELOADERNOEXEC' in os.environ:
21 multiproc_exception =
"Wine does not currently support multiprocessing"
22 except ImportError
as detail:
23 multiproc_exception = str(detail)
29 if 'forkserver' in multiprocessing.get_all_start_methods():
30 return multiprocessing.get_context(
'forkserver')
32 return multiprocessing.get_context()
51 self.spacing = spacing
52 self.resolution = resolution
53 self.threshold = density_threshold
54 self.originx = origin[0]
55 self.originy = origin[1]
56 self.originz = origin[2]
58 self.fits_fn = fits_fn
60 self.num_fits = num_fits
61 self.angles_per_voxel = angles_per_voxel
62 self.ref_pdb = ref_pdb
65 print(
"resolution is:", self.resolution)
66 dmap = IMP.em.read_map(self.em_map)
67 dmap.get_header().set_resolution(self.resolution)
68 dmap.update_voxel_size(self.spacing)
72 dmap.set_was_used(
True)
73 dmap.get_header().
show()
80 fits = ff.do_global_fitting(dmap, self.threshold, mol2fit,
81 self.angle / 180.0 * math.pi,
82 self.num_fits, self.spacing, 0.5,
83 True, self.angles_per_voxel)
84 fits.set_was_used(
True)
85 final_fits = fits.best_fits_
86 if self.ref_pdb !=
'':
90 for i, fit
in enumerate(final_fits):
92 if self.ref_pdb !=
'':
93 trans = fit.get_fit_transformation()
99 fit.set_rmsd_to_reference(rmsd)
101 if self.ref_pdb !=
'':
102 print(
'from all fits, lowest rmsd to ref:', cur_low)
111 desc =
"""Fit subunits into a density map with FFT."""
112 p = ArgumentParser(description=desc)
113 p.add_argument(
"-c",
"--cpu", dest=
"cpus", type=int, default=1,
114 help=
"number of cpus to use (default 1)")
115 p.add_argument(
"-a",
"--angle", dest=
"angle", type=float, default=30,
116 help=
"angle delta (degrees) for FFT rotational "
117 "search (default 30)")
119 p.add_argument(
"-n",
"--num", dest=
"num", type=int,
120 default=100, help=
"Number of fits to report (default 100)")
122 p.add_argument(
"-v",
"--angle_voxel", dest=
"angle_voxel", type=int,
124 help=
"Number of angles to keep per voxel (default 10)")
126 p.add_argument(
"assembly_file", help=
"assembly file name")
133 return p.parse_args()
136 def run(asmb_fn, options):
137 if multiproc_exception
is None and options.cpus > 1:
140 asmb_input.set_was_used(
True)
141 em_map = asmb_input.get_assembly_header().get_dens_fn()
143 spacing = asmb_input.get_assembly_header().get_spacing()
144 origin = asmb_input.get_assembly_header().get_origin()
145 for i
in range(asmb_input.get_number_of_component_headers()):
146 fits_fn = asmb_input.get_component_header(i).get_transformations_fn()
147 pdb_fn = asmb_input.get_component_header(i).get_filename()
153 asmb_input.get_assembly_header().get_threshold(),
159 if multiproc_exception
is None and options.cpus > 1:
165 The Python 'multiprocessing' module (available in Python 2.6 and later) is
166 needed to run on multiple CPUs, and could not be found
167 (Python error: '%s').
168 Running on a single processor.""" % multiproc_exception, file=sys.stderr)
170 if multiproc_exception
is None and options.cpus > 1:
172 nproc = min(options.cpus, asmb_input.get_number_of_component_headers())
174 p = ctx.Pool(processes=nproc)
175 _ = list(p.imap_unordered(do_work, work_units))
181 run(args.assembly_file, args)
184 if __name__ ==
"__main__":
Fit a molecule inside its density by local or global FFT.
SettingsData * read_settings(const char *filename)
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.
double get_rmsd(const Selection &s0, const Selection &s1)
void transform(Hierarchy h, const algebra::Transformation3D &tr)
Transform a hierarchy. This is aware of rigid bodies.
Fitting atomic structures into a cryo-electron microscopy density map.
Basic utilities for handling cryo-electron microscopy 3D density maps.
void write_fitting_solutions(const char *fitting_fn, const FittingSolutionRecords &fit_sols, int num_sols=-1)
Write fitting solutions to a file.
std::ostream & show(Hierarchy h, std::ostream &out=std::cout)
Print the hierarchy using a given decorator to display each node.
IMP::core::RigidBody create_rigid_body(Hierarchy h)
double get_resolution(Model *m, ParticleIndex pi)
Estimate the resolution of the hierarchy as used by Representation.
Functionality for loading, creating, manipulating and scoring atomic structures.