X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=bbtk%2Fsrc%2FbbcreaMaracasVisuTubeFilter.cxx;h=e3989b55f32b3b274128bc5fdc82736f6db175f9;hb=9efc4d8bf8fa16e76153eb2efccc5057a79f3d29;hp=16441924a4e790b274dd027b7d2054fa4e3ecb3c;hpb=8b01430072604956a52213748415d28c87e018a6;p=creaMaracasVisu.git

diff --git a/bbtk/src/bbcreaMaracasVisuTubeFilter.cxx b/bbtk/src/bbcreaMaracasVisuTubeFilter.cxx
index 1644192..e3989b5 100644
--- a/bbtk/src/bbcreaMaracasVisuTubeFilter.cxx
+++ b/bbtk/src/bbcreaMaracasVisuTubeFilter.cxx
@@ -1,3 +1,28 @@
+/*# ---------------------------------------------------------------------
+#
+# Copyright (c) CREATIS (Centre de Recherche en Acquisition et Traitement de l'Image
+#                        pour la Sant�)
+# Authors : Eduardo Davila, Frederic Cervenansky, Claire Mouton
+# Previous Authors : Laurent Guigues, Jean-Pierre Roux
+# CreaTools website : www.creatis.insa-lyon.fr/site/fr/creatools_accueil
+#
+#  This software is governed by the CeCILL-B license under French law and
+#  abiding by the rules of distribution of free software. You can  use,
+#  modify and/ or redistribute the software under the terms of the CeCILL-B
+#  license as circulated by CEA, CNRS and INRIA at the following URL
+#  http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
+#  or in the file LICENSE.txt.
+#
+#  As a counterpart to the access to the source code and  rights to copy,
+#  modify and redistribute granted by the license, users are provided only
+#  with a limited warranty  and the software's author,  the holder of the
+#  economic rights,  and the successive licensors  have only  limited
+#  liability.
+#
+#  The fact that you are presently reading this means that you have had
+#  knowledge of the CeCILL-B license and that you accept its terms.
+# ------------------------------------------------------------------------ */
+
 //===== 
 // 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)
 //===== 
@@ -55,6 +80,13 @@ namespace bbcreaMaracasVisu
 		this->lstPointZ = lstPointZ;
 	}
 	
+	
+	void MaracasTubeFilter::SetlstColour( std::vector<double> lstColour  )
+	{
+		this->lstColour=lstColour;
+	}
+
+	
 	void MaracasTubeFilter::SetlstRadius( std::vector<double> lstRadius  )
 	{
 		this->lstRadius = lstRadius;
@@ -87,56 +119,91 @@ namespace bbcreaMaracasVisu
 			points->InsertPoint(i, lstPointX[i],  lstPointY[i],  lstPointZ[i]);
 		}
 		
-		vtkSmartPointer<vtkCellArray> lines =
-		vtkSmartPointer<vtkCellArray>::New();
+		vtkSmartPointer<vtkCellArray> lines =	vtkSmartPointer<vtkCellArray>::New();
 		lines->InsertNextCell( lstPointX.size() );
 		for (i = 0; i < lstPointX.size(); i++)
 		{
 			lines->InsertCellPoint(i);
 		}
 		
-		vtkSmartPointer<vtkPolyData> polyData =
-		vtkSmartPointer<vtkPolyData>::New();
+		vtkSmartPointer<vtkPolyData> polyData =	vtkSmartPointer<vtkPolyData>::New();
 		polyData->SetPoints(points);
 		polyData->SetLines(lines);
 		
-
+		double radio;
 		// Varying tube radius using sine-function
-		vtkSmartPointer<vtkDoubleArray> tubeRadius =
-		vtkSmartPointer<vtkDoubleArray>::New();
+		vtkSmartPointer<vtkDoubleArray> tubeRadius = vtkSmartPointer<vtkDoubleArray>::New();
 		tubeRadius->SetName("TubeRadius");
-		tubeRadius->SetNumberOfTuples( lstRadius.size() );
-		for (i=0 ;i<lstRadius.size() ; i++)
+		tubeRadius->SetNumberOfTuples( lstPointX.size() );
+		for (i=0 ;i<lstPointX.size() ; i++)
 		{
-			tubeRadius->SetTuple1(i, lstRadius[i] );
+			if (lstRadius.size()==0)
+			{ 
+			   radio=1;
+			} else if (i<lstRadius.size()) {
+			  radio = lstRadius[i];
+			} else if (i>=lstRadius.size()) {
+			  radio= lstRadius[ lstRadius.size()-1 ];
+			}   
+
+			tubeRadius->SetTuple1(i, radio );
 		}
 		polyData->GetPointData()->AddArray(tubeRadius);
 		polyData->GetPointData()->SetActiveScalars("TubeRadius");
 		
 		// RBG array (could add Alpha channel too I guess...)
 		// Varying from blue to red
-		vtkSmartPointer<vtkUnsignedCharArray> colors =
-		vtkSmartPointer<vtkUnsignedCharArray>::New();
+		vtkSmartPointer<vtkUnsignedCharArray> colors = vtkSmartPointer<vtkUnsignedCharArray>::New();
 		colors->SetName("Colors");
 		colors->SetNumberOfComponents(3);
 		colors->SetNumberOfTuples( lstPointX.size() );
