]> Creatis software - bbtk.git/blob - packages/wxvtk/src/bbwxvtkViewer2D.cxx
Try to display planes (from Image Planes)
[bbtk.git] / packages / wxvtk / src / bbwxvtkViewer2D.cxx
1 /*=========================================================================
2                                                                                 
3   Program:   bbtk
4   Module:    $RCSfile: bbwxvtkViewer2D.cxx,v $
5   Language:  C++
6   Date:      $Date: 2008/12/03 09:34:41 $
7   Version:   $Revision: 1.26 $
8                                                                                 
9   Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
10   l'Image). All rights reserved. See Doc/License.txt or
11   http://www.creatis.insa-lyon.fr/Public/bbtk/License.html for details.
12                                                                                 
13      This software is distributed WITHOUT ANY WARRANTY; without even
14      the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
15      PURPOSE.  See the above copyright notices for more information.
16                                                                                 
17 =========================================================================*/
18 /**
19  *  \file 
20  *  \brief 
21  */
22
23
24 #ifdef _USE_WXWIDGETS_
25
26
27 #include "bbwxvtkViewer2D.h"
28 #include "bbwxvtkPackage.h"
29 //#include <wx/dialog.h>
30
31 #include "vtkInteractorStyleImage.h"
32 #include "vtkCamera.h"
33 #include "vtkRenderer.h"
34 #include <vtkInteractorStyleImage.h>
35 #include <vtkCommand.h>
36
37
38
39 namespace bbwxvtk
40 {
41
42   BBTK_ADD_BLACK_BOX_TO_PACKAGE(wxvtk,Viewer2D);
43   
44
45   Viewer2DWidget::Viewer2DWidget(Viewer2D* box,
46                                  wxWindow *parent)
47     :  
48     wxPanel( parent, -1, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL),
49     mBox(box)
50   {
51     wxPanel *panel      = this;
52
53     wxvtkrenderwindowinteractor = new wxVTKRenderWindowInteractor(panel,-1);
54     wxvtkrenderwindowinteractor->UseCaptureMouseOn();   
55
56     imageViewer = vtkImageViewer2::New();
57     imageViewer->SetSlice( 1 );
58     imageViewer->SetupInteractor ( wxvtkrenderwindowinteractor );
59     
60     mDefaultImage = vtkImageData::New();
61     int dim[3];
62     dim[0] = dim[1] = 32 ; 
63     dim[2] = 2; 
64     mDefaultImage->SetDimensions ( dim );
65     mDefaultImage->SetSpacing(1,1,1);
66     mDefaultImage->SetExtent(0,dim[0]-1,0,dim[1]-1,0,dim[2]-1);
67     mDefaultImage->SetWholeExtent(0,dim[0]-1,0,dim[1]-1,0,dim[2]-1);
68
69     mDefaultImage->SetScalarTypeToUnsignedChar();
70     mDefaultImage->AllocateScalars();    
71     for (int i=0;i<dim[0];i++) 
72       for (int j=0;j<dim[1];j++) 
73         for (int k=0;k<dim[2];k++) 
74         mDefaultImage->SetScalarComponentFromFloat(i,j,k,0,0);
75
76     for (int i=0;i<dim[0];i++) 
77         {
78           mDefaultImage->SetScalarComponentFromFloat(dim[0]-i-1,i,0,0,255);
79           mDefaultImage->SetScalarComponentFromFloat(i,i,1,0,255);
80         }
81
82     backImageData = mDefaultImage;
83     imageViewer->SetInput( backImageData );
84     mUpdateCamera = true;
85
86     wxBoxSizer *sizer = new wxBoxSizer(wxVERTICAL);
87     sizer-> Add( wxvtkrenderwindowinteractor , 1, wxEXPAND, 0);
88     panel-> SetSizer(sizer);
89     panel-> SetAutoLayout(true);
90     panel-> Layout();
91
92
93     //  wxvtkrenderwindowinteractor->Refresh();
94     //  wxvtkrenderwindowinteractor->Render();
95     //  imageViewer->GetRenderer()->ResetCameraClippingRange();
96
97   
98   }
99   //-------------------------------------------------------------------------
100   
101   Viewer2DWidget::~Viewer2DWidget()
102   {
103     //    delete wxvtkrenderwindowinteractor;
104     // pourquoi pas wxvtkrenderwindowinteractor->Delete(); // JP
105     // LG : because it has been allocated with operator new not method ::New 
106     // we want to force its destruction without taking care of vtk ref counting
107     // on the contrary the following vtk objects are deleted with vtk Delete
108     //    std::cout << "Viewer2DWidget::~Viewer2DWidget()" << std::endl;
109     mDefaultImage->Delete();
110
111     if (backImageData)
112       {
113         //      backImageData->UnRegister(NULL);
114       }
115     if (imageViewer) 
116       {
117         imageViewer->Delete();
118       }
119     delete wxvtkrenderwindowinteractor;
120     //    std::cout << "Viewer2DWidget::~Viewer2DWidget() OK" << std::endl;
121   }
122   
123   //-------------------------------------------------------------------------
124   
125
126
127
128   void Viewer2DWidget::UpdateView()
129   {
130     //    std::cout << "Viewer2DWidget::UpdateView() "<<mBox->bbGetFullName() << std::endl;
131     //   std::cout << "slice="<<mBox->bbGetInputSlice()<<std::endl;
132
133
134     if ( ( mBox->bbGetInputIn() == NULL ) &&
135          ( backImageData != mDefaultImage ) )
136       {
137         backImageData = mDefaultImage;
138         mUpdateCamera = true;
139         std::cout << "Viewer2DWidget::UpdateView() passe par 1" << std::endl;
140       }
141     else if ( ( mBox->bbGetInputIn() != NULL ) && 
142               (backImageData != mBox->bbGetInputIn()) ) 
143       {
144         backImageData = mBox->bbGetInputIn();
145         backImageData->Update();
146         imageViewer->SetInput( backImageData );
147         mUpdateCamera = true;
148                 std::cout << "Viewer2DWidget::UpdateView() passe par 2" << std::endl;
149       }
150
151     if (mUpdateCamera)
152       {
153         int x1,x2,y1,y2,z1,z2;
154         double spx,spy,spz;
155         backImageData->GetSpacing(spx,spy,spz);
156         backImageData->GetExtent (x1,x2,y1,y2,z1,z2);
157         
158         //      bbtkMessage("Output",2,"Viewer2D '"<<mBox->bbGetName()<<"' - Image :"<<std::endl);
159         //      std::cout << " extent  = "<< x1<<"-"<<x2<<" x "<<y1<<"-"<<y2<<" x "<<z1<<"-"<<z2<<std::endl;
160         //      std::cout << " spacing = "<<spx << " - "<< spy << " - "<<spz<<std::endl;
161         
162         vtkCamera *camera = imageViewer->GetRenderer()->GetActiveCamera();
163
164         camera->SetViewUp    ( spx*0,         -spy*1,        spz*0 );
165         camera->SetPosition  ( spx*(x1+x2)/2, spy*(y1+y2)/2, -spz*10000000 ); 
166         camera->SetFocalPoint( spx*(x1+x2)/2, spy*(y1+y2)/2, spz*0 ); 
167
168         camera->ComputeViewPlaneNormal();
169         camera->SetParallelScale(  spx*(x2-x1)/2.0 );
170         //   
171         //      imageViewer->GetRenderer()->ResetCamera(x1,x2,y1,y2,z1,z2);
172         mUpdateCamera = false;
173         std::cout << "Viewer2DWidget::UpdateView() passe par 3" << std::endl;
174       }
175   
176     //  std::cout << "OW = "<< mBox->bbGetWindow() << std::endl;
177     //  std::cout << "CW = "<< mBox->bbGetContainingWindow() << std::endl;
178     //  std::cout << "Sh = "<< mBox->bbIsShown() << std::endl;
179     
180     int orientation = mBox->bbGetInputOrientation();
181     if (orientation<0 || orientation>2) {
182        bbtkMessage("Output", 2, "Viewer2D : Orientation was not 0< <2 "<<std::endl);
183        orientation=2; // Why not?      
184     }
185     // mBox->bbIsShown() is necessary to be sure that the wxWindow is already 
186     // created and displayed. 
187     // Else when slice != 0 we get an X Window System error with wxGTK
188     if (
189         (backImageData!=NULL)&&
190         (mBox->bbIsShown()))
191     {
192                 int z = mBox->bbGetInputSlice();
193                 //      std::cout << "slice = "<<z<<std::endl;
194                 int ext[6];
195                 backImageData->GetWholeExtent(ext);
196
197                 //      std::cout << "ext = "<<ext[4]<<" - "<<ext[5]<<std::endl;
198
199             switch(orientation){
200                case 0:
201                 if (z<ext[0]) { z=ext[0]; }
202                 if (z>ext[1]) { z=ext[1]; }
203                  
204                 break;
205                                    
206                case 1:
207                 if (z<ext[2]) { z=ext[2]; }
208                 if (z>ext[3]) { z=ext[3]; }
209                 break;
210                                                     
211                case 2:
212                 if (z<ext[4]) { z=ext[4]; }
213                 if (z>ext[5]) { z=ext[5]; }
214                 break;
215                 
216               default:
217                 bbtkMessage("Output", 2, "Viewer2D : Orientation was not 0< <2 (shouldn't arrive here!"<<std::endl);
218                 z = ext[5]; // Why not?
219            }
220
221             //      std::cout << "V2D Render "<<std::endl;
222             //      imageViewer->SetSliceOrientation (orientation);
223             //     imageViewer->SetSlice( z );
224
225            //            wxvtkrenderwindowinteractor->Update();
226            //            wxvtkrenderwindowinteractor->Render();
227            //          imageViewer->GetRenderer()->ResetCameraClippingRange();
228  
229             //xvtkrenderwindowinteractor->Refresh();
230             wxvtkrenderwindowinteractor->Render();
231     imageViewer->GetRenderer()->ResetCameraClippingRange();
232            imageViewer->SetSliceOrientation (orientation);
233            imageViewer->SetSlice( z );
234
235 std::cout << "Viewer2DWidget::UpdateView() passe par 4" << std::endl;       
236 /*
237         #if (VTK_MAJOR_VERSION >= 5)
238                 imageViewer->SetSlice( z );
239         #else
240                 imageViewer->SetZSlice( z );
241         #endif
242 */
243 //           imageViewer->SetSliceOrientation (orientation);
244 //         imageViewer->SetSlice( z );
245                 
246                 //imageViewer->UpdateDisplayExtent();
247   } 
248
249
250 }
251
252
253   vtkRenderer * Viewer2DWidget::GetRenderer()
254   {
255           return imageViewer->GetRenderer();
256   }
257
258
259
260   //--------------------------------------------------------------------------
261   //-------------------------------------------------------------------------
262   //--------------------------------------------------------------------------
263   //--------------------------------------------------------------------------
264
265
266
267
268   BBTK_BLACK_BOX_IMPLEMENTATION(Viewer2D,bbtk::WxBlackBox);
269
270
271   void Viewer2D::Process() 
272   { 
273     // LG : Bug on Linux (wxGTK) when initial slice != 0 on first time 
274     // (see wxvtk/bbs/appli/ExampleSimpleSlicer)
275     // I think there is a problem with rendering before window 
276     // has been reparented ... we have to check this 
277     
278 std::cout << "Viewer2D::Process() passe par 5" << std::endl;       
279     
280           Viewer2DWidget* w = (Viewer2DWidget*)bbGetOutputWidget(); 
281           
282     if (w)
283         {
284 std::cout << "Viewer2D::Process() passe par 6" << std::endl;
285                 w->UpdateView();
286                 bbSetOutputRenderer( ((Viewer2DWidget*)bbGetOutputWidget())->GetRenderer() );
287         }
288           bbSetOutputOut( bbGetInputSlice() );
289           
290   }
291   
292
293   void Viewer2D::bbUserConstructor()
294   {
295     bbSetInputIn(NULL);
296     bbSetInputSlice(0);
297     bbSetOutputWidget(0);
298     bbSetInputOrientation(0);
299   }
300
301   /**
302    * \brief  Create wxWidget . 
303    *
304    *
305    */ 
306   void Viewer2D::CreateWidget(wxWindow* parent)
307   {
308   std::cout << "Viewer2D::CreateWidget() passe par 7" << std::endl;
309     Viewer2DWidget* w = new Viewer2DWidget(this,parent);//bbGetWxParent());
310
311     bbSetOutputWidget(w);
312   }
313
314
315   // This callback is *no more*
316   // necessary to get actual processing of the view 
317   // when window is shown
318   void  Viewer2D::OnShowWidget()
319   {
320     // ((Viewer2DWidget*)bbGetOutputWidget())->Update();
321     // ((Viewer2DWidget*)bbGetOutputWidget())->UpdateView();
322   }
323   
324
325 }//namespace bbtk
326
327 #endif
328