Try to display planes (from Image Planes)

[bbtk.git] / packages / vtk / src / bbvtkImagePlanes.cxx

diff --git a/packages/vtk/src/bbvtkImagePlanes.cxx b/packages/vtk/src/bbvtkImagePlanes.cxx
index 0f5e34b66c4ab462c9fb74165f0ba7d7908936c3..79ab3850b81e61815cc83328d6940080d02b13cb 100644 (file)
--- a/packages/vtk/src/bbvtkImagePlanes.cxx
+++ b/packages/vtk/src/bbvtkImagePlanes.cxx
@@ -2,8 +2,8 @@
   Program:   bbtk
   Module:    $RCSfile: bbvtkImagePlanes.cxx,v $
   Language:  C++
-  Date:      $Date: 2008/10/17 08:18:30 $
-  Version:   $Revision: 1.9 $
+  Date:      $Date: 2008/12/03 09:34:39 $
+  Version:   $Revision: 1.14 $
 =========================================================================*/
 
 /* ---------------------------------------------------------------------
@@ -38,6 +38,7 @@
 #include "bbvtkPackage.h"
 #include "vtkCellPicker.h"
 #include "vtkProperty.h"
+#include "vtkMetaImageWriter.h"
 
 #include "bbstdCast.h"
 namespace bbstd
@@ -63,28 +64,32 @@ namespace bbvtk
 namespace bbvtk
 {
 
-
    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;
      bbSetInputIn(0);
      std::vector<double> vect;
      vect.push_back(512);
      vect.push_back(256);
-     bbSetInputWindowLevel (vect);
+     bbSetInputWindowLevel (vect);  
+
+     bbSetModifiedStatus();
    }
    
-   void ImagePlanes::bbUserCopyConstructor() 
+   void ImagePlanes::bbUserCopyConstructor(bbtk::BlackBox::Pointer) 
    { 
      planeWidgetX = planeWidgetY = planeWidgetZ = 0;
+     imageX       = imageY       = imageZ       = 0;
    }
 
    void ImagePlanes::Init() 
-   { 
+   {  
      if (planeWidgetX != 0) return;
+       
      // The shared picker enables us to use 3 planes at one time
      // and gets the picking order right
      vtkCellPicker* picker = vtkCellPicker::New();
@@ -119,6 +124,12 @@ namespace bbvtk
      bbSetOutputPlaneZ(planeWidgetZ);
 
      picker->UnRegister(NULL);
+     
+     myCallbackPlane *_myCallback = myCallbackPlane::New();
+     _myCallback->SetCurrentBlackBox(this);
+     planeWidgetX->AddObserver(vtkCommand::InteractionEvent,_myCallback);
+     planeWidgetY->AddObserver(vtkCommand::InteractionEvent,_myCallback);
+     planeWidgetZ->AddObserver(vtkCommand::InteractionEvent,_myCallback);  
    }
 
 //---------------------------------------------------------------------
@@ -141,21 +152,56 @@ namespace bbvtk
         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);
+        
         planeWidgetX->SetInput(bbGetInputIn());
-        planeWidgetX->SetPlaneOrientationToXAxes();
-        planeWidgetX->SetSlicePosition((xMax+xMin)/2.); 
+        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();
+        
+        //planeWidgetX->GetResliceOutput()->Print(std::cout);
 
         planeWidgetY->SetInput(bbGetInputIn());
         planeWidgetY->SetPlaneOrientationToYAxes();
-        planeWidgetY->SetSlicePosition((yMax+yMin)/2.); 
-
+        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.); 
-
+        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();
+        
        }
    }
 }//namespace bbtk