//  SMESH SMESHGUI : GUI for SMESH component
//
//  Copyright (C) 2003  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
//  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
//
//  This library is free software; you can redistribute it and/or
//  modify it under the terms of the GNU Lesser General Public
//  License as published by the Free Software Foundation; either
//  version 2.1 of the License.
//
//  This library is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
//  Lesser General Public License for more details.
//
//  You should have received a copy of the GNU Lesser General Public
//  License along with this library; if not, write to the Free Software
//  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
//
//  See http://www.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org
//
//
//
//  File   : SMESHGUI_AddSubMeshDlg.cxx
//  Author : Nicolas REJNERI
//  Module : SMESH
//  $Header$

#include "SMESHGUI_AddSubMeshDlg.h"

#include "SMESHGUI.h"
#include "SMESHGUI_Utils.h"
#include "SMESHGUI_GEOMGenUtils.h"
#include "SMESHGUI_HypothesesUtils.h"

#include "SMESH_TypeFilter.hxx"
#include "SMESH_NumberFilter.hxx"

#include "GEOMBase.h"

#include "SUIT_Session.h"
#include "SUIT_MessageBox.h"
#include "SUIT_OverrideCursor.h"
#include "SUIT_Operation.h"
#include "SUIT_Desktop.h"

#include "SALOMEDSClient_Study.hxx"
#include "SALOMEDS_SObject.hxx"
#include "SALOME_ListIO.hxx"
#include "SALOME_ListIteratorOfListIO.hxx"
#include "SalomeApp_Tools.h"

#include "SVTK_ViewModel.h"

#include "utilities.h"

// QT Includes
#include <qgroupbox.h>
#include <qlabel.h>
#include <qlineedit.h>
#include <qpushbutton.h>
#include <qlayout.h>
#include <qpixmap.h>

using namespace std;

namespace SMESH {
  SMESH::SMESH_subMesh_var AddSubMesh (SMESH::SMESH_Mesh_ptr theMesh,
                                       GEOM::GEOM_Object_ptr theShapeObject,
                                       const char* theMeshName)
  {
    SMESH::SMESH_subMesh_var aSubMesh;
    try {
      if (!theMesh->_is_nil() && !theShapeObject->_is_nil())
	aSubMesh = theMesh->GetSubMesh(theShapeObject, theMeshName);
    } catch (const SALOME::SALOME_Exception& S_ex) {
      SalomeApp_Tools::QtCatchCorbaException(S_ex);
    }
    return aSubMesh._retn();
  }
}

