#include <processor.h>
Inheritance diagram for EMAN::WaveletProcessor:
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 TypeDict | get_param_types () const |
Get processor parameter information in a dictionary. | |
virtual string | get_desc () const |
Get the descrition of this specific processor. | |
Static Public Member Functions | |
static Processor * | NEW () |
Static Public Attributes | |
static const string | NAME = "basis.wavelet" |
type | "daub", "harr", or "bspl" | |
dir | 1 for forward transform, -1 for inverse transform | |
ord | for Daubechies (4,6,8,...,20), for Harr (2), for B-Splines (103, 105, 202, 204, 206, 208, 301, 303, 305 307, 309) |
Definition at line 6508 of file processor.h.
virtual string EMAN::WaveletProcessor::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 6532 of file processor.h.
06533 { 06534 return "Computes the DWT (discrete wavelet transform) of an image in one of 3 possible bases"; 06535 }
virtual string EMAN::WaveletProcessor::get_name | ( | ) | const [inline, virtual] |
Get the processor's name.
Each processor is identified by a unique name.
Implements EMAN::Processor.
Definition at line 6513 of file processor.h.
References NAME.
Referenced by process_inplace().
06514 { 06515 return NAME; 06516 }
virtual TypeDict EMAN::WaveletProcessor::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 6523 of file processor.h.
References EMAN::EMObject::INT, EMAN::TypeDict::put(), and EMAN::EMObject::STRING.
06524 { 06525 TypeDict d; 06526 d.put("type", EMObject::STRING, "'daub', 'harr' or 'bspl'"); 06527 d.put("dir", EMObject::INT, "1 for forward transform, -1 for inverse transform"); 06528 d.put("ord", EMObject::INT, "Daubechies (4,6,8,...,20), for Harr (2), for B-Splines (103, 105, 202, 204, 206, 208, 301, 303, 305 307, 309)"); 06529 return d; 06530 }
static Processor* EMAN::WaveletProcessor::NEW | ( | ) | [inline, static] |
void WaveletProcessor::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 8003 of file processor.cpp.
References get_name(), EMAN::EMData::get_value_at(), EMAN::EMData::get_xsize(), EMAN::EMData::get_ysize(), EMAN::EMData::get_zsize(), ImageDimensionException, InvalidStringException, EMAN::Util::IsPower2(), LOGERR, EMAN::Processor::params, and EMAN::EMData::set_value_at_fast().
08004 { 08005 if (image->get_zsize() != 1) { 08006 LOGERR("%s Processor doesn't support 3D", get_name().c_str()); 08007 throw ImageDimensionException("3D model not supported"); 08008 } 08009 08010 int i,nx,ny; 08011 const gsl_wavelet_type * T; 08012 nx=image->get_xsize(); 08013 ny=image->get_ysize(); 08014 08015 if (nx != ny && ny!=1) throw ImageDimensionException("Wavelet transform only supports square images"); 08016 // float l=log((float)nx)/log(2.0f); 08017 // if (l!=floor(l)) throw ImageDimensionException("Wavelet transform size must be power of 2"); 08018 if( !Util::IsPower2(nx) ) throw ImageDimensionException("Wavelet transform size must be power of 2"); 08019 08020 // Unfortunately GSL works only on double() arrays 08021 // eventually we should put our own wavelet code in here 08022 // but this will work for now 08023 double *cpy = (double *)malloc(nx*ny*sizeof(double)); 08024 08025 for (i=0; i<nx*ny; i++) cpy[i]=image->get_value_at(i,0,0); 08026 08027 string tp = (const char*)params["type"]; 08028 if (tp=="daub") T=gsl_wavelet_daubechies; 08029 else if (tp=="harr") T=gsl_wavelet_haar; 08030 else if (tp=="bspl") T=gsl_wavelet_bspline; 08031 else throw InvalidStringException(tp,"Invalid wavelet name, 'daub', 'harr' or 'bspl'"); 08032 08033 int K=(int)params["ord"]; 08034 gsl_wavelet_direction dir; 08035 if ((int)params["dir"]==1) dir=forward; 08036 else dir=backward; 08037 08038 gsl_wavelet *w = gsl_wavelet_alloc(T, K); 08039 gsl_wavelet_workspace *work = gsl_wavelet_workspace_alloc(nx); 08040 08041 if (ny==1) gsl_wavelet_transform (w,cpy, 1, nx, dir, work); 08042 else gsl_wavelet2d_transform (w, cpy, nx,nx,ny, dir, work); 08043 08044 gsl_wavelet_workspace_free (work); 08045 gsl_wavelet_free (w); 08046 08047 for (i=0; i<nx*ny; i++) image->set_value_at_fast(i,0,0,static_cast<float>(cpy[i])); 08048 08049 free(cpy); 08050 }
const string WaveletProcessor::NAME = "basis.wavelet" [static] |