//===== 
// Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
//===== 
#include "bbcreaVtkApplyMask.h"
#include "bbcreaVtkPackage.h"

#include "creaVtk_MACROS.h"

namespace bbcreaVtk
{

BBTK_ADD_BLACK_BOX_TO_PACKAGE(creaVtk,ApplyMask)
BBTK_BLACK_BOX_IMPLEMENTATION(ApplyMask,bbtk::AtomicBlackBox);
//===== 
// Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
//===== 
void ApplyMask::Process()
{

// THE MAIN PROCESSING METHOD BODY
//   Here we simply set the input 'In' value to the output 'Out'
//   And print out the output value
// INPUT/OUTPUT ACCESSORS ARE OF THE FORM :
//    void bbSet{Input|Output}NAME(const TYPE&)
//    const TYPE& bbGet{Input|Output}NAME() const 
//    Where :
//    * NAME is the name of the input/output
//      (the one provided in the attribute 'name' of the tag 'input')
//    * TYPE is the C++ type of the input/output
//      (the one provided in the attribute 'type' of the tag 'input')

//    bbSetOutputOut( bbGetInputIn() );
//    std::cout << "Output value = " <<bbGetOutputOut() << std::endl;

	if ((bbGetInputImage()!=NULL) && (bbGetInputMask()!=NULL))
	{
		int extImage[6];
		bbGetInputImage()->GetExtent( extImage );
		int dimXImage = extImage[1]-extImage[0]+1;
		int dimYImage = extImage[3]-extImage[2]+1;
		int dimZImage = extImage[5]-extImage[4]+1;

		int extMask[6];
		bbGetInputImage()->GetExtent( extMask );
		int dimXMask = extMask[1]-extMask[0]+1;
		int dimYMask = extMask[3]-extMask[2]+1;
		int dimZMask = extMask[5]-extMask[4]+1;


		if (bbGetOutputOut()!=NULL) 
		{
			bbGetOutputOut()->Delete();
		}	

		vtkImageData *resultImage=NULL;
		resultImage = vtkImageData::New();
		resultImage->Initialize();
		resultImage->SetSpacing( bbGetInputImage()->GetSpacing() );
		resultImage->SetDimensions(  dimXImage, dimYImage, dimZImage );
		resultImage->AllocateScalars( bbGetInputImage()->GetScalarType(),1 );
			

		if ((dimXImage==dimXMask) &&
			(dimYImage==dimYMask) &&
			(dimZImage==dimZMask)) 
			{


				long int i, size = dimXImage * dimYImage * dimZImage;
				#pragma omp parallel for 
				for (i=0; i<size;i++)
				{

					DEF_POINTER_IMAGE_VTK_CREA(vI,ssI,pI,stI, bbGetInputImage() );
					DEF_POINTER_IMAGE_VTK_CREA(vM,ssM,pM,stM, bbGetInputMask() );
					DEF_POINTER_IMAGE_VTK_CREA(vO,ssO,pO,stO, resultImage );

					GETVALUE2_VTK_CREA(vI,pI,stI,i);
					GETVALUE2_VTK_CREA(vM,pM,stM,i);
					if (vM!=0)
					{
						vO = vI;
					} else {
						vO = bbGetInputBackground(); 
					}
					SETVALUE2_VTK_CREA(vO,pO,stO,i); 	
				} // for
		} //if dim
	bbSetOutputOut( resultImage );
	} // if Image && Mask	

}
//===== 
// Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
//===== 
void ApplyMask::bbUserSetDefaultValues()
{

//  SET HERE THE DEFAULT INPUT/OUTPUT VALUES OF YOUR BOX 
//    Here we initialize the input 'In' to 0
   bbSetInputImage(NULL);
   bbSetInputMask(NULL);
   bbSetInputBackground(0);
   bbSetOutputOut(NULL);
  
}
//===== 
// Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
//===== 
void ApplyMask::bbUserInitializeProcessing()
{

//  THE INITIALIZATION METHOD BODY :
//    Here does nothing 
//    but this is where you should allocate the internal/output pointers 
//    if any 

  
}
//===== 
// Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)
//===== 
void ApplyMask::bbUserFinalizeProcessing()
{

//  THE FINALIZATION METHOD BODY :
//    Here does nothing 
//    but this is where you should desallocate the internal/output pointers 
//    if any
  
}
}
// EO namespace bbcreaVtk