//=================================================================================
// function : SMESHGUI_AddSubMeshDlg()
// purpose  : Constructs a SMESHGUI_AddSubMeshDlg which is a child of 'parent', with the
//            name 'name' and widget flags set to 'f'.
//            The dialog will by default be modeless, unless you set 'modal' to
//            TRUE to construct a modal dialog.
//=================================================================================
SMESHGUI_AddSubMeshDlg::SMESHGUI_AddSubMeshDlg( SMESHGUI* theModule, const char* name,
                                                bool modal, WFlags fl)
     : QDialog( SMESH::GetDesktop( theModule ), name, modal, WStyle_Customize | WStyle_NormalBorder |
                WStyle_Title | WStyle_SysMenu | WDestructiveClose),
     mySMESHGUI( theModule ),
     mySelectionMgr( SMESH::GetSelectionMgr( theModule ) )
{
    QPixmap image0 (SMESHGUI::resourceMgr()->loadPixmap("SMESH", tr("ICON_SELECT")));
    if (!name)
      setName("SMESHGUI_AddSubMeshDlg");
    setCaption(tr("SMESH_ADD_SUBMESH"));
    setSizeGripEnabled(TRUE);
    QGridLayout* SMESHGUI_AddSubMeshDlgLayout = new QGridLayout(this);
    SMESHGUI_AddSubMeshDlgLayout->setSpacing(6);
    SMESHGUI_AddSubMeshDlgLayout->setMargin(11);

    /***************************************************************/
    GroupC1 = new QGroupBox (tr("SMESH_ARGUMENTS"), this, "GroupC1");
    GroupC1->setColumnLayout(0, Qt::Vertical);
    GroupC1->layout()->setSpacing(0);
    GroupC1->layout()->setMargin(0);
    QGridLayout* GroupC1Layout = new QGridLayout (GroupC1->layout());
    GroupC1Layout->setAlignment(Qt::AlignTop);
    GroupC1Layout->setSpacing(6);
    GroupC1Layout->setMargin(11);

    TextLabel_NameMesh = new QLabel (tr("SMESH_NAME"), GroupC1, "TextLabel_NameMesh");
    GroupC1Layout->addWidget(TextLabel_NameMesh, 0, 0);
    LineEdit_NameMesh = new QLineEdit (GroupC1, "LineEdit_NameMesh");
    GroupC1Layout->addWidget(LineEdit_NameMesh, 0, 2);

    TextLabelC1A1 = new QLabel (tr("SMESH_OBJECT_MESH"), GroupC1, "TextLabelC1A1");
    GroupC1Layout->addWidget(TextLabelC1A1, 1, 0);
    SelectButtonC1A1 = new QPushButton (GroupC1, "SelectButtonC1A1");
    SelectButtonC1A1->setPixmap(image0);
    SelectButtonC1A1->setToggleButton(FALSE);
    GroupC1Layout->addWidget(SelectButtonC1A1, 1, 1);
    LineEditC1A1 = new QLineEdit (GroupC1, "LineEditC1A1");
    LineEditC1A1->setReadOnly(true);
    GroupC1Layout->addWidget(LineEditC1A1, 1, 2);

    TextLabelC1A2 = new QLabel(tr("SMESH_OBJECT_GEOM"), GroupC1, "TextLabelC1A2");
    GroupC1Layout->addWidget(TextLabelC1A2, 2, 0);
    SelectButtonC1A2 = new QPushButton(GroupC1, "SelectButtonC1A2");
    SelectButtonC1A2->setPixmap(image0);
    SelectButtonC1A2->setToggleButton(FALSE);
    GroupC1Layout->addWidget(SelectButtonC1A2, 2, 1);
    LineEditC1A2 = new QLineEdit(GroupC1, "LineEditC1A2");
    LineEditC1A2->setReadOnly(true);
    GroupC1Layout->addWidget(LineEditC1A2, 2, 2);

    TextLabelC1A1Hyp = new QLabel(tr("SMESH_OBJECT_HYPOTHESIS"), GroupC1, "TextLabelC1A1Hyp");
    GroupC1Layout->addWidget(TextLabelC1A1Hyp, 3, 0);
    SelectButtonC1A1Hyp = new QPushButton(GroupC1, "SelectButtonC1A1Hyp");
    SelectButtonC1A1Hyp->setPixmap(image0);
    GroupC1Layout->addWidget(SelectButtonC1A1Hyp, 3, 1);
    LineEditC1A1Hyp = new QLineEdit(GroupC1, "LineEditC1A1Hyp");
    LineEditC1A1Hyp->setReadOnly(true);
    GroupC1Layout->addWidget(LineEditC1A1Hyp, 3, 2);

    TextLabelC1A1Algo = new QLabel(tr("SMESH_OBJECT_ALGORITHM"), GroupC1, "TextLabelC1A1Algo");
    GroupC1Layout->addWidget(TextLabelC1A1Algo, 4, 0);
    SelectButtonC1A1Algo = new QPushButton(GroupC1, "SelectButtonC1A1Algo");
    SelectButtonC1A1Algo->setPixmap(image0);
    GroupC1Layout->addWidget(SelectButtonC1A1Algo, 4, 1);
    LineEditC1A1Algo = new QLineEdit(GroupC1, "LineEditC1A1Algo");
    LineEditC1A1Algo->setReadOnly(true);
    GroupC1Layout->addWidget(LineEditC1A1Algo, 4, 2);

    SMESHGUI_AddSubMeshDlgLayout->addWidget(GroupC1, 1, 0);

    /***************************************************************/
    GroupButtons = new QGroupBox(this, "GroupButtons");
    GroupButtons->setColumnLayout(0, Qt::Vertical);
    GroupButtons->layout()->setSpacing(0);
    GroupButtons->layout()->setMargin(0);
    QGridLayout* GroupButtonsLayout = new QGridLayout(GroupButtons->layout());
    GroupButtonsLayout->setAlignment(Qt::AlignTop);
    GroupButtonsLayout->setSpacing(6);
    GroupButtonsLayout->setMargin(11);

    buttonOk = new QPushButton(tr("SMESH_BUT_OK"), GroupButtons, "buttonOk");
    buttonOk->setAutoDefault(TRUE);
    buttonOk->setDefault(TRUE);
    GroupButtonsLayout->addWidget(buttonOk, 0, 0);

    buttonApply = new QPushButton(tr("SMESH_BUT_APPLY"), GroupButtons, "buttonApply");
    buttonApply->setAutoDefault(TRUE);
    GroupButtonsLayout->addWidget(buttonApply, 0, 1);

    GroupButtonsLayout->addItem(new QSpacerItem(20, 20, QSizePolicy::Expanding, QSizePolicy::Minimum), 0, 2);

    buttonCancel = new QPushButton(tr("SMESH_BUT_CLOSE"), GroupButtons, "buttonCancel");
    buttonCancel->setAutoDefault(TRUE);
    GroupButtonsLayout->addWidget(buttonCancel, 0, 3);

    SMESHGUI_AddSubMeshDlgLayout->addWidget(GroupButtons, 2, 0);

    /***************************************************************/
    Init();
}

