X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=packages%2Fvtk%2Fsrc%2FbbvtkCreateImage.cxx;h=2c75a26041084cd93d041a8fcc734e6aef1c948f;hb=edb8e9b8948e7216e06b3ef5228b16f18417ca52;hp=c534252a5b3452d7c440b579464a69e342790b80;hpb=af00d6db1dc36f21081781f4c15e4762c2eb25b3;p=bbtk.git

diff --git a/packages/vtk/src/bbvtkCreateImage.cxx b/packages/vtk/src/bbvtkCreateImage.cxx
index c534252..2c75a26 100644
--- a/packages/vtk/src/bbvtkCreateImage.cxx
+++ b/packages/vtk/src/bbvtkCreateImage.cxx
@@ -53,99 +53,119 @@ void CreateImage::Process()
 //    * TYPE is the C++ type of the input/output
 //      (the one provided in the attribute 'type' of the tag 'input')
 
-	double	spc[3];
-	int		dim[3];	
-	int		outputformat = VTK_UNSIGNED_SHORT;
-	
-	if (bbGetInputOutputFormat()=="VTK_BIT")					outputformat = VTK_BIT;                 // 1
-	else if (bbGetInputOutputFormat()=="VTK_CHAR")				outputformat = VTK_CHAR;				// 2
-	else if (bbGetInputOutputFormat()=="MET_CHAR")				outputformat = VTK_CHAR;				// 2
-	else if (bbGetInputOutputFormat()=="VTK_SIGNED_CHAR")		outputformat = VTK_SIGNED_CHAR;     	// 15
-	else if (bbGetInputOutputFormat()=="MET_UCHAR")				outputformat = VTK_UNSIGNED_CHAR;       // 3
-	else if (bbGetInputOutputFormat()=="VTK_UNSIGNED_CHAR")		outputformat = VTK_UNSIGNED_CHAR;       // 3
-	else if (bbGetInputOutputFormat()=="VTK_SHORT")				outputformat = VTK_SHORT;               // 4
-	else if (bbGetInputOutputFormat()=="MET_SHORT")				outputformat = VTK_SHORT;               // 4
-	else if (bbGetInputOutputFormat()=="VTK_UNSIGNED_SHORT")	outputformat = VTK_UNSIGNED_SHORT;      // 5
-	else if (bbGetInputOutputFormat()=="MET_USHORT")			outputformat = VTK_UNSIGNED_SHORT;      // 5
-	else if (bbGetInputOutputFormat()=="VTK_INT")				outputformat = VTK_INT;                 // 6
-	else if (bbGetInputOutputFormat()=="MET_INT")				outputformat = VTK_INT;                 // 6
-	else if (bbGetInputOutputFormat()=="VTK_UNSIGNED_INT")		outputformat = VTK_UNSIGNED_INT;        // 7 
-	else if (bbGetInputOutputFormat()=="MEY_UINT")				outputformat = VTK_UNSIGNED_INT;        // 7 
-	else if (bbGetInputOutputFormat()=="VTK_LONG")				outputformat = VTK_LONG;                // 8
-	else if (bbGetInputOutputFormat()=="MET_LONG")				outputformat = VTK_LONG;                // 8
-	else if (bbGetInputOutputFormat()=="VTK_UNSIGNED_LONG")		outputformat = VTK_UNSIGNED_LONG;       // 9
-	else if (bbGetInputOutputFormat()=="MET_ULONG")				outputformat = VTK_UNSIGNED_LONG;       // 9
-	else if (bbGetInputOutputFormat()=="VTK_FLOAT")				outputformat = VTK_FLOAT;               // 10
-	else if (bbGetInputOutputFormat()=="MET_FLOAT")				outputformat = VTK_FLOAT;               // 10
-	else if (bbGetInputOutputFormat()=="VTK_DOUBLE")			outputformat = VTK_DOUBLE;              // 11  		
-	else if (bbGetInputOutputFormat()=="MET_DOUBLE")			outputformat = VTK_DOUBLE;              // 11  
-
-	spc[0] = bbGetInputSpacing()[0];
-	spc[1] = bbGetInputSpacing()[1];
-	spc[2] = bbGetInputSpacing()[2];
-	dim[0] =  bbGetInputDimensions()[0];
-	dim[1] =  bbGetInputDimensions()[1];
-	dim[2] =  bbGetInputDimensions()[2];
-	if ((dim[0]<=0)||(dim[1]<=0)||(dim[2]<=0))
-	{
-		dim[0]=1;
-		dim[1]=1;
-		dim[2]=1;
-	}
-	if ((spc[0]<=0)||(spc[1]<=0)||(spc[2]<=0))
+    int		dim[3];
+	dim[0] 	= bbGetInputDimensions()[0];
+	dim[1] 	= bbGetInputDimensions()[1];
+	dim[2] 	= bbGetInputDimensions()[2];
+	if (imageoutput!=NULL)
 	{
-		spc[0]=1;
-		spc[1]=1;
-		spc[2]=1;
-	}
-//EED 2017-12-7
-	if (imageoutput!=NULL) {imageoutput->Delete();}	
-	imageoutput = vtkImageData::New();
-	imageoutput->Initialize();
-	imageoutput->SetSpacing( spc );
-	imageoutput->SetDimensions(  dim[0], dim[1], dim[2] );
-
-//EED 2017-01-01 Migration VTK7
-#if (VTK_MAJOR_VERSION <= 5) 
-	imageoutput->SetScalarType( outputformat );	
-	imageoutput->AllocateScalars();
-#endif
-#if (VTK_MAJOR_VERSION >= 6) 
-	imageoutput->AllocateScalars( outputformat,1 );
-#endif
-	
-	int i,j,k;
-	long sizeBlock;	
+		imageoutput->Delete();
+		imageoutput = NULL;
+	} // if imageoutput
 
