// Copyright (C) 2007-2016  CEA/DEN, EDF R&D, OPEN CASCADE
//
// 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, or (at your option) any later version.
//
// 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.salome-platform.org/ or email : webmaster.salome@opencascade.com
//

// File   : SMESHGUI_FindElemByPointDlg.cxx
// Author : Edward AGAPOV, Open CASCADE S.A.S.
//

// SMESH includes
#include "SMESHGUI_FindElemByPointDlg.h"

#include "SMESHGUI.h"
#include "SMESHGUI_MeshUtils.h"
#include "SMESHGUI_VTKUtils.h"
#include "SMESHGUI_SpinBox.h"
#include "SMESHGUI_MeshEditPreview.h"

#include "SMESH_Actor.h"
#include "SMESH_ActorUtils.h"
#include "SMESH_TypeFilter.hxx"
#include "SMESH_LogicalFilter.hxx"

// SALOME GUI includes
#include <LightApp_SelectionMgr.h>
#include <QtxComboBox.h>
#include <SALOME_ListIO.hxx>
#include <SUIT_Desktop.h>
#include <SUIT_MessageBox.h>
#include <SUIT_OverrideCursor.h>
#include <SUIT_ResourceMgr.h>
#include <SVTK_ViewModel.h>
#include <SVTK_ViewWindow.h>
#include <SalomeApp_Tools.h>
#include <SalomeApp_TypeFilter.h>

// Qt includes
#include <QAbstractButton>
#include <QGridLayout>
#include <QGroupBox>
#include <QHBoxLayout>
#include <QLabel>
#include <QLineEdit>
#include <QListWidget>
#include <QPushButton>
#include <QVBoxLayout>

// VTK includes
#include <vtkProperty.h>

// IDL includes
#include <SALOMEconfig.h>
#include CORBA_SERVER_HEADER(SMESH_MeshEditor)

#define SPACING 6
#define MARGIN  11

//=======================================================================
/*!
 * \brief Dialog to find elements by a point coordinates
 */
//=======================================================================

SMESHGUI_FindElemByPointDlg::SMESHGUI_FindElemByPointDlg()
  : SMESHGUI_Dialog( 0, false, true, OK | Help )
{
  setWindowTitle(tr("CAPTION"));

  QVBoxLayout* aDlgLay = new QVBoxLayout (mainFrame());
  aDlgLay->setMargin(0);
  aDlgLay->setSpacing(SPACING);

  QWidget* aMainFrame = createMainFrame  (mainFrame());

  aDlgLay->addWidget(aMainFrame);
  aDlgLay->setStretchFactor(aMainFrame, 1);
}

//================================================================================
/*!
 * \brief Set types of elements available for search
 */
//================================================================================

void SMESHGUI_FindElemByPointDlg::setTypes(SMESH::array_of_ElementType_var & types)
{
  myElemTypeCombo->blockSignals(true);
  myElemTypeCombo->clear();
  int nbTypes = 0, hasNodes = 0;
  for ( int i = 0; i < (int) types->length(); ++i )
  {
    switch ( types[i] ) {
    case SMESH::NODE:
      myElemTypeCombo->addItem( tr( "MEN_NODE" ));
      myElemTypeCombo->setId( nbTypes++, int( SMESH::NODE  ));
      hasNodes = 1;
      break;
    case SMESH::EDGE:
      myElemTypeCombo->addItem( tr( "MEN_EDGE" ));
      myElemTypeCombo->setId( nbTypes++, int( SMESH::EDGE  ));
      break;
    case SMESH::FACE:
      myElemTypeCombo->addItem( tr( "MEN_FACE" ));
      myElemTypeCombo->setId( nbTypes++, int( SMESH::FACE  ));
      break;
    case SMESH::VOLUME:
      myElemTypeCombo->addItem( tr( "MEN_VOLUME_3D" ));
      myElemTypeCombo->setId( nbTypes++, int( SMESH::VOLUME  ));
      break;
    case SMESH::ELEM0D:
      myElemTypeCombo->addItem( tr( "MEN_ELEM0D" ));
      myElemTypeCombo->setId( nbTypes++, int( SMESH::ELEM0D  ));
      break;
    case SMESH::BALL:
      myElemTypeCombo->addItem( tr( "MEN_BALL" ));
      myElemTypeCombo->setId( nbTypes++, int( SMESH::BALL  ));
      break;
    default:;
    }
  }
  if ( nbTypes - hasNodes > 1 )
  {
    myElemTypeCombo->addItem( tr( "MEN_ALL" ));
    myElemTypeCombo->setId( nbTypes++, int( SMESH::ALL   ));
  }
  if ( !hasNodes && nbTypes > 0 )
  {
    myElemTypeCombo->addItem( tr( "MEN_NODE" ));
    myElemTypeCombo->setId( nbTypes++, int( SMESH::NODE  ));
  }
  myElemTypeCombo->blockSignals(false);
}

