#include "vtkMPR3DDataViewer.h"


vtkMPR3DDataViewer::vtkMPR3DDataViewer()
{
	_visiblePosition[0]=false;
	_visiblePosition[1]=false;
	_visiblePosition[2]=false;
	_ctfun = NULL;
	_vtkmprbasedata=NULL;

	_ctfun=NULL;
	_saggitalColors=NULL;
	_saggital=NULL;
	_axialColors=NULL;
	_axial=NULL;
	_coronalColors=NULL;
	_coronal=NULL;
	_mapOutline=NULL;
	_outline=NULL;
	_outlineData=NULL;
}
//-------------------------------------------------------------------
vtkMPR3DDataViewer::~vtkMPR3DDataViewer()
{
	_outlineData-> Delete();
	_mapOutline-> Delete();
	_outline-> Delete();

//	if (_bwLut)		_bwLut->Delete();
//	if (_hueLut)		_hueLut->Delete();
//	if (_satLut)		_satLut->Delete();
	if (_ctfun)		_ctfun->Delete();
	if (_saggitalColors)	_saggitalColors	->Delete();
	if (_saggital)		_saggital->Delete();
	if (_axialColors)	_axialColors->Delete();
	if (_axial)		_axial->Delete();
	if (_coronalColors)	_coronalColors->Delete();
	if (_coronal)		_coronal->Delete();

}
//-------------------------------------------------------------------
vtkActor* vtkMPR3DDataViewer::GetOutlineActor()
{
	return _outline;
}
//-------------------------------------------------------------------
vtkImageActor* vtkMPR3DDataViewer::GetImageActor(int id)
{
	vtkImageActor *tmpVtkActor=NULL;
	if (id==0){
		tmpVtkActor = GetvtkActor_saggital();
	}
	if (id==1){
		tmpVtkActor = GetvtkActor_coronal();
	}
	if (id==2){
		tmpVtkActor = GetvtkActor_axial();
	}
	return tmpVtkActor;
}
//-------------------------------------------------------------------
void vtkMPR3DDataViewer::Refresh()
{
	int x = (int)(_vtkmprbasedata->GetX());
	int y = (int)(_vtkmprbasedata->GetY());
	int z = (int)(_vtkmprbasedata->GetZ());
	SetPositionX( x );
	SetPositionY( y );
	SetPositionZ( z );
}
//-------------------------------------------------------------------
void vtkMPR3DDataViewer::SetVtkMPRBaseData(vtkMPRBaseData *vtkmprbasedata)
{
	_vtkmprbasedata=vtkmprbasedata;
}
//-------------------------------------------------------------------
vtkMPRBaseData* vtkMPR3DDataViewer::GetVtkMPRBaseData()
{
	return _vtkmprbasedata;
}
//-------------------------------------------------------------------

vtkColorTransferFunction *vtkMPR3DDataViewer::GetvtkColorTransferFunction()
{
	return _ctfun;
}
//-------------------------------------------------------------------

std::vector<double>* vtkMPR3DDataViewer::GetctFunVectorPoint()
{
	return &_ctFunVectorPoint;
}
//-------------------------------------------------------------------
std::vector<double>*     vtkMPR3DDataViewer::GetctFunVectorRed()
{
	return &_ctFunVectorRed;
}

//-------------------------------------------------------------------
std::vector<double>*     vtkMPR3DDataViewer::GetctFunVectorGreen()
{
	return &_ctFunVectorGreen;
}

//-------------------------------------------------------------------
std::vector<double>*     vtkMPR3DDataViewer::GetctFunVectorBlue()
{
	return &_ctFunVectorBlue;
}

