]> Creatis software - bbtk.git/blobdiff - packages/vtk/src/bbvtkSegmentationConnectivity.cxx
#3095 BBTK Feature New Normal - Box Segmentation Connectivity add multiple input...
[bbtk.git] / packages / vtk / src / bbvtkSegmentationConnectivity.cxx
index ec416f158e30410f1e644ac0fc8dbc771c0d439c..3b28389b314c7ba907561d6707610220205a9f43 100644 (file)
@@ -1,42 +1,66 @@
+/*
+ # ---------------------------------------------------------------------
+ #
+ # 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.
+ # ------------------------------------------------------------------------ */
+
+
 /*=========================================================================
-                                                                                
   Program:   bbtk
   Module:    $RCSfile: bbvtkSegmentationConnectivity.cxx,v $
   Language:  C++
-  Date:      $Date: 2008/04/24 12:56:41 $
-  Version:   $Revision: 1.1 $
-                                                                                
-  Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
-  l'Image). All rights reserved. See Doc/License.txt or
-  http://www.creatis.insa-lyon.fr/Public/bbtk/License.html for details.
-                                                                                
-     This software is distributed WITHOUT ANY WARRANTY; without even
-     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
-     PURPOSE.  See the above copyright notices for more information.
-                                                                                
+  Date:      $Date: 2012/11/16 08:51:58 $
+  Version:   $Revision: 1.8 $
 =========================================================================*/
+
 /**
  *  \file 
  *  \brief 
  */
 
-
 #ifdef _USE_VTK_
 
-
 #include "bbvtkSegmentationConnectivity.h"
 #include "bbvtkPackage.h"
 