-	for (i=0; i<dim[0]; i++)
+	if ( !((dim[0]<=0)||(dim[1]<=0)||(dim[2]<=0)) )
 	{
-		imageoutput->SetScalarComponentFromDouble(i,0,0, 0, bbGetInputInitialValue() );
-	} // for i
-	
-	sizeBlock= dim[0] * imageoutput->GetScalarSize();
-    for (j=1; j<dim[1]; j++)
-	{
-		memcpy(  imageoutput->GetScalarPointer(0,j,0), imageoutput->GetScalarPointer()  , sizeBlock );
-	} // for j		
-
-	sizeBlock= dim[0] * dim[1] * imageoutput->GetScalarSize();
-	for (k=0; k<dim[2]; k++)
-	{
-		memcpy(  imageoutput->GetScalarPointer(0,0,k) ,imageoutput->GetScalarPointer()  , sizeBlock );
-	} // for k
-	
-
-//EED 2017-01-01 Migration VTK7
-#if (VTK_MAJOR_VERSION <= 5) 
-	imageoutput->Update();
-#endif
-#if (VTK_MAJOR_VERSION >= 6) 
-	// ..
-#endif
-
-	imageoutput->Modified();
+		double	spc[3];
+		int		outputformat = VTK_UNSIGNED_SHORT;
+
+		spc[0]	= 1;
+		spc[1]	= 1;
+		spc[2]	= 1;
+		if  ( bbGetInputSpacing().size()==3 )
+		{
+			spc[0] = bbGetInputSpacing()[0];
+			spc[1] = bbGetInputSpacing()[1];
+			spc[2] = bbGetInputSpacing()[2];
+			if ((spc[0]<=0)||(spc[1]<=0)||(spc[2]<=0))
+			{
+				spc[0] = 1;
+				spc[1] = 1;
+				spc[2] = 1;
+			}
+		} // size spc 3
+		if  ( bbGetInputSpacing().size()==2 )
+		{
+			spc[0] = bbGetInputSpacing()[0];
+			spc[1] = bbGetInputSpacing()[1];
+			if ((spc[0]<=0)||(spc[1]<=0))
+			{
+				spc[0] = 1;
+				spc[1] = 1;
+			}
+		} // if size spc 2
+
+		
+		if (bbGetInputOutputFormat()=="VTK_BIT")					outputformat = VTK_BIT;                 // 1
+		else if (bbGetInputOutputFormat()=="VTK_CHAR")				outputformat = VTK_CHAR;				// 2
+		else if (bbGetInputOutputFormat()=="MET_CHAR")				outputformat = VTK_CHAR;				// 2
+		else if (bbGetInputOutputFormat()=="VTK_SIGNED_CHAR")		outputformat = VTK_SIGNED_CHAR;     	// 15
+		else if (bbGetInputOutputFormat()=="MET_UCHAR")				outputformat = VTK_UNSIGNED_CHAR;       // 3
+		else if (bbGetInputOutputFormat()=="VTK_UNSIGNED_CHAR")		outputformat = VTK_UNSIGNED_CHAR;       // 3
+		else if (bbGetInputOutputFormat()=="VTK_SHORT")				outputformat = VTK_SHORT;               // 4
+		else if (bbGetInputOutputFormat()=="MET_SHORT")				outputformat = VTK_SHORT;               // 4
+		else if (bbGetInputOutputFormat()=="VTK_UNSIGNED_SHORT")	outputformat = VTK_UNSIGNED_SHORT;      // 5
+		else if (bbGetInputOutputFormat()=="MET_USHORT")			outputformat = VTK_UNSIGNED_SHORT;      // 5
+		else if (bbGetInputOutputFormat()=="VTK_INT")				outputformat = VTK_INT;                 // 6
+		else if (bbGetInputOutputFormat()=="MET_INT")				outputformat = VTK_INT;                 // 6
+		else if (bbGetInputOutputFormat()=="VTK_UNSIGNED_INT")		outputformat = VTK_UNSIGNED_INT;        // 7 
+		else if (bbGetInputOutputFormat()=="MEY_UINT")				outputformat = VTK_UNSIGNED_INT;        // 7 
+		else if (bbGetInputOutputFormat()=="VTK_LONG")				outputformat = VTK_LONG;                // 8
+		else if (bbGetInputOutputFormat()=="MET_LONG")				outputformat = VTK_LONG;                // 8
+		else if (bbGetInputOutputFormat()=="VTK_UNSIGNED_LONG")		outputformat = VTK_UNSIGNED_LONG;       // 9
+		else if (bbGetInputOutputFormat()=="MET_ULONG")				outputformat = VTK_UNSIGNED_LONG;       // 9
+		else if (bbGetInputOutputFormat()=="VTK_FLOAT")				outputformat = VTK_FLOAT;               // 10
+		else if (bbGetInputOutputFormat()=="MET_FLOAT")				outputformat = VTK_FLOAT;               // 10
+		else if (bbGetInputOutputFormat()=="VTK_DOUBLE")			outputformat = VTK_DOUBLE;              // 11  		
+		else if (bbGetInputOutputFormat()=="MET_DOUBLE")			outputformat = VTK_DOUBLE;              // 11  
+
+		if ((dim[0]<=0)||(dim[1]<=0)||(dim[2]<=0))
+		{
+			dim[0] = 1;
+			dim[1] = 1;
+			dim[2] = 1;
+		}
+
+		imageoutput = vtkImageData::New();
+		imageoutput->Initialize();
+		imageoutput->SetSpacing( spc );
+		imageoutput->SetDimensions(  dim[0], dim[1], dim[2] );
+
+	//EED 2017-01-01 Migration VTK7
+	#if (VTK_MAJOR_VERSION <= 5) 
+		imageoutput->SetScalarType( outputformat );	
+		imageoutput->AllocateScalars();
+	#endif
+	#if (VTK_MAJOR_VERSION >= 6) 
+		imageoutput->AllocateScalars( outputformat,1 );
+	#endif
+		
+		int i,j,k;
+		long sizeBlock;	
+		for (i=0; i<dim[0]; i++)
+		{
+			imageoutput->SetScalarComponentFromDouble(i,0,0, 0, bbGetInputInitialValue() );
+		} // for i
+		sizeBlock= dim[0] * imageoutput->GetScalarSize();
+		for (j=1; j<dim[1]; j++)
+		{
+			memcpy(  imageoutput->GetScalarPointer(0,j,0), imageoutput->GetScalarPointer()  , sizeBlock );
+		} // for j		
+		sizeBlock= dim[0] * dim[1] * imageoutput->GetScalarSize();
+		for (k=0; k<dim[2]; k++)
+		{
+			memcpy(  imageoutput->GetScalarPointer(0,0,k) ,imageoutput->GetScalarPointer()  , sizeBlock );
+		} // for k
+	//EED 2017-01-01 Migration VTK7
+	#if (VTK_MAJOR_VERSION <= 5) 
+		imageoutput->Update();
+	#endif
+	#if (VTK_MAJOR_VERSION >= 6) 
+		// ..
+	#endif
+		imageoutput->Modified();
+	} // if dim = 0
 	bbSetOutputOut(imageoutput);
-}	
+}
 	
 //===== 
 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)