#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 | |
| Processor * | NEW () |
Static Public Attributes | |
| 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 6564 of file processor.h.
|
|
Get the descrition of this specific processor. This function must be overwritten by a subclass.
Implements EMAN::Processor. Definition at line 6588 of file processor.h. 06589 {
06590 return "Computes the DWT (discrete wavelet transform) of an image in one of 3 possible bases";
06591 }
|
|
|
Get the processor's name. Each processor is identified by a unique name.
Implements EMAN::Processor. Definition at line 6569 of file processor.h. Referenced by process_inplace(). 06570 {
06571 return NAME;
06572 }
|
|
|
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 6579 of file processor.h. References EMAN::TypeDict::put(). 06580 {
06581 TypeDict d;
06582 d.put("type", EMObject::STRING, "'daub', 'harr' or 'bspl'");
06583 d.put("dir", EMObject::INT, "1 for forward transform, -1 for inverse transform");
06584 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)");
06585 return d;
06586 }
|
|
|
Definition at line 6574 of file processor.h. 06575 {
06576 return new WaveletProcessor();
06577 }
|
|
|
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.
Implements EMAN::Processor. Definition at line 8005 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, nx, ny, and EMAN::EMData::set_value_at_fast(). 08006 {
08007 if (image->get_zsize() != 1) {
08008 LOGERR("%s Processor doesn't support 3D", get_name().c_str());
08009 throw ImageDimensionException("3D model not supported");
08010 }
08011
08012 int i,nx,ny;
08013 const gsl_wavelet_type * T;
08014 nx=image->get_xsize();
08015 ny=image->get_ysize();
08016
08017 if (nx != ny && ny!=1) throw ImageDimensionException("Wavelet transform only supports square images");
08018 // float l=log((float)nx)/log(2.0f);
08019 // if (l!=floor(l)) throw ImageDimensionException("Wavelet transform size must be power of 2");
08020 if( !Util::IsPower2(nx) ) throw ImageDimensionException("Wavelet transform size must be power of 2");
08021
08022 // Unfortunately GSL works only on double() arrays
08023 // eventually we should put our own wavelet code in here
08024 // but this will work for now
08025 double *cpy = (double *)malloc(nx*ny*sizeof(double));
08026
08027 for (i=0; i<nx*ny; i++) cpy[i]=image->get_value_at(i,0,0);
08028
08029 string tp = (const char*)params["type"];
08030 if (tp=="daub") T=gsl_wavelet_daubechies;
08031 else if (tp=="harr") T=gsl_wavelet_haar;
08032 else if (tp=="bspl") T=gsl_wavelet_bspline;
08033 else throw InvalidStringException(tp,"Invalid wavelet name, 'daub', 'harr' or 'bspl'");
08034
08035 int K=(int)params["ord"];
08036 gsl_wavelet_direction dir;
08037 if ((int)params["dir"]==1) dir=forward;
08038 else dir=backward;
08039
08040 gsl_wavelet *w = gsl_wavelet_alloc(T, K);
08041 gsl_wavelet_workspace *work = gsl_wavelet_workspace_alloc(nx);
08042
08043 if (ny==1) gsl_wavelet_transform (w,cpy, 1, nx, dir, work);
08044 else gsl_wavelet2d_transform (w, cpy, nx,nx,ny, dir, work);
08045
08046 gsl_wavelet_workspace_free (work);
08047 gsl_wavelet_free (w);
08048
08049 for (i=0; i<nx*ny; i++) image->set_value_at_fast(i,0,0,static_cast<float>(cpy[i]));
08050
08051 free(cpy);
08052 }
|
|
|
Definition at line 211 of file processor.cpp. |
1.3.9.1