//-------------------------------------------------------------------
void vtkMPR3DDataViewer::Configure()
{
/*
  // Start by creatin a black/white lookup table.
	_bwLut = vtkLookupTable::New();
    _bwLut->SetTableRange (0, 2000);
    _bwLut->SetSaturationRange (0, 0);
    _bwLut->SetHueRange (0, 0);
    _bwLut->SetValueRange (0, 1);

  // Now create a lookup table that consists of the full hue circle
  // (from HSV).
	_hueLut = vtkLookupTable::New();
    _hueLut->SetTableRange (0, 2000);
    _hueLut->SetHueRange (0, 1);
    _hueLut->SetSaturationRange (1, 1);
    _hueLut->SetValueRange (1, 1);

  // Finally, create a lookup table with a single hue but having a range
  // in the saturation of the hue.
	_satLut = vtkLookupTable::New();
    _satLut->SetTableRange (0, 2000);
    _satLut->SetHueRange (.6, .6);
    _satLut->SetSaturationRange (0, 1);
    _satLut->SetValueRange (1, 1);
*/


	double range[2];
	double max;
	
	if(_vtkmprbasedata->GetImageData()){
		_vtkmprbasedata->GetImageData()->GetScalarRange(range);
		max = range[1];
	}else{
		max = 1;
	}
	
	
	

	_ctFunVectorPoint.clear();
	_ctFunVectorPoint.push_back(max*0/4);
	_ctFunVectorPoint.push_back(max*1/4);
	_ctFunVectorPoint.push_back(max*2/4);
	_ctFunVectorPoint.push_back(max*3/4);
	_ctFunVectorPoint.push_back(max*4/4);

	_ctFunVectorRed.clear();
	_ctFunVectorRed.push_back(0.0);
	_ctFunVectorRed.push_back(1.0);
	_ctFunVectorRed.push_back(0.0);
	_ctFunVectorRed.push_back(0.0);
	_ctFunVectorRed.push_back(0.0);

	_ctFunVectorGreen.clear();
	_ctFunVectorGreen.push_back(0.0);
	_ctFunVectorGreen.push_back(0.0);
	_ctFunVectorGreen.push_back(0.0);
	_ctFunVectorGreen.push_back(1.0);
	_ctFunVectorGreen.push_back(0.2);

	_ctFunVectorBlue.clear();
	_ctFunVectorBlue.push_back(0.0);
	_ctFunVectorBlue.push_back(0.0);
	_ctFunVectorBlue.push_back(1.0);
	_ctFunVectorBlue.push_back(0.0);
	_ctFunVectorBlue.push_back(0.0);


	if(_ctfun==NULL){
		_ctfun = vtkColorTransferFunction::New();
	}
	_ctfun->RemoveAllPoints();

	int i,size=_ctFunVectorPoint.size();
	for (i=0;i<size;i++)
	{
		_ctfun->AddRGBPoint( _ctFunVectorPoint[i] , _ctFunVectorRed[i],_ctFunVectorGreen[i],_ctFunVectorBlue[i]);
		
	}
/*
	_ctfun->AddRGBPoint( 100 ,0 ,0,0);
	_ctfun->AddRGBPoint( 200 ,1 ,0,0);
	_ctfun->AddRGBPoint( 300 ,0 ,1,0);
	_ctfun->AddRGBPoint( 400 ,0 ,0,1);
*/
//	void *p=this; // JPRx

  // Create the first of the three planes. The filter vtkImageMapToColors
  // maps the data through the corresponding lookup table created above.  The
  // vtkImageActor is a type of vtkProp and conveniently displays an image on
  // a single quadrilateral plane. It does this using texture mapping and as
  // a result is quite fast. (Note: the input image has to be unsigned char
  // values, which the vtkImageMapToColors produces.) Note also that by
  // specifying the DisplayExtent, the pipeline requests data of this extent
  // and the vtkImageMapToColors only processes a slice of data.
	
	if(_saggitalColors==NULL){
		_saggitalColors = vtkImageMapToColors::New();
	}
	_saggitalColors->RemoveAllInputs();
	
	if(_vtkmprbasedata->GetImageData()){
		_saggitalColors->SetInput( _vtkmprbasedata->GetImageData() );
	}
//    _saggitalColors->SetLookupTable(_bwLut);
	
    _saggitalColors->SetLookupTable(_ctfun);
	

	if(_saggital==NULL){
		_saggital = vtkImageActor::New();
		_saggital->SetInput(_saggitalColors->GetOutput());
	}	
	//_saggitalColors->Update();
    


  // Create the second (axial) plane of the three planes. We use the
  // same approach as before except that the extent differs.

	
	if(_axialColors==NULL){
		_axialColors = vtkImageMapToColors::New();
	}

	_axialColors->RemoveAllInputs();
	if(_vtkmprbasedata->GetImageData()){
		_axialColors->SetInput( _vtkmprbasedata->GetImageData() );
	}
//    _axialColors->SetLookupTable(_hueLut);
    _axialColors->SetLookupTable(_ctfun);

	if(_axial==NULL){
		_axial = vtkImageActor::New();
		_axial->SetInput(_axialColors->GetOutput());
	}
	
	//_axialColors->Update();
    

	// Create the third (coronal) plane of the three planes. We use 
  // the same approach as before except that the extent differs.
	if(_coronalColors==NULL){
		_coronalColors = vtkImageMapToColors::New();
	}
	_coronalColors->RemoveAllInputs();
	if(_vtkmprbasedata->GetImageData()){
		_coronalColors->SetInput( _vtkmprbasedata->GetImageData() );
	}
//    _coronalColors->SetLookupTable(_satLut);
    _coronalColors->SetLookupTable(_ctfun);

	if(_coronal==NULL){
		_coronal = vtkImageActor::New();
		_coronal->SetInput(_coronalColors->GetOutput());
	}
	
    

  // An outline provides context around the data.
  //
	if(_outlineData==NULL){
		_outlineData = vtkOutlineFilter::New();
	}
	_outlineData->RemoveAllInputs();
	if(_vtkmprbasedata->GetImageData()){
		_outlineData->SetInput((vtkDataSet *) _vtkmprbasedata->GetImageData() );
	}

	if(_mapOutline==NULL){
		_mapOutline = vtkPolyDataMapper::New();
		_mapOutline->SetInput(_outlineData->GetOutput());
	}
	_mapOutline->Update();    
	//_mapOutline->RemoveAllInputs();
    
	if(_outline==NULL){
		_outline = vtkActor::New();
		_outline->SetMapper(_mapOutline);
		_outline->GetProperty()->SetColor(0,0,0);
	}
	
    

	//int ext[6];
	//_vtkmprbasedata->GetImageData()->GetExtent(ext);

	


}

