]> Creatis software - bbtk.git/blobdiff - packages/vtk/src/bbvtkImagePlanes.cxx
*** empty log message ***
[bbtk.git] / packages / vtk / src / bbvtkImagePlanes.cxx
index 79ab3850b81e61815cc83328d6940080d02b13cb..b405d90f90cc6678f15e4fe59829d818608d92fe 100644 (file)
@@ -2,8 +2,8 @@
   Program:   bbtk
   Module:    $RCSfile: bbvtkImagePlanes.cxx,v $
   Language:  C++
-  Date:      $Date: 2008/12/03 09:34:39 $
-  Version:   $Revision: 1.14 $
+  Date:      $Date: 2008/12/03 13:35:32 $
+  Version:   $Revision: 1.15 $
 =========================================================================*/
 
 /* ---------------------------------------------------------------------
 #include "bbvtkPackage.h"
 #include "vtkCellPicker.h"
 #include "vtkProperty.h"
+
 #include "vtkMetaImageWriter.h"
+#include "vtkPNGWriter.h"
 
 #include "bbstdCast.h"
+#include <vtkCommand.h>
+
+#include "vtkImageData.h"
+//#include "vtkOutlineFilter.h"
+//#include "vtkPolyDataMapper.h"
+//#include "vtkActor.h"
+#include "vtkImagePlaneWidget.h"
+#include "vtkCellPicker.h"
+//#include "vtkProperty.h"
+
+//#include "vtkRenderer.h"
+//#include "vtkCamera.h"
+
+#include "vtkPlaneWidget.h"
+
+
 namespace bbstd
 {
+
   //====================================================================
   BBTK_BLACK_BOX_TEMPLATE2_IMPLEMENTATION(Cast,
                                          bbtk::AtomicBlackBox);
@@ -52,6 +71,8 @@ namespace bbstd
 using namespace bbstd;
 namespace bbvtk
 {
+
+
   //====================================================================
   // Add the specialized adaptors to the package
   typedef vtkImagePlaneWidget* I;
@@ -64,31 +85,66 @@ namespace bbvtk
 namespace bbvtk
 {
 
+  //================================================================
+ class myCallbackPlane : public vtkCommand
+ {
+ public:
+   static myCallbackPlane *New()
+      {
+         return new myCallbackPlane;
+      }
+   virtual void Execute(vtkObject *caller, unsigned long, void*)
+   {
+       currentBlackBox->Process();
+       currentBlackBox->bbSignalOutputModification();
+   } 
+   void SetCurrentBlackBox(ImagePlanes *cBB) {currentBlackBox = cBB;};    
+   void SetVtkPlaneWidget( vtkImagePlaneWidget *planeWidget );
+   myCallbackPlane() {};
+
+ private:
+   vtkPlaneWidget *planeWidget;
+   ImagePlanes *currentBlackBox;
+ };
+  //================================================================
+
+
+
+  //================================================================
+
    BBTK_ADD_BLACK_BOX_TO_PACKAGE(vtk,ImagePlanes)
    BBTK_BLACK_BOX_IMPLEMENTATION(ImagePlanes,bbtk::AtomicBlackBox);
 
    void ImagePlanes::bbUserConstructor() 
    { 
-     planeWidgetX = planeWidgetY = planeWidgetZ = 0;
-     imageX       = imageY       = imageZ       = 0;
+     bbSetOutputPlaneX(0);
+     bbSetOutputPlaneY(0);
+     bbSetOutputPlaneZ(0);
+     bbSetOutputImageX(0);
+     bbSetOutputImageY(0);
+     bbSetOutputImageZ(0);
      bbSetInputIn(0);
      std::vector<double> vect;
-     vect.push_back(512);
-     vect.push_back(256);
+     vect.push_back(0);
+     vect.push_back(0);
      bbSetInputWindowLevel (vect);  
-
-     bbSetModifiedStatus();
    }
    
    void ImagePlanes::bbUserCopyConstructor(bbtk::BlackBox::Pointer) 
    { 
-     planeWidgetX = planeWidgetY = planeWidgetZ = 0;
-     imageX       = imageY       = imageZ       = 0;
+     bbSetOutputPlaneX(0);
+     bbSetOutputPlaneY(0);
+     bbSetOutputPlaneZ(0);
+     bbSetOutputImageX(0);
+     bbSetOutputImageY(0);
+     bbSetOutputImageZ(0);
    }
 
    void ImagePlanes::Init() 
    {  
-     if (planeWidgetX != 0) return;
+
+     /// CREATION DES WIDGETS
+     if (bbGetOutputPlaneX() != 0) return;
        
      // The shared picker enables us to use 3 planes at one time
      // and gets the picking order right
@@ -96,14 +152,14 @@ namespace bbvtk
      picker->SetTolerance(0.005);
   
      // The 3 image plane widgets 
-     planeWidgetX = vtkImagePlaneWidget::New();
+     vtkImagePlaneWidget* planeWidgetX = vtkImagePlaneWidget::New();
      planeWidgetX->DisplayTextOn();
      planeWidgetX->SetPicker(picker);
      planeWidgetX->SetKeyPressActivationValue('x');
      vtkProperty* prop1 = planeWidgetX->GetPlaneProperty();
      prop1->SetColor(1, 0, 0);
 
-     planeWidgetY = vtkImagePlaneWidget::New();
+     vtkImagePlaneWidget* planeWidgetY = vtkImagePlaneWidget::New();
      planeWidgetY->DisplayTextOn();
      planeWidgetY->SetPicker(picker);
      planeWidgetY->SetKeyPressActivationValue('y');
@@ -111,7 +167,7 @@ namespace bbvtk
      prop2->SetColor(1, 1, 0);
      planeWidgetY->SetLookupTable(planeWidgetX->GetLookupTable());
 
-     planeWidgetZ = vtkImagePlaneWidget::New();
+     vtkImagePlaneWidget* planeWidgetZ = vtkImagePlaneWidget::New();
      planeWidgetZ->DisplayTextOn();
      planeWidgetZ->SetPicker(picker);
      planeWidgetZ->SetKeyPressActivationValue('z');
@@ -122,6 +178,9 @@ namespace bbvtk
      bbSetOutputPlaneX(planeWidgetX);
      bbSetOutputPlaneY(planeWidgetY);
      bbSetOutputPlaneZ(planeWidgetZ);
+     bbSetOutputImageX(planeWidgetX->GetResliceOutput());
+     bbSetOutputImageY(planeWidgetY->GetResliceOutput());
+     bbSetOutputImageZ(planeWidgetZ->GetResliceOutput());
 
      picker->UnRegister(NULL);
      
@@ -136,10 +195,9 @@ namespace bbvtk
   void ImagePlanes::bbUserDestructor()
   {
 
-    if (planeWidgetX) planeWidgetX->UnRegister(NULL);
-    if (planeWidgetY) planeWidgetY->UnRegister(NULL);
-    if (planeWidgetZ) planeWidgetZ->UnRegister(NULL);
-
+    if (bbGetOutputPlaneX()) bbGetOutputPlaneX()->UnRegister(NULL);
+    if (bbGetOutputPlaneY()) bbGetOutputPlaneY()->UnRegister(NULL);
+    if (bbGetOutputPlaneZ()) bbGetOutputPlaneZ()->UnRegister(NULL);
   }
   
 //---------------------------------------------------------------------  
@@ -147,61 +205,53 @@ namespace bbvtk
    {
      if (bbGetInputIn()!=0)
        {
+        // Create the widgets if not already done
         Init();
 
-        int xMin, xMax, yMin, yMax, zMin, zMax;
-        bbGetInputIn()->GetExtent(xMin, xMax, yMin, yMax, zMin, zMax);
+        if ( bbGetInputIn() != image)
+          {
+            // Input image has changed : reinitialize planes
+            image = bbGetInputIn();
 
-         // Initial values : center of the volume (in real world, not in pixels!)
-         double xSpacing, ySpacing, zSpacing;
-        bbGetInputIn()->GetSpacing(xSpacing, ySpacing, zSpacing);
-        
-        planeWidgetX->SetInput(bbGetInputIn());
-        planeWidgetX->SetPlaneOrientationToXAxes();     
-        planeWidgetX->SetSlicePosition((xMax+xMin)/2.*xSpacing);
-        planeWidgetX->GetResliceOutput()->Update();
-        bbSetOutput("ImageX", planeWidgetX->GetResliceOutput());
-        
-        
-        vtkMetaImageWriter *writer = vtkMetaImageWriter::New();
-        
-        std::string newFilenameX("newFilenameX.mhd");
-        writer->SetInput(planeWidgetX->GetResliceOutput());
-        writer->SetFileName(newFilenameX.c_str());
-        writer->Update();
+            int xMin, xMax, yMin, yMax, zMin, zMax;
+            bbGetInputIn()->GetExtent(xMin, xMax, yMin, yMax, zMin, zMax);
+            
+            // Initial values : center of the volume (in real world, not in pixels!)
+            double xSpacing, ySpacing, zSpacing;
+            bbGetInputIn()->GetSpacing(xSpacing, ySpacing, zSpacing);
+            
+            bbGetOutputPlaneX()->SetInput(bbGetInputIn());
+            bbGetOutputPlaneX()->SetPlaneOrientationToXAxes();  
+            bbGetOutputPlaneX()->SetSlicePosition((xMax+xMin)/2.*xSpacing);
+            
+            bbGetOutputPlaneY()->SetInput(bbGetInputIn());
+            bbGetOutputPlaneY()->SetPlaneOrientationToYAxes();
+            bbGetOutputPlaneY()->SetSlicePosition((yMax+yMin)/2.*ySpacing);
+            
+            bbGetOutputPlaneZ()->SetInput(bbGetInputIn());
+            bbGetOutputPlaneZ()->SetPlaneOrientationToZAxes();
+            bbGetOutputPlaneZ()->SetSlicePosition((zMax+zMin)/2.*zSpacing);
+            
+            if (bbGetInputWindowLevel()[0]!=0)
+              {
+                bbGetOutputPlaneZ()->SetWindowLevel(bbGetInputWindowLevel()[0],
+                                                    bbGetInputWindowLevel()[1]);
+              }
+            else 
+              {
+                double *range = image->GetScalarRange();
+                bbGetOutputPlaneZ()->SetWindowLevel(range[1] - range[0],
+                                                    0.5*(range[1]+range[0]));
+              }
+          }
         
-        //planeWidgetX->GetResliceOutput()->Print(std::cout);
-
-        planeWidgetY->SetInput(bbGetInputIn());
-        planeWidgetY->SetPlaneOrientationToYAxes();
-        planeWidgetY->SetSlicePosition((yMax+yMin)/2.*ySpacing);
-        planeWidgetY->GetResliceOutput()->Update(); 
-        bbSetOutput("ImageY", planeWidgetY->GetResliceOutput());
-        //planeWidgetY->GetResliceOutput()->Print(std::cout);
         
-        std::string newFilenameY("newFilenameY.mhd");
-        writer->SetInput(planeWidgetY->GetResliceOutput());
-        writer->SetFileName(newFilenameY.c_str());
-        writer->Update();
-                
-        planeWidgetZ->SetInput(bbGetInputIn());
-        planeWidgetZ->SetPlaneOrientationToZAxes();
-        planeWidgetZ->SetSlicePosition((zMax+zMin)/2.*zSpacing);
-        planeWidgetZ->GetResliceOutput()->Update(); 
-        bbSetOutput("ImageZ", planeWidgetZ->GetResliceOutput());
-       // planeWidgetZ->GetResliceOutput()->Print(std::cout);
-       
-       // planeWidgetZ->SetWindowLevel(512,256);
-       
-        std::string newFilenameZ("newFilenameZ.mhd");
-        writer->SetInput(planeWidgetZ->GetResliceOutput());
-        writer->SetFileName(newFilenameZ.c_str());
-        writer->Update();
-       
-         planeWidgetZ->SetWindowLevel(bbGetInputWindowLevel()[0],bbGetInputWindowLevel()[1]);
-       
-       // writer->vtkMetaImageWriter::Delete();
+        // UPDATE DES SORTIES 
+        bbGetOutputPlaneX()->GetResliceOutput()->Update();
+        bbGetOutputPlaneY()->GetResliceOutput()->Update(); 
+        bbGetOutputPlaneZ()->GetResliceOutput()->Update(); 
         
+                
        }
    }
 }//namespace bbtk