1 #include "vvToolBinarize.h"
2 #include "vvSlicerManager.h"
4 #include "vvToolInputSelectorWidget.h"
6 #include "clitkBinarizeImageGenericFilter.h"
8 #include <vtkImageActor.h>
10 #include <vtkImageClip.h>
12 //------------------------------------------------------------------------------
13 // Create the tool and automagically (I like this word) insert it in
14 // the main window menu.
15 ADD_TOOL(vvToolBinarize);
16 //------------------------------------------------------------------------------
19 //------------------------------------------------------------------------------
20 void vvToolBinarize::Initialize() {
21 SetToolName("Binarize");
22 SetToolMenuName("Binarize");
23 SetToolIconFilename(":/new/prefix1/icons/binarize.png");
24 SetToolTip("Image interactive binarization with thresholds.");
26 //------------------------------------------------------------------------------
29 //------------------------------------------------------------------------------
30 vvToolBinarize::vvToolBinarize(vvMainWindowBase * parent, Qt::WindowFlags f)
31 :vvToolWidgetBase(parent,f),
32 vvToolBase<vvToolBinarize>(parent),
36 Ui_vvToolBinarize::setupUi(mToolWidget);
37 mInteractiveDisplayIsEnabled = mCheckBoxInteractiveDisplay->isChecked();
39 // Connect signals & slots
40 connect(mRadioButtonLowerThan, SIGNAL(toggled(bool)), this, SLOT(enableLowerThan(bool)));
41 connect(mCheckBoxUseFG, SIGNAL(toggled(bool)), this, SLOT(useFGBGtoggled(bool)));
42 connect(mCheckBoxUseBG, SIGNAL(toggled(bool)), this, SLOT(useFGBGtoggled(bool)));
43 connect(mCheckBoxInteractiveDisplay, SIGNAL(toggled(bool)), this, SLOT(InteractiveDisplayToggled(bool)));
45 // Initialize some widget
46 mThresholdSlider1->SetText("");
47 mThresholdSlider2->SetText("");
48 mFGSlider->SetText("Foreground value");
49 mBGSlider->SetText("Background value");
52 mFilter = new clitk::BinarizeImageGenericFilter<args_info_clitkBinarizeImage>;
54 // Set how many inputs are needed for this tool
55 AddInputSelector("Select one image", mFilter);
57 //------------------------------------------------------------------------------
60 //------------------------------------------------------------------------------
61 vvToolBinarize::~vvToolBinarize() {
63 //------------------------------------------------------------------------------
66 //------------------------------------------------------------------------------
67 void vvToolBinarize::InteractiveDisplayToggled(bool b) {
68 mInteractiveDisplayIsEnabled = b;
69 if (!mInteractiveDisplayIsEnabled) {
73 for(unsigned int i=0; i<mImageContour.size(); i++) {
74 mImageContour[i]->showActors();
75 if (mRadioButtonLowerThan->isChecked())
76 mImageContourLower[i]->showActors();
78 if (mCurrentSlicerManager)
79 mCurrentSlicerManager->Render();
82 //------------------------------------------------------------------------------
85 //------------------------------------------------------------------------------
86 void vvToolBinarize::RemoveVTKObjects() {
87 for(unsigned int i=0; i<mImageContour.size(); i++) {
88 mImageContour[i]->hideActors();
89 mImageContourLower[i]->hideActors();
91 if (mCurrentSlicerManager)
92 mCurrentSlicerManager->Render();
94 //------------------------------------------------------------------------------
97 //------------------------------------------------------------------------------
98 bool vvToolBinarize::close() {
100 return vvToolWidgetBase::close();
102 //------------------------------------------------------------------------------
105 //------------------------------------------------------------------------------
106 void vvToolBinarize::reject() {
107 // DD("vvToolBinarize::reject");
109 return vvToolWidgetBase::reject();
111 //------------------------------------------------------------------------------
114 //------------------------------------------------------------------------------
115 void vvToolBinarize::enableLowerThan(bool b) {
117 mThresholdSlider1->resetMaximum();
118 for(unsigned int i=0; i<mImageContour.size(); i++) {
119 mImageContourLower[i]->hideActors();
121 mCurrentSlicerManager->Render();
124 valueChangedT1(mThresholdSlider1->GetValue());
125 valueChangedT2(mThresholdSlider2->GetValue());
126 for(unsigned int i=0; i<mImageContour.size(); i++) {
127 mImageContourLower[i]->showActors();
129 mCurrentSlicerManager->Render();
132 //------------------------------------------------------------------------------
135 //------------------------------------------------------------------------------
136 void vvToolBinarize::useFGBGtoggled(bool) {
137 if (!mCheckBoxUseBG->isChecked() && !mCheckBoxUseFG->isChecked())
138 mCheckBoxUseBG->toggle();
140 //------------------------------------------------------------------------------
143 //------------------------------------------------------------------------------
144 // void vvToolBinarize::InputIsSelected(std::vector<vvSlicerManager *> & m) {
145 // DD("vvToolBinarize::InputIsSelected vector in binarize");
148 void vvToolBinarize::InputIsSelected(vvSlicerManager * m) {
149 mCurrentSlicerManager = m;
151 // Specific for this gui
152 mThresholdSlider1->SetValue(0);
153 mThresholdSlider2->SetValue(0);
154 mThresholdSlider1->SetImage(mCurrentImage);
155 mThresholdSlider2->SetImage(mCurrentImage);
156 mFGSlider->SetImage(mCurrentImage);
157 mBGSlider->SetImage(mCurrentImage);
158 // DD(mCurrentSlicerManager->GetFileName().c_str());
159 mFGSlider->SetMaximum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMax());
160 mFGSlider->SetMinimum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMin());
161 mBGSlider->SetMaximum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMax());
162 mBGSlider->SetMinimum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMin());
163 mFGSlider->SetValue(1);
164 mBGSlider->SetValue(0);
165 mFGSlider->SetSingleStep(1);
166 mBGSlider->SetSingleStep(1);
168 // VTK objects for interactive display
169 for(int i=0;i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
170 mImageContour.push_back(new vvImageContour);
171 mImageContour[i]->setSlicer(mCurrentSlicerManager->GetSlicer(i));
172 mImageContour[i]->setColor(1.0, 0.0, 0.0);
173 mImageContourLower.push_back(new vvImageContour);
174 mImageContourLower[i]->setSlicer(mCurrentSlicerManager->GetSlicer(i));
175 mImageContourLower[i]->setColor(0.0, 0.0, 1.0);
177 valueChangedT1(mThresholdSlider1->GetValue());
179 connect(mThresholdSlider1, SIGNAL(valueChanged(double)), this, SLOT(valueChangedT1(double)));
180 connect(mThresholdSlider2, SIGNAL(valueChanged(double)), this, SLOT(valueChangedT2(double)));
182 connect(mCurrentSlicerManager,SIGNAL(UpdateSlice(int,int)),this,SLOT(UpdateSlice(int, int)));
183 connect(mCurrentSlicerManager,SIGNAL(UpdateTSlice(int,int)),this,SLOT(UpdateSlice(int, int)));
186 //------------------------------------------------------------------------------
189 //------------------------------------------------------------------------------
190 void vvToolBinarize::UpdateSlice(int slicer,int slices) {
191 if (!mInteractiveDisplayIsEnabled) return;
192 if (!mCurrentSlicerManager) close();
193 for(int i=0;i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
194 mImageContour[i]->update(mThresholdSlider1->GetValue());
195 if (mRadioButtonLowerThan->isChecked())
196 mImageContourLower[i]->update(mThresholdSlider2->GetValue());
199 mCurrentSlicerManager->Render();
201 //------------------------------------------------------------------------------
203 //------------------------------------------------------------------------------
204 void vvToolBinarize::GetArgsInfoFromGUI() {
206 /* //KEEP THIS FOR READING GGO FROM FILE
208 std::string a = "toto";
209 char * const* argv = new char*;
211 struct cmdline_parser_params p;
212 p.check_required = 0;
213 int good = cmdline_parser_ext(argc, argv, &args_info, &p);
217 mArgsInfo.imagetypes_flag = 0;
218 mArgsInfo.upper_given = 0;
219 mArgsInfo.lower_given = 0;
220 bool inverseBGandFG = false;
222 mArgsInfo.lower_given = 1;
223 mArgsInfo.lower_arg = mThresholdSlider1->GetValue();
224 // DD(mArgsInfo.lower_arg);
225 if (mRadioButtonLowerThan->isChecked()) {
226 mArgsInfo.upper_given = 1;
227 mArgsInfo.upper_arg = mThresholdSlider2->GetValue();
228 if (mArgsInfo.upper_arg<mArgsInfo.lower_arg) {
229 mArgsInfo.upper_given = 0;
230 DD("TODO : lower thres greater than greater thres ! Ignoring ");
234 mArgsInfo.fg_arg = mFGSlider->GetValue();
235 mArgsInfo.bg_arg = mBGSlider->GetValue();
237 if (inverseBGandFG) {
238 mArgsInfo.fg_arg = mFGSlider->GetValue();
239 mArgsInfo.bg_arg = mBGSlider->GetValue();
241 mArgsInfo.fg_given = 1;
242 mArgsInfo.bg_given = 1;
244 if (mCheckBoxUseBG->isChecked()) {
245 if (mCheckBoxUseFG->isChecked()) mArgsInfo.mode_arg = (char*)"both";
246 else mArgsInfo.mode_arg = (char*)"BG";
248 else mArgsInfo.mode_arg = (char*)"FG";
250 mArgsInfo.verbose_flag = false;
252 // Required (even if not used)
253 mArgsInfo.input_given = 0;
254 mArgsInfo.output_given = 0;
256 mArgsInfo.input_arg = new char;
257 mArgsInfo.output_arg = new char;
259 //------------------------------------------------------------------------------
262 //------------------------------------------------------------------------------
263 void vvToolBinarize::apply() {
264 if (!mCurrentSlicerManager) close();
265 QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));
266 GetArgsInfoFromGUI();
269 clitk::BinarizeImageGenericFilter<args_info_clitkBinarizeImage>::Pointer filter =
270 clitk::BinarizeImageGenericFilter<args_info_clitkBinarizeImage>::New();
271 filter->SetInputVVImage(mCurrentImage);
272 filter->SetArgsInfo(mArgsInfo);
273 filter->EnableReadOnDisk(false);
277 vvImage::Pointer output = filter->GetOutputVVImage();
278 std::ostringstream osstream;
279 osstream << "Binarized_" << mCurrentSlicerManager->GetSlicer(0)->GetFileName() << ".mhd";
280 AddImage(output,osstream.str());
281 QApplication::restoreOverrideCursor();
284 //------------------------------------------------------------------------------
287 //------------------------------------------------------------------------------
288 void vvToolBinarize::valueChangedT2(double v) {
289 // DD("valueChangedT2");
290 if (mRadioButtonLowerThan->isChecked()) {
291 mThresholdSlider1->SetMaximum(v);
292 if (!mInteractiveDisplayIsEnabled) return;
293 for(int i=0;i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
294 mImageContourLower[i]->update(v);
296 mCurrentSlicerManager->Render();
299 //------------------------------------------------------------------------------
302 //------------------------------------------------------------------------------
303 void vvToolBinarize::valueChangedT1(double v) {
304 // DD("valueChangedT1");
305 if (!mCurrentSlicerManager) close();
306 mThresholdSlider2->SetMinimum(v);
307 // int m1 = (int)lrint(v);
308 if (!mInteractiveDisplayIsEnabled) return;
309 for(int i=0;i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
310 mImageContour[i]->update(v);
312 mCurrentSlicerManager->Render();
314 //------------------------------------------------------------------------------