1 // Copyright (C) 2007-2016 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
26 #include <BRepPrimAPI_MakeBox.hxx>
27 #include <BRepPrimAPI_MakeCylinder.hxx>
28 #include <BRepPrimAPI_MakeSphere.hxx>
29 #include <BRep_Tool.hxx>
31 #include <TColStd_MapOfAsciiString.hxx>
32 #include <TCollection_AsciiString.hxx>
34 #include <TopExp_Explorer.hxx>
35 #include <TopTools_IndexedMapOfShape.hxx>
36 #include <TopTools_ListIteratorOfListOfShape.hxx>
37 #include <TopTools_ListOfShape.hxx>
38 #include <TopTools_MapOfShape.hxx>
40 #include <TopoDS_CompSolid.hxx>
41 #include <TopoDS_Compound.hxx>
42 #include <TopoDS_Edge.hxx>
43 #include <TopoDS_Face.hxx>
44 #include <TopoDS_Iterator.hxx>
45 #include <TopoDS_Shape.hxx>
46 #include <TopoDS_Shell.hxx>
47 #include <TopoDS_Solid.hxx>
48 #include <TopoDS_Vertex.hxx>
49 #include <TopoDS_Wire.hxx>
58 #include <libgen.h> // for basename function
62 #define LibHandle HMODULE
63 #define LoadLib( name ) LoadLibrary( name )
64 #define GetProc GetProcAddress
65 #define UnLoadLib( handle ) FreeLibrary( handle );
67 #define LibHandle void*
69 #define LoadLib( name ) dlopen( name, RTLD_LAZY | RTLD_LOCAL )
70 #else // DYNLOAD_LOCAL
71 #define LoadLib( name ) dlopen( name, RTLD_LAZY | RTLD_GLOBAL )
72 #endif // DYNLOAD_LOCAL
74 #define UnLoadLib( handle ) dlclose( handle );
77 #include "SMESH_Gen_i.hxx"
78 #include "SMESH_version.h"
80 #include "DriverMED_W_SMESHDS_Mesh.h"
81 #include "DriverMED_R_SMESHDS_Mesh.h"
83 #include "DriverCGNS_Read.hxx"
85 #include "MED_Factory.hxx"
86 #include "SMDS_EdgePosition.hxx"
87 #include "SMDS_FacePosition.hxx"
88 #include "SMDS_SetIterator.hxx"
89 #include "SMDS_SpacePosition.hxx"
90 #include "SMDS_VertexPosition.hxx"
91 #include "SMESHDS_Document.hxx"
92 #include "SMESHDS_Group.hxx"
93 #include "SMESHDS_GroupOnGeom.hxx"
94 #include "SMESH_Algo_i.hxx"
95 #include "SMESH_File.hxx"
96 #include "SMESH_Group.hxx"
97 #include "SMESH_Group_i.hxx"
98 #include "SMESH_Hypothesis.hxx"
99 #include "SMESH_Hypothesis_i.hxx"
100 #include "SMESH_Mesh.hxx"
101 #include "SMESH_MeshEditor.hxx"
102 #include "SMESH_Mesh_i.hxx"
103 #include "SMESH_PreMeshInfo.hxx"
104 #include "SMESH_PythonDump.hxx"
105 #include "SMESH_ControlsDef.hxx"
106 #include "SMESH_TryCatch.hxx" // to include after OCC headers!
108 #include CORBA_SERVER_HEADER(SMESH_Group)
109 #include CORBA_SERVER_HEADER(SMESH_Filter)
110 #include CORBA_SERVER_HEADER(SMESH_MeshEditor)
113 #include <GEOM_Client.hxx>
115 #include <Basics_Utils.hxx>
116 #include <Basics_DirUtils.hxx>
118 #include <OpUtil.hxx>
119 #include <SALOMEDS_Tool.hxx>
120 #include <SALOME_Container_i.hxx>
121 #include <SALOME_LifeCycleCORBA.hxx>
122 #include <SALOME_NamingService.hxx>
123 #include <Utils_CorbaException.hxx>
124 #include <Utils_ExceptHandlers.hxx>
125 #include <Utils_SINGLETON.hxx>
126 #include <utilities.h>
128 #include CORBA_CLIENT_HEADER(SALOME_ModuleCatalog)
129 #include CORBA_CLIENT_HEADER(SALOME_Session)
131 // helpers about SALOME::GenericObj
132 #include <SALOMEDS_wrap.hxx>
133 #include <SALOMEDS_Attributes_wrap.hxx>
134 #include <GEOM_wrap.hxx>
143 using SMESH::TPythonDump;
146 #define NUM_TMP_FILES 2
149 static int MYDEBUG = 0;
151 static int MYDEBUG = 0;
154 // Static variables definition
155 GEOM::GEOM_Gen_var SMESH_Gen_i::myGeomGen = GEOM::GEOM_Gen::_nil();
156 CORBA::ORB_var SMESH_Gen_i::myOrb;
157 PortableServer::POA_var SMESH_Gen_i::myPoa;
158 SALOME_NamingService* SMESH_Gen_i::myNS = NULL;
159 SALOME_LifeCycleCORBA* SMESH_Gen_i::myLCC = NULL;
160 SMESH_Gen_i* SMESH_Gen_i::mySMESHGen = NULL;
163 const int nbElemPerDiagonal = 10;
165 //=============================================================================
167 * GetServant [ static ]
169 * Get servant of the CORBA object
171 //=============================================================================
173 PortableServer::ServantBase_var SMESH_Gen_i::GetServant( CORBA::Object_ptr theObject )
175 if( CORBA::is_nil( theObject ) || CORBA::is_nil( GetPOA() ) )
178 PortableServer::Servant aServant = GetPOA()->reference_to_servant( theObject );
181 catch (PortableServer::POA::ObjectNotActive &ex)
183 INFOS("GetServant: ObjectNotActive");
186 catch (PortableServer::POA::WrongAdapter &ex)
188 INFOS("GetServant: WrongAdapter: OK when several servants used to build several mesh in parallel...");
191 catch (PortableServer::POA::WrongPolicy &ex)
193 INFOS("GetServant: WrongPolicy");
198 INFOS( "GetServant - Unknown exception was caught!!!" );
203 //=============================================================================
205 * SObjectToObject [ static ]
207 * Get CORBA object corresponding to the SALOMEDS::SObject
209 //=============================================================================
211 CORBA::Object_var SMESH_Gen_i::SObjectToObject( SALOMEDS::SObject_ptr theSObject )
213 SALOMEDS::GenericAttribute_wrap anAttr;
214 CORBA::Object_var anObj;
215 if ( !theSObject->_is_nil() ) {
217 if( theSObject->FindAttribute( anAttr.inout(), "AttributeIOR" ) ) {
218 SALOMEDS::AttributeIOR_wrap anIOR = anAttr;
219 CORBA::String_var aValue = anIOR->Value();
220 if( strcmp( aValue, "" ) != 0 )
221 anObj = GetORB()->string_to_object( aValue );
225 INFOS( "SObjectToObject - Unknown exception was caught!!!" );
231 //=============================================================================
235 * Get SALOME_NamingService object
237 //=============================================================================
239 SALOME_NamingService* SMESH_Gen_i::GetNS()
241 if ( myNS == NULL ) {
242 myNS = SINGLETON_<SALOME_NamingService>::Instance();
243 ASSERT(SINGLETON_<SALOME_NamingService>::IsAlreadyExisting());
244 myNS->init_orb( GetORB() );
249 //=============================================================================
253 * Get SALOME_LifeCycleCORBA object
255 //=============================================================================
256 SALOME_LifeCycleCORBA* SMESH_Gen_i::GetLCC() {
257 if ( myLCC == NULL ) {
258 myLCC = new SALOME_LifeCycleCORBA( GetNS() );
264 //=============================================================================
266 * GetGeomEngine [ static ]
268 * Get GEOM::GEOM_Gen reference
270 //=============================================================================
271 GEOM::GEOM_Gen_var SMESH_Gen_i::GetGeomEngine() {
272 //CCRT GEOM::GEOM_Gen_var aGeomEngine =
273 //CCRT GEOM::GEOM_Gen::_narrow( GetLCC()->FindOrLoad_Component("FactoryServer","GEOM") );
274 //CCRT return aGeomEngine._retn();
275 if(CORBA::is_nil(myGeomGen))
277 Engines::EngineComponent_ptr temp=GetLCC()->FindOrLoad_Component("FactoryServer","GEOM");
278 myGeomGen=GEOM::GEOM_Gen::_narrow(temp);
283 //=============================================================================
285 * SMESH_Gen_i::SMESH_Gen_i
287 * Default constructor: not for use
289 //=============================================================================
291 SMESH_Gen_i::SMESH_Gen_i()
295 //=============================================================================
297 * SMESH_Gen_i::SMESH_Gen_i
299 * Standard constructor, used with Container
301 //=============================================================================
303 SMESH_Gen_i::SMESH_Gen_i( CORBA::ORB_ptr orb,
304 PortableServer::POA_ptr poa,
305 PortableServer::ObjectId* contId,
306 const char* instanceName,
307 const char* interfaceName )
308 : Engines_Component_i( orb, poa, contId, instanceName, interfaceName )
311 myOrb = CORBA::ORB::_duplicate(orb);
312 myPoa = PortableServer::POA::_duplicate(poa);
315 _id = myPoa->activate_object( _thisObj );
317 myStudyContext = new StudyContext;
319 myIsEmbeddedMode = false;
320 myIsEnablePublish = true;
321 myShapeReader = NULL; // shape reader
323 myIsHistoricalPythonDump = true;
324 myToForgetMeshDataOnHypModif = false;
326 // set it in standalone mode only
327 //OSD::SetSignal( true );
329 // 0020605: EDF 1190 SMESH: Display performance. 80 seconds for 52000 cells.
330 // find out mode (embedded or standalone) here else
331 // meshes created before calling SMESH_Client::GetSMESHGen(), which calls
332 // SMESH_Gen_i::SetEmbeddedMode(), have wrong IsEmbeddedMode flag
333 if ( SALOME_NamingService* ns = GetNS() )
335 CORBA::Object_var obj = ns->Resolve( "/Kernel/Session" );
336 SALOME::Session_var session = SALOME::Session::_narrow( obj ) ;
337 if ( !session->_is_nil() )
339 CORBA::String_var str_host = session->getHostname();
340 CORBA::Long s_pid = session->getPID();
341 string my_host = Kernel_Utils::GetHostname();
343 long my_pid = (long)_getpid();
345 long my_pid = (long) getpid();
347 SetEmbeddedMode( s_pid == my_pid && my_host == str_host.in() );
352 //=============================================================================
354 * SMESH_Gen_i::~SMESH_Gen_i
358 //=============================================================================
360 SMESH_Gen_i::~SMESH_Gen_i()
362 // delete hypothesis creators
363 map<string, GenericHypothesisCreator_i*>::iterator itHyp, itHyp2;
364 for (itHyp = myHypCreatorMap.begin(); itHyp != myHypCreatorMap.end(); itHyp++)
366 // same creator can be mapped under different names
367 GenericHypothesisCreator_i* creator = (*itHyp).second;
371 for (itHyp2 = itHyp; itHyp2 != myHypCreatorMap.end(); itHyp2++)
372 if ( creator == (*itHyp2).second )
373 (*itHyp2).second = 0;
375 myHypCreatorMap.clear();
377 // Clear study contexts data
378 delete myStudyContext;
380 // delete shape reader
382 delete myShapeReader;
384 //=============================================================================
386 * SMESH_Gen_i::getHypothesisCreator
388 * Get hypothesis creator
390 //=============================================================================
391 GenericHypothesisCreator_i* SMESH_Gen_i::getHypothesisCreator(const char* theHypName,
392 const char* theLibName,
393 std::string& thePlatformLibName)
394 throw (SALOME::SALOME_Exception)
396 std::string aPlatformLibName;
397 /* It's Need to translate lib name for WIN32 or X platform */
398 if ( theLibName && theLibName[0] != '\0' )
400 int libNameLen = strlen(theLibName);
401 //check for old format "libXXXXXXX.so"
402 if (libNameLen > 7 &&
403 !strncmp( theLibName, "lib", 3 ) &&
404 !strcmp( theLibName+libNameLen-3, ".so" ))
408 aPlatformLibName = std::string( theLibName+3, libNameLen-6 ) + ".dll";
409 #elif defined(__APPLE__)
410 aPlatformLibName = std::string( theLibName, libNameLen-3 ) + ".dylib";
412 aPlatformLibName = theLibName;
417 //try to use new format
419 aPlatformLibName = theLibName;
420 aPlatformLibName += ".dll";
421 #elif defined(__APPLE__)
422 aPlatformLibName = std::string( "lib" ) + std::string( theLibName ) + ".dylib";
424 aPlatformLibName = std::string( "lib" ) + std::string( theLibName ) + ".so";
428 thePlatformLibName = aPlatformLibName;
430 Unexpect aCatch(SALOME_SalomeException);
431 if(MYDEBUG) MESSAGE( "Create Hypothesis <" << theHypName << "> from " << aPlatformLibName);
433 typedef GenericHypothesisCreator_i* (*GetHypothesisCreator)(const char* );
434 GenericHypothesisCreator_i* aCreator;
437 // check, if creator for this hypothesis type already exists
438 if (myHypCreatorMap.find(string(theHypName)) == myHypCreatorMap.end())
440 // load plugin library
441 if(MYDEBUG) MESSAGE("Loading server meshers plugin library ...");
442 LibHandle libHandle = LoadLib( aPlatformLibName.c_str() );
445 // report any error, if occurred
447 const char* anError = dlerror();
448 throw(SALOME_Exception(anError));
450 throw(SALOME_Exception(LOCALIZED( "Can't load server meshers plugin library" )));
454 // get method, returning hypothesis creator
455 if(MYDEBUG) MESSAGE("Find GetHypothesisCreator() method ...");
456 GetHypothesisCreator procHandle =
457 (GetHypothesisCreator)GetProc( libHandle, "GetHypothesisCreator" );
460 throw(SALOME_Exception(LOCALIZED("bad hypothesis plugin library")));
461 UnLoadLib(libHandle);
464 // get hypothesis creator
465 if(MYDEBUG) MESSAGE("Get Hypothesis Creator for " << theHypName);
466 aCreator = procHandle(theHypName);
469 throw(SALOME_Exception(LOCALIZED("no such a hypothesis in this plugin")));
471 // map hypothesis creator to a hypothesis name
472 myHypCreatorMap[string(theHypName)] = aCreator;
477 return myHypCreatorMap[string(theHypName)];
480 catch (SALOME_Exception& S_ex)
482 THROW_SALOME_CORBA_EXCEPTION(S_ex.what(), SALOME::BAD_PARAM);
487 //=============================================================================
489 * SMESH_Gen_i::createHypothesis
491 * Create hypothesis of given type
493 //=============================================================================
494 SMESH::SMESH_Hypothesis_ptr SMESH_Gen_i::createHypothesis(const char* theHypName,
495 const char* theLibName)
497 SMESH_Hypothesis_i* myHypothesis_i = 0;
498 SMESH::SMESH_Hypothesis_var hypothesis_i;
499 std::string aPlatformLibName;
500 GenericHypothesisCreator_i* aCreator =
501 getHypothesisCreator(theHypName, theLibName, aPlatformLibName);
503 // create a new hypothesis object, store its ref. in studyContext
504 myHypothesis_i = aCreator->Create(myPoa, &myGen);
507 myHypothesis_i->SetLibName( aPlatformLibName.c_str() ); // for persistency assurance
508 CORBA::String_var hypName = myHypothesis_i->GetName();
509 myHypCreatorMap[ hypName.in() ] = aCreator;
511 // activate the CORBA servant of hypothesis
512 hypothesis_i = myHypothesis_i->_this();
513 int nextId = RegisterObject( hypothesis_i );
514 if(MYDEBUG) { MESSAGE( "Add hypo to map with id = "<< nextId ); }
515 else { nextId = 0; } // avoid "unused variable" warning in release mode
517 return hypothesis_i._retn();
520 //=============================================================================
522 * SMESH_Gen_i::createMesh
524 * Create empty mesh on shape
526 //=============================================================================
527 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::createMesh()
528 throw ( SALOME::SALOME_Exception )
530 Unexpect aCatch(SALOME_SalomeException);
531 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::createMesh" );
533 // Get or create the GEOM_Client instance
535 // create a new mesh object servant, store it in a map in study context
536 SMESH_Mesh_i* meshServant = new SMESH_Mesh_i( GetPOA(), this );
537 // create a new mesh object
538 if(MYDEBUG) MESSAGE("myIsEmbeddedMode " << myIsEmbeddedMode);
539 meshServant->SetImpl( myGen.CreateMesh( myIsEmbeddedMode ));
541 // activate the CORBA servant of Mesh
542 SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( meshServant->_this() );
543 int nextId = RegisterObject( mesh );
544 if(MYDEBUG) { MESSAGE( "Add mesh to map with id = "<< nextId); }
545 else { nextId = 0; } // avoid "unused variable" warning in release mode
548 catch (SALOME_Exception& S_ex) {
549 THROW_SALOME_CORBA_EXCEPTION( S_ex.what(), SALOME::BAD_PARAM );
551 return SMESH::SMESH_Mesh::_nil();
554 //=============================================================================
556 * SMESH_Gen_i::GetShapeReader
560 //=============================================================================
561 GEOM_Client* SMESH_Gen_i::GetShapeReader()
563 // create shape reader if necessary
564 if ( !myShapeReader )
565 myShapeReader = new GEOM_Client(GetContainerRef());
566 ASSERT( myShapeReader );
567 return myShapeReader;
570 //=============================================================================
572 * SMESH_Gen_i::SetGeomEngine
574 * Set GEOM::GEOM_Gen reference
576 //=============================================================================
577 //GEOM::GEOM_Gen_ptr SMESH_Gen_i::SetGeomEngine( const char* containerLoc )
578 void SMESH_Gen_i::SetGeomEngine( GEOM::GEOM_Gen_ptr geomcompo )
580 //Engines::Component_ptr temp=GetLCC()->FindOrLoad_Component(containerLoc,"GEOM");
581 //myGeomGen=GEOM::GEOM_Gen::_narrow(temp);
582 myGeomGen=GEOM::GEOM_Gen::_duplicate(geomcompo);
586 //=============================================================================
588 * SMESH_Gen_i::SetEmbeddedMode
592 //=============================================================================
594 void SMESH_Gen_i::SetEmbeddedMode( CORBA::Boolean theMode )
596 myIsEmbeddedMode = theMode;
598 if ( !myIsEmbeddedMode ) {
599 //PAL10867: disable signals catching with "noexcepthandler" option
600 char* envNoCatchSignals = getenv("NOT_INTERCEPT_SIGNALS");
601 if (!envNoCatchSignals || !atoi(envNoCatchSignals))
606 char* envDisableFPE = getenv("DISABLE_FPE");
607 if (envDisableFPE && atoi(envDisableFPE))
612 OSD::SetSignal( raiseFPE );
614 // else OSD::SetSignal() is called in GUI
618 //=============================================================================
620 * SMESH_Gen_i::IsEmbeddedMode
624 //=============================================================================
626 CORBA::Boolean SMESH_Gen_i::IsEmbeddedMode()
628 return myIsEmbeddedMode;
631 //=============================================================================
633 * SMESH_Gen_i::SetEnablePublish
635 * Set enable publishing in the study
637 //=============================================================================
638 void SMESH_Gen_i::SetEnablePublish( CORBA::Boolean theIsEnablePublish )
640 myIsEnablePublish = theIsEnablePublish;
643 //=============================================================================
645 * SMESH_Gen_i::IsEnablePublish
647 * Check enable publishing
649 //=============================================================================
651 CORBA::Boolean SMESH_Gen_i::IsEnablePublish()
653 return myIsEnablePublish;
656 //=============================================================================
658 * SMESH_Gen_i::UpdateStudy
660 * Update study (needed at switching GEOM->SMESH)
662 //=============================================================================
664 void SMESH_Gen_i::UpdateStudy()
666 if ( !myStudyContext )
667 myStudyContext = new StudyContext;
669 SALOMEDS::Study_var aStudy = getStudyServant();
670 if ( !CORBA::is_nil( aStudy ) ) {
671 SALOMEDS::StudyBuilder_var aStudyBuilder = aStudy->NewBuilder();
672 SALOMEDS::SComponent_wrap GEOM_var = aStudy->FindComponent( "GEOM" );
673 if( !GEOM_var->_is_nil() )
674 aStudyBuilder->LoadWith( GEOM_var, GetGeomEngine() );
675 // NPAL16168, issue 0020210
676 // Let meshes update their data depending on GEOM groups that could change
677 CORBA::String_var compDataType = ComponentDataType();
678 SALOMEDS::SComponent_wrap me = aStudy->FindComponent( compDataType.in() );
679 if ( !me->_is_nil() ) {
680 SALOMEDS::ChildIterator_wrap anIter = aStudy->NewChildIterator( me );
681 for ( ; anIter->More(); anIter->Next() ) {
682 SALOMEDS::SObject_wrap so = anIter->Value();
683 CORBA::Object_var ior = SObjectToObject( so );
684 if ( SMESH_Mesh_i* mesh = SMESH::DownCast<SMESH_Mesh_i*>( ior ))
685 mesh->CheckGeomModif();
691 //=============================================================================
693 * SMESH_Gen_i::GetStudyContext
697 //=============================================================================
698 StudyContext* SMESH_Gen_i::GetStudyContext()
700 return myStudyContext;
703 //=============================================================================
705 * SMESH_Gen_i::CreateHypothesis
707 * Create hypothesis/algorothm of given type and publish it in the study
709 //=============================================================================
711 SMESH::SMESH_Hypothesis_ptr SMESH_Gen_i::CreateHypothesis( const char* theHypName,
712 const char* theLibName )
713 throw ( SALOME::SALOME_Exception )
715 Unexpect aCatch(SALOME_SalomeException);
716 // Create hypothesis/algorithm
717 SMESH::SMESH_Hypothesis_var hyp = this->createHypothesis( theHypName, theLibName );
719 // Publish hypothesis/algorithm in the study
720 if ( CanPublishInStudy( hyp ) ) {
721 SALOMEDS::SObject_wrap aSO = PublishHypothesis( hyp );
722 if ( !aSO->_is_nil() ) {
723 // Update Python script
724 TPythonDump() << aSO << " = " << this << ".CreateHypothesis('"
725 << theHypName << "', '" << theLibName << "')";
732 //================================================================================
734 * \brief Return a hypothesis holding parameter values corresponding either to the mesh
735 * existing on the given geometry or to size of the geometry.
736 * \param theHypType - hypothesis type name
737 * \param theLibName - plugin library name
738 * \param theMesh - The mesh of interest
739 * \param theGeom - The shape to get parameter values from
740 * \retval SMESH::SMESH_Hypothesis_ptr - The returned hypothesis may be the one existing
741 * in a study and used to compute the mesh, or a temporary one created just to pass
744 //================================================================================
746 SMESH::SMESH_Hypothesis_ptr
747 SMESH_Gen_i::GetHypothesisParameterValues (const char* theHypType,
748 const char* theLibName,
749 SMESH::SMESH_Mesh_ptr theMesh,
750 GEOM::GEOM_Object_ptr theGeom,
751 CORBA::Boolean byMesh)
752 throw ( SALOME::SALOME_Exception )
754 Unexpect aCatch(SALOME_SalomeException);
755 if ( byMesh && CORBA::is_nil( theMesh ) )
756 return SMESH::SMESH_Hypothesis::_nil();
757 if ( byMesh && CORBA::is_nil( theGeom ) )
758 return SMESH::SMESH_Hypothesis::_nil();
760 // -----------------------------------------------
761 // find hypothesis used to mesh theGeom
762 // -----------------------------------------------
764 // get mesh and shape
765 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
766 TopoDS_Shape shape = GeomObjectToShape( theGeom );
767 if ( byMesh && ( !meshServant || meshServant->NbNodes()==0 || shape.IsNull() ))
768 return SMESH::SMESH_Hypothesis::_nil();
769 ::SMESH_Mesh* mesh = meshServant ? &meshServant->GetImpl() : (::SMESH_Mesh*)0;
771 // create a temporary hypothesis to know its dimension
772 SMESH::SMESH_Hypothesis_var tmpHyp = this->createHypothesis( theHypType, theLibName );
773 SMESH_Hypothesis_i* hypServant = SMESH::DownCast<SMESH_Hypothesis_i*>( tmpHyp );
775 return SMESH::SMESH_Hypothesis::_nil();
776 ::SMESH_Hypothesis* hyp = hypServant->GetImpl();
779 // look for a hypothesis of theHypType used to mesh the shape
780 if ( myGen.GetShapeDim( shape ) == hyp->GetDim() )
783 SMESH::ListOfHypothesis_var aHypList = theMesh->GetHypothesisList( theGeom );
784 int nbLocalHyps = aHypList->length();
785 for ( int i = 0; i < nbLocalHyps; i++ ) {
786 CORBA::String_var hypName = aHypList[i]->GetName();
787 if ( strcmp( theHypType, hypName.in() ) == 0 ) // FOUND local!
788 return SMESH::SMESH_Hypothesis::_duplicate( aHypList[i] );
790 // check super shapes
791 TopTools_ListIteratorOfListOfShape itShape( mesh->GetAncestors( shape ));
792 while ( nbLocalHyps == 0 && itShape.More() ) {
793 GEOM::GEOM_Object_ptr geomObj = ShapeToGeomObject( itShape.Value() );
794 if ( ! CORBA::is_nil( geomObj )) {
795 SMESH::ListOfHypothesis_var aHypList = theMesh->GetHypothesisList( geomObj );
796 nbLocalHyps = aHypList->length();
797 for ( int i = 0; i < nbLocalHyps; i++ )
798 if ( strcmp( theHypType, aHypList[i]->GetName() ) == 0 ) // FOUND global!
799 return SMESH::SMESH_Hypothesis::_duplicate( aHypList[i] );
805 // let the temporary hypothesis find out somehow parameter values by mesh
806 if ( hyp->SetParametersByMesh( mesh, shape ))
807 return SMESH::SMESH_Hypothesis::_duplicate( tmpHyp );
812 diagonal = mesh->GetShapeDiagonalSize();
814 diagonal = ::SMESH_Mesh::GetShapeDiagonalSize( shape );
815 ::SMESH_Hypothesis::TDefaults dflts;
816 dflts._elemLength = diagonal / myGen.GetBoundaryBoxSegmentation();
817 dflts._nbSegments = myGen.GetDefaultNbSegments();
818 dflts._shape = &shape;
819 // let the temporary hypothesis initialize it's values
820 if ( hyp->SetParametersByDefaults( dflts, mesh ))
821 return SMESH::SMESH_Hypothesis::_duplicate( tmpHyp );
824 return SMESH::SMESH_Hypothesis::_nil();
827 //=============================================================================
829 * Returns \c True if a hypothesis is assigned to a sole sub-mesh in a current Study
830 * \param [in] theHyp - the hypothesis of interest
831 * \param [out] theMesh - the sole mesh using \a theHyp
832 * \param [out] theShape - the sole geometry \a theHyp is assigned to
833 * \return boolean - \c True if \a theMesh and \a theShape are sole using \a theHyp
835 * If two meshes on same shape have theHyp assigned to the same sub-shape, they are
836 * considered as SAME sub-mesh => result is \c true.
837 * This method ids used to initialize SMESHGUI_GenericHypothesisCreator with
838 * a shape to which an hyp being edited is assigned.
840 //=============================================================================
842 CORBA::Boolean SMESH_Gen_i::GetSoleSubMeshUsingHyp( SMESH::SMESH_Hypothesis_ptr theHyp,
843 SMESH::SMESH_Mesh_out theMesh,
844 GEOM::GEOM_Object_out theShape)
846 if ( CORBA::is_nil( theHyp ))
849 // get Mesh component SO
850 CORBA::String_var compDataType = ComponentDataType();
851 SALOMEDS::SComponent_wrap comp = getStudyServant()->FindComponent( compDataType.in() );
852 if ( CORBA::is_nil( comp ))
855 // look for child SO of meshes
856 SMESH::SMESH_Mesh_var foundMesh;
857 TopoDS_Shape foundShape;
859 SALOMEDS::ChildIterator_wrap meshIter = getStudyServant()->NewChildIterator( comp );
860 for ( ; meshIter->More() && isSole; meshIter->Next() )
862 SALOMEDS::SObject_wrap curSO = meshIter->Value();
863 CORBA::Object_var obj = SObjectToObject( curSO );
864 SMESH_Mesh_i* mesh_i = SMESH::DownCast< SMESH_Mesh_i* >( obj );
868 // look for a sole shape where theHyp is assigned
869 bool isHypFound = false;
870 const ShapeToHypothesis & s2hyps = mesh_i->GetImpl().GetMeshDS()->GetHypotheses();
871 ShapeToHypothesis::Iterator s2hypsIt( s2hyps );
872 for ( ; s2hypsIt.More() && isSole; s2hypsIt.Next() )
874 const THypList& hyps = s2hypsIt.Value();
875 THypList::const_iterator h = hyps.begin();
876 for ( ; h != hyps.end(); ++h )
877 if ( (*h)->GetID() == theHyp->GetId() )
879 if ( h != hyps.end()) // theHyp found
882 if ( ! foundShape.IsNull() &&
883 ! foundShape.IsSame( s2hypsIt.Key() )) // not a sole sub-shape
885 foundShape.Nullify();
889 foundShape = s2hypsIt.Key();
891 } // loop on assigned hyps
893 if ( isHypFound && !foundShape.IsNull() ) // a mesh using theHyp is found
895 if ( !foundMesh->_is_nil() ) // not a sole mesh
897 if ( !foundMesh->HasShapeToMesh() ||
898 !mesh_i ->HasShapeToMesh() )
900 isSole = ( foundMesh->HasShapeToMesh() == mesh_i->HasShapeToMesh() );
904 GEOM::GEOM_Object_var s1 = mesh_i ->GetShapeToMesh();
905 GEOM::GEOM_Object_var s2 = foundMesh->GetShapeToMesh();
906 isSole = s1->IsSame( s2 );
909 foundMesh = SMESH::SMESH_Mesh::_narrow( obj );
915 ! foundMesh->_is_nil() &&
916 ! foundShape.IsNull() )
918 theMesh = foundMesh._retn();
919 theShape = ShapeToGeomObject( foundShape );
920 return ( !theMesh->_is_nil() && !theShape->_is_nil() );
925 //=============================================================================
927 * Sets number of segments per diagonal of boundary box of geometry by which
928 * default segment length of appropriate 1D hypotheses is defined
930 //=============================================================================
932 void SMESH_Gen_i::SetBoundaryBoxSegmentation( CORBA::Long theNbSegments )
933 throw ( SALOME::SALOME_Exception )
935 if ( theNbSegments > 0 )
936 myGen.SetBoundaryBoxSegmentation( int( theNbSegments ));
938 THROW_SALOME_CORBA_EXCEPTION( "non-positive number of segments", SALOME::BAD_PARAM );
940 //=============================================================================
942 * \brief Sets default number of segments per edge
944 //=============================================================================
945 void SMESH_Gen_i::SetDefaultNbSegments(CORBA::Long theNbSegments)
946 throw ( SALOME::SALOME_Exception )
948 if ( theNbSegments > 0 )
949 myGen.SetDefaultNbSegments( int(theNbSegments) );
951 THROW_SALOME_CORBA_EXCEPTION( "non-positive number of segments", SALOME::BAD_PARAM );
954 //=============================================================================
958 //=============================================================================
960 void SMESH_Gen_i::SetOption(const char* name, const char* value)
962 if ( name && value && strlen( value ) > 0 )
965 if ( strcmp(name, "historical_python_dump") == 0 )
967 myIsHistoricalPythonDump = ( value[0] == '1' || toupper(value[0]) == 'T' ); // 1 || true
968 msgToGUI = "preferences/SMESH/historical_python_dump/";
969 msgToGUI += myIsHistoricalPythonDump ? "true" : "false";
971 else if ( strcmp(name, "forget_mesh_on_hyp_modif") == 0 )
973 myToForgetMeshDataOnHypModif = ( value[0] == '1' || toupper(value[0]) == 'T' ); // 1 || true
974 msgToGUI = "preferences/SMESH/forget_mesh_on_hyp_modif/";
975 msgToGUI += myToForgetMeshDataOnHypModif ? "true" : "false";
977 else if ( strcmp(name, "default_grp_color") == 0 )
981 // color must be presented as a string of following form:
982 if ( str.at(0) == '#' && str.length() == 7 ) { // hexadecimal color ("#ffaa00", for example)
984 for ( size_t i = 0; i < str.length()/2; i++ )
985 if ( str.at(i*2) >= '0' && str.at(i*2) <= 'f' && str.at(i*2+1) >= '0' && str.at(i*2+1) <= 'f' )
986 color.push_back( strtol( str.substr( i*2, 2 ).c_str(), NULL, 16 ) );
988 else if ( value ) { // rgb color ("255,170,0", for example)
989 string tempValue( value );
990 char* colorValue = strtok( &tempValue[0], "," );
991 while ( colorValue != NULL ) {
992 int c_value = atoi( colorValue );
993 if ( c_value >= 0 && c_value <= 255 )
994 color.push_back( c_value );
995 colorValue = strtok( NULL, "," );
998 if ( color.size() == 3 ) { // color must have three valid component
999 SMESHDS_GroupBase::SetDefaultColor( Quantity_Color( color[0]/255., color[1]/255., color[2]/255., Quantity_TOC_RGB ) );
1000 myDefaultGroupColor = value;
1001 msgToGUI = "preferences/SMESH/default_grp_color/";
1006 // update preferences in case if SetOption() is invoked from python console
1007 if ( !msgToGUI.empty() )
1009 CORBA::Object_var obj = SMESH_Gen_i::GetNS()->Resolve( "/Kernel/Session" );
1010 SALOME::Session_var session = SALOME::Session::_narrow( obj );
1011 if ( !CORBA::is_nil( session ) )
1012 session->emitMessageOneWay(msgToGUI.c_str());
1017 //=============================================================================
1019 Return an option value
1021 //=============================================================================
1023 char* SMESH_Gen_i::GetOption(const char* name)
1027 if ( strcmp(name, "historical_python_dump") == 0 )
1029 return CORBA::string_dup( myIsHistoricalPythonDump ? "true" : "false" );
1031 if ( strcmp(name, "forget_mesh_on_hyp_modif") == 0 )
1033 return CORBA::string_dup( myToForgetMeshDataOnHypModif ? "true" : "false" );
1035 if ( strcmp(name, "default_grp_color") == 0 )
1037 return CORBA::string_dup( myDefaultGroupColor.c_str() );
1040 return CORBA::string_dup( "" );
1043 //=============================================================================
1045 * SMESH_Gen_i::CreateMesh
1047 * Create empty mesh on a shape and publish it in the study
1049 //=============================================================================
1051 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMesh( GEOM::GEOM_Object_ptr theShapeObject )
1052 throw ( SALOME::SALOME_Exception )
1054 Unexpect aCatch(SALOME_SalomeException);
1055 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::CreateMesh" );
1057 SMESH::SMESH_Mesh_var mesh = this->createMesh();
1059 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( mesh );
1060 ASSERT( meshServant );
1061 meshServant->SetShape( theShapeObject );
1063 // publish mesh in the study
1064 if ( CanPublishInStudy( mesh ) ) {
1065 SALOMEDS::StudyBuilder_var aStudyBuilder = getStudyServant()->NewBuilder();
1066 aStudyBuilder->NewCommand(); // There is a transaction
1067 SALOMEDS::SObject_wrap aSO = PublishMesh( mesh.in() );
1068 aStudyBuilder->CommitCommand();
1069 if ( !aSO->_is_nil() ) {
1070 // Update Python script
1071 TPythonDump() << aSO << " = " << this << ".CreateMesh(" << theShapeObject << ")";
1075 return mesh._retn();
1078 //=============================================================================
1080 * SMESH_Gen_i::CreateEmptyMesh
1084 //=============================================================================
1086 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateEmptyMesh()
1087 throw ( SALOME::SALOME_Exception )
1089 Unexpect aCatch(SALOME_SalomeException);
1090 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::CreateMesh" );
1092 SMESH::SMESH_Mesh_var mesh = this->createMesh();
1094 // publish mesh in the study
1095 if ( CanPublishInStudy( mesh ) ) {
1096 SALOMEDS::StudyBuilder_var aStudyBuilder = getStudyServant()->NewBuilder();
1097 aStudyBuilder->NewCommand(); // There is a transaction
1098 SALOMEDS::SObject_wrap aSO = PublishMesh( mesh.in() );
1099 aStudyBuilder->CommitCommand();
1100 if ( !aSO->_is_nil() ) {
1101 // Update Python script
1102 TPythonDump() << aSO << " = " << this << ".CreateEmptyMesh()";
1106 return mesh._retn();
1111 //================================================================================
1113 * \brief Throws an exception in case if the file can't be read
1115 //================================================================================
1117 void checkFileReadable( const char* theFileName ) throw ( SALOME::SALOME_Exception )
1119 SMESH_File f ( theFileName );
1122 if ( !f.error().empty() )
1123 THROW_SALOME_CORBA_EXCEPTION( f.error().c_str(), SALOME::BAD_PARAM);
1125 THROW_SALOME_CORBA_EXCEPTION
1126 (( SMESH_Comment("Can't open for reading the file ") << theFileName ).c_str(),
1127 SALOME::BAD_PARAM );
1132 //=============================================================================
1134 * SMESH_Gen_i::CreateMeshFromUNV
1136 * Create mesh and import data from UNV file
1138 //=============================================================================
1140 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMeshesFromUNV( const char* theFileName )
1141 throw ( SALOME::SALOME_Exception )
1143 Unexpect aCatch(SALOME_SalomeException);
1145 checkFileReadable( theFileName );
1147 SMESH::SMESH_Mesh_var aMesh = createMesh();
1149 // publish mesh in the study
1150 if ( CanPublishInStudy( aMesh ) ) {
1151 SALOMEDS::StudyBuilder_var aStudyBuilder = getStudyServant()->NewBuilder();
1152 aStudyBuilder->NewCommand(); // There is a transaction
1153 SALOMEDS::SObject_wrap aSO = PublishMesh( aMesh.in(), aFileName.c_str() );
1154 aStudyBuilder->CommitCommand();
1155 if ( !aSO->_is_nil() ) {
1156 // Update Python script
1157 TPythonDump() << aSO << " = " << this << ".CreateMeshesFromUNV(r'" << theFileName << "')";
1161 SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
1163 aServant->ImportUNVFile( theFileName );
1165 // Dump creation of groups
1166 SMESH::ListOfGroups_var groups = aServant->GetGroups();
1168 aServant->GetImpl().GetMeshDS()->Modified();
1169 return aMesh._retn();
1172 //=============================================================================
1174 * SMESH_Gen_i::CreateMeshFromMED
1176 * Create mesh and import data from MED file
1178 //=============================================================================
1180 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromMEDorSAUV( const char* theFileName,
1181 SMESH::DriverMED_ReadStatus& theStatus,
1182 const char* theCommandNameForPython,
1183 const char* theFileNameForPython)
1186 char bname[ _MAX_FNAME ];
1187 _splitpath( theFileNameForPython, NULL, NULL, bname, NULL );
1188 string aFileName = bname;
1190 string aFileName = basename( const_cast<char *>(theFileNameForPython) );
1192 // Retrieve mesh names from the file
1193 DriverMED_R_SMESHDS_Mesh myReader;
1194 myReader.SetFile( theFileName );
1195 myReader.SetMeshId( -1 );
1196 Driver_Mesh::Status aStatus;
1197 list<string> aNames = myReader.GetMeshNames(aStatus);
1198 SMESH::mesh_array_var aResult = new SMESH::mesh_array();
1199 theStatus = (SMESH::DriverMED_ReadStatus)aStatus;
1201 { // open a new scope to make aPythonDump die before PythonDump in SMESH_Mesh::GetGroups()
1204 TPythonDump aPythonDump;
1205 aPythonDump << "([";
1207 if (theStatus == SMESH::DRS_OK) {
1208 SALOMEDS::StudyBuilder_var aStudyBuilder;
1209 aStudyBuilder = getStudyServant()->NewBuilder();
1210 aStudyBuilder->NewCommand(); // There is a transaction
1212 aResult->length( aNames.size() );
1215 // Iterate through all meshes and create mesh objects
1216 for ( list<string>::iterator it = aNames.begin(); it != aNames.end(); it++ )
1219 if (i > 0) aPythonDump << ", ";
1222 SMESH::SMESH_Mesh_var mesh = createMesh();
1224 // publish mesh in the study
1225 SALOMEDS::SObject_wrap aSO;
1226 if ( CanPublishInStudy( mesh ) )
1227 // little trick: for MED file theFileName and theFileNameForPython are the same, but they are different for SAUV
1228 // - as names of meshes are stored in MED file, we use them for data publishing
1229 // - as mesh name is not stored in UNV file, we use file name as name of mesh when publishing data
1230 aSO = PublishMesh( mesh.in(), ( theFileName == theFileNameForPython ) ? (*it).c_str() : aFileName.c_str() );
1233 if ( !aSO->_is_nil() ) {
1236 aPythonDump << "mesh_" << i;
1239 // Read mesh data (groups are published automatically by ImportMEDFile())
1240 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( mesh ).in() );
1241 ASSERT( meshServant );
1242 SMESH::DriverMED_ReadStatus status1 =
1243 meshServant->ImportMEDFile( theFileName, (*it).c_str() );
1244 if (status1 > theStatus)
1245 theStatus = status1;
1247 aResult[i++] = SMESH::SMESH_Mesh::_duplicate( mesh );
1248 meshServant->GetImpl().GetMeshDS()->Modified();
1250 if ( !aStudyBuilder->_is_nil() )
1251 aStudyBuilder->CommitCommand();
1254 // Update Python script
1255 aPythonDump << "], status) = " << this << "." << theCommandNameForPython << "(r'" << theFileNameForPython << "')";
1257 // Dump creation of groups
1258 for ( CORBA::ULong i = 0; i < aResult->length(); ++i )
1259 SMESH::ListOfGroups_var groups = aResult[ i ]->GetGroups();
1261 return aResult._retn();
1264 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromMED( const char* theFileName,
1265 SMESH::DriverMED_ReadStatus& theStatus)
1266 throw ( SALOME::SALOME_Exception )
1268 Unexpect aCatch(SALOME_SalomeException);
1269 checkFileReadable( theFileName );
1271 SMESH::mesh_array* result = CreateMeshesFromMEDorSAUV(theFileName, theStatus, "CreateMeshesFromMED", theFileName);
1275 //=============================================================================
1277 * SMESH_Gen_i::CreateMeshFromSAUV
1279 * Create mesh and import data from SAUV file
1281 //=============================================================================
1283 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromSAUV( const char* theFileName,
1284 SMESH::DriverMED_ReadStatus& theStatus)
1285 throw ( SALOME::SALOME_Exception )
1287 Unexpect aCatch(SALOME_SalomeException);
1288 checkFileReadable( theFileName );
1290 std::string sauvfilename(theFileName);
1291 std::string medfilename(theFileName);
1292 medfilename += ".med";
1295 cmd = "%PYTHONBIN% ";
1300 cmd += "from medutilities import convert ; convert(r'" + sauvfilename + "', 'GIBI', 'MED', 1, r'" + medfilename + "')";
1302 system(cmd.c_str());
1303 SMESH::mesh_array* result = CreateMeshesFromMEDorSAUV(medfilename.c_str(), theStatus, "CreateMeshesFromSAUV", sauvfilename.c_str());
1305 cmd = "%PYTHONBIN% ";
1310 cmd += "from medutilities import my_remove ; my_remove(r'" + medfilename + "')";
1312 system(cmd.c_str());
1316 //=============================================================================
1318 * SMESH_Gen_i::CreateMeshFromSTL
1320 * Create mesh and import data from STL file
1322 //=============================================================================
1324 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMeshesFromSTL( const char* theFileName )
1325 throw ( SALOME::SALOME_Exception )
1327 Unexpect aCatch(SALOME_SalomeException);
1328 checkFileReadable( theFileName );
1330 SMESH::SMESH_Mesh_var aMesh = createMesh();
1333 char bname[ _MAX_FNAME ];
1334 _splitpath( theFileName, NULL, NULL, bname, NULL );
1335 string aFileName = bname;
1337 string aFileName = basename( const_cast<char *>(theFileName) );
1339 // publish mesh in the study
1340 if ( CanPublishInStudy( aMesh ) ) {
1341 SALOMEDS::StudyBuilder_var aStudyBuilder = getStudyServant()->NewBuilder();
1342 aStudyBuilder->NewCommand(); // There is a transaction
1343 SALOMEDS::SObject_wrap aSO = PublishInStudy( SALOMEDS::SObject::_nil(), aMesh.in(), aFileName.c_str() );
1344 aStudyBuilder->CommitCommand();
1345 if ( !aSO->_is_nil() ) {
1346 // Update Python script
1347 TPythonDump() << aSO << " = " << this << ".CreateMeshesFromSTL(r'" << theFileName << "')";
1351 SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
1353 aServant->ImportSTLFile( theFileName );
1354 aServant->GetImpl().GetMeshDS()->Modified();
1355 return aMesh._retn();
1358 //================================================================================
1360 * \brief Create meshes and import data from the CGSN file
1362 //================================================================================
1364 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromCGNS( const char* theFileName,
1365 SMESH::DriverMED_ReadStatus& theStatus)
1366 throw ( SALOME::SALOME_Exception )
1368 Unexpect aCatch(SALOME_SalomeException);
1369 checkFileReadable( theFileName );
1371 SMESH::mesh_array_var aResult = new SMESH::mesh_array();
1374 // Retrieve nb meshes from the file
1375 DriverCGNS_Read myReader;
1376 myReader.SetFile( theFileName );
1377 Driver_Mesh::Status aStatus;
1378 int nbMeshes = myReader.GetNbMeshes(aStatus);
1379 theStatus = (SMESH::DriverMED_ReadStatus)aStatus;
1381 aResult->length( nbMeshes );
1383 { // open a new scope to make aPythonDump die before PythonDump in SMESH_Mesh::GetGroups()
1386 TPythonDump aPythonDump;
1387 aPythonDump << "([";
1389 if (theStatus == SMESH::DRS_OK)
1391 SALOMEDS::StudyBuilder_var aStudyBuilder = getStudyServant()->NewBuilder();
1392 aStudyBuilder->NewCommand(); // There is a transaction
1396 // Iterate through all meshes and create mesh objects
1397 for ( ; i < nbMeshes; ++i )
1400 if (i > 0) aPythonDump << ", ";
1403 SMESH::SMESH_Mesh_var mesh = createMesh();
1404 aResult[i] = SMESH::SMESH_Mesh::_duplicate( mesh );
1406 // Read mesh data (groups are published automatically by ImportMEDFile())
1407 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( mesh ).in() );
1408 ASSERT( meshServant );
1410 SMESH::DriverMED_ReadStatus status1 =
1411 meshServant->ImportCGNSFile( theFileName, i, meshName );
1412 if (status1 > theStatus)
1413 theStatus = status1;
1415 meshServant->GetImpl().GetMeshDS()->Modified();
1416 // publish mesh in the study
1417 SALOMEDS::SObject_wrap aSO;
1418 if ( CanPublishInStudy( mesh ) )
1419 aSO = PublishMesh( mesh.in(), meshName.c_str() );
1422 if ( !aSO->_is_nil() ) {
1426 aPythonDump << "mesh_" << i;
1429 aStudyBuilder->CommitCommand();
1432 aPythonDump << "], status) = " << this << ".CreateMeshesFromCGNS(r'" << theFileName << "')";
1434 // Dump creation of groups
1435 for ( CORBA::ULong i = 0; i < aResult->length(); ++i )
1436 SMESH::ListOfGroups_var groups = aResult[ i ]->GetGroups();
1438 THROW_SALOME_CORBA_EXCEPTION("CGNS library is unavailable", SALOME::INTERNAL_ERROR);
1441 return aResult._retn();
1444 //================================================================================
1446 * \brief Create a mesh and import data from a GMF file
1448 //================================================================================
1450 SMESH::SMESH_Mesh_ptr
1451 SMESH_Gen_i::CreateMeshesFromGMF( const char* theFileName,
1452 CORBA::Boolean theMakeRequiredGroups,
1453 SMESH::ComputeError_out theError)
1454 throw ( SALOME::SALOME_Exception )
1456 Unexpect aCatch(SALOME_SalomeException);
1457 checkFileReadable( theFileName );
1459 SMESH::SMESH_Mesh_var aMesh = createMesh();
1461 char bname[ _MAX_FNAME ];
1462 _splitpath( theFileName, NULL, NULL, bname, NULL );
1463 string aFileName = bname;
1465 string aFileName = basename( const_cast<char *>(theFileName) );
1467 // publish mesh in the study
1468 if ( CanPublishInStudy( aMesh ) ) {
1469 SALOMEDS::StudyBuilder_var aStudyBuilder = getStudyServant()->NewBuilder();
1470 aStudyBuilder->NewCommand(); // There is a transaction
1471 SALOMEDS::SObject_wrap aSO = PublishInStudy( SALOMEDS::SObject::_nil(), aMesh.in(), aFileName.c_str() );
1472 aStudyBuilder->CommitCommand();
1473 if ( !aSO->_is_nil() ) {
1474 // Update Python script
1475 TPythonDump() << "("<< aSO << ", error) = " << this << ".CreateMeshesFromGMF(r'"
1476 << theFileName << "', "
1477 << theMakeRequiredGroups << " )";
1480 SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
1482 theError = aServant->ImportGMFFile( theFileName, theMakeRequiredGroups );
1483 aServant->GetImpl().GetMeshDS()->Modified();
1484 return aMesh._retn();
1488 //=============================================================================
1490 * SMESH_Gen_i::IsReadyToCompute
1492 * Returns true if mesh contains enough data to be computed
1494 //=============================================================================
1496 CORBA::Boolean SMESH_Gen_i::IsReadyToCompute( SMESH::SMESH_Mesh_ptr theMesh,
1497 GEOM::GEOM_Object_ptr theShapeObject )
1498 throw ( SALOME::SALOME_Exception )
1500 Unexpect aCatch(SALOME_SalomeException);
1501 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::IsReadyToCompute" );
1503 if ( CORBA::is_nil( theShapeObject ) )
1504 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1505 SALOME::BAD_PARAM );
1507 if ( CORBA::is_nil( theMesh ) )
1508 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
1509 SALOME::BAD_PARAM );
1513 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1514 ASSERT( meshServant );
1515 if ( meshServant ) {
1516 // get local TopoDS_Shape
1517 TopoDS_Shape myLocShape = GeomObjectToShape( theShapeObject );
1518 // call implementation
1519 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1520 return myGen.CheckAlgoState( myLocMesh, myLocShape );
1523 catch ( SALOME_Exception& S_ex ) {
1524 INFOS( "catch exception "<< S_ex.what() );
1529 //================================================================================
1531 * \brief Find SObject for an algo
1533 //================================================================================
1535 SALOMEDS::SObject_ptr SMESH_Gen_i::GetAlgoSO(const ::SMESH_Algo* algo)
1538 SALOMEDS::Study_var aStudy = getStudyServant();
1539 if ( !aStudy->_is_nil() ) {
1540 // find algo in the study
1541 CORBA::String_var compDataType = ComponentDataType();
1542 SALOMEDS::SComponent_wrap father = aStudy->FindComponent( compDataType.in() );
1543 if ( !father->_is_nil() ) {
1544 SALOMEDS::ChildIterator_wrap itBig = aStudy->NewChildIterator( father );
1545 for ( ; itBig->More(); itBig->Next() ) {
1546 SALOMEDS::SObject_wrap gotBranch = itBig->Value();
1547 if ( gotBranch->Tag() == GetAlgorithmsRootTag() ) {
1548 SALOMEDS::ChildIterator_wrap algoIt = aStudy->NewChildIterator( gotBranch );
1549 for ( ; algoIt->More(); algoIt->Next() ) {
1550 SALOMEDS::SObject_wrap algoSO = algoIt->Value();
1551 CORBA::Object_var algoIOR = SObjectToObject( algoSO );
1552 if ( !CORBA::is_nil( algoIOR )) {
1553 SMESH_Hypothesis_i* impl = SMESH::DownCast<SMESH_Hypothesis_i*>( algoIOR );
1554 if ( impl && impl->GetImpl() == algo )
1555 return algoSO._retn();
1557 } // loop on algo SO's
1560 } // SMESH component iterator
1564 return SALOMEDS::SObject::_nil();
1567 //================================================================================
1569 * \brief Return errors of mesh computation
1571 //================================================================================
1573 SMESH::compute_error_array* SMESH_Gen_i::GetComputeErrors( SMESH::SMESH_Mesh_ptr theMesh,
1574 GEOM::GEOM_Object_ptr theSubObject )
1575 throw ( SALOME::SALOME_Exception )
1577 Unexpect aCatch(SALOME_SalomeException);
1578 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetComputeErrors()" );
1580 if ( CORBA::is_nil( theSubObject ) && theMesh->HasShapeToMesh())
1581 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference", SALOME::BAD_PARAM );
1583 if ( CORBA::is_nil( theMesh ) )
1584 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1586 SMESH::compute_error_array_var error_array = new SMESH::compute_error_array;
1588 if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1591 if(theMesh->HasShapeToMesh())
1592 shape = GeomObjectToShape( theSubObject );
1594 shape = SMESH_Mesh::PseudoShape();
1596 ::SMESH_Mesh& mesh = meshServant->GetImpl();
1598 error_array->length( mesh.GetMeshDS()->MaxShapeIndex() );
1601 SMESH_subMesh *sm = mesh.GetSubMesh(shape);
1602 const bool includeSelf = true, complexShapeFirst = true;
1603 SMESH_subMeshIteratorPtr smIt = sm->getDependsOnIterator(includeSelf,
1605 while ( smIt->more() )
1608 // if ( sm->GetSubShape().ShapeType() == TopAbs_VERTEX )
1610 SMESH_ComputeErrorPtr error = sm->GetComputeError();
1611 if ( error && !error->IsOK() )
1613 if ( !( error->myAlgo ) &&
1614 !( error->myAlgo = sm->GetAlgo() ))
1616 SMESH::ComputeError & errStruct = error_array[ nbErr++ ];
1617 errStruct.code = -( error->myName < 0 ? error->myName + 1: error->myName ); // -1 -> 0
1618 errStruct.comment = error->myComment.c_str();
1619 errStruct.subShapeID = sm->GetId();
1620 SALOMEDS::SObject_wrap algoSO = GetAlgoSO( error->myAlgo );
1621 if ( !algoSO->_is_nil() ) {
1622 CORBA::String_var algoName = algoSO->GetName();
1623 errStruct.algoName = algoName;
1626 errStruct.algoName = error->myAlgo->GetName();
1628 errStruct.hasBadMesh = error->HasBadElems();
1631 error_array->length( nbErr );
1634 catch ( SALOME_Exception& S_ex ) {
1635 INFOS( "catch exception "<< S_ex.what() );
1638 return error_array._retn();
1641 //================================================================================
1643 * \brief Return mesh elements preventing computation of a subshape
1645 //================================================================================
1647 SMESH::MeshPreviewStruct*
1648 SMESH_Gen_i::GetBadInputElements( SMESH::SMESH_Mesh_ptr theMesh,
1649 CORBA::Short theSubShapeID )
1650 throw ( SALOME::SALOME_Exception )
1652 Unexpect aCatch(SALOME_SalomeException);
1653 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetBadInputElements()" );
1655 if ( CORBA::is_nil( theMesh ) )
1656 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1658 SMESH::MeshPreviewStruct_var result = new SMESH::MeshPreviewStruct;
1661 if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1663 // mesh implementation
1664 ::SMESH_Mesh& mesh = meshServant->GetImpl();
1665 // submesh by subshape id
1666 if ( SMESH_subMesh * sm = mesh.GetSubMeshContaining( theSubShapeID ))
1669 SMESH_ComputeErrorPtr error = sm->GetComputeError();
1670 if ( error && error->HasBadElems() )
1672 typedef map<const SMDS_MeshElement*, int > TNode2LocalIDMap;
1673 typedef TNode2LocalIDMap::iterator TNodeLocalID;
1675 // get nodes of elements and count elements
1676 TNode2LocalIDMap mapNode2LocalID;
1677 list< TNodeLocalID > connectivity;
1678 int i, nbElements = 0, nbConnNodes = 0;
1680 const list<const SMDS_MeshElement*>& badElems =
1681 static_cast<SMESH_BadInputElements*>( error.get() )->myBadElements;
1682 list<const SMDS_MeshElement*>::const_iterator elemIt = badElems.begin();
1683 list<const SMDS_MeshElement*>::const_iterator elemEnd = badElems.end();
1684 for ( ; elemIt != elemEnd; ++elemIt, ++nbElements )
1686 SMDS_ElemIteratorPtr nIt = (*elemIt)->nodesIterator();
1687 while ( nIt->more() )
1688 connectivity.push_back
1689 ( mapNode2LocalID.insert( make_pair( nIt->next(), ++nbConnNodes)).first );
1691 // fill node coords and assign local ids to the nodes
1692 int nbNodes = mapNode2LocalID.size();
1693 result->nodesXYZ.length( nbNodes );
1694 TNodeLocalID node2ID = mapNode2LocalID.begin();
1695 for ( i = 0; i < nbNodes; ++i, ++node2ID ) {
1696 node2ID->second = i;
1697 const SMDS_MeshNode* node = (const SMDS_MeshNode*) node2ID->first;
1698 result->nodesXYZ[i].x = node->X();
1699 result->nodesXYZ[i].y = node->Y();
1700 result->nodesXYZ[i].z = node->Z();
1702 // fill connectivity
1703 result->elementConnectivities.length( nbConnNodes );
1704 list< TNodeLocalID >::iterator connIt = connectivity.begin();
1705 for ( i = 0; i < nbConnNodes; ++i, ++connIt ) {
1706 result->elementConnectivities[i] = (*connIt)->second;
1708 // fill element types
1709 result->elementTypes.length( nbElements );
1710 for ( i = 0, elemIt = badElems.begin(); i <nbElements; ++i, ++elemIt )
1712 const SMDS_MeshElement* elem = *elemIt;
1713 result->elementTypes[i].SMDS_ElementType = (SMESH::ElementType) elem->GetType();
1714 result->elementTypes[i].isPoly = elem->IsPoly();
1715 result->elementTypes[i].nbNodesInElement = elem->NbNodes();
1721 catch ( SALOME_Exception& S_ex ) {
1722 INFOS( "catch exception "<< S_ex.what() );
1725 return result._retn();
1728 //================================================================================
1730 * \brief Create a group of elements preventing computation of a sub-shape
1732 //================================================================================
1734 SMESH::ListOfGroups*
1735 SMESH_Gen_i::MakeGroupsOfBadInputElements( SMESH::SMESH_Mesh_ptr theMesh,
1736 CORBA::Short theSubShapeID,
1737 const char* theGroupName )
1738 throw ( SALOME::SALOME_Exception )
1740 Unexpect aCatch(SALOME_SalomeException);
1742 SMESH::ListOfGroups_var groups;
1744 if ( CORBA::is_nil( theMesh ) )
1745 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1748 if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1750 groups = meshServant->MakeGroupsOfBadInputElements( theSubShapeID, theGroupName );
1751 TPythonDump() << groups << " = " << this
1752 << ".MakeGroupsOfBadInputElements( "
1753 << theMesh << ", " << theSubShapeID << ", '" << theGroupName << "' )";
1756 catch ( SALOME_Exception& S_ex ) {
1757 INFOS( "catch exception "<< S_ex.what() );
1759 return groups._retn();
1762 //================================================================================
1764 * \brief Returns errors of hypotheses definition
1765 * \param theMesh - the mesh
1766 * \param theSubObject - the main or sub- shape
1767 * \retval SMESH::algo_error_array* - sequence of errors
1769 //================================================================================
1771 SMESH::algo_error_array* SMESH_Gen_i::GetAlgoState( SMESH::SMESH_Mesh_ptr theMesh,
1772 GEOM::GEOM_Object_ptr theSubObject )
1773 throw ( SALOME::SALOME_Exception )
1775 Unexpect aCatch(SALOME_SalomeException);
1776 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetAlgoState()" );
1778 if ( CORBA::is_nil( theSubObject ) && theMesh->HasShapeToMesh())
1779 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference", SALOME::BAD_PARAM );
1781 if ( CORBA::is_nil( theMesh ) )
1782 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1784 SMESH::algo_error_array_var error_array = new SMESH::algo_error_array;
1786 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
1787 ASSERT( meshServant );
1788 if ( meshServant ) {
1789 TopoDS_Shape myLocShape;
1790 if(theMesh->HasShapeToMesh())
1791 myLocShape = GeomObjectToShape( theSubObject );
1793 myLocShape = SMESH_Mesh::PseudoShape();
1795 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1796 list< ::SMESH_Gen::TAlgoStateError > error_list;
1797 list< ::SMESH_Gen::TAlgoStateError >::iterator error;
1798 // call ::SMESH_Gen::GetAlgoState()
1799 myGen.GetAlgoState( myLocMesh, myLocShape, error_list );
1800 error_array->length( error_list.size() );
1802 for ( error = error_list.begin(); error != error_list.end(); ++error )
1804 // fill AlgoStateError structure
1805 SMESH::AlgoStateError & errStruct = error_array[ i++ ];
1806 errStruct.state = SMESH_Mesh_i::ConvertHypothesisStatus( error->_name );
1807 errStruct.algoDim = error->_algoDim;
1808 errStruct.isGlobalAlgo = error->_isGlobalAlgo;
1809 errStruct.algoName = "";
1810 SALOMEDS::SObject_wrap algoSO = GetAlgoSO( error->_algo );
1811 if ( !algoSO->_is_nil() ) {
1812 CORBA::String_var algoName = algoSO->GetName();
1813 errStruct.algoName = algoName.in();
1818 catch ( SALOME_Exception& S_ex ) {
1819 INFOS( "catch exception "<< S_ex.what() );
1821 return error_array._retn();
1824 //=============================================================================
1826 * SMESH_Gen_i::GetSubShapesId
1828 * Get sub-shapes unique ID's list
1830 //=============================================================================
1833 SMESH_Gen_i::GetSubShapesId( GEOM::GEOM_Object_ptr theMainShapeObject,
1834 const SMESH::object_array& theListOfSubShapeObject )
1835 throw ( SALOME::SALOME_Exception )
1837 Unexpect aCatch(SALOME_SalomeException);
1838 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetSubShapesId" );
1840 SMESH::long_array_var shapesId = new SMESH::long_array;
1843 if ( CORBA::is_nil( theMainShapeObject ) )
1844 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference", SALOME::BAD_PARAM );
1848 TopoDS_Shape myMainShape = GeomObjectToShape(theMainShapeObject);
1849 TopTools_IndexedMapOfShape myIndexToShape;
1850 TopExp::MapShapes(myMainShape,myIndexToShape);
1852 for ( CORBA::ULong i = 0; i < theListOfSubShapeObject.length(); i++ )
1854 GEOM::GEOM_Object_var aShapeObject
1855 = GEOM::GEOM_Object::_narrow(theListOfSubShapeObject[i]);
1856 if ( CORBA::is_nil( aShapeObject ) )
1857 THROW_SALOME_CORBA_EXCEPTION ("bad shape object reference", \
1858 SALOME::BAD_PARAM );
1860 TopoDS_Shape locShape = GeomObjectToShape(aShapeObject);
1861 for (TopExp_Explorer exp(locShape,TopAbs_FACE); exp.More(); exp.Next())
1863 const TopoDS_Face& F = TopoDS::Face(exp.Current());
1864 setId.insert(myIndexToShape.FindIndex(F));
1865 if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(F));
1867 for (TopExp_Explorer exp(locShape,TopAbs_EDGE); exp.More(); exp.Next())
1869 const TopoDS_Edge& E = TopoDS::Edge(exp.Current());
1870 setId.insert(myIndexToShape.FindIndex(E));
1871 if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(E));
1873 for (TopExp_Explorer exp(locShape,TopAbs_VERTEX); exp.More(); exp.Next())
1875 const TopoDS_Vertex& V = TopoDS::Vertex(exp.Current());
1876 setId.insert(myIndexToShape.FindIndex(V));
1877 if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(V));
1880 shapesId->length(setId.size());
1881 set<int>::iterator iind;
1883 for (iind = setId.begin(); iind != setId.end(); iind++)
1885 if(MYDEBUG) SCRUTE((*iind));
1886 shapesId[i] = (*iind);
1887 if(MYDEBUG) SCRUTE(shapesId[i]);
1891 catch (SALOME_Exception& S_ex)
1893 THROW_SALOME_CORBA_EXCEPTION(S_ex.what(), SALOME::BAD_PARAM);
1896 return shapesId._retn();
1899 //=============================================================================
1901 * SMESH_Gen_i::Compute
1903 * Compute mesh on a shape
1905 //=============================================================================
1907 CORBA::Boolean SMESH_Gen_i::Compute( SMESH::SMESH_Mesh_ptr theMesh,
1908 GEOM::GEOM_Object_ptr theShapeObject )
1909 throw ( SALOME::SALOME_Exception )
1912 Unexpect aCatch(SALOME_SalomeException);
1913 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Compute" );
1915 if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
1916 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1917 SALOME::BAD_PARAM );
1919 if ( CORBA::is_nil( theMesh ) )
1920 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
1921 SALOME::BAD_PARAM );
1923 // Update Python script
1924 TPythonDump() << "isDone = " << this << ".Compute( "
1925 << theMesh << ", " << theShapeObject << ")";
1929 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1930 ASSERT( meshServant );
1931 if ( meshServant ) {
1932 meshServant->Load();
1933 // NPAL16168: "geometrical group edition from a submesh don't modify mesh computation"
1934 meshServant->CheckGeomModif();
1935 // get local TopoDS_Shape
1936 TopoDS_Shape myLocShape;
1937 if(theMesh->HasShapeToMesh())
1938 myLocShape = GeomObjectToShape( theShapeObject );
1940 myLocShape = SMESH_Mesh::PseudoShape();
1941 // call implementation compute
1942 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1943 myGen.PrepareCompute( myLocMesh, myLocShape );
1944 int how = ::SMESH_Gen::COMPACT_MESH;
1945 if ( myLocShape != myLocMesh.GetShapeToMesh() ) // compute a sub-mesh
1946 how |= ::SMESH_Gen::SHAPE_ONLY;
1947 bool ok = myGen.Compute( myLocMesh, myLocShape, how );
1948 meshServant->CreateGroupServants(); // algos can create groups (issue 0020918)
1949 myLocMesh.GetMeshDS()->Modified();
1950 UpdateIcons( theMesh );
1954 catch ( std::bad_alloc ) {
1955 INFOS( "Compute(): lack of memory" );
1957 catch ( SALOME_Exception& S_ex ) {
1958 INFOS( "Compute(): catch exception "<< S_ex.what() );
1961 INFOS( "Compute(): unknown exception " );
1966 //=============================================================================
1968 * SMESH_Gen_i::CancelCompute
1970 * Cancel Compute mesh on a shape
1972 //=============================================================================
1974 void SMESH_Gen_i::CancelCompute( SMESH::SMESH_Mesh_ptr theMesh,
1975 GEOM::GEOM_Object_ptr theShapeObject )
1977 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1978 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1979 TopoDS_Shape myLocShape;
1980 if(theMesh->HasShapeToMesh())
1981 myLocShape = GeomObjectToShape( theShapeObject );
1983 myLocShape = SMESH_Mesh::PseudoShape();
1984 myGen.CancelCompute( myLocMesh, myLocShape);
1987 //=============================================================================
1989 * SMESH_Gen_i::Precompute
1991 * Compute mesh as preview till indicated dimension on shape
1993 //=============================================================================
1995 SMESH::MeshPreviewStruct* SMESH_Gen_i::Precompute( SMESH::SMESH_Mesh_ptr theMesh,
1996 GEOM::GEOM_Object_ptr theShapeObject,
1997 SMESH::Dimension theDimension,
1998 SMESH::long_array& theShapesId)
1999 throw ( SALOME::SALOME_Exception )
2001 Unexpect aCatch(SALOME_SalomeException);
2002 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Precompute" );
2004 if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
2005 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
2006 SALOME::BAD_PARAM );
2008 if ( CORBA::is_nil( theMesh ) )
2009 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
2010 SALOME::BAD_PARAM );
2012 SMESH::MeshPreviewStruct_var result = new SMESH::MeshPreviewStruct;
2015 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
2016 meshServant->Load();
2017 ASSERT( meshServant );
2018 if ( meshServant ) {
2019 // NPAL16168: "geometrical group edition from a submesh don't modify mesh computation"
2020 meshServant->CheckGeomModif();
2021 // get local TopoDS_Shape
2022 TopoDS_Shape myLocShape;
2023 if(theMesh->HasShapeToMesh())
2024 myLocShape = GeomObjectToShape( theShapeObject );
2026 return result._retn();;
2028 // call implementation compute
2029 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
2031 ::MeshDimension aDim = (MeshDimension)theDimension;
2032 if ( myGen.Compute( myLocMesh, myLocShape, ::SMESH_Gen::COMPACT_MESH, aDim, &shapeIds ) )
2034 int nbShapeId = shapeIds.size();
2035 theShapesId.length( nbShapeId );
2036 // iterates on shapes and collect mesh entities into mesh preview
2037 TSetOfInt::const_iterator idIt = shapeIds.begin();
2038 TSetOfInt::const_iterator idEnd = shapeIds.end();
2039 std::map< int, int > mapOfShIdNb;
2040 std::set< SMESH_TLink > setOfEdge;
2041 std::list< SMDSAbs_ElementType > listOfElemType;
2042 typedef map<const SMDS_MeshElement*, int > TNode2LocalIDMap;
2043 typedef TNode2LocalIDMap::iterator TNodeLocalID;
2044 TNode2LocalIDMap mapNode2LocalID;
2045 list< TNodeLocalID > connectivity;
2046 int i, nbConnNodes = 0;
2047 std::set< const SMESH_subMesh* > setOfVSubMesh;
2048 // iterates on shapes
2049 for ( ; idIt != idEnd; idIt++ )
2051 if ( mapOfShIdNb.find( *idIt ) != mapOfShIdNb.end() )
2053 SMESH_subMesh* sm = myLocMesh.GetSubMeshContaining(*idIt);
2054 if ( !sm || !sm->IsMeshComputed() )
2057 const TopoDS_Shape& aSh = sm->GetSubShape();
2058 const int shDim = myGen.GetShapeDim( aSh );
2059 if ( shDim < 1 || shDim > theDimension )
2062 mapOfShIdNb[ *idIt ] = 0;
2063 theShapesId[ mapOfShIdNb.size() - 1 ] = *idIt;
2065 SMESHDS_SubMesh* smDS = sm->GetSubMeshDS();
2066 if ( !smDS ) continue;
2068 if ( theDimension == SMESH::DIM_2D )
2070 SMDS_ElemIteratorPtr faceIt = smDS->GetElements();
2071 while ( faceIt->more() )
2073 const SMDS_MeshElement* face = faceIt->next();
2074 int aNbNode = face->NbNodes();
2076 aNbNode /= 2; // do not take into account additional middle nodes
2078 SMDS_MeshNode* node1 = (SMDS_MeshNode*)face->GetNode( 0 );
2079 for ( int nIndx = 0; nIndx < aNbNode; nIndx++ )
2081 SMDS_MeshNode* node2 = (SMDS_MeshNode*)face->GetNode( nIndx+1 < aNbNode ? nIndx+1 : 0 );
2082 if ( setOfEdge.insert( SMESH_TLink ( node1, node2 ) ).second )
2084 listOfElemType.push_back( SMDSAbs_Edge );
2085 connectivity.push_back
2086 ( mapNode2LocalID.insert( make_pair( node1, ++nbConnNodes)).first );
2087 connectivity.push_back
2088 ( mapNode2LocalID.insert( make_pair( node2, ++nbConnNodes)).first );
2094 else if ( theDimension == SMESH::DIM_1D )
2096 SMDS_NodeIteratorPtr nodeIt = smDS->GetNodes();
2097 while ( nodeIt->more() )
2099 listOfElemType.push_back( SMDSAbs_Node );
2100 connectivity.push_back
2101 ( mapNode2LocalID.insert( make_pair( nodeIt->next(), ++nbConnNodes)).first );
2103 // add corner nodes by first vertex from edge
2104 SMESH_subMeshIteratorPtr edgeSmIt =
2105 sm->getDependsOnIterator(/*includeSelf*/false,
2106 /*complexShapeFirst*/false);
2107 while ( edgeSmIt->more() )
2109 SMESH_subMesh* vertexSM = edgeSmIt->next();
2110 // check that vertex is not already treated
2111 if ( !setOfVSubMesh.insert( vertexSM ).second )
2113 if ( vertexSM->GetSubShape().ShapeType() != TopAbs_VERTEX )
2116 const SMESHDS_SubMesh* vertexSmDS = vertexSM->GetSubMeshDS();
2117 SMDS_NodeIteratorPtr nodeIt = vertexSmDS->GetNodes();
2118 while ( nodeIt->more() )
2120 listOfElemType.push_back( SMDSAbs_Node );
2121 connectivity.push_back
2122 ( mapNode2LocalID.insert( make_pair( nodeIt->next(), ++nbConnNodes)).first );
2128 // fill node coords and assign local ids to the nodes
2129 int nbNodes = mapNode2LocalID.size();
2130 result->nodesXYZ.length( nbNodes );
2131 TNodeLocalID node2ID = mapNode2LocalID.begin();
2132 for ( i = 0; i < nbNodes; ++i, ++node2ID ) {
2133 node2ID->second = i;
2134 const SMDS_MeshNode* node = (const SMDS_MeshNode*) node2ID->first;
2135 result->nodesXYZ[i].x = node->X();
2136 result->nodesXYZ[i].y = node->Y();
2137 result->nodesXYZ[i].z = node->Z();
2139 // fill connectivity
2140 result->elementConnectivities.length( nbConnNodes );
2141 list< TNodeLocalID >::iterator connIt = connectivity.begin();
2142 for ( i = 0; i < nbConnNodes; ++i, ++connIt ) {
2143 result->elementConnectivities[i] = (*connIt)->second;
2146 // fill element types
2147 result->elementTypes.length( listOfElemType.size() );
2148 std::list< SMDSAbs_ElementType >::const_iterator typeIt = listOfElemType.begin();
2149 std::list< SMDSAbs_ElementType >::const_iterator typeEnd = listOfElemType.end();
2150 for ( i = 0; typeIt != typeEnd; ++i, ++typeIt )
2152 SMDSAbs_ElementType elemType = *typeIt;
2153 result->elementTypes[i].SMDS_ElementType = (SMESH::ElementType)elemType;
2154 result->elementTypes[i].isPoly = false;
2155 result->elementTypes[i].nbNodesInElement = elemType == SMDSAbs_Edge ? 2 : 1;
2158 // correct number of shapes
2159 theShapesId.length( mapOfShIdNb.size() );
2163 catch ( std::bad_alloc ) {
2164 INFOS( "Precompute(): lack of memory" );
2166 catch ( SALOME_Exception& S_ex ) {
2167 INFOS( "Precompute(): catch exception "<< S_ex.what() );
2170 INFOS( "Precompute(): unknown exception " );
2172 return result._retn();
2176 //=============================================================================
2178 * SMESH_Gen_i::Evaluate
2180 * Evaluate mesh on a shape
2182 //=============================================================================
2184 SMESH::long_array* SMESH_Gen_i::Evaluate(SMESH::SMESH_Mesh_ptr theMesh,
2185 GEOM::GEOM_Object_ptr theShapeObject)
2186 // SMESH::long_array& theNbElems)
2187 throw ( SALOME::SALOME_Exception )
2189 Unexpect aCatch(SALOME_SalomeException);
2190 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Evaluate" );
2192 if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
2193 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
2194 SALOME::BAD_PARAM );
2196 if ( CORBA::is_nil( theMesh ) )
2197 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
2198 SALOME::BAD_PARAM );
2200 SMESH::long_array_var nbels = new SMESH::long_array;
2201 nbels->length(SMESH::Entity_Last);
2202 int i = SMESH::Entity_Node;
2203 for (; i < SMESH::Entity_Last; i++)
2206 // Update Python script
2207 TPythonDump() << "theNbElems = " << this << ".Evaluate( "
2208 << theMesh << ", " << theShapeObject << ")";
2212 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
2213 ASSERT( meshServant );
2214 if ( meshServant ) {
2215 meshServant->Load();
2216 // NPAL16168: "geometrical group edition from a submesh don't modify mesh computation"
2217 meshServant->CheckGeomModif();
2218 // get local TopoDS_Shape
2219 TopoDS_Shape myLocShape;
2220 if(theMesh->HasShapeToMesh())
2221 myLocShape = GeomObjectToShape( theShapeObject );
2223 myLocShape = SMESH_Mesh::PseudoShape();
2224 // call implementation compute
2225 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
2226 MapShapeNbElems aResMap;
2227 /*CORBA::Boolean ret =*/ myGen.Evaluate( myLocMesh, myLocShape, aResMap);
2228 MapShapeNbElemsItr anIt = aResMap.begin();
2229 for(; anIt!=aResMap.end(); anIt++) {
2230 const vector<int>& aVec = (*anIt).second;
2231 for ( i = SMESH::Entity_Node; i < (int)aVec.size(); i++ ) {
2232 int nbElem = aVec[i];
2233 if ( nbElem < 0 ) // algo failed, check that it has reported a message
2235 SMESH_subMesh* sm = anIt->first;
2236 SMESH_ComputeErrorPtr& error = sm->GetComputeError();
2237 const SMESH_Algo* algo = sm->GetAlgo();
2238 if ( (algo && !error.get()) || error->IsOK() )
2239 error.reset( new SMESH_ComputeError( COMPERR_ALGO_FAILED,"Failed to evaluate",algo));
2243 nbels[i] += aVec[i];
2247 return nbels._retn();
2250 catch ( std::bad_alloc ) {
2251 INFOS( "Evaluate(): lack of memory" );
2253 catch ( SALOME_Exception& S_ex ) {
2254 INFOS( "Evaluate(): catch exception "<< S_ex.what() );
2257 INFOS( "Evaluate(): unknown exception " );
2260 return nbels._retn();
2263 //================================================================================
2265 * \brief Return geometrical object the given element is built on
2266 * \param theMesh - the mesh the element is in
2267 * \param theElementID - the element ID
2268 * \param theGeomName - the name of the result geom object if it is not yet published
2269 * \retval GEOM::GEOM_Object_ptr - the found or just published geom object (no need to UnRegister())
2271 //================================================================================
2273 GEOM::GEOM_Object_ptr
2274 SMESH_Gen_i::GetGeometryByMeshElement( SMESH::SMESH_Mesh_ptr theMesh,
2275 CORBA::Long theElementID,
2276 const char* theGeomName)
2277 throw ( SALOME::SALOME_Exception )
2279 Unexpect aCatch(SALOME_SalomeException);
2281 GEOM::GEOM_Object_wrap geom = FindGeometryByMeshElement(theMesh, theElementID);
2282 if ( !geom->_is_nil() ) {
2283 GEOM::GEOM_Object_var mainShape = theMesh->GetShapeToMesh();
2284 GEOM::GEOM_Gen_ptr geomGen = GetGeomEngine();
2286 // try to find the corresponding SObject
2287 SALOMEDS::SObject_wrap SObj = ObjectToSObject( geom.in() );
2288 if ( SObj->_is_nil() ) // submesh can be not found even if published
2290 // try to find published submesh
2291 GEOM::ListOfLong_var list = geom->GetSubShapeIndices();
2292 if ( !geom->IsMainShape() && list->length() == 1 ) {
2293 SALOMEDS::SObject_wrap mainSO = ObjectToSObject( mainShape );
2294 SALOMEDS::ChildIterator_wrap it;
2295 if ( !mainSO->_is_nil() ) {
2296 it = getStudyServant()->NewChildIterator( mainSO );
2298 if ( !it->_is_nil() ) {
2299 for ( it->InitEx(true); it->More(); it->Next() ) {
2300 SALOMEDS::SObject_wrap so = it->Value();
2301 CORBA::Object_var obj = SObjectToObject( so );
2302 GEOM::GEOM_Object_var subGeom = GEOM::GEOM_Object::_narrow( obj );
2303 if ( !subGeom->_is_nil() ) {
2304 GEOM::ListOfLong_var subList = subGeom->GetSubShapeIndices();
2305 if ( subList->length() == 1 && list[0] == subList[0] ) {
2315 if ( SObj->_is_nil() ) // publish a new subshape
2316 SObj = geomGen->AddInStudy( geom, theGeomName, mainShape );
2318 // return only published geometry
2319 if ( !SObj->_is_nil() ) {
2320 //return geom._retn(); -- servant of geom must be UnRegister()ed;
2321 CORBA::Object_var obj = SObjectToObject( SObj );
2322 GEOM::GEOM_Object_var go = GEOM::GEOM_Object::_narrow( obj );
2326 return GEOM::GEOM_Object::_nil();
2329 //================================================================================
2331 * \brief Return geometrical object the given element is built on.
2332 * \param theMesh - the mesh the element is in
2333 * \param theElementID - the element ID
2334 * \retval GEOM::GEOM_Object_ptr - the found or created (UnRegister()!) geom object
2336 //================================================================================
2338 GEOM::GEOM_Object_ptr
2339 SMESH_Gen_i::FindGeometryByMeshElement( SMESH::SMESH_Mesh_ptr theMesh,
2340 CORBA::Long theElementID)
2341 throw ( SALOME::SALOME_Exception )
2343 Unexpect aCatch(SALOME_SalomeException);
2344 if ( CORBA::is_nil( theMesh ) )
2345 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference", SALOME::BAD_PARAM );
2347 GEOM::GEOM_Object_var mainShape = theMesh->GetShapeToMesh();
2348 GEOM::GEOM_Gen_ptr geomGen = GetGeomEngine();
2350 // get a core mesh DS
2351 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
2352 if ( meshServant && !geomGen->_is_nil() && !mainShape->_is_nil() )
2354 ::SMESH_Mesh & mesh = meshServant->GetImpl();
2355 SMESHDS_Mesh* meshDS = mesh.GetMeshDS();
2356 // find the element in mesh
2357 if ( const SMDS_MeshElement * elem = meshDS->FindElement( theElementID ) ) {
2358 // find a shape id by the element
2359 if ( int shapeID = ::SMESH_MeshEditor( &mesh ).FindShape( elem )) {
2360 // get a geom object by the shape id
2361 GEOM::GEOM_Object_var geom = ShapeToGeomObject( meshDS->IndexToShape( shapeID ));
2362 if ( geom->_is_nil() ) {
2363 // try to find a published sub-shape
2364 SALOMEDS::SObject_wrap mainSO = ObjectToSObject( mainShape );
2365 SALOMEDS::ChildIterator_wrap it;
2366 if ( !mainSO->_is_nil() ) {
2367 it = getStudyServant()->NewChildIterator( mainSO );
2369 if ( !it->_is_nil() ) {
2370 for ( it->InitEx(true); it->More(); it->Next() ) {
2371 SALOMEDS::SObject_wrap so = it->Value();
2372 CORBA::Object_var obj = SObjectToObject( so );
2373 GEOM::GEOM_Object_var subGeom = GEOM::GEOM_Object::_narrow( obj );
2374 if ( !subGeom->_is_nil() ) {
2375 GEOM::ListOfLong_var subList = subGeom->GetSubShapeIndices();
2376 if ( subList->length() == 1 && shapeID == subList[0] ) {
2384 if ( geom->_is_nil() ) {
2386 GEOM::GEOM_IShapesOperations_wrap op = geomGen->GetIShapesOperations();
2387 if ( !op->_is_nil() )
2388 geom = op->GetSubShape( mainShape, shapeID );
2393 if ( !geom->_is_nil() ) {
2394 GeomObjectToShape( geom ); // let geom client remember the found shape
2395 return geom._retn();
2400 return GEOM::GEOM_Object::_nil();
2403 //================================================================================
2405 * SMESH_Gen_i::Concatenate
2407 * Concatenate the given meshes into one mesh
2409 //================================================================================
2411 SMESH::SMESH_Mesh_ptr
2412 SMESH_Gen_i::Concatenate(const SMESH::ListOfIDSources& theMeshesArray,
2413 CORBA::Boolean theUniteIdenticalGroups,
2414 CORBA::Boolean theMergeNodesAndElements,
2415 CORBA::Double theMergeTolerance)
2416 throw ( SALOME::SALOME_Exception )
2418 return ConcatenateCommon(theMeshesArray,
2419 theUniteIdenticalGroups,
2420 theMergeNodesAndElements,
2425 //================================================================================
2427 * SMESH_Gen_i::ConcatenateWithGroups
2429 * Concatenate the given meshes into one mesh
2430 * Create the groups of all elements from initial meshes
2432 //================================================================================
2434 SMESH::SMESH_Mesh_ptr
2435 SMESH_Gen_i::ConcatenateWithGroups(const SMESH::ListOfIDSources& theMeshesArray,
2436 CORBA::Boolean theUniteIdenticalGroups,
2437 CORBA::Boolean theMergeNodesAndElements,
2438 CORBA::Double theMergeTolerance)
2439 throw ( SALOME::SALOME_Exception )
2441 return ConcatenateCommon(theMeshesArray,
2442 theUniteIdenticalGroups,
2443 theMergeNodesAndElements,
2448 //================================================================================
2450 * SMESH_Gen_i::ConcatenateCommon
2452 * Concatenate the given meshes into one mesh
2454 //================================================================================
2456 SMESH::SMESH_Mesh_ptr
2457 SMESH_Gen_i::ConcatenateCommon(const SMESH::ListOfIDSources& theMeshesArray,
2458 CORBA::Boolean theUniteIdenticalGroups,
2459 CORBA::Boolean theMergeNodesAndElements,
2460 CORBA::Double theMergeTolerance,
2461 CORBA::Boolean theCommonGroups)
2462 throw ( SALOME::SALOME_Exception )
2464 std::unique_ptr< TPythonDump > pPythonDump( new TPythonDump );
2465 TPythonDump& pythonDump = *pPythonDump; // prevent dump of called methods
2468 SMESH::SMESH_Mesh_var newMesh = CreateEmptyMesh();
2469 SMESH_Mesh_i* newImpl = SMESH::DownCast<SMESH_Mesh_i*>( newMesh );
2470 if ( !newImpl ) return newMesh._retn();
2472 ::SMESH_Mesh& locMesh = newImpl->GetImpl();
2473 SMESHDS_Mesh* newMeshDS = locMesh.GetMeshDS();
2475 typedef std::list<SMESH::SMESH_Group_var> TListOfNewGroups;
2476 typedef std::pair<string, SMESH::ElementType > TNameAndType;
2477 typedef std::map< TNameAndType, TListOfNewGroups > TGroupsMap;
2478 TGroupsMap groupsMap;
2479 TListOfNewGroups listOfNewGroups;
2481 ::SMESH_MeshEditor newEditor( &locMesh );
2482 ::SMESH_MeshEditor::ElemFeatures elemType;
2484 // loop on sub-meshes
2485 for ( CORBA::ULong i = 0; i < theMeshesArray.length(); i++ )
2487 if ( CORBA::is_nil( theMeshesArray[i] )) continue;
2488 SMESH::SMESH_Mesh_var initMesh = theMeshesArray[i]->GetMesh();
2489 SMESH_Mesh_i* initImpl = SMESH::DownCast<SMESH_Mesh_i*>( initMesh );
2490 if ( !initImpl ) continue;
2493 // assure that IDs increments by one during iteration
2494 ::SMESH_Mesh& initLocMesh = initImpl->GetImpl();
2495 SMESHDS_Mesh* initMeshDS = initLocMesh.GetMeshDS();
2496 if ( initMeshDS->MaxNodeID() > initMeshDS->NbNodes() ||
2497 initMeshDS->MaxElementID() > initMeshDS->NbElements() )
2499 initMeshDS->Modified();
2500 initMeshDS->CompactMesh();
2503 // remember nb of elements before filling in
2504 SMESH::long_array_var prevState = newMesh->GetNbElementsByType();
2508 std::vector< const SMDS_MeshElement* > newNodes( initMeshDS->NbNodes() + 1, 0 );
2509 SMDS_ElemIteratorPtr elemIt = initImpl->GetElements( theMeshesArray[i], SMESH::NODE );
2510 while ( elemIt->more() )
2512 SMESH_NodeXYZ node = elemIt->next();
2513 newNodes[ node->GetID() ] = newMeshDS->AddNode( node.X(), node.Y(), node.Z() );
2518 std::vector< const SMDS_MeshElement* > newElems( initMeshDS->NbElements() + 1, 0 );
2519 elemIt = initImpl->GetElements( theMeshesArray[i], SMESH::ALL );
2520 while ( elemIt->more() )
2522 const SMDS_MeshElement* elem = elemIt->next();
2523 elemType.myNodes.resize( elem->NbNodes() );
2525 SMDS_NodeIteratorPtr itNodes = elem->nodeIterator();
2526 for ( int k = 0; itNodes->more(); k++)
2528 const SMDS_MeshNode* node = itNodes->next();
2529 elemType.myNodes[ k ] = static_cast< const SMDS_MeshNode*> ( newNodes[ node->GetID() ]);
2532 // creates a corresponding element on existent nodes in new mesh
2533 newElems[ elem->GetID() ] =
2534 newEditor.AddElement( elemType.myNodes, elemType.Init( elem, /*basicOnly=*/false ));
2536 newEditor.ClearLastCreated(); // forget the history
2539 // create groups of just added elements
2540 SMESH::SMESH_Group_var newGroup;
2541 SMESH::ElementType groupType;
2542 if ( theCommonGroups )
2545 const char* typeNames[] = { "All","Nodes","Edges","Faces","Volumes","0DElems","Balls" };
2546 { // check of typeNames: compilation failure mains that NB_ELEMENT_TYPES changed:
2547 const int nbNames = sizeof(typeNames) / sizeof(const char*);
2548 int _assert[( nbNames == SMESH::NB_ELEMENT_TYPES ) ? 2 : -1 ]; _assert[0]=_assert[1]=0;
2551 SMESH::long_array_var curState = newMesh->GetNbElementsByType();
2553 for( groupType = SMESH::NODE;
2554 groupType < SMESH::NB_ELEMENT_TYPES;
2555 groupType = (SMESH::ElementType)( groupType + 1 ))
2557 if ( curState[ groupType ] <= prevState[ groupType ])
2558 continue; // no elements of groupType added from the i-th mesh
2560 // make a group name
2561 std::string groupName = "Gr";
2562 SALOMEDS::SObject_wrap meshSO = ObjectToSObject( theMeshesArray[i] );
2564 CORBA::String_var name = meshSO->GetName();
2568 groupName += typeNames[ groupType ];
2570 // make and fill a group
2571 newGroup = newImpl->CreateGroup( groupType, groupName.c_str() );
2572 std::vector< const SMDS_MeshElement* > & elemVec =
2573 ( groupType == SMESH::NODE ) ? newNodes : newElems;
2574 if ( SMESH_Group_i* grp_i = SMESH::DownCast<SMESH_Group_i*>( newGroup ))
2576 if ( SMESHDS_Group* grpDS = dynamic_cast<SMESHDS_Group*>( grp_i->GetGroupDS() ))
2578 for ( size_t j = 0; j < elemVec.size(); ++j )
2580 if ( elemVec[j] && elemVec[j]->GetType() == grpDS->GetType() )
2581 grpDS->Add( elemVec[j] );
2585 listOfNewGroups.clear();
2586 listOfNewGroups.push_back( newGroup );
2587 groupsMap.insert( std::make_pair( TNameAndType( groupName, groupType ),
2592 if ( SMESH_Mesh_i* initImpl = SMESH::DownCast<SMESH_Mesh_i*>( theMeshesArray[i] ))
2596 SMESH::SMESH_GroupBase_ptr group;
2597 CORBA::String_var groupName;
2598 SMESH::long_array_var newIDs = new SMESH::long_array();
2600 // loop on groups of a source mesh
2601 SMESH::ListOfGroups_var listOfGroups = initImpl->GetGroups();
2602 for ( CORBA::ULong iG = 0; iG < listOfGroups->length(); iG++ )
2604 group = listOfGroups[iG];
2605 groupType = group->GetType();
2606 groupName = group->GetName();
2607 std::string name = groupName.in();
2609 // convert a list of IDs
2610 newIDs->length( group->Size() );
2611 std::vector< const SMDS_MeshElement* > & elemVec =
2612 ( groupType == SMESH::NODE ) ? newNodes : newElems;
2613 SMDS_ElemIteratorPtr itGrElems = initImpl->GetElements( group, SMESH::ALL );
2615 while ( itGrElems->more() )
2617 const SMDS_MeshElement* elem = itGrElems->next();
2618 const SMDS_MeshElement* newElem = elemVec[ elem->GetID() ];
2620 newIDs[ nbElems++ ] = newElem->GetID();
2622 newIDs->length( nbElems );
2624 // check that a current group name and type don't have identical ones in final mesh
2625 listOfNewGroups.clear();
2626 TNameAndType nameAndType( name, groupType );
2627 TGroupsMap::iterator anIter = groupsMap.find( nameAndType );
2628 if ( anIter == groupsMap.end() )
2630 // add a new group in the mesh
2631 newGroup = newImpl->CreateGroup( groupType, groupName.in() );
2632 newGroup->Add( newIDs );
2634 listOfNewGroups.push_back( newGroup );
2635 groupsMap.insert( std::make_pair( nameAndType, listOfNewGroups ));
2637 else if ( theUniteIdenticalGroups )
2639 // unite identical groups
2640 TListOfNewGroups& aNewGroups = anIter->second;
2641 aNewGroups.front()->Add( newIDs );
2645 // rename identical groups
2646 newGroup = newImpl->CreateGroup( groupType, groupName );
2647 newGroup->Add( newIDs );
2649 TListOfNewGroups& newGroups = anIter->second;
2650 std::string newGroupName;
2651 if ( newGroups.size() == 1 )
2653 newGroupName = name + "_1";
2654 newGroups.front()->SetName( newGroupName.c_str() );
2656 newGroupName = name + "_" + SMESH_Comment( newGroups.size() + 1 );
2657 newGroup->SetName( newGroupName.c_str() );
2658 newGroups.push_back( newGroup );
2661 } // if an IDSource is a mesh
2664 if ( theMergeNodesAndElements ) // merge nodes
2666 TIDSortedNodeSet meshNodes; // no input nodes == treat all
2667 SMESH_MeshEditor::TListOfListOfNodes groupsOfNodes;
2668 newEditor.FindCoincidentNodes( meshNodes, theMergeTolerance, groupsOfNodes,
2669 /*SeparateCornersAndMedium=*/ false );
2670 newEditor.MergeNodes( groupsOfNodes );
2672 newEditor.MergeEqualElements();
2675 // Update Python script
2676 pythonDump << newMesh << " = " << this
2677 << "." << ( theCommonGroups ? "ConcatenateWithGroups" : "Concatenate" ) << "("
2678 << theMeshesArray << ", "
2679 << theUniteIdenticalGroups << ", "
2680 << theMergeNodesAndElements << ", "
2681 << TVar( theMergeTolerance ) << ")";
2683 pPythonDump.reset(); // enable python dump from GetGroups()
2685 // 0020577: EDF 1164 SMESH: Bad dump of concatenate with create common groups
2686 if ( !newMesh->_is_nil() )
2688 SMESH::ListOfGroups_var groups = newMesh->GetGroups();
2691 // IPAL21468 Change icon of compound because it need not be computed.
2692 SALOMEDS::SObject_wrap meshSO = ObjectToSObject( newMesh );
2693 SetPixMap( meshSO, "ICON_SMESH_TREE_MESH" );
2695 newMeshDS->Modified();
2697 return newMesh._retn();
2700 //================================================================================
2702 * \brief Create a mesh by copying a part of another mesh
2703 * \param meshPart - a part of mesh to copy
2704 * \param toCopyGroups - to create in the new mesh groups
2705 * the copied elements belongs to
2706 * \param toKeepIDs - to preserve IDs of the copied elements or not
2707 * \retval SMESH::SMESH_Mesh_ptr - the new mesh
2709 //================================================================================
2711 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CopyMesh(SMESH::SMESH_IDSource_ptr meshPart,
2712 const char* meshName,
2713 CORBA::Boolean toCopyGroups,
2714 CORBA::Boolean toKeepIDs)
2716 Unexpect aCatch(SALOME_SalomeException);
2718 TPythonDump* pyDump = new TPythonDump; // prevent dump from CreateMesh()
2720 // 1. Get source mesh
2722 if ( CORBA::is_nil( meshPart ))
2723 THROW_SALOME_CORBA_EXCEPTION( "bad IDSource", SALOME::BAD_PARAM );
2725 SMESH::SMESH_Mesh_var srcMesh = meshPart->GetMesh();
2726 SMESH_Mesh_i* srcMesh_i = SMESH::DownCast<SMESH_Mesh_i*>( srcMesh );
2728 THROW_SALOME_CORBA_EXCEPTION( "bad mesh of IDSource", SALOME::BAD_PARAM );
2730 SMESHDS_Mesh* srcMeshDS = srcMesh_i->GetImpl().GetMeshDS();
2732 // 2. Make a new mesh
2734 SMESH::SMESH_Mesh_var newMesh = CreateMesh(GEOM::GEOM_Object::_nil());
2735 SMESH_Mesh_i* newMesh_i = SMESH::DownCast<SMESH_Mesh_i*>( newMesh );
2737 THROW_SALOME_CORBA_EXCEPTION( "can't create a mesh", SALOME::INTERNAL_ERROR );
2738 SALOMEDS::SObject_wrap meshSO = ObjectToSObject( newMesh );
2739 if ( !meshSO->_is_nil() )
2741 SetName( meshSO, meshName, "Mesh" );
2742 SetPixMap( meshSO, "ICON_SMESH_TREE_MESH_IMPORTED");
2744 SMESHDS_Mesh* newMeshDS = newMesh_i->GetImpl().GetMeshDS();
2745 ::SMESH_MeshEditor editor( &newMesh_i->GetImpl() );
2746 ::SMESH_MeshEditor::ElemFeatures elemType;
2748 // 3. Get elements to copy
2750 SMDS_ElemIteratorPtr srcElemIt; SMDS_NodeIteratorPtr srcNodeIt;
2751 TIDSortedElemSet srcElems;
2752 SMESH::array_of_ElementType_var srcElemTypes = meshPart->GetTypes();
2753 if ( SMESH::DownCast<SMESH_Mesh_i*>( meshPart ))
2756 srcElemIt = srcMeshDS->elementsIterator();
2757 srcNodeIt = srcMeshDS->nodesIterator();
2761 SMESH::long_array_var ids = meshPart->GetIDs();
2762 if ( srcElemTypes->length() == 1 && srcElemTypes[0] == SMESH::NODE ) // group of nodes
2764 for ( CORBA::ULong i=0; i < ids->length(); i++ )
2765 if ( const SMDS_MeshElement * elem = srcMeshDS->FindNode( ids[i] ))
2766 srcElems.insert( elem );
2770 for ( CORBA::ULong i = 0; i < ids->length(); i++ )
2771 if ( const SMDS_MeshElement * elem = srcMeshDS->FindElement( ids[i] ))
2772 srcElems.insert( elem );
2774 if ( srcElems.empty() )
2775 return newMesh._retn();
2777 typedef SMDS_SetIterator< SMDS_pElement, TIDSortedElemSet::const_iterator > ElIter;
2778 srcElemIt = SMDS_ElemIteratorPtr( new ElIter( srcElems.begin(), srcElems.end() ));
2783 typedef map<SMDS_pElement, SMDS_pElement, TIDCompare> TE2EMap;
2784 TE2EMap e2eMapByType[ SMDSAbs_NbElementTypes ];
2785 TE2EMap& n2nMap = e2eMapByType[ SMDSAbs_Node ];
2787 const SMDS_MeshNode *nSrc, *nTgt;
2788 vector< const SMDS_MeshNode* > nodes;
2789 while ( srcElemIt->more() )
2791 const SMDS_MeshElement * elem = srcElemIt->next();
2793 nodes.resize( elem->NbNodes());
2794 SMDS_ElemIteratorPtr nIt = elem->nodesIterator();
2796 for ( iN = 0; nIt->more(); ++iN )
2798 nSrc = static_cast<const SMDS_MeshNode*>( nIt->next() );
2799 nTgt = newMeshDS->FindNode( nSrc->GetID());
2801 nTgt = newMeshDS->AddNodeWithID( nSrc->X(), nSrc->Y(), nSrc->Z(), nSrc->GetID());
2806 for ( iN = 0; nIt->more(); ++iN )
2808 nSrc = static_cast<const SMDS_MeshNode*>( nIt->next() );
2809 TE2EMap::iterator n2n = n2nMap.insert( make_pair( nSrc, SMDS_pNode(0) )).first;
2811 n2n->second = newMeshDS->AddNode( nSrc->X(), nSrc->Y(), nSrc->Z() );
2812 nodes[ iN ] = (const SMDS_MeshNode*) n2n->second;
2816 if ( elem->GetType() != SMDSAbs_Node )
2818 elemType.Init( elem, /*basicOnly=*/false );
2819 if ( toKeepIDs ) elemType.SetID( elem->GetID() );
2821 const SMDS_MeshElement * newElem = editor.AddElement( nodes, elemType );
2822 if ( toCopyGroups && !toKeepIDs )
2823 e2eMapByType[ elem->GetType() ].insert( make_pair( elem, newElem ));
2825 } // while ( srcElemIt->more() )
2827 // 4(b). Copy free nodes
2829 if ( srcNodeIt && srcMeshDS->NbNodes() != newMeshDS->NbNodes() )
2831 while ( srcNodeIt->more() )
2833 nSrc = srcNodeIt->next();
2834 if ( nSrc->NbInverseElements() == 0 )
2837 nTgt = newMeshDS->AddNodeWithID( nSrc->X(), nSrc->Y(), nSrc->Z(), nSrc->GetID());
2839 n2nMap[ nSrc ] = newMeshDS->AddNode( nSrc->X(), nSrc->Y(), nSrc->Z() );
2846 int nbNewGroups = 0;
2849 SMESH_Mesh::GroupIteratorPtr gIt = srcMesh_i->GetImpl().GetGroups();
2850 while ( gIt->more() )
2852 SMESH_Group* group = gIt->next();
2853 const SMESHDS_GroupBase* groupDS = group->GetGroupDS();
2855 // Check group type. We copy nodal groups containing nodes of copied element
2856 SMDSAbs_ElementType groupType = groupDS->GetType();
2857 if ( groupType != SMDSAbs_Node &&
2858 newMeshDS->GetMeshInfo().NbElements( groupType ) == 0 )
2859 continue; // group type differs from types of meshPart
2861 // Find copied elements in the group
2862 vector< const SMDS_MeshElement* > groupElems;
2863 SMDS_ElemIteratorPtr eIt = groupDS->GetElements();
2866 const SMDS_MeshElement* foundElem;
2867 if ( groupType == SMDSAbs_Node )
2869 while ( eIt->more() )
2870 if (( foundElem = newMeshDS->FindNode( eIt->next()->GetID() )))
2871 groupElems.push_back( foundElem );
2875 while ( eIt->more() )
2876 if (( foundElem = newMeshDS->FindElement( eIt->next()->GetID() )))
2877 groupElems.push_back( foundElem );
2882 TE2EMap & e2eMap = e2eMapByType[ groupDS->GetType() ];
2883 if ( e2eMap.empty() ) continue;
2884 int minID = e2eMap.begin()->first->GetID();
2885 int maxID = e2eMap.rbegin()->first->GetID();
2886 TE2EMap::iterator e2e;
2887 while ( eIt->more() && groupElems.size() < e2eMap.size())
2889 const SMDS_MeshElement* e = eIt->next();
2890 if ( e->GetID() < minID || e->GetID() > maxID ) continue;
2891 if ((e2e = e2eMap.find( e )) != e2eMap.end())
2892 groupElems.push_back( e2e->second );
2896 if ( !groupElems.empty() )
2898 SMESH::SMESH_Group_var newGroupObj =
2899 newMesh->CreateGroup( SMESH::ElementType(groupType), group->GetName() );
2900 if ( SMESH_GroupBase_i* newGroup_i = SMESH::DownCast<SMESH_GroupBase_i*>( newGroupObj))
2902 SMESHDS_GroupBase * newGroupDS = newGroup_i->GetGroupDS();
2903 SMDS_MeshGroup& smdsGroup = ((SMESHDS_Group*)newGroupDS)->SMDSGroup();
2904 for ( unsigned i = 0; i < groupElems.size(); ++i )
2905 smdsGroup.Add( groupElems[i] );
2913 newMeshDS->Modified();
2915 *pyDump << newMesh << " = " << this
2916 << ".CopyMesh( " << meshPart << ", "
2917 << "'" << meshName << "', "
2918 << toCopyGroups << ", "
2919 << toKeepIDs << ")";
2921 delete pyDump; pyDump = 0; // allow dump in GetGroups()
2923 if ( nbNewGroups > 0 ) // dump created groups
2924 SMESH::ListOfGroups_var groups = newMesh->GetGroups();
2926 return newMesh._retn();
2929 //================================================================================
2931 * \brief Get version of MED format being used.
2933 //================================================================================
2935 char* SMESH_Gen_i::GetMEDFileVersion()
2937 MED::TInt majeur, mineur, release;
2938 majeur = mineur = release = 0;
2939 MED::GetVersionRelease(majeur, mineur, release);
2940 std::ostringstream version;
2941 version << majeur << "." << mineur << "." << release;
2942 return CORBA::string_dup( version.str().c_str() );
2945 //================================================================================
2947 * SMESH_Gen_i::GetMEDVersion
2949 * Get MED version of the file by its name
2951 //================================================================================
2952 char* SMESH_Gen_i::GetMEDVersion(const char* theFileName)
2954 std::string version = MED::GetMEDVersion( theFileName );
2955 return CORBA::string_dup( version.c_str() );
2958 //================================================================================
2960 * SMESH_Gen_i::CheckCompatibility
2962 * Check compatibility of file with MED format being used.
2964 //================================================================================
2965 CORBA::Boolean SMESH_Gen_i::CheckCompatibility(const char* theFileName)
2967 return MED::CheckCompatibility( theFileName );
2970 //================================================================================
2972 * SMESH_Gen_i::GetMeshNames
2974 * Get names of meshes defined in file with the specified name
2976 //================================================================================
2977 SMESH::string_array* SMESH_Gen_i::GetMeshNames(const char* theFileName)
2979 SMESH::string_array_var aResult = new SMESH::string_array();
2980 MED::PWrapper aMed = MED::CrWrapperR( theFileName );
2982 MED::TInt aNbMeshes = aMed->GetNbMeshes( &anErr );
2984 aResult->length( aNbMeshes );
2985 for( MED::TInt i = 0; i < aNbMeshes; i++ ) {
2986 MED::PMeshInfo aMeshInfo = aMed->GetPMeshInfo( i+1 );
2987 aResult[i] = CORBA::string_dup( aMeshInfo->GetName().c_str() );
2990 return aResult._retn();
2993 //=============================================================================
2997 * Save SMESH module's data
2999 //=============================================================================
3000 SALOMEDS::TMPFile* SMESH_Gen_i::Save( SALOMEDS::SComponent_ptr theComponent,
3004 if (!myStudyContext)
3007 // Store study contents as a set of python commands
3010 SALOMEDS::Study_var aStudy = getStudyServant();
3012 // Declare a byte stream
3013 SALOMEDS::TMPFile_var aStreamFile;
3015 // Obtain a temporary dir
3016 TCollection_AsciiString tmpDir =
3017 ( isMultiFile ) ? TCollection_AsciiString( ( char* )theURL ) : ( char* )SALOMEDS_Tool::GetTmpDir().c_str();
3019 // Create a sequence of files processed
3020 SALOMEDS_Tool::ListOfFiles aFileSeq;
3021 aFileSeq.reserve( NUM_TMP_FILES );
3023 TCollection_AsciiString aStudyName( "" );
3025 aStudyName = ( (char*)SALOMEDS_Tool::GetNameFromPath( Kernel_Utils::encode(aStudy->URL()) ).c_str() );
3027 // Set names of temporary files
3028 TCollection_AsciiString filename =
3029 aStudyName + TCollection_AsciiString( "_SMESH.hdf" ); // for SMESH data itself
3030 TCollection_AsciiString meshfile =
3031 aStudyName + TCollection_AsciiString( "_SMESH_Mesh.med" ); // for mesh data to be stored in MED file
3032 aFileSeq.push_back(CORBA::string_dup( filename.ToCString() ));
3033 aFileSeq.push_back(CORBA::string_dup( meshfile.ToCString() ));
3034 filename = tmpDir + filename;
3035 meshfile = tmpDir + meshfile;
3038 HDFdataset* aDataset;
3039 HDFgroup* aTopGroup;
3041 HDFgroup* aSubGroup;
3042 HDFgroup* aSubSubGroup;
3043 hdf_size aSize[ 1 ];
3046 //Remove the files if they exist: BugID: 11225
3047 #ifndef WIN32 /* unix functionality */
3048 TCollection_AsciiString cmd("rm -f \"");
3050 TCollection_AsciiString cmd("del /F \"");
3057 system(cmd.ToCString());
3059 // MED writer to be used by storage process
3060 DriverMED_W_SMESHDS_Mesh myWriter;
3061 myWriter.SetFile( meshfile.ToCString() );
3064 // SetStoreName() to groups before storing hypotheses to let them refer to
3065 // groups using "store name", which is "Group <group_persistent_id>"
3067 SALOMEDS::ChildIterator_wrap itBig = aStudy->NewChildIterator( theComponent );
3068 for ( ; itBig->More(); itBig->Next() ) {
3069 SALOMEDS::SObject_wrap gotBranch = itBig->Value();
3070 if ( gotBranch->Tag() > GetAlgorithmsRootTag() ) {
3071 CORBA::Object_var anObject = SObjectToObject( gotBranch );
3072 if ( !CORBA::is_nil( anObject ) ) {
3073 SMESH::SMESH_Mesh_var myMesh = SMESH::SMESH_Mesh::_narrow( anObject ) ;
3074 if ( !myMesh->_is_nil() ) {
3075 myMesh->Load(); // load from study file if not yet done
3076 TPythonDump pd; // not to dump GetGroups()
3077 SMESH::ListOfGroups_var groups = myMesh->GetGroups();
3078 pd << ""; // to avoid optimizing pd out
3079 for ( CORBA::ULong i = 0; i < groups->length(); ++i )
3081 SMESH_GroupBase_i* grImpl = SMESH::DownCast<SMESH_GroupBase_i*>( groups[i]);
3084 CORBA::String_var objStr = GetORB()->object_to_string( grImpl->_this() );
3085 int anId = myStudyContext->findId( string( objStr.in() ) );
3087 sprintf( grpName, "Group %d", anId );
3088 SMESHDS_GroupBase* aGrpBaseDS = grImpl->GetGroupDS();
3089 aGrpBaseDS->SetStoreName( grpName );
3099 // ---> create HDF file
3100 aFile = new HDFfile( (char*) filename.ToCString() );
3101 aFile->CreateOnDisk();
3103 // --> iterator for top-level objects
3104 SALOMEDS::ChildIterator_wrap itBig = aStudy->NewChildIterator( theComponent );
3105 for ( ; itBig->More(); itBig->Next() ) {
3106 SALOMEDS::SObject_wrap gotBranch = itBig->Value();
3108 // --> hypotheses root branch (only one for the study)
3109 if ( gotBranch->Tag() == GetHypothesisRootTag() ) {
3110 // create hypotheses root HDF group
3111 aTopGroup = new HDFgroup( "Hypotheses", aFile );
3112 aTopGroup->CreateOnDisk();
3114 // iterator for all hypotheses
3115 SALOMEDS::ChildIterator_wrap it = aStudy->NewChildIterator( gotBranch );
3116 for ( ; it->More(); it->Next() ) {
3117 SALOMEDS::SObject_wrap mySObject = it->Value();
3118 CORBA::Object_var anObject = SObjectToObject( mySObject );
3119 if ( !CORBA::is_nil( anObject ) ) {
3120 SMESH::SMESH_Hypothesis_var myHyp = SMESH::SMESH_Hypothesis::_narrow( anObject );
3121 if ( !myHyp->_is_nil() ) {
3122 SMESH_Hypothesis_i* myImpl = dynamic_cast<SMESH_Hypothesis_i*>( GetServant( myHyp ).in() );
3124 CORBA::String_var hn = myHyp->GetName(), ln = myHyp->GetLibName();
3125 std::string hypname = hn.in();
3126 std::string libname = ln.in();
3128 // Needs for save crossplatform libname, i.e. parth of name ( ".dll" for
3129 // WIN32 and ".so" for X-system) must be deleted
3130 int libname_len = libname.length();
3132 if( libname_len > 4 )
3133 libname.resize( libname_len - 4 );
3135 // PAL17753 (Regression: missing hypothesis in restored study)
3136 // "lib" also should be removed from the beginning