packages/wxvtk/src/bbwxvtkViewer2D.cxx

   1 /*=========================================================================
   2
   3   Program:   bbtk
   4   Module:    $RCSfile: bbwxvtkViewer2D.cxx,v $
   5   Language:  C++
   6   Date:      $Date: 2008/11/25 11:34:01 $
   7   Version:   $Revision: 1.23 $
   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
  26
  27 #include "bbwxvtkViewer2D.h"
  28 #include "bbwxvtkPackage.h"
  29 //#include <wx/dialog.h>
  30
  31 #include "vtkInteractorStyleImage.h"
  32 #include "vtkCamera.h"
  33 #include "vtkRenderer.h"
  34 #include <vtkInteractorStyleImage.h>
  35 #include <vtkCommand.h>
  36
  37
  38
  39 namespace bbwxvtk
  40 {
  41
  42   BBTK_ADD_BLACK_BOX_TO_PACKAGE(wxvtk,Viewer2D);
  43
  44
  45   Viewer2DWidget::Viewer2DWidget(Viewer2D* box,
  46                                  wxWindow *parent)
  47     :
  48     wxPanel( parent, -1, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL),
  49     mBox(box)
  50   {
  51     wxPanel *panel      = this;
  52
  53     wxvtkrenderwindowinteractor = new wxVTKRenderWindowInteractor(panel,-1);
  54     wxvtkrenderwindowinteractor->UseCaptureMouseOn();
  55
  56     imageViewer = vtkImageViewer2::New();
  57     imageViewer->SetSlice( 1 );
  58     imageViewer->SetupInteractor ( wxvtkrenderwindowinteractor );
  59
  60     mDefaultImage = vtkImageData::New();
  61     int dim[3];
  62     dim[0] = dim[1] = 32 ;
  63     dim[2] = 2;
  64     mDefaultImage->SetDimensions ( dim );
  65     mDefaultImage->SetSpacing(1,1,1);
  66     mDefaultImage->SetExtent(0,dim[0]-1,0,dim[1]-1,0,dim[2]-1);
  67     mDefaultImage->SetWholeExtent(0,dim[0]-1,0,dim[1]-1,0,dim[2]-1);
  68
  69     mDefaultImage->SetScalarTypeToUnsignedChar();
  70     mDefaultImage->AllocateScalars();
  71     for (int i=0;i<dim[0];i++)
  72       for (int j=0;j<dim[1];j++)
  73         for (int k=0;k<dim[2];k++)
  74         mDefaultImage->SetScalarComponentFromFloat(i,j,k,0,0);
  75
  76     for (int i=0;i<dim[0];i++)
  77         {
  78           mDefaultImage->SetScalarComponentFromFloat(dim[0]-i-1,i,0,0,255);
  79           mDefaultImage->SetScalarComponentFromFloat(i,i,1,0,255);
  80         }
  81
  82     backImageData = mDefaultImage;
  83     imageViewer->SetInput( backImageData );
  84     mUpdateCamera = true;
  85
  86     wxBoxSizer *sizer = new wxBoxSizer(wxVERTICAL);
  87     sizer-> Add( wxvtkrenderwindowinteractor , 1, wxEXPAND, 0);
  88     panel-> SetSizer(sizer);
  89     panel-> SetAutoLayout(true);
  90     panel-> Layout();
  91
  92
  93     wxvtkrenderwindowinteractor->Refresh();
  94     wxvtkrenderwindowinteractor->Render();
  95     imageViewer->GetRenderer()->ResetCameraClippingRange();
  96
  97
  98   }
  99   //-------------------------------------------------------------------------
 100
 101   Viewer2DWidget::~Viewer2DWidget()
 102   {
 103     //    delete wxvtkrenderwindowinteractor;
 104     // pourquoi pas wxvtkrenderwindowinteractor->Delete(); // JP
 105     // LG : because it has been allocated with operator new not method ::New
 106     // we want to force its destruction without taking care of vtk ref counting
 107     // on the contrary the following vtk objects are deleted with vtk Delete
 108     //    std::cout << "Viewer2DWidget::~Viewer2DWidget()" << std::endl;
 109     mDefaultImage->Delete();
 110
 111     if (backImageData)
 112       {
 113         //      backImageData->UnRegister(NULL);
 114       }
 115     if (imageViewer)
 116       {
 117         imageViewer->Delete();
 118       }
 119     delete wxvtkrenderwindowinteractor;
 120     //    std::cout << "Viewer2DWidget::~Viewer2DWidget() OK" << std::endl;
 121   }
 122
 123   //-------------------------------------------------------------------------
 124
 125
 126
 127
 128   void Viewer2DWidget::UpdateView()
 129   {
 130     //    std::cout << "Viewer2DWidget::UpdateView() "<<mBox->bbGetFullName() << std::endl;
 131     //   std::cout << "slice="<<mBox->bbGetInputSlice()<<std::endl;
 132
 133
 134     if ( ( mBox->bbGetInputIn() == NULL ) &&
 135          ( backImageData != mDefaultImage ) )
 136       {
 137         backImageData = mDefaultImage;
 138         mUpdateCamera = true;
 139       }
 140     else if ( ( mBox->bbGetInputIn() != NULL ) &&
 141               (backImageData != mBox->bbGetInputIn()) )
 142       {
 143         backImageData = mBox->bbGetInputIn();
 144         backImageData->Update();
 145         imageViewer->SetInput( backImageData );
 146         mUpdateCamera = true;
 147       }
 148
 149     if (mUpdateCamera)
 150       {
 151         int x1,x2,y1,y2,z1,z2;
 152         double spx,spy,spz;
 153         backImageData->GetSpacing(spx,spy,spz);
 154         backImageData->GetExtent (x1,x2,y1,y2,z1,z2);
 155
 156         //      bbtkMessage("Output",2,"Viewer2D '"<<mBox->bbGetName()<<"' - Image :"<<std::endl);
 157         //      std::cout << " extent  = "<< x1<<"-"<<x2<<" x "<<y1<<"-"<<y2<<" x "<<z1<<"-"<<z2<<std::endl;
 158         //      std::cout << " spacing = "<<spx << " - "<< spy << " - "<<spz<<std::endl;
 159
 160         vtkCamera *camera = imageViewer->GetRenderer()->GetActiveCamera();
 161
 162         camera->SetViewUp    ( spx*0,         -spy*1,        spz*0 );
 163         camera->SetPosition  ( spx*(x1+x2)/2, spy*(y1+y2)/2, -spz*10000000 );
 164         camera->SetFocalPoint( spx*(x1+x2)/2, spy*(y1+y2)/2, spz*0 );
 165
 166         camera->ComputeViewPlaneNormal();
 167         camera->SetParallelScale(  spx*(x2-x1)/2.0 );
 168         //
 169         //      imageViewer->GetRenderer()->ResetCamera(x1,x2,y1,y2,z1,z2);
 170         mUpdateCamera = false;
 171       }
 172
 173     //  std::cout << "OW = "<< mBox->bbGetWindow() << std::endl;
 174     //  std::cout << "CW = "<< mBox->bbGetContainingWindow() << std::endl;
 175     //  std::cout << "Sh = "<< mBox->bbIsShown() << std::endl;
 176
 177     int orientation = mBox->bbGetInputOrientation();
 178     if (orientation<0 || orientation>2) {
 179        bbtkMessage("Output", 2, "Viewer2D : Orientation was not 0< <2 "<<std::endl);
 180        orientation=2; // Why not?
 181     }
 182     // mBox->bbIsShown() is necessary to be sure that the wxWindow is already
 183     // created and displayed.
 184     // Else when slice != 0 we get an X Window System error with wxGTK
 185     if (
 186         (backImageData!=NULL)&&
 187         (mBox->bbIsShown()))
 188     {
 189                 int z = mBox->bbGetInputSlice();
 190                 //      std::cout << "slice = "<<z<<std::endl;
 191                 int ext[6];
 192                 backImageData->GetWholeExtent(ext);
 193
 194                 //      std::cout << "ext = "<<ext[4]<<" - "<<ext[5]<<std::endl;
 195
 196             switch(orientation){
 197                case 0:
 198                 if (z<ext[0]) { z=ext[0]; }
 199                 if (z>ext[1]) { z=ext[1]; }
 200
 201                 break;
 202
 203                case 1:
 204                 if (z<ext[2]) { z=ext[2]; }
 205                 if (z>ext[3]) { z=ext[3]; }
 206                 break;
 207
 208                case 2:
 209                 if (z<ext[4]) { z=ext[4]; }
 210                 if (z>ext[5]) { z=ext[5]; }
 211                 break;
 212
 213               default:
 214                 bbtkMessage("Output", 2, "Viewer2D : Orientation was not 0< <2 (shouldn't arrive here!"<<std::endl);
 215                 z = ext[5]; // Why not?
 216            }
 217
 218             //      std::cout << "V2D Render "<<std::endl;
 219            imageViewer->SetSliceOrientation (orientation);
 220            imageViewer->SetSlice( z );
 221             wxvtkrenderwindowinteractor->Update();
 222             wxvtkrenderwindowinteractor->Render();
 223             imageViewer->GetRenderer()->ResetCameraClippingRange();
 224
 225 /*
 226         #if (VTK_MAJOR_VERSION >= 5)
 227                 imageViewer->SetSlice( z );
 228         #else
 229                 imageViewer->SetZSlice( z );
 230         #endif
 231 */
 232 //           imageViewer->SetSliceOrientation (orientation);
 233 //         imageViewer->SetSlice( z );
 234
 235                 //imageViewer->UpdateDisplayExtent();
 236   }
 237
 238
 239 }
 240
 241
 242   vtkRenderer * Viewer2DWidget::GetRenderer()
 243   {
 244           return imageViewer->GetRenderer();
 245   }
 246
 247
 248
 249   //--------------------------------------------------------------------------
 250   //-------------------------------------------------------------------------
 251   //--------------------------------------------------------------------------
 252   //--------------------------------------------------------------------------
 253
 254
 255
 256
 257   BBTK_BLACK_BOX_IMPLEMENTATION(Viewer2D,bbtk::WxBlackBox);
 258
 259
 260   void Viewer2D::Process()
 261   {
 262     // LG : Bug on Linux (wxGTK) when initial slice != 0 on first time
 263     // (see wxvtk/bbs/appli/ExampleSimpleSlicer)
 264     // I think there is a problem with rendering before window
 265     // has been reparented ... we have to check this
 266           Viewer2DWidget* w = (Viewer2DWidget*)bbGetOutputWidget();
 267     if (w)
 268         {
 269                 w->UpdateView();
 270                 bbSetOutputRenderer( ((Viewer2DWidget*)bbGetOutputWidget())->GetRenderer() );
 271         }
 272           bbSetOutputOut( bbGetInputSlice() );
 273
 274   }
 275
 276
 277   void Viewer2D::bbUserConstructor()
 278   {
 279     bbSetInputIn(NULL);
 280     bbSetInputSlice(0);
 281     bbSetOutputWidget(0);
 282     bbSetInputOrientation(0);
 283   }
 284
 285   /**
 286    * \brief  Create wxWidget .
 287    *
 288    *
 289    */
 290   void Viewer2D::CreateWidget(wxWindow* parent)
 291   {
 292     Viewer2DWidget* w = new Viewer2DWidget(this,parent);//bbGetWxParent());
 293     bbSetOutputWidget(w);
 294   }
 295
 296
 297   // This callback is necessary to get actual processing of the view
 298   // when window is shown
 299   void  Viewer2D::OnShowWidget()
 300   {
 301     ((Viewer2DWidget*)bbGetOutputWidget())->Update();
 302     ((Viewer2DWidget*)bbGetOutputWidget())->UpdateView();
 303   }
 304
 305
 306 }//namespace bbtk
 307
 308 #endif
 309