+		int numberOfColours = lstColour.size()/3;
+		int indexcolour;
 		for (i = 0; i < lstPointX.size() ;i++)
-		{
-			colors->InsertTuple3(i, 255 , 0.0 , 0.0 );
-		}
+		{			
+			if (numberOfColours==0){
+				colors->InsertTuple3(i, 1 ,  1 ,  1 );
+			} else {
+				if (i<numberOfColours){
+					indexcolour=i*3;
+				} else {
+					indexcolour=(numberOfColours-1)*3;
+				}
+				colors->InsertTuple3(i, 255*lstColour[indexcolour+0] ,  255*lstColour[indexcolour+1] ,  255*lstColour[indexcolour+2] );
+			}
+		} // for
 		polyData->GetPointData()->AddArray(colors);
 		
-		vtkSmartPointer<vtkTubeFilter> tube
-		= vtkSmartPointer<vtkTubeFilter>::New();
+		vtkSmartPointer<vtkTubeFilter> tube = vtkSmartPointer<vtkTubeFilter>::New();
+		tube->CappingOn();
+
+//EED 2017-01-01 Migration VTK7
+#if VTK_MAJOR_VERSION <= 5
 		tube->SetInput(polyData);
+#else
+		tube->SetInputData(polyData);
+#endif
+
 		tube->SetNumberOfSides(nTv);
 		tube->SetVaryRadiusToVaryRadiusByAbsoluteScalar();
 		
-		vtkSmartPointer<vtkPolyDataMapper> mapper =
-		vtkSmartPointer<vtkPolyDataMapper>::New();
+		vtkSmartPointer<vtkPolyDataMapper> mapper =	vtkSmartPointer<vtkPolyDataMapper>::New();
 		mapper->SetInputConnection(tube->GetOutputPort());
+//		mapper->ScalarVisibilityOn();
+//		mapper->SetScalarModeToUsePointFieldData();
+		
 		mapper->ScalarVisibilityOn();
+		
+//		mapper->SetScalarModeToUseCellData();
+//		mapper->SetScalarModeToUseCellFieldData();
+//		mapper->SetScalarModeToUseFieldData();
+//		mapper->SetScalarModeToUsePointData();
 		mapper->SetScalarModeToUsePointFieldData();
+		
+		mapper->SetColorModeToDefault();
+//		mapper->SetColorModeToMapScalars();		
+		
+		
 		mapper->SelectColorArray("Colors");
 		
 		//		vtkSmartPointer<vtkActor> actor =vtkSmartPointer<vtkActor>::New();
@@ -144,7 +211,6 @@ namespace bbcreaMaracasVisu
 		actor->SetMapper(mapper);
 		actor->GetProperty()->SetOpacity( opacity );
 
-		
 		if ( transform!=NULL )
 		{
 			actor->SetUserTransform( transform );
@@ -181,8 +247,6 @@ void TubeFilter::Process()
 //    * TYPE is the C++ type of the input/output
 //      (the one provided in the attribute 'type' of the tag 'input')
 	
-	printf("EED TubeFilter::Process start \n");
-
 	if (tubefilter!=NULL) 
 	{ 
 		delete tubefilter;
@@ -191,12 +255,20 @@ void TubeFilter::Process()
 	tubefilter = new MaracasTubeFilter();
 	tubefilter->SetvtkRenderer( bbGetInputRenderer() );	
 	tubefilter->SetlstPoints( bbGetInputlstPointX(), bbGetInputlstPointY(), bbGetInputlstPointZ() );
+
+	// Sets the default value for radius to 1 everywhere, if the radius input vevctor is empty.
+	if (bbGetInputlstRadio().size() == 0)
+	  {
+	    std::vector<double> radio;
+	    radio.resize(bbGetInputlstPointX().size(), 1.);
+	    bbSetInputlstRadio(radio);
+	  }
 	tubefilter->SetlstRadius( bbGetInputlstRadio() );
 	tubefilter->SetOpacity( bbGetInputOpacity() );
 	tubefilter->SetTransform( bbGetInputTransform() );	
+	tubefilter->SetlstColour( bbGetInputColour() );	
 	tubefilter->Run();
     bbSetOutputOutAxis( tubefilter->GetActor() );	
-	printf("EED TubeFilter::Process end \n");
 }
 	
 //===== 
@@ -208,7 +280,17 @@ void TubeFilter::bbUserSetDefaultValues()
 //    Here we initialize the input 'In' to 0
 	bbSetInputOpacity(1);
 	bbSetInputTransform(NULL);
+	
+	std::vector<double> colour;
+	// 0- gray
+	colour.push_back(1.0);
+	colour.push_back(0.0);
+	colour.push_back(0.0);
+	bbSetInputColour(colour);
 
+ 	std::vector<double> lstRadius;
+	lstRadius.push_back(1);
+        bbSetInputColour(lstRadius);
 }
 	
 //=====