]> Creatis software - bbtk.git/blob - packages/vtk/src/bbvtkImagePlanes.cxx
#3194 BBTK Feature New Normal - Update surfaceTexture box with new lookuptables
[bbtk.git] / packages / vtk / src / bbvtkImagePlanes.cxx
1 /*
2  # ---------------------------------------------------------------------
3  #
4  # Copyright (c) CREATIS (Centre de Recherche en Acquisition et Traitement de l'Image
5  #                        pour la SantÈ)
6  # Authors : Eduardo Davila, Frederic Cervenansky, Claire Mouton
7  # Previous Authors : Laurent Guigues, Jean-Pierre Roux
8  # CreaTools website : www.creatis.insa-lyon.fr/site/fr/creatools_accueil
9  #
10  #  This software is governed by the CeCILL-B license under French law and
11  #  abiding by the rules of distribution of free software. You can  use,
12  #  modify and/ or redistribute the software under the terms of the CeCILL-B
13  #  license as circulated by CEA, CNRS and INRIA at the following URL
14  #  http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
15  #  or in the file LICENSE.txt.
16  #
17  #  As a counterpart to the access to the source code and  rights to copy,
18  #  modify and redistribute granted by the license, users are provided only
19  #  with a limited warranty  and the software's author,  the holder of the
20  #  economic rights,  and the successive licensors  have only  limited
21  #  liability.
22  #
23  #  The fact that you are presently reading this means that you have had
24  #  knowledge of the CeCILL-B license and that you accept its terms.
25  # ------------------------------------------------------------------------ */
26
27
28 /*=========================================================================
29   Program:   bbtk
30   Module:    $RCSfile: bbvtkImagePlanes.cxx,v $
31   Language:  C++
32   Date:      $Date: 2012/11/16 08:51:58 $
33   Version:   $Revision: 1.38 $
34 =========================================================================*/
35
36 /**
37  *  \file 
38  *  \brief 
39  */
40
41 #ifdef _USE_VTK_
42 #include "bbvtkImagePlanes.h"
43 #include "bbvtkPackage.h"
44 #include "vtkCellPicker.h"
45 #include "vtkProperty.h"
46 #include "vtkPolyData.h"
47
48
49 #include "vtkMetaImageWriter.h"
50 #include "vtkPNGWriter.h"
51
52 #include "bbstdCast.h"
53 #include <vtkCommand.h>
54
55 #include "vtkImageData.h"
56 //#include "vtkOutlineFilter.h"
57 //#include "vtkPolyDataMapper.h"
58 //#include "vtkActor.h"
59 #include "vtkImagePlaneWidget.h"
60 #include "vtkCellPicker.h"
61 //#include "vtkProperty.h"
62
63 //#include "vtkRenderer.h"
64 //#include "vtkCamera.h"
65
66 #include "vtkPlaneWidget.h"
67
68 #include <vtkImplicitPlaneWidget.h>
69
70 #include "bbstdRelay.h"
71
72 #include "vtkObjectFactory.h"
73 #include "vtkImageFlip.h"
74
75 #include "vtkImageReslice.h"
76 #include "vtkImageChangeInformation.h"
77
78 namespace bbstd
79 {
80
81   //====================================================================
82   BBTK_BLACK_BOX_TEMPLATE2_IMPLEMENTATION(Cast,
83                                           bbtk::AtomicBlackBox);
84   //====================================================================
85   //====================================================================
86 //  BBTK_BLACK_BOX_TEMPLATE_IMPLEMENTATION(Relay,
87 //                                       bbtk::AtomicBlackBox);
88   //====================================================================
89
90 }
91 using namespace bbstd;
92 /*
93 namespace bbtk
94 {
95         typedef vtkImageData::Pointer vtkImageDataPointer;
96   BBTK_DEFINE_HUMAN_READABLE_TYPE_NAME(vtkImageDataPointer,
97                                        "vtkImageDataPointer");
98 }
99 */
100 namespace bbvtk
101 {
102
103   //====================================================================
104   // Add the specialized adaptors to the package
105   typedef vtkImagePlaneWidget* I;
106   typedef vtkInteractorObserver* O;
107
108   BBTK_ADD_TEMPLATE2_BLACK_BOX_TO_PACKAGE(vtk,Cast,I,O);
109
110   BBTK_DEFINE_RELAY_BLACK_BOX(vtkImageDataPointer,vtk,vtkImageDataPointerRelay);
111   BBTK_BLACK_BOX_IMPLEMENTATION(vtkImageDataPointerRelay,bbtk::AtomicBlackBox);
112
113   BBTK_ADD_BLACK_BOX_TO_PACKAGE(vtk,vtkImageDataPointerRelay);
114  // BBTK_ADD_TEMPLATE_BLACK_BOX_TO_PACKAGE(vtk,Relay,vtkImageDataPointer);
115   //Pointer);
116
117 }
118
119 namespace bbvtk
120 {
121
122   //================================================================
123  class ImagePlanes::VtkCallbackType : public vtkCommand
124  {
125  public:
126    static VtkCallbackType *New()
127    {
128      return new VtkCallbackType;
129    }
130    //vtkTypeRevisionMacro(VtkCallbackType,vtkCommand);
131
132    virtual void Execute(vtkObject *caller, unsigned long, void*)
133    {
134        mBlackBox->Process();
135        mBlackBox->bbSignalOutputModification();
136    } 
137    void SetBlackBox(ImagePlanes *BB) { mBlackBox = BB;};    
138    //   void SetVtkPlaneWidget( vtkImagePlaneWidget *planeWidget );
139    VtkCallbackType() {};
140
141  private:
142    //   vtkPlaneWidget *planeWidget;
143    ImagePlanes *mBlackBox;
144  };
145   //================================================================
146
147   //vtkCxxRevisionMacro(ImagePlanes::VtkCallbackType, "$Revision: 1.38 $");
148
149   //================================================================
150
151    BBTK_ADD_BLACK_BOX_TO_PACKAGE(vtk,ImagePlanes)
152    BBTK_BLACK_BOX_IMPLEMENTATION(ImagePlanes,bbtk::AtomicBlackBox);
153
154    void ImagePlanes::bbUserSetDefaultValues() 
155    { 
156      bbSetOutputPlaneX(NULL);
157      bbSetOutputPlaneY(NULL);
158      bbSetOutputPlaneZ(NULL);
159      bbSetOutputImageX(NULL);
160      bbSetOutputImageY(NULL);
161      bbSetOutputImageZ(NULL);
162      bbSetInputIn(NULL);
163      std::vector<double> vect;
164      vect.push_back(0);
165      vect.push_back(0);
166      bbSetInputWindowLevel (vect);  
167      mVtkCallback = 0;
168
169      std::vector<int> vectpoints;
170
171      bbSetOutputPlane3Pts(0);
172      bbSetOutputImage3Pts(0);
173      bbSetInputPointsX(vectpoints);
174      bbSetInputPointsY(vectpoints);
175      bbSetInputPointsZ(vectpoints);
176
177      _imageReslicer = NULL;
178      image          = NULL;
179      _transform     = NULL;
180      _matrix        = NULL;
181            
182            bbSetInputInterpolation(1);     
183    }
184
185         
186    void ImagePlanes::bbUserInitializeProcessing() 
187    {  
188      /// CREATION DES WIDGETS
189      if (bbGetOutputPlaneX() != 0) return;
190        
191      // The shared picker enables us to use 3 planes at one time
192      // and gets the picking order right
193      vtkCellPicker* picker = vtkCellPicker::New();
194      picker->SetTolerance(0.005);
195   
196      // The 3 image plane widgets 
197      vtkImagePlaneWidget* planeWidgetX = GetPlaneWidget('x', 1, 0, 0, picker);
198      vtkImagePlaneWidget* planeWidgetY = GetPlaneWidget('y', 1, 1, 0, picker);
199      planeWidgetY->SetLookupTable(planeWidgetX->GetLookupTable());
200
201      vtkImagePlaneWidget* planeWidgetZ = GetPlaneWidget('z', 0, 0, 1, picker);     
202      planeWidgetZ->SetLookupTable(planeWidgetX->GetLookupTable());
203
204      vtkImagePlaneWidget* planeWidget3Pts = GetPlaneWidget('3', 0, 1, 1, picker);        
205      planeWidget3Pts->SetLookupTable(planeWidgetX->GetLookupTable());
206
207            // EED MPR view orientation correction..
208      vtkImageFlip  *flipYFilter = vtkImageFlip::New();
209      flipYFilter->SetFilteredAxis(1); // flip y axis
210      flipYFilter->SetInput( planeWidgetX->GetResliceOutput() );
211      flipYFilter->Update();
212
213            vtkImageChangeInformation *image = vtkImageChangeInformation::New();
214            image->SetInput(  planeWidgetY->GetResliceOutput()  );
215            image->SetOutputSpacing( 1,1,1 );       
216            image->CenterImageOn();
217            image->Update();
218            _imageTransform = vtkTransform::New();
219            vtkImageReslice *slicer =vtkImageReslice::New();
220            slicer->SetInput( image->GetOutput() );
221            slicer->SetInformationInput( image->GetOutput() );
222            slicer->SetResliceTransform( _imageTransform );
223            slicer->SetOutputOrigin(0 , 0 , 0 );
224            slicer->SetInterpolationModeToNearestNeighbor();
225            slicer->Update();       
226            vtkImageChangeInformation *imageResult = vtkImageChangeInformation::New();
227            imageResult->SetInput( slicer->GetOutput() );
228            double spc[3];
229             planeWidgetY->GetResliceOutput()->GetSpacing(spc);
230            imageResult->SetOutputSpacing( spc[1], spc[0], spc[2] ); 
231            imageResult->SetOutputOrigin( 0,0,0 ); 
232                    
233            
234      bbSetOutputPlaneX(planeWidgetX);
235      bbSetOutputPlaneY(planeWidgetY);
236      bbSetOutputPlaneZ(planeWidgetZ);
237      bbSetOutputPlane3Pts(planeWidget3Pts);      
238      bbSetOutputImageX( flipYFilter->GetOutput() );             // EED MPR view orientation correction..
239      bbSetOutputImageY( imageResult->GetOutput() );                     // EED MPR view orientation correction..
240      bbSetOutputImageZ(planeWidgetZ->GetResliceOutput());
241      bbSetInputInteractor(0);
242      //bbSetOutputImage3Pts(planeWidget3Pts->GetResliceOutput());
243
244          if(picker != 0)
245      {
246                 picker->UnRegister(NULL);
247          }
248
249      mVtkCallback = VtkCallbackType::New();
250      mVtkCallback->SetBlackBox(this);
251      planeWidgetX->AddObserver(vtkCommand::InteractionEvent,mVtkCallback);
252      planeWidgetY->AddObserver(vtkCommand::InteractionEvent,mVtkCallback);
253      planeWidgetZ->AddObserver(vtkCommand::InteractionEvent,mVtkCallback);       
254
255    }
256
257 //---------------------------------------------------------------------
258   void ImagePlanes::bbUserFinalizeProcessing()
259   {
260
261     if (bbGetOutputPlaneX()) 
262       {
263
264         /*
265           bbGetOutputPlaneX()->RemoveObserver(mVtkCallback);
266           bbGetOutputPlaneY()->RemoveObserver(mVtkCallback);
267           bbGetOutputPlaneZ()->RemoveObserver(mVtkCallback);
268
269         bbGetOutputPlaneX()->Delete();
270         bbGetOutputPlaneY()->Delete();
271         bbGetOutputPlaneZ()->Delete();
272         mVtkCallback->Delete();
273         */
274         //bbGetOutputPlaneX()->SetInput(NULL);
275         //bbGetOutputPlaneY()->SetInput(NULL);
276         //bbGetOutputPlaneZ()->SetInput(NULL);
277
278       }
279   }
280   
281 //---------------------------------------------------------------------  
282   void ImagePlanes::Process()
283   {
284           
285         if (bbGetInputIn()!=0)
286         {
287                 int dim[3];
288                 int ext[6];
289                 bbGetOutputPlaneX()->GetResliceOutput()->GetWholeExtent(ext);
290                 dim[0] = ext[1]-ext[0]+1;
291                 dim[1] = ext[3]-ext[2]+1;
292                 dim[2] = ext[5]-ext[4]+1;
293                 _imageTransform->Identity();
294                 _imageTransform->PostMultiply();
295                 _imageTransform->Translate( (int)(-(dim[0]/2)) , (int)(-(dim[1]/2)) ,0);
296                 _imageTransform->RotateZ(90);
297                 
298                 
299                 if ( image != bbGetInputIn()){//bbGetInputStatus("In") != bbtk::UPTODATE ){
300                         // Input image has changed : reinitialize planes
301                         image = bbGetInputIn();                 
302
303                         // Initial values : center of the volume (in real world, not in pixels!)
304                         int xMin, xMax, yMin, yMax, zMin, zMax;
305                         bbGetInputIn()->GetExtent(xMin, xMax, yMin, yMax, zMin, zMax);
306                         double xSpacing, ySpacing, zSpacing;
307                         bbGetInputIn()->GetSpacing(xSpacing, ySpacing, zSpacing);
308
309                         bbGetOutputPlaneX()->SetInput(bbGetInputIn());
310                         bbGetOutputPlaneX()->SetPlaneOrientationToXAxes();       
311                         bbGetOutputPlaneX()->SetSlicePosition((xMax+xMin)/2.*xSpacing);
312
313                         //                 bbGetOutputPlaneX()->SetOrigin( 58*xSpacing , 80*ySpacing , 82*zSpacing );
314                         //                 bbGetOutputPlaneX()->SetPoint1( 0*xSpacing, 146*ySpacing, 186*zSpacing);
315                         //                 bbGetOutputPlaneX()->SetPoint2( 126*xSpacing, 146*ySpacing, 0*zSpacing);
316
317                         bbGetOutputPlaneY()->SetInput(bbGetInputIn());
318                         bbGetOutputPlaneY()->SetPlaneOrientationToYAxes();
319                         bbGetOutputPlaneY()->SetSlicePosition((yMax+yMin)/2.*ySpacing);
320
321                         bbGetOutputPlaneZ()->SetInput(bbGetInputIn());
322                         bbGetOutputPlaneZ()->SetPlaneOrientationToZAxes();
323                         bbGetOutputPlaneZ()->SetSlicePosition((zMax+zMin)/2.*zSpacing);
324
325                         if (bbGetInputWindowLevel()[0]!=0)
326                         {
327                                 bbGetOutputPlaneZ()->SetWindowLevel(bbGetInputWindowLevel()[0],
328                                                                 bbGetInputWindowLevel()[1]);
329                         } else {
330                                 double *range = image->GetScalarRange();
331                                 bbGetOutputPlaneZ()->SetWindowLevel(range[1]-range[0],
332                                                                0.5*(range[1]+range[0]));
333                         } // windowlevel
334                         updateInteractor();
335                         
336                         dim[0] = xMax-xMin+1;
337                         dim[1] = yMax-yMin+1;
338                         dim[2] = zMax-zMin+1;
339                         _imageTransform->Identity();
340                         _imageTransform->PostMultiply();
341                         _imageTransform->Translate( (int)(-(dim[0]/2)*(1/xSpacing)) , (int)(-(dim[2]/2)*(1/zSpacing)) ,0);
342                         _imageTransform->RotateZ(90);
343                         
344            } // image
345                 
346                         // UPDATE DES SORTIES 
347                 bbGetOutputPlaneX()->SetResliceInterpolate( bbGetInputInterpolation() );
348                 bbGetOutputPlaneY()->SetResliceInterpolate( bbGetInputInterpolation() );
349                 bbGetOutputPlaneZ()->SetResliceInterpolate( bbGetInputInterpolation() );
350
351                 
352                 bbGetOutputPlaneX()->GetResliceOutput()->Update();
353                 bbGetOutputPlaneY()->GetResliceOutput()->Update(); 
354                 bbGetOutputPlaneZ()->GetResliceOutput()->Update();               
355                 
356                 std::vector<int> pointsx = bbGetInputPointsX();
357                 std::vector<int> pointsy = bbGetInputPointsY();
358                 std::vector<int> pointsz = bbGetInputPointsZ();
359
360                 //std::cout<<pointsx.size()<<pointsy.size()<<pointsz.size()<<std::endl;
361
362                 if (pointsx.size()==pointsy.size() && pointsx.size()==pointsz.size()&&pointsx.size()>=3)
363                 {
364
365                         //Get the corresponding three points out of the vectors
366                         double origin[3];
367                         origin[0]                       = pointsx[0];
368                         origin[1]                       = pointsy[0];
369                         origin[2]                       = pointsz[0];
370
371                         double point1[3];
372                         point1[0]                       = pointsx[1];
373                         point1[1]                       = pointsy[1];
374                         point1[2]                       = pointsz[1];
375                         double point2[3];
376                         point2[0]                       = pointsx[2];
377                         point2[1]                       = pointsy[2];
378                         point2[2]                       = pointsz[2];   
379
380                         //With the three points we create the corresponding X, Y and Z vectors all orthogonal to each other
381                         double* vect1           = getNormal(makeVector(origin, point1));
382                         double* vect2           = getNormal(makeVector(origin, point2));                                
383                         double* crossp          = getCrossProduct(vect1, vect2);
384
385                         double *newx            = getCrossProduct(vect2, crossp);
386
387                         int ext[6],factor       =       0;
388                         bbGetInputIn()->GetExtent(ext);
389
390                         factor = ext[0]<ext[3]? ext[3] : ext[0];
391                         factor = factor<ext[5]? ext[5] : factor;
392
393         //for the plane widgets
394                         vtkImagePlaneWidget* plane3pts = (vtkImagePlaneWidget*)bbGetOutputPlane3Pts();
395                         plane3pts->SetInput(bbGetInputIn());                    
396                         double xSpacing, ySpacing, zSpacing;
397                         bbGetInputIn()->GetSpacing(xSpacing, ySpacing, zSpacing);
398                         plane3pts->SetOrigin(pointsx[0]*xSpacing,pointsy[0]*ySpacing,pointsz[0]*zSpacing);      
399                         plane3pts->SetPoint1((origin[0]+newx[0]*factor)*xSpacing,
400                                                                         (origin[1]+newx[1]*factor)*ySpacing,
401                                                                         (origin[2]+newx[2]*factor)*zSpacing);
402                         plane3pts->SetPoint2((origin[0]+vect2[0]*factor)*xSpacing,
403                                                                         (origin[1]+vect2[1]*factor)*ySpacing,
404                                                                         (origin[2]+vect2[2]*factor)*zSpacing);
405                         plane3pts->GetResliceOutput()->Update();
406 //To get the slice of image out of the selected volume
407                         if (_imageReslicer==NULL)
408                         {
409                                 _imageReslicer = vtkImageReslice::New();                                        
410                                 _imageReslicer->SetOutputDimensionality(2);
411                                 _transform = vtkTransform::New();
412                                 _matrix = vtkMatrix4x4::New();  
413                         } // if
414                         _imageReslicer->SetInterpolationMode( bbGetInputInterpolation() );
415                         _imageReslicer->SetInput( bbGetInputIn() );
416                         _imageReslicer->SetInformationInput(bbGetInputIn());    
417                         //fill out the information with the created vectors and using the spacing of the image
418                         _imageReslicer->SetResliceAxesDirectionCosines(newx[0]*xSpacing,newx[1]*xSpacing,newx[2]*xSpacing,
419                                                                         vect2[0]*ySpacing,vect2[1]*ySpacing,vect2[2]*ySpacing,
420                                                                         crossp[0]*zSpacing,crossp[1]*zSpacing,crossp[2]*zSpacing);                      
421                         _imageReslicer->SetResliceAxesOrigin(origin[0]*xSpacing,origin[1]*ySpacing,origin[2]*zSpacing);
422                         _imageReslicer->GetOutput()->Update();
423                         _imageReslicer->GetOutput()->UpdateInformation();
424
425                         bbSetOutputImage3Pts(_imageReslicer->GetOutput());
426
427                         _matrix->Identity();    
428
429                         _matrix->SetElement(0,0,newx[0]*xSpacing);
430                         _matrix->SetElement(1,0,newx[1]*xSpacing);
431                         _matrix->SetElement(2,0,newx[2]*xSpacing);
432                         _matrix->SetElement(0,1,vect2[0]*ySpacing);
433                         _matrix->SetElement(1,1,vect2[1]*ySpacing);
434                         _matrix->SetElement(2,1,vect2[2]*ySpacing);
435                         _matrix->SetElement(0,2,crossp[0]*zSpacing);
436                         _matrix->SetElement(1,2,crossp[1]*zSpacing);
437                         _matrix->SetElement(2,2,crossp[2]*zSpacing);
438                         _matrix->SetElement(0,3,origin[0]*xSpacing);
439                         _matrix->SetElement(1,3,origin[1]*ySpacing);
440                         _matrix->SetElement(2,3,origin[2]*zSpacing);
441
442                         _transform->SetMatrix(_matrix);
443
444                         //set the transformation out to be used by other bbBoxes
445                         bbSetOutputTransform3Pts((vtkLinearTransform*)_transform);                      
446                 }       // pointsx pointsy  pointsz
447         } // bbGetInputIn
448   }
449         
450         void ImagePlanes::updateInteractor()
451         {
452                 vtkRenderWindowInteractor* interactor = bbGetInputInteractor(); 
453                 if(interactor)
454                 {
455                         bbGetOutputPlaneX()->SetInteractor(interactor);
456                         bbGetOutputPlaneX()->EnabledOn();
457                         bbGetOutputPlaneY()->SetInteractor(interactor);
458                         bbGetOutputPlaneY()->EnabledOn();
459                         bbGetOutputPlaneZ()->SetInteractor(interactor);
460                         bbGetOutputPlaneZ()->EnabledOn();
461                         bbGetOutputPlane3Pts()->SetInteractor(interactor);
462                         bbGetOutputPlane3Pts()->EnabledOn();
463                 } // if
464         }
465
466         //-----------------------------------------------------------------     
467         void vtkImageDataPointerRelay::bbUserSetDefaultValues()
468         {
469                 
470         }
471         
472         //-----------------------------------------------------------------     
473   void vtkImageDataPointerRelay::bbUserInitializeProcessing()
474         {
475         }
476         
477         //-----------------------------------------------------------------     
478   void vtkImageDataPointerRelay::bbUserFinalizeProcessing()
479         {
480         }
481         
482   vtkImagePlaneWidget* ImagePlanes::GetPlaneWidget(unsigned char activationkey, double r, double g, double b, vtkCellPicker* picker)
483   {
484                 vtkProperty* prop1 = 0;         
485                 vtkImagePlaneWidget* planeWidget = 0;
486
487                 planeWidget = vtkImagePlaneWidget::New();
488                 planeWidget->DisplayTextOn();
489                 planeWidget->SetPicker(picker);
490                 planeWidget->SetKeyPressActivationValue(activationkey);
491                 prop1 = planeWidget->GetPlaneProperty();
492                 prop1->SetColor(r, g, b);
493
494                 return planeWidget;
495   }
496
497   double* ImagePlanes::getCrossProduct(double* vect0,double* vect1){
498         double* vectCross;
499         vectCross = new double[3];
500         vectCross[0] = vect0[1]*vect1[2]-(vect0[2]*vect1[1]);
501         vectCross[1] = -(vect0[0]*vect1[2]-(vect0[2]*vect1[0]));
502         vectCross[2] = vect0[0]*vect1[1]-(vect0[1]*vect1[0]);
503
504         return vectCross;
505   }
506 /**
507 **      Returns the magnitud of the given vector
508 **/
509   double ImagePlanes::getMagnitud(double* vect){
510
511         double mag;
512         mag = sqrt(pow(vect[0],2) + pow(vect[1],2) + pow(vect[2],2));
513         //std::cout<<"mag "<<mag <<std::endl;
514         return mag;
515   }
516 /**
517 **      returns the unitary vector of the given vector
518 **      u = 1/|vect| . vect
519 **/
520   double* ImagePlanes::getNormal(double* vect){
521
522         double* vectnorm;
523         double mag = getMagnitud(vect);
524
525         vectnorm = new double[3];
526
527         if(mag!=0){
528                 vectnorm[0] = vect[0]/mag;
529                 vectnorm[1] = vect[1]/mag;
530                 vectnorm[2] = vect[2]/mag;
531         }else{
532                 vectnorm[0] = 0;
533                 vectnorm[1] = 0;
534                 vectnorm[2] = 0;
535         }
536
537         return vectnorm;
538   }
539
540   double* ImagePlanes::makeVector(double podouble0[3], double podouble1[3]){
541         double *vect;
542         vect = new double[3];
543
544         vect[0]= podouble1[0]-podouble0[0];
545         vect[1]= podouble1[1]-podouble0[1];
546         vect[2]= podouble1[2]-podouble0[2];
547
548         return vect;
549   }
550         
551 }//namespace bbtk
552
553 #endif // _USE_VTK_