//=================================================================================
// function : ~SMESHGUI_AddSubMeshDlg()
// purpose  : Destroys the object and frees any allocated resources
//=================================================================================
SMESHGUI_AddSubMeshDlg::~SMESHGUI_AddSubMeshDlg()
{
  // no need to delete child widgets, Qt does it all for us
}

//=================================================================================
// function : Init()
// purpose  :
//=================================================================================
void SMESHGUI_AddSubMeshDlg::Init ()
{
  mySMESHGUI->SetActiveDialogBox((QDialog*)this);

  //myGeomFilter       = new SALOME_TypeFilter("GEOM");
  TColStd_MapOfInteger allTypesMap;
  for (int i = 0; i < 10; i++)
    allTypesMap.Add(i);
  myGeomFilter       = new SMESH_NumberFilter ("GEOM", TopAbs_SHAPE, 0, allTypesMap);
  myMeshFilter       = new SMESH_TypeFilter (MESH);
  myAlgorithmFilter  = new SMESH_TypeFilter (ALGORITHM);
  myHypothesisFilter = new SMESH_TypeFilter (HYPOTHESIS);

  /* signals and slots connections */
  connect(buttonOk,     SIGNAL(clicked()), this, SLOT(ClickOnOk()));
  connect(buttonApply,  SIGNAL(clicked()), this, SLOT(ClickOnApply()));
  connect(buttonCancel, SIGNAL(clicked()), this, SLOT(ClickOnCancel()));

  connect(SelectButtonC1A1,     SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));
  connect(SelectButtonC1A2,     SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));
  connect(SelectButtonC1A1Hyp,  SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));
  connect(SelectButtonC1A1Algo, SIGNAL(clicked()), this, SLOT(SetEditCurrentArgument()));

  connect(mySelectionMgr, SIGNAL(currentSelectionChanged()),  this, SLOT(SelectionIntoArgument()));
  connect(mySMESHGUI, SIGNAL(SignalDeactivateActiveDialog()), this, SLOT(DeactivateActiveDialog()));
  connect(mySMESHGUI, SIGNAL(SignalCloseAllDialogs()),        this, SLOT(ClickOnCancel()));

  int x, y;
  mySMESHGUI->DefineDlgPosition(this, x, y);
  this->move(x, y);
  this->show();

  LineEdit_NameMesh->setText(tr("SMESH_SUBMESH"));
  LineEdit_NameMesh->setFocus();
  myEditCurrentArgument = LineEditC1A1;
  mySelectionMgr->clearFilters();
  mySelectionMgr->installFilter(myMeshFilter);

  SelectionIntoArgument();
}

//=================================================================================
// function : ClickOnOk()
// purpose  :
//=================================================================================
void SMESHGUI_AddSubMeshDlg::ClickOnOk()
{
  if (this->ClickOnApply())
    this->ClickOnCancel();
}

