|
|
| (One intermediate revision by one other user not shown) |
| Line 1: |
Line 1: |
| ==ParaviewColormap.cxx== | | {{warning|1=The media wiki content on this page is no longer maintained. The examples presented on the https://itk.org/Wiki/* pages likely require ITK version 4.13 or earlier releases. In many cases, the examples on this page no longer conform to the best practices for modern ITK versions. |
| <source lang="cpp">
| | }} |
| #include "itkCustomColormapFunction.h"
| |
| #include "itkImage.h"
| |
| #include "itkImageRegionConstIterator.h"
| |
| #include "itkScalarToRGBColormapImageFilter.h"
| |
| #include "itkRescaleIntensityImageFilter.h"
| |
| #include "itkImageFileWriter.h"
| |
| #include "itkRGBPixel.h"
| |
|
| |
|
| typedef itk::RGBPixel<unsigned char> RGBPixelType;
| | [https://itk.org/ITKExamples[ITK Sphinx Examples]] |
| typedef itk::Image<RGBPixelType, 2> RGBImageType;
| |
| | |
| typedef itk::Image<float, 2> FloatImageType;
| |
| typedef itk::Image<unsigned char, 2> UnsignedCharImageType;
| |
| | |
| int main( int argc, char *argv[])
| |
| {
| |
| FloatImageType::Pointer image = FloatImageType::New();
| |
| | |
| itk::Index<2> start;
| |
| start.Fill(0);
| |
| | |
| itk::Size<2> size = {{100,20}};
| |
| | |
| itk::ImageRegion<2> region(start, size);
| |
| | |
| image->SetRegions(region);
| |
| image->Allocate();
| |
| | |
| for(unsigned int i = 0; i < size[0]; i++)
| |
| {
| |
| for(unsigned int j = 0; j < size[1]; j++)
| |
| {
| |
| itk::Index<2> pixel;
| |
| pixel[0] = i;
| |
| pixel[1] = j;
| |
| image->SetPixel(pixel, i);
| |
| }
| |
| }
| |
| | |
| // Custom colormap
| |
| //typedef itk::Function::CustomColormapFunction<UnsignedCharImageType::PixelType, RGBImageType::PixelType> ColormapType;
| |
| typedef itk::Function::CustomColormapFunction<FloatImageType::PixelType, RGBImageType::PixelType> ColormapType;
| |
| ColormapType::ChannelType redChannel;
| |
| ColormapType::ChannelType greenChannel;
| |
| ColormapType::ChannelType blueChannel;
| |
|
| |
| redChannel.push_back(static_cast<ColormapType::RealType>(0.231373));
| |
| greenChannel.push_back(static_cast<ColormapType::RealType>(0.298039));
| |
| blueChannel.push_back(static_cast<ColormapType::RealType>(0.752941));
| |
| | |
| redChannel.push_back(static_cast<ColormapType::RealType>(1));
| |
| greenChannel.push_back(static_cast<ColormapType::RealType>(1));
| |
| blueChannel.push_back(static_cast<ColormapType::RealType>(1));
| |
|
| |
| redChannel.push_back(static_cast<ColormapType::RealType>(0.705882));
| |
| greenChannel.push_back(static_cast<ColormapType::RealType>(0.0156863));
| |
| blueChannel.push_back(static_cast<ColormapType::RealType>(0.14902));
| |
|
| |
| ColormapType::Pointer colormap = ColormapType::New();
| |
| colormap->SetRedChannel(redChannel);
| |
| colormap->SetGreenChannel(greenChannel);
| |
| colormap->SetBlueChannel(blueChannel);
| |
| | |
| // Setup conversion
| |
| typedef itk::ScalarToRGBColormapImageFilter<FloatImageType, RGBImageType> RGBFilterType;
| |
| RGBFilterType::Pointer rgbfilter = RGBFilterType::New();
| |
| rgbfilter->SetInput(image);
| |
|
| |
| rgbfilter->SetColormap(colormap);
| |
|
| |
| // Built-in color map
| |
| // rgbfilter->SetColormap( RGBFilterType::Hot );
| |
| | |
| {
| |
| typedef itk::ImageFileWriter< RGBImageType > WriterType;
| |
| WriterType::Pointer writer = WriterType::New();
| |
| writer->SetFileName("colormap.png");
| |
| writer->SetInput(rgbfilter->GetOutput());
| |
| writer->Update();
| |
| }
| |
| | |
| return EXIT_SUCCESS;
| |
| }
| |
| | |
| </source>
| |
| | |
| {{ITKVTKCMakeLists|CustomColormap|}}
| |
