packages/wxvtk/src/bbwxvtkViewer2D.cxx

   1 /*=========================================================================
   2
   3   Program:   bbtk
   4   Module:    $RCSfile: bbwxvtkViewer2D.cxx,v $
   5   Language:  C++
   6   Date:      $Date: 2008/12/15 13:13:49 $
   7   Version:   $Revision: 1.33 $
   8
   9   Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
  10   l'Image). All rights reserved. See Doc/License.txt or
  11   http://www.creatis.insa-lyon.fr/Public/bbtk/License.html for details.
  12
  13      This software is distributed WITHOUT ANY WARRANTY; without even
  14      the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  15      PURPOSE.  See the above copyright notices for more information.
  16
  17 =========================================================================*/
  18 /**
  19  *  \file
  20  *  \brief
  21  */
  22
  23
  24 #ifdef USE_WXWIDGETS
  25 #ifdef USE_VTK
  26
  27
  28 #include "bbwxvtkViewer2D.h"
  29 #include "bbwxvtkPackage.h"
  30 //#include <wx/dialog.h>
  31
  32 #include "vtkInteractorStyleImage.h"
  33 #include "vtkCamera.h"
  34 #include "vtkRenderer.h"
  35 #include <vtkInteractorStyleImage.h>
  36 #include <vtkCommand.h>
  37
  38
  39
  40 namespace bbwxvtk
  41 {
  42
  43
  44
  45
  46   //--------------------------------------------------------------------------
  47   class Viewer2DWidget : public wxPanel
  48   {
  49   public:
  50     Viewer2DWidget(Viewer2D* box, wxWindow *parent);
  51     ~Viewer2DWidget();
  52     void UpdateView();
  53     vtkRenderer         *GetRenderer();
  54   private:
  55     bool                        mFirstTime;
  56     Viewer2D                    *mBox;
  57     wxvtkImageViewer2           *imageViewer;
  58     vtkImageData                *backImageData;
  59     vtkImageData                *mDefaultImage;
  60     wxVTKRenderWindowInteractor *wxvtkrenderwindowinteractor;
  61     bool mUpdateCamera;
  62
  63     vtkImplicitPlaneWidget * maPlane1, * maPlane2, * mbPlane1, * mbPlane2;
  64   };
  65
  66   Viewer2DWidget::Viewer2DWidget(Viewer2D* box,
  67                                  wxWindow *parent)
  68     :
  69     wxPanel( parent, -1, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL),
  70     mFirstTime(true),
  71     mBox(box)
  72   {
  73     wxPanel *panel      = this;
  74
  75     wxvtkrenderwindowinteractor = new wxVTKRenderWindowInteractor(panel,-1);
  76     wxvtkrenderwindowinteractor->UseCaptureMouseOn();
  77
  78     imageViewer = wxvtkImageViewer2::New();
  79     imageViewer->SetSlice( 1 );
  80     imageViewer->SetupInteractor ( wxvtkrenderwindowinteractor );
  81
  82     mDefaultImage = vtkImageData::New();
  83     int dim[3];
  84     dim[0] = dim[1] = 256 ;
  85     dim[2] = 2;
  86     mDefaultImage->SetDimensions ( dim );
  87     mDefaultImage->SetSpacing(1,1,1);
  88     mDefaultImage->SetExtent(0,dim[0]-1,0,dim[1]-1,0,dim[2]-1);
  89     mDefaultImage->SetWholeExtent(0,dim[0]-1,0,dim[1]-1,0,dim[2]-1);
  90
  91     mDefaultImage->SetScalarTypeToUnsignedChar();
  92     mDefaultImage->AllocateScalars();
  93     for (int i=0;i<dim[0];i++)
  94       for (int j=0;j<dim[1];j++)
  95         for (int k=0;k<dim[2];k++)
  96         mDefaultImage->SetScalarComponentFromFloat(i,j,k,0,0);
  97
  98     for (int i=0;i<dim[0];i++)
  99       {
 100         /*
 101         mDefaultImage->SetScalarComponentFromFloat(i,0,0,0,255);
 102         mDefaultImage->SetScalarComponentFromFloat(i,dim[1]-1,0,0,255);
 103         mDefaultImage->SetScalarComponentFromFloat(0,i,0,0,255);
 104         mDefaultImage->SetScalarComponentFromFloat(dim[0]-1,i,0,0,255);
 105         mDefaultImage->SetScalarComponentFromFloat(i,0,1,0,255);
 106         mDefaultImage->SetScalarComponentFromFloat(i,dim[1]-1,1,0,255);
 107         mDefaultImage->SetScalarComponentFromFloat(0,i,1,0,255);
 108         mDefaultImage->SetScalarComponentFromFloat(dim[0]-1,i,1,0,255);
 109         */
 110         mDefaultImage->SetScalarComponentFromFloat(i,i,0,0,255);
 111         mDefaultImage->SetScalarComponentFromFloat(i,dim[1]-i-1,0,0,255);
 112         mDefaultImage->SetScalarComponentFromFloat(i,i,1,0,255);
 113         mDefaultImage->SetScalarComponentFromFloat(i,dim[1]-i-1,1,0,255);
 114       }
 115
 116     backImageData = mDefaultImage;
 117     imageViewer->SetInput( backImageData );
 118     mUpdateCamera = true;
 119
 120     wxBoxSizer *sizer = new wxBoxSizer(wxVERTICAL);
 121     sizer-> Add( wxvtkrenderwindowinteractor , 1, wxEXPAND, 0);
 122     panel-> SetSizer(sizer);
 123     panel-> SetAutoLayout(true);
 124     panel-> Layout();
 125
 126
 127     //  wxvtkrenderwindowinteractor->Refresh();
 128     //  wxvtkrenderwindowinteractor->Render();
 129     //  imageViewer->GetRenderer()->ResetCameraClippingRange();
 130
 131     imageViewer->GetRenderer()->SetBackground(0.1,0.1,0.2);
 132
 133   }
 134   //-------------------------------------------------------------------------
 135
 136   Viewer2DWidget::~Viewer2DWidget()
 137   {
 138     //    delete wxvtkrenderwindowinteractor;
 139     // pourquoi pas wxvtkrenderwindowinteractor->Delete(); // JP
 140     // LG : because it has been allocated with operator new not method ::New
 141     // we want to force its destruction without taking care of vtk ref counting
 142     // on the contrary the following vtk objects are deleted with vtk Delete
 143     //    std::cout << "Viewer2DWidget::~Viewer2DWidget()" << std::endl;
 144     mDefaultImage->Delete();
 145
 146     if (backImageData)
 147       {
 148         //      backImageData->UnRegister(NULL);
 149       }
 150     if (imageViewer)
 151       {
 152         imageViewer->Delete();
 153       }
 154     delete wxvtkrenderwindowinteractor;
 155     //    std::cout << "Viewer2DWidget::~Viewer2DWidget() OK" << std::endl;
 156   }
 157
 158   //-------------------------------------------------------------------------
 159
 160
 161
 162
 163   void Viewer2DWidget::UpdateView()
 164   {
 165     //    std::cout << "Viewer2DWidget::UpdateView() " << std::endl;
 166     //   std::cout << "slice="<<mBox->bbGetInputSlice()<<std::endl;
 167
 168
 169     if ( ( mBox->bbGetInputIn() == NULL ) &&
 170          ( backImageData != mDefaultImage ) )
 171       {
 172         //
 173         //      std::cout << "** Viewer2DWidget::UpdateView() : NULL Input (reset)"               <<std::endl;
 174         backImageData = mDefaultImage;
 175         mUpdateCamera = true;
 176
 177       }
 178     else if ( ( mBox->bbGetInputIn() != NULL ) &&
 179               (( mBox->bbGetInputStatus("In") != bbtk::UPTODATE  ) ||
 180                ( mFirstTime))
 181               )
 182               //(backImageData != mBox->bbGetInputIn()) )
 183       {
 184
 185         //      std::cout << "** Viewer2DWidget::UpdateView() : Input changed"                    <<std::endl;
 186         mFirstTime = false;
 187         backImageData = mBox->bbGetInputIn();
 188         backImageData->Update();
 189         imageViewer->SetInput( backImageData );
 190         mUpdateCamera = true;
 191       }
 192
 193     if (mUpdateCamera)
 194       {
 195         //      std::cout << "** Viewer2DWidget::UpdateView() : Update Camera"
 196         //                <<std::endl;
 197         int x1,x2,y1,y2,z1,z2;
 198         double spx,spy,spz;
 199         backImageData->GetSpacing(spx,spy,spz);
 200         backImageData->GetExtent (x1,x2,y1,y2,z1,z2);
 201
 202         double *range = backImageData->GetScalarRange();
 203         imageViewer->SetColorWindow(range[1] - range[0]);
 204         imageViewer->SetColorLevel(0.5 * (range[1] + range[0]));
 205
 206         imageViewer->GetRenderer()->ResetCamera();
 207         double bounds[6];
 208         imageViewer->GetRenderer()->ComputeVisiblePropBounds(bounds);
 209         imageViewer->GetRenderer()->ResetCameraClippingRange(bounds);
 210
 211         //      bbtkMessage("Output",2,"Viewer2D '"<<mBox->bbGetName()<<"' - Image :"<<std::endl);
 212         //      std::cout << " extent  = "<< x1<<"-"<<x2<<" x "<<y1<<"-"<<y2<<" x "<<z1<<"-"<<z2<<std::endl;
 213         //      std::cout << " spacing = "<<spx << " - "<< spy << " - "<<spz<<std::endl;
 214         /*
 215         vtkCamera *camera = imageViewer->GetRenderer()->GetActiveCamera();
 216
 217         camera->SetViewUp    ( spx*0,         -spy*1,        spz*0 );
 218         camera->SetPosition  ( spx*(x1+x2)/2, spy*(y1+y2)/2, -spz*10000000 );
 219         camera->SetFocalPoint( spx*(x1+x2)/2, spy*(y1+y2)/2, spz*0 );
 220
 221         camera->ComputeViewPlaneNormal();
 222         camera->SetParallelScale(  spx*(x2-x1)/2.0 );
 223         //
 224         //      imageViewer->GetRenderer()->ResetCamera(x1,x2,y1,y2,z1,z2);
 225         mUpdateCamera = false;
 226         //      std::cout << "Viewer2DWidget::UpdateView() passe par 3" << std::endl;
 227         imageViewer->SetColorWindow(512);
 228         imageViewer->SetColorLevel(256);
 229         */
 230          /*
 231         for (int i=0;i<10;i++)
 232           for (int j=0;j<10;j++)
 233             std::cout << mBox->bbGetInputIn()->GetScalarComponentAsFloat(i,j,0,0)
 234                       << std::endl;
 235          */
 236     }
 237
 238     //  std::cout << "OW = "<< mBox->bbGetWindow() << std::endl;
 239     //  std::cout << "CW = "<< mBox->bbGetContainingWindow() << std::endl;
 240     //  std::cout << "Sh = "<< mBox->bbIsShown() << std::endl;
 241
 242     int orientation = mBox->bbGetInputOrientation();
 243     if (orientation<0 || orientation>2) {
 244       bbtkMessage("Output", 2, "Viewer2D : Orientation was not 0< <2 "<<std::endl);
 245        orientation=2; // Why not?
 246     }
 247     // mBox->bbIsShown() is necessary to be sure that the wxWindow is already
 248     // created and displayed.
 249     // Else when slice != 0 we get an X Window System error with wxGTK
 250     if (
 251         (backImageData!=NULL)&&
 252         (mBox->bbIsShown()))
 253     {
 254                 int z = mBox->bbGetInputSlice();
 255                 //      std::cout << "slice = "<<z<<std::endl;
 256                 int ext[6];
 257                 backImageData->GetWholeExtent(ext);
 258                 /*
 259                 std::cout << "ext = "<<ext[0]<<" - "<<ext[1]<<" ; "
 260                           <<ext[2]<<" - "<<ext[3]<<" ; "
 261                           <<ext[4]<<" - "<<ext[5]
 262                           <<std::endl;
 263                 */
 264             switch(orientation){
 265                case 0:
 266                 if (z<ext[0]) { z=ext[0]; }
 267                 if (z>ext[1]) { z=ext[1]; }
 268
 269                 break;
 270
 271                case 1:
 272                 if (z<ext[2]) { z=ext[2]; }
 273                 if (z>ext[3]) { z=ext[3]; }
 274                 break;
 275
 276                case 2:
 277                 if (z<ext[4]) { z=ext[4]; }
 278                 if (z>ext[5]) { z=ext[5]; }
 279                 break;
 280
 281               default:
 282                 bbtkMessage("Output", 2, "Viewer2D : Orientation was not 0< <2 (shouldn't arrive here!"<<std::endl);
 283                 z = ext[5]; // Why not?
 284            }
 285             wxvtkrenderwindowinteractor->Refresh();
 286             wxvtkrenderwindowinteractor->Render();
 287             bbtkDebugMessage("Output",3,"Viewer2D : slice = "<<z<<std::endl);
 288             imageViewer->SetSliceOrientation (orientation);
 289             imageViewer->SetSlice( z );
 290
 291     }
 292
 293
 294   }
 295
 296
 297   vtkRenderer * Viewer2DWidget::GetRenderer()
 298   {
 299           return imageViewer->GetRenderer();
 300   }
 301
 302
 303
 304   //--------------------------------------------------------------------------
 305   //-------------------------------------------------------------------------
 306   //--------------------------------------------------------------------------
 307   //--------------------------------------------------------------------------
 308
 309
 310   BBTK_ADD_BLACK_BOX_TO_PACKAGE(wxvtk,Viewer2D);
 311   BBTK_BLACK_BOX_IMPLEMENTATION(Viewer2D,bbtk::WxBlackBox);
 312
 313
 314   void Viewer2D::Process()
 315   {
 316     //    std::cout << "Viewer2D::Process" << std::endl;
 317     //    std::cout << "In="<<bbtk::GetIOStatusString(bbGetInputStatus("In")) << std::endl;
 318     //    std::cout << "Slice="<<bbtk::GetIOStatusString(bbGetInputStatus("Slice")) << std::endl;
 319     Viewer2DWidget* w = (Viewer2DWidget*)bbGetOutputWidget();
 320     if (w)
 321       {
 322         w->UpdateView();
 323       }
 324     bbSetOutputOut( bbGetInputSlice() );
 325
 326   }
 327
 328
 329   void Viewer2D::bbUserConstructor()
 330   {
 331     bbSetInputIn(NULL);
 332     bbSetInputSlice(0);
 333     bbSetOutputWidget(0);
 334     bbSetOutputRenderer(0);
 335     bbSetInputOrientation(2);
 336   }
 337
 338
 339   void Viewer2D::CreateWidget(wxWindow* parent)
 340   {
 341     Viewer2DWidget* w = new Viewer2DWidget(this,parent);
 342     w->UpdateView();
 343     bbSetOutputWidget(w);
 344     bbSetOutputRenderer( ((Viewer2DWidget*)bbGetOutputWidget())->GetRenderer() );
 345
 346   }
 347
 348
 349
 350 }//namespace bbtk
 351
 352 #endif
 353 #endif
 354