]> Creatis software - clitk.git/blob - vv/vvToolWidgetBase.cxx
Merge branch 'master' of git://git.creatis.insa-lyon.fr/clitk
[clitk.git] / vv / vvToolWidgetBase.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 "vvToolWidgetBase.h"
21 #include "vvMainWindowBase.h"
22 #include "vvSlicerManager.h"
23
24 // Qt
25 #include <QMessageBox>
26 #include <QKeyEvent>
27 #include <QDockWidget>
28 #include <QTabWidget>
29
30 //------------------------------------------------------------------------------
31 // Static initialisation
32 int vvToolWidgetBase::mTabNumber = -1;
33 QWidget * vvToolWidgetBase::mStaticWidgetForTab = NULL;
34 QVBoxLayout * vvToolWidgetBase::mStaticVerticalLayout = NULL;
35 bool vvToolWidgetBase::mIsAnotherToolWaitInput = false;
36
37 //------------------------------------------------------------------------------
38 vvToolWidgetBase::vvToolWidgetBase(vvMainWindowBase * parent, Qt::WindowFlags f, bool initialize):
39   QWidget(parent, f),
40   Ui::vvToolWidgetBase()
41 {
42   mMainWindow = parent;
43   setAttribute(Qt::WA_DeleteOnClose);
44   if (initialize) Initialization();
45   if (isWindow()) { // In this case, the tool is a floating windows
46     // this->grabKeyboard();   // for the ESC key to close the dialog
47     mPreventToUseTwoToolsOnSameInput = false;
48   }
49   else { // In this case it is inserted into a tab
50     DD("Not implemented yet TODO !!");
51     exit(0);
52     // Prevent to load two tools at the same time
53     DD(mIsAnotherToolWaitInput);
54     if (mIsAnotherToolWaitInput) {
55       //      setVisible(false);
56       QWidget::close();
57       DD("before return");
58       return;
59     }
60     else mIsAnotherToolWaitInput = true;
61     mPreventToUseTwoToolsOnSameInput = true;
62     // Setup the UI in a new widget
63
64     mWidgetForTab = new QWidget(this); 
65                                 // <-- try to set mToolWidget instead of this ? NO  (loop)
66                                 // <-- try to set parent instead of this ? NO, change nothing
67
68     QVBoxLayout * verticalLayout = new QVBoxLayout;//(mWidgetForTab);
69     verticalLayout->addWidget(mToolInputSelectionWidget);
70     verticalLayout->addWidget(mToolWidget);
71     verticalLayout->setContentsMargins(1, 1, 1, 1);
72     mWidgetForTab->setLayout(verticalLayout);
73     DD(mWidgetForTab->isVisible());
74     mWidgetForTab->setVisible(true);
75     DD(mWidgetForTab->isVisible());
76
77     // Is this the first time we add a tab ? 
78     if (parent->GetTab()->widget(mTabNumber) == NULL) { // Yes, create main widget
79       DD("Create main widget");
80       mStaticWidgetForTab = new QWidget(parent->GetTab());//parent, f); 
81                                 // <-- try to set mToolWidget instead of nothing ? NO loop
82                                 // <-- try to set parent->GetTab() instead of nothing ? 
83       mStaticVerticalLayout = new QVBoxLayout;//(mStaticWidgetForTab);
84       mStaticWidgetForTab->setLayout(mStaticVerticalLayout);
85
86       /*mWidgetForTab = new QWidget(mStaticWidgetForTab);
87     QVBoxLayout * verticalLayout = new QVBoxLayout;//(mWidgetForTab);
88     verticalLayout->addWidget(mToolInputSelectionWidget);
89     verticalLayout->addWidget(mToolWidget);
90     verticalLayout->setContentsMargins(1, 1, 1, 1);
91     mWidgetForTab->setLayout(verticalLayout);
92     DD(mWidgetForTab->isVisible());
93     mWidgetForTab->setVisible(true);
94     DD(mWidgetForTab->isVisible());*/
95
96       //<----------      mStaticVerticalLayout->addWidget(mWidgetForTab); 
97
98       mTabNumber = parent->GetTab()->addTab(mStaticWidgetForTab, "");
99       DD(mStaticWidgetForTab->isVisible());
100       mStaticWidgetForTab->setVisible(true);
101       DD(mStaticWidgetForTab->isVisible());
102       //      mWidgetForTab->setParent(mStaticWidgetForTab);
103     }
104     else {
105       DD("insert into widget");
106       mStaticVerticalLayout->addWidget(mWidgetForTab);
107       SwapCurrentWidget();
108       mToolWidget->setEnabled(true);
109     }
110     parent->GetTab()->setCurrentIndex(mTabNumber); 
111     mMainButtonBox->hide(); // No OK/Cancel by default in this case
112   }
113 }
114 //------------------------------------------------------------------------------
115
116
117 //------------------------------------------------------------------------------
118 void vvToolWidgetBase::Initialization() 
119 {
120   mCurrentSlicerManager = 0;
121   mIsInitialized = false;
122   mFilter = 0;
123   setWindowModality(Qt::NonModal);
124   // GUI Initialization
125   setupUi(this); ///////////////////////////// TRIAL
126   // Connect signals & slots
127   connect(mMainWindow, SIGNAL(AnImageIsBeingClosed(vvSlicerManager*)), this, SLOT(AnImageIsBeingClosed(vvSlicerManager*)));
128   connect(mMainWindow, SIGNAL(SelectedImageHasChanged(vvSlicerManager*)), this, SLOT(SelectedImageHasChanged(vvSlicerManager*)));
129   connect(mToolInputSelectionWidget, SIGNAL(accepted()), this, SLOT(InputIsSelected()));
130   connect(mToolInputSelectionWidget, SIGNAL(rejected()), this, SLOT(close()));
131   connect(mMainButtonBox, SIGNAL(accepted()), this, SLOT(apply()));
132   connect(mMainButtonBox, SIGNAL(rejected()), this, SLOT(close()));
133
134   // Disable main widget while input image is not selected
135   mToolWidget->setEnabled(false);
136 }
137 //------------------------------------------------------------------------------
138
139
140 //------------------------------------------------------------------------------
141 vvToolWidgetBase::~vvToolWidgetBase()
142 {
143 }
144 //------------------------------------------------------------------------------
145
146
147 //------------------------------------------------------------------------------
148 void vvToolWidgetBase::keyPressEvent(QKeyEvent *event) 
149 {
150   if (event->key() == Qt::Key_Escape) {
151     reject();
152     event->accept();
153     return;
154   } 
155   else {
156     QWidget::keyPressEvent(event);
157   }
158   //  event->ignore();
159   //mMainWindow->keyPressEvent(event);
160   // QWidget::keyPressEvent(event);
161 }
162 //------------------------------------------------------------------------------
163
164
165 //------------------------------------------------------------------------------
166 void vvToolWidgetBase::accept()
167 {
168   apply();
169 }
170 //------------------------------------------------------------------------------
171
172
173 //------------------------------------------------------------------------------
174 void vvToolWidgetBase::reject()
175 {
176   close();
177 }
178 //------------------------------------------------------------------------------
179
180
181 //------------------------------------------------------------------------------
182 void vvToolWidgetBase::AddInputSelector(QString s, clitk::ImageToImageGenericFilterBase * f, bool allowSkip)
183 {
184   int j=0;
185   mFilter = f;
186   mSlicerManagersCompatible.clear();
187   mToolInputSelectionWidget->setToolTip(QString("%1").arg(mFilter->GetAvailableImageTypes().c_str()));
188   mCurrentCompatibleIndex = 0;
189   for(unsigned int i=0; i<mMainWindow->GetSlicerManagers().size(); i++) {
190     vvImage * s = mMainWindow->GetSlicerManagers()[i]->GetImage();
191     if (mFilter->CheckImageType(s->GetNumberOfDimensions(),
192                                 s->GetNumberOfScalarComponents(),
193                                 s->GetScalarTypeAsITKString())) {
194       mSlicerManagersCompatible.push_back(mMainWindow->GetSlicerManagers()[i]);
195       if ((int)i == mMainWindow->GetSlicerManagerCurrentIndex()) mCurrentCompatibleIndex = j;
196       j++;
197     }
198   }
199   if (mSlicerManagersCompatible.size() == 0) {
200     std::ostringstream osstream;
201     osstream << "Sorry, could not perform operation. No (compatible) image. "
202              << mFilter->GetAvailableImageTypes();
203     QMessageBox::information(this, "No image", osstream.str().c_str());
204     reject();
205     return;
206   }
207   if (mPreventToUseTwoToolsOnSameInput) {
208     CheckInputList(mSlicerManagersCompatible, mCurrentCompatibleIndex);
209     if (mSlicerManagersCompatible.size() == 0) {
210       QMessageBox::information(mMainWindow, tr("Error"), "Sorry, no other loaded images can use this tool. Abort");
211       reject();
212       return;
213     }
214   }
215   mToolInputSelectionWidget->AddInputSelector(s, mSlicerManagersCompatible, mCurrentCompatibleIndex, allowSkip);
216 }
217 //------------------------------------------------------------------------------
218
219
220 //------------------------------------------------------------------------------
221 void vvToolWidgetBase::AddInputSelector(QString s, bool allowSkip)
222 {
223   mSlicerManagersCompatible.clear();
224   for(unsigned int i=0; i<mMainWindow->GetSlicerManagers().size(); i++) {
225     mSlicerManagersCompatible.push_back(mMainWindow->GetSlicerManagers()[i]);
226   }
227   if (mMainWindow->GetSlicerManagers().size() == 0) {
228     QMessageBox::information(this, "No image","Sorry, could not perform operation. No (compatible) image.");
229     close();
230     return;
231   }
232   mCurrentCompatibleIndex = mMainWindow->GetSlicerManagerCurrentIndex();
233   if (mPreventToUseTwoToolsOnSameInput) {
234     CheckInputList(mSlicerManagersCompatible,  mCurrentCompatibleIndex);
235     if (mSlicerManagersCompatible.size() == 0) {
236       QMessageBox::information(mMainWindow, tr("Error"), "Sorry, no other loaded images can use this tool. Abort");
237       close();
238       return;
239     }
240   }
241   mToolInputSelectionWidget->AddInputSelector(s, mSlicerManagersCompatible, mCurrentCompatibleIndex, allowSkip);
242 }
243 //------------------------------------------------------------------------------
244
245
246 //------------------------------------------------------------------------------
247 void vvToolWidgetBase::HideInputSelector()
248 {
249   mToolInputSelectionWidget->hide();
250 }
251 //------------------------------------------------------------------------------
252
253
254 //------------------------------------------------------------------------------
255 void vvToolWidgetBase::show()
256 {
257   if (!mIsInitialized) {
258     mToolInputSelectionWidget->Initialize();
259     mIsInitialized = true;
260   }
261   QWidget::show();
262 }
263 //------------------------------------------------------------------------------
264
265
266 //------------------------------------------------------------------------------
267 void vvToolWidgetBase::closeEvent(QCloseEvent *event) 
268 {
269   mIsAnotherToolWaitInput = false;
270   if (isWindow()) {
271     event->accept();//return QWidget::close();
272     return;
273   }
274   else {
275     if (!mStaticWidgetForTab) {
276       event->accept();//return QWidget::close();
277       return;
278     }
279     mStaticVerticalLayout->removeWidget(mWidgetForTab);
280     mWidgetForTab->close();
281     delete mWidgetForTab;
282     QList<QObject*> l =mStaticWidgetForTab->children(); 
283     if (l.size() > 1) {
284       QWidget * c = dynamic_cast<QWidget*>(l[1]);
285       c->setVisible(true);
286     }
287   }
288   event->accept();
289 }
290 //------------------------------------------------------------------------------
291
292
293 //------------------------------------------------------------------------------
294 bool vvToolWidgetBase::close()
295 {
296   QApplication::restoreOverrideCursor();
297   return QWidget::close();
298 }
299 //------------------------------------------------------------------------------
300
301
302 //------------------------------------------------------------------------------
303 void vvToolWidgetBase::AnImageIsBeingClosed(vvSlicerManager * m)
304 {
305   mToolInputSelectionWidget->AnImageIsBeingClosed(m);
306   if (m == mCurrentSlicerManager) {
307     close();
308   }
309 }
310 //------------------------------------------------------------------------------
311
312
313 //------------------------------------------------------------------------------
314 void vvToolWidgetBase::SwapCurrentWidget()
315 {
316   mStaticWidgetForTab->setUpdatesEnabled(false);
317   QList<QObject*> l =mStaticWidgetForTab->children(); 
318   for(int i=1; i<l.size(); i++) {
319     QWidget * c = dynamic_cast<QWidget*>(l[i]);
320     if (l[i] == mWidgetForTab) {
321       c->setVisible(true);
322     }
323     else {
324       c->setVisible(false);
325     }
326   } 
327   mStaticWidgetForTab->setUpdatesEnabled(true);
328 }
329 //------------------------------------------------------------------------------
330
331
332 //------------------------------------------------------------------------------
333 void vvToolWidgetBase::SelectedImageHasChanged(vvSlicerManager * m)
334 {
335   if (!isWindow()) { // When the tool is not in a window, it is in a tab : we only display if needed
336     if (mCurrentSlicerManager == NULL) return;
337     if (mToolWidget == NULL) return;
338     if (m != mCurrentSlicerManager) { // current tool is not selected
339       mToolWidget->setEnabled(false);
340     }
341     else { // The current tool is selected
342       SwapCurrentWidget();
343       mToolWidget->setEnabled(true);
344     }
345   }
346 }
347 //------------------------------------------------------------------------------
348
349
350 //------------------------------------------------------------------------------
351 void vvToolWidgetBase::InitializeInputs()
352 {
353 }
354 //------------------------------------------------------------------------------
355
356
357 //------------------------------------------------------------------------------
358 void vvToolWidgetBase::InputIsSelected()
359 {
360   mMainButtonBox->setEnabled(true);
361   std::vector<vvSlicerManager*> & l = mToolInputSelectionWidget->GetSelectedInputs();
362   mCurrentSlicerManager = l[0];
363   mCurrentImage = mCurrentSlicerManager->GetImage();
364   mToolWidget->setEnabled(true);
365   if (!mCurrentSlicerManager) close();
366   if (l.size() == 1) InputIsSelected(mCurrentSlicerManager);
367   else InputIsSelected(l);
368   mIsAnotherToolWaitInput = false;
369 }
370 //------------------------------------------------------------------------------
371
372
373 //------------------------------------------------------------------------------
374 void vvToolWidgetBase::InputIsSelected(vvSlicerManager * m)
375 {
376   std::cerr << "You MUST overwrite this method vvToolWidgetBase::InputIsSelected(vvSlicerManager * m) if you use one single input" << std::endl;
377   exit(0);
378 }
379 //------------------------------------------------------------------------------
380
381
382 //------------------------------------------------------------------------------
383 void vvToolWidgetBase::InputIsSelected(std::vector<vvSlicerManager*> & l)
384 {
385   mMainButtonBox->setEnabled(true);
386   if (l.size() == 1) InputIsSelected(l[0]);
387   else {
388     std::cerr << "You MUST overwrite this method vvToolWidgetBase::InputIsSelected(vector<vvSlicerManager *> m) if you use several input" << std::endl;
389     exit(0);
390   }
391 }
392 //------------------------------------------------------------------------------