//=======================================================================
// function : createMainFrame()
// purpose  : Create frame containing dialog's input fields
//=======================================================================
QWidget* SMESHGUI_FindElemByPointDlg::createMainFrame (QWidget* theParent)
{
  QWidget* aFrame = new QWidget(theParent);

  //mesh name

  QGroupBox* aMeshGrp = new QGroupBox(tr("MESH_GROUP"), aFrame);
  QHBoxLayout* aMeshGrpLayout = new QHBoxLayout(aMeshGrp);
  aMeshGrpLayout->setMargin(MARGIN);
  aMeshGrpLayout->setSpacing(SPACING);

  myMeshName = new QLineEdit(aMeshGrp);
  aMeshGrpLayout->addWidget(myMeshName);

  // coordinates

  QGroupBox* aCoordGrp = new QGroupBox(tr("SMESH_COORDINATES"), aFrame);
  QHBoxLayout* aCoordGrpLayout = new QHBoxLayout(aCoordGrp);
  aCoordGrpLayout->setMargin(MARGIN);
  aCoordGrpLayout->setSpacing(SPACING);

  QLabel* aXLabel = new QLabel(tr("SMESH_X"), aCoordGrp);
  myX = new SMESHGUI_SpinBox(aCoordGrp);

  QLabel* aYLabel = new QLabel(tr("SMESH_Y"), aCoordGrp);
  myY = new SMESHGUI_SpinBox(aCoordGrp);

  QLabel* aZLabel = new QLabel(tr("SMESH_Z"), aCoordGrp);
  myZ = new SMESHGUI_SpinBox(aCoordGrp);

  myX->RangeStepAndValidator(COORD_MIN, COORD_MAX, 10.0, "length_precision");
  myY->RangeStepAndValidator(COORD_MIN, COORD_MAX, 10.0, "length_precision");
  myZ->RangeStepAndValidator(COORD_MIN, COORD_MAX, 10.0, "length_precision");
  myX->SetValue(0);
  myY->SetValue(0);
  myZ->SetValue(0);

  aCoordGrpLayout->addWidget(aXLabel);
  aCoordGrpLayout->addWidget(myX);
  aCoordGrpLayout->addWidget(aYLabel);
  aCoordGrpLayout->addWidget(myY);
  aCoordGrpLayout->addWidget(aZLabel);
  aCoordGrpLayout->addWidget(myZ);

  // Elements

  QGroupBox* elementGrp = new QGroupBox(tr("Elements"), aFrame);
  QGridLayout* elementGrpLayout = new QGridLayout(elementGrp);
  elementGrpLayout->setSpacing(SPACING);
  elementGrpLayout->setMargin(MARGIN);


  myFindBtn = new QPushButton(elementGrp);
  myFindBtn->setText(tr("Find"));
  //myFindBtn->setCheckable(false);

  myElemTypeCombo = new QtxComboBox(elementGrp);

  myFoundList = new QListWidget(elementGrp);

  elementGrpLayout->addWidget( myFindBtn,       0, 0 );
  elementGrpLayout->addWidget( myElemTypeCombo, 0, 1 );
  elementGrpLayout->addWidget( myFoundList,     1, 0, 2, 2 );


  QVBoxLayout* aLay = new QVBoxLayout(aFrame);
  aLay->setMargin( 0 );
  aLay->setSpacing( SPACING );
  aLay->addWidget(aMeshGrp);
  aLay->addWidget(aCoordGrp);
  aLay->addWidget(elementGrp);

  // OK instead of "Apply and Close"
  if ( QAbstractButton* but = button(OK) )
    but->setText( tr("SMESH_BUT_OK"));

  return aFrame;
}

