// Copyright (C) 2005 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.salome-platform.org/ or email : webmaster.salome@opencascade.com // /** * SMESH SMESHGUI * * Copyright (C) 2005 CEA/DEN, EDF R&D * * * * File : SMESHGUI_MeshOp.h * Author : Sergey LITONIN * Module : SMESHGUI */ #include "SMESHGUI_MeshOp.h" #include "SMESHGUI_MeshDlg.h" #include "SMESHGUI_ShapeByMeshDlg.h" #include "SMESH_TypeFilter.hxx" #include "SMESHGUI.h" #include "SMESHGUI_HypothesesUtils.h" #include "SMESHGUI_Hypotheses.h" #include "SMESHGUI_Utils.h" #include "SMESHGUI_GEOMGenUtils.h" #include "SMESHGUI_VTKUtils.h" #include "SMESH_TypeFilter.hxx" #include "SMESH_NumberFilter.hxx" #include CORBA_CLIENT_HEADER(SMESH_Gen) #include "GEOM_SelectionFilter.h" #include "GEOMBase.h" #include "GeometryGUI.h" #include "SalomeApp_Tools.h" #include "SalomeApp_Application.h" #include "SALOMEDSClient_Study.hxx" #include "SALOMEDSClient_AttributeIOR.hxx" #include "SALOMEDSClient_AttributeName.hxx" #include "SALOMEDS_SComponent.hxx" #include "SALOMEDS_SObject.hxx" #include "LightApp_SelectionMgr.h" #include "LightApp_UpdateFlags.h" #include "SUIT_MessageBox.h" #include "SUIT_Desktop.h" #include "SUIT_OverrideCursor.h" #include "SALOME_InteractiveObject.hxx" #include "SALOME_ListIO.hxx" #include "utilities.h" #include #include #include #include //================================================================================ /*! * \brief Constructor * \param theToCreate - if this parameter is true then operation is used for creation, * for editing otherwise * * Initialize operation */ //================================================================================ SMESHGUI_MeshOp::SMESHGUI_MeshOp( const bool theToCreate, const bool theIsMesh ) : SMESHGUI_SelectionOp(), myToCreate( theToCreate ), myIsMesh( theIsMesh ), myDlg( 0 ), myShapeByMeshOp( 0 ) { if ( GeometryGUI::GetGeomGen()->_is_nil() )// check that GEOM_Gen exists GeometryGUI::InitGeomGen(); myIsOnGeometry = true; } //================================================================================ /*! * \brief Destructor */ //================================================================================ SMESHGUI_MeshOp::~SMESHGUI_MeshOp() { if ( myDlg ) delete myDlg; } //================================================================================ /*! * \brief Gets dialog of this operation * \retval LightApp_Dialog* - pointer to dialog of this operation */ //================================================================================ LightApp_Dialog* SMESHGUI_MeshOp::dlg() const { return myDlg; } //================================================================================ /*! * \brief Creates or edits mesh * \retval bool - TRUE if operation is performed successfully, FALSE otherwise * * Virtual slot redefined from the base class called when "Apply" button is clicked * creates or edits mesh */ //================================================================================ bool SMESHGUI_MeshOp::onApply() { if (isStudyLocked()) return false; QString aMess; if ( !isValid( aMess ) ) { dlg()->show(); if ( aMess != "" ) SUIT_MessageBox::warn1( myDlg, tr( "SMESH_WRN_WARNING" ), aMess, tr( "SMESH_BUT_OK" ) ); return false; } bool aResult = false; aMess = ""; try { if ( myToCreate && myIsMesh ) aResult = createMesh( aMess ); if ( myToCreate && !myIsMesh ) aResult = createSubMesh( aMess ); else if ( !myToCreate ) aResult = editMeshOrSubMesh( aMess ); if ( aResult ) update( UF_ObjBrowser | UF_Model ); } catch ( const SALOME::SALOME_Exception& S_ex ) { SalomeApp_Tools::QtCatchCorbaException( S_ex ); aResult = false; } catch ( ... ) { aResult = false; } if ( aResult ) { if ( myToCreate ) setDefaultName(); } else { if ( aMess == "" ) aMess = tr( "SMESH_OPERATION_FAILED" ); SUIT_MessageBox::warn1( myDlg, tr( "SMESH_ERROR" ), aMess, tr( "SMESH_BUT_OK" ) ); } return aResult; } //================================================================================ /*! * \brief Creates dialog if necessary and shows it * * Virtual method redefined from base class called when operation is started creates * dialog if necessary and shows it, activates selection */ //================================================================================ void SMESHGUI_MeshOp::startOperation() { if (!myDlg) { myDlg = new SMESHGUI_MeshDlg( myToCreate, myIsMesh ); for ( int i = SMESH::DIM_0D; i <= SMESH::DIM_3D; i++ ) { connect( myDlg->tab( i ), SIGNAL( createHyp( const int, const int ) ), this, SLOT( onCreateHyp( const int, const int ) ) ); connect( myDlg->tab( i ), SIGNAL( editHyp( const int, const int ) ), this, SLOT( onEditHyp( const int, const int ) ) ); connect( myDlg->tab( i ), SIGNAL( selectAlgo( const int ) ), this, SLOT( onAlgoSelected( const int ) ) ); } connect( myDlg, SIGNAL( hypoSet( const QString& )), SLOT( onHypoSet( const QString& ))); connect( myDlg, SIGNAL( geomSelectionByMesh( bool )), SLOT( onGeomSelectionByMesh( bool ))); if ( myToCreate ) if ( myIsMesh ) myHelpFileName = "constructing_meshes_page.html"; else myHelpFileName = "constructing_submeshes_page.html"; else myHelpFileName = "editing_meshes_page.html"; } SMESHGUI_SelectionOp::startOperation(); // iterate through dimensions and get available algoritms, set them to the dialog _PTR(SComponent) aFather = SMESH::GetActiveStudyDocument()->FindComponent( "SMESH" ); for ( int i = SMESH::DIM_0D; i <= SMESH::DIM_3D; i++ ) { SMESHGUI_MeshTab* aTab = myDlg->tab( i ); QStringList hypList; // clear available hypotheses aTab->setAvailableHyps( MainHyp, hypList ); aTab->setAvailableHyps( AddHyp, hypList ); aTab->setExistingHyps( MainHyp, hypList ); aTab->setExistingHyps( AddHyp, hypList ); myExistingHyps[ i ][ MainHyp ].clear(); myExistingHyps[ i ][ AddHyp ].clear(); // set algos availableHyps( i, Algo, hypList, myAvailableHypData[i][Algo] ); aTab->setAvailableHyps( Algo, hypList ); } if ( myToCreate ) { setDefaultName(); myDlg->activateObject( myIsMesh ? SMESHGUI_MeshDlg::Geom : SMESHGUI_MeshDlg::Mesh ); } else myDlg->activateObject( SMESHGUI_MeshDlg::Obj ); myDlg->setHypoSets( SMESH::GetHypothesesSets() ); myDlg->setCurrentTab( SMESH::DIM_3D ); myDlg->show(); selectionDone(); myIgnoreAlgoSelection = false; } //================================================================================ /*! * \brief Creates selection filter * \param theId - identifier of current selection widget * \retval SUIT_SelectionFilter* - pointer to the created filter or null * * Creates selection filter in accordance with identifier of current selection widget */ //================================================================================ SUIT_SelectionFilter* SMESHGUI_MeshOp::createFilter( const int theId ) const { if ( theId == SMESHGUI_MeshDlg::Geom ) { // TColStd_MapOfInteger allTypesMap; // for ( int i = 0; i < 10; i++ ) // allTypesMap.Add( i ); // return new SMESH_NumberFilter( "GEOM", TopAbs_SHAPE, 0, allTypesMap ); return new GEOM_SelectionFilter( (SalomeApp_Study*)study(), true ); } else if ( theId == SMESHGUI_MeshDlg::Obj && !myToCreate ) return new SMESH_TypeFilter( MESHorSUBMESH ); else if ( theId == SMESHGUI_MeshDlg::Mesh ) return new SMESH_TypeFilter( MESH ); else return 0; } //================================================================================ /*! * \brief check if selected shape is a subshape of the shape to mesh * \retval bool - check result */ //================================================================================ bool SMESHGUI_MeshOp::isSubshapeOk() const { if ( !myToCreate || myIsMesh ) // not submesh creation return false; // mesh QString aMeshEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Mesh ); _PTR(SObject) pMesh = studyDS()->FindObjectID( aMeshEntry.latin1() ); if (!pMesh) return false; SMESH::SMESH_Mesh_var mesh = SMESH::SObjectToInterface( pMesh ); if (mesh->_is_nil()) return false; // main shape of the mesh GEOM::GEOM_Object_var mainGeom = mesh->GetShapeToMesh(); if (mainGeom->_is_nil()) return false; // geometry QStringList aGEOMs; myDlg->selectedObject(SMESHGUI_MeshDlg::Geom, aGEOMs); if (aGEOMs.count() > 0) { GEOM::GEOM_Gen_var geomGen = SMESH::GetGEOMGen(); _PTR(Study) aStudy = SMESH::GetActiveStudyDocument(); if (geomGen->_is_nil() || !aStudy) return false; GEOM::GEOM_IGroupOperations_var op = geomGen->GetIGroupOperations(aStudy->StudyId()); if (op->_is_nil()) return false; // check all selected shapes QStringList::const_iterator aSubShapesIter = aGEOMs.begin(); for (; aSubShapesIter != aGEOMs.end(); aSubShapesIter++) { QString aSubGeomEntry = (*aSubShapesIter); _PTR(SObject) pSubGeom = studyDS()->FindObjectID(aSubGeomEntry.latin1()); if (!pSubGeom) return false; GEOM::GEOM_Object_var aSubGeomVar = GEOM::GEOM_Object::_narrow(_CAST(SObject,pSubGeom)->GetObject()); if (aSubGeomVar->_is_nil()) return false; // skl for NPAL14695 - implementation of searching of mainObj GEOM::GEOM_Object_var mainObj = op->GetMainShape(aSubGeomVar); //if (mainObj->_is_nil() || // string(mainObj->GetEntry()) != string(mainGeom->GetEntry())) return false; while(1) { if (mainObj->_is_nil()) return false; if (string(mainObj->GetEntry()) == string(mainGeom->GetEntry())) return true; mainObj = op->GetMainShape(mainObj); } } //return true; } return false; } //================================================================================ /*! * \brief find an existing submesh by the selected shape * \retval _PTR(SObject) - the found submesh SObject */ //================================================================================ _PTR(SObject) SMESHGUI_MeshOp::getSubmeshByGeom() const { QString aMeshEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Mesh ); QString aGeomEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Geom ); _PTR(SObject) pMesh = studyDS()->FindObjectID( aMeshEntry.latin1() ); _PTR(SObject) pGeom = studyDS()->FindObjectID( aGeomEntry.latin1() ); if ( pMesh && pGeom ) { GEOM::GEOM_Object_var geom = SMESH::SObjectToInterface( pGeom ); if ( !geom->_is_nil() ) { int tag = -1; switch ( geom->GetShapeType() ) { case GEOM::VERTEX: tag = SMESH::Tag_SubMeshOnVertex ; break; case GEOM::EDGE: tag = SMESH::Tag_SubMeshOnEdge ; break; case GEOM::WIRE: tag = SMESH::Tag_SubMeshOnWire ; break; case GEOM::FACE: tag = SMESH::Tag_SubMeshOnFace ; break; case GEOM::SHELL: tag = SMESH::Tag_SubMeshOnShell ; break; case GEOM::SOLID: tag = SMESH::Tag_SubMeshOnSolid ; break; case GEOM::COMPOUND: tag = SMESH::Tag_SubMeshOnCompound; break; default:; } _PTR(GenericAttribute) anAttr; _PTR(SObject) aSubmeshRoot; _PTR(Study) aStudy = SMESH::GetActiveStudyDocument(); if ( pMesh->FindSubObject( tag, aSubmeshRoot ) ) { _PTR(ChildIterator) smIter = aStudy->NewChildIterator( aSubmeshRoot ); for (; smIter->More(); smIter->Next() ) { _PTR(SObject) aSmObj = smIter->Value(); if ( ! aSmObj->FindAttribute( anAttr, "AttributeIOR" )) continue; _PTR(ChildIterator) anIter1 = aStudy->NewChildIterator(aSmObj); for (; anIter1->More(); anIter1->Next()) { _PTR(SObject) pGeom2 = anIter1->Value(); if ( pGeom2->ReferencedObject( pGeom2 ) && pGeom2->GetID() == pGeom->GetID() ) return aSmObj; } } } } } return _PTR(SObject)(); } //================================================================================ /*! * \brief Updates dialog's look and feel * * Virtual method redefined from the base class updates dialog's look and feel */ //================================================================================ void SMESHGUI_MeshOp::selectionDone() { if (!dlg()->isShown() || !myDlg->isEnabled()) return; SMESHGUI_SelectionOp::selectionDone(); try { myIsOnGeometry = true; //Check geometry for mesh QString anObjEntry = myDlg->selectedObject(SMESHGUI_MeshDlg::Obj); _PTR(SObject) pObj = studyDS()->FindObjectID(anObjEntry.latin1()); if (pObj) { SMESH::SMESH_Mesh_var aMeshVar = SMESH::SMESH_Mesh::_narrow(_CAST(SObject,pObj)->GetObject()); if (!aMeshVar->_is_nil()) { if (!myToCreate && !aMeshVar->HasShapeToMesh()) myIsOnGeometry = false; } } if (myIsOnGeometry) { // Enable tabs according to shape dimension int shapeDim = 3; QStringList aGEOMs; myDlg->selectedObject(SMESHGUI_MeshDlg::Geom, aGEOMs); GEOM::ListOfGO_var aSeq = new GEOM::ListOfGO; if (aGEOMs.count() > 0) { // one or more GEOM shape selected aSeq->length(aGEOMs.count()); QStringList::const_iterator aSubShapesIter = aGEOMs.begin(); int iSubSh = 0; for (; aSubShapesIter != aGEOMs.end(); aSubShapesIter++, iSubSh++) { QString aSubGeomEntry = (*aSubShapesIter); _PTR(SObject) pSubGeom = studyDS()->FindObjectID(aSubGeomEntry.latin1()); GEOM::GEOM_Object_var aSubGeomVar = GEOM::GEOM_Object::_narrow(_CAST(SObject,pSubGeom)->GetObject()); aSeq[iSubSh] = aSubGeomVar; } } else { // get geometry by selected sub-mesh QString anObjEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Obj ); _PTR(SObject) pObj = studyDS()->FindObjectID( anObjEntry.latin1() ); GEOM::GEOM_Object_var aGeomVar = SMESH::GetShapeOnMeshOrSubMesh( pObj ); if (!aGeomVar->_is_nil()) { aSeq->length(1); aSeq[0] = aGeomVar; } } if (aSeq->length() > 0) { shapeDim = 0; for (int iss = 0; iss < aSeq->length() && shapeDim < 3; iss++) { GEOM::GEOM_Object_var aGeomVar = aSeq[iss]; switch ( aGeomVar->GetShapeType() ) { case GEOM::SOLID: case GEOM::SHELL: shapeDim = 3; break; case GEOM::FACE: shapeDim = (shapeDim < 2) ? 2 : shapeDim; break; case GEOM::WIRE: case GEOM::EDGE: shapeDim = (shapeDim < 1) ? 1 : shapeDim; break; case GEOM::VERTEX: break; default: TopoDS_Shape aShape; if ( GEOMBase::GetShape(aGeomVar, aShape)) { TopExp_Explorer exp( aShape, TopAbs_SHELL ); if ( exp.More() ) shapeDim = 3; else if ( exp.Init( aShape, TopAbs_FACE ), exp.More() ) shapeDim = (shapeDim < 2) ? 2 : shapeDim; else if ( exp.Init( aShape, TopAbs_EDGE ), exp.More() ) shapeDim = (shapeDim < 1) ? 1 : shapeDim; else ;//shapeDim = 0; } } } } myDlg->setMaxHypoDim( shapeDim ); if (!myToCreate) // edition: read hypotheses { if (pObj != 0) { SMESH::SMESH_subMesh_var aVar = SMESH::SMESH_subMesh::_narrow( _CAST( SObject,pObj )->GetObject() ); myDlg->setObjectShown( SMESHGUI_MeshDlg::Mesh, !aVar->_is_nil() ); myDlg->setObjectShown( SMESHGUI_MeshDlg::Geom, true ); myDlg->objectWg( SMESHGUI_MeshDlg::Mesh, SMESHGUI_MeshDlg::Btn )->hide(); myDlg->objectWg( SMESHGUI_MeshDlg::Geom, SMESHGUI_MeshDlg::Btn )->hide(); myDlg->updateGeometry(); myDlg->adjustSize(); readMesh(); } else myDlg->reset(); } else if ( !myIsMesh ) // submesh creation { // if a submesh on the selected shape already exist, pass to submesh edition mode if ( _PTR(SObject) pSubmesh = getSubmeshByGeom() ) { SMESH::SMESH_subMesh_var sm = SMESH::SObjectToInterface( pSubmesh ); bool editSubmesh = ( !sm->_is_nil() && SUIT_MessageBox::question2( myDlg, tr( "SMESH_WARNING" ), tr( "EDIT_SUBMESH_QUESTION"), tr( "SMESH_BUT_YES" ), tr( "SMESH_BUT_NO" ), 1, 0, 0 )); if ( editSubmesh ) { selectionMgr()->clearFilters(); selectObject( pSubmesh ); SMESHGUI::GetSMESHGUI()->switchToOperation(704); return; } else { myDlg->selectObject( "", SMESHGUI_MeshDlg::Geom, "" ); selectObject( _PTR(SObject)() ); selectionDone(); } } // enable/disable popup for choice of geom selection way bool enable = false; QString aMeshEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Mesh ); if ( _PTR(SObject) pMesh = studyDS()->FindObjectID( aMeshEntry.latin1() )) { SMESH::SMESH_Mesh_var mesh = SMESH::SObjectToInterface( pMesh ); if ( !mesh->_is_nil() ) enable = ( shapeDim > 1 ) && ( mesh->NbEdges() > 0 ); } myDlg->setGeomPopupEnabled( enable ); } } else { myDlg->enableTab( SMESH::DIM_3D ); QStringList hypList; availableHyps( SMESH::DIM_3D, Algo, hypList, myAvailableHypData[SMESH::DIM_3D][Algo]); SMESHGUI_MeshTab* aTab = myDlg->tab( SMESH::DIM_3D ); aTab->setAvailableHyps( Algo, hypList ); for (int i = SMESH::DIM_0D;i < SMESH::DIM_3D; ++i) { myDlg->disableTab(i); } //Hide labels and fields (Mesh ang Geometry) myDlg->setObjectShown( SMESHGUI_MeshDlg::Mesh, false ); myDlg->setObjectShown( SMESHGUI_MeshDlg::Geom, false ); myDlg->adjustSize(); readMesh(); } } catch ( const SALOME::SALOME_Exception& S_ex ) { SalomeApp_Tools::QtCatchCorbaException( S_ex ); } catch ( ... ) { } } //================================================================================ /*! * \brief Verifies validity of input data * \param theMess - Output parameter intended for returning error message * \retval bool - TRUE if input data is valid, false otherwise * * Verifies validity of input data. This method is called when "Apply" or "OK" button * is pressed before mesh creation or editing. */ //================================================================================ bool SMESHGUI_MeshOp::isValid( QString& theMess ) const { // Selected object to be edited if ( !myToCreate && myDlg->selectedObject( SMESHGUI_MeshDlg::Obj ) == "" ) { theMess = tr( "THERE_IS_NO_OBJECT_FOR_EDITING" ); return false; } // Name QString aMeshName = myDlg->objectText( SMESHGUI_MeshDlg::Obj ); aMeshName = aMeshName.stripWhiteSpace(); if ( aMeshName == "" ) { theMess = myIsMesh ? tr( "NAME_OF_MESH_IS_EMPTY" ) : tr( "NAME_OF_SUBMESH_IS_EMPTY" ); return false; } /* // Imported mesh, if create sub-mesh or edit mesh if ( !myToCreate || ( myToCreate && !myIsMesh )) { QString aMeshEntry = myDlg->selectedObject ( myToCreate ? SMESHGUI_MeshDlg::Mesh : SMESHGUI_MeshDlg::Obj ); if ( _PTR(SObject) pMesh = studyDS()->FindObjectID( aMeshEntry.latin1() )) { SMESH::SMESH_Mesh_var mesh = SMESH::SObjectToInterface( pMesh ); if ( !mesh->_is_nil() && CORBA::is_nil( mesh->GetShapeToMesh() )) { theMess = tr( "IMPORTED_MESH" ); return false; } } }*/ // Geom if ( myToCreate ) { QString aGeomEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Geom ); if ( aGeomEntry == "" ) { theMess = tr( "GEOMETRY_OBJECT_IS_NOT_DEFINED" ); return false; } _PTR(SObject) pGeom = studyDS()->FindObjectID( aGeomEntry.latin1() ); if ( !pGeom || GEOM::GEOM_Object::_narrow( _CAST( SObject,pGeom )->GetObject() )->_is_nil() ) { theMess = tr( "GEOMETRY_OBJECT_IS_NULL" ); return false; } // Mesh if ( !myIsMesh ) // i.e sub-mesh creation, { QString aMeshEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Mesh ); if ( aMeshEntry == "" ) { theMess = tr( "MESH_IS_NOT_DEFINED" ); return false; } _PTR(SObject) pMesh = studyDS()->FindObjectID( aMeshEntry.latin1() ); if ( !pMesh || SMESH::SMESH_Mesh::_narrow( _CAST( SObject,pMesh )->GetObject() )->_is_nil() ) { theMess = tr( "MESH_IS_NULL" ); return false; } if ( !isSubshapeOk() ) { theMess = tr( "INVALID_SUBSHAPE" ); return false; } } } return true; } //================================================================================ /*! * \brief check compatibility of the algorithm and another algorithm or hypothesis * \param theAlgoData - algorithm data * \param theHypData - hypothesis data * \param theHypType - hypothesis type * \param theHypTypeName - hypothesis type name, must be provided if 2-nd arg is not algo * \retval bool - check result */ //================================================================================ static bool isCompatible(const HypothesisData* theAlgoData, const HypothesisData* theHypData, const int theHypType) { if ( !theAlgoData ) return true; if ( theHypType == SMESHGUI_MeshOp::Algo ) return SMESH::IsCompatibleAlgorithm( theAlgoData, theHypData ); bool isOptional; return ( SMESH::IsAvailableHypothesis( theAlgoData, theHypData->TypeName, isOptional )); } //================================================================================ /*! * \brief Gets available hypotheses or algorithms * \param theDim - specifies dimension of returned hypotheses/algorifms * \param theHypType - specifies whether algorims or hypotheses or additional ones * are retrieved (possible values are in HypType enumeration) * \param theHyps - Output list of hypotheses' names * \param theAlgoData - to select hypos able to be used by this algo (optional) * * Gets available hypotheses or algorithm in accordance with input parameters */ //================================================================================ void SMESHGUI_MeshOp::availableHyps( const int theDim, const int theHypType, QStringList& theHyps, THypDataList& theDataList, HypothesisData* theAlgoData ) const { theDataList.clear(); theHyps.clear(); bool isAlgo = ( theHypType == Algo ); bool isAux = ( theHypType == AddHyp ); QStringList aHypTypeNameList = SMESH::GetAvailableHypotheses( isAlgo, theDim, isAux, myIsOnGeometry ); QStringList::const_iterator anIter; for ( anIter = aHypTypeNameList.begin(); anIter != aHypTypeNameList.end(); ++anIter ) { HypothesisData* aData = SMESH::GetHypothesisData( *anIter ); if ( isCompatible ( theAlgoData, aData, theHypType )) { theDataList.append( aData ); theHyps.append( aData->Label ); } } } //================================================================================ /*! * \brief Gets existing hypotheses or algorithms * \param theDim - specifies dimension of returned hypotheses/algorifms * \param theHypType - specifies whether algorims or hypotheses or additional ones * are retrieved (possible values are in HypType enumeration) * \param theFather - start object for finding ( may be component, mesh, or sub-mesh ) * \param theHyps - output list of names. * \param theHypVars - output list of variables. * \param theAlgoData - to select hypos able to be used by this algo (optional) * * Gets existing (i.e. already created) hypotheses or algorithm in accordance with * input parameters */ //================================================================================ void SMESHGUI_MeshOp::existingHyps( const int theDim, const int theHypType, _PTR(SObject) theFather, QStringList& theHyps, THypList& theHypList, HypothesisData* theAlgoData) { // Clear hypoheses list theHyps.clear(); theHypList.clear(); if ( !theFather ) return; const bool isAux = ( theHypType == AddHyp ); _PTR(SObject) aHypRoot; _PTR(GenericAttribute) anAttr; _PTR(AttributeName) aName; _PTR(AttributeIOR) anIOR; bool isMesh = !_CAST( SComponent, theFather ); int aPart = -1; if ( isMesh ) aPart = theHypType == Algo ? SMESH::Tag_RefOnAppliedAlgorithms : SMESH::Tag_RefOnAppliedHypothesis; else aPart = theHypType == Algo ? SMESH::Tag_AlgorithmsRoot : SMESH::Tag_HypothesisRoot; if ( theFather->FindSubObject( aPart, aHypRoot ) ) { _PTR(ChildIterator) anIter = SMESH::GetActiveStudyDocument()->NewChildIterator( aHypRoot ); for (; anIter->More(); anIter->Next() ) { _PTR(SObject) anObj = anIter->Value(); if ( isMesh ) // i.e. mesh or submesh { _PTR(SObject) aRefObj; if ( anObj->ReferencedObject( aRefObj ) ) anObj = aRefObj; else continue; } if ( anObj->FindAttribute( anAttr, "AttributeName" ) ) { aName = anAttr; CORBA::Object_var aVar = _CAST(SObject,anObj)->GetObject(); if ( !CORBA::is_nil( aVar ) ) { SMESH::SMESH_Hypothesis_var aHypVar = SMESH::SMESH_Hypothesis::_narrow( aVar ); if ( !aHypVar->_is_nil() ) { CORBA::String_var aHypType( aHypVar->GetName() ); HypothesisData* aData = SMESH::GetHypothesisData( aHypType ); if ( ( theDim == -1 || aData->Dim.contains( theDim ) ) && ( isCompatible ( theAlgoData, aData, theHypType )) && ( isAux == aData->IsAux )) { std::string aHypName = aName->Value(); theHyps.append( aHypName.c_str() ); theHypList.append( THypItem( aHypVar, aHypName.c_str() ) ); } } } } } } } //================================================================================ /*! * \brief If create or edit a submesh, return a hypothesis holding parameters used * to mesh a subshape * \param aHypType - The hypothesis type name * \param aServerLib - Server library name * \param hypData - The structure holding the hypothesis type etc. * \retval SMESH::SMESH_Hypothesis_var - the hypothesis holding parameter values */ //================================================================================ SMESH::SMESH_Hypothesis_var SMESHGUI_MeshOp::getInitParamsHypothesis( const QString& aHypType, const QString& aServerLib ) const { if ( aHypType.isEmpty() || aServerLib.isEmpty() ) return SMESH::SMESH_Hypothesis::_nil(); const int nbColonsInMeshEntry = 3; bool isSubMesh = myToCreate ? !myIsMesh : myDlg->selectedObject( SMESHGUI_MeshDlg::Obj ).contains(':') > nbColonsInMeshEntry; if ( isSubMesh ) { // get mesh and geom object SMESH::SMESH_Mesh_var aMeshVar = SMESH::SMESH_Mesh::_nil(); GEOM::GEOM_Object_var aGeomVar = GEOM::GEOM_Object::_nil(); QString anEntry = myDlg->selectedObject ( myToCreate ? SMESHGUI_MeshDlg::Mesh : SMESHGUI_MeshDlg::Obj ); if ( _PTR(SObject) pObj = studyDS()->FindObjectID( anEntry.latin1() )) { CORBA::Object_ptr Obj = _CAST( SObject,pObj )->GetObject(); if ( myToCreate ) // mesh and geom may be selected { aMeshVar = SMESH::SMESH_Mesh::_narrow( Obj ); anEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Geom ); if ( _PTR(SObject) pGeom = studyDS()->FindObjectID( anEntry.latin1() )) aGeomVar= GEOM::GEOM_Object::_narrow( _CAST( SObject,pGeom )->GetObject() ); } else // edition: sub-mesh may be selected { SMESH::SMESH_subMesh_var sm = SMESH::SMESH_subMesh::_narrow( Obj ); if ( !sm->_is_nil() ) { aMeshVar = sm->GetFather(); aGeomVar = sm->GetSubShape(); } } } if ( !aMeshVar->_is_nil() && !aGeomVar->_is_nil() ) return SMESHGUI::GetSMESHGen()->GetHypothesisParameterValues( aHypType, aServerLib, aMeshVar, aGeomVar ); } return SMESH::SMESH_Hypothesis::_nil(); } //================================================================================ /*! * \Brief Returns tab dimention * \param tab - the tab in the dlg * \param dlg - my dialogue * \retval int - dimention */ //================================================================================ static int getTabDim (const QObject* tab, SMESHGUI_MeshDlg* dlg ) { int aDim = -1; for (int i = SMESH::DIM_0D; i <= SMESH::DIM_3D; i++) if (tab == dlg->tab(i)) aDim = i; return aDim; } //================================================================================ /*! * \brief Create hypothesis * \param theHypType - hypothesis category (main or additional) * \param theIndex - index of type of hypothesis to be cerated * * Specifies dimension of hypothesis to be created (using sender() method), * specifies its type and calls method for hypothesis creation */ //================================================================================ void SMESHGUI_MeshOp::onCreateHyp( const int theHypType, const int theIndex ) { // Specifies dimension of hypothesis to be created int aDim = getTabDim( sender(), myDlg ); if (aDim == -1) return; // Specifies type of hypothesis to be created THypDataList& dataList = myAvailableHypData[ aDim ][ theHypType ]; if (theIndex < 0 || theIndex >= dataList.count()) return; QString aHypTypeName = dataList[ theIndex ]->TypeName; // Create hypothesis createHypothesis(aDim, theHypType, aHypTypeName); } //================================================================================ /*! * Create hypothesis and update dialog. * \param theDim - dimension of hypothesis to be created * \param theType - hypothesis category (algorithm, hypothesis, additional hypothesis) * \param theTypeName - specifies hypothesis to be created */ //================================================================================ namespace { QString GetUniqueName (const QStringList& theHypNames, const QString& theName, size_t theIteration = 1) { QString aName = theName + "_" + QString::number( theIteration ); if ( theHypNames.contains( aName ) ) return GetUniqueName( theHypNames, theName, ++theIteration ); return aName; } } void SMESHGUI_MeshOp::createHypothesis (const int theDim, const int theType, const QString& theTypeName) { // During a hypothesis creation we might need to select some objects. // Main dialog must not update it's own selected objects in this case. dlg()->deactivateAll(); HypothesisData* aData = SMESH::GetHypothesisData(theTypeName.latin1()); if (!aData) return; QStringList aHypNames; TDim2Type2HypList::const_iterator aDimIter = myExistingHyps.begin(); for (; aDimIter != myExistingHyps.end(); aDimIter++) { const TType2HypList& aType2HypList = aDimIter.data(); TType2HypList::const_iterator aTypeIter = aType2HypList.begin(); for (; aTypeIter != aType2HypList.end(); aTypeIter++) { const THypList& aHypList = aTypeIter.data(); THypList::const_iterator anIter = aHypList.begin(); for (; anIter != aHypList.end(); anIter++) { const THypItem& aHypItem = *anIter; const QString& aHypName = aHypItem.second; aHypNames.append(aHypName); } } } QString aHypName = GetUniqueName( aHypNames, aData->Label); // existing hypos int nbHyp = myExistingHyps[theDim][theType].count(); QString aClientLibName = aData->ClientLibName; if (aClientLibName == "") { // Call hypothesis creation server method (without GUI) SMESH::CreateHypothesis(theTypeName, aHypName, false); } else { // Get hypotheses creator client (GUI) SMESHGUI_GenericHypothesisCreator* aCreator = SMESH::GetHypothesisCreator(theTypeName); // Create hypothesis if (aCreator) { // When create or edit a submesh, try to initialize a new hypothesis // with values used to mesh a subshape SMESH::SMESH_Hypothesis_var initParamHyp = getInitParamsHypothesis(theTypeName, aData->ServerLibName); myDlg->setEnabled( false ); aCreator->create(initParamHyp, aHypName, myDlg); myDlg->setEnabled( true ); } else { SMESH::CreateHypothesis(theTypeName, aHypName, false); } } _PTR(SComponent) aFather = SMESH::GetActiveStudyDocument()->FindComponent("SMESH"); HypothesisData* algoData = hypData( theDim, Algo, currentHyp( theDim, Algo )); QStringList aNewHyps; existingHyps(theDim, theType, aFather, aNewHyps, myExistingHyps[theDim][theType], algoData); if (aNewHyps.count() > nbHyp) { for (int i = nbHyp; i < aNewHyps.count(); i++) myDlg->tab(theDim)->addHyp(theType, aNewHyps[i]); } } //================================================================================ /*! * \brief Calls plugin methods for hypothesis editing * \param theHypType - specifies whether main hypothesis or additional one * is edited * \param theIndex - index of existing hypothesis * * Calls plugin methods for hypothesis editing */ //================================================================================ void SMESHGUI_MeshOp::onEditHyp( const int theHypType, const int theIndex ) { // Speicfies dimension of hypothesis to be created int aDim = getTabDim( sender(), myDlg ); if (aDim == -1) return; const THypList& aList = myExistingHyps[ aDim ][ theHypType ]; if ( theIndex < 0 || theIndex >= aList.count() ) return; const THypItem& aHypItem = aList[ theIndex ]; SMESH::SMESH_Hypothesis_var aHyp = aHypItem.first; if ( aHyp->_is_nil() ) return; CORBA::String_var aTypeName = aHyp->GetName(); SMESHGUI_GenericHypothesisCreator* aCreator = SMESH::GetHypothesisCreator( aTypeName ); if ( aCreator ) { myDlg->setEnabled( false ); aCreator->edit( aHyp.in(), aHypItem.second, dlg() ); myDlg->setEnabled( true ); } } //================================================================================ /*! * \brief access to hypothesis data * \param theDim - hyp dimension * \param theHypType - hyp type (Algo,MainHyp or AddHyp) * \param theIndex - index in the list * \retval HypothesisData* - result data, may be 0 */ //================================================================================ HypothesisData* SMESHGUI_MeshOp::hypData( const int theDim, const int theHypType, const int theIndex) { if ( theDim > -1 && theDim <= SMESH::DIM_3D && theHypType > -1 && theHypType < NbHypTypes && theIndex > -1 && theIndex < myAvailableHypData[ theDim ][ theHypType ].count() ) return myAvailableHypData[ theDim ][ theHypType ][ theIndex ]; return 0; } //================================================================================ /*! * \brief Set available algos and hypos according to the selected algorithm * \param theIndex - algorithm index */ //================================================================================ void SMESHGUI_MeshOp::onAlgoSelected( const int theIndex, const int theDim ) { if ( myIgnoreAlgoSelection ) return; int aDim = theDim < 0 ? getTabDim( sender(), myDlg ): theDim; if (aDim == -1) return; // find highest available dimension, all algos of this dimension are available for choice int aTopDim = -1; for (int i = SMESH::DIM_0D; i <= SMESH::DIM_3D; i++) if (isAccessibleDim( i )) aTopDim = i; if (aTopDim == -1) return; const bool isSubmesh = ( myToCreate ? !myIsMesh : myDlg->isObjectShown( SMESHGUI_MeshDlg::Mesh )); HypothesisData* algoData = hypData( aDim, Algo, theIndex ); HypothesisData* algoByDim[4]; algoByDim[ aDim ] = algoData; QStringList anAvailable; if ( !algoData ) { // all algos becomes available availableHyps( aDim, Algo, anAvailable, myAvailableHypData[ aDim ][ Algo ]); myDlg->tab( aDim )->setAvailableHyps( Algo, anAvailable ); } // check that algorithms of other dimentions are compatible with // the selected one // 2 loops: backward and forward from algo dimension for ( int forward = false; forward <= true; ++forward ) { int dim = aDim + 1, lastDim = SMESH::DIM_3D, dir = 1; if ( !forward ) { dim = aDim - 1; lastDim = SMESH::DIM_0D; dir = -1; } HypothesisData* prevAlgo = algoData; bool noCompatible = false; for (; dim * dir <= lastDim * dir; dim += dir) { if ( !isAccessibleDim( dim )) continue; if ( noCompatible ) { // the selected algo has no compatible ones anAvailable.clear(); myDlg->tab( dim )->setAvailableHyps( Algo, anAvailable ); myAvailableHypData[dim][Algo].clear(); algoByDim[ dim ] = 0; continue; } // get currently selected algo int algoIndex = currentHyp( dim, Algo ); HypothesisData* curAlgo = hypData( dim, Algo, algoIndex ); if ( curAlgo ) { // some algo selected if ( !isCompatible( prevAlgo, curAlgo, Algo )) curAlgo = 0; } // set new available algoritms availableHyps( dim, Algo, anAvailable, myAvailableHypData[dim][Algo], prevAlgo ); HypothesisData* soleCompatible = 0; if ( anAvailable.count() == 1 ) soleCompatible = myAvailableHypData[dim][Algo][0]; if ( dim == aTopDim && prevAlgo ) // all available algoritms should be selectable any way availableHyps( dim, Algo, anAvailable, myAvailableHypData[dim][Algo], 0 ); myDlg->tab( dim )->setAvailableHyps( Algo, anAvailable ); noCompatible = anAvailable.isEmpty(); // restore previously selected algo algoIndex = myAvailableHypData[dim][Algo].findIndex( curAlgo ); if ( !isSubmesh && algoIndex < 0 && soleCompatible && !forward && dim != SMESH::DIM_0D) // select the sole compatible algo algoIndex = myAvailableHypData[dim][Algo].findIndex( soleCompatible ); setCurrentHyp( dim, Algo, algoIndex ); // remember current algo prevAlgo = algoByDim[ dim ] = hypData( dim, Algo, algoIndex ); } } // set hypotheses corresponding to the found algoritms _PTR(SObject) pObj = SMESH::GetActiveStudyDocument()->FindComponent("SMESH"); for ( int dim = SMESH::DIM_0D; dim <= SMESH::DIM_3D; dim++ ) { if ( !isAccessibleDim( dim )) continue; for ( int type = MainHyp; type < NbHypTypes; type++ ) { myAvailableHypData[ dim ][ type ].clear(); QStringList anAvailable, anExisting; HypothesisData* curAlgo = algoByDim[ dim ]; int hypIndex = currentHyp( dim, type ); SMESH::SMESH_Hypothesis_var curHyp; if ( hypIndex >= 0 && hypIndex < myExistingHyps[ dim ][ type ].count() ) curHyp = myExistingHyps[ dim ][ type ][ hypIndex ].first; if ( !myToCreate && !curAlgo && !curHyp->_is_nil() ) { // edition, algo not selected // try to find algo by selected hypothesis in order to keep it selected bool algoDeselectedByUser = ( theDim < 0 && aDim == dim ); CORBA::String_var curHypType = curHyp->GetName(); if ( !algoDeselectedByUser && myObjHyps[ dim ][ type ].count() > 0 && curHypType == myObjHyps[ dim ][ type ].first().first->GetName()) { CORBA::String_var aName = curHyp->GetName(); HypothesisData* hypData = SMESH::GetHypothesisData( aName ); for (int i = 0; i < myAvailableHypData[ dim ][ Algo ].count(); ++i) { curAlgo = myAvailableHypData[ dim ][ Algo ][ i ]; if (curAlgo && hypData && isCompatible(curAlgo, hypData, type)) break; else curAlgo = 0; } } } // get hyps compatible with curAlgo if ( curAlgo ) { // check if a selected hyp is compatible with the curAlgo if ( !curHyp->_is_nil() ) { CORBA::String_var aName = curHyp->GetName(); HypothesisData* hypData = SMESH::GetHypothesisData( aName ); if ( !isCompatible( curAlgo, hypData, type )) curHyp = SMESH::SMESH_Hypothesis::_nil(); } existingHyps( dim, type, pObj, anExisting, myExistingHyps[ dim ][ type ], curAlgo); availableHyps( dim, type, anAvailable, myAvailableHypData[ dim ][ type ], curAlgo); } // set list of hypotheses myDlg->tab( dim )->setAvailableHyps( type, anAvailable ); myDlg->tab( dim )->setExistingHyps( type, anExisting ); // set current existing hypothesis if ( !curHyp->_is_nil() && !anExisting.isEmpty() ) hypIndex = this->find( curHyp, myExistingHyps[ dim ][ type ]); else hypIndex = -1; if ( !isSubmesh && hypIndex < 0 && anExisting.count() == 1 ) { // none is yet selected => select the sole existing if it is not optional CORBA::String_var hypTypeName = myExistingHyps[ dim ][ type ].first().first->GetName(); bool isOptional = true; if ( algoByDim[ dim ] && SMESH::IsAvailableHypothesis( algoByDim[ dim ], hypTypeName.in(), isOptional ) && !isOptional ) hypIndex = 0; } setCurrentHyp( dim, type, hypIndex ); } } } //================================================================================ /*! * \brief Creates and selects hypothesis of hypotheses set * \param theSetName - The name of hypotheses set */ //================================================================================ void SMESHGUI_MeshOp::onHypoSet( const QString& theSetName ) { HypothesesSet* aHypoSet = SMESH::GetHypothesesSet(theSetName); if (!aHypoSet) return; // clear all hyps for (int dim = SMESH::DIM_0D; dim <= SMESH::DIM_3D; dim++) { setCurrentHyp(dim, Algo, -1); setCurrentHyp(dim, AddHyp, -1); setCurrentHyp(dim, MainHyp, -1); } for (int aHypType = Algo; aHypType < AddHyp; aHypType++) { bool isAlgo = (aHypType == Algo); // set hyps from the set QStringList* aHypoList = isAlgo ? &aHypoSet->AlgoList : &aHypoSet->HypoList; for (int i = 0, n = aHypoList->count(); i < n; i++) { const QString& aHypoTypeName = (*aHypoList)[ i ]; HypothesisData* aHypData = SMESH::GetHypothesisData(aHypoTypeName); if (!aHypData) continue; int aDim = aHypData->Dim[0]; // create or/and set if (isAlgo) { int index = myAvailableHypData[aDim][Algo].findIndex( aHypData ); if ( index < 0 ) { QStringList anAvailable; availableHyps( aDim, Algo, anAvailable, myAvailableHypData[aDim][Algo] ); myDlg->tab( aDim )->setAvailableHyps( Algo, anAvailable ); index = myAvailableHypData[aDim][Algo].findIndex( aHypData ); } setCurrentHyp( aDim, Algo, index ); onAlgoSelected( index, aDim ); } else { bool mainHyp = true; QStringList anAvailable; availableHyps( aDim, MainHyp, anAvailable, myAvailableHypData[aDim][MainHyp] ); myDlg->tab( aDim )->setAvailableHyps( MainHyp, anAvailable ); int index = myAvailableHypData[aDim][MainHyp].findIndex( aHypData ); if ( index < 0 ) { mainHyp = false; index = myAvailableHypData[aDim][AddHyp].findIndex( aHypData ); } if (index >= 0) createHypothesis(aDim, mainHyp ? MainHyp : AddHyp, aHypoTypeName); } } // loop on hypos in the set } // loop on algo/hypo } //================================================================================ /*! * \brief Creates mesh * \param theMess - Output parameter intended for returning error message * \retval bool - TRUE if mesh is created, FALSE otherwise * * Creates mesh */ //================================================================================ bool SMESHGUI_MeshOp::createMesh( QString& theMess ) { theMess = ""; //QString aGeomEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Geom ); //QString aGeomEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Geom ); QStringList aList; myDlg->selectedObject( SMESHGUI_MeshDlg::Geom, aList ); QStringList::Iterator it = aList.begin(); for (; it!=aList.end(); it++) { QString aGeomEntry = *it; _PTR(SObject) pGeom = studyDS()->FindObjectID( aGeomEntry.latin1() ); GEOM::GEOM_Object_var aGeomVar = GEOM::GEOM_Object::_narrow( _CAST( SObject,pGeom )->GetObject() ); SMESH::SMESH_Gen_var aSMESHGen = SMESHGUI::GetSMESHGen(); if ( aSMESHGen->_is_nil() ) return false; SUIT_OverrideCursor aWaitCursor; // create mesh SMESH::SMESH_Mesh_var aMeshVar = aSMESHGen->CreateMesh( aGeomVar ); if ( aMeshVar->_is_nil() ) return false; _PTR(SObject) aMeshSO = SMESH::FindSObject( aMeshVar.in() ); if ( aMeshSO ) SMESH::SetName( aMeshSO, myDlg->objectText( SMESHGUI_MeshDlg::Obj ).latin1() ); for ( int aDim = SMESH::DIM_0D; aDim <= SMESH::DIM_3D; aDim++ ) { if ( !isAccessibleDim( aDim )) continue; // assign hypotheses for ( int aHypType = MainHyp; aHypType <= AddHyp; aHypType++ ) { int aHypIndex = currentHyp( aDim, aHypType ); if ( aHypIndex >= 0 && aHypIndex < myExistingHyps[ aDim ][ aHypType ].count() ) { SMESH::SMESH_Hypothesis_var aHypVar = myExistingHyps[ aDim ][ aHypType ][ aHypIndex ].first; if ( !aHypVar->_is_nil() ) SMESH::AddHypothesisOnMesh( aMeshVar, aHypVar ); } } // find or create algorithm SMESH::SMESH_Hypothesis_var anAlgoVar = getAlgo( aDim ); if ( !anAlgoVar->_is_nil() ) SMESH::AddHypothesisOnMesh( aMeshVar, anAlgoVar ); } } return true; } //================================================================================ /*! * \brief Creates sub-mesh * \param theMess - Output parameter intended for returning error message * \retval bool - TRUE if sub-mesh is created, FALSE otherwise * * Creates sub-mesh */ //================================================================================ bool SMESHGUI_MeshOp::createSubMesh( QString& theMess ) { theMess = ""; SMESH::SMESH_Gen_var aSMESHGen = SMESHGUI::GetSMESHGen(); if ( aSMESHGen->_is_nil() ) return false; // get mesh object QString aMeshEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Mesh ); _PTR(SObject) pMesh = studyDS()->FindObjectID( aMeshEntry.latin1() ); SMESH::SMESH_Mesh_var aMeshVar = SMESH::SMESH_Mesh::_narrow( _CAST( SObject,pMesh )->GetObject() ); if (aMeshVar->_is_nil()) return false; // GEOM shape of the main mesh GEOM::GEOM_Object_var mainGeom = aMeshVar->GetShapeToMesh(); // Name for the new sub-mesh QString aName = myDlg->objectText(SMESHGUI_MeshDlg::Obj); // get geom object GEOM::GEOM_Object_var aGeomVar; QStringList aGEOMs; myDlg->selectedObject(SMESHGUI_MeshDlg::Geom, aGEOMs); if (aGEOMs.count() == 1) { //QString aGeomEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Geom ); QString aGeomEntry = aGEOMs.first(); _PTR(SObject) pGeom = studyDS()->FindObjectID( aGeomEntry.latin1() ); aGeomVar = GEOM::GEOM_Object::_narrow( _CAST( SObject,pGeom )->GetObject() ); } else if (aGEOMs.count() > 1) { // create a GEOM group GEOM::GEOM_Gen_var geomGen = SMESH::GetGEOMGen(); _PTR(Study) aStudy = SMESH::GetActiveStudyDocument(); if (!geomGen->_is_nil() && aStudy) { GEOM::GEOM_IGroupOperations_var op = geomGen->GetIGroupOperations(aStudy->StudyId()); if (!op->_is_nil()) { // check and add all selected GEOM objects: they must be // a sub-shapes of the main GEOM and must be of one type int iSubSh = 0; TopAbs_ShapeEnum aGroupType = TopAbs_SHAPE; GEOM::ListOfGO_var aSeq = new GEOM::ListOfGO; aSeq->length(aGEOMs.count()); QStringList::const_iterator aSubShapesIter = aGEOMs.begin(); for (; aSubShapesIter != aGEOMs.end(); aSubShapesIter++, iSubSh++) { QString aSubGeomEntry = (*aSubShapesIter); _PTR(SObject) pSubGeom = studyDS()->FindObjectID(aSubGeomEntry.latin1()); GEOM::GEOM_Object_var aSubGeomVar = GEOM::GEOM_Object::_narrow(_CAST(SObject,pSubGeom)->GetObject()); TopAbs_ShapeEnum aSubShapeType = (TopAbs_ShapeEnum)aSubGeomVar->GetShapeType(); if (iSubSh == 0) { aGroupType = aSubShapeType; } else { if (aSubShapeType != aGroupType) aGroupType = TopAbs_SHAPE; } aSeq[iSubSh] = aSubGeomVar; } // create a group GEOM::GEOM_Object_var aGroupVar = op->CreateGroup(mainGeom, aGroupType); op->UnionList(aGroupVar, aSeq); if (op->IsDone()) { aGeomVar = aGroupVar; // publish the GEOM group in study QString aNewGeomGroupName ("Auto_group_for_"); aNewGeomGroupName += aName; SALOMEDS::SObject_var aNewGroupSO = geomGen->AddInStudy(aSMESHGen->GetCurrentStudy(), aGeomVar, aNewGeomGroupName, mainGeom); } } } } else { } if (aGeomVar->_is_nil()) return false; SUIT_OverrideCursor aWaitCursor; // create sub-mesh SMESH::SMESH_subMesh_var aSubMeshVar = aMeshVar->GetSubMesh( aGeomVar, aName.latin1() ); for ( int aDim = SMESH::DIM_0D; aDim <= SMESH::DIM_3D; aDim++ ) { if ( !isAccessibleDim( aDim )) continue; // find or create algorithm SMESH::SMESH_Hypothesis_var anAlgoVar = getAlgo( aDim ); if ( !anAlgoVar->_is_nil() ) SMESH::AddHypothesisOnSubMesh( aSubMeshVar, anAlgoVar ); // assign hypotheses for ( int aHypType = MainHyp; aHypType <= AddHyp; aHypType++ ) { int aHypIndex = currentHyp( aDim, aHypType ); if ( aHypIndex >= 0 && aHypIndex < myExistingHyps[ aDim ][ aHypType ].count() ) { SMESH::SMESH_Hypothesis_var aHypVar = myExistingHyps[ aDim ][ aHypType ][ aHypIndex ].first; if ( !aHypVar->_is_nil() ) SMESH::AddHypothesisOnSubMesh( aSubMeshVar, aHypVar ); } } } // deselect geometry: next submesh should be created on other subshape myDlg->clearSelection( SMESHGUI_MeshDlg::Geom ); selectObject( _PTR(SObject)() ); selectionDone(); return true; } //================================================================================ /*! * \brief Gets current hypothesis or algorithms * \param theDim - dimension of hypothesis or algorithm * \param theHypType - Type of hypothesis (Algo, MainHyp, AddHyp) * \retval int - current hypothesis or algorithms * * Gets current hypothesis or algorithms */ //================================================================================ int SMESHGUI_MeshOp::currentHyp( const int theDim, const int theHypType ) const { return myDlg->tab( theDim )->currentHyp( theHypType ) - 1; } //================================================================================ /*! * \brief Returns true if hypotheses of given dim can be assigned * \param theDim - hypotheses dimension * \retval bool - result */ //================================================================================ bool SMESHGUI_MeshOp::isAccessibleDim( const int theDim ) const { return myDlg->tab( theDim )->isEnabled(); } //================================================================================ /*! * \brief Sets current hypothesis or algorithms * \param theDim - dimension of hypothesis or algorithm * \param theHypType - Type of hypothesis (Algo, MainHyp, AddHyp) * \param theIndex - Index of hypothesis * * Gets current hypothesis or algorithms */ //================================================================================ void SMESHGUI_MeshOp::setCurrentHyp( const int theDim, const int theHypType, const int theIndex ) { myIgnoreAlgoSelection = true; myDlg->tab( theDim )->setCurrentHyp( theHypType, theIndex + 1 ); myIgnoreAlgoSelection = false; } //================================================================================ /*! * \brief Generates default and sets mesh/submesh name * * Generates and sets default mesh/submesh name(Mesh_1, Mesh_2, etc.) */ //================================================================================ void SMESHGUI_MeshOp::setDefaultName() const { QString aResName; _PTR(Study) aStudy = SMESH::GetActiveStudyDocument(); int i = 1; QString aPrefix = tr( myIsMesh ? "SMESH_OBJECT_MESH" : "SMESH_SUBMESH" ) + "_"; _PTR(SObject) anObj; do { aResName = aPrefix + QString::number( i++ ); anObj = aStudy->FindObject( aResName.latin1() ); } while ( anObj ); QLineEdit* aControl = ( QLineEdit* )myDlg->objectWg( SMESHGUI_MeshDlg::Obj, SMESHGUI_MeshDlg::Control ); aControl->setText( aResName ); } //================================================================================ /*! * \brief Gets algorithm or creates it if necessary * \param theDim - specifies dimension of returned hypotheses/algorifms * \retval SMESH::SMESH_Hypothesis_var - algorithm * * Gets algorithm or creates it if necessary */ //================================================================================ SMESH::SMESH_Hypothesis_var SMESHGUI_MeshOp::getAlgo( const int theDim ) { SMESH::SMESH_Hypothesis_var anAlgoVar; // get type of the selected algo int aHypIndex = currentHyp( theDim, Algo ); THypDataList& dataList = myAvailableHypData[ theDim ][ Algo ]; if ( aHypIndex < 0 || aHypIndex >= dataList.count()) return anAlgoVar; QString aHypName = dataList[ aHypIndex ]->TypeName; // get existing algoritms _PTR(SObject) pObj = SMESH::GetActiveStudyDocument()->FindComponent("SMESH"); QStringList tmp; existingHyps( theDim, Algo, pObj, tmp, myExistingHyps[ theDim ][ Algo ]); // look for anexisting algo of such a type THypList& aHypVarList = myExistingHyps[ theDim ][ Algo ]; THypList::iterator anIter = aHypVarList.begin(); for (; anIter != aHypVarList.end(); anIter++) { SMESH::SMESH_Hypothesis_var aHypVar = (*anIter).first; CORBA::String_var aName = aHypVar->GetName(); if ( !aHypVar->_is_nil() && aHypName == aName ) { anAlgoVar = aHypVar; break; } } if (anAlgoVar->_is_nil()) { HypothesisData* aHypData = SMESH::GetHypothesisData( aHypName ); if (aHypData) { QString aClientLibName = aHypData->ClientLibName; if (aClientLibName == "") { // Call hypothesis creation server method (without GUI) SMESH::CreateHypothesis(aHypName, aHypData->Label, true); } else { // Get hypotheses creator client (GUI) SMESHGUI_GenericHypothesisCreator* aCreator = SMESH::GetHypothesisCreator(aHypName); // Create algorithm if (aCreator) aCreator->create(true, aHypName, myDlg); else SMESH::CreateHypothesis(aHypName, aHypData->Label, true); } QStringList tmpList; _PTR(SComponent) aFather = SMESH::GetActiveStudyDocument()->FindComponent( "SMESH" ); existingHyps( theDim, Algo, aFather, tmpList, myExistingHyps[ theDim ][ Algo ] ); } THypList& aNewHypVarList = myExistingHyps[ theDim ][ Algo ]; for ( anIter = aNewHypVarList.begin(); anIter != aNewHypVarList.end(); ++anIter ) { SMESH::SMESH_Hypothesis_var aHypVar = (*anIter).first; CORBA::String_var aName = aHypVar->GetName(); if ( !aHypVar->_is_nil() && aHypName == aName ) { anAlgoVar = aHypVar; break; } } } return anAlgoVar._retn(); } //================================================================================ /*! * \brief Reads parameters of edited mesh and assigns them to the dialog * * Reads parameters of edited mesh and assigns them to the dialog (called when * mesh is edited only) */ //================================================================================ void SMESHGUI_MeshOp::readMesh() { QString anObjEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Obj ); _PTR(SObject) pObj = studyDS()->FindObjectID( anObjEntry.latin1() ); if ( !pObj ) return; if (myIsOnGeometry) { // Get name of mesh if current object is sub-mesh SMESH::SMESH_subMesh_var aSubMeshVar = SMESH::SMESH_subMesh::_narrow( _CAST( SObject,pObj )->GetObject() ); if ( !aSubMeshVar->_is_nil() ) { SMESH::SMESH_Mesh_var aMeshVar = aSubMeshVar->GetFather(); if ( !aMeshVar->_is_nil() ) { _PTR(SObject) aMeshSO = SMESH::FindSObject( aMeshVar ); QString aMeshName = name( aMeshSO ); myDlg->setObjectText( SMESHGUI_MeshDlg::Mesh, aMeshName ); } } // Get name of geometry object GEOM::GEOM_Object_var aGeomVar = SMESH::GetShapeOnMeshOrSubMesh( pObj ); if ( !aGeomVar->_is_nil() ) { _PTR(SObject) aGeomSO = studyDS()->FindObjectID( aGeomVar->GetStudyEntry() ); QString aShapeName = name( aGeomSO ); myDlg->setObjectText( SMESHGUI_MeshDlg::Geom, aShapeName ); } } // Get hypotheses and algorithms assigned to the mesh/sub-mesh QStringList anExisting; const int aDim = ( myIsOnGeometry ) ? SMESH::DIM_0D : SMESH::DIM_3D; for ( int dim = aDim; dim <= SMESH::DIM_3D; dim++ ) { // get algorithm existingHyps( dim, Algo, pObj, anExisting, myObjHyps[ dim ][ Algo ] ); // find algo index among available ones int aHypIndex = -1; if ( myObjHyps[ dim ][ Algo ].count() > 0 ) { SMESH::SMESH_Hypothesis_var aVar = myObjHyps[ dim ][ Algo ].first().first; CORBA::String_var aHypTypeName = aVar->GetName(); HypothesisData* algoData = SMESH::GetHypothesisData( aHypTypeName ); aHypIndex = myAvailableHypData[ dim ][ Algo ].findIndex ( algoData ); // if ( aHypIndex < 0 && algoData ) { // // assigned algo is incompatible with other algorithms // myAvailableHypData[ dim ][ Algo ].push_back( algoData ); // aHypIndex = myAvailableHypData[ dim ][ hypType ].count() - 1; // } } setCurrentHyp( dim, Algo, aHypIndex ); // set existing and available hypothesis according to the selected algo onAlgoSelected( aHypIndex, dim ); } // get hypotheses bool hypWithoutAlgo = false; for ( int dim = aDim; dim <= SMESH::DIM_3D; dim++ ) { for ( int hypType = MainHyp; hypType <= AddHyp; hypType++ ) { // get hypotheses existingHyps( dim, hypType, pObj, anExisting, myObjHyps[ dim ][ hypType ] ); // find index of requered hypothesis among existing ones for this dimension and type int aHypIndex = -1; if ( myObjHyps[ dim ][ hypType ].count() > 0 ) { aHypIndex = find( myObjHyps[ dim ][ hypType ].first().first, myExistingHyps[ dim ][ hypType ] ); if ( aHypIndex < 0 ) { // assigned hypothesis is incompatible with the algorithm if ( currentHyp( dim, Algo ) < 0 ) { // none algo selected; it is edition for sure, of submesh maybe hypWithoutAlgo = true; myExistingHyps[ dim ][ hypType ].push_back( myObjHyps[ dim ][ hypType ].first() ); aHypIndex = myExistingHyps[ dim ][ hypType ].count() - 1; myDlg->tab( dim )->setExistingHyps( hypType, anExisting ); } } } setCurrentHyp( dim, hypType, aHypIndex ); } } // make available other hyps of same type as one without algo if ( hypWithoutAlgo ) onAlgoSelected( currentHyp( 0, Algo ), 0 ); } //================================================================================ /*! * \brief Gets name of object * \param theSO - SObject * \retval QString - name of object * * Gets name of object */ //================================================================================ QString SMESHGUI_MeshOp::name( _PTR(SObject) theSO ) const { QString aResName; if ( theSO ) { _PTR(GenericAttribute) anAttr; _PTR(AttributeName) aNameAttr; if ( theSO->FindAttribute( anAttr, "AttributeName" ) ) { aNameAttr = anAttr; aResName = aNameAttr->Value().c_str(); } } return aResName; } //================================================================================ /*! * \brief Finds hypothesis in input list * \param theHyp - hypothesis to be found * \param theHypList - input list of hypotheses * \retval int - index of hypothesis or -1 if it is not found * * Finds position of hypothesis in input list */ //================================================================================ int SMESHGUI_MeshOp::find( const SMESH::SMESH_Hypothesis_var& theHyp, const THypList& theHypList ) const { int aRes = -1; if ( !theHyp->_is_nil() ) { int i = 0; THypList::const_iterator anIter = theHypList.begin(); for (; anIter != theHypList.end(); ++ anIter) { if ( theHyp->_is_equivalent( (*anIter).first ) ) { aRes = i; break; } i++; } } return aRes; } //================================================================================ /*! * \brief Edits mesh or sub-mesh * \param theMess - Output parameter intended for returning error message * \retval bool - TRUE if mesh is edited succesfully, FALSE otherwise * * Assigns new name hypotheses and algoriths to the mesh or sub-mesh */ //================================================================================ bool SMESHGUI_MeshOp::editMeshOrSubMesh( QString& theMess ) { theMess = ""; SMESH::SMESH_Gen_var aSMESHGen = SMESHGUI::GetSMESHGen(); if ( aSMESHGen->_is_nil() ) return false; QString anObjEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Obj ); _PTR(SObject) pObj = studyDS()->FindObjectID( anObjEntry.latin1() ); if ( !pObj ) return false; SUIT_OverrideCursor aWaitCursor; // Set new name QString aName = myDlg->objectText( SMESHGUI_MeshDlg::Obj ); SMESH::SetName( pObj, aName.latin1() ); int aDim = ( myIsOnGeometry ) ? SMESH::DIM_0D : SMESH::DIM_3D; // First, remove old algos in order to avoid messages on algorithm hiding for ( int dim = aDim; dim <= SMESH::DIM_3D; dim++ ) { if ( isAccessibleDim( dim ) && myObjHyps[ dim ][ Algo ].count() > 0 ) { SMESH::SMESH_Hypothesis_var anOldAlgo = myObjHyps[ dim ][ Algo ].first().first; SMESH::SMESH_Hypothesis_var anAlgoVar = getAlgo( dim ); if ( anAlgoVar->_is_nil() || // no new algo selected or strcmp(anOldAlgo->GetName(), anAlgoVar->GetName()) ) // algo change { // remove old algorithm SMESH::RemoveHypothesisOrAlgorithmOnMesh ( pObj, myObjHyps[ dim ][ Algo ].first().first ); myObjHyps[ dim ][ Algo ].clear(); } } } // Assign new algorithms and hypotheses for ( int dim = aDim; dim <= SMESH::DIM_3D; dim++ ) { if ( !isAccessibleDim( dim )) continue; // find or create algorithm SMESH::SMESH_Hypothesis_var anAlgoVar = getAlgo( dim ); // assign new algorithm if ( !anAlgoVar->_is_nil() && // some algo selected and myObjHyps[ dim ][ Algo ].count() == 0 ) // no algo assigned { SALOMEDS_SObject* aSObject = _CAST(SObject, pObj); CORBA::Object_var anObject = aSObject->GetObject(); SMESH::SMESH_Mesh_var aMeshVar = SMESH::SMESH_Mesh::_narrow( anObject ); bool isMesh = !aMeshVar->_is_nil(); if ( isMesh ) { SMESH::AddHypothesisOnMesh( aMeshVar, anAlgoVar ); } else { SMESH::SMESH_subMesh_var aVar = SMESH::SMESH_subMesh::_narrow( _CAST(SObject,pObj)->GetObject() ); if ( !aVar->_is_nil() ) SMESH::AddHypothesisOnSubMesh( aVar, anAlgoVar ); } myObjHyps[ dim ][ Algo ].append( THypItem( anAlgoVar, aName) ); } // assign hypotheses for ( int hypType = MainHyp; hypType <= AddHyp; hypType++ ) { int aNewHypIndex = currentHyp( dim, hypType ); int anOldHypIndex = -1; // remove old hypotheses if ( myObjHyps[ dim ][ hypType ].count() > 0 ) { anOldHypIndex = find( myObjHyps[ dim ][ hypType ].first().first , myExistingHyps[ dim ][ hypType ] ); if ( aNewHypIndex != anOldHypIndex || // different hyps anOldHypIndex == -1 ) // hyps of different algos { SMESH::RemoveHypothesisOrAlgorithmOnMesh ( pObj, myObjHyps[ dim ][ hypType ].first().first ); myObjHyps[ dim ][ hypType ].clear(); } } // assign new hypotheses if ( aNewHypIndex != anOldHypIndex && aNewHypIndex != -1 ) { SMESH::SMESH_Mesh_var aMeshVar = SMESH::SMESH_Mesh::_narrow( _CAST(SObject,pObj)->GetObject() ); bool isMesh = !aMeshVar->_is_nil(); if ( isMesh ) { SMESH::AddHypothesisOnMesh (aMeshVar, myExistingHyps[ dim ][ hypType ][ aNewHypIndex ].first ); } else { SMESH::SMESH_subMesh_var aVar = SMESH::SMESH_subMesh::_narrow( _CAST(SObject,pObj)->GetObject() ); if ( !aVar->_is_nil() ) SMESH::AddHypothesisOnSubMesh ( aVar, myExistingHyps[ dim ][ hypType ][ aNewHypIndex ].first ); } } // reread all hypotheses of mesh if necessary QStringList anExisting; existingHyps( dim, hypType, pObj, anExisting, myObjHyps[ dim ][ hypType ] ); } } return true; } //================================================================================ /*! * \brief Verifies whether given operator is valid for this one * \param theOtherOp - other operation * \return Returns TRUE if the given operator is valid for this one, FALSE otherwise * * method redefined from base class verifies whether given operator is valid for * this one (i.e. can be started "above" this operator). In current implementation method * retuns false if theOtherOp operation is not intended for deleting objects or mesh * elements. */ //================================================================================ bool SMESHGUI_MeshOp::isValid( SUIT_Operation* theOp ) const { return SMESHGUI_Operation::isValid( theOp ) && !theOp->inherits( "SMESHGUI_MeshOp" ); } //================================================================================ /*! * \brief SLOT. Is called when the user selects a way of geometry selection * \param theByMesh - true if the user wants to find geometry by mesh element */ //================================================================================ void SMESHGUI_MeshOp::onGeomSelectionByMesh( bool theByMesh ) { if ( theByMesh ) { if ( !myShapeByMeshOp ) { myShapeByMeshOp = new SMESHGUI_ShapeByMeshOp(); connect(myShapeByMeshOp, SIGNAL(committed(SUIT_Operation*)), SLOT(onPublishShapeByMeshDlg(SUIT_Operation*))); connect(myShapeByMeshOp, SIGNAL(aborted(SUIT_Operation*)), SLOT(onCloseShapeByMeshDlg(SUIT_Operation*))); } // set mesh object to SMESHGUI_ShapeByMeshOp and start it QString aMeshEntry = myDlg->selectedObject( SMESHGUI_MeshDlg::Mesh ); if ( _PTR(SObject) pMesh = studyDS()->FindObjectID( aMeshEntry.latin1() )) { SMESH::SMESH_Mesh_var aMeshVar = SMESH::SMESH_Mesh::_narrow( _CAST( SObject,pMesh )->GetObject() ); if ( !aMeshVar->_is_nil() ) { myDlg->hide(); // stop processing selection myShapeByMeshOp->setModule( getSMESHGUI() ); myShapeByMeshOp->setStudy( 0 ); // it's really necessary myShapeByMeshOp->SetMesh( aMeshVar ); myShapeByMeshOp->start(); } } } } //================================================================================ /*! * \brief SLOT. Is called when Ok is pressed in SMESHGUI_ShapeByMeshDlg */ //================================================================================ void SMESHGUI_MeshOp::onPublishShapeByMeshDlg(SUIT_Operation* op) { if ( myShapeByMeshOp == op ) { SMESHGUI::GetSMESHGUI()->getApp()->updateObjectBrowser(); //MZN: 24.11.2006 IPAL13980 - Object Browser update added myDlg->show(); // Select a found geometry object GEOM::GEOM_Object_var aGeomVar = myShapeByMeshOp->GetShape(); if ( !aGeomVar->_is_nil() ) { QString ID = aGeomVar->GetStudyEntry(); if ( _PTR(SObject) aGeomSO = studyDS()->FindObjectID( ID.latin1() )) { selectObject( aGeomSO ); selectionDone(); } } } } //================================================================================ /*! * \brief SLOT. Is called when Close is pressed in SMESHGUI_ShapeByMeshDlg */ //================================================================================ void SMESHGUI_MeshOp::onCloseShapeByMeshDlg(SUIT_Operation* op) { if ( myShapeByMeshOp == op && myDlg ) { myDlg->show(); } } //================================================================================ /*! * \brief Selects a SObject * \param theSObj - the SObject to select */ //================================================================================ void SMESHGUI_MeshOp::selectObject( _PTR(SObject) theSObj ) const { if ( LightApp_SelectionMgr* sm = selectionMgr() ) { SALOME_ListIO anIOList; if ( theSObj ) { Handle(SALOME_InteractiveObject) anIO = new SALOME_InteractiveObject ( theSObj->GetID().c_str(), "SMESH", theSObj->GetName().c_str() ); anIOList.Append( anIO ); } sm->setSelectedObjects( anIOList, false ); } }