ITK  5.2.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 "itkRGBPixel.h"
#include "itkImage.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>;
// Create and setup a reader
ReaderType::Pointer reader = ReaderType::New();
reader->SetFileName(argv[1]);
// Create and setup a median filter
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>;
CastType::Pointer cast = CastType::New();
cast->SetInput(medianFilter->GetOutput());
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: itkAnnulusOperator.h:24
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