#2453 creaVtk Feature New Normal - Stream Lines segmentation by Scalars

[creaVtk.git] / lib / creaVtk / creaVtkStreamLineScalarSeg.cpp

diff --git a/lib/creaVtk/creaVtkStreamLineScalarSeg.cpp b/lib/creaVtk/creaVtkStreamLineScalarSeg.cpp
new file mode 100644 (file)
index 0000000..95fd4c5
--- /dev/null
+++ b/lib/creaVtk/creaVtkStreamLineScalarSeg.cpp
@@ -0,0 +1,143 @@
+/*
+# ---------------------------------------------------------------------
+#
+# Copyright (c) CREATIS (Centre de Recherche en Acquisition et Traitement de l'Image
+#                        pour la Sante)
+# 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.
+# ------------------------------------------------------------------------
+*/
+
+#include "creaVtkStreamLineScalarSeg.h"
+#include "vtkPointData.h"
+#include "vtkDataArray.h"
+#include "vtkDoubleArray.h"
+
+
+// ------------------------------------------------------------------------
+creaVtkStreamLineScalarSeg::creaVtkStreamLineScalarSeg()
+{
+       _vtkidlist                      = vtkIdList::New();
+   _newvtkpolydata     = vtkPolyData::New();   
+   _newvtkcellarray    = vtkCellArray::New();  
+}
+
+// ------------------------------------------------------------------------
+creaVtkStreamLineScalarSeg::~creaVtkStreamLineScalarSeg()
+{
+}
+
+// ------------------------------------------------------------------------
+void creaVtkStreamLineScalarSeg::SetScalarStreamLinesIn( vtkPolyData* polydata)
+{
+       _StreamLinesIn = polydata;
+}
+
+// ------------------------------------------------------------------------
+void creaVtkStreamLineScalarSeg::SetScalarArrayName( std::string name)
+{
+       _ScalarArrayName = name;
+}
+
+// ------------------------------------------------------------------------
+void creaVtkStreamLineScalarSeg::SetThresholdLower( double value)
+{
+       _ThresholdLower = value;        
+}
+
+// ------------------------------------------------------------------------
+void creaVtkStreamLineScalarSeg::SetThresholdUpper( double value )
+{
+       _ThresholdUpper = value;        
+}
+
+// ------------------------------------------------------------------------
+void creaVtkStreamLineScalarSeg::Process()
+{
+       if (_ScalarArrayName=="")
+       {
+               printf(".\n ");
+               printf(".\n ");
+               printf("BBTK Warnning! ScalarArrayName is empty in creaVtkStreamLineScalarSeg \n");
+               printf(".\n ");
+               printf(".\n ");
+       }
+
+       if ((_StreamLinesIn!=NULL) && (_ScalarArrayName!="") )
+       {
+               int iLine;
+
+               _newvtkcellarray->Initialize();
+
+       vtkCellArray    *vtkcellarray           = _StreamLinesIn->GetLines();
+       int                             nLinesCell                      = vtkcellarray->GetNumberOfCells(); 
+//     int                             vtkcellarraysize        = vtkcellarray->GetSize();
+       int                             numberofids; 
+       int                             ii;
+               vtkDataArray    *scalarArray = _StreamLinesIn->GetPointData()->GetArray( _ScalarArrayName.c_str() );
+
+
+               vtkDoubleArray  *newScalarArray                         = vtkDoubleArray::New();
+               newScalarArray->SetName( _ScalarArrayName.c_str() );            
+               newScalarArray->SetNumberOfComponents(1); 
+
+       // FOR EACH LINE
+               double scalarValue;
+               long int iLineNew=0;
+       ii=0;
+       for ( iLine=0 ; iLine<nLinesCell ; iLine++ )
+       {
+                       vtkcellarray->GetCell(ii, _vtkidlist );
+                       numberofids = _vtkidlist->GetNumberOfIds();
+                       scalarValue = scalarArray->GetTuple1(iLine);
+                       if ( ( scalarValue>=_ThresholdLower ) && ( scalarValue<=_ThresholdUpper ) )
+                       {
+                               _newvtkcellarray->InsertNextCell( _vtkidlist );
+                               newScalarArray->InsertTuple1( iLineNew, scalarValue );
+                               iLineNew++;
+                       }
+
+                       ii=ii+numberofids+1;
+       } // for iLine
+
+               _newvtkpolydata->SetPoints( _StreamLinesIn->GetPoints() );
+               _newvtkpolydata->SetLines( _newvtkcellarray );
+               int iArrays,arraysSize = _StreamLinesIn->GetPointData()->GetNumberOfArrays();
+               for ( iArrays=0 ; iArrays<arraysSize ; iArrays++ )
+               {       
+                       _newvtkpolydata->GetPointData()->AddArray( _StreamLinesIn->GetPointData()->GetArray(iArrays) );
+               }               
+               _newvtkpolydata->GetPointData()->RemoveArray( _ScalarArrayName.c_str() );
+               _newvtkpolydata->GetPointData()->AddArray( newScalarArray );
+               _newvtkpolydata->Update();
+
+       _StreamLinesOut = _newvtkpolydata;
+
+       } // _StreamLinesIn
+
+}
+
+// ------------------------------------------------------------------------
+vtkPolyData* creaVtkStreamLineScalarSeg::GetStreamLinesOut()
+{
+       return _StreamLinesOut;
+}
+
+