]> Creatis software - bbtk.git/blob - packages/wxvtk/src/bbwxvtkViewer2D.cxx
New widget pipeline processing : should be almost ok
[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/11/25 11:31:08 $
7   Version:   $Revision: 1.22 $
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       }
140     else if ( ( mBox->bbGetInputIn() != NULL ) && 
141               (backImageData != mBox->bbGetInputIn()) ) 
142       {
143         backImageData = mBox->bbGetInputIn();
144         backImageData->Update();
145         imageViewer->SetInput( backImageData );
146         mUpdateCamera = true;
147       }
148
149     if (mUpdateCamera)
150       {
151         int x1,x2,y1,y2,z1,z2;
152         double spx,spy,spz;
153         backImageData->GetSpacing(spx,spy,spz);
154         backImageData->GetExtent (x1,x2,y1,y2,z1,z2);
155         
156         //      bbtkMessage("Output",2,"Viewer2D '"<<mBox->bbGetName()<<"' - Image :"<<std::endl);
157         //      std::cout << " extent  = "<< x1<<"-"<<x2<<" x "<<y1<<"-"<<y2<<" x "<<z1<<"-"<<z2<<std::endl;
158         //      std::cout << " spacing = "<<spx << " - "<< spy << " - "<<spz<<std::endl;
159         
160         vtkCamera *camera = imageViewer->GetRenderer()->GetActiveCamera();
161
162         camera->SetViewUp    ( spx*0,         -spy*1,        spz*0 );
163         camera->SetPosition  ( spx*(x1+x2)/2, spy*(y1+y2)/2, -spz*10000000 ); 
164         camera->SetFocalPoint( spx*(x1+x2)/2, spy*(y1+y2)/2, spz*0 ); 
165
166         camera->ComputeViewPlaneNormal();
167         camera->SetParallelScale(  spx*(x2-x1)/2.0 );
168         //   
169         //      imageViewer->GetRenderer()->ResetCamera(x1,x2,y1,y2,z1,z2);
170         mUpdateCamera = false;
171       }
172   
173     //  std::cout << "OW = "<< mBox->bbGetWindow() << std::endl;
174     //  std::cout << "CW = "<< mBox->bbGetContainingWindow() << std::endl;
175     //  std::cout << "Sh = "<< mBox->bbIsShown() << std::endl;
176     
177     int orientation = mBox->bbGetInputOrientation();
178     if (orientation<0 || orientation>2) {
179        bbtkMessage("Output", 2, "Viewer2D : Orientation was not 0< <2 "<<std::endl);
180        orientation=2; // Why not?      
181     }
182     // mBox->bbIsShown() is necessary to be sure that the wxWindow is already 
183     // created and displayed. 
184     // Else when slice != 0 we get an X Window System error with wxGTK
185     if (
186         (backImageData!=NULL)&&
187         (mBox->bbIsShown()))
188     {
189                 int z = mBox->bbGetInputSlice();
190                 //      std::cout << "slice = "<<z<<std::endl;
191                 int ext[6];
192                 backImageData->GetWholeExtent(ext);
193
194                 //      std::cout << "ext = "<<ext[4]<<" - "<<ext[5]<<std::endl;
195
196             switch(orientation){
197                case 0:
198                 if (z<ext[0]) { z=ext[0]; }
199                 if (z>ext[1]) { z=ext[1]; }
200                  
201                 break;
202                                    
203                case 1:
204                 if (z<ext[2]) { z=ext[2]; }
205                 if (z>ext[3]) { z=ext[3]; }
206                 break;
207                                                     
208                case 2:
209                 if (z<ext[4]) { z=ext[4]; }
210                 if (z>ext[5]) { z=ext[5]; }
211                 break;
212                 
213               default:
214                 bbtkMessage("Output", 2, "Viewer2D : Orientation was not 0< <2 (shouldn't arrive here!"<<std::endl);
215                 z = ext[5]; // Why not?
216            }
217
218             std::cout << "V2D Render "<<std::endl;
219             wxvtkrenderwindowinteractor->Update();
220             wxvtkrenderwindowinteractor->Render();
221             imageViewer->GetRenderer()->ResetCameraClippingRange();
222             
223 /*
224         #if (VTK_MAJOR_VERSION >= 5)
225                 imageViewer->SetSlice( z );
226         #else
227                 imageViewer->SetZSlice( z );
228         #endif
229 */
230            imageViewer->SetSliceOrientation (orientation);
231            imageViewer->SetSlice( z );
232                 
233                 //imageViewer->UpdateDisplayExtent();
234   } 
235
236
237 }
238
239
240   vtkRenderer * Viewer2DWidget::GetRenderer()
241   {
242           return imageViewer->GetRenderer();
243   }
244
245
246
247   //--------------------------------------------------------------------------
248   //-------------------------------------------------------------------------
249   //--------------------------------------------------------------------------
250   //--------------------------------------------------------------------------
251
252
253
254
255   BBTK_BLACK_BOX_IMPLEMENTATION(Viewer2D,bbtk::WxBlackBox);
256
257
258   void Viewer2D::Process() 
259   { 
260     // LG : Bug on Linux (wxGTK) when initial slice != 0 on first time 
261     // (see wxvtk/bbs/appli/ExampleSimpleSlicer)
262     // I think there is a problem with rendering before window 
263     // has been reparented ... we have to check this 
264           Viewer2DWidget* w = (Viewer2DWidget*)bbGetOutputWidget(); 
265     if (w)
266         {
267                 w->UpdateView();
268                 bbSetOutputRenderer( ((Viewer2DWidget*)bbGetOutputWidget())->GetRenderer() );
269         }
270           bbSetOutputOut( bbGetInputSlice() );
271           
272   }
273   
274
275   void Viewer2D::bbUserConstructor()
276   {
277     bbSetInputIn(NULL);
278     bbSetInputSlice(0);
279     bbSetOutputWidget(0);
280     bbSetInputOrientation(0);
281   }
282
283   /**
284    * \brief  Create wxWidget . 
285    *
286    *
287    */ 
288   void Viewer2D::CreateWidget(wxWindow* parent)
289   {
290     bbtkDebugMessageInc("Core",9,"Viewer2D::CreateWidget()"<<std::endl);
291
292           Viewer2DWidget* w = new Viewer2DWidget(this,parent);//bbGetWxParent());
293     bbSetOutputWidget(w);
294
295     bbtkDebugDecTab("Core",9);
296   }
297
298
299   // This callback is necessary to get actual processing of the view 
300   // when window is shown
301   void  Viewer2D::OnShowWidget()
302   {
303     std::cout << "V2D::OnShow"<<std::endl;
304     ((Viewer2DWidget*)bbGetOutputWidget())->Update();
305     ((Viewer2DWidget*)bbGetOutputWidget())->UpdateView();
306 #if defined(_WIN32)
307 //    ((Viewer2DWidget*)bbGetOutputWidget())->Refresh();
308     ((Viewer2DWidget*)bbGetOutputWidget())->Update();
309     ((Viewer2DWidget*)bbGetOutputWidget())->UpdateView();
310 #endif
311      /*
312       // BUGS ON LINUX !!
313     ((Viewer2DWidget*)bbGetOutputWidget())->Update();
314     ((Viewer2DWidget*)bbGetOutputWidget())->UpdateView();
315     */
316     std::cout << "V2D::OnShow OK"<<std::endl;
317   }
318   
319
320 }//namespace bbtk
321
322 #endif
323