1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
/*=========================================================================
*
* Copyright Insight Software Consortium
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0.txt
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*=========================================================================*/
#include "itkImageFileReader.h"
#include "itkImageFileWriter.h"
#include "itkSimpleFilterWatcher.h"
#include "itkLabelImageToLabelMapFilter.h"
#include "itkLabelMapMaskImageFilter.h"
int itkLabelMapMaskImageFilterTest(int argc, char * argv[])
{
if( argc != 9 )
{
std::cerr << "usage: " << argv[0] << " labelImage input output label bg neg crop cropBorder" << std::endl;
// std::cerr << " : " << std::endl;
exit(1);
}
// the filters are able to work in any dimension. Lets choose 3, so the program can be tested on
// 2D and 2D image.
const int dim = 3;
// declare the input image type
typedef itk::Image< unsigned char, dim > ImageType;
// and the label object type to use. The input image is a label image, so the
// type of the label can be the same type than the pixel type. itk::LabelObject is
// chosen, because only the mask feature is tested here, so we don't need any
// attribute.
typedef itk::LabelObject< unsigned char, dim > LabelObjectType;
typedef itk::LabelMap< LabelObjectType > LabelMapType;
// read the label image and the input image to be masked.
typedef itk::ImageFileReader< ImageType > ReaderType;
ReaderType::Pointer reader = ReaderType::New();
reader->SetFileName( argv[1] );
ReaderType::Pointer reader2 = ReaderType::New();
reader2->SetFileName( argv[2] );
// convert the label image to a label collection image.
typedef itk::LabelImageToLabelMapFilter< ImageType, LabelMapType> I2LType;
I2LType::Pointer i2l = I2LType::New();
i2l->SetInput( reader->GetOutput() );
// i2l->SetUseBackground( true );
// then mask the image. Two inputs are required (the label collection image, and
// the image to be masked). The label used to mask the image is passed with the
// SetLabel() method. The background in the output image, where the image is masked,
// is passed with SetBackground(). The user can choose to mask the image outside the
// label object (that's the default behavior), or inside the label object with the
// chosen label, by calling SetNegated(). Finally, the image can be cropped to the
// masked region, by calling SetCrop( true ), or to a region padded by a border, by
// calling both SetCrop() and SetCropBorder(). The crop border defaults to 0, and the
// image is not cropped by default.
typedef itk::LabelMapMaskImageFilter< LabelMapType, ImageType > MaskType;
MaskType::Pointer mask = MaskType::New();
mask->SetInput( i2l->GetOutput() );
mask->SetFeatureImage( reader2->GetOutput() );
mask->SetLabel( atoi(argv[4]) );
mask->SetBackgroundValue( atoi(argv[5]) );
mask->SetNegated( atoi(argv[6]) );
mask->SetCrop( atoi(argv[7]) );
MaskType::SizeType border;
border.Fill( atoi(argv[8]) );
mask->SetCropBorder( border );
itk::SimpleFilterWatcher watcher6(mask, "filter");
// Finally, save the output image.
typedef itk::ImageFileWriter< ImageType > WriterType;
WriterType::Pointer writer = WriterType::New();
writer->SetInput( mask->GetOutput() );
writer->SetFileName( argv[3] );
writer->Update();
return 0;
}
|
