#include "volumerenderermanagerdata.h"

/********************************************************************************************
** Start of data viewmanagerData
*********************************************************************************************/

VolumeRendererManagerData::VolumeRendererManagerData(vtkImageData* vol, std::string dataname){

    Initialize(vol, dataname);


}

VolumeRendererManagerData::VolumeRendererManagerData(vtkImageData* vol, vtkRenderer* render, std::string dataname){

    Initialize(vol, dataname);
    render->AddActor(this->getProp3D());
}

void VolumeRendererManagerData::Initialize(vtkImageData* vol, std::string dataname){
    _vol = vol;
    _dataname = dataname;

    _tfun = vtkPiecewiseFunction::New();
    _ctfun = vtkColorTransferFunction::New();

    _volumePlanes  = vtkPlanes::New();
    _compositeFunction = vtkVolumeRayCastCompositeFunction::New();
    _MIPFunction = vtkVolumeRayCastMIPFunction::New();

    _volumeMapper = vtkVolumeRayCastMapper::New();
    _volumeMapper->SetVolumeRayCastFunction(_compositeFunction);
    _volumeMapper->SetClippingPlanes( _volumePlanes );
    _volumeMapper->AutoAdjustSampleDistancesOn();
    _volumeProperty = vtkVolumeProperty::New();
    _volumeProperty->SetInterpolationTypeToLinear();
    //_volumeProperty->ShadeOn();
    _volumeProperty->DisableGradientOpacityOn();
    _volumeProperty->SetColor(_ctfun);
    _volumeProperty->SetScalarOpacity(_tfun );

    _newvol = vtkVolume::New();
    _newvol->SetMapper(_volumeMapper );
    _newvol->SetProperty(_volumeProperty );
    _volumeMapper->SetInput( _vol );
    _volumeMapper->Update();
    _newvol->Update();
}


VolumeRendererManagerData::~VolumeRendererManagerData(){

        _tfun->Delete();
        _ctfun->Delete();
        _volumePlanes->Delete();
        _compositeFunction->Delete();
        _volumeMapper->Delete();
        _volumeProperty->Delete();
        _newvol->Delete();

}
/**
**	Set Volume Color
**/
void VolumeRendererManagerData::setVolumeColor(std::vector<double>& greylevelcolors,
                                                                std::vector<double>& red,
                                                                std::vector<double>& green,
                                                                std::vector<double>& blue)
{

        _ctfun->RemoveAllPoints();
        for(int i = 0; i < greylevelcolors.size();i++){
                _ctfun->AddRGBPoint(greylevelcolors[i], red[i],green[i], blue[i]);
                                //std::cout<<"VolumeRendererManagerData::setVolumeColor "<<greylevelcolors[i]<<" "<<red[i]<<" "<<green[i]<<" "<<blue[i]<<std::endl;
        }
        _newvol->Update();
}

/**
**	Volume Opacity
**/
void VolumeRendererManagerData::setVolumeOpacity(std::vector<double> greylevel,std::vector<double> value){

        _tfun->RemoveAllPoints();
        for(int i = 0; i < greylevel.size();i++){
                _tfun->AddPoint(greylevel[i], value[i]);
                        //std::cout<<"VolumeRendererManagerData::setVolumeOpacity "<<greylevel[i]<<" "<< value[i]<<std::endl;
        }
    _newvol->Update();
}

/**
**	Check if the variables are setted correctly
**/
void VolumeRendererManagerData::checkInvariant()throw (char *){
        if(!_compositeFunction){
                throw "No composite function initialized";
        }
        if(!_MIPFunction){
                throw "No MIP function initialized";
        }
        if(!_volumeMapper){
                throw "No volume mapper initialized";
        }
}

/**
**	Check if the variables are setted correctly
**/
void VolumeRendererManagerData::Update(){
         _newvol->Update();
}
/**
**	get the prop3D
**/
vtkProp3D* VolumeRendererManagerData::getProp3D(){
        return this->_newvol;
}
/**
**	return the id from the daat
**/
int VolumeRendererManagerData::getId(){
        return _id;
}
/**
**	set data id
**/
void VolumeRendererManagerData::setId(int propid){
        _id = propid;
}

/**
**	Get the filanme
**/
std::string VolumeRendererManagerData::getDataname(){
        return _dataname;
}
/**
** Set the filanme
**/
void VolumeRendererManagerData::setDataname(std::string dataname){
        _dataname = dataname;
}

void VolumeRendererManagerData::changeCompositeMIPFunction(int function) throw (char *){
        checkInvariant();
        if(function == 0){
                _volumeMapper->SetVolumeRayCastFunction(_compositeFunction);
        }else{
                _volumeMapper->SetVolumeRayCastFunction(_MIPFunction);
        }

}

void VolumeRendererManagerData::SetLookupTable(vtkLookupTable* lookuptable){
     _ctfun->RemoveAllPoints();
     _ctfun->DeepCopy((vtkColorTransferFunction*)lookuptable);
}