Class that generates a grid over an image in order to visualize the deformation field...

[creaRigidRegistration.git] / lib / MyGridOnImageGenerator.cpp

diff --git a/lib/MyGridOnImageGenerator.cpp b/lib/MyGridOnImageGenerator.cpp
new file mode 100755 (executable)
index 0000000..8b7f566
--- /dev/null
+++ b/lib/MyGridOnImageGenerator.cpp
@@ -0,0 +1,111 @@
+//----------------------------------------------------------------------------
+// Class definition include
+//----------------------------------------------------------------------------
+
+#include "MyGridOnImageGenerator.h"
+
+//----------------------------------------------------------------------------
+// Class implementation
+//----------------------------------------------------------------------------
+//----------------------------------------------------------------------------
+// Builder
+//----------------------------------------------------------------------------
+
+MyGridOnImageGenerator::MyGridOnImageGenerator( vtkImageData* nImage, double nScpX, double nScpY, double nScpZ)
+{
+       spcX = nScpX;
+       spcY = nScpY;
+       spcZ = nScpZ;
+       image = nImage;
+}
+
+//----------------------------------------------------------------------------
+// Destructor
+//----------------------------------------------------------------------------
+
+MyGridOnImageGenerator::~MyGridOnImageGenerator( )
+{
+
+}
+//----------------------------------------------------------------------------
+// Methods
+//----------------------------------------------------------------------------
+
+
+void MyGridOnImageGenerator::setScpX(double nScpX)
+{
+       spcX = nScpX;
+}
+
+void MyGridOnImageGenerator::setScpY(double nScpY)
+{
+       spcY = nScpY;
+}
+
+void MyGridOnImageGenerator::setScpZ(double nScpZ)
+{
+       spcZ = nScpZ;
+}
+
+
+void MyGridOnImageGenerator::setImage(vtkImageData* nImage)
+{
+       image = nImage;
+}
+
+vtkImageData* MyGridOnImageGenerator::getGridOnImage( )
+{
+       //Variables
+       double spc[3];
+       int dims[3];
+       int sizeX = 0, sizeY = 0, sizeZ = 0;
+       double minMaxValue[2];
+       double minValue = 0, maxValue = 0;
+       int spcVoxelX = 0, spcVoxelY = 0, spcVoxelZ = 0;
+
+       vtkImageData* gridImage = vtkImageData::New();
+       gridImage->ShallowCopy(image);
+       gridImage->Update();
+
+       //Get image spacing, size and mix and max values
+       gridImage->GetSpacing(spc);
+       gridImage->GetDimensions(dims);
+       sizeX = dims[0];
+       sizeY = dims[1];
+       sizeZ = dims[2];
+       gridImage->GetScalarRange(minMaxValue);
+       minValue = minMaxValue[0];
+       maxValue = minMaxValue[1];
+
+       //Grid spacing in voxels
+       spcVoxelX = spcX / spc[0];
+       spcVoxelY = spcY / spc[1];
+       spcVoxelZ = spcZ / spc[2];
+
+       std::cout << "spcVoxelX:" << spcVoxelX << ", spcVoxelY:" << spcVoxelY << ", spcVoxelZ:" << spcVoxelZ <<std::endl;
+       std::cout << "minValue:" << minValue << ", maxValue:" << maxValue << std::endl;
+       std::cout << "sizeX:" << sizeX << ", sizeY:" << sizeY << ", sizeZ:" << sizeZ <<std::endl;
+
+       for(int px = 0; px < sizeX; px++)
+       {
+               for(int py = 0; py < sizeY; py++)
+               {
+                       for(int pz = 0; pz < sizeZ; pz++)
+                       {
+                               if(px % spcVoxelX == 0 || py % spcVoxelY == 0 || pz % spcVoxelZ == 0)
+                               {
+                                       //std::cout << "In" ;
+                                       unsigned char* pointer = (unsigned char*) gridImage->GetScalarPointer(px, py, pz);
+                                       if( (*pointer - minValue) < (maxValue - *pointer) )
+                                               *pointer = maxValue;
+                                       else
+                                               *pointer = minValue;
+                               }
+                       }
+               }
+       }
+
+
+       return gridImage;
+}
+