void TestExtractChannels()
{
typedef itk::VectorImage<float, 2> VectorImageType;
VectorImageType::Pointer image = VectorImageType::New();
itk::Index<2> corner = {{0,0}};
itk::Size<2> size = {{100,100}};
itk::ImageRegion<2> region(corner, size);
image->SetRegions(region);
image->SetNumberOfComponentsPerPixel(3);
image->Allocate();
// Extract the first two channels
std::vector<unsigned int> channels;
channels.push_back(0);
channels.push_back(1);
typedef itk::VectorImage<float, 2> FloatScalarImageType;
FloatScalarImageType::Pointer floatScalarImage = FloatScalarImageType::New();
ITKHelpers::ExtractChannels(image.GetPointer(), channels, floatScalarImage.GetPointer());
typedef itk::VectorImage<unsigned char, 2> UnsignedCharScalarImageType;
UnsignedCharScalarImageType::Pointer unsignedCharScalarImage = UnsignedCharScalarImageType::New();
ITKHelpers::ExtractChannels(image.GetPointer(), channels, unsignedCharScalarImage.GetPointer());
}
int main(int argc, char*argv[])
{
Mask::Pointer mask = Mask::New();
CreateMask(mask);
OutputHelpers::WriteImage(mask.GetPointer(), "mask.png");
UnsignedCharScalarImageType::Pointer image = UnsignedCharScalarImageType::New();
CreateImage(image);
OutputHelpers::WriteImage(image.GetPointer(), "image.png");
FloatScalarImageType::Pointer output = FloatScalarImageType::New();
MaskOperations::MaskedLaplacian(image.GetPointer(), mask.GetPointer(), output.GetPointer());
OutputHelpers::WriteImage(output.GetPointer(), "laplacian.mha");
return EXIT_SUCCESS;
}
void ComputeColorIsophotesInRegion(const TVectorImageType* const image, const Mask* const mask,
const itk::ImageRegion<2>& region , TIsophoteImageType* const isophotes)
{
/*
* 'isophotes' must already be initialized to the right size and allocated.
*/
//EnterFunction("ComputeIsophotes()");
RGBImageType::Pointer rgbImage = RGBImageType::New();
ITKHelpers::VectorImageToRGBImage(image, rgbImage);
//HelpersOutput::WriteImageConditional<RGBImageType>(rgbImage, "Debug/Initialize.rgb.mha", this->DebugImages);
typedef itk::RGBToLuminanceImageFilter< RGBImageType, FloatScalarImageType > LuminanceFilterType;
LuminanceFilterType::Pointer luminanceFilter = LuminanceFilterType::New();
luminanceFilter->SetInput(rgbImage);
luminanceFilter->Update();
FloatScalarImageType::Pointer luminanceImage = FloatScalarImageType::New();
ITKHelpers::DeepCopy(luminanceFilter->GetOutput(), luminanceImage.GetPointer());
FloatScalarImageType::Pointer blurredLuminance = FloatScalarImageType::New();
// Blur with a Gaussian kernel. From TestIsophotes.cpp, it actually seems like not blurring,
// but using a masked sobel operator produces the most reliable isophotes.
unsigned int kernelRadius = 0;
MaskOperations::MaskedBlur(luminanceFilter->GetOutput(), mask,
kernelRadius, blurredLuminance.GetPointer());
//HelpersOutput::WriteImageConditional(blurredLuminance,
// "Debug/Initialize.blurredLuminance.mha", true);
//ITKHelpers::InitializeImage(isophotes, image->GetLargestPossibleRegion());
Isophotes::ComputeMaskedIsophotesInRegion(blurredLuminance.GetPointer(), mask, region, isophotes);
// if(this->DebugImages)
// {
// HelpersOutput::Write2DVectorImage(this->IsophoteImage, "Debug/Initialize.IsophoteImage.mha");
// }
//LeaveFunction("ComputeIsophotes()");
}
void TestExtractChannel()
{
typedef itk::VectorImage<float, 2> VectorImageType;
VectorImageType::Pointer image = VectorImageType::New();
itk::Index<2> corner = {{0,0}};
itk::Size<2> size = {{100,100}};
itk::ImageRegion<2> region(corner, size);
image->SetRegions(region);
image->SetNumberOfComponentsPerPixel(2);
image->Allocate();
typedef itk::Image<float, 2> FloatScalarImageType;
FloatScalarImageType::Pointer floatScalarImage = FloatScalarImageType::New();
ITKHelpers::ExtractChannel(image.GetPointer(), 0, floatScalarImage.GetPointer());
typedef itk::Image<unsigned char, 2> UnsignedCharScalarImageType;
UnsignedCharScalarImageType::Pointer unsignedCharScalarImage = UnsignedCharScalarImageType::New();
ITKHelpers::ExtractChannel(image.GetPointer(), 0, unsignedCharScalarImage.GetPointer());
}
int main(int argc, char *argv[])
{
if(argc != 3)
{
std::cerr << "Required arguments: image mask" << std::endl;
return EXIT_FAILURE;
}
std::string imageFilename = argv[1];
std::string maskFilename = argv[2];
std::cout << "Reading image: " << imageFilename << std::endl;
std::cout << "Reading mask: " << maskFilename << std::endl;
typedef itk::ImageFileReader<FloatVectorImageType> ImageReaderType;
ImageReaderType::Pointer imageReader = ImageReaderType::New();
imageReader->SetFileName(imageFilename.c_str());
imageReader->Update();
std::cout << "Read image " << imageReader->GetOutput()->GetLargestPossibleRegion() << std::endl;
typedef itk::ImageFileReader<Mask> MaskReaderType;
MaskReaderType::Pointer maskReader = MaskReaderType::New();
maskReader->SetFileName(maskFilename.c_str());
maskReader->Update();
std::cout << "Read mask " << maskReader->GetOutput()->GetLargestPossibleRegion() << std::endl;
// Prepare image
RGBImageType::Pointer rgbImage = RGBImageType::New();
// Helpers::VectorImageToRGBImage(imageReader->GetOutput(), rgbImage);
// TODO: Update this call to new API
//maskReader->GetOutput()->ApplyToImage(rgbImage.GetPointer(), Qt::black);
OutputHelpers::WriteImage<RGBImageType>(rgbImage, "Test/TestIsophotes.rgb.mha");
typedef itk::RGBToLuminanceImageFilter< RGBImageType, FloatScalarImageType > LuminanceFilterType;
LuminanceFilterType::Pointer luminanceFilter = LuminanceFilterType::New();
luminanceFilter->SetInput(rgbImage);
luminanceFilter->Update();
OutputHelpers::WriteImage<FloatScalarImageType>(luminanceFilter->GetOutput(), "Test/Luminance.mha");
// PatchBasedInpainting inpainting;
// inpainting.SetDebugImages(true);
// inpainting.SetMask(maskReader->GetOutput());
// inpainting.SetImage(imageReader->GetOutput());
//Helpers::Write2DVectorImage(inpainting.GetIsophoteImage(), "Test/TestIsophotes.isophotes.mha");
//inpainting.FindBoundary();
// After blurVariance == 4, you cannot tell the difference in the output.
for(unsigned int blurVariance = 0; blurVariance < 5; ++blurVariance)
{
std::string fileNumber = Helpers::ZeroPad(blurVariance, 2);
FloatScalarImageType::Pointer blurredLuminance = FloatScalarImageType::New();
// Blur with a Gaussian kernel
MaskOperations::MaskedBlur(luminanceFilter->GetOutput(), maskReader->GetOutput(),
blurVariance, blurredLuminance.GetPointer());
std::stringstream ssBlurredLuminance;
ssBlurredLuminance << "Test/BlurredLuminance_" << fileNumber << ".mha";
OutputHelpers::WriteImage(blurredLuminance.GetPointer(), ssBlurredLuminance.str());
//Helpers::WriteImage<FloatScalarImageType>(blurredLuminance, "Test/TestIsophotes.blurred.mha");
FloatVector2ImageType::Pointer gradient = FloatVector2ImageType::New();
Derivatives::MaskedGradient(blurredLuminance.GetPointer(), maskReader->GetOutput(), gradient.GetPointer());
// Boundary gradient
typedef itk::MaskImageFilter< FloatVector2ImageType, UnsignedCharScalarImageType, FloatVector2ImageType > MaskFilterType;
MaskFilterType::Pointer maskFilter = MaskFilterType::New();
maskFilter->SetInput(gradient);
//maskFilter->SetMaskImage(inpainting.GetBoundaryImage());
maskFilter->Update();
vtkSmartPointer<vtkPolyData> boundaryGradient = vtkSmartPointer<vtkPolyData>::New();
// TODO: Convert this call to new API
//Helpers::ConvertNonZeroPixelsToVectors(maskFilter->GetOutput(), boundaryGradient);
std::stringstream ssPolyData;
ssPolyData << "Test/BoundaryGradient_" << fileNumber << ".vtp";
OutputHelpers::WritePolyData(boundaryGradient, ssPolyData.str());
}
return EXIT_SUCCESS;
}
bool TestExtractChannel()
{
// VectorImage
{
typedef itk::VectorImage<float, 2> VectorImageType;
VectorImageType::Pointer image = VectorImageType::New();
itk::Index<2> corner = {{0,0}};
itk::Size<2> size = {{100,100}};
itk::ImageRegion<2> region(corner, size);
image->SetRegions(region);
image->SetNumberOfComponentsPerPixel(2);
image->Allocate();
typedef itk::Image<float, 2> FloatScalarImageType;
FloatScalarImageType::Pointer floatScalarImage = FloatScalarImageType::New();
ITKHelpers::ExtractChannel(image.GetPointer(), 0, floatScalarImage.GetPointer());
typedef itk::Image<unsigned char, 2> UnsignedCharScalarImageType;
UnsignedCharScalarImageType::Pointer unsignedCharScalarImage = UnsignedCharScalarImageType::New();
ITKHelpers::ExtractChannel(image.GetPointer(), 0, unsignedCharScalarImage.GetPointer());
}
// VectorImage different output type
{
typedef itk::VectorImage<float, 2> VectorImageType;
VectorImageType::Pointer image = VectorImageType::New();
itk::Index<2> corner = {{0,0}};
itk::Size<2> size = {{100,100}};
itk::ImageRegion<2> region(corner, size);
image->SetRegions(region);
image->SetNumberOfComponentsPerPixel(2);
image->Allocate();
typedef itk::Image<unsigned char, 2> UnsignedCharScalarImageType;
UnsignedCharScalarImageType::Pointer unsignedCharScalarImage = UnsignedCharScalarImageType::New();
ITKHelpers::ExtractChannel(image.GetPointer(), 0, unsignedCharScalarImage.GetPointer());
}
// Scalar Image
{
typedef itk::Image<float, 2> VectorImageType;
VectorImageType::Pointer image = VectorImageType::New();
itk::Index<2> corner = {{0,0}};
itk::Size<2> size = {{100,100}};
itk::ImageRegion<2> region(corner, size);
image->SetRegions(region);
image->Allocate();
typedef itk::Image<float, 2> FloatScalarImageType;
FloatScalarImageType::Pointer floatScalarImage = FloatScalarImageType::New();
ITKHelpers::ExtractChannel(image.GetPointer(), 0, floatScalarImage.GetPointer());
typedef itk::Image<unsigned char, 2> UnsignedCharScalarImageType;
UnsignedCharScalarImageType::Pointer unsignedCharScalarImage = UnsignedCharScalarImageType::New();
ITKHelpers::ExtractChannel(image.GetPointer(), 0, unsignedCharScalarImage.GetPointer());
}
// Image<CovariantVector>
{
typedef itk::Image<itk::CovariantVector<float, 3>, 2> VectorImageType;
VectorImageType::Pointer image = VectorImageType::New();
itk::Index<2> corner = {{0,0}};
itk::Size<2> size = {{100,100}};
itk::ImageRegion<2> region(corner, size);
image->SetRegions(region);
image->Allocate();
typedef itk::Image<float, 2> FloatScalarImageType;
FloatScalarImageType::Pointer floatScalarImage = FloatScalarImageType::New();
ITKHelpers::ExtractChannel(image.GetPointer(), 0, floatScalarImage.GetPointer());
}
// Image<Vector>
{
typedef itk::Image<itk::Vector<float, 3>, 2> VectorImageType;
VectorImageType::Pointer image = VectorImageType::New();
itk::Index<2> corner = {{0,0}};
itk::Size<2> size = {{100,100}};
itk::ImageRegion<2> region(corner, size);
image->SetRegions(region);
image->Allocate();
typedef itk::Image<float, 2> FloatScalarImageType;
FloatScalarImageType::Pointer floatScalarImage = FloatScalarImageType::New();
ITKHelpers::ExtractChannel(image.GetPointer(), 0, floatScalarImage.GetPointer());
}
return true;
}
