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
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkStreamingImageFilterTest3.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 <iostream>
#include "itkImage.h"
#include "itkImageRegionIterator.h"
#include "itkImageFileReader.h"
#include "itkImageFileWriter.h"
#include "itkShiftScaleImageFilter.h"
#include "itkStreamingImageFilter.h"
#include "itkImageRegionMultidimensionalSplitter.h"
#include "../IO/itkPipelineMonitorImageFilter.h"
int itkStreamingImageFilterTest3(int argc, char*argv [] )
{
if( argc < 3 )
{
std::cerr << "Usage: " << std::endl;
std::cerr << argv[0] << " inputImageFile outputImageFile numberOfStreamDivisions" << std::endl;
return EXIT_FAILURE;
}
const std::string inputFilename = argv[1];
const std::string outputFilename = argv[2];
unsigned int numberOfStreamDivisions = atoi(argv[3]);
typedef unsigned char PixelType;
typedef itk::Image< PixelType, 2 > ImageType;
typedef itk::ImageFileReader<ImageType> ReaderType;
ReaderType::Pointer reader = ReaderType::New();
reader->SetFileName( inputFilename );
typedef itk::ShiftScaleImageFilter<ImageType, ImageType> SomeFilter;
SomeFilter::Pointer filter = SomeFilter::New();
filter->SetInput( reader->GetOutput() );
// monitor what's going on
itk::PipelineMonitorImageFilter<ImageType>::Pointer monitor;
monitor = itk::PipelineMonitorImageFilter<ImageType>::New();
monitor->SetInput( filter->GetOutput() );
itk::ImageRegionMultidimensionalSplitter<2>::Pointer splitter;
splitter = itk::ImageRegionMultidimensionalSplitter<2>::New();
itk::StreamingImageFilter<ImageType, ImageType>::Pointer streamer;
streamer = itk::StreamingImageFilter<ImageType, ImageType>::New();
streamer->SetInput( monitor->GetOutput() );
streamer->SetNumberOfStreamDivisions( numberOfStreamDivisions );
streamer->SetRegionSplitter( splitter );
typedef itk::ImageFileWriter<ImageType> WriterType;
WriterType::Pointer writer = WriterType::New();
writer->SetFileName( outputFilename );
writer->SetInput( streamer->GetOutput() );
writer->Update();
unsigned int expectedNumberOfStreams =
splitter->GetNumberOfSplits(streamer->GetOutput()->GetLargestPossibleRegion(), numberOfStreamDivisions);
std::cout << "ExpectedNumberOfStreams: " << expectedNumberOfStreams << std::endl;
if (!monitor->VerifyAllInputCanStream(expectedNumberOfStreams))
{
std::cout << "Filter failed to execute as expected!" << std::endl;
std::cout << monitor;
return EXIT_FAILURE;
}
return EXIT_SUCCESS;
}
|