//================================================================================
/*!
 * \brief Constructor
*/
//================================================================================

SMESHGUI_FindElemByPointOp::SMESHGUI_FindElemByPointOp()
  :SMESHGUI_SelectionOp()
{
  mySimulation = 0;
  mySMESHGUI = 0;
  myDlg = new SMESHGUI_FindElemByPointDlg;
  myHelpFileName = "find_element_by_point_page.html";

  QList<SUIT_SelectionFilter*> filters;
  filters.append( new SMESH_TypeFilter( SMESH::MESH ) );
  filters.append( new SMESH_TypeFilter( SMESH::GROUP ) );
  myFilter = new SMESH_LogicalFilter( filters, SMESH_LogicalFilter::LO_OR );

  myPreview = new SMESH::MeshPreviewStruct();

  myPreview->nodesXYZ.length(1);
  myPreview->nodesXYZ[0].x = myDlg->myX->GetValue();
  myPreview->nodesXYZ[0].y = myDlg->myY->GetValue();
  myPreview->nodesXYZ[0].z = myDlg->myZ->GetValue();

  myPreview->elementTypes.length(1);
  myPreview->elementTypes[0].SMDS_ElementType = SMESH::NODE;
  myPreview->elementTypes[0].isPoly = false;
  myPreview->elementTypes[0].nbNodesInElement = 1;

  myPreview->elementConnectivities.length(1);
  myPreview->elementConnectivities[0] = 0;

  // connect signals and slots
  connect(myDlg->myFindBtn,      SIGNAL(clicked()),               this, SLOT(onFind()));
  connect(myDlg->myX,            SIGNAL(valueChanged(double)),    this, SLOT(redisplayPreview()));
  connect(myDlg->myY,            SIGNAL(valueChanged(double)),    this, SLOT(redisplayPreview()));
  connect(myDlg->myZ,            SIGNAL(valueChanged(double)),    this, SLOT(redisplayPreview()));
  connect(myDlg->myFoundList,    SIGNAL(itemSelectionChanged()),  this, SLOT(onElemSelected()));
  connect(myDlg->myElemTypeCombo,SIGNAL(currentIndexChanged(int)),this, SLOT(onElemTypeChange(int)));
  connect(myDlg,                 SIGNAL(rejectedDlg()),           this, SLOT(onRejectedDlg()));
}

//=======================================================================
// function : startOperation()
// purpose  : Init dialog fields, connect signals and slots, show dialog
//=======================================================================
void SMESHGUI_FindElemByPointOp::startOperation()
{
  // init simulation with a current View
  if ( mySimulation ) delete mySimulation;
  mySMESHGUI = getSMESHGUI();
  mySimulation = new SMESHGUI_MeshEditPreview(SMESH::GetViewWindow( mySMESHGUI ) );
  connect(mySMESHGUI, SIGNAL (SignalActivatedViewManager()), this, SLOT(onOpenView()));
  connect(mySMESHGUI, SIGNAL (SignalCloseView()), this, SLOT(onCloseView()));
  vtkProperty* aProp = vtkProperty::New();
  aProp->SetRepresentationToWireframe();
  aProp->SetColor(250, 0, 250);
  aProp->SetPointSize(5);
  aProp->SetLineWidth( SMESH::GetFloat("SMESH:element_width",1) + 1);
  mySimulation->GetActor()->SetProperty(aProp);
  aProp->Delete();

  SMESHGUI_SelectionOp::startOperation();
  myDlg->show();
  redisplayPreview();

  onSelectionDone(); // init myMeshOrPart
}

//================================================================================
/*!
 * \brief Stops operation
 */
//================================================================================

void SMESHGUI_FindElemByPointOp::stopOperation()
{
  if ( mySimulation )
  {
    mySimulation->SetVisibility(false);
    delete mySimulation;
    mySimulation = 0;
  }
  disconnect(mySMESHGUI, SIGNAL (SignalActivatedViewManager()), this, SLOT(onOpenView()));
  disconnect(mySMESHGUI, SIGNAL (SignalCloseView()), this, SLOT(onCloseView()));
  selectionMgr()->removeFilter( myFilter );
  SMESHGUI_SelectionOp::stopOperation();
}

