X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=lib%2FcreaVtk%2FcreaVtkUnMosaicVectorVtkImageData.cpp;h=93faba51325d36df492f786bc2c5283684e52ddc;hb=3582c0f9b8380fde739856aa6a188da3262cf03b;hp=2e7847c5c8141a423c6e897d74951750a262aafa;hpb=b5ae034170e154e060837d657257f22c422d4b21;p=creaVtk.git

diff --git a/lib/creaVtk/creaVtkUnMosaicVectorVtkImageData.cpp b/lib/creaVtk/creaVtkUnMosaicVectorVtkImageData.cpp
index 2e7847c..93faba5 100644
--- a/lib/creaVtk/creaVtkUnMosaicVectorVtkImageData.cpp
+++ b/lib/creaVtk/creaVtkUnMosaicVectorVtkImageData.cpp
@@ -27,6 +27,9 @@
 
 #include "creaVtkUnMosaicVectorVtkImageData.h"
 
+#define _USE_MATH_DEFINES // for C
+#include <math.h>
+
 creaVtkUnMosaicVectorVtkImageData::creaVtkUnMosaicVectorVtkImageData()
 {
 }
@@ -35,52 +38,74 @@ creaVtkUnMosaicVectorVtkImageData::~creaVtkUnMosaicVectorVtkImageData()
 {
 }
 
-std::vector<vtkImageData*> creaVtkUnMosaicVectorVtkImageData::unMosaicVectorVtkImageData (int NbImagesPerRow, int NbImagesInMosaic, std::vector<vtkImageData*> bbGetInputIn)
+std::vector<vtkImageData*> creaVtkUnMosaicVectorVtkImageData::unMosaicVectorVtkImageData (std::vector<vtkImageData*> imageInput, std::vector<int> NbImagesInMosaicVector)
 {
-	int nbImagesPerRow =   NbImagesPerRow;
-	int nbImagesInMosaic = NbImagesInMosaic;	
-
-	if (nbImagesPerRow == 0 || nbImagesInMosaic == 0) 
-        {
+	std::vector<int> nbImagesPerRow;
+	std::vector<int> nbImagesInMosaic;
+	double imagesInMosaic;
+	for(int i = 0; i < NbImagesInMosaicVector.size(); i++)
+	{
+		imagesInMosaic = NbImagesInMosaicVector[i];
+		nbImagesPerRow.push_back( ceil(sqrt(imagesInMosaic)) );
+		nbImagesInMosaic.push_back(NbImagesInMosaicVector[i]);
+	} // for i
+	if(imageInput.size() != NbImagesInMosaicVector.size())
+	{
+		for(int j = NbImagesInMosaicVector.size(); j < imageInput.size(); j++)	
+		{
+			imagesInMosaic = NbImagesInMosaicVector[NbImagesInMosaicVector.size()-1];
+			nbImagesPerRow.push_back( ceil(sqrt(imagesInMosaic)) );		
+			nbImagesInMosaic.push_back( NbImagesInMosaicVector[NbImagesInMosaicVector.size()-1] );
+		} // for j
+	} // for imageInput size
+	if (nbImagesPerRow.size() == 0 || nbImagesInMosaic.size() == 0) 
+    {
    		std::cout << "VtkUnMosaicVectorVtkImageData ERROR: The number of Images by mosaic is not set " << std::endl; 
-	}
-
-	std::vector<vtkImageData*> imageIn = bbGetInputIn;
-
+	} // if nbImagesPerRow size
+	std::vector<vtkImageData*> imageIn = imageInput;
 	std::vector<vtkImageData*> mImageOut;
-	for (int i = 0 ; i < (unsigned int)imageIn.size() ; i++) {
-			mImageOut.push_back( unMosaic(imageIn[i], NbImagesPerRow, NbImagesInMosaic) );
-	}
-
+	for (int i = 0 ; i < (unsigned int)imageIn.size() ; i++) 
+	{
+		mImageOut.push_back( unMosaic(imageIn[i], nbImagesPerRow[i], nbImagesInMosaic[i]) );
+	} // for i
 	return mImageOut;	
 }
 
+//------------------------------------------------------------------------------
 vtkImageData * creaVtkUnMosaicVectorVtkImageData::unMosaic(vtkImageData *imageIn, int nbImagesPerRow, int numberOfImagesInMosaic)
 {   
-   int inputdims[3];
-   int outputdims[3];
-   imageIn->GetDimensions (inputdims);
-   unsigned short *input = (unsigned short *)(imageIn->GetScalarPointer());
-   imageIn->Update();
-     
-   unsigned int div = (unsigned int)ceil(sqrt( (double)numberOfImagesInMosaic ) );
-   outputdims[0] = inputdims[0] / div;
-   outputdims[1] = inputdims[1] / div;
-   outputdims[2] = numberOfImagesInMosaic;
+   	int inputdims[3];
+   	int outputdims[3];
+   	imageIn->GetDimensions (inputdims);
+   	unsigned short *input = (unsigned short *)(imageIn->GetScalarPointer());
+//EED 2017-01-01 Migration VTK7
+#if VTK_MAJOR_VERSION <= 5
+ 	imageIn->Update();
+#else
+ 	// ..
+#endif 
+   	unsigned int div = (unsigned int)ceil(sqrt( (double)numberOfImagesInMosaic ) );
+   	outputdims[0] = inputdims[0] / div;
+   	outputdims[1] = inputdims[1] / div;
+   	outputdims[2] = numberOfImagesInMosaic;
  
     vtkImageData *vtkImageOut;
     vtkImageOut = vtkImageData::New();
     vtkImageOut->SetDimensions( outputdims );
     vtkImageOut->SetExtent(0,outputdims[0]-1,0,outputdims[1]-1,0,outputdims[2]-1);
+    
+//EED 2017-01-01 Migration VTK7
+#if VTK_MAJOR_VERSION <= 5
     vtkImageOut->SetWholeExtent(0,outputdims[0]-1,0,outputdims[1]-1,0,outputdims[2]-1);
     vtkImageOut->SetNumberOfScalarComponents(1);
 //vtkImageOut->SetSpacing( blabla );
     vtkImageOut->SetScalarType( VTK_UNSIGNED_SHORT );
     vtkImageOut->AllocateScalars();
     vtkImageOut->Update();
-    
+#else
+    vtkImageOut->AllocateScalars(VTK_UNSIGNED_SHORT,1);
+#endif
     unsigned short *output =(unsigned short *)(vtkImageOut->GetScalarPointer());
-
     unsigned short *dest = output;
     int dimXImageElem = outputdims[0];
     int dimYImageElem = outputdims[1];
@@ -97,18 +122,8 @@ vtkImageData * creaVtkUnMosaicVectorVtkImageData::unMosaic(vtkImageData *imageIn
           memcpy(dest, input+debImage, dimXImageElem*sizeof(unsigned short));
           debImage += dimXImageElem*nbImagesPerRow;
           dest += dimXImageElem;
-       }
-    }
+       } // for j
+    } // for i
     return  vtkImageOut;    
 }
 
-//---------------------------------------------
-//Method template
-//---------------------------------------------
-/*
-void creaVtkUnMosaicVectorVtkImageData::FunctionName(int& parameterA)
-{
-  parameterA = 2 * parameterA;
-  return;
-}
-*/