/*=========================================================================
  Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv

  Authors belong to:
  - University of LYON              http://www.universite-lyon.fr/
  - Léon Bérard cancer center       http://www.centreleonberard.fr
  - CREATIS CNRS laboratory         http://www.creatis.insa-lyon.fr

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.  See the copyright notices for more information.

  It is distributed under dual licence

  - BSD        See included LICENSE.txt file
  - CeCILL-B   http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
===========================================================================**/

#include "vvToolImageArithm.h"
#include "vvSlicer.h"
#include "clitkImageArithmGenericFilter.h"
#include <QMessageBox>

//------------------------------------------------------------------------------
// Create the tool and automagically (I like this word) insert it in
// the main window menu.
ADD_TOOL(vvToolImageArithm);
//------------------------------------------------------------------------------

//------------------------------------------------------------------------------
vvToolImageArithm::vvToolImageArithm(vvMainWindowBase * parent, Qt::WindowFlags f)
  :vvToolWidgetBase(parent, f),
   vvToolBase<vvToolImageArithm>(parent),
   Ui::vvToolImageArithm()
{
  // Setup the UI
  Ui_vvToolImageArithm::setupUi(mToolWidget);

  // Main filter
  mFilter = new clitk::ImageArithmGenericFilter<args_info_clitkImageArithm>;

  // Set how many inputs are needed for this tool
  AddInputSelector("Select first image (A)", mFilter);
  AddInputSelector("Select second image (B)", mFilter, true);
}
//------------------------------------------------------------------------------


//------------------------------------------------------------------------------
vvToolImageArithm::~vvToolImageArithm()
{
}
//------------------------------------------------------------------------------


//------------------------------------------------------------------------------
void vvToolImageArithm::Initialize()
{
  SetToolName("ImageArithm");
  SetToolMenuName("ImageArithm");
  SetToolIconFilename(":/common/icons/arithm.png");
  SetToolTip("Perform simple arithmetic operations on one or two images.");
  SetToolExperimental(false);
}
//------------------------------------------------------------------------------


//------------------------------------------------------------------------------
void vvToolImageArithm::InputIsSelected(std::vector<vvSlicerManager *> & l)
{
  mInput1 = l[0];
  mInput2 = l[1];
  mTwoInputs = true;
  mGroupBoxOneInput->setEnabled(false);
  mGroupBoxTwoInputs->setEnabled(true);
}
//------------------------------------------------------------------------------


//------------------------------------------------------------------------------
void vvToolImageArithm::InputIsSelected(vvSlicerManager * l)
{
  mInput1 = l;
  mTwoInputs = false;
  mGroupBoxTwoInputs->setEnabled(false);
  mGroupBoxOneInput->setEnabled(true);
}
//------------------------------------------------------------------------------


