#include <processor.h>
Inheritance diagram for EMAN::FlipProcessor:


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 = "xform.flip" |
| axis | 'x', 'y', or 'z' axis. 'x' means horizonal flip; 'y' means vertical flip; |
Definition at line 4465 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 4487 of file processor.h. 04488 {
04489 return "flip an image around an axis.";
04490 }
|
|
|
Get the processor's name. Each processor is identified by a unique name.
Implements EMAN::Processor. Definition at line 4470 of file processor.h. 04471 {
04472 return NAME;
04473 }
|
|
|
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 4480 of file processor.h. References EMAN::TypeDict::put(). 04481 {
04482 TypeDict d;
04483 d.put("axis", EMObject::STRING, "'x', 'y', or 'z' axis. 'x' means horizonal flip; 'y' means vertical flip;");
04484 return d;
04485 }
|
|
|
Definition at line 4475 of file processor.h. 04476 {
04477 return new FlipProcessor();
04478 }
|
|
|
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 4272 of file processor.cpp. References EMAN::EMData::get_data(), EMAN::EMData::get_xsize(), EMAN::EMData::get_ysize(), EMAN::EMData::get_zsize(), LOGWARN, nx, ny, EMAN::Processor::params, EMAN::EMData::process_inplace(), t, EMAN::EMData::update(), x, and y. 04273 {
04274 ENTERFUNC;
04275 if (!image) {
04276 LOGWARN("NULL Image");
04277 return;
04278 }
04279 string axis = (const char*)params["axis"];
04280
04281 #ifdef EMAN2_USING_CUDA
04282 if (image->getcudarwdata()) {
04283 //cout << "flip processor" << endl;
04284 float array[12] = {1.0, 0.0, 0.0, 0.0,
04285 0.0, 1.0, 0.0, 0.0,
04286 0.0, 0.0, 1.0, 0.0};
04287 if (axis == "x" || axis == "X") { // horizontal flip
04288 array[0] = -1.0;
04289 }else if (axis == "y" || axis == "Y") { // vertical flip
04290 array[5] = -1.0;
04291 }
04292 else if (axis == "z" || axis == "Z") { // vertical flip
04293 array[10] = -1.0;
04294 }
04295 Transform t(array);
04296 Dict params("transform",(Transform*)&t);
04297 image->process_inplace("xform",params);
04298
04299 EXITFUNC;
04300 return;
04301 }
04302 #endif
04303
04304
04305 float *d = image->get_data();
04306 int nx = image->get_xsize();
04307 int ny = image->get_ysize();
04308 int nz = image->get_zsize();
04309
04310 size_t nxy = nx * ny;
04311
04312
04313 // Note in all cases the origin is nx/2, ny/2 and nz/2
04314 // This means when flipping even sized dimensions that some pixels are redundant.
04315 // Here redundant pixels are set to zero, however, should this change to something
04316 // like the mean.
04317 if (axis == "x" || axis == "X") { // Horizontal flip
04318 int offset = (nx%2 == 0);
04319 size_t idx1, idx2;
04320 for(int z = 0; z < nz; ++z) {
04321 for(int y = 0; y < ny; ++y) {
04322 if (offset != 0 ) {
04323 idx1 = z*nxy + y*nx;
04324 d[idx1] = 0; // Here's where you'd make it the mean
04325 }
04326 for(int x = offset; x < nx / 2; ++x) {
04327 idx1 = z*nxy + y*nx + x;
04328 idx2 = z*nxy + y*nx + (nx-x-1+offset);
04329 std::swap(d[idx1], d[idx2]);
04330 }
04331
04332 }
04333 }
04334 }
04335
04336 else if (axis == "y" || axis == "Y") { // vertical flip
04337 int offset = (ny%2 == 0);
04338 for(int z=0; z<nz; ++z) {
04339 if (offset != 0) {
04340 std::fill(d+z*nxy,d+(size_t)z*nxy+nx,0); // So if we have change it to the mean it's easy to do so. (instead of using memset)
04341 }
04342 for(int y=offset; y<ny/2; ++y) {
04343 for(int x=0; x<nx; ++x) {
04344 std::swap(d[(size_t)z*nxy + y*nx +x], d[(size_t)z*nxy + (ny -y -1+offset)*nx +x]);
04345 }
04346 }
04347 }
04348 }
04349 else if (axis == "z" || axis == "Z") { //z axis flip
04350 int offset = (nz%2 == 0);
04351 if (offset != 0) {
04352 std::fill(d,d+nxy,0);// So if we have change it to the mean it's easy to do so. (instead of using memset)
04353 }
04354 size_t idx1, idx2;
04355 for(int z=offset; z<nz/2; ++z) {
04356 for(int y=0; y<ny; ++y) {
04357 for(int x=0; x<nx; ++x) {
04358 idx1 = (size_t)z*nxy + y*nx + x;
04359 idx2 = (size_t)(nz-z-1+offset)*nxy + y*nx + x;
04360 std::swap(d[idx1], d[idx2]);
04361 }
04362 }
04363 }
04364 }
04365
04366 image->update();
04367 EXITFUNC;
04368 }
|
|
|
Definition at line 157 of file processor.cpp. |
1.3.9.1