packages/wxvtk/src/bbwxvtkViewer2D.cxx

   1 /*=========================================================================
   2
   3   Program:   bbtk
   4   Module:    $RCSfile: bbwxvtkViewer2D.cxx,v $
   5   Language:  C++
   6   Date:      $Date: 2008/12/09 12:54:07 $
   7   Version:   $Revision: 1.29 $
   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 = wxvtkImageViewer2::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     imageViewer->GetRenderer()->SetBackground(0.1,0.1,0.2);
  98
  99   }
 100   //-------------------------------------------------------------------------
 101
 102   Viewer2DWidget::~Viewer2DWidget()
 103   {
 104     //    delete wxvtkrenderwindowinteractor;
 105     // pourquoi pas wxvtkrenderwindowinteractor->Delete(); // JP
 106     // LG : because it has been allocated with operator new not method ::New
 107     // we want to force its destruction without taking care of vtk ref counting
 108     // on the contrary the following vtk objects are deleted with vtk Delete
 109     //    std::cout << "Viewer2DWidget::~Viewer2DWidget()" << std::endl;
 110     mDefaultImage->Delete();
 111
 112     if (backImageData)
 113       {
 114         //      backImageData->UnRegister(NULL);
 115       }
 116     if (imageViewer)
 117       {
 118         imageViewer->Delete();
 119       }
 120     delete wxvtkrenderwindowinteractor;
 121     //    std::cout << "Viewer2DWidget::~Viewer2DWidget() OK" << std::endl;
 122   }
 123
 124   //-------------------------------------------------------------------------
 125
 126
 127
 128
 129   void Viewer2DWidget::UpdateView()
 130   {
 131     //    std::cout << "Viewer2DWidget::UpdateView() "<<mBox->bbGetFullName() << std::endl;
 132     //   std::cout << "slice="<<mBox->bbGetInputSlice()<<std::endl;
 133
 134
 135     if ( ( mBox->bbGetInputIn() == NULL ) &&
 136          ( backImageData != mDefaultImage ) )
 137       {
 138         //
 139         //      std::cout << "** Viewer2DWidget::UpdateView() : NULL Input (reset)"
 140         //                <<std::endl;
 141         backImageData = mDefaultImage;
 142         mUpdateCamera = true;
 143
 144       }
 145     else if ( ( mBox->bbGetInputIn() != NULL ) &&
 146               ( mBox->bbGetInputStatus("In") != bbtk::UPTODATE  )
 147               )
 148               //(backImageData != mBox->bbGetInputIn()) )
 149       {
 150         //      std::cout << "** Viewer2DWidget::UpdateView() : Input changed"
 151         //                <<std::endl;
 152         backImageData = mBox->bbGetInputIn();
 153         backImageData->Update();
 154         imageViewer->SetInput( backImageData );
 155         mUpdateCamera = true;
 156       }
 157
 158     if (mUpdateCamera)
 159       {
 160         //      std::cout << "** Viewer2DWidget::UpdateView() : Update Camera"
 161         //                <<std::endl;
 162         int x1,x2,y1,y2,z1,z2;
 163         double spx,spy,spz;
 164         backImageData->GetSpacing(spx,spy,spz);
 165         backImageData->GetExtent (x1,x2,y1,y2,z1,z2);
 166
 167         double *range = backImageData->GetScalarRange();
 168         imageViewer->SetColorWindow(range[1] - range[0]);
 169         imageViewer->SetColorLevel(0.5 * (range[1] + range[0]));
 170
 171         imageViewer->GetRenderer()->ResetCamera();
 172         double bounds[6];
 173         imageViewer->GetRenderer()->ComputeVisiblePropBounds(bounds);
 174         imageViewer->GetRenderer()->ResetCameraClippingRange(bounds);
 175
 176         //      bbtkMessage("Output",2,"Viewer2D '"<<mBox->bbGetName()<<"' - Image :"<<std::endl);
 177         //      std::cout << " extent  = "<< x1<<"-"<<x2<<" x "<<y1<<"-"<<y2<<" x "<<z1<<"-"<<z2<<std::endl;
 178         //      std::cout << " spacing = "<<spx << " - "<< spy << " - "<<spz<<std::endl;
 179         /*
 180         vtkCamera *camera = imageViewer->GetRenderer()->GetActiveCamera();
 181
 182         camera->SetViewUp    ( spx*0,         -spy*1,        spz*0 );
 183         camera->SetPosition  ( spx*(x1+x2)/2, spy*(y1+y2)/2, -spz*10000000 );
 184         camera->SetFocalPoint( spx*(x1+x2)/2, spy*(y1+y2)/2, spz*0 );
 185
 186         camera->ComputeViewPlaneNormal();
 187         camera->SetParallelScale(  spx*(x2-x1)/2.0 );
 188         //
 189         //      imageViewer->GetRenderer()->ResetCamera(x1,x2,y1,y2,z1,z2);
 190         mUpdateCamera = false;
 191         //      std::cout << "Viewer2DWidget::UpdateView() passe par 3" << std::endl;
 192         imageViewer->SetColorWindow(512);
 193         imageViewer->SetColorLevel(256);
 194         */
 195          /*
 196         for (int i=0;i<10;i++)
 197           for (int j=0;j<10;j++)
 198             std::cout << mBox->bbGetInputIn()->GetScalarComponentAsFloat(i,j,0,0)
 199                       << std::endl;
 200          */
 201     }
 202
 203     //  std::cout << "OW = "<< mBox->bbGetWindow() << std::endl;
 204     //  std::cout << "CW = "<< mBox->bbGetContainingWindow() << std::endl;
 205     //  std::cout << "Sh = "<< mBox->bbIsShown() << std::endl;
 206
 207     int orientation = mBox->bbGetInputOrientation();
 208     if (orientation<0 || orientation>2) {
 209       bbtkMessage("Output", 2, "Viewer2D : Orientation was not 0< <2 "<<std::endl);
 210        orientation=2; // Why not?
 211     }
 212     // mBox->bbIsShown() is necessary to be sure that the wxWindow is already
 213     // created and displayed.
 214     // Else when slice != 0 we get an X Window System error with wxGTK
 215     if (
 216         (backImageData!=NULL)&&
 217         (mBox->bbIsShown()))
 218     {
 219                 int z = mBox->bbGetInputSlice();
 220                 //      std::cout << "slice = "<<z<<std::endl;
 221                 int ext[6];
 222                 backImageData->GetWholeExtent(ext);
 223                 /*
 224                 std::cout << "ext = "<<ext[0]<<" - "<<ext[1]<<" ; "
 225                           <<ext[2]<<" - "<<ext[3]<<" ; "
 226                           <<ext[4]<<" - "<<ext[5]
 227                           <<std::endl;
 228                 */
 229             switch(orientation){
 230                case 0:
 231                 if (z<ext[0]) { z=ext[0]; }
 232                 if (z>ext[1]) { z=ext[1]; }
 233
 234                 break;
 235
 236                case 1:
 237                 if (z<ext[2]) { z=ext[2]; }
 238                 if (z>ext[3]) { z=ext[3]; }
 239                 break;
 240
 241                case 2:
 242                 if (z<ext[4]) { z=ext[4]; }
 243                 if (z>ext[5]) { z=ext[5]; }
 244                 break;
 245
 246               default:
 247                 bbtkMessage("Output", 2, "Viewer2D : Orientation was not 0< <2 (shouldn't arrive here!"<<std::endl);
 248                 z = ext[5]; // Why not?
 249            }
 250             wxvtkrenderwindowinteractor->Refresh();
 251             wxvtkrenderwindowinteractor->Render();
 252             bbtkDebugMessage("Output",3,"Viewer2D : slice = "<<z<<std::endl);
 253             imageViewer->SetSliceOrientation (orientation);
 254             imageViewer->SetSlice( z );
 255
 256     }
 257
 258
 259   }
 260
 261
 262   vtkRenderer * Viewer2DWidget::GetRenderer()
 263   {
 264           return imageViewer->GetRenderer();
 265   }
 266
 267
 268
 269   //--------------------------------------------------------------------------
 270   //-------------------------------------------------------------------------
 271   //--------------------------------------------------------------------------
 272   //--------------------------------------------------------------------------
 273
 274
 275
 276
 277   BBTK_BLACK_BOX_IMPLEMENTATION(Viewer2D,bbtk::WxBlackBox);
 278
 279
 280   void Viewer2D::Process()
 281   {
 282     Viewer2DWidget* w = (Viewer2DWidget*)bbGetOutputWidget();
 283     if (w)
 284       {
 285         w->UpdateView();
 286         bbSetOutputRenderer( ((Viewer2DWidget*)bbGetOutputWidget())->GetRenderer() );
 287       }
 288     bbSetOutputOut( bbGetInputSlice() );
 289
 290   }
 291
 292
 293   void Viewer2D::bbUserConstructor()
 294   {
 295     bbSetInputIn(NULL);
 296     bbSetInputSlice(0);
 297     bbSetOutputWidget(0);
 298     bbSetInputOrientation(2);
 299   }
 300
 301   /**
 302    * \brief  Create wxWidget .
 303    *
 304    *
 305    */
 306   void Viewer2D::CreateWidget(wxWindow* parent)
 307   {
 308     Viewer2DWidget* w = new Viewer2DWidget(this,parent);//bbGetWxParent());
 309     bbSetOutputWidget(w);
 310   }
 311
 312
 313   // This callback is *no more*
 314   // necessary to get actual processing of the view
 315   // when window is shown
 316   void  Viewer2D::OnShowWidget()
 317   {
 318     // ((Viewer2DWidget*)bbGetOutputWidget())->Update();
 319     // ((Viewer2DWidget*)bbGetOutputWidget())->UpdateView();
 320   }
 321
 322
 323 }//namespace bbtk
 324
 325 #endif
 326