+#include "bbtkStaticLecture.h"
+
 namespace bbvtk
 {
-
-       BBTK_ADD_BLACK_BOX_TO_PACKAGE(vtk,SegmentationConnectivity)
-       BBTK_USER_BLACK_BOX_IMPLEMENTATION(SegmentationConnectivity,bbtk::AtomicBlackBox);
-
-
-       void SegmentationConnectivity::bbUserConstructor() 
-   { 
-      bbSetInputIn(NULL);
+  BBTK_ADD_BLACK_BOX_TO_PACKAGE(vtk,SegmentationConnectivity);
+  BBTK_BLACK_BOX_IMPLEMENTATION(SegmentationConnectivity,
+                               bbtk::AtomicBlackBox);
+
+// --------------------------------------------------------------
+
+  void SegmentationConnectivity::bbUserSetDefaultValues() 
+  { 
+         thresh2       = NULL;
+//       cast2         = NULL;
+         connect2      = NULL;
+         cast4         = NULL;
+               
+         bbSetInputIn(NULL);
          std::vector<int> position;
          position.push_back(0);
          position.push_back(0);
@@ -47,47 +71,137 @@ namespace bbvtk
          threshold.push_back(0);
          bbSetInputThresholdMinMax(threshold);
          bbSetOutputOut(NULL);
-         thresh2 = vtkImageThreshold::New();
-         thresh2->SetInValue(255);
-         thresh2->SetOutputScalarTypeToUnsignedShort();
-         thresh2->SetOutValue(0);
-         cast2 = vtkImageCast::New();
-         cast2->SetInput(thresh2->GetOutput());
-         cast2->SetOutputScalarTypeToUnsignedChar();
-         connect2 = vtkImageSeedConnectivity::New();
-         connect2->SetInput(cast2->GetOutput());
-         connect2->SetInputConnectValue(255);
-         connect2->SetOutputConnectedValue(255);
-         connect2->SetOutputUnconnectedValue(0);
-         cast4 = vtkImageCast::New();
-         cast4->SetInput(connect2->GetOutput());
-         cast4->SetOutputScalarTypeToUnsignedShort();
-   }
+
+         bbSetInputOutputFormat("SAME");
+         bbSetInputValue(255);
 
 
-  void SegmentationConnectivity::DoProcess()
-  {
-       vtkImageData *imagedata = bbGetInputIn();
-       imagedata->UpdateInformation();
-       imagedata->SetUpdateExtent(imagedata->GetWholeExtent());
-       imagedata->Update();
-       thresh2->ThresholdBetween(3000, 3001);
-       thresh2->SetInput(imagedata);
-       thresh2->ThresholdBetween(bbGetInputThresholdMinMax()[0], bbGetInputThresholdMinMax()[1]);
-       thresh2->Update();
-       cast2->Update();
-       connect2->RemoveAllSeeds ();
-       connect2->AddSeed( bbGetInputPositionXYZ()[0] , bbGetInputPositionXYZ()[1] , bbGetInputPositionXYZ()[2] );
-       connect2->Update();
-       cast4->Update();
-
-    bbSetOutputOut(cast4->GetOutput() );         
+  }
+
+// --------------------------------------------------------------
+               
+       void SegmentationConnectivity::bbUserInitializeProcessing()
+       {
+
+               thresh2 = vtkImageThreshold::New();
+                       thresh2->SetInValue(255);
+                       thresh2->SetOutputScalarTypeToUnsignedChar();
+               connect2 = vtkImageSeedConnectivity::New();
+                       connect2->SetInput(thresh2->GetOutput());
+                       connect2->SetInputConnectValue(255);
+                       connect2->SetOutputConnectedValue( 255 );
+                       connect2->SetOutputUnconnectedValue(0);
+               cast4 = vtkImageCast::New();
+                       cast4->SetInput(connect2->GetOutput());
+
+// EED 2016 06 15
+
+/*
+               thresh2 = vtkImageThreshold::New();
+                       thresh2->SetInValue(255);
+                       thresh2->SetOutputScalarTypeToUnsignedShort();
+                       thresh2->SetOutValue(0);
+               cast2 = vtkImageCast::New();
+                       cast2->SetInput(thresh2->GetOutput());
+                       cast2->SetOutputScalarTypeToUnsignedChar();
+               connect2 = vtkImageSeedConnectivity::New();
+                       connect2->SetInput(cast2->GetOutput());
+                       connect2->SetInputConnectValue(255);
+                       connect2->SetOutputConnectedValue(255);
+                       connect2->SetOutputUnconnectedValue(0);
+               cast4 = vtkImageCast::New();
+                       cast4->SetInput(connect2->GetOutput());
+                       cast4->SetOutputScalarTypeToUnsignedShort();
+*/
+
+       }
+
+// --------------------------------------------------------------
+               
+  void SegmentationConnectivity::bbUserFinalizeProcessing() 
+  { 
+         if (thresh2!=NULL)
+         {
+                 thresh2->Delete();
+                 thresh2=NULL;
+         }
+//       if (cast2!=NULL)
+//       {
+//               cast2->Delete();
+//               cast2=NULL;
+//       }
+         if (connect2!=NULL)
+         {
+                 connect2->Delete();
+                 connect2=NULL;
+         }
+         if (cast4!=NULL)
+         {
+                 cast4->Delete();
+                 cast4=NULL;
+         }  
   }
 
 
-}// EO namespace bbvtk
 
 
+// --------------------------------------------------------------
+               
+  void SegmentationConnectivity::DoProcess()
+  {
+         
+       int x,y,z;
+       int i,numPoints = bbGetInputPositionXYZ().size()/3;
+       int verif = (bbGetInputPositionXYZ().size()-1) % 3;
+       if (( numPoints>=1) && (verif==2) && (bbGetInputIn()!=NULL)) 
+       {
+               int outputformat = bbtkStaticLecture::GetTypeFormat( bbGetInputOutputFormat() , bbGetInputIn() );
+               vtkImageData *imagedata = bbGetInputIn();
+               imagedata->UpdateInformation();
+               imagedata->SetUpdateExtent(imagedata->GetWholeExtent());
+               imagedata->Update();
+               thresh2->ThresholdBetween(3000, 3001);
+               thresh2->SetInput(imagedata);
+               thresh2->ThresholdBetween(bbGetInputThresholdMinMax()[0], bbGetInputThresholdMinMax()[1]);
+               thresh2->Modified();
+               thresh2->Update();
+               connect2->RemoveAllSeeds ();
+
+               int ext[6];
+               imagedata->GetWholeExtent(ext);
+               int maxX = ext[1]-ext[0]+1;
+               int maxY = ext[3]-ext[2]+1;
+               int maxZ = ext[5]-ext[4]+1;
+
+               for (i=0;i<numPoints;i++)
+               {
+                       x=bbGetInputPositionXYZ()[i*3+0];
+                       y=bbGetInputPositionXYZ()[i*3+1];
+                       z=bbGetInputPositionXYZ()[i*3+2];
+                       if ( (x>=0) || (x<maxX) || (y>=0) || (y<maxY) || (z>=0) || (z<maxZ) )
+                       {
+                               connect2->AddSeed( x,y,z );
+                       } else {
+                               printf("EED WARNNIG! In box: SegmentationConnectivity point out of range \n");
+                       } // if point inside image 
+
+               } // for i
+               connect2->SetOutputConnectedValue( bbGetInputValue() );
+               cast4->SetOutputScalarType( outputformat );
+               connect2->Modified();
+               connect2->Update();
+               cast4->Update();                        
+               bbSetOutputOut(cast4->GetOutput() );
+       }  else {
+               printf("EED WARNNIG! In box: SegmentationConnectivity List of points are not coherent. \n");
+               bbSetOutputOut( NULL );
+       }// if numPoints verif
+
+}
+
+  
+}// EO namespace bbvtk
+
 #endif //_USE_VTK_