]> Creatis software - creaMaracasVisu.git/blobdiff - lib/Kernel/VTKObjects/VolumeRenderer/volumerenderermanagerdata.cxx
changes in qt for volume rendering with gpu support, if you are working with volume...
[creaMaracasVisu.git] / lib / Kernel / VTKObjects / VolumeRenderer / volumerenderermanagerdata.cxx
index e820153968c759b55971d87535cbfeee9bfd378d..3e6ad74dda2afae136efb66e602c061bfdbf8297 100644 (file)
@@ -5,6 +5,24 @@
 ** Start of data viewmanagerData
 *********************************************************************************************/
 
+#include "vtkStripper.h"
+#include "boxSurfaceObserver.h"
+#include "vtkProperty.h"
+
+#include <vtkVolumeRayCastMapper.h>
+
+#if VTK_MAJOR_VERSION >= 5
+    #if VTK_MINOR_VERSION >= 6
+        #include "vtkGPUVolumeRayCastMapper.h"
+    #endif
+#endif
+
+using namespace std;
+
+VolumeRendererManagerData::VolumeRendererManagerData(vtkImageData* vol, bool usegpu){
+    Initialize(vol, "", usegpu);
+}
+
 VolumeRendererManagerData::VolumeRendererManagerData(vtkImageData* vol, std::string dataname){
 
     Initialize(vol, dataname);
@@ -16,50 +34,170 @@ VolumeRendererManagerData::VolumeRendererManagerData(vtkImageData* vol, vtkRende
 
     Initialize(vol, dataname);
     render->AddActor(this->getProp3D());
+
 }
 
-void VolumeRendererManagerData::Initialize(vtkImageData* vol, std::string dataname){
+void VolumeRendererManagerData::Initialize(vtkImageData* vol, std::string dataname, bool usegpu){
+    _id = 0;
     _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->DisableGradientOpacityOn();    
+    _tfun = vtkPiecewiseFunction::New();
     _volumeProperty->SetScalarOpacity(_tfun );
+    _ctfun = vtkColorTransferFunction::New();
 
     _newvol = vtkVolume::New();
-    _newvol->SetMapper(_volumeMapper );
-    _newvol->SetProperty(_volumeProperty );
-    _volumeMapper->SetInput( _vol );
-    _volumeMapper->Update();
+    _newvol->SetProperty(_volumeProperty );    
+
+
+    _compositeFunction = 0;
+    _MIPFunction = 0;
+
+    BoxWidget = 0;
+
+/* EED9Juin2011
+    if(usegpu && _vol->GetDataDimension() > 2){
+        _volumeMappergpu = vtkGPUVolumeRayCastMapper::New();
+        _volumeMappergpu->SetClippingPlanes( _volumePlanes );
+        _volumeMappergpu->AutoAdjustSampleDistancesOn();
+
+        _newvol->SetMapper(_volumeMappergpu );
+
+        _volumeMappergpu->SetInput( _vol );
+        _volumeMappergpu->Update();
+    }else{
+        _compositeFunction = vtkVolumeRayCastCompositeFunction::New();
+        _MIPFunction = vtkVolumeRayCastMIPFunction::New();
+
+        _volumeMapper = vtkVolumeRayCastMapper::New();
+        _volumeMapper->SetVolumeRayCastFunction(_compositeFunction);
+        _volumeMapper->SetClippingPlanes( _volumePlanes );
+        _volumeMapper->AutoAdjustSampleDistancesOn();
+
+        _newvol->SetMapper(_volumeMapper );
+
+        _volumeMapper->SetInput( _vol );
+        _volumeMapper->Update();
+    }
+*/
+
+       VolumeMapper = 0;
+#if VTK_MAJOR_VERSION >= 5
+    #if VTK_MINOR_VERSION >= 6
+       vtkGPUVolumeRayCastMapper * volumeMappergpu = vtkGPUVolumeRayCastMapper::New();
+       volumeMappergpu->AutoAdjustSampleDistancesOn();
+       VolumeMapper = volumeMappergpu;
+    #endif
+#else
+
+       _volumeProperty->SetColor(_ctfun);
+
+
+       _compositeFunction                                                              = vtkVolumeRayCastCompositeFunction::New();
+       _MIPFunction                                                                    = vtkVolumeRayCastMIPFunction::New();
+       vtkVolumeRayCastMapper* volumeMapper                                                                    = vtkVolumeRayCastMapper::New();
+       volumeMapper->SetVolumeRayCastFunction(_compositeFunction);
+               volumeMapper->AutoAdjustSampleDistancesOn();
+               VolumeMapper                                                                    = volumeMapper;
+#endif
+
+
+        VolumeMapper->SetClippingPlanes( _volumePlanes );
+        _newvol->SetMapper(VolumeMapper );
+        VolumeMapper->SetInput( _vol );
+        VolumeMapper->Update();
     _newvol->Update();
 }
 
 
-VolumeRendererManagerData::~VolumeRendererManagerData(){
+VolumeRendererManagerData::~VolumeRendererManagerData()
+{
 
         _tfun->Delete();
         _ctfun->Delete();
         _volumePlanes->Delete();
-        _compositeFunction->Delete();
-        _volumeMapper->Delete();
         _volumeProperty->Delete();
         _newvol->Delete();
 
+        if(_compositeFunction)
+            _compositeFunction->Delete();
+        if(_MIPFunction)
+            _MIPFunction->Delete();
+        if(VolumeMapper)
+            VolumeMapper->Delete();
+
+        if(BoxWidget){
+            DisableBoundingBox();
+        }
+
+}
+void VolumeRendererManagerData::SetIndependentComponents(bool independent){
+
+    if(!independent){
+        _volumeProperty->IndependentComponentsOff();
+    }else{
+        _volumeProperty->IndependentComponentsOn();
+    }
 }
+
+void VolumeRendererManagerData::EnableBoundingBox(vtkRenderWindowInteractor* interactor)
+{
+//EED9Juin2011    if(_volumeMappergpu){
+
+    if(!BoxWidget){
+        BoxWidget = vtkBoxWidget::New();
+        BoxWidget->SetInteractor( interactor );
+        BoxWidget->SetPlaceFactor(1.01);
+
+        BoxWidget->SetInput( _vol );
+        BoxWidget->InsideOutOn();
+        BoxWidget->PlaceWidget();
+
+
+        vtkBoxWidgetCallback *callback = vtkBoxWidgetCallback::New();
+
+
+        callback->SetMapper(VolumeMapper);
+
+
+        BoxWidget->AddObserver(vtkCommand::InteractionEvent, callback);
+        callback->Delete();
+
+        BoxWidget->EnabledOn();
+        BoxWidget->GetSelectedFaceProperty()->SetOpacity(0.0);
+
+        cout<<"JPRG::VolumeRendererManagerData::EnableBoundingBox::CREATE"<<endl;
+
+    }else{
+
+        BoxWidget->EnabledOn();
+
+        cout<<"JPRG::VolumeRendererManagerData::EnableBoundingBox"<<endl;
+
+    }
+//EED9Juin2011  }
+}
+
+void VolumeRendererManagerData::DisableBoundingBox(){
+
+    if(BoxWidget){
+        BoxWidget->EnabledOff();
+        //BoxWidget->Delete();
+        //BoxWidget = 0;
+
+        cout<<"JPRG::VolumeRendererManagerData::DisableBoundingBox"<<endl;
+
+
+    }
+}
+
 /**
 **     Set Volume Color
 **/
@@ -70,7 +208,7 @@ void VolumeRendererManagerData::setVolumeColor(std::vector<double>& greylevelcol
 {
 
         _ctfun->RemoveAllPoints();
-        for(int i = 0; i < greylevelcolors.size();i++){
+        for(unsigned 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;
         }
@@ -83,13 +221,24 @@ void VolumeRendererManagerData::setVolumeColor(std::vector<double>& greylevelcol
 void VolumeRendererManagerData::setVolumeOpacity(std::vector<double> greylevel,std::vector<double> value){
 
         _tfun->RemoveAllPoints();
-        for(int i = 0; i < greylevel.size();i++){
+        for(unsigned int i = 0; i < greylevel.size();i++){
                 _tfun->AddPoint(greylevel[i], value[i]);
-                        //std::cout<<"VolumeRendererManagerData::setVolumeOpacity "<<greylevel[i]<<" "<< value[i]<<std::endl;
+                //std::cout<<"VolumeRendererManagerData::setVolumeOpacity "<<greylevel[i]<<" "<< value[i]<<std::endl;
         }
     _newvol->Update();
 }
 
+/**
+**     Volume Opacity
+**/
+void VolumeRendererManagerData::setVolumeOpacity(std::vector<double> greylevel,double value){
+    std::vector<double> valuevector;
+    for(unsigned i = 0; i < greylevel.size(); i++){
+        valuevector.push_back(value);
+    }
+    setVolumeOpacity(greylevel, valuevector);
+}
+
 /**
 **     Check if the variables are setted correctly
 **/
@@ -100,7 +249,7 @@ void VolumeRendererManagerData::checkInvariant()throw (char *){
         if(!_MIPFunction){
                 throw "No MIP function initialized";
         }
-        if(!_volumeMapper){
+        if(!VolumeMapper){
                 throw "No volume mapper initialized";
         }
 }
@@ -143,17 +292,69 @@ 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::changeCompositeMIPFunction(int function){
+        //checkInvariant();
+        if(VolumeMapper){
+#if VTK_MAJOR_VERSION >= 5
+    #if VTK_MINOR_VERSION >= 6
+            if(dynamic_cast<vtkVolumeRayCastMapper*>(VolumeMapper)){
+                vtkVolumeRayCastMapper* volumemapper = dynamic_cast<vtkVolumeRayCastMapper*>(VolumeMapper);
+                if(function == 0){
+                        volumemapper->SetVolumeRayCastFunction(_compositeFunction);
+                }else{
+                        volumemapper->SetVolumeRayCastFunction(_MIPFunction);
+                }
+            }else if(dynamic_cast<vtkGPUVolumeRayCastMapper*>(VolumeMapper)){
+                vtkGPUVolumeRayCastMapper* volumemapper = dynamic_cast<vtkGPUVolumeRayCastMapper*>(VolumeMapper);
+                if(function == 0){
+                    volumemapper->SetBlendModeToComposite();
+                }else if(function == 1){
+                    volumemapper->SetBlendModeToMaximumIntensity();
+                }else if(function == 2){
+                    volumemapper->SetBlendModeToMinimumIntensity();
+                }
+            }
+    #endif
+#else
+            vtkGPUVolumeRayCastMapper* volumemapper = dynamic_cast<vtkGPUVolumeRayCastMapper*>(VolumeMapper);
+            if(function == 0){
+                volumemapper->SetBlendModeToComposite();
+            }else if(function == 1){
+                volumemapper->SetBlendModeToMaximumIntensity();
+            }else if(function == 2){
+                volumemapper->SetBlendModeToMinimumIntensity();
+            }
+#endif
         }
 
 }
 
 void VolumeRendererManagerData::SetLookupTable(vtkLookupTable* lookuptable){
-     _ctfun->RemoveAllPoints();
-     _ctfun->DeepCopy((vtkColorTransferFunction*)lookuptable);
+    if(lookuptable){
+         _ctfun->RemoveAllPoints();
+         vtkColorTransferFunction* colort = (vtkColorTransferFunction*)lookuptable;
+         for(int i = 0; i < colort->GetSize(); i++){
+             double val[6];
+             colort->GetNodeValue(i, val);
+             cout<< "JPRG::VolumeRendererManagerData::SetLookupTable::"<<val[0]<<" "<< val[1]<<" "<<val[2]<<" "<< val[3]<<endl;
+             _ctfun->AddRGBPoint(val[0], val[1], val[2], val[3]);
+         }
+         _newvol->Update();
+    }
+
+}
+
+void VolumeRendererManagerData::changeInterpolationType(int type){
+    //checkInvariant();
+    if(type == 0){
+        _volumeProperty->SetInterpolationTypeToLinear();
+    }else if(type == 1){
+        _volumeProperty->SetInterpolationTypeToNearest();
+    }
+}
+
+void VolumeRendererManagerData::SetColorTransferFunction(int i, vtkColorTransferFunction* colorf){
+
+    _volumeProperty->SetColor(i, colorf);
+
 }