packages/wxvtk/src/bbwxvtkViewer2D.cxx

   1 /*
   2  # ---------------------------------------------------------------------
   3  #
   4  # Copyright (c) CREATIS (Centre de Recherche en Acquisition et Traitement de l'Image
   5  #                        pour la SantÈ)
   6  # Authors : Eduardo Davila, Frederic Cervenansky, Claire Mouton
   7  # Previous Authors : Laurent Guigues, Jean-Pierre Roux
   8  # CreaTools website : www.creatis.insa-lyon.fr/site/fr/creatools_accueil
   9  #
  10  #  This software is governed by the CeCILL-B license under French law and
  11  #  abiding by the rules of distribution of free software. You can  use,
  12  #  modify and/ or redistribute the software under the terms of the CeCILL-B
  13  #  license as circulated by CEA, CNRS and INRIA at the following URL
  14  #  http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
  15  #  or in the file LICENSE.txt.
  16  #
  17  #  As a counterpart to the access to the source code and  rights to copy,
  18  #  modify and redistribute granted by the license, users are provided only
  19  #  with a limited warranty  and the software's author,  the holder of the
  20  #  economic rights,  and the successive licensors  have only  limited
  21  #  liability.
  22  #
  23  #  The fact that you are presently reading this means that you have had
  24  #  knowledge of the CeCILL-B license and that you accept its terms.
  25  # ------------------------------------------------------------------------ */
  26
  27
  28 /*=========================================================================
  29
  30   Program:   bbtk
  31   Module:    $RCSfile: bbwxvtkViewer2D.cxx,v $
  32   Language:  C++
  33   Date:      $Date: 2012/11/16 08:52:36 $
  34   Version:   $Revision: 1.39 $
  35
  36 =========================================================================*/
  37 /**
  38  *  \file
  39  *  \brief
  40  */
  41
  42
  43 #ifdef USE_WXWIDGETS
  44 #ifdef USE_VTK
  45
  46 #include "bbwxvtkViewer2D.h"
  47 #include "bbwxvtkPackage.h"
  48 //#include <wx/dialog.h>
  49
  50 #include "vtkInteractorStyleImage.h"
  51 #include "vtkCamera.h"
  52 #include "vtkRenderer.h"
  53 #include <vtkInteractorStyleImage.h>
  54 #include <vtkCommand.h>
  55 #include <vtkImageActor.h>
  56 namespace bbwxvtk
  57 {
  58   //--------------------------------------------------------------------------
  59   class Viewer2DWidget : public wxPanel
  60   {
  61   public:
  62     Viewer2DWidget(Viewer2D* box, wxWindow *parent);
  63     ~Viewer2DWidget();
  64     void UpdateView();
  65     vtkRenderer         *GetRenderer();
  66     vtkImageActor       *GetImageActor();
  67   private:
  68     bool                        mFirstTime;
  69     Viewer2D                    *mBox;
  70     wxvtkImageViewer2           *imageViewer;
  71     vtkImageData                *backImageData;
  72     vtkImageData                *mDefaultImage;
  73     wxVTKRenderWindowInteractor *wxvtkrenderwindowinteractor;
  74     bool mUpdateCamera;
  75
  76     vtkImplicitPlaneWidget * maPlane1, * maPlane2, * mbPlane1, * mbPlane2;
  77   };
  78
  79   Viewer2DWidget::Viewer2DWidget(Viewer2D* box,
  80                                  wxWindow *parent)
  81     :
  82     wxPanel( parent, -1, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL),
  83     mFirstTime(true),
  84     mBox(box)
  85   {
  86     wxPanel *panel      = this;
  87
  88     wxvtkrenderwindowinteractor = new wxVTKRenderWindowInteractor(panel,-1);
  89     wxvtkrenderwindowinteractor->UseCaptureMouseOn();
  90
  91     imageViewer = wxvtkImageViewer2::New();
  92     imageViewer->SetSlice( 1 );
  93     imageViewer->SetupInteractor ( wxvtkrenderwindowinteractor );
  94
  95     mDefaultImage = vtkImageData::New();
  96     int dim[3];
  97     dim[0] = dim[1] = 256 ;
  98     dim[2] = 2;
  99     mDefaultImage->SetDimensions ( dim );
 100     mDefaultImage->SetSpacing(1,1,1);
 101     mDefaultImage->SetExtent(0,dim[0]-1,0,dim[1]-1,0,dim[2]-1);
 102
 103 //EED 2017-01-01 Migration VTK7
 104 #if VTK_MAJOR_VERSION <= 5
 105     mDefaultImage->SetWholeExtent(0,dim[0]-1,0,dim[1]-1,0,dim[2]-1);
 106     mDefaultImage->SetScalarTypeToUnsignedChar();
 107     mDefaultImage->AllocateScalars();
 108 #else
 109     mDefaultImage->AllocateScalars(VTK_UNSIGNED_CHAR,1);
 110 #endif
 111
 112     for (int i=0;i<dim[0];i++)
 113       for (int j=0;j<dim[1];j++)
 114         for (int k=0;k<dim[2];k++)
 115            mDefaultImage->SetScalarComponentFromFloat(i,j,k,0,0);
 116
 117     for (int i=0;i<dim[0];i++)
 118       {
 119         /*
 120         mDefaultImage->SetScalarComponentFromFloat(i,0,0,0,255);
 121         mDefaultImage->SetScalarComponentFromFloat(i,dim[1]-1,0,0,255);
 122         mDefaultImage->SetScalarComponentFromFloat(0,i,0,0,255);
 123         mDefaultImage->SetScalarComponentFromFloat(dim[0]-1,i,0,0,255);
 124         mDefaultImage->SetScalarComponentFromFloat(i,0,1,0,255);
 125         mDefaultImage->SetScalarComponentFromFloat(i,dim[1]-1,1,0,255);
 126         mDefaultImage->SetScalarComponentFromFloat(0,i,1,0,255);
 127         mDefaultImage->SetScalarComponentFromFloat(dim[0]-1,i,1,0,255);
 128         */
 129         mDefaultImage->SetScalarComponentFromFloat(i,i,0,0,255);
 130         mDefaultImage->SetScalarComponentFromFloat(i,dim[1]-i-1,0,0,255);
 131         mDefaultImage->SetScalarComponentFromFloat(i,i,1,0,255);
 132         mDefaultImage->SetScalarComponentFromFloat(i,dim[1]-i-1,1,0,255);
 133       }
 134
 135     mDefaultImage->Modified();
 136
 137     backImageData = mDefaultImage;
 138     imageViewer->SetInput( backImageData );
 139     mUpdateCamera = true;
 140
 141     wxBoxSizer *sizer = new wxBoxSizer(wxVERTICAL);
 142     sizer->Add( wxvtkrenderwindowinteractor, 1, wxEXPAND, 0);
 143     panel->SetSizer(sizer);
 144     panel->SetAutoLayout(true);
 145     panel->Layout();
 146
 147
 148     //  wxvtkrenderwindowinteractor->Refresh();
 149     //  wxvtkrenderwindowinteractor->Render();
 150     //  imageViewer->GetRenderer()->ResetCameraClippingRange();
 151
 152     imageViewer->GetRenderer()->SetBackground(0.1,0.1,0.2);
 153
 154   }
 155   //-------------------------------------------------------------------------
 156
 157   Viewer2DWidget::~Viewer2DWidget()
 158   {
 159     //    delete wxvtkrenderwindowinteractor;
 160     // pourquoi pas wxvtkrenderwindowinteractor->Delete(); // JP
 161     // LG : because it has been allocated with operator new not method ::New
 162     // we want to force its destruction without taking care of vtk ref counting
 163     // on the contrary the following vtk objects are deleted with vtk Delete
 164     //    std::cout << "Viewer2DWidget::~Viewer2DWidget()" << std::endl;
 165     mDefaultImage->Delete();
 166
 167     if (backImageData)
 168       {
 169         //      backImageData->UnRegister(NULL);
 170       }
 171     if (imageViewer)
 172       {
 173         imageViewer->Delete();
 174       }
 175     delete wxvtkrenderwindowinteractor;
 176     //    std::cout << "Viewer2DWidget::~Viewer2DWidget() OK" << std::endl;
 177   }
 178
 179   //-------------------------------------------------------------------------
 180
 181
 182   void Viewer2DWidget::UpdateView()
 183   {
 184     //    std::cout << "Viewer2DWidget::UpdateView() " << std::endl;
 185     //    std::cout << "slice="<<mBox->bbGetInputSlice()<<std::endl;
 186     mUpdateCamera = false; //JPR
 187     if ( ( mBox->bbGetInputIn() == NULL ) &&
 188          ( backImageData != mDefaultImage ) )
 189       {
 190         //
 191         //      std::cout << "** Viewer2DWidget::UpdateView() : NULL Input (reset)" <<std::endl;
 192         backImageData = mDefaultImage;
 193         mUpdateCamera = true;
 194       }
 195     else if ( ( mBox->bbGetInputIn() != NULL ) &&
 196               (( mBox->bbGetInputStatus("In") != bbtk::UPTODATE  ) ||
 197                ( mFirstTime))
 198               )
 199               //(backImageData != mBox->bbGetInputIn()) )
 200       {
 201         //      std::cout << "** Viewer2DWidget::UpdateView() : Input changed"  <<std::endl;
 202         mFirstTime = false;
 203         if (backImageData != mBox->bbGetInputIn()) // Don't update if just Slice changed
 204           {
 205           backImageData = mBox->bbGetInputIn();
 206 //EED 2017-01-01 Migration VTK7
 207 #if VTK_MAJOR_VERSION <= 5
 208           backImageData->Update();
 209 #else
 210         // ..
 211 #endif
 212           imageViewer->SetInput( backImageData );
 213           mUpdateCamera = true;
 214           }
 215       }
 216
 217     if (mUpdateCamera)
 218       {
 219         //      std::cout << "** Viewer2DWidget::UpdateView() : Update Camera"
 220         //                <<std::endl;
 221         int x1,x2,y1,y2,z1,z2;
 222         double spx,spy,spz;
 223         backImageData->GetSpacing(spx,spy,spz);
 224         backImageData->GetExtent (x1,x2,y1,y2,z1,z2);
 225
 226         double *range = backImageData->GetScalarRange();
 227         imageViewer->SetColorWindow(range[1] - range[0]);
 228         imageViewer->SetColorLevel(0.5 * (range[1] + range[0]));
 229
 230         imageViewer->GetRenderer()->ResetCamera();
 231         double bounds[6];
 232         imageViewer->GetRenderer()->ComputeVisiblePropBounds(bounds);
 233         imageViewer->GetRenderer()->ResetCameraClippingRange(bounds);
 234
 235         //      bbtkMessage("Output",2,"Viewer2D '"<<mBox->bbGetName()<<"' - Image :"<<std::endl);
 236         //      std::cout << " extent  = "<< x1<<"-"<<x2<<" x "<<y1<<"-"<<y2<<" x "<<z1<<"-"<<z2<<std::endl;
 237         //      std::cout << " spacing = "<<spx << " - "<< spy << " - "<<spz<<std::endl;
 238         /*
 239         vtkCamera *camera = imageViewer->GetRenderer()->GetActiveCamera();
 240
 241         camera->SetViewUp    ( spx*0,         -spy*1,        spz*0 );
 242         camera->SetPosition  ( spx*(x1+x2)/2, spy*(y1+y2)/2, -spz*10000000 );
 243         camera->SetFocalPoint( spx*(x1+x2)/2, spy*(y1+y2)/2, spz*0 );
 244
 245         camera->ComputeViewPlaneNormal();
 246         camera->SetParallelScale(  spx*(x2-x1)/2.0 );
 247         //
 248         //      imageViewer->GetRenderer()->ResetCamera(x1,x2,y1,y2,z1,z2);
 249         mUpdateCamera = false;
 250         //      std::cout << "Viewer2DWidget::UpdateView() passe par 3" << std::endl;
 251         imageViewer->SetColorWindow(512);
 252         imageViewer->SetColorLevel(256);
 253         */
 254          /*
 255         for (int i=0;i<10;i++)
 256           for (int j=0;j<10;j++)
 257             std::cout << mBox->bbGetInputIn()->GetScalarComponentAsFloat(i,j,0,0)
 258                       << std::endl;
 259          */
 260     }
 261
 262     //  std::cout << "OW = "<< mBox->bbGetWindow() << std::endl;
 263     //  std::cout << "CW = "<< mBox->bbGetContainingWindow() << std::endl;
 264     //  std::cout << "Sh = "<< mBox->bbIsShown() << std::endl;
 265
 266     bool interp = mBox->bbGetInputInterpolate();
 267     if ( interp )
 268         imageViewer->GetImageActor()->InterpolateOn();
 269     else
 270         imageViewer->GetImageActor()->InterpolateOff();
 271
 272
 273     int orientation = mBox->bbGetInputOrientation();
 274     if (orientation<0 || orientation>2) {
 275        bbtkMessage("Output", 2, "Viewer2D : Orientation was not 0< <2 "<<std::endl);
 276        orientation=2; // Why not?
 277     }
 278
 279     // mBox->bbIsShown() is necessary to be sure that the wxWindow is already
 280     // created and displayed.
 281     // Else when slice != 0 we get an X Window System error with wxGTK
 282     if (
 283         (backImageData!=NULL)&&
 284         (mBox->bbIsShown()))
 285     {
 286                 int z = mBox->bbGetInputSlice();
 287                 int ext[6];
 288
 289 //EED 2017-01-01 Migration VTK7
 290 #if VTK_MAJOR_VERSION <= 5
 291                 backImageData->GetWholeExtent(ext);
 292 #else
 293                 backImageData->GetExtent(ext);
 294 #endif
 295
 296                 /*
 297                 std::cout << "ext = "<<ext[0]<<" - "<<ext[1]<<" ; "
 298                           <<ext[2]<<" - "<<ext[3]<<" ; "
 299                           <<ext[4]<<" - "<<ext[5]
 300                           <<std::endl;
 301                 */
 302             switch(orientation){
 303                case 0:
 304                 if (z<ext[0]) { z=ext[0]; }
 305                 if (z>ext[1]) { z=ext[1]; }
 306
 307                 break;
 308
 309                case 1:
 310                 if (z<ext[2]) { z=ext[2]; }
 311                 if (z>ext[3]) { z=ext[3]; }
 312                 break;
 313
 314                case 2:
 315                 if (z<ext[4]) { z=ext[4]; }
 316                 if (z>ext[5]) { z=ext[5]; }
 317                 break;
 318
 319               default:
 320                 bbtkMessage("Output", 2, "Viewer2D : Orientation was not 0< <2 (shouldn't arrive here!"<<std::endl);
 321                 z = ext[5]; // Why not?
 322            }
 323             wxvtkrenderwindowinteractor->Refresh();
 324             wxvtkrenderwindowinteractor->Render();
 325             bbtkDebugMessage("Output",3,"Viewer2D : slice = "<<z<<std::endl);
 326             imageViewer->SetSliceOrientation (orientation);
 327             imageViewer->SetSlice( z );
 328     }
 329   }
 330
 331
 332   vtkRenderer * Viewer2DWidget::GetRenderer()
 333   {
 334           return imageViewer->GetRenderer();
 335   }
 336
 337
 338
 339   //--------------------------------------------------------------------------
 340   //-------------------------------------------------------------------------
 341   //--------------------------------------------------------------------------
 342   //--------------------------------------------------------------------------
 343
 344
 345   BBTK_ADD_BLACK_BOX_TO_PACKAGE(wxvtk,Viewer2D);
 346   BBTK_BLACK_BOX_IMPLEMENTATION(Viewer2D,bbtk::WxBlackBox);
 347
 348
 349         //-----------------------------------------------------------------
 350         void Viewer2D::bbUserSetDefaultValues()
 351         {
 352                 bbSetInputIn(NULL);
 353                 bbSetInputInterpolate(true);  // JP
 354                 bbSetInputSlice(0);
 355                 bbSetOutputWidget(0);
 356                 bbSetOutputRenderer(0);
 357                 bbSetInputOrientation(2);
 358         }
 359
 360         //-----------------------------------------------------------------
 361         void Viewer2D::bbUserInitializeProcessing()
 362         {
 363         }
 364
 365         //-----------------------------------------------------------------
 366         void Viewer2D::bbUserFinalizeProcessing()
 367         {
 368         }
 369
 370         //--------------------------------------------------------------------------
 371   void Viewer2D::Process()
 372   {
 373     //    std::cout << "Viewer2D::Process" << std::endl;
 374     //    std::cout << "In="<<bbtk::GetIOStatusString(bbGetInputStatus("In")) << std::endl;
 375     //    std::cout << "Slice="<<bbtk::GetIOStatusString(bbGetInputStatus("Slice")) << std::endl;
 376     Viewer2DWidget* w = (Viewer2DWidget*)bbGetOutputWidget();
 377     if (w)
 378       {
 379         w->UpdateView();
 380       }
 381     bbSetOutputOut( bbGetInputSlice() );
 382   }
 383
 384
 385
 386
 387         //--------------------------------------------------------------------------
 388   void Viewer2D::CreateWidget(wxWindow* parent)
 389   {
 390     Viewer2DWidget* w = new Viewer2DWidget(this,parent);
 391     w->UpdateView();
 392     bbSetOutputWidget(w);
 393     bbSetOutputRenderer( ((Viewer2DWidget*)bbGetOutputWidget())->GetRenderer() );
 394   }
 395
 396
 397
 398 }//namespace bbtk
 399
 400 #endif
 401 #endif
 402