/*=========================================================================
  
  Filter:    ComposeRGBA
  Program:   Insight Segmentation & Registration Toolkit
  Module:    ComposeRGBATests.cxx
  Language:  C++
  Date:      200/05/11
  Version:   1.0

  Portions of this code are covered under the ITK and VTK copyright.
  See VTKCopyright.txt or http://www.kitware.com/VTKCopyright.htm for details.
  See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.  See the above copyright notices for more information.
=========================================================================*/

#if defined(_MSC_VER)
//Warning about: identifier was truncated to '255' characters in the debug information (MVC6.0 Debug)
#pragma warning( disable : 4786 )
#endif

// General includes
#include <iostream>

// ITK includes
#include "itkNumericTraits.h"
#include "itkRGBAPixel.h"
#include "itkImage.h"
#include "itkImageFileReader.h"
#include "itkImageFileWriter.h"
#include "itkComposeRGBImageFilter.h"
#include "itkComposeRGBAImageFilter.h"

/////////////////////////////////////////////////////////////
// Template for ComposeRGBA Tests
template <int VDimensions>
int Test_ComposeRGBA_ND(int argc, char* argv[])
{
    try
	{
		// ARGUMENTS:
		//argv[0] = Executable name
		//argv[1] = Output file name and path
		//argv[2] = Input 1 file name and path
		//argv[2] = Input 2 file name and path
		//argv[2] = Input 3 file name and path
		//argv[2] = Input 4 file name and path

		// Get arguments
		char* OutputFilename  = argv[1];
		char* Input1Filename  = argv[2];
		char* Input2Filename  = argv[3];
		char* Input3Filename  = argv[4];
		char* Input4Filename  = argv[5];

		// Typedefs
		typedef unsigned char ScalarPixelType;
		typedef itk::RGBAPixel< ScalarPixelType >
            RGBAPixelType;
		typedef itk::Image< ScalarPixelType, VDimensions >
            ScalarImageType;
		typedef itk::Image< RGBAPixelType, VDimensions >
            RGBAImageType;
		typedef itk::ImageFileReader< ScalarImageType >
            ReaderType;
		typedef itk::ImageFileWriter< RGBAImageType >
            WriterType;
		typedef itk::ComposeRGBAImageFilter< ScalarImageType, RGBAImageType >
            ComposeFilterType;

		// Read input1
		ReaderType::Pointer reader1 = ReaderType::New();
		reader1->SetFileName( Input1Filename );
		reader1->Update();
		
		// Read input2
		ReaderType::Pointer reader2 = ReaderType::New();
		reader2->SetFileName( Input2Filename );
		reader2->Update();

		// Read input3
		ReaderType::Pointer reader3 = ReaderType::New();
		reader3->SetFileName( Input3Filename );
		reader3->Update();

		// Read input4
		ReaderType::Pointer reader4 = ReaderType::New();
		reader4->SetFileName( Input4Filename );
		reader4->Update();

		// Test ComposeRGBA filter
		ComposeFilterType::Pointer filterCompose = ComposeFilterType::New();
		filterCompose->SetInput( 0, reader1->GetOutput() );
		filterCompose->SetInput( 1, reader2->GetOutput() );
		filterCompose->SetInput( 2, reader3->GetOutput() );
		filterCompose->SetInput( 3, reader4->GetOutput() );
		filterCompose->Update();

        // Write output
		WriterType::Pointer writer = WriterType::New();
		writer->SetFileName( OutputFilename );
		writer->SetInput( filterCompose->GetOutput() );
		writer->Update();
	}
	catch (itk::ExceptionObject & err) 
    { 
		std::cerr << "ExceptionObject caught !" << std::endl; 
		std::cerr << err << std::endl; 
		return EXIT_FAILURE;
    } 

	//Return
	return EXIT_SUCCESS;
}

/////////////////////////////////////////////////////////////
// ComposeRGBA Test for 2D images
int Test_ComposeRGBA_2D(int argc, char* argv[])
{
	return Test_ComposeRGBA_ND<2>(argc, argv);
}