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
95
96
97
98
99
100
101
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkVectorGradientMagnitudeImageFilterTest1.cxx
Language: C++
Date: $Date$
Version: $Revision$
Copyright (c) Insight Software Consortium. All rights reserved.
See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm 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.
=========================================================================*/
#if defined(_MSC_VER)
#pragma warning ( disable : 4786 )
#endif
#include <fstream>
#include "itkVectorGradientMagnitudeImageFilter.h"
#include "itkPNGImageIO.h"
#include "itkImageFileReader.h"
#include "itkImageFileWriter.h"
#include "itkImageRegionIterator.h"
#include "itkRescaleIntensityImageFilter.h"
#include "itkRGBPixel.h"
int itkVectorGradientMagnitudeImageFilterTest1(int ac, char* av[] )
{
typedef itk::RGBPixel<unsigned short> RGBPixelType;
typedef itk::Image<unsigned char, 2> CharImageType;
typedef itk::Image<RGBPixelType, 2> RGBImageType;
typedef itk::VectorGradientMagnitudeImageFilter<RGBImageType> FilterType;
typedef itk::ImageFileReader<RGBImageType> ReaderType;
typedef itk::RescaleIntensityImageFilter<FilterType::OutputImageType,
CharImageType> RescaleFilterType;
typedef itk::ImageFileWriter<CharImageType> WriterType;
if(ac < 4)
{
std::cerr << "Usage: " << av[0] << " InputImage OutputImage Mode\n";
return EXIT_FAILURE;
}
// Create a reader and filter
ReaderType::Pointer reader = ReaderType::New();
reader->SetFileName(av[1]);
FilterType::Pointer filter = FilterType::New();
filter->SetInput(reader->GetOutput());
int mode = ::atoi( av[3] );
if ( mode == 1)
{
filter->SetUsePrincipleComponentsOn();
}
else
{
filter->SetUsePrincipleComponentsOff();
}
RescaleFilterType::Pointer rescale = RescaleFilterType::New();
rescale->SetOutputMinimum(0);
rescale->SetOutputMaximum(255);
rescale->SetInput( filter->GetOutput() );
WriterType::Pointer writer = WriterType::New();
writer->SetInput( rescale->GetOutput() );
writer->SetFileName( av[2] );
try
{
writer->Update();
}
catch (itk::ExceptionObject& e)
{
std::cerr << "Exception detected: " << e.GetDescription();
return EXIT_FAILURE;
}
catch (...)
{
std::cerr << "Some other exception occurred" << std::endl;
return EXIT_FAILURE;
}
// Exercise the Print method
std::cout << "-- Test of the Print method --------------" << std::endl;
filter->Print( std::cout );
std::cout << "-- End of Print method test --------------" << std::endl;
std::cout << "The gradient image range was (low, high) = ("
<< rescale->GetInputMinimum() << ", " << rescale->GetInputMaximum()
<< ")" << std::endl;
std::cout << "Output was scaled, shifted = " << rescale->GetScale() << ", "
<< rescale->GetShift() << std::endl;
return EXIT_SUCCESS;
}
|