//------------------------------------------------------------------------------
void vvToolImageArithm::GetArgsInfoFromGUI()
{
  mArgsInfo.input1_given = false;
  if (mTwoInputs) {
    mArgsInfo.input2_given = true;
    mArgsInfo.input2_arg = new char; // need to indicate that there are two inputs
    mArgsInfo.scalar_given = false;
    if (radioButtonSum->isChecked()) mArgsInfo.operation_arg = 0;
    if (radioButtonMultiply->isChecked()) mArgsInfo.operation_arg = 1;
    if (radioButtonDivide->isChecked()) mArgsInfo.operation_arg = 2;
    if (radioButtonMax->isChecked()) mArgsInfo.operation_arg = 3;
    if (radioButtonMin->isChecked()) mArgsInfo.operation_arg = 4;
    if (radioButtonAbsDiff->isChecked()) mArgsInfo.operation_arg = 5;
    if (radioButtonSquaredDiff->isChecked()) mArgsInfo.operation_arg = 6;
    if (radioButtonDifference->isChecked()) mArgsInfo.operation_arg = 7;
    if (radioButtonRelativeDiff->isChecked()) mArgsInfo.operation_arg = 8;
  } else {
    mArgsInfo.input2_given = false;
    mArgsInfo.scalar_given = true;
    if (radioButtonSumV->isChecked()) mArgsInfo.operation_arg = 0;
    if (radioButtonMultiplyV->isChecked()) mArgsInfo.operation_arg = 1;
    if (radioButtonInverseV->isChecked()) mArgsInfo.operation_arg = 2;
    if (radioButtonMaxV->isChecked()) mArgsInfo.operation_arg = 3;
    if (radioButtonMinV->isChecked()) mArgsInfo.operation_arg = 4;
    if (radioButtonAbsDiffV->isChecked()) mArgsInfo.operation_arg = 5;
    if (radioButtonSquaredDiffV->isChecked()) mArgsInfo.operation_arg = 6;
    if (radioButtonLogAlone->isChecked()) mArgsInfo.operation_arg = 7;
    if (radioButtonExpAlone->isChecked()) mArgsInfo.operation_arg = 8;
    if (radioButtonSqrtV->isChecked()) mArgsInfo.operation_arg = 9;
    if (radioButtonDivideV->isChecked()) mArgsInfo.operation_arg = 11;
    if (radioButtonNormalize->isChecked()) mArgsInfo.operation_arg = 12;
    if (radioButtonLogV->isChecked()) mArgsInfo.operation_arg = 13;
    mArgsInfo.scalar_given = true;
    mArgsInfo.scalar_arg = mValueSpinBox->value();
  }
  mArgsInfo.output_given = false;
  mArgsInfo.verbose_flag = false;
  mArgsInfo.setFloatOutput_flag = mCheckBoxUseFloatOutputType->isChecked();
  mArgsInfo.imagetypes_flag = false;
}
//------------------------------------------------------------------------------


//------------------------------------------------------------------------------
void vvToolImageArithm::apply()
{
  if (!mCurrentSlicerManager) close();
  QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));
  GetArgsInfoFromGUI();

  std::vector<vvImage::Pointer> inputs;
  if (mTwoInputs) {
    // Input
    inputs.push_back(mInput1->GetImage());
    inputs.push_back(mInput2->GetImage());

    // Check input type
    if (inputs[0]->GetScalarTypeAsITKString() != inputs[1]->GetScalarTypeAsITKString()) {
      std::cerr << "Sorry inputs should have the same pixeltype." << std::endl;
      std::cerr << "Input1 = " << inputs[0]->GetScalarTypeAsITKString() << std::endl;
      std::cerr << "Input2 = " << inputs[1]->GetScalarTypeAsITKString() << std::endl;
      QApplication::restoreOverrideCursor();
      QMessageBox::information(this, "Wrong image type","Sorry, could not perform operation. Please select inputs with same pixel type.");
      close();
      return;
    }
    
    // Check size
    if (!mInput1->GetImage()->HaveSameSizeAndSpacingThan(mInput2->GetImage())) {
      std::cerr << "Sorry inputs should have the same size and spacing." << std::endl;
      QApplication::restoreOverrideCursor();
      QMessageBox::information(this, "Wrong images size","Sorry, could not perform operation. Please select inputs with same size and spacing.");
      close();
      return;
    }
  } else {
    // Input
    inputs.push_back(mInput1->GetImage());
  }

  // Main filter
  clitk::ImageArithmGenericFilter<args_info_clitkImageArithm>::Pointer filter =
    clitk::ImageArithmGenericFilter<args_info_clitkImageArithm>::New();
  filter->SetInputVVImages(inputs);
  filter->SetArgsInfo(mArgsInfo);
  filter->EnableReadOnDisk(false);
  filter->EnableOverwriteInputImage(false);
  filter->Update();

  // Output
  vvImage::Pointer output = filter->GetOutputVVImage();
  std::ostringstream osstream;
  osstream << "Arithm_" << mArgsInfo.operation_arg << "_ "
           << mCurrentSlicerManager->GetSlicer(0)->GetFileName() << ".mhd";
  AddImage(output,osstream.str());
  QApplication::restoreOverrideCursor();
  close();
}
//------------------------------------------------------------------------------