//=================================================================================
/*!
 * \brief SLOT called when the viewer opened
 */
//=================================================================================
void SMESHGUI_FindElemByPointOp::onOpenView()
{
  if ( mySimulation ) {
    mySimulation->SetVisibility(false);
  }
  else {
    mySimulation = new SMESHGUI_MeshEditPreview(SMESH::GetViewWindow( mySMESHGUI ));
  }
}

//=================================================================================
/*!
 * \brief SLOT called when the viewer closed
 */
//=================================================================================
void SMESHGUI_FindElemByPointOp::onCloseView()
{
  delete mySimulation;
  mySimulation = 0;
}
//================================================================================
/*!
 * \brief hilight found selected elements
 */
//================================================================================

void SMESHGUI_FindElemByPointOp::onElemSelected()
{
  if ( !myMeshIO.IsNull() )
  {
    Selection_Mode selMode =
      myDlg->myElemTypeCombo->currentId() == int(SMESH::NODE) ? NodeSelection : CellSelection;
    if ( selectionMode() != selMode )
      setSelectionMode( selMode );

    QList<QListWidgetItem *> ids = myDlg->myFoundList->selectedItems();
    QList<QListWidgetItem*>::iterator id = ids.begin();
    TColStd_MapOfInteger idMap;
    for ( ; id != ids.end(); ++id )
      idMap.Add( (*id)->text().toInt() );

    addOrRemoveIndex( myMeshIO, idMap, false );

    SALOME_ListIO aList;
    aList.Append(myMeshIO);
    selectionMgr()->setSelectedObjects(aList,false);
  }
}

//================================================================================
/*!
 * \brief Set selection mode according to element type
 */
//================================================================================

void SMESHGUI_FindElemByPointOp::onElemTypeChange(int index)
{
  Selection_Mode newMode = (index == 1) ? NodeSelection : CellSelection;
  if ( selectionMode() != newMode )
  {
    selectionMgr()->clearFilters();
    setSelectionMode( newMode );
  }
  myDlg->myFoundList->clear();
}

//================================================================================
/*!
 * \brief Method needed for internal cuisine
 */
//================================================================================

void SMESHGUI_FindElemByPointDlg::reject()
{
  emit rejectedDlg();
  QtxDialog::reject();
}

//================================================================================
/*!
 * \brief Method needed for internal cuisine
 */
//================================================================================

void SMESHGUI_FindElemByPointOp::onRejectedDlg()
{
  myMeshIO.Nullify(); 
}

//================================================================================
/*!
 * \brief perform it's intention action: find elements
 */
//================================================================================

void SMESHGUI_FindElemByPointOp::onFind()
{
  if ( myMeshOrPart->_is_nil() )
    return;

  try {
    SUIT_OverrideCursor wc;

    SMESH::SMESH_Mesh_var aMesh = myMeshOrPart->GetMesh();
    if ( aMesh->_is_nil() )
      return;
    SMESH::SMESH_MeshEditor_var aMeshEditor = aMesh->GetMeshEditor();
    if (aMeshEditor->_is_nil())
      return;

    SMESH::long_array_var foundIds;
    if ( aMesh->_is_equivalent( myMeshOrPart ) )
      foundIds =
        aMeshEditor->FindElementsByPoint( myDlg->myX->GetValue(),
                                          myDlg->myY->GetValue(),
                                          myDlg->myZ->GetValue(),
                                          SMESH::ElementType( myDlg->myElemTypeCombo->currentId()));
    else
      foundIds =
        aMeshEditor->FindAmongElementsByPoint( myMeshOrPart,
                                               myDlg->myX->GetValue(),
                                               myDlg->myY->GetValue(),
                                               myDlg->myZ->GetValue(),
                                               SMESH::ElementType( myDlg->myElemTypeCombo->currentId()));
    myDlg->myFoundList->clear();
    for ( int i = 0; i < (int) foundIds->length(); ++i )
      myDlg->myFoundList->addItem( QString::number( foundIds[i] ));

    if ( foundIds->length() > 0 )
      myDlg->myFoundList->setCurrentRow(0);
  }
  catch (const SALOME::SALOME_Exception& S_ex) {
    SalomeApp_Tools::QtCatchCorbaException(S_ex);
  }
  catch (...) {
  }
}

