src/creaImageIOWxViewer.cpp

   1
   2 #include <creaImageIOWxViewer.h>
   3 #include <creaImageIOSystem.h>
   4 #include <fstream>
   5 #include <vtkCamera.h>
   6 #include <vtkRenderer.h>
   7 #include <vtkImageData.h>
   8 #include <creawxVTKRenderWindowInteractor.h>
   9 #include <creaMessageManager.h>
  10 #include <stdio.h>
  11 #include <time.h>
  12
  13
  14 using namespace crea;
  15
  16 namespace creaImageIO
  17 {
  18
  19   //=====================================================================
  20
  21   //=====================================================================
  22   class WxViewerPlayer: public wxThread
  23   {
  24   public:
  25     WxViewerPlayer(WxViewer* v) :
  26       mWxViewer(v)
  27     {}
  28
  29     void* Entry();
  30     void  OnExit();
  31
  32   private:
  33
  34     WxViewer* mWxViewer;
  35   };
  36
  37   //=====================================================================
  38
  39
  40
  41
  42
  43   //=====================================================================
  44   // CTor
  45   WxViewer::WxViewer(wxWindow *parent,
  46                      wxWindowID id,
  47                      wxString title,
  48                      const wxPoint& pos,
  49                      const wxSize& size)
  50     :   wxPanel( parent,
  51                  id,
  52                  pos,
  53                  size)
  54   {
  55     wxMutexLocker lock(mMutex);
  56     GimmickDebugMessage(6,"WxViewer::WxViewer"
  57                         <<std::endl);
  58
  59     mNeedRefresh = false;
  60     mLastImageShown = NULL;
  61
  62         // previewer
  63     mInteractor = new crea::creawxVTKRenderWindowInteractor(this,-1);
  64     mInteractor->UseCaptureMouseOn();
  65
  66     mViewer     = vtkImageViewer2::New();
  67     mViewer->SetupInteractor ( mInteractor );
  68
  69     mCurrent = 0;
  70         mPlayer = 0;
  71
  72         // Grid to place checkbox and slider
  73         mflexSizer = new wxFlexGridSizer(1,2,1,1);
  74         //Slider
  75         mslide = new wxSlider(this,-1,0,0,1, wxDefaultPosition, wxSize(400,40), wxSL_HORIZONTAL | wxSL_LABELS);
  76         Connect( mslide->GetId(), wxEVT_COMMAND_SLIDER_UPDATED , (wxObjectEventFunction) &WxViewer::OnSlide );
  77         //CheckBox
  78         mcheck = new wxCheckBox(this,5123,crea::std2wx("Cine Loop"));
  79         Connect( mcheck->GetId(), wxEVT_COMMAND_CHECKBOX_CLICKED , (wxObjectEventFunction) &WxViewer::OnCineLoop );
  80         mcheck->SetValue(false);
  81         mflexSizer->Add(mcheck,0, wxFIXED_MINSIZE);
  82         mflexSizer-> Add( mslide,1,wxALIGN_CENTER | wxFIXED_MINSIZE );
  83
  84         // Sizer for Previewer and GridSizer
  85         mtopSizer = new wxBoxSizer(wxVERTICAL);
  86         mtopSizer->Add(mflexSizer,0);
  87     mtopSizer-> Add( mInteractor ,1,wxGROW,0);
  88         SetSizer(mtopSizer,true);
  89
  90         Update();
  91     Layout();
  92   }
  93   //=====================================================================
  94
  95   //=====================================================================
  96   /// Destructor
  97   WxViewer::~WxViewer()
  98   {
  99     wxMutexLocker lock(mMutex);
 100     GimmickDebugMessage(6,"WxViewer::~WxViewer"
 101                         <<std::endl);
 102     // TO DO : desallocate cleanly
 103         if(mPlayer)
 104         {
 105                 mPlayer->Pause();
 106                 mPlayer->Delete();
 107                 mPlayer = 0;
 108         }
 109         delete mInteractor;
 110         //delete mslide;
 111         //delete mflexSizer;
 112   }
 113   //=====================================================================
 114
 115   //================================================================
 116    void WxViewer::SetImageVector(std::vector<boost::shared_ptr<ImagePointerHolder> >& pointers)
 117   {
 118         wxMutexLocker lock(mMutex);
 119         GimmickDebugMessage(6,"WxViewer::SetImageVector"<<std::endl);
 120         imagePointers=pointers;
 121
 122         mslide->SetMax(pointers.size());
 123         mslide->SetMin(1);
 124         // Refresh don't work, TO MODIFY
 125         mslide->Refresh();
 126         mslide->ClearTicks();
 127         mslide->Hide();
 128         mslide->Show();
 129         StartPlayer();
 130   }
 131
 132   //================================================================
 133
 134   void WxViewer::ShowNextImage()
 135   {
 136
 137
 138           mMutex.Unlock();
 139         wxMutexLocker lock(mMutex);
 140
 141
 142     GimmickMessage(2,"WxViewer::ShowNextImage() "
 143                    <<mCurrent+1<<"/"
 144                    <<imagePointers.size()<<std::endl);
 145
 146     if(imagePointers.size()>0)
 147         {
 148                 if (mCurrent<imagePointers.size())
 149                 {
 150                         boost::shared_ptr<ImagePointerHolder> iph = imagePointers[mCurrent];
 151                         //ImagePointerHolder* iph= imagePointers[mCurrent];
 152                         vtkImageData* currIm=iph->Get();
 153                         ShowImage(currIm);
 154                         if ( currIm != mLastImageShown )
 155                         {
 156                                 mNeedRefresh = true;
 157                                 mLastImageShown = currIm;
 158                         }
 159                 mCurrent++;
 160             }
 161                 else
 162                 {
 163                         mCurrent = 0;
 164                         //ImagePointerHolder* iph=imagePointers[mCurrent];
 165                         boost::shared_ptr<ImagePointerHolder> iph = imagePointers[mCurrent];
 166                         vtkImageData* currIm=iph->Get();
 167                         ShowImage(currIm);
 168                         if ( currIm != mLastImageShown )
 169                         {
 170                                 mNeedRefresh = true;
 171                                 mLastImageShown = currIm;
 172                         }
 173                         mCurrent++;
 174                 }
 175         }
 176   }
 177   //================================================================
 178
 179   //=====================================================================
 180   void WxViewer::ShowImage(vtkImageData* im)
 181   {
 182     GimmickDebugMessage(6,"WxViewer::ShowImage"
 183                         <<std::endl);
 184     if (im==0) return;
 185
 186     mViewer->SetInput(im);
 187
 188     mViewer->SetSlice( 0 );
 189
 190     int x1,x2,y1,y2,z1,z2;
 191     double spx,spy,spz;
 192     im->Update();
 193
 194 //std::cout << "in WxViewer::ShowImage PrintSelf() =";
 195 //im->PrintSelf(std::cout, vtkIndent(2));
 196
 197     im->GetSpacing(spx,spy,spz);
 198     //im->GetExtent (x1,x2,y1,y2,z1,z2);  // JPR
 199     im->GetWholeExtent (x1,x2,y1,y2,z1,z2);
 200 /*
 201 std::cout << "in WxViewer::ShowImage GetWholeExtent ext =";
 202         std::cout << "   [x1]=" << x1;
 203         std::cout << "   [x2]=" << x2;
 204         std::cout << "   [y1]=" << y1;
 205         std::cout << "   [y2]=" << y2;
 206         std::cout << "   [z1]=" << z1;
 207         std::cout << "   [z2]=" << z2;
 208 std::cout << std::endl;
 209 */
 210     if ((x1!=mx1) ||
 211         (x2!=mx2) ||
 212         (y1!=my1) ||
 213         (y2!=my2) ||
 214         (z1!=mz1) ||
 215         (z2!=mz2) ||
 216         (spx!=mspx) ||
 217         (spy!=mspy) ||
 218         (spz!=mspz)
 219         )
 220       {
 221         mx1 = x1;
 222         mx2 = x2;
 223         my1 = y1;
 224         my2 = y2;
 225         mz1 = z1;
 226         mz2 = z2;
 227         mspx = spx;
 228         mspy = spy;
 229         mspz = spz;
 230
 231         double *range = im->GetScalarRange();
 232         mViewer->SetColorWindow(range[1] - range[0]);
 233         mViewer->SetColorLevel(0.5 * (range[1] + range[0]));
 234
 235         mViewer->GetRenderer()->ResetCamera();
 236         double bounds[6];
 237
 238         mViewer->GetRenderer()->ComputeVisiblePropBounds(bounds);
 239
 240         mViewer->GetRenderer()->ResetCameraClippingRange(bounds);
 241         mViewer->GetRenderer()->SetBackground(0.1,0.1,0.2);
 242       }
 243   }
 244   //================================================================
 245
 246   //================================================================
 247   bool WxViewer::RefreshIfNecessary()
 248   {
 249     if (mNeedRefresh)
 250       {
 251         GimmickDebugMessage(10,"WxViewer : Refreshing"<<std::endl);
 252
 253         mInteractor->Render();
 254         mNeedRefresh = false;
 255         return true;
 256       }
 257     return false;
 258   }
 259   //================================================================
 260
 261   //==================================================
 262   void WxViewer::StopPlayer()
 263   {
 264           wxMutexLocker lock(mMutex);
 265           if (mPlayer==0 ) return;
 266           mPlayer->Delete();
 267           mPlayer=0;
 268           mMutex.Unlock();
 269   }
 270   //================================================================
 271
 272   //==================================================
 273   void WxViewer::StartPlayer()
 274         {
 275                 if(mcheck->IsChecked())
 276                 {
 277                         //      wxMutexLocker lock(mMutex);
 278                         if (mPlayer != 0) return;
 279                         mPlayer = new WxViewerPlayer(this);
 280                         mPlayer->Create();
 281                         mPlayer->Run();
 282                 }
 283                 else
 284                 {
 285                         ShowNextImage();
 286                 }
 287         }
 288
 289   //================================================================
 290
 291   //==================================================
 292
 293     void WxViewer::OnCineLoop(wxCommandEvent &Event)
 294         {
 295                 if(!mcheck->IsChecked())
 296                 {
 297                         mPlayer->Pause();
 298                         mPlayer->Delete();
 299                         mPlayer = 0;
 300                 }
 301                 StartPlayer();
 302         }
 303
 304  //================================================================
 305
 306   //==================================================
 307
 308         void WxViewer::OnSlide(wxCommandEvent &Event)
 309          {
 310                  mCurrent = mslide->GetValue() -1;
 311                  StartPlayer();
 312          }
 313          //================================================================
 314
 315   //==================================================
 316
 317          void WxViewer::SetValue()
 318          {
 319                  mslide->SetValue(mCurrent);
 320          }
 321
 322   //  BEGIN_EVENT_TABLE(WxGimmickFrame, wxDialog)
 323   //    END_EVENT_TABLE()
 324   //================================================================
 325
 326 //========================================================================
 327 //========================================================================
 328 //========================================================================
 329 //========================================================================
 330 //========================================================================
 331 //========================================================================
 332 //========================================================================
 333 //========================================================================
 334
 335   void*  WxViewerPlayer::Entry()
 336   {
 337
 338     GimmickDebugMessage(6,"WxViewerPlayer::Entry()"<<std::endl);
 339
 340         while(!TestDestroy())
 341             {
 342
 343                         mWxViewer->ShowNextImage();
 344                         mWxViewer->SetValue();
 345                         ::wxWakeUpIdle();
 346                         clock_t endwait;
 347                         endwait = clock () + 0.2 * CLOCKS_PER_SEC ;
 348                         while (clock() < endwait ) {}
 349
 350       }
 351     return 0;
 352   }
 353
 354   //=====================================================================
 355
 356   //=====================================================================
 357   void WxViewerPlayer::OnExit()
 358   {
 359     GimmickDebugMessage(6,"WxViewerPlayer::OnExit() "<<std::endl);
 360   }
 361
 362
 363 } // EO namespace creaImageIO
 364