1 // Copyright (C) 2007-2015 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License, or (at your option) any later version.
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 // Lesser General Public License for more details.
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
22 // File : SMESH_Gen_i.cxx
23 // Author : Paul RASCLE, EDF
27 #include <TopExp_Explorer.hxx>
29 #include <TopoDS_Iterator.hxx>
30 #include <TopoDS_Compound.hxx>
31 #include <TopoDS_CompSolid.hxx>
32 #include <TopoDS_Solid.hxx>
33 #include <TopoDS_Shell.hxx>
34 #include <TopoDS_Face.hxx>
35 #include <TopoDS_Wire.hxx>
36 #include <TopoDS_Edge.hxx>
37 #include <TopoDS_Vertex.hxx>
38 #include <TopoDS_Shape.hxx>
39 #include <TopTools_MapOfShape.hxx>
40 #include <TopTools_IndexedMapOfShape.hxx>
41 #include <TopTools_ListOfShape.hxx>
42 #include <TopTools_ListIteratorOfListOfShape.hxx>
44 #include <BRep_Tool.hxx>
45 #include <TCollection_AsciiString.hxx>
56 #define LibHandle HMODULE
57 #define LoadLib( name ) LoadLibrary( name )
58 #define GetProc GetProcAddress
59 #define UnLoadLib( handle ) FreeLibrary( handle );
61 #define LibHandle void*
62 #define LoadLib( name ) dlopen( name, RTLD_LAZY )
64 #define UnLoadLib( handle ) dlclose( handle );
67 #include "SMESH_Gen_i.hxx"
68 #include "SMESH_version.h"
70 #include "DriverMED_W_SMESHDS_Mesh.h"
71 #include "DriverMED_R_SMESHDS_Mesh.h"
73 #include "DriverCGNS_Read.hxx"
75 #include "MED_Factory.hxx"
76 #include "SMDS_EdgePosition.hxx"
77 #include "SMDS_FacePosition.hxx"
78 #include "SMDS_PolyhedralVolumeOfNodes.hxx"
79 #include "SMDS_SetIterator.hxx"
80 #include "SMDS_SpacePosition.hxx"
81 #include "SMDS_VertexPosition.hxx"
82 #include "SMESHDS_Document.hxx"
83 #include "SMESHDS_Group.hxx"
84 #include "SMESHDS_GroupOnGeom.hxx"
85 #include "SMESH_Algo_i.hxx"
86 #include "SMESH_File.hxx"
87 #include "SMESH_Group.hxx"
88 #include "SMESH_Group_i.hxx"
89 #include "SMESH_Hypothesis.hxx"
90 #include "SMESH_Hypothesis_i.hxx"
91 #include "SMESH_Mesh.hxx"
92 #include "SMESH_MeshEditor.hxx"
93 #include "SMESH_Mesh_i.hxx"
94 #include "SMESH_PreMeshInfo.hxx"
95 #include "SMESH_PythonDump.hxx"
96 #include "SMESH_TryCatch.hxx" // to include after OCC headers!
98 #include CORBA_SERVER_HEADER(SMESH_Group)
99 #include CORBA_SERVER_HEADER(SMESH_Filter)
100 #include CORBA_SERVER_HEADER(SMESH_MeshEditor)
103 #include <GEOM_Client.hxx>
105 #include <Basics_Utils.hxx>
106 #include <Basics_DirUtils.hxx>
108 #include <OpUtil.hxx>
109 #include <SALOMEDS_Tool.hxx>
110 #include <SALOME_Container_i.hxx>
111 #include <SALOME_DataContainer_i.hxx>
112 #include <SALOME_LifeCycleCORBA.hxx>
113 #include <SALOME_NamingService.hxx>
114 #include <Utils_CorbaException.hxx>
115 #include <Utils_ExceptHandlers.hxx>
116 #include <Utils_SINGLETON.hxx>
117 #include <utilities.h>
119 #include CORBA_CLIENT_HEADER(SALOME_ModuleCatalog)
120 #include CORBA_CLIENT_HEADER(SALOME_Session)
122 // helpers about SALOME::GenericObj
123 #include <SALOMEDS_wrap.hxx>
124 #include <SALOMEDS_Attributes_wrap.hxx>
125 #include <GEOM_wrap.hxx>
133 using SMESH::TPythonDump;
136 #define NUM_TMP_FILES 2
139 static int MYDEBUG = 0;
141 static int MYDEBUG = 0;
144 // Static variables definition
145 GEOM::GEOM_Gen_var SMESH_Gen_i::myGeomGen = GEOM::GEOM_Gen::_nil();
146 CORBA::ORB_var SMESH_Gen_i::myOrb;
147 PortableServer::POA_var SMESH_Gen_i::myPoa;
148 SALOME_NamingService* SMESH_Gen_i::myNS = NULL;
149 SALOME_LifeCycleCORBA* SMESH_Gen_i::myLCC = NULL;
150 SMESH_Gen_i* SMESH_Gen_i::mySMESHGen = NULL;
153 const int nbElemPerDiagonal = 10;
155 //=============================================================================
157 * GetServant [ static ]
159 * Get servant of the CORBA object
161 //=============================================================================
163 PortableServer::ServantBase_var SMESH_Gen_i::GetServant( CORBA::Object_ptr theObject )
165 if( CORBA::is_nil( theObject ) || CORBA::is_nil( GetPOA() ) )
168 PortableServer::Servant aServant = GetPOA()->reference_to_servant( theObject );
172 INFOS( "GetServant - Unknown exception was caught!!!" );
177 //=============================================================================
179 * SObjectToObject [ static ]
181 * Get CORBA object corresponding to the SALOMEDS::SObject
183 //=============================================================================
185 CORBA::Object_var SMESH_Gen_i::SObjectToObject( SALOMEDS::SObject_ptr theSObject )
187 SALOMEDS::GenericAttribute_wrap anAttr;
188 CORBA::Object_var anObj;
189 if ( !theSObject->_is_nil() ) {
191 if( theSObject->FindAttribute( anAttr.inout(), "AttributeIOR" ) ) {
192 SALOMEDS::AttributeIOR_wrap anIOR = anAttr;
193 CORBA::String_var aValue = anIOR->Value();
194 if( strcmp( aValue, "" ) != 0 )
195 anObj = GetORB()->string_to_object( aValue );
199 INFOS( "SObjectToObject - Unknown exception was caught!!!" );
205 //=============================================================================
209 * Get SALOME_NamingService object
211 //=============================================================================
213 SALOME_NamingService* SMESH_Gen_i::GetNS()
215 if ( myNS == NULL ) {
216 myNS = SINGLETON_<SALOME_NamingService>::Instance();
217 ASSERT(SINGLETON_<SALOME_NamingService>::IsAlreadyExisting());
218 myNS->init_orb( GetORB() );
223 //=============================================================================
227 * Get SALOME_LifeCycleCORBA object
229 //=============================================================================
230 SALOME_LifeCycleCORBA* SMESH_Gen_i::GetLCC() {
231 if ( myLCC == NULL ) {
232 myLCC = new SALOME_LifeCycleCORBA( GetNS() );
238 //=============================================================================
240 * GetGeomEngine [ static ]
242 * Get GEOM::GEOM_Gen reference
244 //=============================================================================
245 GEOM::GEOM_Gen_var SMESH_Gen_i::GetGeomEngine() {
246 //CCRT GEOM::GEOM_Gen_var aGeomEngine =
247 //CCRT GEOM::GEOM_Gen::_narrow( GetLCC()->FindOrLoad_Component("FactoryServer","GEOM") );
248 //CCRT return aGeomEngine._retn();
249 if(CORBA::is_nil(myGeomGen))
251 Engines::EngineComponent_ptr temp=GetLCC()->FindOrLoad_Component("FactoryServer","GEOM");
252 myGeomGen=GEOM::GEOM_Gen::_narrow(temp);
257 //=============================================================================
259 * SMESH_Gen_i::SMESH_Gen_i
261 * Default constructor: not for use
263 //=============================================================================
265 SMESH_Gen_i::SMESH_Gen_i()
267 INFOS( "SMESH_Gen_i::SMESH_Gen_i : default constructor" );
270 //=============================================================================
272 * SMESH_Gen_i::SMESH_Gen_i
274 * Standard constructor, used with Container
276 //=============================================================================
278 SMESH_Gen_i::SMESH_Gen_i( CORBA::ORB_ptr orb,
279 PortableServer::POA_ptr poa,
280 PortableServer::ObjectId* contId,
281 const char* instanceName,
282 const char* interfaceName )
283 : Engines_Component_i( orb, poa, contId, instanceName, interfaceName )
285 MESSAGE( "SMESH_Gen_i::SMESH_Gen_i : standard constructor" );
287 myOrb = CORBA::ORB::_duplicate(orb);
288 myPoa = PortableServer::POA::_duplicate(poa);
291 _id = myPoa->activate_object( _thisObj );
293 myIsEmbeddedMode = false;
294 myShapeReader = NULL; // shape reader
296 myIsHistoricalPythonDump = true;
297 myToForgetMeshDataOnHypModif = false;
299 myImportedStudyChanged = true;
300 myImportedStudyId = 0;
302 // set it in standalone mode only
303 //OSD::SetSignal( true );
305 // 0020605: EDF 1190 SMESH: Display performance. 80 seconds for 52000 cells.
306 // find out mode (embedded or standalone) here else
307 // meshes created before calling SMESH_Client::GetSMESHGen(), which calls
308 // SMESH_Gen_i::SetEmbeddedMode(), have wrong IsEmbeddedMode flag
309 if ( SALOME_NamingService* ns = GetNS() )
311 CORBA::Object_var obj = ns->Resolve( "/Kernel/Session" );
312 SALOME::Session_var session = SALOME::Session::_narrow( obj ) ;
313 if ( !session->_is_nil() )
315 CORBA::String_var str_host = session->getHostname();
316 CORBA::Long s_pid = session->getPID();
317 string my_host = Kernel_Utils::GetHostname();
319 long my_pid = (long)_getpid();
321 long my_pid = (long) getpid();
323 SetEmbeddedMode( s_pid == my_pid && my_host == str_host.in() );
328 //=============================================================================
330 * SMESH_Gen_i::~SMESH_Gen_i
334 //=============================================================================
336 SMESH_Gen_i::~SMESH_Gen_i()
338 MESSAGE( "SMESH_Gen_i::~SMESH_Gen_i" );
340 // delete hypothesis creators
341 map<string, GenericHypothesisCreator_i*>::iterator itHyp;
342 for (itHyp = myHypCreatorMap.begin(); itHyp != myHypCreatorMap.end(); itHyp++)
344 delete (*itHyp).second;
346 myHypCreatorMap.clear();
348 // Clear study contexts data
349 map<int, StudyContext*>::iterator it;
350 for ( it = myStudyContextMap.begin(); it != myStudyContextMap.end(); ++it ) {
353 myStudyContextMap.clear();
354 // delete shape reader
356 delete myShapeReader;
358 //=============================================================================
360 * SMESH_Gen_i::getHypothesisCreator
362 * Get hypothesis creator
364 //=============================================================================
365 GenericHypothesisCreator_i* SMESH_Gen_i::getHypothesisCreator(const char* theHypName,
366 const char* theLibName,
367 std::string& thePlatformLibName)
368 throw (SALOME::SALOME_Exception)
370 std::string aPlatformLibName;
371 /* It's Need to tranlate lib name for WIN32 or X platform */
372 if ( theLibName && theLibName[0] != '\0' )
374 int libNameLen = strlen(theLibName);
375 //check for old format "libXXXXXXX.so"
376 if (libNameLen > 7 &&
377 !strncmp( theLibName, "lib", 3 ) &&
378 !strcmp( theLibName+libNameLen-3, ".so" ))
382 aPlatformLibName = std::string( theLibName+3, libNameLen-6 ) + ".dll";
384 aPlatformLibName = theLibName;
389 //try to use new format
391 aPlatformLibName = theLibName;
392 aPlatformLibName += ".dll";
394 aPlatformLibName = "lib" + std::string( theLibName ) + ".so";
398 thePlatformLibName = aPlatformLibName;
400 Unexpect aCatch(SALOME_SalomeException);
401 if(MYDEBUG) MESSAGE( "Create Hypothesis <" << theHypName << "> from " << aPlatformLibName);
403 typedef GenericHypothesisCreator_i* (*GetHypothesisCreator)(const char* );
404 GenericHypothesisCreator_i* aCreator;
407 // check, if creator for this hypothesis type already exists
408 if (myHypCreatorMap.find(string(theHypName)) == myHypCreatorMap.end())
410 // load plugin library
411 if(MYDEBUG) MESSAGE("Loading server meshers plugin library ...");
412 LibHandle libHandle = LoadLib( aPlatformLibName.c_str() );
415 // report any error, if occured
417 const char* anError = dlerror();
418 throw(SALOME_Exception(anError));
420 throw(SALOME_Exception(LOCALIZED( "Can't load server meshers plugin library" )));
424 // get method, returning hypothesis creator
425 if(MYDEBUG) MESSAGE("Find GetHypothesisCreator() method ...");
426 GetHypothesisCreator procHandle =
427 (GetHypothesisCreator)GetProc( libHandle, "GetHypothesisCreator" );
430 throw(SALOME_Exception(LOCALIZED("bad hypothesis plugin library")));
431 UnLoadLib(libHandle);
434 // get hypothesis creator
435 if(MYDEBUG) MESSAGE("Get Hypothesis Creator for " << theHypName);
436 aCreator = procHandle(theHypName);
439 throw(SALOME_Exception(LOCALIZED("no such a hypothesis in this plugin")));
441 // map hypothesis creator to a hypothesis name
442 myHypCreatorMap[string(theHypName)] = aCreator;
447 return myHypCreatorMap[string(theHypName)];
450 catch (SALOME_Exception& S_ex)
452 THROW_SALOME_CORBA_EXCEPTION(S_ex.what(), SALOME::BAD_PARAM);
457 //=============================================================================
459 * SMESH_Gen_i::createHypothesis
461 * Create hypothesis of given type
463 //=============================================================================
464 SMESH::SMESH_Hypothesis_ptr SMESH_Gen_i::createHypothesis(const char* theHypName,
465 const char* theLibName)
467 SMESH_Hypothesis_i* myHypothesis_i = 0;
468 SMESH::SMESH_Hypothesis_var hypothesis_i;
469 std::string aPlatformLibName;
470 GenericHypothesisCreator_i* aCreator =
471 getHypothesisCreator(theHypName, theLibName, aPlatformLibName);
473 // create a new hypothesis object, store its ref. in studyContext
474 myHypothesis_i = aCreator->Create(myPoa, GetCurrentStudyID(), &myGen);
477 myHypothesis_i->SetLibName( aPlatformLibName.c_str() ); // for persistency assurance
478 CORBA::String_var hypName = myHypothesis_i->GetName();
479 myHypCreatorMap[ hypName.in() ] = aCreator;
481 // activate the CORBA servant of hypothesis
482 hypothesis_i = myHypothesis_i->_this();
483 int nextId = RegisterObject( hypothesis_i );
484 if(MYDEBUG) { MESSAGE( "Add hypo to map with id = "<< nextId ); }
485 else { nextId = 0; } // avoid "unused variable" warning in release mode
487 return hypothesis_i._retn();
490 //=============================================================================
492 * SMESH_Gen_i::createMesh
494 * Create empty mesh on shape
496 //=============================================================================
497 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::createMesh()
498 throw ( SALOME::SALOME_Exception )
500 Unexpect aCatch(SALOME_SalomeException);
501 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::createMesh" );
503 // Get or create the GEOM_Client instance
505 // create a new mesh object servant, store it in a map in study context
506 SMESH_Mesh_i* meshServant = new SMESH_Mesh_i( GetPOA(), this, GetCurrentStudyID() );
507 // create a new mesh object
508 MESSAGE("myIsEmbeddedMode " << myIsEmbeddedMode);
509 meshServant->SetImpl( myGen.CreateMesh( GetCurrentStudyID(), myIsEmbeddedMode ));
511 // activate the CORBA servant of Mesh
512 SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( meshServant->_this() );
513 int nextId = RegisterObject( mesh );
514 if(MYDEBUG) { MESSAGE( "Add mesh to map with id = "<< nextId); }
515 else { nextId = 0; } // avoid "unused variable" warning in release mode
518 catch (SALOME_Exception& S_ex) {
519 THROW_SALOME_CORBA_EXCEPTION( S_ex.what(), SALOME::BAD_PARAM );
521 return SMESH::SMESH_Mesh::_nil();
524 //=============================================================================
526 * SMESH_Gen_i::GetShapeReader
530 //=============================================================================
531 GEOM_Client* SMESH_Gen_i::GetShapeReader()
533 // create shape reader if necessary
534 if ( !myShapeReader )
535 myShapeReader = new GEOM_Client(GetContainerRef());
536 ASSERT( myShapeReader );
537 return myShapeReader;
540 //=============================================================================
542 * SMESH_Gen_i::SetGeomEngine
544 * Set GEOM::GEOM_Gen reference
546 //=============================================================================
547 //GEOM::GEOM_Gen_ptr SMESH_Gen_i::SetGeomEngine( const char* containerLoc )
548 void SMESH_Gen_i::SetGeomEngine( GEOM::GEOM_Gen_ptr geomcompo )
550 //Engines::Component_ptr temp=GetLCC()->FindOrLoad_Component(containerLoc,"GEOM");
551 //myGeomGen=GEOM::GEOM_Gen::_narrow(temp);
552 myGeomGen=GEOM::GEOM_Gen::_duplicate(geomcompo);
556 //=============================================================================
558 * SMESH_Gen_i::SetEmbeddedMode
562 //=============================================================================
564 void SMESH_Gen_i::SetEmbeddedMode( CORBA::Boolean theMode )
566 myIsEmbeddedMode = theMode;
567 MESSAGE("myIsEmbeddedMode " << myIsEmbeddedMode);
569 if ( !myIsEmbeddedMode ) {
570 //PAL10867: disable signals catching with "noexcepthandler" option
571 char* envNoCatchSignals = getenv("NOT_INTERCEPT_SIGNALS");
572 if (!envNoCatchSignals || !atoi(envNoCatchSignals))
577 char* envDisableFPE = getenv("DISABLE_FPE");
578 if (envDisableFPE && atoi(envDisableFPE))
583 OSD::SetSignal( raiseFPE );
585 // else OSD::SetSignal() is called in GUI
589 //=============================================================================
591 * SMESH_Gen_i::IsEmbeddedMode
595 //=============================================================================
597 CORBA::Boolean SMESH_Gen_i::IsEmbeddedMode()
599 return myIsEmbeddedMode;
602 //=============================================================================
604 * SMESH_Gen_i::SetCurrentStudy
608 //=============================================================================
610 void SMESH_Gen_i::SetCurrentStudy( SALOMEDS::Study_ptr theStudy )
612 setCurrentStudy( theStudy );
615 void SMESH_Gen_i::setCurrentStudy( SALOMEDS::Study_ptr theStudy,
616 bool theStudyIsBeingClosed)
618 int curStudyId = GetCurrentStudyID();
619 myCurrentStudy = SALOMEDS::Study::_duplicate( theStudy );
620 // create study context, if it doesn't exist and set current study
621 int studyId = GetCurrentStudyID();
622 if ( myStudyContextMap.find( studyId ) == myStudyContextMap.end() )
623 myStudyContextMap[ studyId ] = new StudyContext;
625 // myCurrentStudy may be nil
626 if ( !theStudyIsBeingClosed && !CORBA::is_nil( myCurrentStudy ) ) {
627 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
628 SALOMEDS::SComponent_wrap GEOM_var = myCurrentStudy->FindComponent( "GEOM" );
629 if( !GEOM_var->_is_nil() )
630 aStudyBuilder->LoadWith( GEOM_var, GetGeomEngine() );
631 // NPAL16168, issue 0020210
632 // Let meshes update their data depending on GEOM groups that could change
633 if ( curStudyId != studyId )
635 CORBA::String_var compDataType = ComponentDataType();
636 SALOMEDS::SComponent_wrap me = myCurrentStudy->FindComponent( compDataType.in() );
637 if ( !me->_is_nil() ) {
638 SALOMEDS::ChildIterator_wrap anIter = myCurrentStudy->NewChildIterator( me );
639 for ( ; anIter->More(); anIter->Next() ) {
640 SALOMEDS::SObject_wrap so = anIter->Value();
641 CORBA::Object_var ior = SObjectToObject( so );
642 if ( SMESH_Mesh_i* mesh = SMESH::DownCast<SMESH_Mesh_i*>( ior ))
643 mesh->CheckGeomModif();
650 //=============================================================================
652 * SMESH_Gen_i::GetCurrentStudy
656 //=============================================================================
658 SALOMEDS::Study_ptr SMESH_Gen_i::GetCurrentStudy()
660 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetCurrentStudy: study Id = " << GetCurrentStudyID() );
661 if ( GetCurrentStudyID() < 0 )
662 return SALOMEDS::Study::_nil();
663 return SALOMEDS::Study::_duplicate( myCurrentStudy );
666 //=============================================================================
668 * SMESH_Gen_i::GetCurrentStudyContext
670 * Get current study context
672 //=============================================================================
673 StudyContext* SMESH_Gen_i::GetCurrentStudyContext()
675 if ( !CORBA::is_nil( myCurrentStudy ) &&
676 myStudyContextMap.find( GetCurrentStudyID() ) != myStudyContextMap.end() )
677 return myStudyContextMap[ myCurrentStudy->StudyId() ];
682 //=============================================================================
684 * SMESH_Gen_i::CreateHypothesis
686 * Create hypothesis/algorothm of given type and publish it in the study
688 //=============================================================================
690 SMESH::SMESH_Hypothesis_ptr SMESH_Gen_i::CreateHypothesis( const char* theHypName,
691 const char* theLibName )
692 throw ( SALOME::SALOME_Exception )
694 Unexpect aCatch(SALOME_SalomeException);
695 // Create hypothesis/algorithm
696 SMESH::SMESH_Hypothesis_var hyp = this->createHypothesis( theHypName, theLibName );
698 // Publish hypothesis/algorithm in the study
699 if ( CanPublishInStudy( hyp ) ) {
700 SALOMEDS::SObject_wrap aSO = PublishHypothesis( myCurrentStudy, hyp );
701 if ( !aSO->_is_nil() ) {
702 // Update Python script
703 TPythonDump() << aSO << " = " << this << ".CreateHypothesis('"
704 << theHypName << "', '" << theLibName << "')";
711 //================================================================================
713 * \brief Return a hypothesis holding parameter values corresponding either to the mesh
714 * existing on the given geometry or to size of the geometry.
715 * \param theHypType - hypothesis type name
716 * \param theLibName - plugin library name
717 * \param theMesh - The mesh of interest
718 * \param theGeom - The shape to get parameter values from
719 * \retval SMESH::SMESH_Hypothesis_ptr - The returned hypothesis may be the one existing
720 * in a study and used to compute the mesh, or a temporary one created just to pass
723 //================================================================================
725 SMESH::SMESH_Hypothesis_ptr
726 SMESH_Gen_i::GetHypothesisParameterValues (const char* theHypType,
727 const char* theLibName,
728 SMESH::SMESH_Mesh_ptr theMesh,
729 GEOM::GEOM_Object_ptr theGeom,
730 CORBA::Boolean byMesh)
731 throw ( SALOME::SALOME_Exception )
733 Unexpect aCatch(SALOME_SalomeException);
734 if ( byMesh && CORBA::is_nil( theMesh ) )
735 return SMESH::SMESH_Hypothesis::_nil();
736 if ( byMesh && CORBA::is_nil( theGeom ) )
737 return SMESH::SMESH_Hypothesis::_nil();
739 // -----------------------------------------------
740 // find hypothesis used to mesh theGeom
741 // -----------------------------------------------
743 // get mesh and shape
744 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
745 TopoDS_Shape shape = GeomObjectToShape( theGeom );
746 if ( byMesh && ( !meshServant || meshServant->NbNodes()==0 || shape.IsNull() ))
747 return SMESH::SMESH_Hypothesis::_nil();
748 ::SMESH_Mesh* mesh = meshServant ? &meshServant->GetImpl() : (::SMESH_Mesh*)0;
750 // create a temporary hypothesis to know its dimention
751 SMESH::SMESH_Hypothesis_var tmpHyp = this->createHypothesis( theHypType, theLibName );
752 SMESH_Hypothesis_i* hypServant = SMESH::DownCast<SMESH_Hypothesis_i*>( tmpHyp );
754 return SMESH::SMESH_Hypothesis::_nil();
755 ::SMESH_Hypothesis* hyp = hypServant->GetImpl();
758 // look for a hypothesis of theHypType used to mesh the shape
759 if ( myGen.GetShapeDim( shape ) == hyp->GetDim() )
762 SMESH::ListOfHypothesis_var aHypList = theMesh->GetHypothesisList( theGeom );
763 int nbLocalHyps = aHypList->length();
764 for ( int i = 0; i < nbLocalHyps; i++ ) {
765 CORBA::String_var hypName = aHypList[i]->GetName();
766 if ( strcmp( theHypType, hypName.in() ) == 0 ) // FOUND local!
767 return SMESH::SMESH_Hypothesis::_duplicate( aHypList[i] );
769 // check super shapes
770 TopTools_ListIteratorOfListOfShape itShape( mesh->GetAncestors( shape ));
771 while ( nbLocalHyps == 0 && itShape.More() ) {
772 GEOM::GEOM_Object_ptr geomObj = ShapeToGeomObject( itShape.Value() );
773 if ( ! CORBA::is_nil( geomObj )) {
774 SMESH::ListOfHypothesis_var aHypList = theMesh->GetHypothesisList( geomObj );
775 nbLocalHyps = aHypList->length();
776 for ( int i = 0; i < nbLocalHyps; i++ )
777 if ( strcmp( theHypType, aHypList[i]->GetName() ) == 0 ) // FOUND global!
778 return SMESH::SMESH_Hypothesis::_duplicate( aHypList[i] );
784 // let the temporary hypothesis find out somehow parameter values by mesh
785 if ( hyp->SetParametersByMesh( mesh, shape ))
786 return SMESH::SMESH_Hypothesis::_duplicate( tmpHyp );
791 diagonal = mesh->GetShapeDiagonalSize();
793 diagonal = ::SMESH_Mesh::GetShapeDiagonalSize( shape );
794 ::SMESH_Hypothesis::TDefaults dflts;
795 dflts._elemLength = diagonal / myGen.GetBoundaryBoxSegmentation();
796 dflts._nbSegments = myGen.GetDefaultNbSegments();
797 dflts._shape = &shape;
798 // let the temporary hypothesis initialize it's values
799 if ( hyp->SetParametersByDefaults( dflts, mesh ))
800 return SMESH::SMESH_Hypothesis::_duplicate( tmpHyp );
803 return SMESH::SMESH_Hypothesis::_nil();
806 //=============================================================================
808 * Returns \c True if a hypothesis is assigned to a sole sub-mesh in a current Study
809 * \param [in] theHyp - the hypothesis of interest
810 * \param [out] theMesh - the sole mesh using \a theHyp
811 * \param [out] theShape - the sole geometry \a theHyp is assigned to
812 * \return boolean - \c True if \a theMesh and \a theShape are sole using \a theHyp
814 * If two meshes on same shape have theHyp assigned to the same sub-shape, they are
815 * considered as SAME sub-mesh => result is \c true.
816 * This method ids used to initialize SMESHGUI_GenericHypothesisCreator with
817 * a shape to which an hyp being edited is assigned.
819 //=============================================================================
821 CORBA::Boolean SMESH_Gen_i::GetSoleSubMeshUsingHyp( SMESH::SMESH_Hypothesis_ptr theHyp,
822 SMESH::SMESH_Mesh_out theMesh,
823 GEOM::GEOM_Object_out theShape)
825 if ( GetCurrentStudyID() < 0 || CORBA::is_nil( theHyp ))
828 // get Mesh component SO
829 CORBA::String_var compDataType = ComponentDataType();
830 SALOMEDS::SComponent_wrap comp = myCurrentStudy->FindComponent( compDataType.in() );
831 if ( CORBA::is_nil( comp ))
834 // look for child SO of meshes
835 SMESH::SMESH_Mesh_var foundMesh;
836 TopoDS_Shape foundShape;
838 SALOMEDS::ChildIterator_wrap meshIter = myCurrentStudy->NewChildIterator( comp );
839 for ( ; meshIter->More() && isSole; meshIter->Next() )
841 SALOMEDS::SObject_wrap curSO = meshIter->Value();
842 CORBA::Object_var obj = SObjectToObject( curSO );
843 SMESH_Mesh_i* mesh_i = SMESH::DownCast< SMESH_Mesh_i* >( obj );
847 // look for a sole shape where theHyp is assigned
848 bool isHypFound = false;
849 const ShapeToHypothesis & s2hyps = mesh_i->GetImpl().GetMeshDS()->GetHypotheses();
850 ShapeToHypothesis::Iterator s2hypsIt( s2hyps );
851 for ( ; s2hypsIt.More() && isSole; s2hypsIt.Next() )
853 const THypList& hyps = s2hypsIt.Value();
854 THypList::const_iterator h = hyps.begin();
855 for ( ; h != hyps.end(); ++h )
856 if ( (*h)->GetID() == theHyp->GetId() )
858 if ( h != hyps.end()) // theHyp found
861 if ( ! foundShape.IsNull() &&
862 ! foundShape.IsSame( s2hypsIt.Key() )) // not a sole sub-shape
864 foundShape.Nullify();
868 foundShape = s2hypsIt.Key();
870 } // loop on assigned hyps
872 if ( isHypFound && !foundShape.IsNull() ) // a mesh using theHyp is found
874 if ( !foundMesh->_is_nil() ) // not a sole mesh
876 GEOM::GEOM_Object_var s1 = mesh_i ->GetShapeToMesh();
877 GEOM::GEOM_Object_var s2 = foundMesh->GetShapeToMesh();
878 if ( ! ( isSole = s1->IsSame( s2 )))
881 foundMesh = SMESH::SMESH_Mesh::_narrow( obj );
887 ! foundMesh->_is_nil() &&
888 ! foundShape.IsNull() )
890 theMesh = foundMesh._retn();
891 theShape = ShapeToGeomObject( foundShape );
892 return ( !theMesh->_is_nil() && !theShape->_is_nil() );
897 //=============================================================================
899 * Sets number of segments per diagonal of boundary box of geometry by which
900 * default segment length of appropriate 1D hypotheses is defined
902 //=============================================================================
904 void SMESH_Gen_i::SetBoundaryBoxSegmentation( CORBA::Long theNbSegments )
905 throw ( SALOME::SALOME_Exception )
907 if ( theNbSegments > 0 )
908 myGen.SetBoundaryBoxSegmentation( int( theNbSegments ));
910 THROW_SALOME_CORBA_EXCEPTION( "non-positive number of segments", SALOME::BAD_PARAM );
912 //=============================================================================
914 * \brief Sets default number of segments per edge
916 //=============================================================================
917 void SMESH_Gen_i::SetDefaultNbSegments(CORBA::Long theNbSegments)
918 throw ( SALOME::SALOME_Exception )
920 if ( theNbSegments > 0 )
921 myGen.SetDefaultNbSegments( int(theNbSegments) );
923 THROW_SALOME_CORBA_EXCEPTION( "non-positive number of segments", SALOME::BAD_PARAM );
926 //=============================================================================
930 //=============================================================================
932 void SMESH_Gen_i::SetOption(const char* name, const char* value)
934 if ( name && value && strlen( value ) > 0 )
937 if ( strcmp(name, "historical_python_dump") == 0 )
939 myIsHistoricalPythonDump = ( value[0] == '1' || toupper(value[0]) == 'T' ); // 1 || true
940 msgToGUI = "preferences/SMESH/historical_python_dump/";
941 msgToGUI += myIsHistoricalPythonDump ? "true" : "false";
943 else if ( strcmp(name, "forget_mesh_on_hyp_modif") == 0 )
945 myToForgetMeshDataOnHypModif = ( value[0] == '1' || toupper(value[0]) == 'T' ); // 1 || true
946 msgToGUI = "preferences/SMESH/forget_mesh_on_hyp_modif/";
947 msgToGUI += myToForgetMeshDataOnHypModif ? "true" : "false";
950 // update preferences in case if SetOption() is invoked from python console
951 if ( !msgToGUI.empty() )
953 CORBA::Object_var obj = SMESH_Gen_i::GetNS()->Resolve( "/Kernel/Session" );
954 SALOME::Session_var session = SALOME::Session::_narrow( obj );
955 if ( !CORBA::is_nil( session ) )
956 session->emitMessageOneWay(msgToGUI.c_str());
961 //=============================================================================
963 Return an option value
965 //=============================================================================
967 char* SMESH_Gen_i::GetOption(const char* name)
971 if ( strcmp(name, "historical_python_dump") == 0 )
973 return CORBA::string_dup( myIsHistoricalPythonDump ? "true" : "false" );
975 if ( strcmp(name, "forget_mesh_on_hyp_modif") == 0 )
977 return CORBA::string_dup( myToForgetMeshDataOnHypModif ? "true" : "false" );
980 return CORBA::string_dup( "" );
983 //=============================================================================
985 * SMESH_Gen_i::CreateMesh
987 * Create empty mesh on a shape and publish it in the study
989 //=============================================================================
991 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMesh( GEOM::GEOM_Object_ptr theShapeObject )
992 throw ( SALOME::SALOME_Exception )
994 Unexpect aCatch(SALOME_SalomeException);
995 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::CreateMesh" );
997 SMESH::SMESH_Mesh_var mesh = this->createMesh();
999 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( mesh );
1000 ASSERT( meshServant );
1001 meshServant->SetShape( theShapeObject );
1003 // publish mesh in the study
1004 if ( CanPublishInStudy( mesh ) ) {
1005 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1006 aStudyBuilder->NewCommand(); // There is a transaction
1007 SALOMEDS::SObject_wrap aSO = PublishMesh( myCurrentStudy, mesh.in() );
1008 aStudyBuilder->CommitCommand();
1009 if ( !aSO->_is_nil() ) {
1010 // Update Python script
1011 TPythonDump() << aSO << " = " << this << ".CreateMesh(" << theShapeObject << ")";
1015 return mesh._retn();
1018 //=============================================================================
1020 * SMESH_Gen_i::CreateEmptyMesh
1024 //=============================================================================
1026 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateEmptyMesh()
1027 throw ( SALOME::SALOME_Exception )
1029 Unexpect aCatch(SALOME_SalomeException);
1030 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::CreateMesh" );
1032 SMESH::SMESH_Mesh_var mesh = this->createMesh();
1034 // publish mesh in the study
1035 if ( CanPublishInStudy( mesh ) ) {
1036 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1037 aStudyBuilder->NewCommand(); // There is a transaction
1038 SALOMEDS::SObject_wrap aSO = PublishMesh( myCurrentStudy, mesh.in() );
1039 aStudyBuilder->CommitCommand();
1040 if ( !aSO->_is_nil() ) {
1041 // Update Python script
1042 TPythonDump() << aSO << " = " << this << ".CreateEmptyMesh()";
1046 return mesh._retn();
1051 //================================================================================
1053 * \brief Throws an exception in case if the file can't be read
1055 //================================================================================
1057 void checkFileReadable( const char* theFileName ) throw ( SALOME::SALOME_Exception )
1059 SMESH_File f ( theFileName );
1062 if ( !f.error().empty() )
1063 THROW_SALOME_CORBA_EXCEPTION( f.error().c_str(), SALOME::BAD_PARAM);
1065 THROW_SALOME_CORBA_EXCEPTION
1066 (( SMESH_Comment("Can't open for reading the file ") << theFileName ).c_str(),
1067 SALOME::BAD_PARAM );
1072 //=============================================================================
1074 * SMESH_Gen_i::CreateMeshFromUNV
1076 * Create mesh and import data from UNV file
1078 //=============================================================================
1080 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMeshesFromUNV( const char* theFileName )
1081 throw ( SALOME::SALOME_Exception )
1083 Unexpect aCatch(SALOME_SalomeException);
1085 checkFileReadable( theFileName );
1087 SMESH::SMESH_Mesh_var aMesh = createMesh();
1089 // publish mesh in the study
1090 if ( CanPublishInStudy( aMesh ) ) {
1091 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1092 aStudyBuilder->NewCommand(); // There is a transaction
1093 SALOMEDS::SObject_wrap aSO = PublishMesh( myCurrentStudy, aMesh.in(), aFileName.c_str() );
1094 aStudyBuilder->CommitCommand();
1095 if ( !aSO->_is_nil() ) {
1096 // Update Python script
1097 TPythonDump() << aSO << " = " << this << ".CreateMeshesFromUNV(r'" << theFileName << "')";
1101 SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
1103 aServant->ImportUNVFile( theFileName );
1105 // Dump creation of groups
1106 SMESH::ListOfGroups_var groups = aServant->GetGroups();
1108 aServant->GetImpl().GetMeshDS()->Modified();
1109 return aMesh._retn();
1112 //=============================================================================
1114 * SMESH_Gen_i::CreateMeshFromMED
1116 * Create mesh and import data from MED file
1118 //=============================================================================
1120 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromMEDorSAUV( const char* theFileName,
1121 SMESH::DriverMED_ReadStatus& theStatus,
1122 const char* theCommandNameForPython,
1123 const char* theFileNameForPython)
1126 char bname[ _MAX_FNAME ];
1127 _splitpath( theFileNameForPython, NULL, NULL, bname, NULL );
1128 string aFileName = bname;
1130 string aFileName = basename( theFileNameForPython );
1132 // Retrieve mesh names from the file
1133 DriverMED_R_SMESHDS_Mesh myReader;
1134 myReader.SetFile( theFileName );
1135 myReader.SetMeshId( -1 );
1136 Driver_Mesh::Status aStatus;
1137 list<string> aNames = myReader.GetMeshNames(aStatus);
1138 SMESH::mesh_array_var aResult = new SMESH::mesh_array();
1139 theStatus = (SMESH::DriverMED_ReadStatus)aStatus;
1141 { // open a new scope to make aPythonDump die before PythonDump in SMESH_Mesh::GetGroups()
1144 TPythonDump aPythonDump;
1145 aPythonDump << "([";
1147 if (theStatus == SMESH::DRS_OK) {
1148 SALOMEDS::StudyBuilder_var aStudyBuilder;
1149 if ( GetCurrentStudyID() > -1 )
1151 aStudyBuilder = myCurrentStudy->NewBuilder();
1152 aStudyBuilder->NewCommand(); // There is a transaction
1154 aResult->length( aNames.size() );
1157 // Iterate through all meshes and create mesh objects
1158 for ( list<string>::iterator it = aNames.begin(); it != aNames.end(); it++ )
1161 if (i > 0) aPythonDump << ", ";
1164 SMESH::SMESH_Mesh_var mesh = createMesh();
1166 // publish mesh in the study
1167 SALOMEDS::SObject_wrap aSO;
1168 if ( CanPublishInStudy( mesh ) )
1169 // little trick: for MED file theFileName and theFileNameForPython are the same, but they are different for SAUV
1170 // - as names of meshes are stored in MED file, we use them for data publishing
1171 // - as mesh name is not stored in UNV file, we use file name as name of mesh when publishing data
1172 aSO = PublishMesh( myCurrentStudy, mesh.in(), ( theFileName == theFileNameForPython ) ? (*it).c_str() : aFileName.c_str() );
1173 if ( !aSO->_is_nil() ) {
1178 aPythonDump << "mesh_" << i;
1181 // Read mesh data (groups are published automatically by ImportMEDFile())
1182 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( mesh ).in() );
1183 ASSERT( meshServant );
1184 SMESH::DriverMED_ReadStatus status1 =
1185 meshServant->ImportMEDFile( theFileName, (*it).c_str() );
1186 if (status1 > theStatus)
1187 theStatus = status1;
1189 aResult[i++] = SMESH::SMESH_Mesh::_duplicate( mesh );
1190 meshServant->GetImpl().GetMeshDS()->Modified();
1192 if ( !aStudyBuilder->_is_nil() )
1193 aStudyBuilder->CommitCommand();
1196 // Update Python script
1197 aPythonDump << "], status) = " << this << "." << theCommandNameForPython << "(r'" << theFileNameForPython << "')";
1199 // Dump creation of groups
1200 for ( int i = 0; i < aResult->length(); ++i )
1201 SMESH::ListOfGroups_var groups = aResult[ i ]->GetGroups();
1203 return aResult._retn();
1206 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromMED( const char* theFileName,
1207 SMESH::DriverMED_ReadStatus& theStatus)
1208 throw ( SALOME::SALOME_Exception )
1210 Unexpect aCatch(SALOME_SalomeException);
1211 checkFileReadable( theFileName );
1213 SMESH::mesh_array* result = CreateMeshesFromMEDorSAUV(theFileName, theStatus, "CreateMeshesFromMED", theFileName);
1217 //=============================================================================
1219 * SMESH_Gen_i::CreateMeshFromSAUV
1221 * Create mesh and import data from SAUV file
1223 //=============================================================================
1225 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromSAUV( const char* theFileName,
1226 SMESH::DriverMED_ReadStatus& theStatus)
1227 throw ( SALOME::SALOME_Exception )
1229 Unexpect aCatch(SALOME_SalomeException);
1230 checkFileReadable( theFileName );
1232 std::string sauvfilename(theFileName);
1233 std::string medfilename(theFileName);
1234 medfilename += ".med";
1237 cmd = "%PYTHONBIN% ";
1242 cmd += "from medutilities import convert ; convert(r'" + sauvfilename + "', 'GIBI', 'MED', 1, r'" + medfilename + "')";
1244 system(cmd.c_str());
1245 SMESH::mesh_array* result = CreateMeshesFromMEDorSAUV(medfilename.c_str(), theStatus, "CreateMeshesFromSAUV", sauvfilename.c_str());
1247 cmd = "%PYTHONBIN% ";
1252 cmd += "from medutilities import my_remove ; my_remove(r'" + medfilename + "')";
1254 system(cmd.c_str());
1258 //=============================================================================
1260 * SMESH_Gen_i::CreateMeshFromSTL
1262 * Create mesh and import data from STL file
1264 //=============================================================================
1266 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMeshesFromSTL( const char* theFileName )
1267 throw ( SALOME::SALOME_Exception )
1269 Unexpect aCatch(SALOME_SalomeException);
1270 checkFileReadable( theFileName );
1272 SMESH::SMESH_Mesh_var aMesh = createMesh();
1275 char bname[ _MAX_FNAME ];
1276 _splitpath( theFileName, NULL, NULL, bname, NULL );
1277 string aFileName = bname;
1279 string aFileName = basename( theFileName );
1281 // publish mesh in the study
1282 if ( CanPublishInStudy( aMesh ) ) {
1283 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1284 aStudyBuilder->NewCommand(); // There is a transaction
1285 SALOMEDS::SObject_wrap aSO = PublishInStudy
1286 ( myCurrentStudy, SALOMEDS::SObject::_nil(), aMesh.in(), aFileName.c_str() );
1287 aStudyBuilder->CommitCommand();
1288 if ( !aSO->_is_nil() ) {
1289 // Update Python script
1290 TPythonDump() << aSO << " = " << this << ".CreateMeshesFromSTL(r'" << theFileName << "')";
1294 SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
1296 aServant->ImportSTLFile( theFileName );
1297 aServant->GetImpl().GetMeshDS()->Modified();
1298 return aMesh._retn();
1301 //================================================================================
1303 * \brief Create meshes and import data from the CGSN file
1305 //================================================================================
1307 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromCGNS( const char* theFileName,
1308 SMESH::DriverMED_ReadStatus& theStatus)
1309 throw ( SALOME::SALOME_Exception )
1311 Unexpect aCatch(SALOME_SalomeException);
1312 checkFileReadable( theFileName );
1314 SMESH::mesh_array_var aResult = new SMESH::mesh_array();
1317 // Retrieve nb meshes from the file
1318 DriverCGNS_Read myReader;
1319 myReader.SetFile( theFileName );
1320 Driver_Mesh::Status aStatus;
1321 int nbMeshes = myReader.GetNbMeshes(aStatus);
1322 theStatus = (SMESH::DriverMED_ReadStatus)aStatus;
1324 aResult->length( nbMeshes );
1326 { // open a new scope to make aPythonDump die before PythonDump in SMESH_Mesh::GetGroups()
1329 TPythonDump aPythonDump;
1330 aPythonDump << "([";
1332 if (theStatus == SMESH::DRS_OK)
1334 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1335 aStudyBuilder->NewCommand(); // There is a transaction
1339 // Iterate through all meshes and create mesh objects
1340 for ( ; i < nbMeshes; ++i )
1343 if (i > 0) aPythonDump << ", ";
1346 SMESH::SMESH_Mesh_var mesh = createMesh();
1347 aResult[i] = SMESH::SMESH_Mesh::_duplicate( mesh );
1349 // Read mesh data (groups are published automatically by ImportMEDFile())
1350 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( mesh ).in() );
1351 ASSERT( meshServant );
1353 SMESH::DriverMED_ReadStatus status1 =
1354 meshServant->ImportCGNSFile( theFileName, i, meshName );
1355 if (status1 > theStatus)
1356 theStatus = status1;
1358 meshServant->GetImpl().GetMeshDS()->Modified();
1359 // publish mesh in the study
1360 SALOMEDS::SObject_wrap aSO;
1361 if ( CanPublishInStudy( mesh ) )
1362 aSO = PublishMesh( myCurrentStudy, mesh.in(), meshName.c_str() );
1365 if ( !aSO->_is_nil() ) {
1369 aPythonDump << "mesh_" << i;
1372 aStudyBuilder->CommitCommand();
1375 aPythonDump << "], status) = " << this << ".CreateMeshesFromCGNS(r'" << theFileName << "')";
1377 // Dump creation of groups
1378 for ( int i = 0; i < aResult->length(); ++i )
1379 SMESH::ListOfGroups_var groups = aResult[ i ]->GetGroups();
1381 THROW_SALOME_CORBA_EXCEPTION("CGNS library is unavailable", SALOME::INTERNAL_ERROR);
1384 return aResult._retn();
1387 //================================================================================
1389 * \brief Create a mesh and import data from a GMF file
1391 //================================================================================
1393 SMESH::SMESH_Mesh_ptr
1394 SMESH_Gen_i::CreateMeshesFromGMF( const char* theFileName,
1395 CORBA::Boolean theMakeRequiredGroups,
1396 SMESH::ComputeError_out theError)
1397 throw ( SALOME::SALOME_Exception )
1399 Unexpect aCatch(SALOME_SalomeException);
1400 checkFileReadable( theFileName );
1402 SMESH::SMESH_Mesh_var aMesh = createMesh();
1404 char bname[ _MAX_FNAME ];
1405 _splitpath( theFileName, NULL, NULL, bname, NULL );
1406 string aFileName = bname;
1408 string aFileName = basename( theFileName );
1410 // publish mesh in the study
1411 if ( CanPublishInStudy( aMesh ) ) {
1412 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1413 aStudyBuilder->NewCommand(); // There is a transaction
1414 SALOMEDS::SObject_wrap aSO = PublishInStudy
1415 ( myCurrentStudy, SALOMEDS::SObject::_nil(), aMesh.in(), aFileName.c_str() );
1416 aStudyBuilder->CommitCommand();
1417 if ( !aSO->_is_nil() ) {
1418 // Update Python script
1419 TPythonDump() << "("<< aSO << ", error) = " << this << ".CreateMeshesFromGMF(r'"
1420 << theFileName << "', "
1421 << theMakeRequiredGroups << " )";
1424 SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
1426 theError = aServant->ImportGMFFile( theFileName, theMakeRequiredGroups );
1427 aServant->GetImpl().GetMeshDS()->Modified();
1428 return aMesh._retn();
1432 //=============================================================================
1434 * SMESH_Gen_i::IsReadyToCompute
1436 * Returns true if mesh contains enough data to be computed
1438 //=============================================================================
1440 CORBA::Boolean SMESH_Gen_i::IsReadyToCompute( SMESH::SMESH_Mesh_ptr theMesh,
1441 GEOM::GEOM_Object_ptr theShapeObject )
1442 throw ( SALOME::SALOME_Exception )
1444 Unexpect aCatch(SALOME_SalomeException);
1445 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::IsReadyToCompute" );
1447 if ( CORBA::is_nil( theShapeObject ) )
1448 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1449 SALOME::BAD_PARAM );
1451 if ( CORBA::is_nil( theMesh ) )
1452 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
1453 SALOME::BAD_PARAM );
1457 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1458 ASSERT( meshServant );
1459 if ( meshServant ) {
1460 // get local TopoDS_Shape
1461 TopoDS_Shape myLocShape = GeomObjectToShape( theShapeObject );
1462 // call implementation
1463 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1464 return myGen.CheckAlgoState( myLocMesh, myLocShape );
1467 catch ( SALOME_Exception& S_ex ) {
1468 INFOS( "catch exception "<< S_ex.what() );
1473 //================================================================================
1475 * \brief Find SObject for an algo
1477 //================================================================================
1479 SALOMEDS::SObject_ptr SMESH_Gen_i::GetAlgoSO(const ::SMESH_Algo* algo)
1482 if ( !myCurrentStudy->_is_nil() ) {
1483 // find algo in the study
1484 CORBA::String_var compDataType = ComponentDataType();
1485 SALOMEDS::SComponent_wrap father = myCurrentStudy->FindComponent( compDataType.in() );
1486 if ( !father->_is_nil() ) {
1487 SALOMEDS::ChildIterator_wrap itBig = myCurrentStudy->NewChildIterator( father );
1488 for ( ; itBig->More(); itBig->Next() ) {
1489 SALOMEDS::SObject_wrap gotBranch = itBig->Value();
1490 if ( gotBranch->Tag() == GetAlgorithmsRootTag() ) {
1491 SALOMEDS::ChildIterator_wrap algoIt = myCurrentStudy->NewChildIterator( gotBranch );
1492 for ( ; algoIt->More(); algoIt->Next() ) {
1493 SALOMEDS::SObject_wrap algoSO = algoIt->Value();
1494 CORBA::Object_var algoIOR = SObjectToObject( algoSO );
1495 if ( !CORBA::is_nil( algoIOR )) {
1496 SMESH_Hypothesis_i* impl = SMESH::DownCast<SMESH_Hypothesis_i*>( algoIOR );
1497 if ( impl && impl->GetImpl() == algo )
1498 return algoSO._retn();
1500 } // loop on algo SO's
1503 } // SMESH component iterator
1507 return SALOMEDS::SObject::_nil();
1510 //================================================================================
1512 * \brief Return errors of mesh computation
1514 //================================================================================
1516 SMESH::compute_error_array* SMESH_Gen_i::GetComputeErrors( SMESH::SMESH_Mesh_ptr theMesh,
1517 GEOM::GEOM_Object_ptr theSubObject )
1518 throw ( SALOME::SALOME_Exception )
1520 Unexpect aCatch(SALOME_SalomeException);
1521 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetComputeErrors()" );
1523 if ( CORBA::is_nil( theSubObject ) && theMesh->HasShapeToMesh())
1524 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference", SALOME::BAD_PARAM );
1526 if ( CORBA::is_nil( theMesh ) )
1527 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1529 SMESH::compute_error_array_var error_array = new SMESH::compute_error_array;
1531 if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1534 if(theMesh->HasShapeToMesh())
1535 shape = GeomObjectToShape( theSubObject );
1537 shape = SMESH_Mesh::PseudoShape();
1539 ::SMESH_Mesh& mesh = meshServant->GetImpl();
1541 error_array->length( mesh.GetMeshDS()->MaxShapeIndex() );
1544 SMESH_subMesh *sm = mesh.GetSubMesh(shape);
1545 const bool includeSelf = true, complexShapeFirst = true;
1546 SMESH_subMeshIteratorPtr smIt = sm->getDependsOnIterator(includeSelf,
1548 while ( smIt->more() )
1551 // if ( sm->GetSubShape().ShapeType() == TopAbs_VERTEX )
1553 SMESH_ComputeErrorPtr error = sm->GetComputeError();
1554 if ( error && !error->IsOK() )
1556 if ( !( error->myAlgo ) &&
1557 !( error->myAlgo = sm->GetAlgo() ))
1559 SMESH::ComputeError & errStruct = error_array[ nbErr++ ];
1560 errStruct.code = -( error->myName < 0 ? error->myName + 1: error->myName ); // -1 -> 0
1561 errStruct.comment = error->myComment.c_str();
1562 errStruct.subShapeID = sm->GetId();
1563 SALOMEDS::SObject_wrap algoSO = GetAlgoSO( error->myAlgo );
1564 if ( !algoSO->_is_nil() ) {
1565 CORBA::String_var algoName = algoSO->GetName();
1566 errStruct.algoName = algoName;
1569 errStruct.algoName = error->myAlgo->GetName();
1571 errStruct.hasBadMesh = !error->myBadElements.empty();
1574 error_array->length( nbErr );
1577 catch ( SALOME_Exception& S_ex ) {
1578 INFOS( "catch exception "<< S_ex.what() );
1581 return error_array._retn();
1584 //================================================================================
1586 * \brief Return mesh elements preventing computation of a subshape
1588 //================================================================================
1590 SMESH::MeshPreviewStruct*
1591 SMESH_Gen_i::GetBadInputElements( SMESH::SMESH_Mesh_ptr theMesh,
1592 CORBA::Short theSubShapeID )
1593 throw ( SALOME::SALOME_Exception )
1595 Unexpect aCatch(SALOME_SalomeException);
1596 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetBadInputElements()" );
1598 if ( CORBA::is_nil( theMesh ) )
1599 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1601 SMESH::MeshPreviewStruct_var result = new SMESH::MeshPreviewStruct;
1604 if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1606 // mesh implementation
1607 ::SMESH_Mesh& mesh = meshServant->GetImpl();
1608 // submesh by subshape id
1609 if ( SMESH_subMesh * sm = mesh.GetSubMeshContaining( theSubShapeID ))
1612 SMESH_ComputeErrorPtr error = sm->GetComputeError();
1613 if ( error && !error->myBadElements.empty())
1615 typedef map<const SMDS_MeshElement*, int > TNode2LocalIDMap;
1616 typedef TNode2LocalIDMap::iterator TNodeLocalID;
1618 // get nodes of elements and count elements
1619 TNode2LocalIDMap mapNode2LocalID;
1620 list< TNodeLocalID > connectivity;
1621 int i, nbElements = 0, nbConnNodes = 0;
1623 list<const SMDS_MeshElement*>::iterator elemIt = error->myBadElements.begin();
1624 list<const SMDS_MeshElement*>::iterator elemEnd = error->myBadElements.end();
1625 for ( ; elemIt != elemEnd; ++elemIt, ++nbElements )
1627 SMDS_ElemIteratorPtr nIt = (*elemIt)->nodesIterator();
1628 while ( nIt->more() )
1629 connectivity.push_back
1630 ( mapNode2LocalID.insert( make_pair( nIt->next(), ++nbConnNodes)).first );
1632 // fill node coords and assign local ids to the nodes
1633 int nbNodes = mapNode2LocalID.size();
1634 result->nodesXYZ.length( nbNodes );
1635 TNodeLocalID node2ID = mapNode2LocalID.begin();
1636 for ( i = 0; i < nbNodes; ++i, ++node2ID ) {
1637 node2ID->second = i;
1638 const SMDS_MeshNode* node = (const SMDS_MeshNode*) node2ID->first;
1639 result->nodesXYZ[i].x = node->X();
1640 result->nodesXYZ[i].y = node->Y();
1641 result->nodesXYZ[i].z = node->Z();
1643 // fill connectivity
1644 result->elementConnectivities.length( nbConnNodes );
1645 list< TNodeLocalID >::iterator connIt = connectivity.begin();
1646 for ( i = 0; i < nbConnNodes; ++i, ++connIt ) {
1647 result->elementConnectivities[i] = (*connIt)->second;
1649 // fill element types
1650 result->elementTypes.length( nbElements );
1651 for ( i = 0, elemIt = error->myBadElements.begin(); i <nbElements; ++i, ++elemIt )
1653 const SMDS_MeshElement* elem = *elemIt;
1654 result->elementTypes[i].SMDS_ElementType = (SMESH::ElementType) elem->GetType();
1655 result->elementTypes[i].isPoly = elem->IsPoly();
1656 result->elementTypes[i].nbNodesInElement = elem->NbNodes();
1662 catch ( SALOME_Exception& S_ex ) {
1663 INFOS( "catch exception "<< S_ex.what() );
1666 return result._retn();
1669 //================================================================================
1671 * \brief Create a group of elements preventing computation of a sub-shape
1673 //================================================================================
1675 SMESH::ListOfGroups*
1676 SMESH_Gen_i::MakeGroupsOfBadInputElements( SMESH::SMESH_Mesh_ptr theMesh,
1677 CORBA::Short theSubShapeID,
1678 const char* theGroupName )
1679 throw ( SALOME::SALOME_Exception )
1681 Unexpect aCatch(SALOME_SalomeException);
1683 SMESH::ListOfGroups_var groups;
1685 if ( CORBA::is_nil( theMesh ) )
1686 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1689 if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1691 groups = meshServant->MakeGroupsOfBadInputElements( theSubShapeID, theGroupName );
1692 TPythonDump() << groups << " = " << this
1693 << ".MakeGroupsOfBadInputElements( "
1694 << theMesh << ", " << theSubShapeID << ", '" << theGroupName << "' )";
1697 catch ( SALOME_Exception& S_ex ) {
1698 INFOS( "catch exception "<< S_ex.what() );
1700 return groups._retn();
1703 //================================================================================
1705 * \brief Returns errors of hypotheses definintion
1706 * \param theMesh - the mesh
1707 * \param theSubObject - the main or sub- shape
1708 * \retval SMESH::algo_error_array* - sequence of errors
1710 //================================================================================
1712 SMESH::algo_error_array* SMESH_Gen_i::GetAlgoState( SMESH::SMESH_Mesh_ptr theMesh,
1713 GEOM::GEOM_Object_ptr theSubObject )
1714 throw ( SALOME::SALOME_Exception )
1716 Unexpect aCatch(SALOME_SalomeException);
1717 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetAlgoState()" );
1719 if ( CORBA::is_nil( theSubObject ) && theMesh->HasShapeToMesh())
1720 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference", SALOME::BAD_PARAM );
1722 if ( CORBA::is_nil( theMesh ) )
1723 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1725 SMESH::algo_error_array_var error_array = new SMESH::algo_error_array;
1727 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
1728 ASSERT( meshServant );
1729 if ( meshServant ) {
1730 TopoDS_Shape myLocShape;
1731 if(theMesh->HasShapeToMesh())
1732 myLocShape = GeomObjectToShape( theSubObject );
1734 myLocShape = SMESH_Mesh::PseudoShape();
1736 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1737 list< ::SMESH_Gen::TAlgoStateError > error_list;
1738 list< ::SMESH_Gen::TAlgoStateError >::iterator error;
1739 // call ::SMESH_Gen::GetAlgoState()
1740 myGen.GetAlgoState( myLocMesh, myLocShape, error_list );
1741 error_array->length( error_list.size() );
1743 for ( error = error_list.begin(); error != error_list.end(); ++error )
1745 // fill AlgoStateError structure
1746 SMESH::AlgoStateError & errStruct = error_array[ i++ ];
1747 errStruct.state = SMESH_Mesh_i::ConvertHypothesisStatus( error->_name );
1748 errStruct.algoDim = error->_algoDim;
1749 errStruct.isGlobalAlgo = error->_isGlobalAlgo;
1750 errStruct.algoName = "";
1751 SALOMEDS::SObject_wrap algoSO = GetAlgoSO( error->_algo );
1752 if ( !algoSO->_is_nil() ) {
1753 CORBA::String_var algoName = algoSO->GetName();
1754 errStruct.algoName = algoName.in();
1759 catch ( SALOME_Exception& S_ex ) {
1760 INFOS( "catch exception "<< S_ex.what() );
1762 return error_array._retn();
1765 //=============================================================================
1767 * SMESH_Gen_i::GetSubShapesId
1769 * Get sub-shapes unique ID's list
1771 //=============================================================================
1773 SMESH::long_array* SMESH_Gen_i::GetSubShapesId( GEOM::GEOM_Object_ptr theMainShapeObject,
1774 const SMESH::object_array& theListOfSubShapeObject )
1775 throw ( SALOME::SALOME_Exception )
1777 Unexpect aCatch(SALOME_SalomeException);
1778 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetSubShapesId" );
1780 SMESH::long_array_var shapesId = new SMESH::long_array;
1783 if ( CORBA::is_nil( theMainShapeObject ) )
1784 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1785 SALOME::BAD_PARAM );
1789 TopoDS_Shape myMainShape = GeomObjectToShape(theMainShapeObject);
1790 TopTools_IndexedMapOfShape myIndexToShape;
1791 TopExp::MapShapes(myMainShape,myIndexToShape);
1793 for ( int i = 0; i < theListOfSubShapeObject.length(); i++ )
1795 GEOM::GEOM_Object_var aShapeObject
1796 = GEOM::GEOM_Object::_narrow(theListOfSubShapeObject[i]);
1797 if ( CORBA::is_nil( aShapeObject ) )
1798 THROW_SALOME_CORBA_EXCEPTION ("bad shape object reference", \
1799 SALOME::BAD_PARAM );
1801 TopoDS_Shape locShape = GeomObjectToShape(aShapeObject);
1802 for (TopExp_Explorer exp(locShape,TopAbs_FACE); exp.More(); exp.Next())
1804 const TopoDS_Face& F = TopoDS::Face(exp.Current());
1805 setId.insert(myIndexToShape.FindIndex(F));
1806 if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(F));
1808 for (TopExp_Explorer exp(locShape,TopAbs_EDGE); exp.More(); exp.Next())
1810 const TopoDS_Edge& E = TopoDS::Edge(exp.Current());
1811 setId.insert(myIndexToShape.FindIndex(E));
1812 if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(E));
1814 for (TopExp_Explorer exp(locShape,TopAbs_VERTEX); exp.More(); exp.Next())
1816 const TopoDS_Vertex& V = TopoDS::Vertex(exp.Current());
1817 setId.insert(myIndexToShape.FindIndex(V));
1818 if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(V));
1821 shapesId->length(setId.size());
1822 set<int>::iterator iind;
1824 for (iind = setId.begin(); iind != setId.end(); iind++)
1826 if(MYDEBUG) SCRUTE((*iind));
1827 shapesId[i] = (*iind);
1828 if(MYDEBUG) SCRUTE(shapesId[i]);
1832 catch (SALOME_Exception& S_ex)
1834 THROW_SALOME_CORBA_EXCEPTION(S_ex.what(), SALOME::BAD_PARAM);
1837 return shapesId._retn();
1840 //=============================================================================
1842 * SMESH_Gen_i::Compute
1844 * Compute mesh on a shape
1846 //=============================================================================
1848 CORBA::Boolean SMESH_Gen_i::Compute( SMESH::SMESH_Mesh_ptr theMesh,
1849 GEOM::GEOM_Object_ptr theShapeObject )
1850 throw ( SALOME::SALOME_Exception )
1853 Unexpect aCatch(SALOME_SalomeException);
1854 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Compute" );
1856 if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
1857 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1858 SALOME::BAD_PARAM );
1860 if ( CORBA::is_nil( theMesh ) )
1861 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
1862 SALOME::BAD_PARAM );
1864 // Update Python script
1865 TPythonDump() << "isDone = " << this << ".Compute( "
1866 << theMesh << ", " << theShapeObject << ")";
1870 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1871 meshServant->Load();
1872 ASSERT( meshServant );
1873 if ( meshServant ) {
1874 // NPAL16168: "geometrical group edition from a submesh don't modifiy mesh computation"
1875 meshServant->CheckGeomModif();
1876 // get local TopoDS_Shape
1877 TopoDS_Shape myLocShape;
1878 if(theMesh->HasShapeToMesh())
1879 myLocShape = GeomObjectToShape( theShapeObject );
1881 myLocShape = SMESH_Mesh::PseudoShape();
1882 // call implementation compute
1883 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1884 myGen.PrepareCompute( myLocMesh, myLocShape);
1885 bool ok = myGen.Compute( myLocMesh, myLocShape);
1886 meshServant->CreateGroupServants(); // algos can create groups (issue 0020918)
1887 myLocMesh.GetMeshDS()->Modified();
1891 catch ( std::bad_alloc ) {
1892 INFOS( "Compute(): lack of memory" );
1894 catch ( SALOME_Exception& S_ex ) {
1895 INFOS( "Compute(): catch exception "<< S_ex.what() );
1898 INFOS( "Compute(): unknown exception " );
1903 //=============================================================================
1905 * SMESH_Gen_i::CancelCompute
1907 * Cancel Compute mesh on a shape
1909 //=============================================================================
1911 void SMESH_Gen_i::CancelCompute( SMESH::SMESH_Mesh_ptr theMesh,
1912 GEOM::GEOM_Object_ptr theShapeObject )
1914 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1915 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1916 TopoDS_Shape myLocShape;
1917 if(theMesh->HasShapeToMesh())
1918 myLocShape = GeomObjectToShape( theShapeObject );
1920 myLocShape = SMESH_Mesh::PseudoShape();
1921 myGen.CancelCompute( myLocMesh, myLocShape);
1924 //=============================================================================
1926 * SMESH_Gen_i::Precompute
1928 * Compute mesh as preview till indicated dimension on shape
1930 //=============================================================================
1932 SMESH::MeshPreviewStruct* SMESH_Gen_i::Precompute( SMESH::SMESH_Mesh_ptr theMesh,
1933 GEOM::GEOM_Object_ptr theShapeObject,
1934 SMESH::Dimension theDimension,
1935 SMESH::long_array& theShapesId)
1936 throw ( SALOME::SALOME_Exception )
1938 Unexpect aCatch(SALOME_SalomeException);
1939 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Precompute" );
1941 if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
1942 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1943 SALOME::BAD_PARAM );
1945 if ( CORBA::is_nil( theMesh ) )
1946 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
1947 SALOME::BAD_PARAM );
1949 SMESH::MeshPreviewStruct_var result = new SMESH::MeshPreviewStruct;
1952 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1953 meshServant->Load();
1954 ASSERT( meshServant );
1955 if ( meshServant ) {
1956 // NPAL16168: "geometrical group edition from a submesh don't modifiy mesh computation"
1957 meshServant->CheckGeomModif();
1958 // get local TopoDS_Shape
1959 TopoDS_Shape myLocShape;
1960 if(theMesh->HasShapeToMesh())
1961 myLocShape = GeomObjectToShape( theShapeObject );
1963 return result._retn();;
1965 // call implementation compute
1966 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1968 ::MeshDimension aDim = (MeshDimension)theDimension;
1969 if ( myGen.Compute( myLocMesh, myLocShape, false, false, aDim, &shapeIds ) )
1971 int nbShapeId = shapeIds.size();
1972 theShapesId.length( nbShapeId );
1973 // iterates on shapes and collect mesh entities into mesh preview
1974 TSetOfInt::const_iterator idIt = shapeIds.begin();
1975 TSetOfInt::const_iterator idEnd = shapeIds.end();
1976 std::map< int, int > mapOfShIdNb;
1977 std::set< SMESH_TLink > setOfEdge;
1978 std::list< SMDSAbs_ElementType > listOfElemType;
1979 typedef map<const SMDS_MeshElement*, int > TNode2LocalIDMap;
1980 typedef TNode2LocalIDMap::iterator TNodeLocalID;
1981 TNode2LocalIDMap mapNode2LocalID;
1982 list< TNodeLocalID > connectivity;
1983 int i, nbConnNodes = 0;
1984 std::set< const SMESH_subMesh* > setOfVSubMesh;
1985 // iterates on shapes
1986 for ( ; idIt != idEnd; idIt++ )
1988 if ( mapOfShIdNb.find( *idIt ) != mapOfShIdNb.end() )
1990 SMESH_subMesh* sm = myLocMesh.GetSubMeshContaining(*idIt);
1991 if ( !sm || !sm->IsMeshComputed() )
1994 const TopoDS_Shape& aSh = sm->GetSubShape();
1995 const int shDim = myGen.GetShapeDim( aSh );
1996 if ( shDim < 1 || shDim > theDimension )
1999 mapOfShIdNb[ *idIt ] = 0;
2000 theShapesId[ mapOfShIdNb.size() - 1 ] = *idIt;
2002 SMESHDS_SubMesh* smDS = sm->GetSubMeshDS();
2003 if ( !smDS ) continue;
2005 if ( theDimension == SMESH::DIM_2D )
2007 SMDS_ElemIteratorPtr faceIt = smDS->GetElements();
2008 while ( faceIt->more() )
2010 const SMDS_MeshElement* face = faceIt->next();
2011 int aNbNode = face->NbNodes();
2013 aNbNode /= 2; // do not take into account additional middle nodes
2015 SMDS_MeshNode* node1 = (SMDS_MeshNode*)face->GetNode( 0 );
2016 for ( int nIndx = 0; nIndx < aNbNode; nIndx++ )
2018 SMDS_MeshNode* node2 = (SMDS_MeshNode*)face->GetNode( nIndx+1 < aNbNode ? nIndx+1 : 0 );
2019 if ( setOfEdge.insert( SMESH_TLink ( node1, node2 ) ).second )
2021 listOfElemType.push_back( SMDSAbs_Edge );
2022 connectivity.push_back
2023 ( mapNode2LocalID.insert( make_pair( node1, ++nbConnNodes)).first );
2024 connectivity.push_back
2025 ( mapNode2LocalID.insert( make_pair( node2, ++nbConnNodes)).first );
2031 else if ( theDimension == SMESH::DIM_1D )
2033 SMDS_NodeIteratorPtr nodeIt = smDS->GetNodes();
2034 while ( nodeIt->more() )
2036 listOfElemType.push_back( SMDSAbs_Node );
2037 connectivity.push_back
2038 ( mapNode2LocalID.insert( make_pair( nodeIt->next(), ++nbConnNodes)).first );
2040 // add corner nodes by first vertex from edge
2041 SMESH_subMeshIteratorPtr edgeSmIt =
2042 sm->getDependsOnIterator(/*includeSelf*/false,
2043 /*complexShapeFirst*/false);
2044 while ( edgeSmIt->more() )
2046 SMESH_subMesh* vertexSM = edgeSmIt->next();
2047 // check that vertex is not already treated
2048 if ( !setOfVSubMesh.insert( vertexSM ).second )
2050 if ( vertexSM->GetSubShape().ShapeType() != TopAbs_VERTEX )
2053 const SMESHDS_SubMesh* vertexSmDS = vertexSM->GetSubMeshDS();
2054 SMDS_NodeIteratorPtr nodeIt = vertexSmDS->GetNodes();
2055 while ( nodeIt->more() )
2057 listOfElemType.push_back( SMDSAbs_Node );
2058 connectivity.push_back
2059 ( mapNode2LocalID.insert( make_pair( nodeIt->next(), ++nbConnNodes)).first );
2065 // fill node coords and assign local ids to the nodes
2066 int nbNodes = mapNode2LocalID.size();
2067 result->nodesXYZ.length( nbNodes );
2068 TNodeLocalID node2ID = mapNode2LocalID.begin();
2069 for ( i = 0; i < nbNodes; ++i, ++node2ID ) {
2070 node2ID->second = i;
2071 const SMDS_MeshNode* node = (const SMDS_MeshNode*) node2ID->first;
2072 result->nodesXYZ[i].x = node->X();
2073 result->nodesXYZ[i].y = node->Y();
2074 result->nodesXYZ[i].z = node->Z();
2076 // fill connectivity
2077 result->elementConnectivities.length( nbConnNodes );
2078 list< TNodeLocalID >::iterator connIt = connectivity.begin();
2079 for ( i = 0; i < nbConnNodes; ++i, ++connIt ) {
2080 result->elementConnectivities[i] = (*connIt)->second;
2083 // fill element types
2084 result->elementTypes.length( listOfElemType.size() );
2085 std::list< SMDSAbs_ElementType >::const_iterator typeIt = listOfElemType.begin();
2086 std::list< SMDSAbs_ElementType >::const_iterator typeEnd = listOfElemType.end();
2087 for ( i = 0; typeIt != typeEnd; ++i, ++typeIt )
2089 SMDSAbs_ElementType elemType = *typeIt;
2090 result->elementTypes[i].SMDS_ElementType = (SMESH::ElementType)elemType;
2091 result->elementTypes[i].isPoly = false;
2092 result->elementTypes[i].nbNodesInElement = elemType == SMDSAbs_Edge ? 2 : 1;
2095 // correct number of shapes
2096 theShapesId.length( mapOfShIdNb.size() );
2100 catch ( std::bad_alloc ) {
2101 INFOS( "Precompute(): lack of memory" );
2103 catch ( SALOME_Exception& S_ex ) {
2104 INFOS( "Precompute(): catch exception "<< S_ex.what() );
2107 INFOS( "Precompute(): unknown exception " );
2109 return result._retn();
2113 //=============================================================================
2115 * SMESH_Gen_i::Evaluate
2117 * Evaluate mesh on a shape
2119 //=============================================================================
2121 SMESH::long_array* SMESH_Gen_i::Evaluate(SMESH::SMESH_Mesh_ptr theMesh,
2122 GEOM::GEOM_Object_ptr theShapeObject)
2123 // SMESH::long_array& theNbElems)
2124 throw ( SALOME::SALOME_Exception )
2126 Unexpect aCatch(SALOME_SalomeException);
2127 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Evaluate" );
2129 if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
2130 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
2131 SALOME::BAD_PARAM );
2133 if ( CORBA::is_nil( theMesh ) )
2134 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
2135 SALOME::BAD_PARAM );
2137 SMESH::long_array_var nbels = new SMESH::long_array;
2138 nbels->length(SMESH::Entity_Last);
2139 int i = SMESH::Entity_Node;
2140 for (; i < SMESH::Entity_Last; i++)
2143 // Update Python script
2144 TPythonDump() << "theNbElems = " << this << ".Evaluate( "
2145 << theMesh << ", " << theShapeObject << ")";
2149 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
2150 ASSERT( meshServant );
2151 if ( meshServant ) {
2152 // NPAL16168: "geometrical group edition from a submesh don't modifiy mesh computation"
2153 meshServant->CheckGeomModif();
2154 // get local TopoDS_Shape
2155 TopoDS_Shape myLocShape;
2156 if(theMesh->HasShapeToMesh())
2157 myLocShape = GeomObjectToShape( theShapeObject );
2159 myLocShape = SMESH_Mesh::PseudoShape();
2160 // call implementation compute
2161 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
2162 MapShapeNbElems aResMap;
2163 /*CORBA::Boolean ret =*/ myGen.Evaluate( myLocMesh, myLocShape, aResMap);
2164 MapShapeNbElemsItr anIt = aResMap.begin();
2165 for(; anIt!=aResMap.end(); anIt++) {
2166 const vector<int>& aVec = (*anIt).second;
2167 for(i = SMESH::Entity_Node; i < aVec.size(); i++) {
2168 int nbElem = aVec[i];
2169 if ( nbElem < 0 ) // algo failed, check that it has reported a message
2171 SMESH_subMesh* sm = anIt->first;
2172 SMESH_ComputeErrorPtr& error = sm->GetComputeError();
2173 const SMESH_Algo* algo = myGen.GetAlgo( myLocMesh, sm->GetSubShape());
2174 if ( (algo && !error.get()) || error->IsOK() )
2175 error.reset( new SMESH_ComputeError( COMPERR_ALGO_FAILED,"Failed to evaluate",algo));
2179 nbels[i] += aVec[i];
2183 return nbels._retn();
2186 catch ( std::bad_alloc ) {
2187 INFOS( "Evaluate(): lack of memory" );
2189 catch ( SALOME_Exception& S_ex ) {
2190 INFOS( "Evaluate(): catch exception "<< S_ex.what() );
2193 INFOS( "Evaluate(): unknown exception " );
2196 return nbels._retn();
2199 //================================================================================
2201 * \brief Return geometrical object the given element is built on
2202 * \param theMesh - the mesh the element is in
2203 * \param theElementID - the element ID
2204 * \param theGeomName - the name of the result geom object if it is not yet published
2205 * \retval GEOM::GEOM_Object_ptr - the found or just published geom object (no need to UnRegister())
2207 //================================================================================
2209 GEOM::GEOM_Object_ptr
2210 SMESH_Gen_i::GetGeometryByMeshElement( SMESH::SMESH_Mesh_ptr theMesh,
2211 CORBA::Long theElementID,
2212 const char* theGeomName)
2213 throw ( SALOME::SALOME_Exception )
2215 Unexpect aCatch(SALOME_SalomeException);
2217 GEOM::GEOM_Object_wrap geom = FindGeometryByMeshElement(theMesh, theElementID);
2218 if ( !geom->_is_nil() ) {
2219 GEOM::GEOM_Object_var mainShape = theMesh->GetShapeToMesh();
2220 GEOM::GEOM_Gen_ptr geomGen = GetGeomEngine();
2222 // try to find the corresponding SObject
2223 SALOMEDS::SObject_wrap SObj = ObjectToSObject( myCurrentStudy, geom.in() );
2224 if ( SObj->_is_nil() ) // submesh can be not found even if published
2226 // try to find published submesh
2227 GEOM::ListOfLong_var list = geom->GetSubShapeIndices();
2228 if ( !geom->IsMainShape() && list->length() == 1 ) {
2229 SALOMEDS::SObject_wrap mainSO = ObjectToSObject( myCurrentStudy, mainShape );
2230 SALOMEDS::ChildIterator_wrap it;
2231 if ( !mainSO->_is_nil() ) {
2232 it = myCurrentStudy->NewChildIterator( mainSO );
2234 if ( !it->_is_nil() ) {
2235 for ( it->InitEx(true); it->More(); it->Next() ) {
2236 SALOMEDS::SObject_wrap so = it->Value();
2237 CORBA::Object_var obj = SObjectToObject( so );
2238 GEOM::GEOM_Object_var subGeom = GEOM::GEOM_Object::_narrow( obj );
2239 if ( !subGeom->_is_nil() ) {
2240 GEOM::ListOfLong_var subList = subGeom->GetSubShapeIndices();
2241 if ( subList->length() == 1 && list[0] == subList[0] ) {
2251 if ( SObj->_is_nil() ) // publish a new subshape
2252 SObj = geomGen->AddInStudy( myCurrentStudy, geom, theGeomName, mainShape );
2254 // return only published geometry
2255 if ( !SObj->_is_nil() ) {
2256 //return geom._retn(); -- servant of geom must be UnRegister()ed;
2257 CORBA::Object_var obj = SObjectToObject( SObj );
2258 GEOM::GEOM_Object_var go = GEOM::GEOM_Object::_narrow( obj );
2262 return GEOM::GEOM_Object::_nil();
2265 //================================================================================
2267 * \brief Return geometrical object the given element is built on.
2268 * \param theMesh - the mesh the element is in
2269 * \param theElementID - the element ID
2270 * \retval GEOM::GEOM_Object_ptr - the found or created (UnRegister()!) geom object
2272 //================================================================================
2274 GEOM::GEOM_Object_ptr
2275 SMESH_Gen_i::FindGeometryByMeshElement( SMESH::SMESH_Mesh_ptr theMesh,
2276 CORBA::Long theElementID)
2277 throw ( SALOME::SALOME_Exception )
2279 Unexpect aCatch(SALOME_SalomeException);
2280 if ( CORBA::is_nil( theMesh ) )
2281 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference", SALOME::BAD_PARAM );
2283 GEOM::GEOM_Object_var mainShape = theMesh->GetShapeToMesh();
2284 GEOM::GEOM_Gen_ptr geomGen = GetGeomEngine();
2286 // get a core mesh DS
2287 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
2288 if ( meshServant && !geomGen->_is_nil() && !mainShape->_is_nil() )
2290 ::SMESH_Mesh & mesh = meshServant->GetImpl();
2291 SMESHDS_Mesh* meshDS = mesh.GetMeshDS();
2292 // find the element in mesh
2293 if ( const SMDS_MeshElement * elem = meshDS->FindElement( theElementID ) ) {
2294 // find a shape id by the element
2295 if ( int shapeID = ::SMESH_MeshEditor( &mesh ).FindShape( elem )) {
2296 // get a geom object by the shape id
2297 GEOM::GEOM_Object_var geom = ShapeToGeomObject( meshDS->IndexToShape( shapeID ));
2298 if ( geom->_is_nil() ) {
2299 // try to find a published sub-shape
2300 SALOMEDS::SObject_wrap mainSO = ObjectToSObject( myCurrentStudy, mainShape );
2301 SALOMEDS::ChildIterator_wrap it;
2302 if ( !mainSO->_is_nil() ) {
2303 it = myCurrentStudy->NewChildIterator( mainSO );
2305 if ( !it->_is_nil() ) {
2306 for ( it->InitEx(true); it->More(); it->Next() ) {
2307 SALOMEDS::SObject_wrap so = it->Value();
2308 CORBA::Object_var obj = SObjectToObject( so );
2309 GEOM::GEOM_Object_var subGeom = GEOM::GEOM_Object::_narrow( obj );
2310 if ( !subGeom->_is_nil() ) {
2311 GEOM::ListOfLong_var subList = subGeom->GetSubShapeIndices();
2312 if ( subList->length() == 1 && shapeID == subList[0] ) {
2320 if ( geom->_is_nil() ) {
2322 GEOM::GEOM_IShapesOperations_wrap op =
2323 geomGen->GetIShapesOperations( GetCurrentStudyID() );
2324 if ( !op->_is_nil() )
2325 geom = op->GetSubShape( mainShape, shapeID );
2330 if ( !geom->_is_nil() ) {
2331 GeomObjectToShape( geom ); // let geom client remember the found shape
2332 return geom._retn();
2337 return GEOM::GEOM_Object::_nil();
2340 //================================================================================
2342 * SMESH_Gen_i::Concatenate
2344 * Concatenate the given meshes into one mesh
2346 //================================================================================
2348 SMESH::SMESH_Mesh_ptr
2349 SMESH_Gen_i::Concatenate(const SMESH::ListOfIDSources& theMeshesArray,
2350 CORBA::Boolean theUniteIdenticalGroups,
2351 CORBA::Boolean theMergeNodesAndElements,
2352 CORBA::Double theMergeTolerance)
2353 throw ( SALOME::SALOME_Exception )
2355 return ConcatenateCommon(theMeshesArray,
2356 theUniteIdenticalGroups,
2357 theMergeNodesAndElements,
2362 //================================================================================
2364 * SMESH_Gen_i::ConcatenateWithGroups
2366 * Concatenate the given meshes into one mesh
2367 * Create the groups of all elements from initial meshes
2369 //================================================================================
2371 SMESH::SMESH_Mesh_ptr
2372 SMESH_Gen_i::ConcatenateWithGroups(const SMESH::ListOfIDSources& theMeshesArray,
2373 CORBA::Boolean theUniteIdenticalGroups,
2374 CORBA::Boolean theMergeNodesAndElements,
2375 CORBA::Double theMergeTolerance)
2376 throw ( SALOME::SALOME_Exception )
2378 return ConcatenateCommon(theMeshesArray,
2379 theUniteIdenticalGroups,
2380 theMergeNodesAndElements,
2385 //================================================================================
2387 * SMESH_Gen_i::ConcatenateCommon
2389 * Concatenate the given meshes into one mesh
2391 //================================================================================
2393 SMESH::SMESH_Mesh_ptr
2394 SMESH_Gen_i::ConcatenateCommon(const SMESH::ListOfIDSources& theMeshesArray,
2395 CORBA::Boolean theUniteIdenticalGroups,
2396 CORBA::Boolean theMergeNodesAndElements,
2397 CORBA::Double theMergeTolerance,
2398 CORBA::Boolean theCommonGroups)
2399 throw ( SALOME::SALOME_Exception )
2401 typedef list<SMESH::SMESH_Group_var> TListOfNewGroups;
2402 typedef map< pair<string, SMESH::ElementType>, TListOfNewGroups > TGroupsMap;
2404 TPythonDump* pPythonDump = new TPythonDump;
2405 TPythonDump& aPythonDump = *pPythonDump; // prevent dump of called methods
2408 SMESH::SMESH_Mesh_var aNewMesh = CreateEmptyMesh();
2410 if ( aNewMesh->_is_nil() )
2411 return aNewMesh._retn();
2413 SMESH_Mesh_i* aNewImpl = SMESH::DownCast<SMESH_Mesh_i*>( aNewMesh );
2415 return aNewMesh._retn();
2417 ::SMESH_Mesh& aLocMesh = aNewImpl->GetImpl();
2418 SMESHDS_Mesh* aNewMeshDS = aLocMesh.GetMeshDS();
2420 TGroupsMap aGroupsMap;
2421 TListOfNewGroups aListOfNewGroups;
2422 ::SMESH_MeshEditor aNewEditor(&aLocMesh);
2423 SMESH::ListOfGroups_var aListOfGroups;
2425 std::vector<const SMDS_MeshNode*> aNodesArray;
2427 // loop on sub-meshes
2428 for ( int i = 0; i < theMeshesArray.length(); i++)
2430 SMESH::SMESH_Mesh_var anInitMesh = theMeshesArray[i]->GetMesh();
2431 if ( anInitMesh->_is_nil() ) continue;
2432 SMESH_Mesh_i* anInitImpl = SMESH::DownCast<SMESH_Mesh_i*>( anInitMesh );
2433 if ( !anInitImpl ) continue;
2436 ::SMESH_Mesh& aInitLocMesh = anInitImpl->GetImpl();
2437 SMESHDS_Mesh* anInitMeshDS = aInitLocMesh.GetMeshDS();
2439 // remember nb of elements before filling in
2440 SMESH::long_array_var prevState = aNewMesh->GetNbElementsByType();
2442 typedef std::map<const SMDS_MeshElement*, const SMDS_MeshElement*, TIDCompare > TEEMap;
2443 TEEMap elemsMap, nodesMap;
2445 // loop on elements of a sub-mesh
2446 SMDS_ElemIteratorPtr itElems = anInitImpl->GetElements( theMeshesArray[i], SMESH::ALL );
2447 const SMDS_MeshElement* anElem;
2448 const SMDS_MeshElement* aNewElem;
2449 const SMDS_MeshNode* aNode;
2450 const SMDS_MeshNode* aNewNode;
2453 while ( itElems->more() )
2455 anElem = itElems->next();
2456 anElemNbNodes = anElem->NbNodes();
2457 aNodesArray.resize( anElemNbNodes );
2459 // loop on nodes of an element
2460 SMDS_ElemIteratorPtr itNodes = anElem->nodesIterator();
2461 for ( int k = 0; itNodes->more(); k++)
2463 aNode = static_cast<const SMDS_MeshNode*>( itNodes->next() );
2464 TEEMap::iterator n2nnIt = nodesMap.find( aNode );
2465 if ( n2nnIt == nodesMap.end() )
2467 aNewNode = aNewMeshDS->AddNode(aNode->X(), aNode->Y(), aNode->Z());
2468 nodesMap.insert( make_pair( aNode, aNewNode ));
2472 aNewNode = static_cast<const SMDS_MeshNode*>( n2nnIt->second );
2474 aNodesArray[k] = aNewNode;
2477 // creates a corresponding element on existent nodes in new mesh
2479 switch ( anElem->GetEntityType() )
2481 case SMDSEntity_Polyhedra:
2482 if ( const SMDS_VtkVolume* aVolume =
2483 dynamic_cast<const SMDS_VtkVolume*> (anElem))
2485 aNewElem = aNewMeshDS->AddPolyhedralVolume( aNodesArray,
2486 aVolume->GetQuantities() );
2489 case SMDSEntity_Ball:
2490 if ( const SMDS_BallElement* aBall =
2491 dynamic_cast<const SMDS_BallElement*> (anElem))
2493 aNewElem = aNewEditor.AddElement( aNodesArray, SMDSAbs_Ball,
2494 /*isPoly=*/false, /*id=*/0,
2495 aBall->GetDiameter() );
2498 case SMDSEntity_Node:
2501 aNewElem = aNewEditor.AddElement( aNodesArray, anElem->GetType(), anElem->IsPoly() );
2504 elemsMap.insert( make_pair( anElem, aNewElem ));
2508 aNewEditor.ClearLastCreated(); // forget the history
2511 // create groups of just added elements
2512 SMESH::SMESH_Group_var aNewGroup;
2513 SMESH::ElementType aGroupType;
2514 if ( theCommonGroups )
2516 SMESH::long_array_var curState = aNewMesh->GetNbElementsByType();
2518 for( aGroupType = SMESH::NODE;
2519 aGroupType < SMESH::NB_ELEMENT_TYPES;
2520 aGroupType = (SMESH::ElementType)( aGroupType + 1 ))
2522 if ( curState[ aGroupType ] <= prevState[ aGroupType ])
2525 // make a group name
2526 const char* typeNames[] = { "All","Nodes","Edges","Faces","Volumes","0DElems","Balls" };
2527 { // check of typeNames, compilation failure mains that NB_ELEMENT_TYPES changed:
2528 const int nbNames = sizeof(typeNames) / sizeof(const char*);
2529 int _assert[( nbNames == SMESH::NB_ELEMENT_TYPES ) ? 1 : -1 ];
2531 string groupName = "Gr";
2532 SALOMEDS::SObject_wrap aMeshSObj = ObjectToSObject( myCurrentStudy, theMeshesArray[i] );
2534 CORBA::String_var name = aMeshSObj->GetName();
2538 groupName += typeNames[ aGroupType ];
2540 // make and fill a group
2541 TEEMap & e2neMap = ( aGroupType == SMESH::NODE ) ? nodesMap : elemsMap;
2542 aNewGroup = aNewImpl->CreateGroup( aGroupType, groupName.c_str() );
2543 if ( SMESH_Group_i* grp_i = SMESH::DownCast<SMESH_Group_i*>( aNewGroup ))
2545 if ( SMESHDS_Group* grpDS = dynamic_cast<SMESHDS_Group*>( grp_i->GetGroupDS() ))
2547 TEEMap::iterator e2neIt = e2neMap.begin();
2548 for ( ; e2neIt != e2neMap.end(); ++e2neIt )
2550 aNewElem = e2neIt->second;
2551 if ( aNewElem->GetType() == grpDS->GetType() )
2553 grpDS->Add( aNewElem );
2555 if ( prevState[ aGroupType ]++ >= curState[ aGroupType ] )
2561 aListOfNewGroups.clear();
2562 aListOfNewGroups.push_back(aNewGroup);
2563 aGroupsMap.insert(make_pair( make_pair(groupName, aGroupType), aListOfNewGroups ));
2567 if ( SMESH_Mesh_i* anSrcImpl = SMESH::DownCast<SMESH_Mesh_i*>( theMeshesArray[i] ))
2569 // copy orphan nodes
2570 if ( anSrcImpl->NbNodes() > nodesMap.size() )
2572 SMDS_ElemIteratorPtr itNodes = anInitImpl->GetElements( theMeshesArray[i], SMESH::NODE );
2573 while ( itNodes->more() )
2575 const SMDS_MeshNode* aNode = static_cast< const SMDS_MeshNode* >( itNodes->next() );
2576 if ( aNode->NbInverseElements() == 0 )
2578 aNewNode = aNewMeshDS->AddNode(aNode->X(), aNode->Y(), aNode->Z());
2579 nodesMap.insert( make_pair( aNode, aNewNode ));
2586 SMESH::SMESH_GroupBase_ptr aGroup;
2587 CORBA::String_var aGroupName;
2588 SMESH::long_array_var anNewIDs = new SMESH::long_array();
2590 // loop on groups of a source mesh
2591 aListOfGroups = anSrcImpl->GetGroups();
2592 for (int iG = 0; iG < aListOfGroups->length(); iG++)
2594 aGroup = aListOfGroups[iG];
2595 aGroupType = aGroup->GetType();
2596 aGroupName = aGroup->GetName();
2598 // convert a list of IDs
2599 anNewIDs->length( aGroup->Size() );
2600 TEEMap & e2neMap = ( aGroupType == SMESH::NODE ) ? nodesMap : elemsMap;
2601 SMDS_ElemIteratorPtr itGrElems = anSrcImpl->GetElements( aGroup, SMESH::ALL );
2603 while ( itGrElems->more() )
2605 anElem = itGrElems->next();
2606 TEEMap::iterator e2neIt = e2neMap.find( anElem );
2607 if ( e2neIt != e2neMap.end() )
2608 anNewIDs[ iElem++ ] = e2neIt->second->GetID();
2610 anNewIDs->length( iElem );
2612 // check a current group name and type don't have identical ones in final mesh
2613 aListOfNewGroups.clear();
2614 TGroupsMap::iterator anIter = aGroupsMap.find( make_pair( aGroupName, aGroupType ));
2615 if ( anIter == aGroupsMap.end() ) {
2616 // add a new group in the mesh
2617 aNewGroup = aNewImpl->CreateGroup( aGroupType, aGroupName );
2618 // add elements into new group
2619 aNewGroup->Add( anNewIDs );
2621 aListOfNewGroups.push_back(aNewGroup);
2622 aGroupsMap.insert(make_pair( make_pair(aGroupName, aGroupType), aListOfNewGroups ));
2625 else if ( theUniteIdenticalGroups ) {
2626 // unite identical groups
2627 TListOfNewGroups& aNewGroups = anIter->second;
2628 aNewGroups.front()->Add( anNewIDs );
2632 // rename identical groups
2633 aNewGroup = aNewImpl->CreateGroup(aGroupType, aGroupName);
2634 aNewGroup->Add( anNewIDs );
2636 TListOfNewGroups& aNewGroups = anIter->second;
2637 string aNewGroupName;
2638 if (aNewGroups.size() == 1) {
2639 aNewGroupName = string(aGroupName) + "_1";
2640 aNewGroups.front()->SetName(aNewGroupName.c_str());
2642 char aGroupNum[128];
2643 sprintf(aGroupNum, "%u", aNewGroups.size()+1);
2644 aNewGroupName = string(aGroupName) + "_" + string(aGroupNum);
2645 aNewGroup->SetName(aNewGroupName.c_str());
2646 aNewGroups.push_back(aNewGroup);
2649 } // if an IDSource is a mesh
2652 if (theMergeNodesAndElements) {
2654 TIDSortedNodeSet aMeshNodes; // no input nodes
2655 SMESH_MeshEditor::TListOfListOfNodes aGroupsOfNodes;
2656 aNewEditor.FindCoincidentNodes( aMeshNodes, theMergeTolerance, aGroupsOfNodes );
2657 aNewEditor.MergeNodes( aGroupsOfNodes );
2659 aNewEditor.MergeEqualElements();
2662 // Update Python script
2663 aPythonDump << aNewMesh << " = " << this << "."
2664 << ( theCommonGroups ? "ConcatenateWithGroups" : "Concatenate" )
2666 for ( int i = 0; i < theMeshesArray.length(); i++) {
2667 if (i > 0) aPythonDump << ", ";
2668 aPythonDump << theMeshesArray[i];
2670 aPythonDump << "], ";
2671 aPythonDump << theUniteIdenticalGroups << ", "
2672 << theMergeNodesAndElements << ", "
2673 << TVar( theMergeTolerance ) << ")";
2675 delete pPythonDump; // enable python dump from GetGroups()
2677 // 0020577: EDF 1164 SMESH: Bad dump of concatenate with create common groups
2678 if ( !aNewMesh->_is_nil() )
2680 SMESH::ListOfGroups_var groups = aNewMesh->GetGroups();
2683 // IPAL21468 Change icon of compound because it need not be computed.
2684 SALOMEDS::SObject_wrap aMeshSObj = ObjectToSObject( myCurrentStudy, aNewMesh );
2685 SetPixMap( aMeshSObj, "ICON_SMESH_TREE_MESH" );
2688 aNewMeshDS->Modified();
2690 return aNewMesh._retn();
2693 //================================================================================
2695 * \brief Create a mesh by copying a part of another mesh
2696 * \param meshPart - a part of mesh to copy
2697 * \param toCopyGroups - to create in the new mesh groups
2698 * the copied elements belongs to
2699 * \param toKeepIDs - to preserve IDs of the copied elements or not
2700 * \retval SMESH::SMESH_Mesh_ptr - the new mesh
2702 //================================================================================
2704 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CopyMesh(SMESH::SMESH_IDSource_ptr meshPart,
2705 const char* meshName,
2706 CORBA::Boolean toCopyGroups,
2707 CORBA::Boolean toKeepIDs)
2709 Unexpect aCatch(SALOME_SalomeException);
2711 TPythonDump* pyDump = new TPythonDump; // prevent dump from CreateMesh()
2713 // 1. Get source mesh
2715 if ( CORBA::is_nil( meshPart ))
2716 THROW_SALOME_CORBA_EXCEPTION( "bad IDSource", SALOME::BAD_PARAM );
2718 SMESH::SMESH_Mesh_var srcMesh = meshPart->GetMesh();
2719 SMESH_Mesh_i* srcMesh_i = SMESH::DownCast<SMESH_Mesh_i*>( srcMesh );
2721 THROW_SALOME_CORBA_EXCEPTION( "bad mesh of IDSource", SALOME::BAD_PARAM );
2723 SMESHDS_Mesh* srcMeshDS = srcMesh_i->GetImpl().GetMeshDS();
2725 // 2. Make a new mesh
2727 SMESH::SMESH_Mesh_var newMesh = CreateMesh(GEOM::GEOM_Object::_nil());
2728 SMESH_Mesh_i* newMesh_i = SMESH::DownCast<SMESH_Mesh_i*>( newMesh );
2730 THROW_SALOME_CORBA_EXCEPTION( "can't create a mesh", SALOME::INTERNAL_ERROR );
2731 SALOMEDS::SObject_wrap meshSO = ObjectToSObject(myCurrentStudy, newMesh );
2732 if ( !meshSO->_is_nil() )
2734 SetName( meshSO, meshName, "Mesh" );
2735 SetPixMap( meshSO, "ICON_SMESH_TREE_MESH_IMPORTED");
2737 SMESHDS_Mesh* newMeshDS = newMesh_i->GetImpl().GetMeshDS();
2738 ::SMESH_MeshEditor editor( &newMesh_i->GetImpl() );
2740 // 3. Get elements to copy
2742 SMDS_ElemIteratorPtr srcElemIt; SMDS_NodeIteratorPtr srcNodeIt;
2743 TIDSortedElemSet srcElems;
2744 SMESH::array_of_ElementType_var srcElemTypes = meshPart->GetTypes();
2745 if ( SMESH::DownCast<SMESH_Mesh_i*>( meshPart ))
2748 srcElemIt = srcMeshDS->elementsIterator();
2749 srcNodeIt = srcMeshDS->nodesIterator();
2753 SMESH::long_array_var ids = meshPart->GetIDs();
2754 if ( srcElemTypes->length() == 1 && srcElemTypes[0] == SMESH::NODE ) // group of nodes
2756 for (int i=0; i < ids->length(); i++)
2757 if ( const SMDS_MeshElement * elem = srcMeshDS->FindNode( ids[i] ))
2758 srcElems.insert( elem );
2762 for (int i=0; i < ids->length(); i++)
2763 if ( const SMDS_MeshElement * elem = srcMeshDS->FindElement( ids[i] ))
2764 srcElems.insert( elem );
2766 if ( srcElems.empty() )
2767 return newMesh._retn();
2769 typedef SMDS_SetIterator< SMDS_pElement, TIDSortedElemSet::const_iterator > ElIter;
2770 srcElemIt = SMDS_ElemIteratorPtr( new ElIter( srcElems.begin(), srcElems.end() ));
2775 typedef map<SMDS_pElement, SMDS_pElement, TIDCompare> TE2EMap;
2776 TE2EMap e2eMapByType[ SMDSAbs_NbElementTypes ];
2777 TE2EMap& n2nMap = e2eMapByType[ SMDSAbs_Node ];
2779 const SMDS_MeshNode *nSrc, *nTgt;
2780 vector< const SMDS_MeshNode* > nodes;
2781 while ( srcElemIt->more() )
2783 const SMDS_MeshElement * elem = srcElemIt->next();
2785 nodes.resize( elem->NbNodes());
2786 SMDS_ElemIteratorPtr nIt = elem->nodesIterator();
2788 for ( iN = 0; nIt->more(); ++iN )
2790 nSrc = static_cast<const SMDS_MeshNode*>( nIt->next() );
2791 nTgt = newMeshDS->FindNode( nSrc->GetID());
2793 nTgt = newMeshDS->AddNodeWithID( nSrc->X(), nSrc->Y(), nSrc->Z(), nSrc->GetID());
2798 for ( iN = 0; nIt->more(); ++iN )
2800 nSrc = static_cast<const SMDS_MeshNode*>( nIt->next() );
2801 TE2EMap::iterator n2n = n2nMap.insert( make_pair( nSrc, SMDS_pNode(0) )).first;
2803 n2n->second = newMeshDS->AddNode( nSrc->X(), nSrc->Y(), nSrc->Z() );
2804 nodes[ iN ] = (const SMDS_MeshNode*) n2n->second;
2808 if ( elem->GetType() != SMDSAbs_Node )
2810 int ID = toKeepIDs ? elem->GetID() : 0;
2811 const SMDS_MeshElement * newElem;
2812 switch ( elem->GetEntityType() ) {
2813 case SMDSEntity_Polyhedra:
2815 newElem = editor.GetMeshDS()->
2816 AddPolyhedralVolumeWithID( nodes,
2817 static_cast<const SMDS_VtkVolume*>(elem)->GetQuantities(),
2820 newElem = editor.GetMeshDS()->
2821 AddPolyhedralVolume( nodes,
2822 static_cast<const SMDS_VtkVolume*>(elem)->GetQuantities());
2824 case SMDSEntity_Ball:
2825 newElem = editor.AddElement( nodes, SMDSAbs_Ball, false, ID,
2826 static_cast<const SMDS_BallElement*>(elem)->GetDiameter());
2829 newElem = editor.AddElement( nodes,elem->GetType(),elem->IsPoly(),ID);
2831 if ( toCopyGroups && !toKeepIDs )
2832 e2eMapByType[ elem->GetType() ].insert( make_pair( elem, newElem ));
2835 } // while ( srcElemIt->more() )
2837 // 4(b). Copy free nodes
2839 if ( srcNodeIt && srcMeshDS->NbNodes() != newMeshDS->NbNodes() )
2841 while ( srcNodeIt->more() )
2843 nSrc = srcNodeIt->next();
2844 if ( nSrc->NbInverseElements() == 0 )
2847 nTgt = newMeshDS->AddNodeWithID( nSrc->X(), nSrc->Y(), nSrc->Z(), nSrc->GetID());
2849 n2nMap[ nSrc ] = newMeshDS->AddNode( nSrc->X(), nSrc->Y(), nSrc->Z() );
2856 int nbNewGroups = 0;
2859 SMESH_Mesh::GroupIteratorPtr gIt = srcMesh_i->GetImpl().GetGroups();
2860 while ( gIt->more() )
2862 SMESH_Group* group = gIt->next();
2863 const SMESHDS_GroupBase* groupDS = group->GetGroupDS();
2865 // Check group type. We copy nodal groups containing nodes of copied element
2866 SMDSAbs_ElementType groupType = groupDS->GetType();
2867 if ( groupType != SMDSAbs_Node &&
2868 newMeshDS->GetMeshInfo().NbElements( groupType ) == 0 )
2869 continue; // group type differs from types of meshPart
2871 // Find copied elements in the group
2872 vector< const SMDS_MeshElement* > groupElems;
2873 SMDS_ElemIteratorPtr eIt = groupDS->GetElements();
2876 const SMDS_MeshElement* foundElem;
2877 if ( groupType == SMDSAbs_Node )
2879 while ( eIt->more() )
2880 if (( foundElem = newMeshDS->FindNode( eIt->next()->GetID() )))
2881 groupElems.push_back( foundElem );
2885 while ( eIt->more() )
2886 if (( foundElem = newMeshDS->FindElement( eIt->next()->GetID() )))
2887 groupElems.push_back( foundElem );
2892 TE2EMap & e2eMap = e2eMapByType[ groupDS->GetType() ];
2893 if ( e2eMap.empty() ) continue;
2894 int minID = e2eMap.begin()->first->GetID();
2895 int maxID = e2eMap.rbegin()->first->GetID();
2896 TE2EMap::iterator e2e;
2897 while ( eIt->more() && groupElems.size() < e2eMap.size())
2899 const SMDS_MeshElement* e = eIt->next();
2900 if ( e->GetID() < minID || e->GetID() > maxID ) continue;
2901 if ((e2e = e2eMap.find( e )) != e2eMap.end())
2902 groupElems.push_back( e2e->second );
2906 if ( !groupElems.empty() )
2908 SMESH::SMESH_Group_var newGroupObj =
2909 newMesh->CreateGroup( SMESH::ElementType(groupType), group->GetName() );
2910 if ( SMESH_GroupBase_i* newGroup_i = SMESH::DownCast<SMESH_GroupBase_i*>( newGroupObj))
2912 SMESHDS_GroupBase * newGroupDS = newGroup_i->GetGroupDS();
2913 SMDS_MeshGroup& smdsGroup = ((SMESHDS_Group*)newGroupDS)->SMDSGroup();
2914 for ( unsigned i = 0; i < groupElems.size(); ++i )
2915 smdsGroup.Add( groupElems[i] );
2923 newMeshDS->Modified();
2925 *pyDump << newMesh << " = " << this
2926 << ".CopyMesh( " << meshPart << ", "
2927 << "'" << meshName << "', "
2928 << toCopyGroups << ", "
2929 << toKeepIDs << ")";
2931 delete pyDump; pyDump = 0; // allow dump in GetGroups()
2933 if ( nbNewGroups > 0 ) // dump created groups
2934 SMESH::ListOfGroups_var groups = newMesh->GetGroups();
2936 return newMesh._retn();
2939 //================================================================================
2941 * SMESH_Gen_i::GetMEDVersion
2943 * Get MED version of the file by its name
2945 //================================================================================
2946 CORBA::Boolean SMESH_Gen_i::GetMEDVersion(const char* theFileName,
2947 SMESH::MED_VERSION& theVersion)
2949 theVersion = SMESH::MED_V2_1;
2950 MED::EVersion aVersion = MED::GetVersionId( theFileName );
2951 switch( aVersion ) {
2952 case MED::eV2_1 : theVersion = SMESH::MED_V2_1; return true;
2953 case MED::eV2_2 : theVersion = SMESH::MED_V2_2; return true;
2954 case MED::eVUnknown : return false;
2959 //================================================================================
2961 * SMESH_Gen_i::GetMeshNames
2963 * Get names of meshes defined in file with the specified name
2965 //================================================================================
2966 SMESH::string_array* SMESH_Gen_i::GetMeshNames(const char* theFileName)
2968 SMESH::string_array_var aResult = new SMESH::string_array();
2969 MED::PWrapper aMed = MED::CrWrapper( theFileName );
2971 MED::TInt aNbMeshes = aMed->GetNbMeshes( &anErr );
2973 aResult->length( aNbMeshes );
2974 for( MED::TInt i = 0; i < aNbMeshes; i++ ) {
2975 MED::PMeshInfo aMeshInfo = aMed->GetPMeshInfo( i+1 );
2976 aResult[i] = CORBA::string_dup( aMeshInfo->GetName().c_str() );
2979 return aResult._retn();
2982 //=============================================================================
2986 * Save SMESH module's data
2988 //=============================================================================
2989 SALOMEDS::TMPFile* SMESH_Gen_i::Save( SALOMEDS::SComponent_ptr theComponent,
2993 INFOS( "SMESH_Gen_i::Save" );
2995 // ASSERT( theComponent->GetStudy()->StudyId() == myCurrentStudy->StudyId() )
2996 // san -- in case <myCurrentStudy> differs from theComponent's study,
2997 // use that of the component
2998 if ( theComponent->GetStudy()->StudyId() != GetCurrentStudyID() )
2999 SetCurrentStudy( theComponent->GetStudy() );
3001 // Store study contents as a set of python commands
3002 SavePython(myCurrentStudy);
3004 StudyContext* myStudyContext = GetCurrentStudyContext();
3006 // Declare a byte stream
3007 SALOMEDS::TMPFile_var aStreamFile;
3009 // Obtain a temporary dir
3010 TCollection_AsciiString tmpDir =
3011 ( isMultiFile ) ? TCollection_AsciiString( ( char* )theURL ) : ( char* )SALOMEDS_Tool::GetTmpDir().c_str();
3013 // Create a sequence of files processed
3014 SALOMEDS::ListOfFileNames_var aFileSeq = new SALOMEDS::ListOfFileNames;
3015 aFileSeq->length( NUM_TMP_FILES );
3017 TCollection_AsciiString aStudyName( "" );
3019 aStudyName = ( (char*)SALOMEDS_Tool::GetNameFromPath( myCurrentStudy->URL() ).c_str() );
3021 // Set names of temporary files
3022 TCollection_AsciiString filename =
3023 aStudyName + TCollection_AsciiString( "_SMESH.hdf" ); // for SMESH data itself
3024 TCollection_AsciiString meshfile =
3025 aStudyName + TCollection_AsciiString( "_SMESH_Mesh.med" ); // for mesh data to be stored in MED file
3026 aFileSeq[ 0 ] = CORBA::string_dup( filename.ToCString() );
3027 aFileSeq[ 1 ] = CORBA::string_dup( meshfile.ToCString() );
3028 filename = tmpDir + filename;
3029 meshfile = tmpDir + meshfile;
3032 HDFdataset* aDataset;
3033 HDFgroup* aTopGroup;
3035 HDFgroup* aSubGroup;
3036 HDFgroup* aSubSubGroup;
3037 hdf_size aSize[ 1 ];
3040 //Remove the files if they exist: BugID: 11225
3041 #ifndef WIN32 /* unix functionality */
3042 TCollection_AsciiString cmd("rm -f \"");
3044 TCollection_AsciiString cmd("del /F \"");
3051 system(cmd.ToCString());
3053 // MED writer to be used by storage process
3054 DriverMED_W_SMESHDS_Mesh myWriter;
3055 myWriter.SetFile( meshfile.ToCString() );
3058 // SetStoreName() to groups before storing hypotheses to let them refer to
3059 // groups using "store name", which is "Group <group_persistent_id>"
3061 SALOMEDS::ChildIterator_wrap itBig = myCurrentStudy->NewChildIterator( theComponent );
3062 for ( ; itBig->More(); itBig->Next() ) {
3063 SALOMEDS::SObject_wrap gotBranch = itBig->Value();
3064 if ( gotBranch->Tag() > GetAlgorithmsRootTag() ) {
3065 CORBA::Object_var anObject = SObjectToObject( gotBranch );
3066 if ( !CORBA::is_nil( anObject ) ) {
3067 SMESH::SMESH_Mesh_var myMesh = SMESH::SMESH_Mesh::_narrow( anObject ) ;
3068 if ( !myMesh->_is_nil() ) {
3069 myMesh->Load(); // load from study file if not yet done
3070 TPythonDump pd; // not to dump GetGroups()
3071 SMESH::ListOfGroups_var groups = myMesh->GetGroups();
3072 pd << ""; // to avoid optimizing pd out
3073 for ( int i = 0; i < groups->length(); ++i )
3075 SMESH_GroupBase_i* grImpl = SMESH::DownCast<SMESH_GroupBase_i*>( groups[i]);
3078 CORBA::String_var objStr = GetORB()->object_to_string( grImpl->_this() );
3079 int anId = myStudyContext->findId( string( objStr.in() ) );
3081 sprintf( grpName, "Group %d", anId );
3082 SMESHDS_GroupBase* aGrpBaseDS = grImpl->GetGroupDS();
3083 aGrpBaseDS->SetStoreName( grpName );
3093 // ---> create HDF file
3094 aFile = new HDFfile( (char*) filename.ToCString() );
3095 aFile->CreateOnDisk();
3097 // --> iterator for top-level objects
3098 SALOMEDS::ChildIterator_wrap itBig = myCurrentStudy->NewChildIterator( theComponent );
3099 for ( ; itBig->More(); itBig->Next() ) {
3100 SALOMEDS::SObject_wrap gotBranch = itBig->Value();
3102 // --> hypotheses root branch (only one for the study)
3103 if ( gotBranch->Tag() == GetHypothesisRootTag() ) {
3104 // create hypotheses root HDF group
3105 aTopGroup = new HDFgroup( "Hypotheses", aFile );
3106 aTopGroup->CreateOnDisk();
3108 // iterator for all hypotheses
3109 SALOMEDS::ChildIterator_wrap it = myCurrentStudy->NewChildIterator( gotBranch );
3110 for ( ; it->More(); it->Next() ) {
3111 SALOMEDS::SObject_wrap mySObject = it->Value();
3112 CORBA::Object_var anObject = SObjectToObject( mySObject );
3113 if ( !CORBA::is_nil( anObject ) ) {
3114 SMESH::SMESH_Hypothesis_var myHyp = SMESH::SMESH_Hypothesis::_narrow( anObject );
3115 if ( !myHyp->_is_nil() ) {
3116 SMESH_Hypothesis_i* myImpl = dynamic_cast<SMESH_Hypothesis_i*>( GetServant( myHyp ).in() );
3118 string hypname = string( myHyp->GetName() );
3119 string libname = string( myHyp->GetLibName() );
3121 // Needs for save crossplatform libname, i.e. parth of name ( ".dll" for
3122 // WIN32 and ".so" for X-system) must be deleted
3123 int libname_len = libname.length();
3125 if( libname_len > 4 )
3126 libname.resize( libname_len - 4 );