/*=========================================================================

  Program:   wxMaracas
  Module:    $RCSfile: wxMaracasMultipleVolumeRendererManager.cxx,v $
  Language:  C++
  Date:      $Date: 2009/11/20 17:09:05 $
  Version:   $Revision: 1.9 $

  Copyright: (c) 2002, 2003
  License:

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

=========================================================================*/
#include "wxMaracasMultipleVolumeRendererManager.h"

#include <vtkMetaImageReader.h>
#include <vtkImageCast.h>
/**
**	Start of the manager class
**/
wxMaracasMultipleVolumeRendererManager::wxMaracasMultipleVolumeRendererManager(){
	_renderer = NULL;
	_idCount=0;
}
wxMaracasMultipleVolumeRendererManager::~wxMaracasMultipleVolumeRendererManager(){
}

/**
**	Sets the renderer to manage the prop3D from the view
**/
void wxMaracasMultipleVolumeRendererManager::setRenderer(vtkRenderer*  renderer){
	_renderer = renderer;
}

/**
** Gets the renderer which manage the prop3D from the view
**/
vtkRenderer* wxMaracasMultipleVolumeRendererManager::getRenderer(){
	return _renderer;
}

/**
** Updates Volume
**/
void wxMaracasMultipleVolumeRendererManager::Update(int ppid){
	wxMaracasMultipleVolumeRendererManagerData* data = this->getViewData(ppid);
	data->Update();
	_renderer->Render();
}

/**
**	Adds a prop3D to the manager and returns the identifier
**/
int wxMaracasMultipleVolumeRendererManager::addVolume(int idTP, vtkImageData* vol, std::string dataname) throw(char*){
	checkInvariant();
	image = vol;
	if(vol != NULL){
		wxMaracasMultipleVolumeRendererManagerData* data = new wxMaracasMultipleVolumeRendererManagerData(vol, dataname);
		prop3Dvect.push_back(data);
		_renderer->AddActor(data->getProp3D());
		if(idTP == -1)
		{
			data->setId(_idCount);
			_idCount++;
		}
		else
		{
			data->setId(idTP);
		}
		printf("wxMaracasMultipleVolumeRendererManager::addVolume->idVolumeRenderer: %i\n", data->getId());
		return data->getId();
	}else{
		throw "Check mhd imagefile file or input";
	}
	return -1;
}
/**
**	adds or removes an actor depending of the bool value
**/

void wxMaracasMultipleVolumeRendererManager::addRemoveActor(int propid, bool addremove)  throw(char*){
	checkInvariant();
	
	wxMaracasMultipleVolumeRendererManagerData* data = this->getViewData(propid);		
	if(data->getProp3D()!=NULL){
		if(addremove){
			_renderer->AddViewProp(data->getProp3D());
		}else{
			_renderer->RemoveViewProp(data->getProp3D());
		}
		_renderer->Render();
	}
	
}
/**
**	Changes the opacity in a prop3D
**/
void wxMaracasMultipleVolumeRendererManager::setVolumeOpacity(int propid, std::vector<double> greylevel,std::vector<double> value)  throw(char*){
	checkInvariant();		

	this->getViewData(propid)->setVolumeOpacity(greylevel, value);
	
	_renderer->Render();

}

/**
**	Set Volume Color
**/
void wxMaracasMultipleVolumeRendererManager::setVolumeColor(int volid, std::vector<double> greylevel,
								std::vector<double> red,
								std::vector<double> green,
								std::vector<double> blue)throw(char*){
	checkInvariant();		

	this->getViewData(volid)->setVolumeColor(greylevel, red, green, blue);
	
	_renderer->Render();
}

vtkImageData* wxMaracasMultipleVolumeRendererManager::getImageData(std::string filename){
	if(filename.compare("")!= 0){	
		
		vtkMetaImageReader* reader =  vtkMetaImageReader::New();	
		reader->SetFileName(filename.c_str());
		reader->Update();
		vtkImageData* img = reader->GetOutput();	

		vtkImageCast* cast = vtkImageCast::New(); 
		cast->SetInput(img);
		cast->SetOutputScalarTypeToUnsignedShort();
		cast->Update();			
		//reader->Delete();
		//img->Delete();
		return cast->GetOutput();
		//return img;
	}	
	return NULL;
}

vtkImageData* wxMaracasMultipleVolumeRendererManager::getImageData(){	
	return image;
}

void wxMaracasMultipleVolumeRendererManager::checkInvariant()  throw(char*){
	if(this->_renderer==NULL){
		throw "Renderer not set";
	}
}

wxMaracasMultipleVolumeRendererManagerData* wxMaracasMultipleVolumeRendererManager::getViewData(int id) throw(char*){
    int i;
	for(i = 0; i < (int)(prop3Dvect.size());i++){
		if(prop3Dvect[i]->getId() == id){
			return prop3Dvect[i];
		}
	}
	throw "id not found in the data";

	return NULL;
}

void wxMaracasMultipleVolumeRendererManager::deleteActor(int propid) throw (char *){
	checkInvariant();	

	this->addRemoveActor(propid, false);

	int i,n;
	bool exit = false;
	for(i = 0; i < (int)(prop3Dvect.size())&&!exit;i++){
		if(prop3Dvect[i]->getId() == propid){			
			n=i;
			exit = true;
		}
	}
	if(exit){
		wxMaracasMultipleVolumeRendererManagerData* data = prop3Dvect[n];			
		int j;
		for(j = i; j < (int)(prop3Dvect.size())-1;j++){
			prop3Dvect[j] = prop3Dvect[j+1];
		}		
		delete data;
		prop3Dvect.pop_back();
	}else{
		throw "id not found in the data";
	}	
    
}

vtkPiecewiseFunction* wxMaracasMultipleVolumeRendererManager::GetTransferFunction(int volumeid){
	return getViewData(volumeid)->GetTransferFunction();
}
vtkColorTransferFunction* wxMaracasMultipleVolumeRendererManager::GetColorFunction(int volumeid){

	return getViewData(volumeid)->GetColorFunction();
}

void wxMaracasMultipleVolumeRendererManager::changeCompositeMIPFunction(int id, int function) throw (char *){
	getViewData(id)->changeCompositeMIPFunction(function);
}