]> Creatis software - creaMaracasVisu.git/blobdiff - bbtk/src/bbcreaMaracasVisuTubeFilter.cxx
#2989 creaMaracasVisu Bug New Normal - ManualContourModel duplicate last point...
[creaMaracasVisu.git] / bbtk / src / bbcreaMaracasVisuTubeFilter.cxx
index 16441924a4e790b274dd027b7d2054fa4e3ecb3c..dbb0a9e457551b8ab38f588d65e0fe566418746b 100644 (file)
@@ -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,84 @@ 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();
                tube->SetInput(polyData);
                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 +204,6 @@ namespace bbcreaMaracasVisu
                actor->SetMapper(mapper);
                actor->GetProperty()->SetOpacity( opacity );
 
-               
                if ( transform!=NULL )
                {
                        actor->SetUserTransform( transform );
@@ -181,8 +240,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 +248,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 +273,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);
 }
        
 //=====