//-------------------------------------------------------------------
vtkImageActor * vtkMPR3DDataViewer::GetvtkActor_saggital()
{
	return _saggital;
}
//-------------------------------------------------------------------
vtkImageActor * vtkMPR3DDataViewer::GetvtkActor_coronal()
{
	return _coronal;
}
//-------------------------------------------------------------------
vtkImageActor * vtkMPR3DDataViewer::GetvtkActor_axial()
{
	return _axial;
}

//------------------------------------------------------------------------
void vtkMPR3DDataViewer::SetPositionX(int pos){
	int x1,x2,y1,y2,z1,z2;
	_vtkmprbasedata->GetDimensionExtention(&x1,&x2,&y1,&y2,&z1,&z2);
	_saggital->SetDisplayExtent( pos , pos , y1 ,y2 , z1 , z2 );
}
//------------------------------------------------------------------------
void vtkMPR3DDataViewer::SetPositionY(int pos){
	int x1,x2,y1,y2,z1,z2;
	_vtkmprbasedata->GetDimensionExtention(&x1,&x2,&y1,&y2,&z1,&z2);
	_coronal->SetDisplayExtent(x1,x2, pos,pos, z1,z2);
}
//------------------------------------------------------------------------
void vtkMPR3DDataViewer::SetPositionZ(int pos){
	int x1,x2,y1,y2,z1,z2;
	_vtkmprbasedata->GetDimensionExtention(&x1,&x2,&y1,&y2,&z1,&z2);
	_axial->SetDisplayExtent(x1,x2, y1,y2, pos,pos);
}
//-------------------------------------------------------------------
void vtkMPR3DDataViewer::SetVisiblePosition(int idPosition, bool visible)
{
	_visiblePosition[idPosition]=visible;
}
//-------------------------------------------------------------------
bool vtkMPR3DDataViewer::GetVisiblePosition(int idPosition)
{
	return _visiblePosition[idPosition];
}

void vtkMPR3DDataViewer::setColorTransferFunction(vtkColorTransferFunction* colortable){
	_saggitalColors->SetLookupTable(colortable);
	 _coronalColors->SetLookupTable(colortable);
	_axialColors->SetLookupTable(colortable);
}