#include <processor.h>
Inheritance diagram for EMAN::ApplyPolynomialProfileToHelix:
Public Member Functions | |
void | process_inplace (EMData *in) |
To process an image in-place. | |
string | get_name () const |
Get the processor's name. | |
string | get_desc () const |
Get the descrition of this specific processor. | |
virtual TypeDict | get_param_types () const |
Get processor parameter information in a dictionary. | |
Static Public Member Functions | |
static Processor * | NEW () |
Static Public Attributes | |
static const string | NAME = "math.poly_radial_profile" |
Definition at line 6927 of file processor.h.
string EMAN::ApplyPolynomialProfileToHelix::get_desc | ( | ) | const [inline, virtual] |
Get the descrition of this specific processor.
This function must be overwritten by a subclass.
Implements EMAN::Processor.
Definition at line 6942 of file processor.h.
06943 { 06944 return "Finds the CM of each z-axis slice and applies a polynomial radial profile about it."; 06945 }
string EMAN::ApplyPolynomialProfileToHelix::get_name | ( | ) | const [inline, virtual] |
Get the processor's name.
Each processor is identified by a unique name.
Implements EMAN::Processor.
Definition at line 6932 of file processor.h.
References NAME.
06933 { 06934 return NAME; 06935 }
virtual TypeDict EMAN::ApplyPolynomialProfileToHelix::get_param_types | ( | ) | const [inline, virtual] |
Get processor parameter information in a dictionary.
Each parameter has one record in the dictionary. Each record contains its name, data-type, and description.
Reimplemented from EMAN::Processor.
Definition at line 6946 of file processor.h.
References EMAN::EMObject::FLOAT, EMAN::EMObject::INT, and EMAN::TypeDict::put().
06947 { 06948 TypeDict d; 06949 d.put("length", EMObject::FLOAT, "Helix length in angstroms."); 06950 d.put("z0", EMObject::INT, "z coordinate in pixels for the midpoint of the cylinder's axis, defaults to center of map"); 06951 return d; 06952 }
static Processor* EMAN::ApplyPolynomialProfileToHelix::NEW | ( | ) | [inline, static] |
Definition at line 6937 of file processor.h.
06938 { 06939 return new ApplyPolynomialProfileToHelix(); 06940 }
void ApplyPolynomialProfileToHelix::process_inplace | ( | EMData * | in | ) | [virtual] |
To process an image in-place.
For those processors which can only be processed out-of-place, override this function to just print out some error message to remind user call the out-of-place version.
image | The image to be processed. |
Implements EMAN::Processor.
Definition at line 9746 of file processor.cpp.
References EMAN::EMData::get_attr(), EMAN::EMData::get_data(), EMAN::EMData::get_xsize(), EMAN::EMData::get_ysize(), EMAN::EMData::get_zsize(), in, EMAN::Processor::params, EMAN::ModelHelixProcessor::radprofile(), EMAN::Util::round(), and EMAN::Dict::set_default().
09747 { 09748 EMData * cyl = in; 09749 int nx = cyl->get_xsize(); 09750 int ny = cyl->get_ysize(); 09751 int nz = cyl->get_zsize(); 09752 float apix_x = cyl->get_attr("apix_x"); //TODO: Ask Matt if I correctly handled cases where apix_x != apix_y or apix_x != apix_z are not equal 09753 float apix_y = cyl->get_attr("apix_y"); 09754 float apix_z = cyl->get_attr("apix_z"); 09755 float lengthAngstroms = params["length"]; 09756 int z0 = params.set_default("z0", -1); //in voxels 09757 09758 if (z0 < 0 || z0 >= nz) 09759 z0 = nz / 2; 09760 09761 int z_start = Util::round( z0 - 0.5*lengthAngstroms/apix_z ); 09762 int z_stop = Util::round( z0 + 0.5*lengthAngstroms/apix_z ); 09763 09764 float * dat = cyl->get_data(); 09765 double rho_x_sum, rho_y_sum, rho_sum, x_cm, y_cm, radius; 09766 09767 for (int k = 0; k < nz; ++k) //taking slices along z axis 09768 { 09769 rho_x_sum = rho_y_sum = rho_sum = 0; //Set to zero for a new slice 09770 09771 if (k >= z_start && k <= z_stop) 09772 //Apply the radial profile only between z_start and z_stop on the z axis 09773 { 09774 //Calculating CM for the slice... 09775 for (int j = 0; j < ny; ++j) 09776 { 09777 for (int i = 0; i < nx; ++i, ++dat) 09778 { 09779 rho_x_sum += (*dat)*i; 09780 rho_y_sum += (*dat)*j; 09781 rho_sum += *dat; 09782 } 09783 } 09784 if (rho_sum != 0) 09785 { 09786 dat -= nx*ny;//going back to the beginning of the dat array 09787 x_cm = rho_x_sum/rho_sum; 09788 y_cm = rho_y_sum/rho_sum; 09789 09790 //Applying radial profile... 09791 for (int j=0; j<ny;++j) 09792 { 09793 for (int i=0;i<nx;++i,++dat) 09794 { 09795 radius = hypot( (i-x_cm)*apix_x, (j-y_cm)*apix_y ); 09796 *dat = radprofile((float)radius, 2);//Type 2 is the polynomial radial profile. 09797 } 09798 } 09799 } 09800 } 09801 else 09802 //Clear the map, setting the density to zero everywhere else. 09803 { 09804 for (int j=0; j<ny; j++) 09805 for(int i=0; i<nx; i++) 09806 { 09807 *dat = 0; 09808 dat++; 09809 } 09810 } 09811 09812 } 09813 }
const string ApplyPolynomialProfileToHelix::NAME = "math.poly_radial_profile" [static] |