]> Creatis software - clitk.git/blob - vv/vvToolImageArithm.cxx
Add clitkImage2DicomSeries tool to write a Dicom Series from an image without a corre...
[clitk.git] / vv / vvToolImageArithm.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 #include "vvToolImageArithm.h"
20 #include "vvSlicer.h"
21 #include "clitkImageArithmGenericFilter.h"
22 #include <QMessageBox>
23
24 //------------------------------------------------------------------------------
25 // Create the tool and automagically (I like this word) insert it in
26 // the main window menu.
27 ADD_TOOL(vvToolImageArithm);
28 //------------------------------------------------------------------------------
29
30 //------------------------------------------------------------------------------
31 vvToolImageArithm::vvToolImageArithm(vvMainWindowBase * parent, Qt::WindowFlags f)
32   :vvToolWidgetBase(parent, f),
33    vvToolBase<vvToolImageArithm>(parent),
34    Ui::vvToolImageArithm()
35 {
36   // Setup the UI
37   Ui_vvToolImageArithm::setupUi(mToolWidget);
38
39   // Main filter
40   mFilter = new clitk::ImageArithmGenericFilter<args_info_clitkImageArithm>;
41
42   // Set how many inputs are needed for this tool
43   AddInputSelector("Select first image (A)", mFilter);
44   AddInputSelector("Select second image (B)", mFilter, true);
45 }
46 //------------------------------------------------------------------------------
47
48
49 //------------------------------------------------------------------------------
50 vvToolImageArithm::~vvToolImageArithm()
51 {
52 }
53 //------------------------------------------------------------------------------
54
55
56 //------------------------------------------------------------------------------
57 void vvToolImageArithm::Initialize()
58 {
59   SetToolName("ImageArithm");
60   SetToolMenuName("ImageArithm");
61   SetToolIconFilename(":/common/icons/arithm.png");
62   SetToolTip("Perform simple arithmetic operations on one or two images.");
63   SetToolExperimental(false);
64 }
65 //------------------------------------------------------------------------------
66
67
68 //------------------------------------------------------------------------------
69 void vvToolImageArithm::InputIsSelected(std::vector<vvSlicerManager *> & l)
70 {
71   mInput1 = l[0];
72   mInput2 = l[1];
73   mTwoInputs = true;
74   mGroupBoxOneInput->setEnabled(false);
75   mGroupBoxTwoInputs->setEnabled(true);
76 }
77 //------------------------------------------------------------------------------
78
79
80 //------------------------------------------------------------------------------
81 void vvToolImageArithm::InputIsSelected(vvSlicerManager * l)
82 {
83   mInput1 = l;
84   mTwoInputs = false;
85   mGroupBoxTwoInputs->setEnabled(false);
86   mGroupBoxOneInput->setEnabled(true);
87 }
88 //------------------------------------------------------------------------------
89
90
91 //------------------------------------------------------------------------------
92 void vvToolImageArithm::GetArgsInfoFromGUI()
93 {
94   mArgsInfo.input1_given = false;
95   if (mTwoInputs) {
96     mArgsInfo.input2_given = true;
97     mArgsInfo.input2_arg = new char; // need to indicate that there are two inputs
98     mArgsInfo.scalar_given = false;
99     if (radioButtonSum->isChecked()) mArgsInfo.operation_arg = 0;
100     if (radioButtonMultiply->isChecked()) mArgsInfo.operation_arg = 1;
101     if (radioButtonDivide->isChecked()) mArgsInfo.operation_arg = 2;
102     if (radioButtonMax->isChecked()) mArgsInfo.operation_arg = 3;
103     if (radioButtonMin->isChecked()) mArgsInfo.operation_arg = 4;
104     if (radioButtonAbsDiff->isChecked()) mArgsInfo.operation_arg = 5;
105     if (radioButtonSquaredDiff->isChecked()) mArgsInfo.operation_arg = 6;
106   } else {
107     mArgsInfo.input2_given = false;
108     mArgsInfo.scalar_given = true;
109     if (radioButtonSumV->isChecked()) mArgsInfo.operation_arg = 0;
110     if (radioButtonMultiplyV->isChecked()) mArgsInfo.operation_arg = 1;
111     if (radioButtonInverseV->isChecked()) mArgsInfo.operation_arg = 2;
112     if (radioButtonMaxV->isChecked()) mArgsInfo.operation_arg = 3;
113     if (radioButtonMinV->isChecked()) mArgsInfo.operation_arg = 4;
114     if (radioButtonAbsDiffV->isChecked()) mArgsInfo.operation_arg = 5;
115     if (radioButtonSquaredDiffV->isChecked()) mArgsInfo.operation_arg = 6;
116     if (radioButtonLogV->isChecked()) mArgsInfo.operation_arg = 7;
117     if (radioButtonExpV->isChecked()) mArgsInfo.operation_arg = 8;
118     if (radioButtonSqrtV->isChecked()) mArgsInfo.operation_arg = 9;
119     mArgsInfo.scalar_given = true;
120     mArgsInfo.scalar_arg = mValueSpinBox->value();
121   }
122   mArgsInfo.output_given = false;
123   mArgsInfo.verbose_flag = false;
124   mArgsInfo.setFloatOutput_flag = mCheckBoxUseFloatOutputType->isChecked();
125   mArgsInfo.imagetypes_flag = false;
126 }
127 //------------------------------------------------------------------------------
128
129
130 //------------------------------------------------------------------------------
131 void vvToolImageArithm::apply()
132 {
133   if (!mCurrentSlicerManager) close();
134   QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));
135   GetArgsInfoFromGUI();
136
137   std::vector<vvImage::Pointer> inputs;
138   if (mTwoInputs) {
139     // Input
140     inputs.push_back(mInput1->GetImage());
141     inputs.push_back(mInput2->GetImage());
142
143     // Check input type
144     if (inputs[0]->GetScalarTypeAsITKString() != inputs[1]->GetScalarTypeAsITKString()) {
145       std::cerr << "Sorry inputs should have the same pixeltype." << std::endl;
146       std::cerr << "Input1 = " << inputs[0]->GetScalarTypeAsITKString() << std::endl;
147       std::cerr << "Input2 = " << inputs[1]->GetScalarTypeAsITKString() << std::endl;
148       QApplication::restoreOverrideCursor();
149       QMessageBox::information(this, "Wrong image type","Sorry, could not perform operation. Please select inputs with same pixel type.");
150       close();
151       return;
152     }
153     
154     // Check size
155     if (!mInput1->GetImage()->HaveSameSizeAndSpacingThan(mInput2->GetImage())) {
156       std::cerr << "Sorry inputs should have the same size and spacing." << std::endl;
157       QApplication::restoreOverrideCursor();
158       QMessageBox::information(this, "Wrong images size","Sorry, could not perform operation. Please select inputs with same size and spacing.");
159       close();
160       return;
161     }
162   } else {
163     // Input
164     inputs.push_back(mInput1->GetImage());
165   }
166
167   // Main filter
168   clitk::ImageArithmGenericFilter<args_info_clitkImageArithm>::Pointer filter =
169     clitk::ImageArithmGenericFilter<args_info_clitkImageArithm>::New();
170   filter->SetInputVVImages(inputs);
171   filter->SetArgsInfo(mArgsInfo);
172   filter->EnableReadOnDisk(false);
173   filter->EnableOverwriteInputImage(false);
174   filter->Update();
175
176   // Output
177   vvImage::Pointer output = filter->GetOutputVVImage();
178   std::ostringstream osstream;
179   osstream << "Arithm_" << mArgsInfo.operation_arg << "_ "
180            << mCurrentSlicerManager->GetSlicer(0)->GetFileName() << ".mhd";
181   AddImage(output,osstream.str());
182   QApplication::restoreOverrideCursor();
183   close();
184 }
185 //------------------------------------------------------------------------------