ITK
5.1.0
Insight Toolkit
SphinxExamples/src/Filtering/Smoothing/MedianFilteringOfAnRGBImage/Code.cxx
/*=========================================================================
*
* Copyright NumFOCUS
*
* 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 "
itkMedianImageFilter.h
"
#include "
itkRGBPixel.h
"
#include "
itkImage.h
"
#include "
itkImageFileReader.h
"
#include "
itkImageFileWriter.h
"
namespace
itk
{
template
<
typename
TComponent =
unsigned
short
>
class
myRGBPixel :
public
RGBPixel<TComponent>
{
public
:
using
Self = myRGBPixel;
using
Superclass = RGBPixel<TComponent>;
using
RGBPixel<TComponent>::operator=;
bool
operator<=
(
const
Self & r)
const
{
return
(this->GetLuminance() <= r.GetLuminance());
}
bool
operator<
(
const
Self & r)
const
{
return
(this->GetLuminance() < r.GetLuminance());
}
};
}
// namespace itk
int
main(
int
argc,
char
* argv[])
{
// Verify command line arguments
if
(argc != 4)
{
std::cerr <<
"Usage: "
<< std::endl;
std::cerr << argv[0] <<
" <InputImageFile> <OutputImageFile> <radius>"
<< std::endl;
return
EXIT_FAILURE;
}
// Setup types
constexpr
unsigned
int
Dimension
= 2;
using
MyPixelType = itk::myRGBPixel<unsigned char>;
using
MyImageType =
itk::Image<MyPixelType, Dimension>
;
// Create and setup a reader
using
ReaderType =
itk::ImageFileReader<MyImageType>
;
ReaderType::Pointer reader = ReaderType::New();
reader->SetFileName(argv[1]);
// Create and setup a median filter
using
FilterType =
itk::MedianImageFilter<MyImageType, MyImageType>
;
FilterType::Pointer medianFilter = FilterType::New();
FilterType::InputSizeType radius;
radius.Fill(std::stoi(argv[3]));
medianFilter->SetRadius(radius);
medianFilter->SetInput(reader->GetOutput());
// Cast the custom myRBGPixel's to RGBPixel's
using
RGBPixelType =
itk::RGBPixel<unsigned char>
;
using
RGBImageType =
itk::Image<RGBPixelType, Dimension>
;
using
CastType =
itk::CastImageFilter<MyImageType, RGBImageType>
;
CastType::Pointer cast = CastType::New();
cast->SetInput(medianFilter->GetOutput());
using
WriterType =
itk::ImageFileWriter<RGBImageType>
;
WriterType::Pointer writer = WriterType::New();
writer->SetInput(cast->GetOutput());
writer->SetFileName(argv[2]);
try
{
writer->Update();
}
catch
(itk::ExceptionObject & error)
{
std::cerr <<
"Error: "
<< error << std::endl;
return
EXIT_FAILURE;
}
return
EXIT_SUCCESS;
}
itk::CastImageFilter
Casts input pixels to output pixel type.
Definition:
itkCastImageFilter.h:104
itk::RGBPixel
Represent Red, Green and Blue components for color images.
Definition:
itkRGBPixel.h:58
itk::operator<
bool operator<(const Index< VDimension > &one, const Index< VDimension > &two)
Definition:
itkIndex.h:546
itkRGBPixel.h
itk::operator<=
bool operator<=(const Index< VDimension > &one, const Index< VDimension > &two)
Definition:
itkIndex.h:560
itkMedianImageFilter.h
itkImageFileReader.h
itkImage.h
itk::ImageFileReader
Data source that reads image data from a single file.
Definition:
itkImageFileReader.h:75
itk::ImageFileWriter
Writes image data to a single file.
Definition:
itkImageFileWriter.h:87
itkImageFileWriter.h
itk
The "itk" namespace contains all Insight Segmentation and Registration Toolkit (ITK) classes....
Definition:
itkArray.h:26
itk::Image
Templated n-dimensional image class.
Definition:
itkImage.h:86
itk::MedianImageFilter
Applies a median filter to an image.
Definition:
itkMedianImageFilter.h:53
itk::GTest::TypedefsAndConstructors::Dimension2::Dimension
constexpr unsigned int Dimension
Definition:
itkGTestTypedefsAndConstructors.h:44
Generated on Sat Apr 25 2020 01:34:08 for ITK by
1.8.16