#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 5643 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 5653 of file processor.h.
05654 { 05655 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."; 05656 }
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 5648 of file processor.h.
References NAME.
05649 { 05650 return NAME; 05651 }
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 5663 of file processor.h.
References EMAN::EMObject::FLOAT, and EMAN::TypeDict::put().
05664 { 05665 TypeDict d; 05666 d.put("val1", EMObject::FLOAT, "number of pixels to expand"); 05667 d.put("val2", EMObject::FLOAT, "number of Gaussian pixels to expand, following the first expansion"); 05668 return d; 05669 }
static Processor* EMAN::IterBinMaskProcessor::NEW | ( | ) | [inline, static] |
Definition at line 5658 of file processor.h.
05659 { 05660 return new IterBinMaskProcessor(); 05661 }
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 6528 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().
06529 { 06530 if (!image) { 06531 LOGWARN("NULL Image"); 06532 return; 06533 } 06534 06535 float val1 = params["val1"]; 06536 float val2 = params["val2"]; 06537 06538 int nx = image->get_xsize(); 06539 int ny = image->get_ysize(); 06540 int nz = image->get_zsize(); 06541 EMData *image2 = new EMData(nx,ny,nz); 06542 06543 // Got a compile warning complaining that these things were never used. Hope I didn't break anything - apologies if so (d.woolford) 06544 // float *dat = image->get_data(); 06545 // float *dat2 = image2->get_data(); 06546 06547 06548 float *d = image->get_data(); 06549 float *d2 = image2->get_data(); 06550 06551 const int nxy = nx * ny; 06552 size_t size = (size_t)nx * ny * nz; 06553 06554 // TODO: THIS IS EXTREMELY INEFFICIENT 06555 if (nz != 1) { 06556 for (int l = 1; l <= (int) val1+val2; ++l) { 06557 for (size_t i=0; i<size; i++) d2[i]=d[i]; 06558 for (int k = 1; k < nz - 1; ++k) { 06559 for (int j = 1; j < ny - 1; ++j) { 06560 for (int i = 1; i < nx - 1; ++i) { 06561 size_t t = i + j*nx+(size_t)k*nx*ny; 06562 if (d[t]) continue; 06563 if (d2[t - 1] || d2[t + 1] || d2[t + nx] || d2[t - nx] || d2[t + nxy] || d2[t - nxy]) d[t] = (float) l + 1; 06564 } 06565 } 06566 } 06567 } 06568 } 06569 else { 06570 for (int l = 1; l <= (int) val1+val2; ++l) { 06571 for (size_t i=0; i<size; i++) d2[i]=d[i]; 06572 for (int j = 1; j < ny - 1; ++j) { 06573 for (int i = 1; i < nx - 1; ++i) { 06574 size_t t = i + j * nx; 06575 if (d[t]) continue; 06576 if (d2[t - 1] || d2[t + 1] || d2[t + nx] || d2[t - nx]) d[t] = (float) l + 1; 06577 } 06578 } 06579 } 06580 } 06581 06582 vector<float> vec; 06583 for (int i=0; i<val1+2; i++) vec.push_back(1.0); 06584 for (int i=0; i<val2; i++) { 06585 vec.push_back(exp(-pow(2.0f*i/(val2),2.0f))); 06586 // printf("%f\n",exp(-pow(2.0*i/(val1-val2),2.0))); 06587 } 06588 for (size_t i = 0; i < size; ++i) if (d[i]) d[i]=vec[(int)d[i]]; 06589 06590 image->update(); 06591 delete image2; 06592 }
const string IterBinMaskProcessor::NAME = "mask.addshells.gauss" [static] |