--- /dev/null
+// 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_AddSubMeshOp.h"
+#include <SMESHGUI_AddSubMeshDlg.h>
+#include <SMESHGUI_GEOMGenUtils.h>
+#include <SMESHGUI_HypothesesUtils.h>
+#include <SMESHGUI_Utils.h>
+#include <SMESHGUI.h>
+
+#include <SMESH_NumberFilter.hxx>
+#include <SMESH_TypeFilter.hxx>
+
+#include <SUIT_SelectionFilter.h>
+#include <SUIT_OverrideCursor.h>
+#include <SUIT_MessageBox.h>
+
+#include <SalomeApp_SelectionMgr.h>
+#include <SalomeApp_UpdateFlags.h>
+#include <SalomeApp_Tools.h>
+
+#include <TColStd_MapOfInteger.hxx>
+
+#include <SALOMEDS_SObject.hxx>
+
+#include <utilities.h>
+
+//=================================================================================
+// function : Constructor
+// purpose :
+//=================================================================================
+SMESHGUI_AddSubMeshOp::SMESHGUI_AddSubMeshOp()
+: SMESHGUI_Operation(),
+ myDlg( 0 ),
+ myMeshFilter( 0 ),
+ myGeomFilter( 0 ),
+ myHypothesisFilter( 0 ),
+ myAlgorithmFilter( 0 )
+{
+}
+
+//=================================================================================
+// function : Destructor
+// purpose :
+//=================================================================================
+SMESHGUI_AddSubMeshOp::~SMESHGUI_AddSubMeshOp()
+{
+ if( myDlg )
+ delete myDlg;
+
+ if( myMeshFilter )
+ delete myMeshFilter;
+
+ if( myGeomFilter )
+ delete myGeomFilter;
+
+ if( myAlgorithmFilter )
+ delete myAlgorithmFilter;
+
+ if( myHypothesisFilter )
+ delete myHypothesisFilter;
+}
+
+//=================================================================================
+// function : startOperation
+// purpose :
+//=================================================================================
+void SMESHGUI_AddSubMeshOp::startOperation()
+{
+ if( !myDlg )
+ {
+ myDlg = new SMESHGUI_AddSubMeshDlg( getSMESHGUI() );
+ connect( myDlg, SIGNAL( objectActivated( int ) ), this, SLOT( onActivateObject( int ) ) );
+ connect( myDlg, SIGNAL( selectionChanged( int ) ), this, SLOT( onSelectionChanged( int ) ) );
+ }
+
+ SMESHGUI_Operation::startOperation();
+
+ if( !myGeomFilter )
+ {
+ TColStd_MapOfInteger allTypesMap;
+ for (int i = 0; i < 10; i++)
+ allTypesMap.Add(i);
+ myGeomFilter = new SMESH_NumberFilter ("GEOM", TopAbs_SHAPE, 0, allTypesMap);
+ }
+
+ if( !myMeshFilter )
+ myMeshFilter = new SMESH_TypeFilter (MESH);
+
+ if( !myAlgorithmFilter )
+ myAlgorithmFilter = new SMESH_TypeFilter (ALGORITHM);
+
+ if( !myHypothesisFilter )
+ myHypothesisFilter = new SMESH_TypeFilter (HYPOTHESIS);
+
+ init();
+ myDlg->show();
+}
+
+//=================================================================================
+// function : dlg
+// purpose :
+//=================================================================================
+SalomeApp_Dialog* SMESHGUI_AddSubMeshOp::dlg() const
+{
+ return myDlg;
+}
+
+//=================================================================================
+// function : selectionDone
+// purpose :
+//=================================================================================
+void SMESHGUI_AddSubMeshOp::selectionDone()
+{
+ QStringList names, ids;
+ SMESHGUI_Dialog::TypesList types;
+ selected( names, types, ids );
+ if( myDlg )
+ {
+ myDlg->selectObject( names, types, ids );
+ myDlg->updateControlState( isValid() );
+ }
+}
+
+//=================================================================================
+// function : onActivateObject
+// purpose :
+//=================================================================================
+void SMESHGUI_AddSubMeshOp::onActivateObject( int obj )
+{
+ SalomeApp_SelectionMgr* mgr = selectionMgr();
+
+ if( !mgr )
+ return;
+
+ mgr->clearFilters();
+
+ if( obj==SMESHGUI_AddSubMeshDlg::MeshObj )
+ mgr->installFilter( myMeshFilter );
+
+ else if( obj==SMESHGUI_AddSubMeshDlg::GeomObj )
+ mgr->installFilter( myGeomFilter );
+
+ else if( obj==SMESHGUI_AddSubMeshDlg::Hypo )
+ mgr->installFilter( myHypothesisFilter );
+
+ else if( obj==SMESHGUI_AddSubMeshDlg::Algo )
+ mgr->installFilter( myAlgorithmFilter );
+}
+
+//=================================================================================
+// function : commitOperation
+// purpose :
+//=================================================================================
+bool SMESHGUI_AddSubMeshOp::onApply()
+{
+ if( getSMESHGUI()->isActiveStudyLocked() )
+ return false;
+
+ QString myNameSubMesh = myDlg->subMeshName();
+ if (myNameSubMesh.isEmpty())
+ {
+ SUIT_MessageBox::warn1( dlg(), tr("SMESH_WRN_WARNING"),
+ tr("SMESH_WRN_EMPTY_NAME"), tr("SMESH_BUT_OK"));
+ return false;
+ }
+
+ QStringList selMesh, selGeom, selHypo, selAlgo;
+ myDlg->selectedObject( SMESHGUI_AddSubMeshDlg::MeshObj, selMesh );
+ myDlg->selectedObject( SMESHGUI_AddSubMeshDlg::GeomObj, selGeom );
+ myDlg->selectedObject( SMESHGUI_AddSubMeshDlg::Hypo, selHypo );
+ myDlg->selectedObject( SMESHGUI_AddSubMeshDlg::Algo, selAlgo );
+
+ _PTR(Study) study = studyDS();
+ _PTR(SObject) aMeshSO = study->FindObjectID( selMesh.first() );
+ SMESH::SMESH_Mesh_var myMesh = SMESH::SMESH_Mesh::_narrow( _CAST( SObject, aMeshSO )->GetObject() );
+ GEOM::GEOM_Object_var myMainShape = SMESH::GetShapeOnMeshOrSubMesh(aMeshSO);
+ if (myMainShape->_is_nil())
+ return false;
+
+ SUIT_OverrideCursor wc;
+
+ _PTR(SObject) aGeomSO = study->FindObjectID( selGeom.first() );
+ GEOM::GEOM_Object_var myGeomShape = GEOM::GEOM_Object::_narrow( _CAST( SObject, aGeomSO )->GetObject() );
+
+ // create submesh
+ SMESH::SMESH_subMesh_var aSubMesh = addSubMesh(myMesh, myGeomShape, myNameSubMesh);
+ int nbSuccess = 0;
+
+ if (!aSubMesh->_is_nil())
+ {
+ // assign hypotheses
+ int nbAlgo = selAlgo.count();
+ int nbHyps = selHypo.count() + nbAlgo;
+ for (int i = 0; i < nbHyps; i++)
+ {
+ _PTR(SObject) aHypSOClient =
+ study->FindObjectID
+ (i < nbAlgo ? selAlgo[i].latin1() : selHypo[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());
+ }
+
+ update( UF_Model | UF_ObjBrowser );
+ init();
+
+ return (nbSuccess > 0);
+}
+
+//=================================================================================
+// function : init()
+// purpose :
+//=================================================================================
+void SMESHGUI_AddSubMeshOp::init()
+{
+ if( myDlg )
+ {
+ myDlg->init();
+ myDlg->updateControlState( false );
+ }
+}
+
+//=================================================================================
+// function : isValid()
+// purpose :
+//=================================================================================
+bool SMESHGUI_AddSubMeshOp::isValid() const
+{
+ bool isEnabled = !myDlg->subMeshName().isEmpty() &&
+ myDlg->hasSelection( SMESHGUI_AddSubMeshDlg::MeshObj ) &&
+ myDlg->hasSelection( SMESHGUI_AddSubMeshDlg::GeomObj ) &&
+ myDlg->hasSelection( SMESHGUI_AddSubMeshDlg::Hypo ) &&
+ myDlg->hasSelection( SMESHGUI_AddSubMeshDlg::Algo );
+
+ bool isImportedMesh = false;
+
+ QStringList selMesh;
+ myDlg->selectedObject( SMESHGUI_AddSubMeshDlg::MeshObj, selMesh );
+ _PTR(SObject) SO = studyDS()->FindObjectID( selMesh.first() );
+ GEOM::GEOM_Object_var myGeomShape = SMESH::GetShapeOnMeshOrSubMesh(SO);
+ isImportedMesh = myGeomShape->_is_nil();
+
+ return isEnabled && !isImportedMesh;
+}
+
+//=================================================================================
+// function : IsFatherOf()
+// purpose :
+//=================================================================================
+static bool IsFatherOf (_PTR(SObject) SO, _PTR(SObject) fatherSO)
+{
+ if( SO && fatherSO )
+ {
+ _PTR(SObject) aSO = SO->GetFather();
+ while (aSO->GetID().length() >= fatherSO->GetID().length())
+ {
+ if (aSO->GetID() == fatherSO->GetID())
+ return true;
+ aSO = aSO->GetFather();
+ }
+ }
+ return false;
+}
+
+//=================================================================================
+// function : onSelectionChanged()
+// purpose :
+//=================================================================================
+void SMESHGUI_AddSubMeshOp::onSelectionChanged( int id )
+{
+ if( !myDlg->hasSelection( id ) )
+ return;
+
+ if( id==SMESHGUI_AddSubMeshDlg::MeshObj )
+ myDlg->clearSelection( SMESHGUI_AddSubMeshDlg::GeomObj );
+
+ else if( id==SMESHGUI_AddSubMeshDlg::GeomObj )
+ {
+ QStringList selMesh, selGeom;
+ myDlg->selectedObject( SMESHGUI_AddSubMeshDlg::MeshObj, selMesh );
+ myDlg->selectedObject( SMESHGUI_AddSubMeshDlg::GeomObj, selGeom );
+
+ _PTR(SObject) aMeshSO = studyDS()->FindObjectID( selMesh.first() );
+
+ GEOM::GEOM_Object_var aMainGeomShape = SMESH::GetShapeOnMeshOrSubMesh(aMeshSO);
+ _PTR(SObject) aMainGeomShapeSO = SMESH::FindSObject(aMainGeomShape);
+ if( !aMainGeomShapeSO || !IsFatherOf( studyDS()->FindObjectID( selGeom.first() ), aMainGeomShapeSO ) )
+ myDlg->clearSelection( id );
+ }
+}
+
+//=================================================================================
+// function : onSelectionChanged()
+// purpose :
+//=================================================================================
+SMESH::SMESH_subMesh_var SMESHGUI_AddSubMeshOp::addSubMesh( SMESH::SMESH_Mesh_ptr theMesh,
+ GEOM::GEOM_Object_ptr theShapeObject,
+ const QString& 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();
+}
