]> Creatis software - clitk.git/blob - vv/vvToolProfile.cxx
Creation of the Profile Tool
[clitk.git] / vv / vvToolProfile.cxx
1 /*=========================================================================
2   Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
3
4   Authors belong to:
5   - University of LYON              http://www.universite-lyon.fr/
6   - Léon Bérard cancer center       http://www.centreleonberard.fr
7   - CREATIS CNRS laboratory         http://www.creatis.insa-lyon.fr
8
9   This software is distributed WITHOUT ANY WARRANTY; without even
10   the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
11   PURPOSE.  See the copyright notices for more information.
12
13   It is distributed under dual licence
14
15   - BSD        See included LICENSE.txt file
16   - CeCILL-B   http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
17   ===========================================================================**/
18
19 // vv
20 #include "vvToolProfile.h"
21 #include "vvSlicerManager.h"
22 #include "vvSlicer.h"
23 #include "vvToolInputSelectorWidget.h"
24
25 // clitk
26 #include "clitkProfileImageGenericFilter.h"
27
28 // vtk
29 #include <vtkImageActor.h>
30 #include <vtkCamera.h>
31 #include <vtkImageClip.h>
32 #include <vtkRenderWindow.h>
33
34
35 //------------------------------------------------------------------------------
36 // Create the tool and automagically (I like this word) insert it in
37 // the main window menu.
38 ADD_TOOL(vvToolProfile);
39 //------------------------------------------------------------------------------
40
41
42 //------------------------------------------------------------------------------
43 void vvToolProfile::Initialize()
44
45   SetToolName("Profile");
46   SetToolMenuName("Intensity Profile");
47   SetToolIconFilename(":/common/icons/profile.png");
48   SetToolTip("Display the intensity profile between 2 points of the image.");
49   SetToolExperimental(false);
50 }
51 //------------------------------------------------------------------------------
52
53
54 //------------------------------------------------------------------------------
55 vvToolProfile::vvToolProfile(vvMainWindowBase * parent, Qt::WindowFlags f)
56   :vvToolWidgetBase(parent,f),
57    vvToolBase<vvToolProfile>(parent),
58    Ui::vvToolProfile()
59
60   // GUI Initialization
61   Ui_vvToolProfile::setupUi(mToolWidget);
62   mInteractiveDisplayIsEnabled = mCheckBoxInteractiveDisplay->isChecked();
63
64   // Connect signals & slots
65   connect(mRadioButtonLowerThan, SIGNAL(toggled(bool)), this, SLOT(enableLowerThan(bool)));
66   connect(mCheckBoxUseFG, SIGNAL(toggled(bool)), this, SLOT(useFGBGtoggled(bool)));
67   connect(mCheckBoxUseBG, SIGNAL(toggled(bool)), this, SLOT(useFGBGtoggled(bool)));
68   connect(mCheckBoxInteractiveDisplay, SIGNAL(toggled(bool)), this, SLOT(InteractiveDisplayToggled(bool)));
69
70   // Initialize some widget
71   mThresholdSlider1->SetText("");
72   mThresholdSlider2->SetText("");
73   mFGSlider->SetText("Foreground value");
74   mBGSlider->SetText("Background value");
75
76   // Main filter
77   mFilter = clitk::ProfileImageGenericFilter::New();
78
79   // Set how many inputs are needed for this tool
80   AddInputSelector("Select one image", mFilter);
81 }
82 //------------------------------------------------------------------------------
83
84
85 //------------------------------------------------------------------------------
86 vvToolProfile::~vvToolProfile()
87
88 }
89 //------------------------------------------------------------------------------
90
91
92 //------------------------------------------------------------------------------
93 void vvToolProfile::InteractiveDisplayToggled(bool b)
94
95   mInteractiveDisplayIsEnabled = b;
96   if (!mInteractiveDisplayIsEnabled) {
97     RemoveVTKObjects();
98   } else {
99     for(unsigned int i=0; i<mImageContour.size(); i++) {
100       mImageContour[i]->ShowActors();
101       if (mRadioButtonLowerThan->isChecked())
102         mImageContourLower[i]->ShowActors();
103     }
104     if (mCurrentSlicerManager)
105       mCurrentSlicerManager->Render();
106   }
107 }
108 //------------------------------------------------------------------------------
109
110
111 //------------------------------------------------------------------------------
112 void vvToolProfile::RemoveVTKObjects()
113
114   for(unsigned int i=0; i<mImageContour.size(); i++) {
115     mImageContour[i]->HideActors();
116     mImageContourLower[i]->HideActors();    
117   }
118   if (mCurrentSlicerManager)
119     mCurrentSlicerManager->Render();
120 }
121 //------------------------------------------------------------------------------
122
123
124 //------------------------------------------------------------------------------
125 bool vvToolProfile::close()
126
127   // RemoveVTKObjects();
128   return vvToolWidgetBase::close();
129 }
130 //------------------------------------------------------------------------------
131
132
133 //------------------------------------------------------------------------------
134 void vvToolProfile::closeEvent(QCloseEvent *event)
135
136   RemoveVTKObjects();
137   event->accept();
138 }
139 //------------------------------------------------------------------------------
140
141
142 //------------------------------------------------------------------------------
143 void vvToolProfile::reject()
144
145   // DD("vvToolProfile::reject");
146   RemoveVTKObjects();
147   return vvToolWidgetBase::reject();
148 }
149 //------------------------------------------------------------------------------
150
151
152 //------------------------------------------------------------------------------
153 void vvToolProfile::enableLowerThan(bool b)
154
155   if (!b) {
156     mThresholdSlider1->resetMaximum();
157     for(unsigned int i=0; i<mImageContour.size(); i++) {
158       mImageContourLower[i]->HideActors();    
159     }
160     mCurrentSlicerManager->Render();
161   } else {
162     valueChangedT1(mThresholdSlider1->GetValue());
163     valueChangedT2(mThresholdSlider2->GetValue());
164     for(unsigned int i=0; i<mImageContour.size(); i++) {
165       mImageContourLower[i]->ShowActors();    
166     }
167     mCurrentSlicerManager->Render();
168   }
169 }
170 //------------------------------------------------------------------------------
171
172
173 //------------------------------------------------------------------------------
174 void vvToolProfile::useFGBGtoggled(bool)
175
176   if (!mCheckBoxUseBG->isChecked() && !mCheckBoxUseFG->isChecked())
177     mCheckBoxUseBG->toggle();
178 }
179 //------------------------------------------------------------------------------
180
181
182 //------------------------------------------------------------------------------
183 // void vvToolProfile::InputIsSelected(std::vector<vvSlicerManager *> & m) {
184 //   DD("vvToolProfile::InputIsSelected vector in Profile");
185 //   DD(m.size());
186 // }
187 void vvToolProfile::InputIsSelected(vvSlicerManager * m)
188
189   mCurrentSlicerManager = m;
190
191   // Specific for this gui
192   mThresholdSlider1->SetValue(0);
193   mThresholdSlider2->SetValue(0);
194   mThresholdSlider1->SetImage(mCurrentImage);
195   mThresholdSlider2->SetImage(mCurrentImage);
196   mFGSlider->SetImage(mCurrentImage);
197   mBGSlider->SetImage(mCurrentImage);
198   //  DD(mCurrentSlicerManager->GetFileName().c_str());
199   //  mFGSlider->SetMaximum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMax());
200   //   mFGSlider->SetMinimum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMin());
201   //   mBGSlider->SetMaximum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMax());
202   //   mBGSlider->SetMinimum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMin());
203
204   // Output is uchar ...
205   mFGSlider->SetMaximum(255);
206   mFGSlider->SetMinimum(0);
207   mBGSlider->SetMaximum(255);
208   mBGSlider->SetMinimum(0);
209
210   mFGSlider->SetValue(1);
211   mBGSlider->SetValue(0);
212   mFGSlider->SetSingleStep(1);
213   mBGSlider->SetSingleStep(1);
214
215   // VTK objects for interactive display
216   for(int i=0; i<mCurrentSlicerManager->GetNumberOfSlicers(); i++) {
217     mImageContour.push_back(vvImageContour::New());
218     mImageContour[i]->SetSlicer(mCurrentSlicerManager->GetSlicer(i));
219     mImageContour[i]->SetColor(1.0, 0.0, 0.0);
220     mImageContour[i]->SetDepth(0); // to be in front of (whe used with ROI tool)
221     mImageContourLower.push_back(vvImageContour::New());
222     mImageContourLower[i]->SetSlicer(mCurrentSlicerManager->GetSlicer(i));
223     mImageContourLower[i]->SetColor(0.0, 0.0, 1.0);
224     mImageContourLower[i]->SetDepth(100); // to be in front of (whe used with ROI tool)
225   }
226   valueChangedT1(mThresholdSlider1->GetValue());
227
228   connect(mThresholdSlider1, SIGNAL(valueChanged(double)), this, SLOT(valueChangedT1(double)));
229   connect(mThresholdSlider2, SIGNAL(valueChanged(double)), this, SLOT(valueChangedT2(double)));
230
231   connect(mCurrentSlicerManager,SIGNAL(UpdateSlice(int,int)),this,SLOT(UpdateSlice(int, int)));
232   connect(mCurrentSlicerManager,SIGNAL(UpdateTSlice(int,int)),this,SLOT(UpdateSlice(int, int)));
233   
234   connect(mCurrentSlicerManager,SIGNAL(UpdateOrientation(int,int)),this,SLOT(UpdateOrientation(int, int)));
235
236   //  connect(mCurrentSlicerManager, SIGNAL(LeftButtonReleaseSignal(int)), SLOT(LeftButtonReleaseEvent(int)));
237   InteractiveDisplayToggled(mInteractiveDisplayIsEnabled);
238 }
239 //------------------------------------------------------------------------------
240
241
242 //------------------------------------------------------------------------------
243 // void vvToolProfile::LeftButtonReleaseEvent(int slicer) {
244 //   DD("LeftButtonReleaseEvent");
245 //   for(int i=0; i<mCurrentSlicerManager->GetNumberOfSlicers(); i++) {
246 //     if (i == slicer);
247 //     mCurrentSlicerManager->GetSlicer(i)->GetRenderWindow()->Render();
248 //   }
249 // }
250 //------------------------------------------------------------------------------
251
252
253 //------------------------------------------------------------------------------
254 void vvToolProfile::UpdateOrientation(int slicer,int orientation)
255
256   Update(slicer);
257 }
258 //------------------------------------------------------------------------------
259
260 //------------------------------------------------------------------------------
261 void vvToolProfile::UpdateSlice(int slicer,int slices)
262
263   Update(slicer);
264 }
265 //------------------------------------------------------------------------------
266
267 //------------------------------------------------------------------------------
268 void vvToolProfile::Update(int slicer)
269
270   if (!mInteractiveDisplayIsEnabled) return;
271   if (!mCurrentSlicerManager) close();
272   mImageContour[slicer]->Update(mThresholdSlider1->GetValue());
273   if (mRadioButtonLowerThan->isChecked()) 
274     mImageContourLower[slicer]->Update(mThresholdSlider2->GetValue());
275 }
276 //------------------------------------------------------------------------------
277
278 //------------------------------------------------------------------------------
279 void vvToolProfile::GetArgsInfoFromGUI()
280
281
282   /* //KEEP THIS FOR READING GGO FROM FILE
283      int argc=1;
284      std::string a = "toto";
285      char * const* argv = new char*;
286      //a.c_str();
287      struct cmdline_parser_params p;
288      p.check_required = 0;
289      int good = cmdline_parser_ext(argc, argv, &args_info, &p);
290      DD(good);
291   */
292   cmdline_parser_clitkProfileImage_init(&mArgsInfo); // Initialisation to default
293   bool inverseBGandFG = false;
294
295   mArgsInfo.lower_given = 1;
296   mArgsInfo.lower_arg = mThresholdSlider1->GetValue();
297   if (mRadioButtonLowerThan->isChecked()) {
298     mArgsInfo.upper_given = 1;
299     mArgsInfo.upper_arg = mThresholdSlider2->GetValue();
300     if (mArgsInfo.upper_arg<mArgsInfo.lower_arg) {
301       mArgsInfo.upper_given = 0;
302       DD("TODO : lower thres greater than greater thres ! Ignoring ");
303     }
304   }
305
306   mArgsInfo.fg_arg = mFGSlider->GetValue();
307   mArgsInfo.bg_arg = mBGSlider->GetValue();
308
309   if (inverseBGandFG) {
310     mArgsInfo.fg_arg = mFGSlider->GetValue();
311     mArgsInfo.bg_arg = mBGSlider->GetValue();
312   }
313   mArgsInfo.fg_given = 1;
314   mArgsInfo.bg_given = 1;
315
316   if (mCheckBoxUseBG->isChecked()) {
317     if (mCheckBoxUseFG->isChecked()) mArgsInfo.mode_arg = (char*)"both";
318     else mArgsInfo.mode_arg = (char*)"BG";
319   } else mArgsInfo.mode_arg = (char*)"FG";
320
321   mArgsInfo.verbose_flag = false;
322
323   // // Required (even if not used)
324   // mArgsInfo.input_given = 0;
325   // mArgsInfo.output_given = 0;
326
327   // mArgsInfo.input_arg = new char;
328   // mArgsInfo.output_arg = new char;
329 }
330 //------------------------------------------------------------------------------
331
332
333 //------------------------------------------------------------------------------
334 void vvToolProfile::apply()
335
336   if (!mCurrentSlicerManager) close();
337   QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));
338   GetArgsInfoFromGUI();  
339   
340   // Main filter
341   clitk::ProfileImageGenericFilter::Pointer filter =
342     clitk::ProfileImageGenericFilter::New();
343   filter->SetInputVVImage(mCurrentImage);
344   filter->SetArgsInfo(mArgsInfo);
345   filter->Update();
346
347
348   // Output
349   vvImage::Pointer output = filter->GetOutputVVImage();  
350   std::ostringstream osstream;
351   osstream << "Profiled_" << mCurrentSlicerManager->GetSlicer(0)->GetFileName() << ".mhd";
352   AddImage(output,osstream.str());
353   QApplication::restoreOverrideCursor();
354   close();
355 }
356 //------------------------------------------------------------------------------
357
358
359 //------------------------------------------------------------------------------
360 void vvToolProfile::valueChangedT2(double v)
361
362   //  DD("valueChangedT2");
363   if (mRadioButtonLowerThan->isChecked()) {
364     mThresholdSlider1->SetMaximum(v);
365     if (!mInteractiveDisplayIsEnabled) return;
366     for(int i=0;i<mCurrentSlicerManager->GetNumberOfSlicers(); i++) {
367       mImageContourLower[i]->Update(v);
368     }
369     mCurrentSlicerManager->Render();
370   }
371 }
372 //------------------------------------------------------------------------------
373
374
375 //------------------------------------------------------------------------------
376 void vvToolProfile::valueChangedT1(double v)
377
378   //  DD("valueChangedT1");
379   if (!mCurrentSlicerManager) close();
380   mThresholdSlider2->SetMinimum(v);
381   //  int m1 = (int)lrint(v);
382   if (!mInteractiveDisplayIsEnabled) return;
383   for(int i=0;i<mCurrentSlicerManager->GetNumberOfSlicers(); i++) {
384     mImageContour[i]->Update(v);
385   }
386   mCurrentSlicerManager->Render();
387 }
388 //------------------------------------------------------------------------------