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         // Refresh don't work, TO MODIFY
 124         mslide->Refresh();
 125         mslide->ClearTicks();
 126         mslide->Hide();
 127         mslide->Show();
 128         StartPlayer();
 129   }
 130
 131   //================================================================
 132
 133   void WxViewer::ShowNextImage()
 134   {
 135
 136
 137           mMutex.Unlock();
 138         wxMutexLocker lock(mMutex);
 139
 140
 141     GimmickMessage(2,"WxViewer::ShowNextImage() "
 142                    <<mCurrent+1<<"/"
 143                    <<imagePointers.size()<<std::endl);
 144
 145     if(imagePointers.size()>0)
 146         {
 147                 if (mCurrent<imagePointers.size())
 148                 {
 149                         boost::shared_ptr<ImagePointerHolder> iph = imagePointers[mCurrent];
 150                         //ImagePointerHolder* iph= imagePointers[mCurrent];
 151                         vtkImageData* currIm=iph->Get();
 152                         ShowImage(currIm);
 153                         if ( currIm != mLastImageShown )
 154                         {
 155                                 mNeedRefresh = true;
 156                                 mLastImageShown = currIm;
 157                         }
 158                 mCurrent++;
 159             }
 160                 else
 161                 {
 162                         mCurrent = 0;
 163                         //ImagePointerHolder* iph=imagePointers[mCurrent];
 164                         boost::shared_ptr<ImagePointerHolder> iph = imagePointers[mCurrent];
 165                         vtkImageData* currIm=iph->Get();
 166                         ShowImage(currIm);
 167                         if ( currIm != mLastImageShown )
 168                         {
 169                                 mNeedRefresh = true;
 170                                 mLastImageShown = currIm;
 171                         }
 172                         mCurrent++;
 173                 }
 174         }
 175   }
 176   //================================================================
 177
 178   //=====================================================================
 179   void WxViewer::ShowImage(vtkImageData* im)
 180   {
 181     GimmickDebugMessage(6,"WxViewer::ShowImage"
 182                         <<std::endl);
 183     if (im==0) return;
 184
 185     mViewer->SetInput(im);
 186
 187     mViewer->SetSlice( 0 );
 188
 189     int x1,x2,y1,y2,z1,z2;
 190     double spx,spy,spz;
 191     im->Update();
 192
 193 //std::cout << "in WxViewer::ShowImage PrintSelf() =";
 194 //im->PrintSelf(std::cout, vtkIndent(2));
 195
 196     im->GetSpacing(spx,spy,spz);
 197     //im->GetExtent (x1,x2,y1,y2,z1,z2);  // JPR
 198     im->GetWholeExtent (x1,x2,y1,y2,z1,z2);
 199 /*
 200 std::cout << "in WxViewer::ShowImage GetWholeExtent ext =";
 201         std::cout << "   [x1]=" << x1;
 202         std::cout << "   [x2]=" << x2;
 203         std::cout << "   [y1]=" << y1;
 204         std::cout << "   [y2]=" << y2;
 205         std::cout << "   [z1]=" << z1;
 206         std::cout << "   [z2]=" << z2;
 207 std::cout << std::endl;
 208 */
 209     if ((x1!=mx1) ||
 210         (x2!=mx2) ||
 211         (y1!=my1) ||
 212         (y2!=my2) ||
 213         (z1!=mz1) ||
 214         (z2!=mz2) ||
 215         (spx!=mspx) ||
 216         (spy!=mspy) ||
 217         (spz!=mspz)
 218         )
 219       {
 220         mx1 = x1;
 221         mx2 = x2;
 222         my1 = y1;
 223         my2 = y2;
 224         mz1 = z1;
 225         mz2 = z2;
 226         mspx = spx;
 227         mspy = spy;
 228         mspz = spz;
 229
 230         double *range = im->GetScalarRange();
 231         mViewer->SetColorWindow(range[1] - range[0]);
 232         mViewer->SetColorLevel(0.5 * (range[1] + range[0]));
 233
 234         mViewer->GetRenderer()->ResetCamera();
 235         double bounds[6];
 236
 237         mViewer->GetRenderer()->ComputeVisiblePropBounds(bounds);
 238
 239         mViewer->GetRenderer()->ResetCameraClippingRange(bounds);
 240         mViewer->GetRenderer()->SetBackground(0.1,0.1,0.2);
 241       }
 242   }
 243   //================================================================
 244
 245   //================================================================
 246   bool WxViewer::RefreshIfNecessary()
 247   {
 248     if (mNeedRefresh)
 249       {
 250         GimmickDebugMessage(10,"WxViewer : Refreshing"<<std::endl);
 251
 252         mInteractor->Render();
 253         mNeedRefresh = false;
 254         return true;
 255       }
 256     return false;
 257   }
 258   //================================================================
 259
 260   //==================================================
 261   void WxViewer::StopPlayer()
 262   {
 263           wxMutexLocker lock(mMutex);
 264           if (mPlayer==0 ) return;
 265           mPlayer->Delete();
 266           mPlayer=0;
 267   }
 268   //================================================================
 269
 270   //==================================================
 271   void WxViewer::StartPlayer()
 272         {
 273                 if(mcheck->IsChecked())
 274                 {
 275                         //      wxMutexLocker lock(mMutex);
 276                         if (mPlayer != 0) return;
 277                         mPlayer = new WxViewerPlayer(this);
 278                         mPlayer->Create();
 279                         mPlayer->Run();
 280                 }
 281                 else
 282                 {
 283                         ShowNextImage();
 284                 }
 285         }
 286
 287   //================================================================
 288
 289   //==================================================
 290
 291     void WxViewer::OnCineLoop(wxCommandEvent &Event)
 292         {
 293                 if(!mcheck->IsChecked())
 294                 {
 295                         mPlayer->Pause();
 296                         mPlayer->Delete();
 297                         mPlayer = 0;
 298                 }
 299                 StartPlayer();
 300         }
 301
 302  //================================================================
 303
 304   //==================================================
 305
 306         void WxViewer::OnSlide(wxCommandEvent &Event)
 307          {
 308                  mCurrent = mslide->GetValue();
 309                  StartPlayer();
 310          }
 311          //================================================================
 312
 313   //==================================================
 314
 315          void WxViewer::SetValue()
 316          {
 317                  mslide->SetValue(mCurrent);
 318          }
 319
 320   //  BEGIN_EVENT_TABLE(WxGimmickFrame, wxDialog)
 321   //    END_EVENT_TABLE()
 322   //================================================================
 323
 324 //========================================================================
 325 //========================================================================
 326 //========================================================================
 327 //========================================================================
 328 //========================================================================
 329 //========================================================================
 330 //========================================================================
 331 //========================================================================
 332
 333   void*  WxViewerPlayer::Entry()
 334   {
 335
 336     GimmickDebugMessage(6,"WxViewerPlayer::Entry()"<<std::endl);
 337
 338         while(!TestDestroy())
 339             {
 340
 341                         mWxViewer->ShowNextImage();
 342                         mWxViewer->SetValue();
 343                         ::wxWakeUpIdle();
 344                         clock_t endwait;
 345                         endwait = clock () + 0.2 * CLOCKS_PER_SEC ;
 346                         while (clock() < endwait ) {}
 347
 348       }
 349     return 0;
 350   }
 351
 352   //=====================================================================
 353
 354   //=====================================================================
 355   void WxViewerPlayer::OnExit()
 356   {
 357     GimmickDebugMessage(6,"WxViewerPlayer::OnExit() "<<std::endl);
 358   }
 359
 360
 361 } // EO namespace creaImageIO
 362