2 # ---------------------------------------------------------------------
4 # Copyright (c) CREATIS (Centre de Recherche en Acquisition et Traitement de l'Image
6 # Authors : Eduardo Davila, Frederic Cervenansky, Claire Mouton
7 # Previous Authors : Laurent Guigues, Jean-Pierre Roux
8 # CreaTools website : www.creatis.insa-lyon.fr/site/fr/creatools_accueil
10 # This software is governed by the CeCILL-B license under French law and
11 # abiding by the rules of distribution of free software. You can use,
12 # modify and/ or redistribute the software under the terms of the CeCILL-B
13 # license as circulated by CEA, CNRS and INRIA at the following URL
14 # http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
15 # or in the file LICENSE.txt.
17 # As a counterpart to the access to the source code and rights to copy,
18 # modify and redistribute granted by the license, users are provided only
19 # with a limited warranty and the software's author, the holder of the
20 # economic rights, and the successive licensors have only limited
23 # The fact that you are presently reading this means that you have had
24 # knowledge of the CeCILL-B license and that you accept its terms.
25 # ------------------------------------------------------------------------ */
28 /*=========================================================================
30 Module: $RCSfile: bbvtkSegmentationConnectivity.cxx,v $
32 Date: $Date: 2012/11/16 08:51:58 $
33 Version: $Revision: 1.8 $
34 =========================================================================*/
43 #include "bbvtkSegmentationConnectivity.h"
44 #include "bbvtkPackage.h"
46 #include "bbtkStaticLecture.h"
50 BBTK_ADD_BLACK_BOX_TO_PACKAGE(vtk,SegmentationConnectivity);
51 BBTK_BLACK_BOX_IMPLEMENTATION(SegmentationConnectivity,
52 bbtk::AtomicBlackBox);
54 // --------------------------------------------------------------
56 void SegmentationConnectivity::bbUserSetDefaultValues()
64 std::vector<int> position;
65 position.push_back(0);
66 position.push_back(0);
67 position.push_back(0);
68 bbSetInputPositionXYZ(position);
69 std::vector<int> threshold;
70 threshold.push_back(0);
71 threshold.push_back(0);
72 bbSetInputThresholdMinMax(threshold);
75 bbSetInputOutputFormat("SAME");
81 // --------------------------------------------------------------
83 void SegmentationConnectivity::bbUserInitializeProcessing()
86 thresh2 = vtkImageThreshold::New();
87 thresh2->SetInValue(255);
88 thresh2->SetOutputScalarTypeToUnsignedChar();
89 connect2 = vtkImageSeedConnectivity::New();
90 connect2->SetInput(thresh2->GetOutput());
91 connect2->SetInputConnectValue(255);
92 connect2->SetOutputConnectedValue( 255 );
93 connect2->SetOutputUnconnectedValue(0);
94 cast4 = vtkImageCast::New();
95 cast4->SetInput(connect2->GetOutput());
100 thresh2 = vtkImageThreshold::New();
101 thresh2->SetInValue(255);
102 thresh2->SetOutputScalarTypeToUnsignedShort();
103 thresh2->SetOutValue(0);
104 cast2 = vtkImageCast::New();
105 cast2->SetInput(thresh2->GetOutput());
106 cast2->SetOutputScalarTypeToUnsignedChar();
107 connect2 = vtkImageSeedConnectivity::New();
108 connect2->SetInput(cast2->GetOutput());
109 connect2->SetInputConnectValue(255);
110 connect2->SetOutputConnectedValue(255);
111 connect2->SetOutputUnconnectedValue(0);
112 cast4 = vtkImageCast::New();
113 cast4->SetInput(connect2->GetOutput());
114 cast4->SetOutputScalarTypeToUnsignedShort();
119 // --------------------------------------------------------------
121 void SegmentationConnectivity::bbUserFinalizeProcessing()
148 // --------------------------------------------------------------
150 void SegmentationConnectivity::DoProcess()
154 int i,numPoints = bbGetInputPositionXYZ().size()/3;
155 int verif = (bbGetInputPositionXYZ().size()-1) % 3;
156 if (( numPoints>=1) && (verif==2) && (bbGetInputIn()!=NULL))
158 int outputformat = bbtkStaticLecture::GetTypeFormat( bbGetInputOutputFormat() , bbGetInputIn() );
159 vtkImageData *imagedata = bbGetInputIn();
160 imagedata->UpdateInformation();
161 imagedata->SetUpdateExtent(imagedata->GetWholeExtent());
163 thresh2->ThresholdBetween(3000, 3001);
164 thresh2->SetInput(imagedata);
165 thresh2->ThresholdBetween(bbGetInputThresholdMinMax()[0], bbGetInputThresholdMinMax()[1]);
168 connect2->RemoveAllSeeds ();
171 imagedata->GetWholeExtent(ext);
172 int maxX = ext[1]-ext[0]+1;
173 int maxY = ext[3]-ext[2]+1;
174 int maxZ = ext[5]-ext[4]+1;
176 for (i=0;i<numPoints;i++)
178 x=bbGetInputPositionXYZ()[i*3+0];
179 y=bbGetInputPositionXYZ()[i*3+1];
180 z=bbGetInputPositionXYZ()[i*3+2];
181 if ( (x>=0) || (x<maxX) || (y>=0) || (y<maxY) || (z>=0) || (z<maxZ) )
183 connect2->AddSeed( x,y,z );
185 printf("EED WARNNIG! In box: SegmentationConnectivity point out of range \n");
186 } // if point inside image
189 connect2->SetOutputConnectedValue( bbGetInputValue() );
190 cast4->SetOutputScalarType( outputformat );
191 connect2->Modified();
194 bbSetOutputOut(cast4->GetOutput() );
196 printf("EED WARNNIG! In box: SegmentationConnectivity List of points are not coherent. \n");
197 bbSetOutputOut( NULL );
198 }// if numPoints verif
203 }// EO namespace bbvtk