9 #ifndef IMPEM_DENSITY_MAP_H
10 #define IMPEM_DENSITY_MAP_H
12 #include <IMP/em/em_config.h>
20 #include <boost/scoped_array.hpp>
44 IMPEMEXPORT DensityMap *
read_map(std::string filename, MapReaderWriter *reader);
54 IMPEMEXPORT DensityMap *
read_map(std::string filename);
59 IMPEMEXPORT
void write_map(DensityMap *m, std::string filename,
60 MapReaderWriter *writer);
71 IMPEMEXPORT
void write_map(DensityMap *m, std::string filename);
101 DensityMap(std::string name =
"DensityMap%1%");
108 void reset_data(
float value = 0.0);
118 void std_normalize();
120 inline bool is_normalized()
const {
return normalized_; }
128 float get_location_in_dim_by_voxel(
long index,
int dim)
const;
137 return z * header_.get_nx() * header_.get_ny() + y * header_.get_nx() + x;
147 long get_voxel_by_location(
float x,
float y,
float z)
const;
155 return get_voxel_by_location(v[0], v[1], v[2]);
169 "invalid map index");
172 "locations should be calculated prior to calling this function");
176 bool is_xyz_ind_part_of_volume(
int ix,
int iy,
int iz)
const;
184 bool is_part_of_volume(
float x,
float y,
float z)
const;
190 return is_part_of_volume(v[0], v[1], v[2]);
202 emreal get_value(
float x,
float y,
float z)
const;
204 return get_value(point[0], point[1], point[2]);
210 emreal get_value(
long index)
const;
216 void set_value(
long index, emreal value);
219 void set_value(
float x,
float y,
float z, emreal value);
226 void set_origin(
float x,
float y,
float z);
228 set_origin(v[0], v[1], v[2]);
233 header_.get_origin(2));
247 #if !defined(IMP_DOXYGEN) && !defined(SWIG)
252 float *get_x_loc()
const {
254 "x location requested before being calculated");
261 float *get_y_loc()
const {
263 "y location requested before being calculated");
270 float *get_z_loc()
const {
272 "z location requested before being calculated");
276 emreal *get_data()
const {
return data_.get(); }
283 bool same_origin(
const DensityMap *other)
const;
289 bool same_dimensions(
const DensityMap *other)
const;
295 bool same_voxel_size(
const DensityMap *other)
const;
301 emreal get_max_value()
const;
303 emreal get_min_value()
const;
311 void add(
const DensityMap *other);
318 void pick_max(
const DensityMap *other);
321 long get_number_of_voxels()
const;
329 void set_void_map(
int nx,
int ny,
int nz);
341 void pad(
int nx,
int ny,
int nz,
float val = 0.0);
360 DensityMap *pad_margin(
int mrg_x,
int mrg_y,
int mrg_z,
float val = 0.0);
368 DensityMap *get_cropped(
float threshold);
379 float get_maximum_value_in_xy_plane(
int z_ind);
381 float get_maximum_value_in_xz_plane(
int y_ind);
383 float get_maximum_value_in_yz_plane(
int x_ind);
392 std::string get_locations_string(
float t);
395 void update_voxel_size(
float new_apix);
402 void calc_all_voxel2loc();
406 #if !defined(IMP_DOXYGEN) && !defined(SWIG)
414 void convolute_kernel(
DensityMap *other,
double *kernel,
int dim_len);
421 void convolute_kernel(
double *kernel,
int dim_len) {
423 cmap->set_was_used(
true);
424 convolute_kernel(cmap, kernel, dim_len);
428 int lower_voxel_shift(emreal loc, emreal kdist, emreal orig,
int ndim)
const;
429 int upper_voxel_shift(emreal loc, emreal kdist, emreal orig,
int ndim)
const;
430 inline bool get_rms_calculated()
const {
return rms_calculated_; }
431 int get_dim_index_by_location(
float loc_val,
int ind)
const;
435 void update_header();
436 void reset_all_voxel2loc();
438 void allocated_data();
439 void float2real(
float *f_data, boost::scoped_array<emreal> &r_data);
440 void real2float(emreal *r_data, boost::scoped_array<float> &f_data);
442 DensityHeader header_;
443 boost::scoped_array<emreal> data_;
444 bool data_allocated_;
449 boost::scoped_array<float>
x_loc_, y_loc_, z_loc_;
454 bool rms_calculated_;
469 #if !defined(IMP_DOXYGEN) && !defined(SWIG)
484 inline void calc_local_bounding_box(
const em::DensityMap *d_map,
double x,
485 double y,
double z,
float kdist,
int &iminx,
486 int &iminy,
int &iminz,
int &imaxx,
487 int &imaxy,
int &imaxz) {
489 iminx = d_map->lower_voxel_shift(x, kdist, h->get_xorigin(), h->get_nx());
490 iminy = d_map->lower_voxel_shift(y, kdist, h->get_yorigin(), h->get_ny());
491 iminz = d_map->lower_voxel_shift(z, kdist, h->get_zorigin(), h->get_nz());
492 imaxx = d_map->upper_voxel_shift(x, kdist, h->get_xorigin(), h->get_nx());
493 imaxy = d_map->upper_voxel_shift(y, kdist, h->get_yorigin(), h->get_ny());
494 imaxz = d_map->upper_voxel_shift(z, kdist, h->get_zorigin(), h->get_nz());
539 IMPEMEXPORT
void get_transformed_into2(
const DensityMap *source,
543 inline bool get_interiors_intersect(
const DensityMap *d1,
564 IMPEMEXPORT DensityMap *
get_segment(DensityMap *map_to_segment,
int nx_start,
565 int nx_end,
int ny_start,
int ny_end,
566 int nz_start,
int nz_end);
569 IMPEMEXPORT DensityMap *
get_segment(DensityMap *map_to_segment,
574 float mas_threshold);
582 IMPEMEXPORT DensityMap *
binarize(DensityMap *orig_map,
float threshold,
583 bool reverse =
false);
596 IMPEMEXPORT DensityMap *
multiply(
const DensityMap *m1,
const DensityMap *m2);
601 IMPEMEXPORT
double convolute(
const DensityMap *m1,
const DensityMap *m2);
604 IMPEMEXPORT
double get_sum(
const DensityMap *m1);
625 template <
class S,
class V,
class E>
631 std::abs(arg.get_unit_cell()[0] - arg.get_unit_cell()[1]) < .01,
632 "The passed grid does not seem to have cubic voxels");
634 algebra::get_bounding_box(arg), arg.get_unit_cell()[0]);
636 static_cast<unsigned int>(ret->get_header()->get_nx()),
637 "X voxels don't match");
639 static_cast<unsigned int>(ret->get_header()->get_ny()),
640 "Y voxels don't match");
642 static_cast<unsigned int>(ret->get_header()->get_nz()),
643 "Z voxels don't match");
644 IMP_FOREACH(
typename Grid::Index i, arg.get_all_indexes()) {
645 long vi = ret->xyz_ind2voxel(i[0], i[1], i[2]);
646 ret->set_value(vi, arg[vi]);
O * release()
Relinquish control of the raw pointer stored in the Pointer.
IMP::Vector< IMP::Pointer< DensityMap > > DensityMaps
The base class to handle reading and writing of density maps.
const DensityHeader * get_header() const
Returns a read-only pointer to the header of the map.
DensityMap * interpolate_map(DensityMap *in_map, double new_spacing)
Return a new map with an updated spacing.
BoundingBoxD< 3 > BoundingBox3D
Typedef for Python.
DensityMap * get_segment(DensityMap *map_to_segment, algebra::Vector3Ds vecs, float dist)
Get a segment of the map covered by the input points.
DensityMap * binarize(DensityMap *orig_map, float threshold, bool reverse=false)
Return a map with 0 for all voxels below the threshold and 1 for those above.
#define IMP_FUNCTION_LOG
Beginning logging for a non-member function.
double get_sum(const DensityMap *m1)
Return the sum of all voxels.
DensityMap * read_map(std::string filename, MapReaderWriter *reader)
Read a density map from a file (using the given reader) and return it.
#define IMP_OBJECT_METHODS(Name)
Define the basic things needed by any Object.
void add_to_map(DensityMap *dm, const Particles &pis)
Rasterize the particles into an existing density map.
A voxel grid in d-dimensional space.
DensityMap * get_threshold_map(const DensityMap *orig_map, float threshold)
Return a map with 0 for all voxels below the threshold.
long get_voxel_by_location(const algebra::Vector3D &v) const
Calculate the voxel of a given location.
DensityMap * get_segment_by_masking(DensityMap *map_to_segment, DensityMap *mask, float mas_threshold)
Get a segment of the map covered by another map.
bool loc_calculated_
true if the locations have already been computed
A smart pointer to a reference counted object.
algebra::GridD< 3, algebra::DenseGridStorageD< 3, float >, float > get_grid(DensityMap *in_map)
Return a dense grid containing the voxels of the passed density map.
Class for handling density maps.
double convolute(const DensityMap *m1, const DensityMap *m2)
Return a convolution between density maps m1 and m2.
DensityMap * get_resampled(DensityMap *input, double scaling)
Get a resampled version of the map.
void get_transformed_into(const DensityMap *source, const algebra::Transformation3D &tr, DensityMap *into, bool calc_rms=true)
Rotate a density map into another map.
void write_map(DensityMap *m, std::string filename, MapReaderWriter *writer)
Write a density map to a file using the given writer.
Common base class for heavy weight IMP objects.
DensityMap * multiply(const DensityMap *m1, const DensityMap *m2)
Return a density map for which voxel i contains the result of m1[i]*m2[i].
An abstract class for reading a map.
DensityMap * create_density_map(const algebra::GridD< 3, algebra::DenseGridStorageD< 3, float >, float > &grid)
Return a density map with the values taken from the grid.
BoundingBoxD< 3 > get_bounding_box(const Cone3D &g)
boost::scoped_array< float > x_loc_
A bounding box in D dimensions.
algebra::Vector3D get_location_by_voxel(long index) const
Calculate the location of a given voxel.
double get_density(const DensityMap *m, const algebra::Vector3D &v)
Get density value at point v, interpolating linearly from the sample values.
DensityMap * get_max_map(DensityMaps maps)
Return a map where each voxel is the maximum value from the input maps.
DensityMap * create_density_map(const DensityMap *other)
Create a copy of another map.
DensityHeader * get_header_writable()
Returns a pointer to the header of the map in a writable version.
#define IMP_NO_SWIG(x)
Hide the line when SWIG is compiled or parses it.
DensityMap * get_binarized_interior(DensityMap *dmap)
Return a binarized map with 1 for voxels that are internal in the input map.
bool is_part_of_volume(const algebra::Vector3D &v) const
Checks whether a given point is in the grid.
Float approximate_molecular_mass(DensityMap *m, Float threshold)
Estimate the molecular mass from a map.
Vector3D get_centroid(const Vector3Ds &ps)
Returns the centroid of a set of vectors.
#define IMP_OBJECTS(Name, PluralName)
Define the types for storing sets of objects.
A nullptr-initialized pointer to an IMP Object.
A shared base class to help in debugging and things.
Float get_spacing() const
Get the voxel size of the map.
double Float
Basic floating-point value (could be float, double...)
long xyz_ind2voxel(int x, int y, int z) const
Calculate the voxel of a given xyz index.
#define IMP_USAGE_CHECK(expr, message)
A runtime test for incorrect usage of a class or method.
All grids that are in the Python API should be defined here.
DensityMap * get_transformed(const DensityMap *input, const algebra::Transformation3D &tr, double threshold)
Return a new density map containing a rotated version of the old one.