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