1 /*=========================================================================
2 Program: vv http://www.creatis.insa-lyon.fr/rio/vv
5 - University of LYON http://www.universite-lyon.fr/
6 - Léon Bérard cancer center http://oncora1.lyon.fnclcc.fr
7 - CREATIS CNRS laboratory http://www.creatis.insa-lyon.fr
9 This software is distributed WITHOUT ANY WARRANTY; without even
10 the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
11 PURPOSE. See the copyright notices for more information.
13 It is distributed under dual licence
15 - BSD See included LICENSE.txt file
16 - CeCILL-B http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
17 ======================================================================-====*/
20 #include "clitkSetBackgroundImageFilter.h"
23 #include <itkConnectedComponentImageFilter.h>
24 #include <itkRelabelComponentImageFilter.h>
25 #include <itkBinaryThresholdImageFilter.h>
26 #include <itkPasteImageFilter.h>
28 //--------------------------------------------------------------------
29 template<class ImageType>
30 void clitk::ComputeBBFromImageRegion(typename ImageType::Pointer image,
31 typename ImageType::RegionType region,
32 typename itk::BoundingBox<unsigned long,
33 ImageType::ImageDimension>::Pointer bb) {
34 typedef typename ImageType::IndexType IndexType;
37 for(unsigned int i=0; i<image->GetImageDimension(); i++) {
38 firstIndex[i] = region.GetIndex()[i];
39 lastIndex[i] = region.GetSize()[i];
42 typedef itk::BoundingBox<unsigned long,
43 ImageType::ImageDimension> BBType;
44 typedef typename BBType::PointType PointType;
47 image->TransformIndexToPhysicalPoint(firstIndex, firstPoint);
48 image->TransformIndexToPhysicalPoint(lastIndex, lastPoint);
50 bb->SetMaximum(lastPoint);
51 bb->SetMinimum(firstPoint);
53 //--------------------------------------------------------------------
56 //--------------------------------------------------------------------
57 template<int Dimension>
58 void clitk::ComputeBBIntersection(typename itk::BoundingBox<unsigned long, Dimension>::Pointer bbo,
59 typename itk::BoundingBox<unsigned long, Dimension>::Pointer bbi1,
60 typename itk::BoundingBox<unsigned long, Dimension>::Pointer bbi2) {
62 typedef itk::BoundingBox<unsigned long, Dimension> BBType;
63 typedef typename BBType::PointType PointType;
67 for(unsigned int i=0; i<Dimension; i++) {
68 firstPoint[i] = std::max(bbi1->GetMinimum()[i],
69 bbi2->GetMinimum()[i]);
70 lastPoint[i] = std::min(bbi1->GetMaximum()[i],
71 bbi2->GetMaximum()[i]);
74 bbo->SetMaximum(lastPoint);
75 bbo->SetMinimum(firstPoint);
77 //--------------------------------------------------------------------
80 //--------------------------------------------------------------------
81 template<class ImageType>
82 void clitk::ComputeRegionFromBB(typename ImageType::Pointer image,
83 const typename itk::BoundingBox<unsigned long,
84 ImageType::ImageDimension>::Pointer bb,
85 typename ImageType::RegionType & region) {
87 typedef typename ImageType::IndexType IndexType;
88 typedef typename ImageType::PointType PointType;
89 typedef typename ImageType::RegionType RegionType;
90 typedef typename ImageType::SizeType SizeType;
92 // Region starting point
93 IndexType regionStart;
94 PointType start = bb->GetMinimum();
95 image->TransformPhysicalPointToIndex(start, regionStart);
99 PointType maxs = bb->GetMaximum();
100 PointType mins = bb->GetMinimum();
101 for(unsigned int i=0; i<ImageType::ImageDimension; i++) {
102 regionSize[i] = floor((maxs[i] - mins[i])/image->GetSpacing()[i]);
106 region.SetIndex(regionStart);
107 region.SetSize(regionSize);
109 //--------------------------------------------------------------------
111 //--------------------------------------------------------------------
112 template<class ImageType, class TMaskImageType>
113 typename ImageType::Pointer
114 clitk::SetBackground(typename ImageType::ConstPointer input,
115 typename TMaskImageType::ConstPointer mask,
116 typename TMaskImageType::PixelType maskBG,
117 typename ImageType::PixelType outValue) {
118 typedef clitk::SetBackgroundImageFilter<ImageType, TMaskImageType, ImageType> SetBackgroundImageFilterType;
119 typename SetBackgroundImageFilterType::Pointer setBackgroundFilter = SetBackgroundImageFilterType::New();
120 setBackgroundFilter->SetInput(input);
121 setBackgroundFilter->SetInput2(mask);
122 setBackgroundFilter->SetMaskValue(maskBG);
123 setBackgroundFilter->SetOutsideValue(outValue);
124 setBackgroundFilter->Update();
125 return setBackgroundFilter->GetOutput();
127 //--------------------------------------------------------------------
130 //--------------------------------------------------------------------
131 template<class ImageType>
132 typename ImageType::Pointer
133 clitk::Labelize(typename ImageType::Pointer input,
134 typename ImageType::PixelType BG,
135 bool isFullyConnected,
136 int minimalComponentSize) {
137 // Connected Component label
138 typedef itk::ConnectedComponentImageFilter<ImageType, ImageType> ConnectFilterType;
139 typename ConnectFilterType::Pointer connectFilter = ConnectFilterType::New();
140 connectFilter->SetInput(input);
141 connectFilter->SetBackgroundValue(BG);
142 connectFilter->SetFullyConnected(isFullyConnected);
144 // Sort by size and remove too small area.
145 typedef itk::RelabelComponentImageFilter<ImageType, ImageType> RelabelFilterType;
146 typename RelabelFilterType::Pointer relabelFilter = RelabelFilterType::New();
147 relabelFilter->InPlaceOn();
148 relabelFilter->SetInput(connectFilter->GetOutput());
149 relabelFilter->SetMinimumObjectSize(minimalComponentSize);
150 relabelFilter->Update();
153 return relabelFilter->GetOutput();
155 //--------------------------------------------------------------------
158 //--------------------------------------------------------------------
159 template<class ImageType>
160 typename ImageType::Pointer
161 clitk::RemoveLabels(typename ImageType::Pointer input,
162 typename ImageType::PixelType BG,
163 std::vector<typename ImageType::PixelType> & labelsToRemove) {
164 typename ImageType::Pointer working_image = input;
165 for (unsigned int i=0; i <labelsToRemove.size(); i++) {
166 typedef clitk::SetBackgroundImageFilter<ImageType, ImageType> SetBackgroundImageFilterType;
167 typename SetBackgroundImageFilterType::Pointer setBackgroundFilter = SetBackgroundImageFilterType::New();
168 setBackgroundFilter->SetInput(input);
169 setBackgroundFilter->SetInput2(input);
170 setBackgroundFilter->SetMaskValue(labelsToRemove[i]);
171 setBackgroundFilter->SetOutsideValue(BG);
172 setBackgroundFilter->Update();
173 working_image = setBackgroundFilter->GetOutput();
175 return working_image;
177 //--------------------------------------------------------------------
180 //--------------------------------------------------------------------
181 template<class ImageType>
182 typename ImageType::Pointer
183 clitk::KeepLabels(typename ImageType::Pointer input,
184 typename ImageType::PixelType BG,
185 typename ImageType::PixelType FG,
186 typename ImageType::PixelType firstKeep,
187 typename ImageType::PixelType lastKeep,
189 typedef itk::BinaryThresholdImageFilter<ImageType, ImageType> BinarizeFilterType;
190 typename BinarizeFilterType::Pointer binarizeFilter = BinarizeFilterType::New();
191 binarizeFilter->SetInput(input);
192 binarizeFilter->SetLowerThreshold(firstKeep);
193 if (useLastKeep) binarizeFilter->SetUpperThreshold(lastKeep);
194 binarizeFilter->SetInsideValue(FG);
195 binarizeFilter->SetOutsideValue(BG);
196 binarizeFilter->Update();
197 return binarizeFilter->GetOutput();
199 //--------------------------------------------------------------------
202 //--------------------------------------------------------------------
203 template<class ImageType>
204 typename ImageType::Pointer
205 clitk::LabelizeAndSelectLabels(typename ImageType::Pointer input,
206 typename ImageType::PixelType BG,
207 typename ImageType::PixelType FG,
208 bool isFullyConnected,
209 int minimalComponentSize,
210 LabelizeParameters<typename ImageType::PixelType> * param)
212 typename ImageType::Pointer working_image;
213 working_image = Labelize<ImageType>(input, BG, isFullyConnected, minimalComponentSize);
214 working_image = RemoveLabels<ImageType>(working_image, BG, param->GetLabelsToRemove());
215 working_image = KeepLabels<ImageType>(working_image,
217 param->GetFirstKeep(),
218 param->GetLastKeep(),
219 param->GetUseLastKeep());
220 return working_image;
222 //--------------------------------------------------------------------
225 //--------------------------------------------------------------------
226 template<class ImageType>
227 typename ImageType::Pointer
228 clitk::EnlargeImageLike(typename ImageType::Pointer input,
229 typename ImageType::Pointer like,
230 typename ImageType::PixelType backgroundValue)
232 if (!HaveSameSpacing<ImageType, ImageType>(input, like)) {
233 FATAL("Images must have the same spacing");
236 typename ImageType::Pointer output = ImageType::New();
237 typename ImageType::SizeType size;
238 for(unsigned int i=0; i<ImageType::ImageDimension; i++) {
239 size[i] = lrint((like->GetLargestPossibleRegion().GetSize()[i]*like->GetSpacing()[i])/
240 (double)like->GetSpacing()[i]);
243 typename ImageType::RegionType region;
244 region.SetSize(size);
245 output->SetRegions(region);
246 output->SetSpacing(like->GetSpacing());
247 output->SetOrigin(like->GetOrigin());
249 output->FillBuffer(backgroundValue);
250 typedef itk::PasteImageFilter<ImageType,ImageType> PasteFilterType;
251 typename PasteFilterType::Pointer pasteFilter = PasteFilterType::New();
252 typename PasteFilterType::InputImageIndexType index;
253 for(unsigned int i=0; i<ImageType::ImageDimension; i++) {
254 index[i] = lrint((input->GetOrigin()[i] - like->GetOrigin()[i])/(double)input->GetSpacing()[i]);
257 pasteFilter->SetSourceImage(input);
258 pasteFilter->SetDestinationImage(output);
259 pasteFilter->SetDestinationIndex(index);
260 pasteFilter->SetSourceRegion(input->GetLargestPossibleRegion());
261 pasteFilter->Update();
262 return pasteFilter->GetOutput();
264 //--------------------------------------------------------------------