#ifndef _SENSE_GRADIENT_CXX
#define _SENSE_GRADIENT_CXX

#include "Sense_Gradient.h"


namespace mial
{
	//Default constructor
	template<class DataType, class TInputImage, class TGradientImage, int nDims>
		Sense_Gradient<DataType,TInputImage,TGradientImage,nDims>
		::Sense_Gradient()
		: Sensor()
	{
		//Instantiate the filters
		magnitudeFilter = MagnitudeFilterType::New();
		gradientFilter = GradientFilterType::New();
		sensorOutput = sensorOut::New();
		//Set the base class pointer
		Sensor::sensorOutput = sensorOutput.GetPointer();
	}

	template<class DataType, class TInputImage, class TGradientImage, int nDims>
		void 
		Sense_Gradient<DataType,TInputImage,TGradientImage,nDims>
		::writeImageToFile(OutputImagePointer img,char * name)
	{
		typedef typename itk::ImageFileWriter<OutputImageType> WriterType;
		typename WriterType::Pointer writer = WriterType::New(); 
		writer->SetFileName(name);
		writer->SetInput(img);
		writer->Update();
	}

	template<class DataType, class TInputImage, class TGradientImage, int nDims>
		void 
		Sense_Gradient<DataType, TInputImage, TGradientImage, nDims>
		::run(typename Sensor::sensorIn * const i)
	{
		typename sensorIn::Pointer in = reinterpret_cast<sensorIn *>(i); //Typecast the input to its desired form

		//Calculate the gradient magnitude at scale sigma
		magnitudeFilter->SetSigma(in->sigma);
		magnitudeFilter->SetNormalizeAcrossScale(1); //Set so magnitude is normalized across scales.
		magnitudeFilter->SetInput(in->imageIn);

		//testing
		//this->writeImageToFile(magnitudeFilter->GetOutput(),"mag.mhd");
		//this->sensorOutput.imageOut = magnitudeFilter->GetOutput();
		//magnitudeFilter->Update();

		//Calculate the gradients
		gradientFilter->SetSigma(in->sigma);
		gradientFilter->SetNormalizeAcrossScale(1); //Set so magnitude is normalized across scales.
		gradientFilter->SetInput(magnitudeFilter->GetOutput());

		gradientFilter->Update();

		this->sensorOutput->imageOut = gradientFilter->GetOutput();

	}
}
#endif