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     backImageData = mDefaultImage;
 136     imageViewer->SetInput( backImageData );
 137     mUpdateCamera = true;
 138
 139     wxBoxSizer *sizer = new wxBoxSizer(wxVERTICAL);
 140     sizer->Add( wxvtkrenderwindowinteractor, 1, wxEXPAND, 0);
 141     panel->SetSizer(sizer);
 142     panel->SetAutoLayout(true);
 143     panel->Layout();
 144
 145
 146     //  wxvtkrenderwindowinteractor->Refresh();
 147     //  wxvtkrenderwindowinteractor->Render();
 148     //  imageViewer->GetRenderer()->ResetCameraClippingRange();
 149
 150     imageViewer->GetRenderer()->SetBackground(0.1,0.1,0.2);
 151
 152   }
 153   //-------------------------------------------------------------------------
 154
 155   Viewer2DWidget::~Viewer2DWidget()
 156   {
 157     //    delete wxvtkrenderwindowinteractor;
 158     // pourquoi pas wxvtkrenderwindowinteractor->Delete(); // JP
 159     // LG : because it has been allocated with operator new not method ::New
 160     // we want to force its destruction without taking care of vtk ref counting
 161     // on the contrary the following vtk objects are deleted with vtk Delete
 162     //    std::cout << "Viewer2DWidget::~Viewer2DWidget()" << std::endl;
 163     mDefaultImage->Delete();
 164
 165     if (backImageData)
 166       {
 167         //      backImageData->UnRegister(NULL);
 168       }
 169     if (imageViewer)
 170       {
 171         imageViewer->Delete();
 172       }
 173     delete wxvtkrenderwindowinteractor;
 174     //    std::cout << "Viewer2DWidget::~Viewer2DWidget() OK" << std::endl;
 175   }
 176
 177   //-------------------------------------------------------------------------
 178
 179
 180   void Viewer2DWidget::UpdateView()
 181   {
 182     //    std::cout << "Viewer2DWidget::UpdateView() " << std::endl;
 183     //    std::cout << "slice="<<mBox->bbGetInputSlice()<<std::endl;
 184     mUpdateCamera = false; //JPR
 185     if ( ( mBox->bbGetInputIn() == NULL ) &&
 186          ( backImageData != mDefaultImage ) )
 187       {
 188         //
 189         //      std::cout << "** Viewer2DWidget::UpdateView() : NULL Input (reset)" <<std::endl;
 190         backImageData = mDefaultImage;
 191         mUpdateCamera = true;
 192       }
 193     else if ( ( mBox->bbGetInputIn() != NULL ) &&
 194               (( mBox->bbGetInputStatus("In") != bbtk::UPTODATE  ) ||
 195                ( mFirstTime))
 196               )
 197               //(backImageData != mBox->bbGetInputIn()) )
 198       {
 199         //      std::cout << "** Viewer2DWidget::UpdateView() : Input changed"  <<std::endl;
 200         mFirstTime = false;
 201         if (backImageData != mBox->bbGetInputIn()) // Don't update if just Slice changed
 202           {
 203           backImageData = mBox->bbGetInputIn();
 204 //EED 2017-01-01 Migration VTK7
 205 #if VTK_MAJOR_VERSION <= 5
 206           backImageData->Update();
 207 #else
 208         // ..
 209 #endif
 210           imageViewer->SetInput( backImageData );
 211           mUpdateCamera = true;
 212           }
 213       }
 214
 215     if (mUpdateCamera)
 216       {
 217         //      std::cout << "** Viewer2DWidget::UpdateView() : Update Camera"
 218         //                <<std::endl;
 219         int x1,x2,y1,y2,z1,z2;
 220         double spx,spy,spz;
 221         backImageData->GetSpacing(spx,spy,spz);
 222         backImageData->GetExtent (x1,x2,y1,y2,z1,z2);
 223
 224         double *range = backImageData->GetScalarRange();
 225         imageViewer->SetColorWindow(range[1] - range[0]);
 226         imageViewer->SetColorLevel(0.5 * (range[1] + range[0]));
 227
 228         imageViewer->GetRenderer()->ResetCamera();
 229         double bounds[6];
 230         imageViewer->GetRenderer()->ComputeVisiblePropBounds(bounds);
 231         imageViewer->GetRenderer()->ResetCameraClippingRange(bounds);
 232
 233         //      bbtkMessage("Output",2,"Viewer2D '"<<mBox->bbGetName()<<"' - Image :"<<std::endl);
 234         //      std::cout << " extent  = "<< x1<<"-"<<x2<<" x "<<y1<<"-"<<y2<<" x "<<z1<<"-"<<z2<<std::endl;
 235         //      std::cout << " spacing = "<<spx << " - "<< spy << " - "<<spz<<std::endl;
 236         /*
 237         vtkCamera *camera = imageViewer->GetRenderer()->GetActiveCamera();
 238
 239         camera->SetViewUp    ( spx*0,         -spy*1,        spz*0 );
 240         camera->SetPosition  ( spx*(x1+x2)/2, spy*(y1+y2)/2, -spz*10000000 );
 241         camera->SetFocalPoint( spx*(x1+x2)/2, spy*(y1+y2)/2, spz*0 );
 242
 243         camera->ComputeViewPlaneNormal();
 244         camera->SetParallelScale(  spx*(x2-x1)/2.0 );
 245         //
 246         //      imageViewer->GetRenderer()->ResetCamera(x1,x2,y1,y2,z1,z2);
 247         mUpdateCamera = false;
 248         //      std::cout << "Viewer2DWidget::UpdateView() passe par 3" << std::endl;
 249         imageViewer->SetColorWindow(512);
 250         imageViewer->SetColorLevel(256);
 251         */
 252          /*
 253         for (int i=0;i<10;i++)
 254           for (int j=0;j<10;j++)
 255             std::cout << mBox->bbGetInputIn()->GetScalarComponentAsFloat(i,j,0,0)
 256                       << std::endl;
 257          */
 258     }
 259
 260     //  std::cout << "OW = "<< mBox->bbGetWindow() << std::endl;
 261     //  std::cout << "CW = "<< mBox->bbGetContainingWindow() << std::endl;
 262     //  std::cout << "Sh = "<< mBox->bbIsShown() << std::endl;
 263
 264     bool interp = mBox->bbGetInputInterpolate();
 265     if ( interp )
 266         imageViewer->GetImageActor()->InterpolateOn();
 267     else
 268         imageViewer->GetImageActor()->InterpolateOff();
 269
 270
 271     int orientation = mBox->bbGetInputOrientation();
 272     if (orientation<0 || orientation>2) {
 273        bbtkMessage("Output", 2, "Viewer2D : Orientation was not 0< <2 "<<std::endl);
 274        orientation=2; // Why not?
 275     }
 276
 277     // mBox->bbIsShown() is necessary to be sure that the wxWindow is already
 278     // created and displayed.
 279     // Else when slice != 0 we get an X Window System error with wxGTK
 280     if (
 281         (backImageData!=NULL)&&
 282         (mBox->bbIsShown()))
 283     {
 284                 int z = mBox->bbGetInputSlice();
 285                 int ext[6];
 286
 287 //EED 2017-01-01 Migration VTK7
 288 #if VTK_MAJOR_VERSION <= 5
 289                 backImageData->GetWholeExtent(ext);
 290 #else
 291                 backImageData->GetExtent(ext);
 292 #endif
 293
 294                 /*
 295                 std::cout << "ext = "<<ext[0]<<" - "<<ext[1]<<" ; "
 296                           <<ext[2]<<" - "<<ext[3]<<" ; "
 297                           <<ext[4]<<" - "<<ext[5]
 298                           <<std::endl;
 299                 */
 300             switch(orientation){
 301                case 0:
 302                 if (z<ext[0]) { z=ext[0]; }
 303                 if (z>ext[1]) { z=ext[1]; }
 304
 305                 break;
 306
 307                case 1:
 308                 if (z<ext[2]) { z=ext[2]; }
 309                 if (z>ext[3]) { z=ext[3]; }
 310                 break;
 311
 312                case 2:
 313                 if (z<ext[4]) { z=ext[4]; }
 314                 if (z>ext[5]) { z=ext[5]; }
 315                 break;
 316
 317               default:
 318                 bbtkMessage("Output", 2, "Viewer2D : Orientation was not 0< <2 (shouldn't arrive here!"<<std::endl);
 319                 z = ext[5]; // Why not?
 320            }
 321             wxvtkrenderwindowinteractor->Refresh();
 322             wxvtkrenderwindowinteractor->Render();
 323             bbtkDebugMessage("Output",3,"Viewer2D : slice = "<<z<<std::endl);
 324             imageViewer->SetSliceOrientation (orientation);
 325             imageViewer->SetSlice( z );
 326
 327     }
 328   }
 329
 330
 331   vtkRenderer * Viewer2DWidget::GetRenderer()
 332   {
 333           return imageViewer->GetRenderer();
 334   }
 335
 336
 337
 338   //--------------------------------------------------------------------------
 339   //-------------------------------------------------------------------------
 340   //--------------------------------------------------------------------------
 341   //--------------------------------------------------------------------------
 342
 343
 344   BBTK_ADD_BLACK_BOX_TO_PACKAGE(wxvtk,Viewer2D);
 345   BBTK_BLACK_BOX_IMPLEMENTATION(Viewer2D,bbtk::WxBlackBox);
 346
 347
 348         //-----------------------------------------------------------------
 349         void Viewer2D::bbUserSetDefaultValues()
 350         {
 351                 bbSetInputIn(NULL);
 352                 bbSetInputInterpolate(true);  // JP
 353                 bbSetInputSlice(0);
 354                 bbSetOutputWidget(0);
 355                 bbSetOutputRenderer(0);
 356                 bbSetInputOrientation(2);
 357         }
 358
 359         //-----------------------------------------------------------------
 360         void Viewer2D::bbUserInitializeProcessing()
 361         {
 362         }
 363
 364         //-----------------------------------------------------------------
 365         void Viewer2D::bbUserFinalizeProcessing()
 366         {
 367         }
 368
 369         //--------------------------------------------------------------------------
 370   void Viewer2D::Process()
 371   {
 372     //    std::cout << "Viewer2D::Process" << std::endl;
 373     //    std::cout << "In="<<bbtk::GetIOStatusString(bbGetInputStatus("In")) << std::endl;
 374     //    std::cout << "Slice="<<bbtk::GetIOStatusString(bbGetInputStatus("Slice")) << std::endl;
 375     Viewer2DWidget* w = (Viewer2DWidget*)bbGetOutputWidget();
 376     if (w)
 377       {
 378         w->UpdateView();
 379       }
 380     bbSetOutputOut( bbGetInputSlice() );
 381   }
 382
 383
 384
 385
 386         //--------------------------------------------------------------------------
 387   void Viewer2D::CreateWidget(wxWindow* parent)
 388   {
 389     Viewer2DWidget* w = new Viewer2DWidget(this,parent);
 390     w->UpdateView();
 391     bbSetOutputWidget(w);
 392     bbSetOutputRenderer( ((Viewer2DWidget*)bbGetOutputWidget())->GetRenderer() );
 393   }
 394
 395
 396
 397 }//namespace bbtk
 398
 399 #endif
 400 #endif
 401