#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 5731 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 5741 of file processor.h.
05742 { 05743 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."; 05744 }
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 5736 of file processor.h.
References NAME.
05737 { 05738 return NAME; 05739 }
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 5751 of file processor.h.
References EMAN::EMObject::FLOAT, and EMAN::TypeDict::put().
05752 { 05753 TypeDict d; 05754 d.put("val1", EMObject::FLOAT, "number of pixels to expand"); 05755 d.put("val2", EMObject::FLOAT, "number of Gaussian pixels to expand, following the first expansion"); 05756 return d; 05757 }
static Processor* EMAN::IterBinMaskProcessor::NEW | ( | ) | [inline, static] |
Definition at line 5746 of file processor.h.
05747 { 05748 return new IterBinMaskProcessor(); 05749 }
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 6495 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().
06496 { 06497 if (!image) { 06498 LOGWARN("NULL Image"); 06499 return; 06500 } 06501 06502 float val1 = params["val1"]; 06503 float val2 = params["val2"]; 06504 06505 int nx = image->get_xsize(); 06506 int ny = image->get_ysize(); 06507 int nz = image->get_zsize(); 06508 EMData *image2 = new EMData(nx,ny,nz); 06509 06510 // Got a compile warning complaining that these things were never used. Hope I didn't break anything - apologies if so (d.woolford) 06511 // float *dat = image->get_data(); 06512 // float *dat2 = image2->get_data(); 06513 06514 06515 float *d = image->get_data(); 06516 float *d2 = image2->get_data(); 06517 06518 const int nxy = nx * ny; 06519 size_t size = nx * ny * nz; 06520 06521 // TODO: THIS IS EXTREMELY INEFFICIENT 06522 if (nz != 1) { 06523 for (int l = 1; l <= (int) val1+val2; ++l) { 06524 for (size_t i=0; i<size; i++) d2[i]=d[i]; 06525 for (int k = 1; k < nz - 1; ++k) { 06526 for (int j = 1; j < ny - 1; ++j) { 06527 for (int i = 1; i < nx - 1; ++i) { 06528 size_t t = i + j*nx+k*nx*ny; 06529 if (d[t]) continue; 06530 if (d2[t - 1] || d2[t + 1] || d2[t + nx] || d2[t - nx] || d2[t + nxy] || d2[t - nxy]) d[t] = (float) l + 1; 06531 } 06532 } 06533 } 06534 } 06535 } 06536 else { 06537 for (int l = 1; l <= (int) val1+val2; ++l) { 06538 for (size_t i=0; i<size; i++) d2[i]=d[i]; 06539 for (int j = 1; j < ny - 1; ++j) { 06540 for (int i = 1; i < nx - 1; ++i) { 06541 size_t t = i + j * nx; 06542 if (d[t]) continue; 06543 if (d2[t - 1] || d2[t + 1] || d2[t + nx] || d2[t - nx]) d[t] = (float) l + 1; 06544 } 06545 } 06546 } 06547 } 06548 06549 vector<float> vec; 06550 for (int i=0; i<val1+2; i++) vec.push_back(1.0); 06551 for (int i=0; i<val2; i++) { 06552 vec.push_back(exp(-pow(2.0f*i/(val2),2.0f))); 06553 // printf("%f\n",exp(-pow(2.0*i/(val1-val2),2.0))); 06554 } 06555 for (size_t i = 0; i < size; i++) if (d[i]) d[i]=vec[(int)d[i]]; 06556 06557 image->update(); 06558 delete image2; 06559 }
const string IterBinMaskProcessor::NAME = "mask.addshells.gauss" [static] |