//=================================================================================
// function : ClickOnApply()
// purpose  :
//=================================================================================
bool SMESHGUI_AddSubMeshDlg::ClickOnApply()
{
  if (mySMESHGUI->isActiveStudyLocked())
    return false;

  QString myNameSubMesh = LineEdit_NameMesh->text().stripWhiteSpace();
  if (myNameSubMesh.isEmpty()) {
    SUIT_MessageBox::warn1(this, tr("SMESH_WRN_WARNING"),
                           tr("SMESH_WRN_EMPTY_NAME"), tr("SMESH_BUT_OK"));
    return false;
  }

  if (myMesh->_is_nil() || myGeomShape->_is_nil() ||
      (!HypoList.count() && !AlgoList.count()))
    return false;

  _PTR(SObject) aMeshSO = SMESH::FindSObject(myMesh);
  GEOM::GEOM_Object_var myMainShape = SMESH::GetShapeOnMeshOrSubMesh(aMeshSO);
  if (myMainShape->_is_nil())
    return false;

  SUIT_OverrideCursor wc;

  SUIT_Operation* op =
    new SUIT_Operation (SUIT_Session::session()->activeApplication());

  // start transaction
  op->start();

  // create submesh
  SMESH::SMESH_subMesh_var aSubMesh = SMESH::AddSubMesh(myMesh, myGeomShape, myNameSubMesh);
  int nbSuccess = 0;

  if (!aSubMesh->_is_nil()) {
    // assign hypotheses
    int nbAlgo = AlgoList.count();
    int nbHyps = HypoList.count() + nbAlgo;
    for (int i = 0; i < nbHyps; i++) {
      _PTR(SObject) aHypSOClient =
        SMESH::GetActiveStudyDocument()->FindObjectID
          (i < nbAlgo ? AlgoList[i].latin1() : HypoList[i-nbAlgo].latin1());
      if (aHypSOClient) {
	CORBA::Object_var anObject = _CAST(SObject,aHypSOClient)->GetObject();
	if (!CORBA::is_nil(anObject)) {
	  SMESH::SMESH_Hypothesis_var aHyp = SMESH::SMESH_Hypothesis::_narrow(anObject);
	  if (!aHyp->_is_nil())
	    if (SMESH::AddHypothesisOnSubMesh(aSubMesh, aHyp))
              nbSuccess++;
	} else {
          SCRUTE(CORBA::is_nil(anObject));
        }
      } else {
        SCRUTE(!aHypSOClient);
      }
    }
  } else {
    SCRUTE(aSubMesh->_is_nil());
  }

  // commit transaction
  op->commit();
  mySMESHGUI->updateObjBrowser();
  return (nbSuccess > 0);
}

//=================================================================================
// function : ClickOnCancel()
// purpose  :
//=================================================================================
void SMESHGUI_AddSubMeshDlg::ClickOnCancel()
{
  close();
}

//=================================================================================
// function : IsFatherOf()
// purpose  :
//=================================================================================
static bool IsFatherOf (_PTR(SObject) SO, _PTR(SObject) fatherSO)
{
  if (SO && fatherSO) {
    _PTR(SObject) aSO = SO->GetFather();
    //while (strlen(aSO->GetID()) >= strlen(fatherSO->GetID())) {
    while (aSO->GetID().length() >= fatherSO->GetID().length()) {
      //if (QString(aSO->GetID()) == QString(fatherSO->GetID()))
      if (aSO->GetID() == fatherSO->GetID())
	return true;
      aSO = aSO->GetFather();
    }
  }
  return false;
}

//=================================================================================
// function : SelectionIntoArgument()
// purpose  : Called when selection as changed or other case
//=================================================================================
void SMESHGUI_AddSubMeshDlg::SelectionIntoArgument()
{
  QString aString = "";

  SALOME_ListIO aList;
  mySelectionMgr->selectedObjects(aList, SVTK_Viewer::Type());
  int nbSel = SMESH::GetNameOfSelectedIObjects(mySelectionMgr, aString);

  if (myEditCurrentArgument == LineEditC1A1) {
    // mesh
    if (nbSel != 1) {
      myMesh = SMESH::SMESH_Mesh::_nil();
      aString = "";
    } else {
      Handle(SALOME_InteractiveObject) IO = aList.First();
      myMesh = SMESH::IObjectToInterface<SMESH::SMESH_Mesh>(IO);
      if (myMesh->_is_nil()) {
	aString = "";
      }
    }
    myGeomShape = GEOM::GEOM_Object::_nil();
    LineEditC1A2->setText("");

  } else if (myEditCurrentArgument == LineEditC1A2) {
    // geom shape
    if (nbSel != 1) {
      myGeomShape = GEOM::GEOM_Object::_nil();
      aString = "";
    } else {
      Handle(SALOME_InteractiveObject) IO = aList.First();
      myGeomShape = SMESH::IObjectToInterface<GEOM::GEOM_Object>(IO);
      if (myGeomShape->_is_nil() || !GEOMBase::IsShape(myGeomShape)) {
        myGeomShape = GEOM::GEOM_Object::_nil();
        aString = "";
      }
      if (!myMesh->_is_nil()) {
	_PTR(SObject) aMeshSO = SMESH::FindSObject(myMesh);
	GEOM::GEOM_Object_var aMainGeomShape = SMESH::GetShapeOnMeshOrSubMesh(aMeshSO);
	_PTR(SObject) aMainGeomShapeSO = SMESH::FindSObject(aMainGeomShape);
	if (!aMainGeomShapeSO ||
            !IsFatherOf(SMESH::GetActiveStudyDocument()->FindObjectID
                        (IO->getEntry()), aMainGeomShapeSO)) {
	  myGeomShape = GEOM::GEOM_Object::_nil();
	  aString = "";
	}
      }
    }
  } else if (myEditCurrentArgument == LineEditC1A1Hyp) {
    // hypotheses
    HypoList.clear();
    if (nbSel >= 1) {
      SALOME_ListIteratorOfListIO Itinit (aList);
      for (; Itinit.More(); Itinit.Next()) {
	HypoList.append(Itinit.Value()->getEntry());
      }
      if (nbSel > 1)
	aString = tr("%1 Hypothesis").arg(nbSel);
    } else {
      aString = "";
    }
  } else if (myEditCurrentArgument == LineEditC1A1Algo) {
    // algorithms
    AlgoList.clear();
    if (nbSel >= 1) {
      SALOME_ListIteratorOfListIO Itinit (aList);
      for (; Itinit.More(); Itinit.Next()) {
	AlgoList.append(Itinit.Value()->getEntry());
      }
      if (nbSel > 1)
	aString = tr("%1 Algorithms").arg(nbSel);
    } else {
      aString = "";
    }
  } else {
  }

  myEditCurrentArgument->setText(aString);
  myEditCurrentArgument->setCursorPosition( 0 );

  UpdateControlState();
}

