lib/maracasVisuLib/src/interface/wxWindows/widgets/wxMaracasMultipleVolumeRendererManager.cxx

   1 /*=========================================================================
   2
   3   Program:   wxMaracas
   4   Module:    $RCSfile: wxMaracasMultipleVolumeRendererManager.cxx,v $
   5   Language:  C++
   6   Date:      $Date: 2009/11/09 19:14:56 $
   7   Version:   $Revision: 1.6 $
   8
   9   Copyright: (c) 2002, 2003
  10   License:
  11
  12      This software is distributed WITHOUT ANY WARRANTY; without even
  13      the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  14      PURPOSE.  See the above copyright notice for more information.
  15
  16 =========================================================================*/
  17 #include "wxMaracasMultipleVolumeRendererManager.h"
  18
  19 #include <vtkMetaImageReader.h>
  20 #include <vtkImageCast.h>
  21 /**
  22 **      Start of the manager class
  23 **/
  24 wxMaracasMultipleVolumeRendererManager::wxMaracasMultipleVolumeRendererManager(){
  25         _renderer = NULL;
  26         _idCount=0;
  27 }
  28 wxMaracasMultipleVolumeRendererManager::~wxMaracasMultipleVolumeRendererManager(){
  29 }
  30
  31 /**
  32 **      Sets the renderer to manage the prop3D from the view
  33 **/
  34 void wxMaracasMultipleVolumeRendererManager::setRenderer(vtkRenderer*  renderer){
  35         _renderer = renderer;
  36 }
  37
  38 /**
  39 ** Gets the renderer which manage the prop3D from the view
  40 **/
  41 vtkRenderer* wxMaracasMultipleVolumeRendererManager::getRenderer(){
  42         return _renderer;
  43 }
  44
  45 /**
  46 ** Updates Volume
  47 **/
  48 void wxMaracasMultipleVolumeRendererManager::Update(int ppid){
  49         wxMaracasMultipleVolumeRendererManagerData* data = this->getViewData(ppid);
  50         data->Update();
  51         _renderer->Render();
  52 }
  53
  54 /**
  55 **      Adds a prop3D to the manager and returns the identifier
  56 **/
  57 int wxMaracasMultipleVolumeRendererManager::addVolume(vtkImageData* vol, std::string dataname) throw(char*){
  58         checkInvariant();
  59         image = vol;
  60         if(vol != NULL){
  61                 wxMaracasMultipleVolumeRendererManagerData* data = new wxMaracasMultipleVolumeRendererManagerData(vol, dataname);
  62                 prop3Dvect.push_back(data);
  63                 _renderer->AddActor(data->getProp3D());
  64                 data->setId(_idCount);
  65                 _idCount++;
  66                 return data->getId();
  67         }else{
  68                 throw "Check mhd imagefile file or input";
  69         }
  70         return -1;
  71 }
  72 /**
  73 **      adds or removes an actor depending of the bool value
  74 **/
  75
  76 void wxMaracasMultipleVolumeRendererManager::addRemoveActor(int propid, bool addremove)  throw(char*){
  77         checkInvariant();
  78
  79         wxMaracasMultipleVolumeRendererManagerData* data = this->getViewData(propid);
  80         if(data->getProp3D()!=NULL){
  81                 if(addremove){
  82                         _renderer->AddViewProp(data->getProp3D());
  83                 }else{
  84                         _renderer->RemoveViewProp(data->getProp3D());
  85                 }
  86                 _renderer->Render();
  87         }
  88
  89 }
  90 /**
  91 **      Changes the opacity in a prop3D
  92 **/
  93 void wxMaracasMultipleVolumeRendererManager::setVolumeOpacity(int propid, std::vector<double> greylevel,std::vector<double> value)  throw(char*){
  94         checkInvariant();
  95
  96         this->getViewData(propid)->setVolumeOpacity(greylevel, value);
  97
  98         _renderer->Render();
  99
 100 }
 101
 102 /**
 103 **      Set Volume Color
 104 **/
 105 void wxMaracasMultipleVolumeRendererManager::setVolumeColor(int volid, std::vector<double> greylevel,
 106                                                                 std::vector<double> red,
 107                                                                 std::vector<double> green,
 108                                                                 std::vector<double> blue)throw(char*){
 109         checkInvariant();
 110
 111         this->getViewData(volid)->setVolumeColor(greylevel, red, green, blue);
 112
 113         _renderer->Render();
 114 }
 115
 116 vtkImageData* wxMaracasMultipleVolumeRendererManager::getImageData(std::string filename){
 117         if(filename.compare("")!= 0){
 118
 119                 vtkMetaImageReader* reader =  vtkMetaImageReader::New();
 120                 reader->SetFileName(filename.c_str());
 121                 reader->Update();
 122                 vtkImageData* img = reader->GetOutput();
 123
 124                 vtkImageCast* cast = vtkImageCast::New();
 125                 cast->SetInput(img);
 126                 cast->SetOutputScalarTypeToUnsignedShort();
 127                 cast->Update();
 128                 //reader->Delete();
 129                 //img->Delete();
 130                 return cast->GetOutput();
 131                 //return img;
 132         }
 133         return NULL;
 134 }
 135
 136 vtkImageData* wxMaracasMultipleVolumeRendererManager::getImageData(){
 137         return image;
 138 }
 139
 140 void wxMaracasMultipleVolumeRendererManager::checkInvariant()  throw(char*){
 141         if(this->_renderer==NULL){
 142                 throw "Renderer not set";
 143         }
 144 }
 145
 146 wxMaracasMultipleVolumeRendererManagerData* wxMaracasMultipleVolumeRendererManager::getViewData(int id) throw(char*){
 147     int i;
 148         for(i = 0; i < (int)(prop3Dvect.size());i++){
 149                 if(prop3Dvect[i]->getId() == id){
 150                         return prop3Dvect[i];
 151                 }
 152         }
 153         throw "id not found in the data";
 154
 155         return NULL;
 156 }
 157
 158 void wxMaracasMultipleVolumeRendererManager::deleteActor(int propid) throw (char *){
 159         checkInvariant();
 160
 161         this->addRemoveActor(propid, false);
 162
 163         int i,n;
 164         bool exit = false;
 165         for(i = 0; i < (int)(prop3Dvect.size())&&!exit;i++){
 166                 if(prop3Dvect[i]->getId() == propid){
 167                         n=i;
 168                         exit = true;
 169                 }
 170         }
 171         if(exit){
 172                 wxMaracasMultipleVolumeRendererManagerData* data = prop3Dvect[n];
 173                 int j;
 174                 for(j = i; j < (int)(prop3Dvect.size())-1;j++){
 175                         prop3Dvect[j] = prop3Dvect[j+1];
 176                 }
 177                 delete data;
 178                 prop3Dvect.pop_back();
 179         }else{
 180                 throw "id not found in the data";
 181         }
 182
 183 }
 184
 185 vtkPiecewiseFunction* wxMaracasMultipleVolumeRendererManager::GetTransferFunction(int volumeid){
 186         return getViewData(volumeid)->GetTransferFunction();
 187 }
 188 vtkColorTransferFunction* wxMaracasMultipleVolumeRendererManager::GetColorFunction(int volumeid){
 189
 190         return getViewData(volumeid)->GetColorFunction();
 191 }