//================================================================================
/*!
 * \brief Method needed for internal cuisine
 */
//================================================================================

bool SMESHGUI_FindElemByPointOp::onApply()
{
  onRejectedDlg();
  return true;
}

//================================================================================
/*!
 * \brief SLOT called when selection changed
 */
//================================================================================

void SMESHGUI_FindElemByPointOp::onSelectionDone()
{
  if ( !myDlg->isVisible() || !myDlg->isEnabled() )
    return;

  QString oldMeshEntry, newMeshEntry;
  if ( !myMeshIO.IsNull() && myMeshIO->hasEntry() )
    oldMeshEntry = myMeshIO->getEntry();

  try {
    SALOME_ListIO aList;
    selectionMgr()->selectedObjects(aList, SVTK_Viewer::Type());
    if (aList.Extent() == 1 && aList.First()->hasEntry())
    {
      Handle(SALOME_InteractiveObject) anIO = aList.First();
      _PTR(SObject) pObj = studyDS()->FindObjectID(anIO->getEntry());
      CORBA::Object_var anObj = SMESH::IObjectToObject( anIO );
      newMeshEntry = anIO->getEntry();
      SMESH::SMESH_IDSource_var aMeshOrPart = SMESH::SMESH_IDSource::_narrow(anObj);
      if ( pObj && !aMeshOrPart->_is_nil() && oldMeshEntry != newMeshEntry )
      {
        myMeshOrPart = aMeshOrPart;
        myMeshIO.Nullify();
        myMeshIO = anIO;
        std::string name = pObj->GetName();
        myDlg->myMeshName->setText("");
        myDlg->myMeshName->setText( QString( name.c_str() ).trimmed() );
        SMESH::array_of_ElementType_var  types = myMeshOrPart->GetTypes();
        myDlg->setTypes( types );
      }
    }
  }
  catch (...) {
  }

  if ( oldMeshEntry != newMeshEntry || newMeshEntry.isEmpty() )
    myDlg->myFoundList->clear();

  myDlg->myFindBtn->setEnabled( !myMeshIO.IsNull() );
}

//================================================================================
/*!
 * \brief show point by coordinates
 */
//================================================================================

void SMESHGUI_FindElemByPointOp::redisplayPreview()
{
  myDlg->myFoundList->clear();

  myPreview->nodesXYZ[0].x = myDlg->myX->GetValue();
  myPreview->nodesXYZ[0].y = myDlg->myY->GetValue();
  myPreview->nodesXYZ[0].z = myDlg->myZ->GetValue();
  if (!mySimulation)
    mySimulation = new SMESHGUI_MeshEditPreview(SMESH::GetViewWindow( mySMESHGUI ));
  mySimulation->SetData(&myPreview.in());
}

//================================================================================
/*!
 * \brief install filter on meshes
 */
//================================================================================

void SMESHGUI_FindElemByPointOp::activateSelection()
{
  selectionMgr()->clearFilters();
  selectionMgr()->installFilter( myFilter );
}

//================================================================================
/*!
 * \brief Destructor
*/
//================================================================================

SMESHGUI_FindElemByPointOp::~SMESHGUI_FindElemByPointOp()
{
  if ( myDlg )         { delete myDlg;        myDlg = 0; }
  if ( mySimulation )  { delete mySimulation; mySimulation = 0; }
  if ( myFilter )
  {
    QList<SUIT_SelectionFilter*> sub =  ((SMESH_LogicalFilter*)myFilter)->getFilters();
    delete sub.front();
    delete sub.back();
    delete myFilter;
    myFilter = 0;
  }
}

//================================================================================
/*!
 * \brief Gets dialog of this operation
 * \retval LightApp_Dialog* - pointer to dialog of this operation
 */
//================================================================================

LightApp_Dialog* SMESHGUI_FindElemByPointOp::dlg() const
{
  return myDlg;
}