//=================================================================================
// function : SetEditCurrentArgument()
// purpose  :
//=================================================================================
void SMESHGUI_AddSubMeshDlg::SetEditCurrentArgument()
{
  QPushButton* send = (QPushButton*)sender();
  if(send == SelectButtonC1A1) {
    LineEditC1A1->setFocus();
    mySelectionMgr->clearFilters();
    mySelectionMgr->installFilter(myMeshFilter);
    myEditCurrentArgument = LineEditC1A1;
  } else if (send == SelectButtonC1A2) {
    LineEditC1A2->setFocus();
    mySelectionMgr->clearFilters();
    mySelectionMgr->installFilter(myGeomFilter);
    myEditCurrentArgument = LineEditC1A2;
  } else if(send == SelectButtonC1A1Hyp) {
    LineEditC1A1Hyp->setFocus();
    mySelectionMgr->clearFilters();
    mySelectionMgr->installFilter(myHypothesisFilter);
    myEditCurrentArgument = LineEditC1A1Hyp;
  } else if(send == SelectButtonC1A1Algo) {
    LineEditC1A1Algo->setFocus();
    mySelectionMgr->clearFilters();
    mySelectionMgr->installFilter(myAlgorithmFilter);
    myEditCurrentArgument = LineEditC1A1Algo;
  }
  SelectionIntoArgument();
}

//=================================================================================
// function : DeactivateActiveDialog()
// purpose  :
//=================================================================================
void SMESHGUI_AddSubMeshDlg::DeactivateActiveDialog()
{
  if (GroupC1->isEnabled()) {
    disconnect(mySelectionMgr, 0, this, 0);
    GroupC1->setEnabled(false);
    GroupButtons->setEnabled(false);
  }
}

//=================================================================================
// function : ActivateThisDialog()
// purpose  :
//=================================================================================
void SMESHGUI_AddSubMeshDlg::ActivateThisDialog()
{
  mySMESHGUI->EmitSignalDeactivateDialog();
  GroupC1->setEnabled(true);
  GroupButtons->setEnabled(true);
  connect (mySelectionMgr, SIGNAL(currentSelectionChanged()), this, SLOT(SelectionIntoArgument()));
}

//=================================================================================
// function : enterEvent()
// purpose  :
//=================================================================================
void SMESHGUI_AddSubMeshDlg::enterEvent (QEvent*)
{
  if (!GroupC1->isEnabled())
    ActivateThisDialog();
}

//=================================================================================
// function : closeEvent()
// purpose  :
//=================================================================================
void SMESHGUI_AddSubMeshDlg::closeEvent (QCloseEvent* e)
{
  disconnect(mySelectionMgr, 0, this, 0);
  mySMESHGUI->ResetState();
  mySelectionMgr->clearFilters();
  QDialog::closeEvent(e);
}

//=================================================================================
// function : UpdateControlState()
// purpose  :
//=================================================================================
void SMESHGUI_AddSubMeshDlg::UpdateControlState()
{
  bool isEnabled = (!myMesh->_is_nil() &&
                    !myGeomShape->_is_nil() &&
                    (HypoList.count() || AlgoList.count()));
  bool isImportedMesh = false;
  if (!myMesh->_is_nil()) {
    _PTR(SObject) aMeshSO = SMESH::FindSObject(myMesh);
    GEOM::GEOM_Object_var myGeomShape = SMESH::GetShapeOnMeshOrSubMesh(aMeshSO);
    isImportedMesh = myGeomShape->_is_nil();
  }

  buttonOk   ->setEnabled(isEnabled && !isImportedMesh);
  buttonApply->setEnabled(isEnabled && !isImportedMesh);
}