packages/wxvtk/src/bbwxvtkViewer2D.cxx

   1 /*=========================================================================
   2
   3   Program:   bbtk
   4   Module:    $RCSfile: bbwxvtkViewer2D.cxx,v $
   5   Language:  C++
   6   Date:      $Date: 2008/10/10 13:57:42 $
   7   Version:   $Revision: 1.16 $
   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        orientation=2; // Why not?
 180
 181     // mBox->bbIsShown() is necessary to be sure that the wxWindow is already
 182     // created and displayed.
 183     // Else when slice != 0 we get an X Window System error with wxGTK
 184     if (
 185         (backImageData!=NULL)&&
 186         (mBox->bbIsShown()))
 187     {
 188                 int z = mBox->bbGetInputSlice();
 189                 //      std::cout << "slice = "<<z<<std::endl;
 190                 int ext[6];
 191                 backImageData->GetWholeExtent(ext);
 192
 193                 //      std::cout << "ext = "<<ext[4]<<" - "<<ext[5]<<std::endl;
 194
 195             switch(orientation){
 196                case 0:
 197                 if (z<ext[0]) { z=ext[0]; }
 198                 if (z>ext[1]) { z=ext[1]; }
 199
 200                 break;
 201
 202                case 1:
 203                 if (z<ext[2]) { z=ext[2]; }
 204                 if (z>ext[3]) { z=ext[3]; }
 205                 break;
 206
 207                case 2:
 208                 if (z<ext[4]) { z=ext[4]; }
 209                 if (z>ext[5]) { z=ext[5]; }
 210                 break;
 211
 212               default:
 213                 z = ext[5]; // Why not?
 214            }
 215
 216             wxvtkrenderwindowinteractor->Update();
 217             wxvtkrenderwindowinteractor->Render();
 218             imageViewer->GetRenderer()->ResetCameraClippingRange();
 219
 220 /*
 221         #if (VTK_MAJOR_VERSION >= 5)
 222                 imageViewer->SetSlice( z );
 223         #else
 224                 imageViewer->SetZSlice( z );
 225         #endif
 226 */
 227            imageViewer->SetSliceOrientation (orientation);
 228            imageViewer->SetSlice( z );
 229
 230                 //imageViewer->UpdateDisplayExtent();
 231   }
 232
 233
 234 }
 235
 236
 237   vtkRenderer * Viewer2DWidget::GetRenderer()
 238   {
 239           return imageViewer->GetRenderer();
 240   }
 241
 242
 243
 244   //--------------------------------------------------------------------------
 245   //-------------------------------------------------------------------------
 246   //--------------------------------------------------------------------------
 247   //--------------------------------------------------------------------------
 248
 249
 250
 251
 252   BBTK_BLACK_BOX_IMPLEMENTATION(Viewer2D,bbtk::WxBlackBox);
 253
 254
 255   void Viewer2D::Process()
 256   {
 257     // LG : Bug on Linux (wxGTK) when initial slice != 0 on first time
 258     // (see wxvtk/bbs/appli/ExampleSimpleSlicer)
 259     // I think there is a problem with rendering before window
 260     // has been reparented ... we have to check this
 261
 262     ((Viewer2DWidget*)bbGetOutputWidget())->UpdateView();
 263     bbSetOutputOut( bbGetInputSlice() );
 264     bbSetOutputRenderer( ((Viewer2DWidget*)bbGetOutputWidget())->GetRenderer() );
 265
 266   }
 267
 268
 269   void Viewer2D::bbUserConstructor()
 270   {
 271     bbSetInputIn(NULL);
 272     bbSetInputSlice(0);
 273   }
 274
 275   /**
 276    * \brief  Create wxWidget .
 277    *
 278    *
 279    */
 280   void Viewer2D::CreateWidget()
 281   {
 282     bbtkDebugMessageInc("Core",9,"Viewer2D::CreateWidget()"<<std::endl);
 283
 284     Viewer2DWidget* w = new Viewer2DWidget(this,bbGetWxParent());
 285     bbSetOutputWidget(w);
 286
 287     bbtkDebugDecTab("Core",9);
 288   }
 289
 290
 291   // This callback is necessary to get actual processing of the view
 292   // when window is shown
 293   void  Viewer2D::bbUserOnShow()
 294   {
 295
 296 #if defined(_WIN32)
 297 //    ((Viewer2DWidget*)bbGetOutputWidget())->Refresh();
 298     ((Viewer2DWidget*)bbGetOutputWidget())->Update();
 299     ((Viewer2DWidget*)bbGetOutputWidget())->UpdateView();
 300 #endif
 301     /*
 302     ((Viewer2DWidget*)bbGetOutputWidget())->Update();
 303     ((Viewer2DWidget*)bbGetOutputWidget())->UpdateView();
 304     */
 305   }
 306
 307
 308 }//namespace bbtk
 309
 310 #endif
 311