#include <processor.h>
Inheritance diagram for EMAN::AreaProcessor:
Public Member Functions | |
AreaProcessor () | |
void | process_inplace (EMData *image) |
To process an image in-place. | |
void | set_params (const Dict &new_params) |
Set the processor parameters using a key/value dictionary. | |
TypeDict | get_param_types () const |
Get processor parameter information in a dictionary. | |
string | get_desc () const |
Get the descrition of this specific processor. | |
Protected Member Functions | |
virtual void | process_pixel (float *pixel, float, float, float, float *area_matrix) const |
virtual void | create_kernel () const =0 |
Protected Attributes | |
int | areasize |
int | matrix_size |
float * | kernel |
int | nx |
int | ny |
int | nz |
This is the base class. Specific AreaProcessor needs to implement function create_kernel().
areasize | The width of the area to process (not radius) |
Definition at line 2824 of file processor.h.
EMAN::AreaProcessor::AreaProcessor | ( | ) | [inline] |
virtual void EMAN::AreaProcessor::create_kernel | ( | ) | const [protected, pure virtual] |
Implemented in EMAN::LaplacianProcessor, and EMAN::ZeroConstantProcessor.
Referenced by process_inplace().
string EMAN::AreaProcessor::get_desc | ( | ) | const [inline, virtual] |
Get the descrition of this specific processor.
This function must be overwritten by a subclass.
Implements EMAN::Processor.
Reimplemented in EMAN::LaplacianProcessor, and EMAN::ZeroConstantProcessor.
Definition at line 2846 of file processor.h.
02847 { 02848 return "AreaProcessor use pixel values and coordinates of a real-space square area. This is the base class. Specific AreaProcessor needs to implement function create_kernel()."; 02849 }
TypeDict EMAN::AreaProcessor::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 2839 of file processor.h.
References EMAN::EMObject::INT, and EMAN::TypeDict::put().
02840 { 02841 TypeDict d; 02842 d.put("areasize", EMObject::INT, "The width of the area to process (not radius)"); 02843 return d; 02844 }
void AreaProcessor::process_inplace | ( | EMData * | image | ) | [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 1623 of file processor.cpp.
References areasize, create_kernel(), data, EMAN::EMData::get_data(), EMAN::EMData::get_xsize(), EMAN::EMData::get_ysize(), EMAN::EMData::get_zsize(), kernel, LOGWARN, matrix(), matrix_size, nx, ny, nz, process_pixel(), EMAN::EMData::update(), x, and y.
01624 { 01625 if (!image) { 01626 LOGWARN("NULL Image"); 01627 return; 01628 } 01629 01630 float *data = image->get_data(); 01631 01632 nx = image->get_xsize(); 01633 ny = image->get_ysize(); 01634 nz = image->get_zsize(); 01635 01636 int n = (areasize - 1) / 2; 01637 matrix_size = areasize * areasize; 01638 01639 if (nz > 1) { 01640 matrix_size *= areasize; 01641 } 01642 01643 float *matrix = new float[matrix_size]; 01644 kernel = new float[matrix_size]; 01645 01646 size_t cpysize = areasize * sizeof(float); 01647 size_t start = (nx * ny + nx + 1) * n; 01648 01649 int xend = nx - n; 01650 int yend = ny - n; 01651 01652 int zstart = n; 01653 int zend = nz - n; 01654 01655 int zbox_start = 0; 01656 int zbox_end = areasize; 01657 01658 if (nz == 1) { 01659 zstart = 0; 01660 zend = 1; 01661 zbox_end = 1; 01662 } 01663 01664 size_t nsec = (size_t)nx * (size_t)ny; 01665 int box_nsec = areasize * areasize; 01666 01667 create_kernel(); 01668 01669 size_t total_size = (size_t)nx * (size_t)ny * (size_t)nz; 01670 float *data2 = new float[total_size]; 01671 memcpy(data2, data, total_size * sizeof(float)); 01672 01673 size_t k; 01674 for (int z = zstart; z < zend; z++) { 01675 for (int y = n; y < yend; y++) { 01676 for (int x = n; x < xend; x++) { 01677 01678 k = (size_t)z * nsec + y * nx + x; 01679 01680 for (int bz = zbox_start; bz < zbox_end; bz++) { 01681 for (int by = 0; by < areasize; by++) { 01682 memcpy(&matrix[(size_t)bz * box_nsec + by * areasize], 01683 &data2[k - start + bz * nsec + by * nx], cpysize); 01684 } 01685 } 01686 01687 process_pixel(&data[k], (float) x, (float) y, (float) z, matrix); 01688 } 01689 } 01690 } 01691 01692 if( matrix ) 01693 { 01694 delete[]matrix; 01695 matrix = 0; 01696 } 01697 01698 if( kernel ) 01699 { 01700 delete[]kernel; 01701 kernel = 0; 01702 } 01703 image->update(); 01704 }
virtual void EMAN::AreaProcessor::process_pixel | ( | float * | pixel, | |
float | , | |||
float | , | |||
float | , | |||
float * | area_matrix | |||
) | const [inline, protected, virtual] |
Reimplemented in EMAN::ZeroConstantProcessor.
Definition at line 2852 of file processor.h.
References kernel, and matrix_size.
Referenced by process_inplace().
02853 { 02854 for (int i = 0; i < matrix_size; i++) 02855 { 02856 *pixel += area_matrix[i] * kernel[i]; 02857 } 02858 }
void EMAN::AreaProcessor::set_params | ( | const Dict & | new_params | ) | [inline, virtual] |
Set the processor parameters using a key/value dictionary.
new_params | A dictionary containing the new parameters. |
Reimplemented from EMAN::Processor.
Definition at line 2833 of file processor.h.
References areasize, and EMAN::Processor::params.
int EMAN::AreaProcessor::areasize [protected] |
Definition at line 2862 of file processor.h.
Referenced by EMAN::LaplacianProcessor::create_kernel(), process_inplace(), and set_params().
float* EMAN::AreaProcessor::kernel [protected] |
Definition at line 2864 of file processor.h.
Referenced by EMAN::LaplacianProcessor::create_kernel(), process_inplace(), and process_pixel().
int EMAN::AreaProcessor::matrix_size [protected] |
int EMAN::AreaProcessor::nx [protected] |
int EMAN::AreaProcessor::ny [protected] |
int EMAN::AreaProcessor::nz [protected] |
Definition at line 2867 of file processor.h.
Referenced by EMAN::LaplacianProcessor::create_kernel(), and process_inplace().