#include <processor.h>
Inheritance diagram for EMAN::IterBinMaskProcessor:
Public Member Functions | |
virtual void | process_inplace (EMData *image) |
To process an image in-place. | |
virtual string | get_name () const |
Get the processor's name. | |
virtual 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 = "mask.addshells.gauss" |
val1 | number of pixels to expand | |
val2 | number of Gaussian pixels to expand, following the first expansion |
Definition at line 5768 of file processor.h.
virtual string EMAN::IterBinMaskProcessor::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 5778 of file processor.h.
05779 { 05780 return "Iterative expansion of a binary mask, val1 is number of pixels to expand, if val2!=0 will make a soft Gaussian edge starting after val2 pixels."; 05781 }
virtual string EMAN::IterBinMaskProcessor::get_name | ( | ) | const [inline, virtual] |
Get the processor's name.
Each processor is identified by a unique name.
Implements EMAN::Processor.
Definition at line 5773 of file processor.h.
References NAME.
05774 { 05775 return NAME; 05776 }
virtual TypeDict EMAN::IterBinMaskProcessor::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 5788 of file processor.h.
References EMAN::EMObject::FLOAT, and EMAN::TypeDict::put().
05789 { 05790 TypeDict d; 05791 d.put("val1", EMObject::FLOAT, "number of pixels to expand"); 05792 d.put("val2", EMObject::FLOAT, "number of Gaussian pixels to expand, following the first expansion"); 05793 return d; 05794 }
static Processor* EMAN::IterBinMaskProcessor::NEW | ( | ) | [inline, static] |
Definition at line 5783 of file processor.h.
05784 { 05785 return new IterBinMaskProcessor(); 05786 }
void IterBinMaskProcessor::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 6644 of file processor.cpp.
References EMAN::EMData::get_data(), EMAN::EMData::get_xsize(), EMAN::EMData::get_ysize(), EMAN::EMData::get_zsize(), LOGWARN, EMAN::Processor::params, t, and EMAN::EMData::update().
06645 { 06646 if (!image) { 06647 LOGWARN("NULL Image"); 06648 return; 06649 } 06650 06651 float val1 = params["val1"]; 06652 float val2 = params["val2"]; 06653 06654 int nx = image->get_xsize(); 06655 int ny = image->get_ysize(); 06656 int nz = image->get_zsize(); 06657 EMData *image2 = new EMData(nx,ny,nz); 06658 06659 // Got a compile warning complaining that these things were never used. Hope I didn't break anything - apologies if so (d.woolford) 06660 // float *dat = image->get_data(); 06661 // float *dat2 = image2->get_data(); 06662 06663 06664 float *d = image->get_data(); 06665 float *d2 = image2->get_data(); 06666 06667 const int nxy = nx * ny; 06668 size_t size = (size_t)nx * ny * nz; 06669 06670 // TODO: THIS IS EXTREMELY INEFFICIENT 06671 if (nz != 1) { 06672 for (int l = 1; l <= (int) val1+val2; ++l) { 06673 for (size_t i=0; i<size; i++) d2[i]=d[i]; 06674 for (int k = 1; k < nz - 1; ++k) { 06675 for (int j = 1; j < ny - 1; ++j) { 06676 for (int i = 1; i < nx - 1; ++i) { 06677 size_t t = i + j*nx+(size_t)k*nx*ny; 06678 if (d[t]) continue; 06679 if (d2[t - 1] || d2[t + 1] || d2[t + nx] || d2[t - nx] || d2[t + nxy] || d2[t - nxy]) d[t] = (float) l + 1; 06680 } 06681 } 06682 } 06683 } 06684 } 06685 else { 06686 for (int l = 1; l <= (int) val1+val2; ++l) { 06687 for (size_t i=0; i<size; i++) d2[i]=d[i]; 06688 for (int j = 1; j < ny - 1; ++j) { 06689 for (int i = 1; i < nx - 1; ++i) { 06690 size_t t = i + j * nx; 06691 if (d[t]) continue; 06692 if (d2[t - 1] || d2[t + 1] || d2[t + nx] || d2[t - nx]) d[t] = (float) l + 1; 06693 } 06694 } 06695 } 06696 } 06697 06698 vector<float> vec; 06699 for (int i=0; i<val1+2; i++) vec.push_back(1.0); 06700 for (int i=0; i<val2; i++) { 06701 vec.push_back(exp(-pow(2.0f*i/(val2),2.0f))); 06702 // printf("%f\n",exp(-pow(2.0*i/(val1-val2),2.0))); 06703 } 06704 for (size_t i = 0; i < size; ++i) if (d[i]) d[i]=vec[(int)d[i]]; 06705 06706 image->update(); 06707 delete image2; 06708 }
const string IterBinMaskProcessor::NAME = "mask.addshells.gauss" [static] |