ITK  4.9.0
Insight Segmentation and Registration Toolkit
WikiExamples/SimpleOperations/CustomColormap.cxx
#include "itkImage.h"
#include "itkRGBPixel.h"
#include "QuickView.h"
typedef itk::RGBPixel<unsigned char> RGBPixelType;
typedef itk::Image<RGBPixelType, 2> RGBImageType;
typedef itk::Image<unsigned char, 2> ImageType;
ImageType::PixelType, RGBImageType::PixelType> ColormapType;
static void CreateRandomImage(
ImageType::Pointer image, unsigned int width = 200, unsigned height = 300);
static void CreateRandomColormap(
ColormapType::Pointer colormap, unsigned int size = 255);
int main(int, char *[])
{
ImageType::Pointer image = ImageType::New();
CreateRandomImage(image);
ColormapFilterType::Pointer colormapFilter1 =
ColormapFilterType::New();
ColormapType::Pointer largeColormap
= ColormapType::New();
CreateRandomColormap(largeColormap, 255);
colormapFilter1->SetInput (image);
colormapFilter1->SetColormap(largeColormap);
QuickView viewer;
viewer.AddRGBImage(
colormapFilter1->GetOutput(),
true);
viewer.Visualize();
return EXIT_SUCCESS;
}
void CreateRandomImage(ImageType::Pointer image,\
unsigned int width,
unsigned height)
{
// Create an image with 2 connected components
ImageType::RegionType region;
ImageType::IndexType start;
start[0] = 0;
start[1] = 0;
ImageType::SizeType size;
size[0] = width;
size[1] = height;
region.SetSize(size);
region.SetIndex(start);
image->SetRegions(region);
image->Allocate();
itk::ImageRegionIterator<ImageType> imageIterator(image,region);
unsigned int cnt = 0;
ImageType::PixelType value = random->GetUniformVariate(0.0, 255.0);
unsigned int bandSize = 5;
while(!imageIterator.IsAtEnd())
{
if (cnt % (size[0] * bandSize) == 0)
{
value = random->GetUniformVariate(0.0, 255.0);
bandSize = static_cast<unsigned int>(random->GetUniformVariate(5, 10));
}
imageIterator.Set(static_cast<ImageType::PixelType>(value));
++imageIterator;
++cnt;
}
}
void CreateRandomColormap(ColormapType::Pointer colormap, unsigned int size)
{
ColormapType::ChannelType redChannel;
ColormapType::ChannelType greenChannel;
ColormapType::ChannelType blueChannel;
random->SetSeed ( 8775070 );
for (unsigned int i = 0; i < size; ++i)
{
redChannel.push_back(static_cast<ColormapType::RealType>
(random->GetUniformVariate(.3, 1.0)));
greenChannel.push_back(static_cast<ColormapType::RealType>
(random->GetUniformVariate(.3, 1.0)));
blueChannel.push_back(static_cast<ColormapType::RealType>
(random->GetUniformVariate(.3, 1.0)));
}
colormap->SetRedChannel(redChannel);
colormap->SetGreenChannel(greenChannel);
colormap->SetBlueChannel(blueChannel);
}