em::FittingSolutions convert_multifit_to_em_format(
const FittingSolutionRecords &multifit_fits) {
em::FittingSolutions output;
for(FittingSolutionRecords::const_iterator it = multifit_fits.begin();
it != multifit_fits.end();it++) {
for(unsigned int i=0;i<multifit_fits.size();i++)
output.add_solution(it->get_fit_transformation(),
it->get_fitting_score());
}
return output;
}
IMPMULTIFIT_BEGIN_NAMESPACE
FittingSolutionRecords get_close_to_point(
const FittingSolutionRecords &fit_sols, atom::Hierarchy mh,
IMP::Particle *ap, Float dist) {
FittingSolutionRecords pruned_fit_sols;
Float dist2 = dist * dist;
IMP::algebra::Vector3D ap_centroid = IMP::core::XYZ(ap).get_coordinates();
IMP::algebra::Vector3D mh_centroid =
IMP::core::get_centroid(IMP::core::XYZs(IMP::core::get_leaves(mh)));
for (FittingSolutionRecords::const_iterator it = fit_sols.begin();
it != fit_sols.end(); it++) {
if (IMP::algebra::get_squared_distance(
it->get_fit_transformation().get_transformed(mh_centroid),
ap_centroid) < dist2) {
pruned_fit_sols.push_back(*it);
}
}
return pruned_fit_sols;
}