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"
107 // to pass CORBA exception through SMESH_TRY
108 #define SMY_OWN_CATCH catch( SALOME::SALOME_Exception& se ) { throw se; }
109 #include "SMESH_TryCatch.hxx" // to include after OCC headers!
111 #include CORBA_SERVER_HEADER(SMESH_Group)
112 #include CORBA_SERVER_HEADER(SMESH_Filter)
113 #include CORBA_SERVER_HEADER(SMESH_MeshEditor)
116 #include <GEOMImpl_Types.hxx>
117 #include <GEOM_Client.hxx>
119 #include <Basics_Utils.hxx>
120 #include <Basics_DirUtils.hxx>
122 #include <OpUtil.hxx>
123 #include <SALOMEDS_Tool.hxx>
124 #include <SALOME_Container_i.hxx>
125 #include <SALOME_LifeCycleCORBA.hxx>
126 #include <SALOME_NamingService.hxx>
127 #include <Utils_CorbaException.hxx>
128 #include <Utils_ExceptHandlers.hxx>
129 #include <Utils_SINGLETON.hxx>
130 #include <utilities.h>
132 #include CORBA_CLIENT_HEADER(SALOME_ModuleCatalog)
133 #include CORBA_CLIENT_HEADER(SALOME_Session)
135 // helpers about SALOME::GenericObj
136 #include <SALOMEDS_wrap.hxx>
137 #include <SALOMEDS_Attributes_wrap.hxx>
138 #include <GEOM_wrap.hxx>
148 using SMESH::TPythonDump;
151 #define NUM_TMP_FILES 2
154 static int MYDEBUG = 0;
156 static int MYDEBUG = 0;
159 // Static variables definition
160 GEOM::GEOM_Gen_var SMESH_Gen_i::myGeomGen = GEOM::GEOM_Gen::_nil();
161 CORBA::ORB_var SMESH_Gen_i::myOrb;
162 PortableServer::POA_var SMESH_Gen_i::myPoa;
163 SALOME_NamingService* SMESH_Gen_i::myNS = NULL;
164 SALOME_LifeCycleCORBA* SMESH_Gen_i::myLCC = NULL;
165 SMESH_Gen_i* SMESH_Gen_i::mySMESHGen = NULL;
168 const int nbElemPerDiagonal = 10;
170 //=============================================================================
172 * GetServant [ static ]
174 * Get servant of the CORBA object
176 //=============================================================================
178 PortableServer::ServantBase_var SMESH_Gen_i::GetServant( CORBA::Object_ptr theObject )
180 if( CORBA::is_nil( theObject ) || CORBA::is_nil( GetPOA() ) )
183 PortableServer::Servant aServant = GetPOA()->reference_to_servant( theObject );
186 catch (PortableServer::POA::ObjectNotActive &ex)
188 INFOS("GetServant: ObjectNotActive");
191 catch (PortableServer::POA::WrongAdapter &ex)
193 INFOS("GetServant: WrongAdapter: OK when several servants used to build several mesh in parallel...");
196 catch (PortableServer::POA::WrongPolicy &ex)
198 INFOS("GetServant: WrongPolicy");
203 INFOS( "GetServant - Unknown exception was caught!!!" );
208 //=============================================================================
210 * SObjectToObject [ static ]
212 * Get CORBA object corresponding to the SALOMEDS::SObject
214 //=============================================================================
216 CORBA::Object_var SMESH_Gen_i::SObjectToObject( SALOMEDS::SObject_ptr theSObject )
218 SALOMEDS::GenericAttribute_wrap anAttr;
219 CORBA::Object_var anObj;
220 if ( !theSObject->_is_nil() ) {
222 if( theSObject->FindAttribute( anAttr.inout(), "AttributeIOR" ) ) {
223 SALOMEDS::AttributeIOR_wrap anIOR = anAttr;
224 CORBA::String_var aValue = anIOR->Value();
225 if( strcmp( aValue, "" ) != 0 )
226 anObj = GetORB()->string_to_object( aValue );
230 INFOS( "SObjectToObject - Unknown exception was caught!!!" );
236 //=============================================================================
240 * Get SALOME_NamingService object
242 //=============================================================================
244 SALOME_NamingService* SMESH_Gen_i::GetNS()
246 if ( myNS == NULL ) {
247 myNS = SINGLETON_<SALOME_NamingService>::Instance();
248 ASSERT(SINGLETON_<SALOME_NamingService>::IsAlreadyExisting());
249 myNS->init_orb( GetORB() );
254 //=============================================================================
258 * Get SALOME_LifeCycleCORBA object
260 //=============================================================================
261 SALOME_LifeCycleCORBA* SMESH_Gen_i::GetLCC() {
262 if ( myLCC == NULL ) {
263 myLCC = new SALOME_LifeCycleCORBA( GetNS() );
269 //=============================================================================
271 * GetGeomEngine [ static ]
273 * Get GEOM::GEOM_Gen reference
275 //=============================================================================
276 GEOM::GEOM_Gen_var SMESH_Gen_i::GetGeomEngine() {
277 //CCRT GEOM::GEOM_Gen_var aGeomEngine =
278 //CCRT GEOM::GEOM_Gen::_narrow( GetLCC()->FindOrLoad_Component("FactoryServer","GEOM") );
279 //CCRT return aGeomEngine._retn();
280 if(CORBA::is_nil(myGeomGen))
282 Engines::EngineComponent_ptr temp=GetLCC()->FindOrLoad_Component("FactoryServer","GEOM");
283 myGeomGen=GEOM::GEOM_Gen::_narrow(temp);
288 //=============================================================================
290 * SMESH_Gen_i::SMESH_Gen_i
292 * Default constructor: not for use
294 //=============================================================================
296 SMESH_Gen_i::SMESH_Gen_i()
300 //=============================================================================
302 * SMESH_Gen_i::SMESH_Gen_i
304 * Standard constructor, used with Container
306 //=============================================================================
308 SMESH_Gen_i::SMESH_Gen_i( CORBA::ORB_ptr orb,
309 PortableServer::POA_ptr poa,
310 PortableServer::ObjectId* contId,
311 const char* instanceName,
312 const char* interfaceName )
313 : Engines_Component_i( orb, poa, contId, instanceName, interfaceName )
316 myOrb = CORBA::ORB::_duplicate(orb);
317 myPoa = PortableServer::POA::_duplicate(poa);
320 _id = myPoa->activate_object( _thisObj );
322 myStudyContext = new StudyContext;
324 myIsEmbeddedMode = false;
325 myIsEnablePublish = true;
326 myShapeReader = NULL; // shape reader
328 myIsHistoricalPythonDump = true;
329 myToForgetMeshDataOnHypModif = false;
331 // set it in standalone mode only
332 //OSD::SetSignal( true );
334 // 0020605: EDF 1190 SMESH: Display performance. 80 seconds for 52000 cells.
335 // find out mode (embedded or standalone) here else
336 // meshes created before calling SMESH_Client::GetSMESHGen(), which calls
337 // SMESH_Gen_i::SetEmbeddedMode(), have wrong IsEmbeddedMode flag
338 if ( SALOME_NamingService* ns = GetNS() )
340 CORBA::Object_var obj = ns->Resolve( "/Kernel/Session" );
341 SALOME::Session_var session = SALOME::Session::_narrow( obj ) ;
342 if ( !session->_is_nil() )
344 CORBA::String_var str_host = session->getHostname();
345 CORBA::Long s_pid = session->getPID();
346 string my_host = Kernel_Utils::GetHostname();
348 long my_pid = (long)_getpid();
350 long my_pid = (long) getpid();
352 SetEmbeddedMode( s_pid == my_pid && my_host == str_host.in() );
357 //=============================================================================
359 * SMESH_Gen_i::~SMESH_Gen_i
363 //=============================================================================
365 SMESH_Gen_i::~SMESH_Gen_i()
367 // delete hypothesis creators
368 map<string, GenericHypothesisCreator_i*>::iterator itHyp, itHyp2;
369 for (itHyp = myHypCreatorMap.begin(); itHyp != myHypCreatorMap.end(); itHyp++)
371 // same creator can be mapped under different names
372 GenericHypothesisCreator_i* creator = (*itHyp).second;
376 for (itHyp2 = itHyp; itHyp2 != myHypCreatorMap.end(); itHyp2++)
377 if ( creator == (*itHyp2).second )
378 (*itHyp2).second = 0;
380 myHypCreatorMap.clear();
382 // Clear study contexts data
383 delete myStudyContext;
385 // delete shape reader
387 delete myShapeReader;
389 //=============================================================================
391 * SMESH_Gen_i::getHypothesisCreator
393 * Get hypothesis creator
395 //=============================================================================
396 GenericHypothesisCreator_i* SMESH_Gen_i::getHypothesisCreator(const char* theHypName,
397 const char* theLibName,
398 std::string& thePlatformLibName)
399 throw (SALOME::SALOME_Exception)
401 std::string aPlatformLibName;
402 /* It's Need to translate lib name for WIN32 or X platform */
403 if ( theLibName && theLibName[0] != '\0' )
405 int libNameLen = strlen(theLibName);
406 //check for old format "libXXXXXXX.so"
407 if (libNameLen > 7 &&
408 !strncmp( theLibName, "lib", 3 ) &&
409 !strcmp( theLibName+libNameLen-3, ".so" ))
413 aPlatformLibName = std::string( theLibName+3, libNameLen-6 ) + ".dll";
414 #elif defined(__APPLE__)
415 aPlatformLibName = std::string( theLibName, libNameLen-3 ) + ".dylib";
417 aPlatformLibName = theLibName;
422 //try to use new format
424 aPlatformLibName = theLibName;
425 aPlatformLibName += ".dll";
426 #elif defined(__APPLE__)
427 aPlatformLibName = std::string( "lib" ) + std::string( theLibName ) + ".dylib";
429 aPlatformLibName = std::string( "lib" ) + std::string( theLibName ) + ".so";
433 thePlatformLibName = aPlatformLibName;
435 Unexpect aCatch(SALOME_SalomeException);
436 if(MYDEBUG) MESSAGE( "Create Hypothesis <" << theHypName << "> from " << aPlatformLibName);
438 typedef GenericHypothesisCreator_i* (*GetHypothesisCreator)(const char* );
439 GenericHypothesisCreator_i* aCreator;
442 // check, if creator for this hypothesis type already exists
443 if (myHypCreatorMap.find(string(theHypName)) == myHypCreatorMap.end())
445 // load plugin library
446 if(MYDEBUG) MESSAGE("Loading server meshers plugin library ...");
449 const wchar_t* path = Kernel_Utils::decode_s(aPlatformLibName);
451 const char* path = aPlatformLibName.c_str();
454 const char* path = aPlatformLibName.c_str();
456 LibHandle libHandle = LoadLib( path );
457 #if defined(WIN32) && defined(UNICODE)
462 // report any error, if occurred
464 const char* anError = dlerror();
465 throw(SALOME_Exception(anError));
467 throw(SALOME_Exception(LOCALIZED( "Can't load server meshers plugin library" )));
471 // get method, returning hypothesis creator
472 if(MYDEBUG) MESSAGE("Find GetHypothesisCreator() method ...");
473 GetHypothesisCreator procHandle =
474 (GetHypothesisCreator)GetProc( libHandle, "GetHypothesisCreator" );
477 throw(SALOME_Exception(LOCALIZED("bad hypothesis plugin library")));
478 UnLoadLib(libHandle);
481 // get hypothesis creator
482 if(MYDEBUG) MESSAGE("Get Hypothesis Creator for " << theHypName);
483 aCreator = procHandle(theHypName);
486 throw(SALOME_Exception(LOCALIZED("no such a hypothesis in this plugin")));
488 // map hypothesis creator to a hypothesis name
489 myHypCreatorMap[string(theHypName)] = aCreator;
494 return myHypCreatorMap[string(theHypName)];
497 catch (SALOME_Exception& S_ex)
499 THROW_SALOME_CORBA_EXCEPTION(S_ex.what(), SALOME::BAD_PARAM);
504 //=============================================================================
506 * SMESH_Gen_i::createHypothesis
508 * Create hypothesis of given type
510 //=============================================================================
511 SMESH::SMESH_Hypothesis_ptr SMESH_Gen_i::createHypothesis(const char* theHypName,
512 const char* theLibName)
514 SMESH_Hypothesis_i* myHypothesis_i = 0;
515 SMESH::SMESH_Hypothesis_var hypothesis_i;
516 std::string aPlatformLibName;
517 GenericHypothesisCreator_i* aCreator =
518 getHypothesisCreator(theHypName, theLibName, aPlatformLibName);
520 // create a new hypothesis object, store its ref. in studyContext
521 myHypothesis_i = aCreator->Create(myPoa, &myGen);
524 myHypothesis_i->SetLibName( aPlatformLibName.c_str() ); // for persistency assurance
525 CORBA::String_var hypName = myHypothesis_i->GetName();
526 myHypCreatorMap[ hypName.in() ] = aCreator;
528 // activate the CORBA servant of hypothesis
529 hypothesis_i = myHypothesis_i->_this();
530 int nextId = RegisterObject( hypothesis_i );
531 if(MYDEBUG) { MESSAGE( "Add hypo to map with id = "<< nextId ); }
532 else { nextId = 0; } // avoid "unused variable" warning in release mode
534 return hypothesis_i._retn();
537 //=============================================================================
539 * SMESH_Gen_i::createMesh
541 * Create empty mesh on shape
543 //=============================================================================
544 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::createMesh()
545 throw ( SALOME::SALOME_Exception )
547 Unexpect aCatch(SALOME_SalomeException);
548 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::createMesh" );
550 // Get or create the GEOM_Client instance
552 // create a new mesh object servant, store it in a map in study context
553 SMESH_Mesh_i* meshServant = new SMESH_Mesh_i( GetPOA(), this );
554 // create a new mesh object
555 if(MYDEBUG) MESSAGE("myIsEmbeddedMode " << myIsEmbeddedMode);
556 meshServant->SetImpl( myGen.CreateMesh( myIsEmbeddedMode ));
558 // activate the CORBA servant of Mesh
559 SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( meshServant->_this() );
560 int nextId = RegisterObject( mesh );
561 if(MYDEBUG) { MESSAGE( "Add mesh to map with id = "<< nextId); }
562 else { nextId = 0; } // avoid "unused variable" warning in release mode
565 catch (SALOME_Exception& S_ex) {
566 THROW_SALOME_CORBA_EXCEPTION( S_ex.what(), SALOME::BAD_PARAM );
568 return SMESH::SMESH_Mesh::_nil();
571 //=============================================================================
573 * SMESH_Gen_i::GetShapeReader
577 //=============================================================================
578 GEOM_Client* SMESH_Gen_i::GetShapeReader()
580 // create shape reader if necessary
581 if ( !myShapeReader )
582 myShapeReader = new GEOM_Client(GetContainerRef());
583 ASSERT( myShapeReader );
584 return myShapeReader;
587 //=============================================================================
589 * SMESH_Gen_i::SetGeomEngine
591 * Set GEOM::GEOM_Gen reference
593 //=============================================================================
594 //GEOM::GEOM_Gen_ptr SMESH_Gen_i::SetGeomEngine( const char* containerLoc )
595 void SMESH_Gen_i::SetGeomEngine( GEOM::GEOM_Gen_ptr geomcompo )
597 //Engines::Component_ptr temp=GetLCC()->FindOrLoad_Component(containerLoc,"GEOM");
598 //myGeomGen=GEOM::GEOM_Gen::_narrow(temp);
599 myGeomGen=GEOM::GEOM_Gen::_duplicate(geomcompo);
603 //=============================================================================
605 * SMESH_Gen_i::SetEmbeddedMode
609 //=============================================================================
611 void SMESH_Gen_i::SetEmbeddedMode( CORBA::Boolean theMode )
613 myIsEmbeddedMode = theMode;
615 if ( !myIsEmbeddedMode ) {
616 //PAL10867: disable signals catching with "noexcepthandler" option
617 char* envNoCatchSignals = getenv("NOT_INTERCEPT_SIGNALS");
618 if (!envNoCatchSignals || !atoi(envNoCatchSignals))
623 char* envDisableFPE = getenv("DISABLE_FPE");
624 if (envDisableFPE && atoi(envDisableFPE))
629 OSD::SetSignal( raiseFPE );
631 // else OSD::SetSignal() is called in GUI
635 //=============================================================================
637 * SMESH_Gen_i::IsEmbeddedMode
641 //=============================================================================
643 CORBA::Boolean SMESH_Gen_i::IsEmbeddedMode()
645 return myIsEmbeddedMode;
648 //=============================================================================
650 * SMESH_Gen_i::SetEnablePublish
652 * Set enable publishing in the study
654 //=============================================================================
655 void SMESH_Gen_i::SetEnablePublish( CORBA::Boolean theIsEnablePublish )
657 myIsEnablePublish = theIsEnablePublish;
660 //=============================================================================
662 * SMESH_Gen_i::IsEnablePublish
664 * Check enable publishing
666 //=============================================================================
668 CORBA::Boolean SMESH_Gen_i::IsEnablePublish()
670 return myIsEnablePublish;
673 //=============================================================================
675 * SMESH_Gen_i::UpdateStudy
677 * Update study (needed at switching GEOM->SMESH)
679 //=============================================================================
681 void SMESH_Gen_i::UpdateStudy()
683 if ( !myStudyContext )
684 myStudyContext = new StudyContext;
686 SALOMEDS::Study_var aStudy = getStudyServant();
687 if ( !CORBA::is_nil( aStudy ) ) {
688 SALOMEDS::StudyBuilder_var aStudyBuilder = aStudy->NewBuilder();
689 SALOMEDS::SComponent_wrap GEOM_var = aStudy->FindComponent( "GEOM" );
690 if( !GEOM_var->_is_nil() )
691 aStudyBuilder->LoadWith( GEOM_var, GetGeomEngine() );
692 // NPAL16168, issue 0020210
693 // Let meshes update their data depending on GEOM groups that could change
694 CORBA::String_var compDataType = ComponentDataType();
695 SALOMEDS::SComponent_wrap me = aStudy->FindComponent( compDataType.in() );
696 if ( !me->_is_nil() ) {
697 SALOMEDS::ChildIterator_wrap anIter = aStudy->NewChildIterator( me );
698 for ( ; anIter->More(); anIter->Next() ) {
699 SALOMEDS::SObject_wrap so = anIter->Value();
700 CORBA::Object_var ior = SObjectToObject( so );
701 if ( SMESH_Mesh_i* mesh = SMESH::DownCast<SMESH_Mesh_i*>( ior ))
702 mesh->CheckGeomModif();
708 //=============================================================================
710 * SMESH_Gen_i::GetStudyContext
714 //=============================================================================
715 StudyContext* SMESH_Gen_i::GetStudyContext()
717 return myStudyContext;
720 //=============================================================================
722 * SMESH_Gen_i::CreateHypothesis
724 * Create hypothesis/algorothm of given type and publish it in the study
726 //=============================================================================
728 SMESH::SMESH_Hypothesis_ptr SMESH_Gen_i::CreateHypothesis( const char* theHypName,
729 const char* theLibName )
730 throw ( SALOME::SALOME_Exception )
732 Unexpect aCatch(SALOME_SalomeException);
733 // Create hypothesis/algorithm
734 SMESH::SMESH_Hypothesis_var hyp = this->createHypothesis( theHypName, theLibName );
736 // Publish hypothesis/algorithm in the study
737 if ( CanPublishInStudy( hyp ) ) {
738 SALOMEDS::SObject_wrap aSO = PublishHypothesis( hyp );
739 if ( !aSO->_is_nil() ) {
740 // Update Python script
741 TPythonDump() << aSO << " = " << this << ".CreateHypothesis('"
742 << theHypName << "', '" << theLibName << "')";
749 //================================================================================
751 * \brief Return a hypothesis holding parameter values corresponding either to the mesh
752 * existing on the given geometry or to size of the geometry.
753 * \param theHypType - hypothesis type name
754 * \param theLibName - plugin library name
755 * \param theMesh - The mesh of interest
756 * \param theGeom - The shape to get parameter values from
757 * \retval SMESH::SMESH_Hypothesis_ptr - The returned hypothesis may be the one existing
758 * in a study and used to compute the mesh, or a temporary one created just to pass
761 //================================================================================
763 SMESH::SMESH_Hypothesis_ptr
764 SMESH_Gen_i::GetHypothesisParameterValues (const char* theHypType,
765 const char* theLibName,
766 SMESH::SMESH_Mesh_ptr theMesh,
767 GEOM::GEOM_Object_ptr theGeom,
768 CORBA::Boolean byMesh)
769 throw ( SALOME::SALOME_Exception )
771 Unexpect aCatch(SALOME_SalomeException);
772 if ( byMesh && CORBA::is_nil( theMesh ) )
773 return SMESH::SMESH_Hypothesis::_nil();
774 if ( byMesh && CORBA::is_nil( theGeom ) )
775 return SMESH::SMESH_Hypothesis::_nil();
777 // -----------------------------------------------
778 // find hypothesis used to mesh theGeom
779 // -----------------------------------------------
781 // get mesh and shape
782 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
783 TopoDS_Shape shape = GeomObjectToShape( theGeom );
784 if ( byMesh && ( !meshServant || meshServant->NbNodes()==0 || shape.IsNull() ))
785 return SMESH::SMESH_Hypothesis::_nil();
786 ::SMESH_Mesh* mesh = meshServant ? &meshServant->GetImpl() : (::SMESH_Mesh*)0;
788 // create a temporary hypothesis to know its dimension
789 SMESH::SMESH_Hypothesis_var tmpHyp = this->createHypothesis( theHypType, theLibName );
790 SMESH_Hypothesis_i* hypServant = SMESH::DownCast<SMESH_Hypothesis_i*>( tmpHyp );
792 return SMESH::SMESH_Hypothesis::_nil();
793 ::SMESH_Hypothesis* hyp = hypServant->GetImpl();
796 // look for a hypothesis of theHypType used to mesh the shape
797 if ( myGen.GetShapeDim( shape ) == hyp->GetDim() )
800 SMESH::ListOfHypothesis_var aHypList = theMesh->GetHypothesisList( theGeom );
801 int nbLocalHyps = aHypList->length();
802 for ( int i = 0; i < nbLocalHyps; i++ ) {
803 CORBA::String_var hypName = aHypList[i]->GetName();
804 if ( strcmp( theHypType, hypName.in() ) == 0 ) // FOUND local!
805 return SMESH::SMESH_Hypothesis::_duplicate( aHypList[i] );
807 // check super shapes
808 TopTools_ListIteratorOfListOfShape itShape( mesh->GetAncestors( shape ));
809 while ( nbLocalHyps == 0 && itShape.More() ) {
810 GEOM::GEOM_Object_ptr geomObj = ShapeToGeomObject( itShape.Value() );
811 if ( ! CORBA::is_nil( geomObj )) {
812 SMESH::ListOfHypothesis_var aHypList = theMesh->GetHypothesisList( geomObj );
813 nbLocalHyps = aHypList->length();
814 for ( int i = 0; i < nbLocalHyps; i++ )
815 if ( strcmp( theHypType, aHypList[i]->GetName() ) == 0 ) // FOUND global!
816 return SMESH::SMESH_Hypothesis::_duplicate( aHypList[i] );
822 // let the temporary hypothesis find out somehow parameter values by mesh
823 if ( hyp->SetParametersByMesh( mesh, shape ))
824 return SMESH::SMESH_Hypothesis::_duplicate( tmpHyp );
829 diagonal = mesh->GetShapeDiagonalSize();
831 diagonal = ::SMESH_Mesh::GetShapeDiagonalSize( shape );
832 ::SMESH_Hypothesis::TDefaults dflts;
833 dflts._elemLength = diagonal / myGen.GetBoundaryBoxSegmentation();
834 dflts._nbSegments = myGen.GetDefaultNbSegments();
835 dflts._shape = &shape;
836 // let the temporary hypothesis initialize it's values
837 if ( hyp->SetParametersByDefaults( dflts, mesh ))
838 return SMESH::SMESH_Hypothesis::_duplicate( tmpHyp );
841 return SMESH::SMESH_Hypothesis::_nil();
844 //=============================================================================
846 * Returns \c True if a hypothesis is assigned to a sole sub-mesh in a current Study
847 * \param [in] theHyp - the hypothesis of interest
848 * \param [out] theMesh - the sole mesh using \a theHyp
849 * \param [out] theShape - the sole geometry \a theHyp is assigned to
850 * \return boolean - \c True if \a theMesh and \a theShape are sole using \a theHyp
852 * If two meshes on same shape have theHyp assigned to the same sub-shape, they are
853 * considered as SAME sub-mesh => result is \c true.
854 * This method ids used to initialize SMESHGUI_GenericHypothesisCreator with
855 * a shape to which an hyp being edited is assigned.
857 //=============================================================================
859 CORBA::Boolean SMESH_Gen_i::GetSoleSubMeshUsingHyp( SMESH::SMESH_Hypothesis_ptr theHyp,
860 SMESH::SMESH_Mesh_out theMesh,
861 GEOM::GEOM_Object_out theShape)
863 if ( CORBA::is_nil( theHyp ))
866 // get Mesh component SO
867 CORBA::String_var compDataType = ComponentDataType();
868 SALOMEDS::SComponent_wrap comp = getStudyServant()->FindComponent( compDataType.in() );
869 if ( CORBA::is_nil( comp ))
872 // look for child SO of meshes
873 SMESH::SMESH_Mesh_var foundMesh;
874 TopoDS_Shape foundShape;
876 SALOMEDS::ChildIterator_wrap meshIter = getStudyServant()->NewChildIterator( comp );
877 for ( ; meshIter->More() && isSole; meshIter->Next() )
879 SALOMEDS::SObject_wrap curSO = meshIter->Value();
880 CORBA::Object_var obj = SObjectToObject( curSO );
881 SMESH_Mesh_i* mesh_i = SMESH::DownCast< SMESH_Mesh_i* >( obj );
885 // look for a sole shape where theHyp is assigned
886 bool isHypFound = false;
887 const ShapeToHypothesis & s2hyps = mesh_i->GetImpl().GetMeshDS()->GetHypotheses();
888 ShapeToHypothesis::Iterator s2hypsIt( s2hyps );
889 for ( ; s2hypsIt.More() && isSole; s2hypsIt.Next() )
891 const THypList& hyps = s2hypsIt.Value();
892 THypList::const_iterator h = hyps.begin();
893 for ( ; h != hyps.end(); ++h )
894 if ( (*h)->GetID() == theHyp->GetId() )
896 if ( h != hyps.end()) // theHyp found
899 if ( ! foundShape.IsNull() &&
900 ! foundShape.IsSame( s2hypsIt.Key() )) // not a sole sub-shape
902 foundShape.Nullify();
906 foundShape = s2hypsIt.Key();
908 } // loop on assigned hyps
910 if ( isHypFound && !foundShape.IsNull() ) // a mesh using theHyp is found
912 if ( !foundMesh->_is_nil() ) // not a sole mesh
914 if ( !foundMesh->HasShapeToMesh() ||
915 !mesh_i ->HasShapeToMesh() )
917 isSole = ( foundMesh->HasShapeToMesh() == mesh_i->HasShapeToMesh() );
921 GEOM::GEOM_Object_var s1 = mesh_i ->GetShapeToMesh();
922 GEOM::GEOM_Object_var s2 = foundMesh->GetShapeToMesh();
923 isSole = s1->IsSame( s2 );
926 foundMesh = SMESH::SMESH_Mesh::_narrow( obj );
932 ! foundMesh->_is_nil() &&
933 ! foundShape.IsNull() )
935 theMesh = foundMesh._retn();
936 theShape = ShapeToGeomObject( foundShape );
937 return ( !theMesh->_is_nil() && !theShape->_is_nil() );
942 //=============================================================================
944 * Sets number of segments per diagonal of boundary box of geometry by which
945 * default segment length of appropriate 1D hypotheses is defined
947 //=============================================================================
949 void SMESH_Gen_i::SetBoundaryBoxSegmentation( CORBA::Long theNbSegments )
950 throw ( SALOME::SALOME_Exception )
952 if ( theNbSegments > 0 )
953 myGen.SetBoundaryBoxSegmentation( int( theNbSegments ));
955 THROW_SALOME_CORBA_EXCEPTION( "non-positive number of segments", SALOME::BAD_PARAM );
957 //=============================================================================
959 * \brief Sets default number of segments per edge
961 //=============================================================================
962 void SMESH_Gen_i::SetDefaultNbSegments(CORBA::Long theNbSegments)
963 throw ( SALOME::SALOME_Exception )
965 if ( theNbSegments > 0 )
966 myGen.SetDefaultNbSegments( int(theNbSegments) );
968 THROW_SALOME_CORBA_EXCEPTION( "non-positive number of segments", SALOME::BAD_PARAM );
971 //=============================================================================
975 //=============================================================================
977 void SMESH_Gen_i::SetOption(const char* name, const char* value)
979 if ( name && value && strlen( value ) > 0 )
982 if ( strcmp(name, "historical_python_dump") == 0 )
984 myIsHistoricalPythonDump = ( value[0] == '1' || toupper(value[0]) == 'T' ); // 1 || true
985 msgToGUI = "preferences/SMESH/historical_python_dump/";
986 msgToGUI += myIsHistoricalPythonDump ? "true" : "false";
988 else if ( strcmp(name, "forget_mesh_on_hyp_modif") == 0 )
990 myToForgetMeshDataOnHypModif = ( value[0] == '1' || toupper(value[0]) == 'T' ); // 1 || true
991 msgToGUI = "preferences/SMESH/forget_mesh_on_hyp_modif/";
992 msgToGUI += myToForgetMeshDataOnHypModif ? "true" : "false";
994 else if ( strcmp(name, "default_grp_color") == 0 )
998 // color must be presented as a string of following form:
999 if ( str.at(0) == '#' && str.length() == 7 ) { // hexadecimal color ("#ffaa00", for example)
1000 str = str.substr(1);
1001 for ( size_t i = 0; i < str.length()/2; i++ )
1002 if ( str.at(i*2) >= '0' && str.at(i*2) <= 'f' && str.at(i*2+1) >= '0' && str.at(i*2+1) <= 'f' )
1003 color.push_back( strtol( str.substr( i*2, 2 ).c_str(), NULL, 16 ) );
1005 else if ( value ) { // rgb color ("255,170,0", for example)
1006 string tempValue( value );
1007 char* colorValue = strtok( &tempValue[0], "," );
1008 while ( colorValue != NULL ) {
1009 int c_value = atoi( colorValue );
1010 if ( c_value >= 0 && c_value <= 255 )
1011 color.push_back( c_value );
1012 colorValue = strtok( NULL, "," );
1015 if ( color.size() == 3 ) { // color must have three valid component
1016 SMESHDS_GroupBase::SetDefaultColor( Quantity_Color( color[0]/255., color[1]/255., color[2]/255., Quantity_TOC_RGB ) );
1017 myDefaultGroupColor = value;
1018 msgToGUI = "preferences/SMESH/default_grp_color/";
1023 // update preferences in case if SetOption() is invoked from python console
1024 if ( !msgToGUI.empty() )
1026 CORBA::Object_var obj = SMESH_Gen_i::GetNS()->Resolve( "/Kernel/Session" );
1027 SALOME::Session_var session = SALOME::Session::_narrow( obj );
1028 if ( !CORBA::is_nil( session ) )
1029 session->emitMessageOneWay(msgToGUI.c_str());
1034 //=============================================================================
1036 Return an option value
1038 //=============================================================================
1040 char* SMESH_Gen_i::GetOption(const char* name)
1044 if ( strcmp(name, "historical_python_dump") == 0 )
1046 return CORBA::string_dup( myIsHistoricalPythonDump ? "true" : "false" );
1048 if ( strcmp(name, "forget_mesh_on_hyp_modif") == 0 )
1050 return CORBA::string_dup( myToForgetMeshDataOnHypModif ? "true" : "false" );
1052 if ( strcmp(name, "default_grp_color") == 0 )
1054 return CORBA::string_dup( myDefaultGroupColor.c_str() );
1057 return CORBA::string_dup( "" );
1060 //=============================================================================
1062 * SMESH_Gen_i::CreateMesh
1064 * Create empty mesh on a shape and publish it in the study
1066 //=============================================================================
1068 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMesh( GEOM::GEOM_Object_ptr theShapeObject )
1069 throw ( SALOME::SALOME_Exception )
1071 Unexpect aCatch(SALOME_SalomeException);
1072 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::CreateMesh" );
1074 SMESH::SMESH_Mesh_var mesh = this->createMesh();
1076 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( mesh );
1077 ASSERT( meshServant );
1078 meshServant->SetShape( theShapeObject );
1080 // publish mesh in the study
1081 if ( CanPublishInStudy( mesh ) ) {
1082 SALOMEDS::StudyBuilder_var aStudyBuilder = getStudyServant()->NewBuilder();
1083 aStudyBuilder->NewCommand(); // There is a transaction
1084 SALOMEDS::SObject_wrap aSO = PublishMesh( mesh.in() );
1085 aStudyBuilder->CommitCommand();
1086 if ( !aSO->_is_nil() ) {
1087 // Update Python script
1088 TPythonDump() << aSO << " = " << this << ".CreateMesh(" << theShapeObject << ")";
1092 return mesh._retn();
1095 //=============================================================================
1097 * SMESH_Gen_i::CreateEmptyMesh
1101 //=============================================================================
1103 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateEmptyMesh()
1104 throw ( SALOME::SALOME_Exception )
1106 Unexpect aCatch(SALOME_SalomeException);
1107 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::CreateMesh" );
1109 SMESH::SMESH_Mesh_var mesh = this->createMesh();
1111 // publish mesh in the study
1112 if ( CanPublishInStudy( mesh ) ) {
1113 SALOMEDS::StudyBuilder_var aStudyBuilder = getStudyServant()->NewBuilder();
1114 aStudyBuilder->NewCommand(); // There is a transaction
1115 SALOMEDS::SObject_wrap aSO = PublishMesh( mesh.in() );
1116 aStudyBuilder->CommitCommand();
1117 if ( !aSO->_is_nil() ) {
1118 // Update Python script
1119 TPythonDump() << aSO << " = " << this << ".CreateEmptyMesh()";
1123 return mesh._retn();
1128 //================================================================================
1130 * \brief Throws an exception in case if the file can't be read
1132 //================================================================================
1134 void checkFileReadable( const char* theFileName ) throw ( SALOME::SALOME_Exception )
1136 SMESH_File f ( theFileName );
1139 if ( !f.error().empty() )
1140 THROW_SALOME_CORBA_EXCEPTION( f.error().c_str(), SALOME::BAD_PARAM);
1142 THROW_SALOME_CORBA_EXCEPTION
1143 (( SMESH_Comment("Can't open for reading the file ") << theFileName ).c_str(),
1144 SALOME::BAD_PARAM );
1149 //=============================================================================
1151 * SMESH_Gen_i::CreateMeshFromUNV
1153 * Create mesh and import data from UNV file
1155 //=============================================================================
1157 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMeshesFromUNV( const char* theFileName )
1158 throw ( SALOME::SALOME_Exception )
1160 Unexpect aCatch(SALOME_SalomeException);
1162 checkFileReadable( theFileName );
1164 SMESH::SMESH_Mesh_var aMesh = createMesh();
1166 // publish mesh in the study
1167 if ( CanPublishInStudy( aMesh ) ) {
1168 SALOMEDS::StudyBuilder_var aStudyBuilder = getStudyServant()->NewBuilder();
1169 aStudyBuilder->NewCommand(); // There is a transaction
1170 SALOMEDS::SObject_wrap aSO = PublishMesh( aMesh.in(), aFileName.c_str() );
1171 aStudyBuilder->CommitCommand();
1172 if ( !aSO->_is_nil() ) {
1173 // Update Python script
1174 TPythonDump() << aSO << " = " << this << ".CreateMeshesFromUNV(r'" << theFileName << "')";
1178 SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
1180 aServant->ImportUNVFile( theFileName );
1182 // Dump creation of groups
1183 SMESH::ListOfGroups_var groups = aServant->GetGroups();
1185 aServant->GetImpl().GetMeshDS()->Modified();
1186 return aMesh._retn();
1189 //=============================================================================
1191 * SMESH_Gen_i::CreateMeshFromMED
1193 * Create mesh and import data from MED file
1195 //=============================================================================
1197 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromMEDorSAUV( const char* theFileName,
1198 SMESH::DriverMED_ReadStatus& theStatus,
1199 const char* theCommandNameForPython,
1200 const char* theFileNameForPython)
1203 char bname[ _MAX_FNAME ];
1204 _splitpath( theFileNameForPython, NULL, NULL, bname, NULL );
1205 string aFileName = bname;
1207 string aFileName = basename( const_cast<char *>(theFileNameForPython) );
1209 // Retrieve mesh names from the file
1210 DriverMED_R_SMESHDS_Mesh myReader;
1211 myReader.SetFile( theFileName );
1212 myReader.SetMeshId( -1 );
1213 Driver_Mesh::Status aStatus;
1214 list<string> aNames = myReader.GetMeshNames(aStatus);
1215 SMESH::mesh_array_var aResult = new SMESH::mesh_array();
1216 theStatus = (SMESH::DriverMED_ReadStatus)aStatus;
1218 { // open a new scope to make aPythonDump die before PythonDump in SMESH_Mesh::GetGroups()
1221 TPythonDump aPythonDump;
1222 aPythonDump << "([";
1224 if (theStatus == SMESH::DRS_OK) {
1225 SALOMEDS::StudyBuilder_var aStudyBuilder;
1226 aStudyBuilder = getStudyServant()->NewBuilder();
1227 aStudyBuilder->NewCommand(); // There is a transaction
1229 aResult->length( aNames.size() );
1232 // Iterate through all meshes and create mesh objects
1233 for ( list<string>::iterator it = aNames.begin(); it != aNames.end(); it++ )
1236 if (i > 0) aPythonDump << ", ";
1239 SMESH::SMESH_Mesh_var mesh = createMesh();
1241 // publish mesh in the study
1242 SALOMEDS::SObject_wrap aSO;
1243 if ( CanPublishInStudy( mesh ) )
1244 // little trick: for MED file theFileName and theFileNameForPython are the same, but they are different for SAUV
1245 // - as names of meshes are stored in MED file, we use them for data publishing
1246 // - as mesh name is not stored in UNV file, we use file name as name of mesh when publishing data
1247 aSO = PublishMesh( mesh.in(), ( theFileName == theFileNameForPython ) ? (*it).c_str() : aFileName.c_str() );
1250 if ( !aSO->_is_nil() ) {
1253 aPythonDump << "mesh_" << i;
1256 // Read mesh data (groups are published automatically by ImportMEDFile())
1257 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( mesh ).in() );
1258 ASSERT( meshServant );
1259 SMESH::DriverMED_ReadStatus status1 =
1260 meshServant->ImportMEDFile( theFileName, (*it).c_str() );
1261 if (status1 > theStatus)
1262 theStatus = status1;
1264 aResult[i++] = SMESH::SMESH_Mesh::_duplicate( mesh );
1265 meshServant->GetImpl().GetMeshDS()->Modified();
1267 if ( !aStudyBuilder->_is_nil() )
1268 aStudyBuilder->CommitCommand();
1271 // Update Python script
1272 aPythonDump << "], status) = " << this << "." << theCommandNameForPython << "(r'" << theFileNameForPython << "')";
1274 // Dump creation of groups
1275 for ( CORBA::ULong i = 0; i < aResult->length(); ++i )
1276 SMESH::ListOfGroups_var groups = aResult[ i ]->GetGroups();
1278 return aResult._retn();
1281 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromMED( const char* theFileName,
1282 SMESH::DriverMED_ReadStatus& theStatus)
1283 throw ( SALOME::SALOME_Exception )
1285 Unexpect aCatch(SALOME_SalomeException);
1286 checkFileReadable( theFileName );
1288 SMESH::mesh_array* result = CreateMeshesFromMEDorSAUV(theFileName, theStatus, "CreateMeshesFromMED", theFileName);
1292 //=============================================================================
1294 * SMESH_Gen_i::CreateMeshFromSAUV
1296 * Create mesh and import data from SAUV file
1298 //=============================================================================
1300 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromSAUV( const char* theFileName,
1301 SMESH::DriverMED_ReadStatus& theStatus)
1302 throw ( SALOME::SALOME_Exception )
1304 Unexpect aCatch(SALOME_SalomeException);
1305 checkFileReadable( theFileName );
1307 std::string sauvfilename(theFileName);
1308 std::string medfilename(theFileName);
1309 medfilename += ".med";
1312 cmd = "%PYTHONBIN% ";
1317 cmd += "from medutilities import convert ; convert(r'" + sauvfilename + "', 'GIBI', 'MED', 1, r'" + medfilename + "')";
1319 system(cmd.c_str());
1320 SMESH::mesh_array* result = CreateMeshesFromMEDorSAUV(medfilename.c_str(), theStatus, "CreateMeshesFromSAUV", sauvfilename.c_str());
1322 cmd = "%PYTHONBIN% ";
1327 cmd += "from medutilities import my_remove ; my_remove(r'" + medfilename + "')";
1329 system(cmd.c_str());
1333 //=============================================================================
1335 * SMESH_Gen_i::CreateMeshFromSTL
1337 * Create mesh and import data from STL file
1339 //=============================================================================
1341 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMeshesFromSTL( const char* theFileName )
1342 throw ( SALOME::SALOME_Exception )
1344 Unexpect aCatch(SALOME_SalomeException);
1345 checkFileReadable( theFileName );
1347 SMESH::SMESH_Mesh_var aMesh = createMesh();
1350 char bname[ _MAX_FNAME ];
1351 _splitpath( theFileName, NULL, NULL, bname, NULL );
1352 string aFileName = bname;
1354 string aFileName = basename( const_cast<char *>(theFileName) );
1356 // publish mesh in the study
1357 if ( CanPublishInStudy( aMesh ) ) {
1358 SALOMEDS::StudyBuilder_var aStudyBuilder = getStudyServant()->NewBuilder();
1359 aStudyBuilder->NewCommand(); // There is a transaction
1360 SALOMEDS::SObject_wrap aSO = PublishInStudy( SALOMEDS::SObject::_nil(), aMesh.in(), aFileName.c_str() );
1361 aStudyBuilder->CommitCommand();
1362 if ( !aSO->_is_nil() ) {
1363 // Update Python script
1364 TPythonDump() << aSO << " = " << this << ".CreateMeshesFromSTL(r'" << theFileName << "')";
1368 SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
1370 aServant->ImportSTLFile( theFileName );
1371 aServant->GetImpl().GetMeshDS()->Modified();
1372 return aMesh._retn();
1375 //================================================================================
1377 * \brief Create meshes and import data from the CGSN file
1379 //================================================================================
1381 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromCGNS( const char* theFileName,
1382 SMESH::DriverMED_ReadStatus& theStatus)
1383 throw ( SALOME::SALOME_Exception )
1385 Unexpect aCatch(SALOME_SalomeException);
1386 checkFileReadable( theFileName );
1388 SMESH::mesh_array_var aResult = new SMESH::mesh_array();
1391 // Retrieve nb meshes from the file
1392 DriverCGNS_Read myReader;
1393 myReader.SetFile( theFileName );
1394 Driver_Mesh::Status aStatus;
1395 int nbMeshes = myReader.GetNbMeshes(aStatus);
1396 theStatus = (SMESH::DriverMED_ReadStatus)aStatus;
1398 aResult->length( nbMeshes );
1400 { // open a new scope to make aPythonDump die before PythonDump in SMESH_Mesh::GetGroups()
1403 TPythonDump aPythonDump;
1404 aPythonDump << "([";
1406 if (theStatus == SMESH::DRS_OK)
1408 SALOMEDS::StudyBuilder_var aStudyBuilder = getStudyServant()->NewBuilder();
1409 aStudyBuilder->NewCommand(); // There is a transaction
1413 // Iterate through all meshes and create mesh objects
1414 for ( ; i < nbMeshes; ++i )
1417 if (i > 0) aPythonDump << ", ";
1420 SMESH::SMESH_Mesh_var mesh = createMesh();
1421 aResult[i] = SMESH::SMESH_Mesh::_duplicate( mesh );
1423 // Read mesh data (groups are published automatically by ImportMEDFile())
1424 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( mesh ).in() );
1425 ASSERT( meshServant );
1427 SMESH::DriverMED_ReadStatus status1 =
1428 meshServant->ImportCGNSFile( theFileName, i, meshName );
1429 if (status1 > theStatus)
1430 theStatus = status1;
1432 meshServant->GetImpl().GetMeshDS()->Modified();
1433 // publish mesh in the study
1434 SALOMEDS::SObject_wrap aSO;
1435 if ( CanPublishInStudy( mesh ) )
1436 aSO = PublishMesh( mesh.in(), meshName.c_str() );
1439 if ( !aSO->_is_nil() ) {
1443 aPythonDump << "mesh_" << i;
1446 aStudyBuilder->CommitCommand();
1449 aPythonDump << "], status) = " << this << ".CreateMeshesFromCGNS(r'" << theFileName << "')";
1451 // Dump creation of groups
1452 for ( CORBA::ULong i = 0; i < aResult->length(); ++i )
1453 SMESH::ListOfGroups_var groups = aResult[ i ]->GetGroups();
1455 THROW_SALOME_CORBA_EXCEPTION("CGNS library is unavailable", SALOME::INTERNAL_ERROR);
1458 return aResult._retn();
1461 //================================================================================
1463 * \brief Create a mesh and import data from a GMF file
1465 //================================================================================
1467 SMESH::SMESH_Mesh_ptr
1468 SMESH_Gen_i::CreateMeshesFromGMF( const char* theFileName,
1469 CORBA::Boolean theMakeRequiredGroups,
1470 SMESH::ComputeError_out theError)
1471 throw ( SALOME::SALOME_Exception )
1473 Unexpect aCatch(SALOME_SalomeException);
1474 checkFileReadable( theFileName );
1476 SMESH::SMESH_Mesh_var aMesh = createMesh();
1478 char bname[ _MAX_FNAME ];
1479 _splitpath( theFileName, NULL, NULL, bname, NULL );
1480 string aFileName = bname;
1482 string aFileName = basename( const_cast<char *>(theFileName) );
1484 // publish mesh in the study
1485 if ( CanPublishInStudy( aMesh ) ) {
1486 SALOMEDS::StudyBuilder_var aStudyBuilder = getStudyServant()->NewBuilder();
1487 aStudyBuilder->NewCommand(); // There is a transaction
1488 SALOMEDS::SObject_wrap aSO = PublishInStudy( SALOMEDS::SObject::_nil(), aMesh.in(), aFileName.c_str() );
1489 aStudyBuilder->CommitCommand();
1490 if ( !aSO->_is_nil() ) {
1491 // Update Python script
1492 TPythonDump() << "("<< aSO << ", error) = " << this << ".CreateMeshesFromGMF(r'"
1493 << theFileName << "', "
1494 << theMakeRequiredGroups << " )";
1497 SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
1499 theError = aServant->ImportGMFFile( theFileName, theMakeRequiredGroups );
1500 aServant->GetImpl().GetMeshDS()->Modified();
1501 return aMesh._retn();
1505 //=============================================================================
1507 * SMESH_Gen_i::IsReadyToCompute
1509 * Returns true if mesh contains enough data to be computed
1511 //=============================================================================
1513 CORBA::Boolean SMESH_Gen_i::IsReadyToCompute( SMESH::SMESH_Mesh_ptr theMesh,
1514 GEOM::GEOM_Object_ptr theShapeObject )
1515 throw ( SALOME::SALOME_Exception )
1517 Unexpect aCatch(SALOME_SalomeException);
1518 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::IsReadyToCompute" );
1520 if ( CORBA::is_nil( theShapeObject ) )
1521 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1522 SALOME::BAD_PARAM );
1524 if ( CORBA::is_nil( theMesh ) )
1525 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
1526 SALOME::BAD_PARAM );
1530 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1531 ASSERT( meshServant );
1532 if ( meshServant ) {
1533 // get local TopoDS_Shape
1534 TopoDS_Shape myLocShape = GeomObjectToShape( theShapeObject );
1535 // call implementation
1536 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1537 return myGen.CheckAlgoState( myLocMesh, myLocShape );
1540 catch ( SALOME_Exception& S_ex ) {
1541 INFOS( "catch exception "<< S_ex.what() );
1546 //================================================================================
1548 * \brief Find SObject for an algo
1550 //================================================================================
1552 SALOMEDS::SObject_ptr SMESH_Gen_i::GetAlgoSO(const ::SMESH_Algo* algo)
1555 SALOMEDS::Study_var aStudy = getStudyServant();
1556 if ( !aStudy->_is_nil() ) {
1557 // find algo in the study
1558 CORBA::String_var compDataType = ComponentDataType();
1559 SALOMEDS::SComponent_wrap father = aStudy->FindComponent( compDataType.in() );
1560 if ( !father->_is_nil() ) {
1561 SALOMEDS::ChildIterator_wrap itBig = aStudy->NewChildIterator( father );
1562 for ( ; itBig->More(); itBig->Next() ) {
1563 SALOMEDS::SObject_wrap gotBranch = itBig->Value();
1564 if ( gotBranch->Tag() == GetAlgorithmsRootTag() ) {
1565 SALOMEDS::ChildIterator_wrap algoIt = aStudy->NewChildIterator( gotBranch );
1566 for ( ; algoIt->More(); algoIt->Next() ) {
1567 SALOMEDS::SObject_wrap algoSO = algoIt->Value();
1568 CORBA::Object_var algoIOR = SObjectToObject( algoSO );
1569 if ( !CORBA::is_nil( algoIOR )) {
1570 SMESH_Hypothesis_i* impl = SMESH::DownCast<SMESH_Hypothesis_i*>( algoIOR );
1571 if ( impl && impl->GetImpl() == algo )
1572 return algoSO._retn();
1574 } // loop on algo SO's
1577 } // SMESH component iterator
1581 return SALOMEDS::SObject::_nil();
1584 //================================================================================
1586 * \brief Return errors of mesh computation
1588 //================================================================================
1590 SMESH::compute_error_array* SMESH_Gen_i::GetComputeErrors( SMESH::SMESH_Mesh_ptr theMesh,
1591 GEOM::GEOM_Object_ptr theSubObject )
1592 throw ( SALOME::SALOME_Exception )
1594 Unexpect aCatch(SALOME_SalomeException);
1595 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetComputeErrors()" );
1597 if ( CORBA::is_nil( theSubObject ) && theMesh->HasShapeToMesh())
1598 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference", SALOME::BAD_PARAM );
1600 if ( CORBA::is_nil( theMesh ) )
1601 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1603 SMESH::compute_error_array_var error_array = new SMESH::compute_error_array;
1605 if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1608 if(theMesh->HasShapeToMesh())
1609 shape = GeomObjectToShape( theSubObject );
1611 shape = SMESH_Mesh::PseudoShape();
1613 ::SMESH_Mesh& mesh = meshServant->GetImpl();
1615 error_array->length( mesh.GetMeshDS()->MaxShapeIndex() );
1618 SMESH_subMesh *sm = mesh.GetSubMesh(shape);
1619 const bool includeSelf = true, complexShapeFirst = true;
1620 SMESH_subMeshIteratorPtr smIt = sm->getDependsOnIterator(includeSelf,
1622 while ( smIt->more() )
1625 // if ( sm->GetSubShape().ShapeType() == TopAbs_VERTEX )
1627 SMESH_ComputeErrorPtr error = sm->GetComputeError();
1628 if ( error && !error->IsOK() )
1630 if ( !( error->myAlgo ) &&
1631 !( error->myAlgo = sm->GetAlgo() ))
1633 SMESH::ComputeError & errStruct = error_array[ nbErr++ ];
1634 errStruct.code = -( error->myName < 0 ? error->myName + 1: error->myName ); // -1 -> 0
1635 errStruct.comment = error->myComment.c_str();
1636 errStruct.subShapeID = sm->GetId();
1637 SALOMEDS::SObject_wrap algoSO = GetAlgoSO( error->myAlgo );
1638 if ( !algoSO->_is_nil() ) {
1639 CORBA::String_var algoName = algoSO->GetName();
1640 errStruct.algoName = algoName;
1643 errStruct.algoName = error->myAlgo->GetName();
1645 errStruct.hasBadMesh = error->HasBadElems();
1648 error_array->length( nbErr );
1651 catch ( SALOME_Exception& S_ex ) {
1652 INFOS( "catch exception "<< S_ex.what() );
1655 return error_array._retn();
1658 //================================================================================
1660 * \brief Return mesh elements preventing computation of a subshape
1662 //================================================================================
1664 SMESH::MeshPreviewStruct*
1665 SMESH_Gen_i::GetBadInputElements( SMESH::SMESH_Mesh_ptr theMesh,
1666 CORBA::Short theSubShapeID )
1667 throw ( SALOME::SALOME_Exception )
1669 Unexpect aCatch(SALOME_SalomeException);
1670 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetBadInputElements()" );
1672 if ( CORBA::is_nil( theMesh ) )
1673 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1675 SMESH::MeshPreviewStruct_var result = new SMESH::MeshPreviewStruct;
1678 if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1680 // mesh implementation
1681 ::SMESH_Mesh& mesh = meshServant->GetImpl();
1682 // submesh by subshape id
1683 if ( SMESH_subMesh * sm = mesh.GetSubMeshContaining( theSubShapeID ))
1686 SMESH_ComputeErrorPtr error = sm->GetComputeError();
1687 if ( error && error->HasBadElems() )
1689 typedef map<const SMDS_MeshElement*, int > TNode2LocalIDMap;
1690 typedef TNode2LocalIDMap::iterator TNodeLocalID;
1692 // get nodes of elements and count elements
1693 TNode2LocalIDMap mapNode2LocalID;
1694 list< TNodeLocalID > connectivity;
1695 int i, nbElements = 0, nbConnNodes = 0;
1697 const list<const SMDS_MeshElement*>& badElems =
1698 static_cast<SMESH_BadInputElements*>( error.get() )->myBadElements;
1699 list<const SMDS_MeshElement*>::const_iterator elemIt = badElems.begin();
1700 list<const SMDS_MeshElement*>::const_iterator elemEnd = badElems.end();
1701 for ( ; elemIt != elemEnd; ++elemIt, ++nbElements )
1703 SMDS_ElemIteratorPtr nIt = (*elemIt)->nodesIterator();
1704 while ( nIt->more() )
1705 connectivity.push_back
1706 ( mapNode2LocalID.insert( make_pair( nIt->next(), ++nbConnNodes)).first );
1708 // fill node coords and assign local ids to the nodes
1709 int nbNodes = mapNode2LocalID.size();
1710 result->nodesXYZ.length( nbNodes );
1711 TNodeLocalID node2ID = mapNode2LocalID.begin();
1712 for ( i = 0; i < nbNodes; ++i, ++node2ID ) {
1713 node2ID->second = i;
1714 const SMDS_MeshNode* node = (const SMDS_MeshNode*) node2ID->first;
1715 result->nodesXYZ[i].x = node->X();
1716 result->nodesXYZ[i].y = node->Y();
1717 result->nodesXYZ[i].z = node->Z();
1719 // fill connectivity
1720 result->elementConnectivities.length( nbConnNodes );
1721 list< TNodeLocalID >::iterator connIt = connectivity.begin();
1722 for ( i = 0; i < nbConnNodes; ++i, ++connIt ) {
1723 result->elementConnectivities[i] = (*connIt)->second;
1725 // fill element types
1726 result->elementTypes.length( nbElements );
1727 for ( i = 0, elemIt = badElems.begin(); i <nbElements; ++i, ++elemIt )
1729 const SMDS_MeshElement* elem = *elemIt;
1730 result->elementTypes[i].SMDS_ElementType = (SMESH::ElementType) elem->GetType();
1731 result->elementTypes[i].isPoly = elem->IsPoly();
1732 result->elementTypes[i].nbNodesInElement = elem->NbNodes();
1738 catch ( SALOME_Exception& S_ex ) {
1739 INFOS( "catch exception "<< S_ex.what() );
1742 return result._retn();
1745 //================================================================================
1747 * \brief Create a group of elements preventing computation of a sub-shape
1749 //================================================================================
1751 SMESH::ListOfGroups*
1752 SMESH_Gen_i::MakeGroupsOfBadInputElements( SMESH::SMESH_Mesh_ptr theMesh,
1753 CORBA::Short theSubShapeID,
1754 const char* theGroupName )
1755 throw ( SALOME::SALOME_Exception )
1757 Unexpect aCatch(SALOME_SalomeException);
1759 SMESH::ListOfGroups_var groups;
1761 if ( CORBA::is_nil( theMesh ) )
1762 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1765 if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1767 groups = meshServant->MakeGroupsOfBadInputElements( theSubShapeID, theGroupName );
1768 TPythonDump() << groups << " = " << this
1769 << ".MakeGroupsOfBadInputElements( "
1770 << theMesh << ", " << theSubShapeID << ", '" << theGroupName << "' )";
1773 catch ( SALOME_Exception& S_ex ) {
1774 INFOS( "catch exception "<< S_ex.what() );
1776 return groups._retn();
1779 //================================================================================
1781 * \brief Returns errors of hypotheses definition
1782 * \param theMesh - the mesh
1783 * \param theSubObject - the main or sub- shape
1784 * \retval SMESH::algo_error_array* - sequence of errors
1786 //================================================================================
1788 SMESH::algo_error_array* SMESH_Gen_i::GetAlgoState( SMESH::SMESH_Mesh_ptr theMesh,
1789 GEOM::GEOM_Object_ptr theSubObject )
1790 throw ( SALOME::SALOME_Exception )
1792 Unexpect aCatch(SALOME_SalomeException);
1793 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetAlgoState()" );
1795 if ( CORBA::is_nil( theSubObject ) && theMesh->HasShapeToMesh())
1796 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference", SALOME::BAD_PARAM );
1798 if ( CORBA::is_nil( theMesh ) )
1799 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1801 SMESH::algo_error_array_var error_array = new SMESH::algo_error_array;
1803 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
1804 ASSERT( meshServant );
1805 if ( meshServant ) {
1806 TopoDS_Shape myLocShape;
1807 if(theMesh->HasShapeToMesh())
1808 myLocShape = GeomObjectToShape( theSubObject );
1810 myLocShape = SMESH_Mesh::PseudoShape();
1812 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1813 list< ::SMESH_Gen::TAlgoStateError > error_list;
1814 list< ::SMESH_Gen::TAlgoStateError >::iterator error;
1815 // call ::SMESH_Gen::GetAlgoState()
1816 myGen.GetAlgoState( myLocMesh, myLocShape, error_list );
1817 error_array->length( error_list.size() );
1819 for ( error = error_list.begin(); error != error_list.end(); ++error )
1821 // fill AlgoStateError structure
1822 SMESH::AlgoStateError & errStruct = error_array[ i++ ];
1823 errStruct.state = SMESH_Mesh_i::ConvertHypothesisStatus( error->_name );
1824 errStruct.algoDim = error->_algoDim;
1825 errStruct.isGlobalAlgo = error->_isGlobalAlgo;
1826 errStruct.algoName = "";
1827 SALOMEDS::SObject_wrap algoSO = GetAlgoSO( error->_algo );
1828 if ( !algoSO->_is_nil() ) {
1829 CORBA::String_var algoName = algoSO->GetName();
1830 errStruct.algoName = algoName.in();
1835 catch ( SALOME_Exception& S_ex ) {
1836 INFOS( "catch exception "<< S_ex.what() );
1838 return error_array._retn();
1841 //=============================================================================
1843 * SMESH_Gen_i::GetSubShapesId
1845 * Get sub-shapes unique ID's list
1847 //=============================================================================
1850 SMESH_Gen_i::GetSubShapesId( GEOM::GEOM_Object_ptr theMainShapeObject,
1851 const SMESH::object_array& theListOfSubShapeObject )
1852 throw ( SALOME::SALOME_Exception )
1854 Unexpect aCatch(SALOME_SalomeException);
1855 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetSubShapesId" );
1857 SMESH::long_array_var shapesId = new SMESH::long_array;
1860 if ( CORBA::is_nil( theMainShapeObject ) )
1861 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference", SALOME::BAD_PARAM );
1865 TopoDS_Shape myMainShape = GeomObjectToShape(theMainShapeObject);
1866 TopTools_IndexedMapOfShape myIndexToShape;
1867 TopExp::MapShapes(myMainShape,myIndexToShape);
1869 for ( CORBA::ULong i = 0; i < theListOfSubShapeObject.length(); i++ )
1871 GEOM::GEOM_Object_var aShapeObject
1872 = GEOM::GEOM_Object::_narrow(theListOfSubShapeObject[i]);
1873 if ( CORBA::is_nil( aShapeObject ) )
1874 THROW_SALOME_CORBA_EXCEPTION ("bad shape object reference", \
1875 SALOME::BAD_PARAM );
1877 TopoDS_Shape locShape = GeomObjectToShape(aShapeObject);
1878 for (TopExp_Explorer exp(locShape,TopAbs_FACE); exp.More(); exp.Next())
1880 const TopoDS_Face& F = TopoDS::Face(exp.Current());
1881 setId.insert(myIndexToShape.FindIndex(F));
1882 if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(F));
1884 for (TopExp_Explorer exp(locShape,TopAbs_EDGE); exp.More(); exp.Next())
1886 const TopoDS_Edge& E = TopoDS::Edge(exp.Current());
1887 setId.insert(myIndexToShape.FindIndex(E));
1888 if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(E));
1890 for (TopExp_Explorer exp(locShape,TopAbs_VERTEX); exp.More(); exp.Next())
1892 const TopoDS_Vertex& V = TopoDS::Vertex(exp.Current());
1893 setId.insert(myIndexToShape.FindIndex(V));
1894 if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(V));
1897 shapesId->length(setId.size());
1898 set<int>::iterator iind;
1900 for (iind = setId.begin(); iind != setId.end(); iind++)
1902 if(MYDEBUG) SCRUTE((*iind));
1903 shapesId[i] = (*iind);
1904 if(MYDEBUG) SCRUTE(shapesId[i]);
1908 catch (SALOME_Exception& S_ex)
1910 THROW_SALOME_CORBA_EXCEPTION(S_ex.what(), SALOME::BAD_PARAM);
1913 return shapesId._retn();
1916 //=============================================================================
1918 * SMESH_Gen_i::Compute
1920 * Compute mesh on a shape
1922 //=============================================================================
1924 CORBA::Boolean SMESH_Gen_i::Compute( SMESH::SMESH_Mesh_ptr theMesh,
1925 GEOM::GEOM_Object_ptr theShapeObject )
1926 throw ( SALOME::SALOME_Exception )
1929 Unexpect aCatch(SALOME_SalomeException);
1930 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Compute" );
1932 if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
1933 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1934 SALOME::BAD_PARAM );
1936 if ( CORBA::is_nil( theMesh ) )
1937 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
1938 SALOME::BAD_PARAM );
1940 // Update Python script
1941 TPythonDump() << "isDone = " << this << ".Compute( "
1942 << theMesh << ", " << theShapeObject << ")";
1946 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
1947 ASSERT( meshServant );
1948 if ( meshServant ) {
1949 meshServant->Load();
1950 // NPAL16168: "geometrical group edition from a submesh don't modify mesh computation"
1951 meshServant->CheckGeomModif();
1952 // get local TopoDS_Shape
1953 TopoDS_Shape myLocShape;
1954 if(theMesh->HasShapeToMesh())
1955 myLocShape = GeomObjectToShape( theShapeObject );
1957 myLocShape = SMESH_Mesh::PseudoShape();
1958 // call implementation compute
1959 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1960 myGen.PrepareCompute( myLocMesh, myLocShape );
1961 int how = ::SMESH_Gen::COMPACT_MESH;
1962 if ( myLocShape != myLocMesh.GetShapeToMesh() ) // compute a sub-mesh
1963 how |= ::SMESH_Gen::SHAPE_ONLY;
1964 bool ok = myGen.Compute( myLocMesh, myLocShape, how );
1965 meshServant->CreateGroupServants(); // algos can create groups (issue 0020918)
1966 myLocMesh.GetMeshDS()->Modified();
1967 UpdateIcons( theMesh );
1969 HighLightInvalid( theMesh, /*isInvalid=*/false );
1973 catch ( std::bad_alloc ) {
1974 INFOS( "Compute(): lack of memory" );
1976 catch ( SALOME_Exception& S_ex ) {
1977 INFOS( "Compute(): catch exception "<< S_ex.what() );
1980 INFOS( "Compute(): unknown exception " );
1985 //=============================================================================
1987 * SMESH_Gen_i::CancelCompute
1989 * Cancel Compute mesh on a shape
1991 //=============================================================================
1993 void SMESH_Gen_i::CancelCompute( SMESH::SMESH_Mesh_ptr theMesh,
1994 GEOM::GEOM_Object_ptr theShapeObject )
1996 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1997 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1998 TopoDS_Shape myLocShape;
1999 if(theMesh->HasShapeToMesh())
2000 myLocShape = GeomObjectToShape( theShapeObject );
2002 myLocShape = SMESH_Mesh::PseudoShape();
2003 myGen.CancelCompute( myLocMesh, myLocShape);
2006 //=============================================================================
2008 * SMESH_Gen_i::Precompute
2010 * Compute mesh as preview till indicated dimension on shape
2012 //=============================================================================
2014 SMESH::MeshPreviewStruct* SMESH_Gen_i::Precompute( SMESH::SMESH_Mesh_ptr theMesh,
2015 GEOM::GEOM_Object_ptr theShapeObject,
2016 SMESH::Dimension theDimension,
2017 SMESH::long_array& theShapesId)
2018 throw ( SALOME::SALOME_Exception )
2020 Unexpect aCatch(SALOME_SalomeException);
2021 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Precompute" );
2023 if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
2024 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
2025 SALOME::BAD_PARAM );
2027 if ( CORBA::is_nil( theMesh ) )
2028 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
2029 SALOME::BAD_PARAM );
2031 SMESH::MeshPreviewStruct_var result = new SMESH::MeshPreviewStruct;
2034 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
2035 meshServant->Load();
2036 ASSERT( meshServant );
2037 if ( meshServant ) {
2038 // NPAL16168: "geometrical group edition from a submesh don't modify mesh computation"
2039 meshServant->CheckGeomModif();
2040 // get local TopoDS_Shape
2041 TopoDS_Shape myLocShape;
2042 if(theMesh->HasShapeToMesh())
2043 myLocShape = GeomObjectToShape( theShapeObject );
2045 return result._retn();;
2047 // call implementation compute
2048 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
2050 ::MeshDimension aDim = (MeshDimension)theDimension;
2051 if ( myGen.Compute( myLocMesh, myLocShape, ::SMESH_Gen::COMPACT_MESH, aDim, &shapeIds ) )
2053 int nbShapeId = shapeIds.size();
2054 theShapesId.length( nbShapeId );
2055 // iterates on shapes and collect mesh entities into mesh preview
2056 TSetOfInt::const_iterator idIt = shapeIds.begin();
2057 TSetOfInt::const_iterator idEnd = shapeIds.end();
2058 std::map< int, int > mapOfShIdNb;
2059 std::set< SMESH_TLink > setOfEdge;
2060 std::list< SMDSAbs_ElementType > listOfElemType;
2061 typedef map<const SMDS_MeshElement*, int > TNode2LocalIDMap;
2062 typedef TNode2LocalIDMap::iterator TNodeLocalID;
2063 TNode2LocalIDMap mapNode2LocalID;
2064 list< TNodeLocalID > connectivity;
2065 int i, nbConnNodes = 0;
2066 std::set< const SMESH_subMesh* > setOfVSubMesh;
2067 // iterates on shapes
2068 for ( ; idIt != idEnd; idIt++ )
2070 if ( mapOfShIdNb.find( *idIt ) != mapOfShIdNb.end() )
2072 SMESH_subMesh* sm = myLocMesh.GetSubMeshContaining(*idIt);
2073 if ( !sm || !sm->IsMeshComputed() )
2076 const TopoDS_Shape& aSh = sm->GetSubShape();
2077 const int shDim = myGen.GetShapeDim( aSh );
2078 if ( shDim < 1 || shDim > theDimension )
2081 mapOfShIdNb[ *idIt ] = 0;
2082 theShapesId[ mapOfShIdNb.size() - 1 ] = *idIt;
2084 SMESHDS_SubMesh* smDS = sm->GetSubMeshDS();
2085 if ( !smDS ) continue;
2087 if ( theDimension == SMESH::DIM_2D )
2089 SMDS_ElemIteratorPtr faceIt = smDS->GetElements();
2090 while ( faceIt->more() )
2092 const SMDS_MeshElement* face = faceIt->next();
2093 int aNbNode = face->NbNodes();
2095 aNbNode /= 2; // do not take into account additional middle nodes
2097 SMDS_MeshNode* node1 = (SMDS_MeshNode*)face->GetNode( 0 );
2098 for ( int nIndx = 0; nIndx < aNbNode; nIndx++ )
2100 SMDS_MeshNode* node2 = (SMDS_MeshNode*)face->GetNode( nIndx+1 < aNbNode ? nIndx+1 : 0 );
2101 if ( setOfEdge.insert( SMESH_TLink ( node1, node2 ) ).second )
2103 listOfElemType.push_back( SMDSAbs_Edge );
2104 connectivity.push_back
2105 ( mapNode2LocalID.insert( make_pair( node1, ++nbConnNodes)).first );
2106 connectivity.push_back
2107 ( mapNode2LocalID.insert( make_pair( node2, ++nbConnNodes)).first );
2113 else if ( theDimension == SMESH::DIM_1D )
2115 SMDS_NodeIteratorPtr nodeIt = smDS->GetNodes();
2116 while ( nodeIt->more() )
2118 listOfElemType.push_back( SMDSAbs_Node );
2119 connectivity.push_back
2120 ( mapNode2LocalID.insert( make_pair( nodeIt->next(), ++nbConnNodes)).first );
2122 // add corner nodes by first vertex from edge
2123 SMESH_subMeshIteratorPtr edgeSmIt =
2124 sm->getDependsOnIterator(/*includeSelf*/false,
2125 /*complexShapeFirst*/false);
2126 while ( edgeSmIt->more() )
2128 SMESH_subMesh* vertexSM = edgeSmIt->next();
2129 // check that vertex is not already treated
2130 if ( !setOfVSubMesh.insert( vertexSM ).second )
2132 if ( vertexSM->GetSubShape().ShapeType() != TopAbs_VERTEX )
2135 const SMESHDS_SubMesh* vertexSmDS = vertexSM->GetSubMeshDS();
2136 SMDS_NodeIteratorPtr nodeIt = vertexSmDS->GetNodes();
2137 while ( nodeIt->more() )
2139 listOfElemType.push_back( SMDSAbs_Node );
2140 connectivity.push_back
2141 ( mapNode2LocalID.insert( make_pair( nodeIt->next(), ++nbConnNodes)).first );
2147 // fill node coords and assign local ids to the nodes
2148 int nbNodes = mapNode2LocalID.size();
2149 result->nodesXYZ.length( nbNodes );
2150 TNodeLocalID node2ID = mapNode2LocalID.begin();
2151 for ( i = 0; i < nbNodes; ++i, ++node2ID ) {
2152 node2ID->second = i;
2153 const SMDS_MeshNode* node = (const SMDS_MeshNode*) node2ID->first;
2154 result->nodesXYZ[i].x = node->X();
2155 result->nodesXYZ[i].y = node->Y();
2156 result->nodesXYZ[i].z = node->Z();
2158 // fill connectivity
2159 result->elementConnectivities.length( nbConnNodes );
2160 list< TNodeLocalID >::iterator connIt = connectivity.begin();
2161 for ( i = 0; i < nbConnNodes; ++i, ++connIt ) {
2162 result->elementConnectivities[i] = (*connIt)->second;
2165 // fill element types
2166 result->elementTypes.length( listOfElemType.size() );
2167 std::list< SMDSAbs_ElementType >::const_iterator typeIt = listOfElemType.begin();
2168 std::list< SMDSAbs_ElementType >::const_iterator typeEnd = listOfElemType.end();
2169 for ( i = 0; typeIt != typeEnd; ++i, ++typeIt )
2171 SMDSAbs_ElementType elemType = *typeIt;
2172 result->elementTypes[i].SMDS_ElementType = (SMESH::ElementType)elemType;
2173 result->elementTypes[i].isPoly = false;
2174 result->elementTypes[i].nbNodesInElement = elemType == SMDSAbs_Edge ? 2 : 1;
2177 // correct number of shapes
2178 theShapesId.length( mapOfShIdNb.size() );
2182 catch ( std::bad_alloc ) {
2183 INFOS( "Precompute(): lack of memory" );
2185 catch ( SALOME_Exception& S_ex ) {
2186 INFOS( "Precompute(): catch exception "<< S_ex.what() );
2189 INFOS( "Precompute(): unknown exception " );
2191 return result._retn();
2195 //=============================================================================
2197 * SMESH_Gen_i::Evaluate
2199 * Evaluate mesh on a shape
2201 //=============================================================================
2203 SMESH::long_array* SMESH_Gen_i::Evaluate(SMESH::SMESH_Mesh_ptr theMesh,
2204 GEOM::GEOM_Object_ptr theShapeObject)
2205 // SMESH::long_array& theNbElems)
2206 throw ( SALOME::SALOME_Exception )
2208 Unexpect aCatch(SALOME_SalomeException);
2209 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Evaluate" );
2211 if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
2212 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
2213 SALOME::BAD_PARAM );
2215 if ( CORBA::is_nil( theMesh ) )
2216 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
2217 SALOME::BAD_PARAM );
2219 SMESH::long_array_var nbels = new SMESH::long_array;
2220 nbels->length(SMESH::Entity_Last);
2221 int i = SMESH::Entity_Node;
2222 for (; i < SMESH::Entity_Last; i++)
2225 // Update Python script
2226 TPythonDump() << "theNbElems = " << this << ".Evaluate( "
2227 << theMesh << ", " << theShapeObject << ")";
2231 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
2232 ASSERT( meshServant );
2233 if ( meshServant ) {
2234 meshServant->Load();
2235 // NPAL16168: "geometrical group edition from a submesh don't modify mesh computation"
2236 meshServant->CheckGeomModif();
2237 // get local TopoDS_Shape
2238 TopoDS_Shape myLocShape;
2239 if(theMesh->HasShapeToMesh())
2240 myLocShape = GeomObjectToShape( theShapeObject );
2242 myLocShape = SMESH_Mesh::PseudoShape();
2243 // call implementation compute
2244 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
2245 MapShapeNbElems aResMap;
2246 /*CORBA::Boolean ret =*/ myGen.Evaluate( myLocMesh, myLocShape, aResMap);
2247 MapShapeNbElemsItr anIt = aResMap.begin();
2248 for(; anIt!=aResMap.end(); anIt++) {
2249 const vector<int>& aVec = (*anIt).second;
2250 for ( i = SMESH::Entity_Node; i < (int)aVec.size(); i++ ) {
2251 int nbElem = aVec[i];
2252 if ( nbElem < 0 ) // algo failed, check that it has reported a message
2254 SMESH_subMesh* sm = anIt->first;
2255 SMESH_ComputeErrorPtr& error = sm->GetComputeError();
2256 const SMESH_Algo* algo = sm->GetAlgo();
2257 if ( (algo && !error.get()) || error->IsOK() )
2258 error.reset( new SMESH_ComputeError( COMPERR_ALGO_FAILED,"Failed to evaluate",algo));
2262 nbels[i] += aVec[i];
2266 return nbels._retn();
2269 catch ( std::bad_alloc ) {
2270 INFOS( "Evaluate(): lack of memory" );
2272 catch ( SALOME_Exception& S_ex ) {
2273 INFOS( "Evaluate(): catch exception "<< S_ex.what() );
2276 INFOS( "Evaluate(): unknown exception " );
2279 return nbels._retn();
2282 //================================================================================
2284 * \brief Return geometrical object the given element is built on
2285 * \param theMesh - the mesh the element is in
2286 * \param theElementID - the element ID
2287 * \param theGeomName - the name of the result geom object if it is not yet published
2288 * \retval GEOM::GEOM_Object_ptr - the found or just published geom object (no need to UnRegister())
2290 //================================================================================
2292 GEOM::GEOM_Object_ptr
2293 SMESH_Gen_i::GetGeometryByMeshElement( SMESH::SMESH_Mesh_ptr theMesh,
2294 CORBA::Long theElementID,
2295 const char* theGeomName)
2296 throw ( SALOME::SALOME_Exception )
2298 Unexpect aCatch(SALOME_SalomeException);
2300 GEOM::GEOM_Object_wrap geom = FindGeometryByMeshElement(theMesh, theElementID);
2301 if ( !geom->_is_nil() ) {
2302 GEOM::GEOM_Object_var mainShape = theMesh->GetShapeToMesh();
2303 GEOM::GEOM_Gen_ptr geomGen = GetGeomEngine();
2305 // try to find the corresponding SObject
2306 SALOMEDS::SObject_wrap SObj = ObjectToSObject( geom.in() );
2307 if ( SObj->_is_nil() ) // submesh can be not found even if published
2309 // try to find published submesh
2310 GEOM::ListOfLong_var list = geom->GetSubShapeIndices();
2311 if ( !geom->IsMainShape() && list->length() == 1 ) {
2312 SALOMEDS::SObject_wrap mainSO = ObjectToSObject( mainShape );
2313 SALOMEDS::ChildIterator_wrap it;
2314 if ( !mainSO->_is_nil() ) {
2315 it = getStudyServant()->NewChildIterator( mainSO );
2317 if ( !it->_is_nil() ) {
2318 for ( it->InitEx(true); it->More(); it->Next() ) {
2319 SALOMEDS::SObject_wrap so = it->Value();
2320 CORBA::Object_var obj = SObjectToObject( so );
2321 GEOM::GEOM_Object_var subGeom = GEOM::GEOM_Object::_narrow( obj );
2322 if ( !subGeom->_is_nil() ) {
2323 GEOM::ListOfLong_var subList = subGeom->GetSubShapeIndices();
2324 if ( subList->length() == 1 && list[0] == subList[0] ) {
2334 if ( SObj->_is_nil() ) // publish a new subshape
2335 SObj = geomGen->AddInStudy( geom, theGeomName, mainShape );
2337 // return only published geometry
2338 if ( !SObj->_is_nil() ) {
2339 //return geom._retn(); -- servant of geom must be UnRegister()ed;
2340 CORBA::Object_var obj = SObjectToObject( SObj );
2341 GEOM::GEOM_Object_var go = GEOM::GEOM_Object::_narrow( obj );
2345 return GEOM::GEOM_Object::_nil();
2348 //================================================================================
2350 * \brief Return geometrical object the given element is built on.
2351 * \param theMesh - the mesh the element is in
2352 * \param theElementID - the element ID
2353 * \retval GEOM::GEOM_Object_ptr - the found or created (UnRegister()!) geom object
2355 //================================================================================
2357 GEOM::GEOM_Object_ptr
2358 SMESH_Gen_i::FindGeometryByMeshElement( SMESH::SMESH_Mesh_ptr theMesh,
2359 CORBA::Long theElementID)
2360 throw ( SALOME::SALOME_Exception )
2362 Unexpect aCatch(SALOME_SalomeException);
2363 if ( CORBA::is_nil( theMesh ) )
2364 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference", SALOME::BAD_PARAM );
2366 GEOM::GEOM_Object_var mainShape = theMesh->GetShapeToMesh();
2367 GEOM::GEOM_Gen_ptr geomGen = GetGeomEngine();
2369 // get a core mesh DS
2370 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
2371 if ( meshServant && !geomGen->_is_nil() && !mainShape->_is_nil() )
2373 ::SMESH_Mesh & mesh = meshServant->GetImpl();
2374 SMESHDS_Mesh* meshDS = mesh.GetMeshDS();
2375 // find the element in mesh
2376 if ( const SMDS_MeshElement * elem = meshDS->FindElement( theElementID ) ) {
2377 // find a shape id by the element
2378 if ( int shapeID = ::SMESH_MeshEditor( &mesh ).FindShape( elem )) {
2379 // get a geom object by the shape id
2380 GEOM::GEOM_Object_var geom = ShapeToGeomObject( meshDS->IndexToShape( shapeID ));
2381 if ( geom->_is_nil() ) {
2382 // try to find a published sub-shape
2383 SALOMEDS::SObject_wrap mainSO = ObjectToSObject( mainShape );
2384 SALOMEDS::ChildIterator_wrap it;
2385 if ( !mainSO->_is_nil() ) {
2386 it = getStudyServant()->NewChildIterator( mainSO );
2388 if ( !it->_is_nil() ) {
2389 for ( it->InitEx(true); it->More(); it->Next() ) {
2390 SALOMEDS::SObject_wrap so = it->Value();
2391 CORBA::Object_var obj = SObjectToObject( so );
2392 GEOM::GEOM_Object_var subGeom = GEOM::GEOM_Object::_narrow( obj );
2393 if ( !subGeom->_is_nil() ) {
2394 GEOM::ListOfLong_var subList = subGeom->GetSubShapeIndices();
2395 if ( subList->length() == 1 && shapeID == subList[0] ) {
2403 if ( geom->_is_nil() ) {
2405 GEOM::GEOM_IShapesOperations_wrap op = geomGen->GetIShapesOperations();
2406 if ( !op->_is_nil() )
2407 geom = op->GetSubShape( mainShape, shapeID );
2412 if ( !geom->_is_nil() ) {
2413 GeomObjectToShape( geom ); // let geom client remember the found shape
2414 return geom._retn();
2419 return GEOM::GEOM_Object::_nil();
2422 //================================================================================
2424 * SMESH_Gen_i::Concatenate
2426 * Concatenate the given meshes into one mesh
2428 //================================================================================
2430 SMESH::SMESH_Mesh_ptr
2431 SMESH_Gen_i::Concatenate(const SMESH::ListOfIDSources& theMeshesArray,
2432 CORBA::Boolean theUniteIdenticalGroups,
2433 CORBA::Boolean theMergeNodesAndElements,
2434 CORBA::Double theMergeTolerance)
2435 throw ( SALOME::SALOME_Exception )
2437 return ConcatenateCommon(theMeshesArray,
2438 theUniteIdenticalGroups,
2439 theMergeNodesAndElements,
2444 //================================================================================
2446 * SMESH_Gen_i::ConcatenateWithGroups
2448 * Concatenate the given meshes into one mesh
2449 * Create the groups of all elements from initial meshes
2451 //================================================================================
2453 SMESH::SMESH_Mesh_ptr
2454 SMESH_Gen_i::ConcatenateWithGroups(const SMESH::ListOfIDSources& theMeshesArray,
2455 CORBA::Boolean theUniteIdenticalGroups,
2456 CORBA::Boolean theMergeNodesAndElements,
2457 CORBA::Double theMergeTolerance)
2458 throw ( SALOME::SALOME_Exception )
2460 return ConcatenateCommon(theMeshesArray,
2461 theUniteIdenticalGroups,
2462 theMergeNodesAndElements,
2467 //================================================================================
2469 * SMESH_Gen_i::ConcatenateCommon
2471 * Concatenate the given meshes into one mesh
2473 //================================================================================
2475 SMESH::SMESH_Mesh_ptr
2476 SMESH_Gen_i::ConcatenateCommon(const SMESH::ListOfIDSources& theMeshesArray,
2477 CORBA::Boolean theUniteIdenticalGroups,
2478 CORBA::Boolean theMergeNodesAndElements,
2479 CORBA::Double theMergeTolerance,
2480 CORBA::Boolean theCommonGroups)
2481 throw ( SALOME::SALOME_Exception )
2483 std::unique_ptr< TPythonDump > pPythonDump( new TPythonDump );
2484 TPythonDump& pythonDump = *pPythonDump; // prevent dump of called methods
2487 SMESH::SMESH_Mesh_var newMesh = CreateEmptyMesh();
2488 SMESH_Mesh_i* newImpl = SMESH::DownCast<SMESH_Mesh_i*>( newMesh );
2489 if ( !newImpl ) return newMesh._retn();
2491 ::SMESH_Mesh& locMesh = newImpl->GetImpl();
2492 SMESHDS_Mesh* newMeshDS = locMesh.GetMeshDS();
2494 typedef std::list<SMESH::SMESH_Group_var> TListOfNewGroups;
2495 typedef std::pair<string, SMESH::ElementType > TNameAndType;
2496 typedef std::map< TNameAndType, TListOfNewGroups > TGroupsMap;
2497 TGroupsMap groupsMap;
2498 TListOfNewGroups listOfNewGroups;
2500 ::SMESH_MeshEditor newEditor( &locMesh );
2501 ::SMESH_MeshEditor::ElemFeatures elemType;
2503 // loop on sub-meshes
2504 for ( CORBA::ULong i = 0; i < theMeshesArray.length(); i++ )
2506 if ( CORBA::is_nil( theMeshesArray[i] )) continue;
2507 SMESH::SMESH_Mesh_var initMesh = theMeshesArray[i]->GetMesh();
2508 SMESH_Mesh_i* initImpl = SMESH::DownCast<SMESH_Mesh_i*>( initMesh );
2509 if ( !initImpl ) continue;
2512 // assure that IDs increments by one during iteration
2513 ::SMESH_Mesh& initLocMesh = initImpl->GetImpl();
2514 SMESHDS_Mesh* initMeshDS = initLocMesh.GetMeshDS();
2515 if ( initMeshDS->MaxNodeID() > initMeshDS->NbNodes() ||
2516 initMeshDS->MaxElementID() > initMeshDS->NbElements() )
2518 initMeshDS->Modified();
2519 initMeshDS->CompactMesh();
2522 // remember nb of elements before filling in
2523 SMESH::long_array_var prevState = newMesh->GetNbElementsByType();
2527 std::vector< const SMDS_MeshElement* > newNodes( initMeshDS->NbNodes() + 1, 0 );
2528 SMDS_ElemIteratorPtr elemIt = initImpl->GetElements( theMeshesArray[i], SMESH::NODE );
2529 while ( elemIt->more() )
2531 SMESH_NodeXYZ node = elemIt->next();
2532 newNodes[ node->GetID() ] = newMeshDS->AddNode( node.X(), node.Y(), node.Z() );
2537 std::vector< const SMDS_MeshElement* > newElems( initMeshDS->NbElements() + 1, 0 );
2538 elemIt = initImpl->GetElements( theMeshesArray[i], SMESH::ALL );
2539 while ( elemIt->more() )
2541 const SMDS_MeshElement* elem = elemIt->next();
2542 elemType.myNodes.resize( elem->NbNodes() );
2544 SMDS_NodeIteratorPtr itNodes = elem->nodeIterator();
2545 for ( int k = 0; itNodes->more(); k++)
2547 const SMDS_MeshNode* node = itNodes->next();
2548 elemType.myNodes[ k ] = static_cast< const SMDS_MeshNode*> ( newNodes[ node->GetID() ]);
2551 // creates a corresponding element on existent nodes in new mesh
2552 newElems[ elem->GetID() ] =
2553 newEditor.AddElement( elemType.myNodes, elemType.Init( elem, /*basicOnly=*/false ));
2555 newEditor.ClearLastCreated(); // forget the history
2558 // create groups of just added elements
2559 SMESH::SMESH_Group_var newGroup;
2560 SMESH::ElementType groupType;
2561 if ( theCommonGroups )
2564 const char* typeNames[] = { "All","Nodes","Edges","Faces","Volumes","0DElems","Balls" };
2565 { // check of typeNames: compilation failure mains that NB_ELEMENT_TYPES changed:
2566 const int nbNames = sizeof(typeNames) / sizeof(const char*);
2567 int _assert[( nbNames == SMESH::NB_ELEMENT_TYPES ) ? 2 : -1 ]; _assert[0]=_assert[1]=0;
2570 SMESH::long_array_var curState = newMesh->GetNbElementsByType();
2572 for( groupType = SMESH::NODE;
2573 groupType < SMESH::NB_ELEMENT_TYPES;
2574 groupType = (SMESH::ElementType)( groupType + 1 ))
2576 if ( curState[ groupType ] <= prevState[ groupType ])
2577 continue; // no elements of groupType added from the i-th mesh
2579 // make a group name
2580 std::string groupName = "Gr";
2581 SALOMEDS::SObject_wrap meshSO = ObjectToSObject( theMeshesArray[i] );
2583 CORBA::String_var name = meshSO->GetName();
2587 groupName += typeNames[ groupType ];
2589 // make and fill a group
2590 newGroup = newImpl->CreateGroup( groupType, groupName.c_str() );
2591 std::vector< const SMDS_MeshElement* > & elemVec =
2592 ( groupType == SMESH::NODE ) ? newNodes : newElems;
2593 if ( SMESH_Group_i* grp_i = SMESH::DownCast<SMESH_Group_i*>( newGroup ))
2595 if ( SMESHDS_Group* grpDS = dynamic_cast<SMESHDS_Group*>( grp_i->GetGroupDS() ))
2597 for ( size_t j = 0; j < elemVec.size(); ++j )
2599 if ( elemVec[j] && elemVec[j]->GetType() == grpDS->GetType() )
2600 grpDS->Add( elemVec[j] );
2604 listOfNewGroups.clear();
2605 listOfNewGroups.push_back( newGroup );
2606 groupsMap.insert( std::make_pair( TNameAndType( groupName, groupType ),
2611 if ( SMESH_Mesh_i* initImpl = SMESH::DownCast<SMESH_Mesh_i*>( theMeshesArray[i] ))
2615 SMESH::SMESH_GroupBase_ptr group;
2616 CORBA::String_var groupName;
2617 SMESH::long_array_var newIDs = new SMESH::long_array();
2619 // loop on groups of a source mesh
2620 SMESH::ListOfGroups_var listOfGroups = initImpl->GetGroups();
2621 for ( CORBA::ULong iG = 0; iG < listOfGroups->length(); iG++ )
2623 group = listOfGroups[iG];
2624 groupType = group->GetType();
2625 groupName = group->GetName();
2626 std::string name = groupName.in();
2628 // convert a list of IDs
2629 newIDs->length( group->Size() );
2630 std::vector< const SMDS_MeshElement* > & elemVec =
2631 ( groupType == SMESH::NODE ) ? newNodes : newElems;
2632 SMDS_ElemIteratorPtr itGrElems = initImpl->GetElements( group, SMESH::ALL );
2634 while ( itGrElems->more() )
2636 const SMDS_MeshElement* elem = itGrElems->next();
2637 const SMDS_MeshElement* newElem = elemVec[ elem->GetID() ];
2639 newIDs[ nbElems++ ] = newElem->GetID();
2641 newIDs->length( nbElems );
2643 // check that a current group name and type don't have identical ones in final mesh
2644 listOfNewGroups.clear();
2645 TNameAndType nameAndType( name, groupType );
2646 TGroupsMap::iterator anIter = groupsMap.find( nameAndType );
2647 if ( anIter == groupsMap.end() )
2649 // add a new group in the mesh
2650 newGroup = newImpl->CreateGroup( groupType, groupName.in() );
2651 newGroup->Add( newIDs );
2653 listOfNewGroups.push_back( newGroup );
2654 groupsMap.insert( std::make_pair( nameAndType, listOfNewGroups ));
2656 else if ( theUniteIdenticalGroups )
2658 // unite identical groups
2659 TListOfNewGroups& aNewGroups = anIter->second;
2660 aNewGroups.front()->Add( newIDs );
2664 // rename identical groups
2665 newGroup = newImpl->CreateGroup( groupType, groupName );
2666 newGroup->Add( newIDs );
2668 TListOfNewGroups& newGroups = anIter->second;
2669 std::string newGroupName;
2670 if ( newGroups.size() == 1 )
2672 newGroupName = name + "_1";
2673 newGroups.front()->SetName( newGroupName.c_str() );
2675 newGroupName = name + "_" + SMESH_Comment( newGroups.size() + 1 );
2676 newGroup->SetName( newGroupName.c_str() );
2677 newGroups.push_back( newGroup );
2680 } // if an IDSource is a mesh
2683 if ( theMergeNodesAndElements ) // merge nodes
2685 TIDSortedNodeSet meshNodes; // no input nodes == treat all
2686 SMESH_MeshEditor::TListOfListOfNodes groupsOfNodes;
2687 newEditor.FindCoincidentNodes( meshNodes, theMergeTolerance, groupsOfNodes,
2688 /*SeparateCornersAndMedium=*/ false );
2689 newEditor.MergeNodes( groupsOfNodes );
2691 newEditor.MergeEqualElements();
2694 // Update Python script
2695 pythonDump << newMesh << " = " << this
2696 << "." << ( theCommonGroups ? "ConcatenateWithGroups" : "Concatenate" ) << "("
2697 << theMeshesArray << ", "
2698 << theUniteIdenticalGroups << ", "
2699 << theMergeNodesAndElements << ", "
2700 << TVar( theMergeTolerance ) << ")";
2702 pPythonDump.reset(); // enable python dump from GetGroups()
2704 // 0020577: EDF 1164 SMESH: Bad dump of concatenate with create common groups
2705 if ( !newMesh->_is_nil() )
2707 SMESH::ListOfGroups_var groups = newMesh->GetGroups();
2710 // IPAL21468 Change icon of compound because it need not be computed.
2711 SALOMEDS::SObject_wrap meshSO = ObjectToSObject( newMesh );
2712 SetPixMap( meshSO, "ICON_SMESH_TREE_MESH" );
2714 newMeshDS->Modified();
2716 return newMesh._retn();
2719 //================================================================================
2721 * \brief Create a mesh by copying a part of another mesh
2722 * \param meshPart - a part of mesh to copy
2723 * \param toCopyGroups - to create in the new mesh groups
2724 * the copied elements belongs to
2725 * \param toKeepIDs - to preserve IDs of the copied elements or not
2726 * \retval SMESH::SMESH_Mesh_ptr - the new mesh
2728 //================================================================================
2730 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CopyMesh(SMESH::SMESH_IDSource_ptr meshPart,
2731 const char* meshName,
2732 CORBA::Boolean toCopyGroups,
2733 CORBA::Boolean toKeepIDs)
2734 throw ( SALOME::SALOME_Exception )
2736 Unexpect aCatch(SALOME_SalomeException);
2738 TPythonDump* pyDump = new TPythonDump; // prevent dump from CreateMesh()
2739 std::unique_ptr<TPythonDump> pyDumpDeleter( pyDump );
2741 // 1. Get source mesh
2743 if ( CORBA::is_nil( meshPart ))
2744 THROW_SALOME_CORBA_EXCEPTION( "bad IDSource", SALOME::BAD_PARAM );
2746 SMESH::SMESH_Mesh_var srcMesh = meshPart->GetMesh();
2747 SMESH_Mesh_i* srcMesh_i = SMESH::DownCast<SMESH_Mesh_i*>( srcMesh );
2749 THROW_SALOME_CORBA_EXCEPTION( "bad mesh of IDSource", SALOME::BAD_PARAM );
2751 SMESHDS_Mesh* srcMeshDS = srcMesh_i->GetImpl().GetMeshDS();
2753 // 2. Make a new mesh
2755 SMESH::SMESH_Mesh_var newMesh = CreateMesh(GEOM::GEOM_Object::_nil());
2756 SMESH_Mesh_i* newMesh_i = SMESH::DownCast<SMESH_Mesh_i*>( newMesh );
2758 THROW_SALOME_CORBA_EXCEPTION( "can't create a mesh", SALOME::INTERNAL_ERROR );
2759 SALOMEDS::SObject_wrap meshSO = ObjectToSObject( newMesh );
2760 if ( !meshSO->_is_nil() )
2762 SetName( meshSO, meshName, "Mesh" );
2763 SetPixMap( meshSO, "ICON_SMESH_TREE_MESH_IMPORTED");
2765 SMESHDS_Mesh* newMeshDS = newMesh_i->GetImpl().GetMeshDS();
2766 ::SMESH_MeshEditor editor( &newMesh_i->GetImpl() );
2767 ::SMESH_MeshEditor::ElemFeatures elemType;
2769 // 3. Get elements to copy
2771 SMDS_ElemIteratorPtr srcElemIt; SMDS_NodeIteratorPtr srcNodeIt;
2772 TIDSortedElemSet srcElems;
2773 SMESH::array_of_ElementType_var srcElemTypes = meshPart->GetTypes();
2774 if ( SMESH::DownCast<SMESH_Mesh_i*>( meshPart ))
2777 srcElemIt = srcMeshDS->elementsIterator();
2778 srcNodeIt = srcMeshDS->nodesIterator();
2782 SMESH::long_array_var ids = meshPart->GetIDs();
2783 if ( srcElemTypes->length() == 1 && srcElemTypes[0] == SMESH::NODE ) // group of nodes
2785 for ( CORBA::ULong i=0; i < ids->length(); i++ )
2786 if ( const SMDS_MeshElement * elem = srcMeshDS->FindNode( ids[i] ))
2787 srcElems.insert( elem );
2791 for ( CORBA::ULong i = 0; i < ids->length(); i++ )
2792 if ( const SMDS_MeshElement * elem = srcMeshDS->FindElement( ids[i] ))
2793 srcElems.insert( elem );
2795 if ( srcElems.empty() )
2796 return newMesh._retn();
2798 typedef SMDS_SetIterator< SMDS_pElement, TIDSortedElemSet::const_iterator > ElIter;
2799 srcElemIt = SMDS_ElemIteratorPtr( new ElIter( srcElems.begin(), srcElems.end() ));
2804 typedef map<SMDS_pElement, SMDS_pElement, TIDCompare> TE2EMap;
2805 TE2EMap e2eMapByType[ SMDSAbs_NbElementTypes ];
2806 TE2EMap& n2nMap = e2eMapByType[ SMDSAbs_Node ];
2808 const SMDS_MeshNode *nSrc, *nTgt;
2809 vector< const SMDS_MeshNode* > nodes;
2810 while ( srcElemIt->more() )
2812 const SMDS_MeshElement * elem = srcElemIt->next();
2814 nodes.resize( elem->NbNodes());
2815 SMDS_ElemIteratorPtr nIt = elem->nodesIterator();
2817 for ( iN = 0; nIt->more(); ++iN )
2819 nSrc = static_cast<const SMDS_MeshNode*>( nIt->next() );
2820 nTgt = newMeshDS->FindNode( nSrc->GetID());
2822 nTgt = newMeshDS->AddNodeWithID( nSrc->X(), nSrc->Y(), nSrc->Z(), nSrc->GetID());
2827 for ( iN = 0; nIt->more(); ++iN )
2829 nSrc = static_cast<const SMDS_MeshNode*>( nIt->next() );
2830 TE2EMap::iterator n2n = n2nMap.insert( make_pair( nSrc, SMDS_pNode(0) )).first;
2832 n2n->second = newMeshDS->AddNode( nSrc->X(), nSrc->Y(), nSrc->Z() );
2833 nodes[ iN ] = (const SMDS_MeshNode*) n2n->second;
2837 if ( elem->GetType() != SMDSAbs_Node )
2839 elemType.Init( elem, /*basicOnly=*/false );
2840 if ( toKeepIDs ) elemType.SetID( elem->GetID() );
2842 const SMDS_MeshElement * newElem = editor.AddElement( nodes, elemType );
2843 if ( toCopyGroups && !toKeepIDs )
2844 e2eMapByType[ elem->GetType() ].insert( make_pair( elem, newElem ));
2846 } // while ( srcElemIt->more() )
2848 // 4(b). Copy free nodes
2850 if ( srcNodeIt && srcMeshDS->NbNodes() != newMeshDS->NbNodes() )
2852 while ( srcNodeIt->more() )
2854 nSrc = srcNodeIt->next();
2855 if ( nSrc->NbInverseElements() == 0 )
2858 nTgt = newMeshDS->AddNodeWithID( nSrc->X(), nSrc->Y(), nSrc->Z(), nSrc->GetID());
2860 n2nMap[ nSrc ] = newMeshDS->AddNode( nSrc->X(), nSrc->Y(), nSrc->Z() );
2867 int nbNewGroups = 0;
2870 SMESH_Mesh::GroupIteratorPtr gIt = srcMesh_i->GetImpl().GetGroups();
2871 while ( gIt->more() )
2873 SMESH_Group* group = gIt->next();
2874 const SMESHDS_GroupBase* groupDS = group->GetGroupDS();
2876 // Check group type. We copy nodal groups containing nodes of copied element
2877 SMDSAbs_ElementType groupType = groupDS->GetType();
2878 if ( groupType != SMDSAbs_Node &&
2879 newMeshDS->GetMeshInfo().NbElements( groupType ) == 0 )
2880 continue; // group type differs from types of meshPart
2882 // Find copied elements in the group
2883 vector< const SMDS_MeshElement* > groupElems;
2884 SMDS_ElemIteratorPtr eIt = groupDS->GetElements();
2887 const SMDS_MeshElement* foundElem;
2888 if ( groupType == SMDSAbs_Node )
2890 while ( eIt->more() )
2891 if (( foundElem = newMeshDS->FindNode( eIt->next()->GetID() )))
2892 groupElems.push_back( foundElem );
2896 while ( eIt->more() )
2897 if (( foundElem = newMeshDS->FindElement( eIt->next()->GetID() )))
2898 groupElems.push_back( foundElem );
2903 TE2EMap & e2eMap = e2eMapByType[ groupDS->GetType() ];
2904 if ( e2eMap.empty() ) continue;
2905 int minID = e2eMap.begin()->first->GetID();
2906 int maxID = e2eMap.rbegin()->first->GetID();
2907 TE2EMap::iterator e2e;
2908 while ( eIt->more() && groupElems.size() < e2eMap.size())
2910 const SMDS_MeshElement* e = eIt->next();
2911 if ( e->GetID() < minID || e->GetID() > maxID ) continue;
2912 if ((e2e = e2eMap.find( e )) != e2eMap.end())
2913 groupElems.push_back( e2e->second );
2917 if ( !groupElems.empty() )
2919 SMESH::SMESH_Group_var newGroupObj =
2920 newMesh->CreateGroup( SMESH::ElementType(groupType), group->GetName() );
2921 if ( SMESH_GroupBase_i* newGroup_i = SMESH::DownCast<SMESH_GroupBase_i*>( newGroupObj))
2923 SMESHDS_GroupBase * newGroupDS = newGroup_i->GetGroupDS();
2924 SMDS_MeshGroup& smdsGroup = ((SMESHDS_Group*)newGroupDS)->SMDSGroup();
2925 for ( unsigned i = 0; i < groupElems.size(); ++i )
2926 smdsGroup.Add( groupElems[i] );
2934 newMeshDS->Modified();
2936 *pyDump << newMesh << " = " << this
2937 << ".CopyMesh( " << meshPart << ", "
2938 << "'" << meshName << "', "
2939 << toCopyGroups << ", "
2940 << toKeepIDs << ")";
2942 pyDumpDeleter.reset(); // allow dump in GetGroups()
2944 if ( nbNewGroups > 0 ) // dump created groups
2945 SMESH::ListOfGroups_var groups = newMesh->GetGroups();
2947 return newMesh._retn();
2951 namespace // utils for CopyMeshWithGeom()
2953 typedef std::map< std::string, std::string > TStr2StrMap;
2954 typedef std::map< std::string, std::set< std::string > > TStr2StrSetMap;
2956 //================================================================================
2958 * \brief Return a new sub-shape corresponding to an old one
2960 //================================================================================
2964 SMESH_Mesh_i* mySrcMesh_i;
2965 SMESH_Mesh_i* myNewMesh_i;
2966 SMESH_Gen_i* myGen_i;
2970 TStr2StrMap myOld2NewEntryMap; // map of study entries
2973 GEOM::ListOfListOfLong_var myGIPMap; // filled by GetInPlaceMap()
2975 // not directly relating to shape search
2976 TStr2StrSetMap myInvalidMap; // blame shape -> invalid objects
2978 //================================================================================
2980 * \brief Constructor
2982 ShapeMapper( SMESH_Mesh_i* srcMesh_i,
2983 SMESH_Mesh_i* newMesh_i,
2984 SMESH_Gen_i* smeshGen_i )
2985 : mySrcMesh_i( srcMesh_i ),
2986 myNewMesh_i( newMesh_i ),
2987 myGen_i ( smeshGen_i ),
2988 myToPublish( smeshGen_i->IsEnablePublish() ),
2989 myGIPMapDone( false )
2991 // retrieve from the study shape mapping made thanks to
2992 // "Set presentation parameters and sub-shapes from arguments" option
2994 GEOM::GEOM_Object_var mainShapeNew = myNewMesh_i->GetShapeToMesh();
2995 GEOM::GEOM_Object_var mainShapeOld = mySrcMesh_i->GetShapeToMesh();
2996 SALOMEDS::SObject_wrap oldSO = myGen_i->ObjectToSObject( mainShapeOld );
2997 SALOMEDS::SObject_wrap newSO = myGen_i->ObjectToSObject( mainShapeNew );
2998 if ( newSO->_is_nil() )
3000 myToPublish = false;
3003 if (( myIsSameGeom = mainShapeNew->_is_equivalent( mainShapeOld )))
3005 CORBA::String_var oldEntry = oldSO->GetID();
3006 CORBA::String_var newEntry = newSO->GetID();
3007 myOld2NewEntryMap.insert( std::make_pair( std::string( oldEntry.in() ),
3008 std::string( newEntry.in() )));
3010 SALOMEDS::Study_var study = myGen_i->getStudyServant();
3011 GEOM::GEOM_Gen_var geomGen = myGen_i->GetGeomEngine();
3012 GEOM::GEOM_IShapesOperations_wrap op = geomGen->GetIShapesOperations();
3013 GEOM::ListOfGO_var subShapes = op->GetExistingSubObjects( mainShapeNew,
3014 /*groupsOnly=*/false );
3015 for ( CORBA::ULong i = 0; i < subShapes->length(); ++i )
3017 newSO = myGen_i->ObjectToSObject( subShapes[ i ]);
3018 SALOMEDS::ChildIterator_wrap anIter = study->NewChildIterator( newSO );
3019 for ( ; anIter->More(); anIter->Next() )
3021 SALOMEDS::SObject_wrap so = anIter->Value();
3022 if ( so->ReferencedObject( oldSO.inout() ))
3024 oldEntry = oldSO->GetID();
3025 newEntry = newSO->GetID();
3026 myOld2NewEntryMap.insert( std::make_pair( std::string( oldEntry.in() ),
3027 std::string( newEntry.in() )));
3033 //================================================================================
3035 * \brief Find a new sub-shape corresponding to an old one
3037 GEOM::GEOM_Object_ptr FindNew( GEOM::GEOM_Object_ptr oldShape )
3040 return GEOM::GEOM_Object::_duplicate( oldShape );
3042 GEOM::GEOM_Object_var newShape;
3044 if ( CORBA::is_nil( oldShape ))
3045 return newShape._retn();
3047 if ( !isChildOfOld( oldShape ))
3048 return GEOM::GEOM_Object::_duplicate( oldShape ); // shape independent of the old shape
3050 GEOM::GEOM_Object_var mainShapeNew = myNewMesh_i->GetShapeToMesh();
3051 GEOM::GEOM_Gen_var geomGen = myGen_i->GetGeomEngine();
3053 // try to find by entry
3056 CORBA::String_var oldEntry = oldShape->GetStudyEntry();
3057 TStr2StrMap::iterator o2nID = myOld2NewEntryMap.find( oldEntry.in() );
3058 if ( o2nID != myOld2NewEntryMap.end() )
3060 newShape = getShapeByEntry( o2nID->second );
3064 if ( newShape->_is_nil() )
3066 // try to construct a new sub-shape using myGIPMap
3068 std::vector< int > newIndices;
3069 GEOM::ListOfLong_var oldIndices = oldShape->GetSubShapeIndices();
3070 for ( CORBA::ULong i = 0; i < oldIndices->length(); ++i )
3072 findNewIDs( oldIndices[i], newIndices );
3074 if ( !newIndices.empty() )
3078 if ( newIndices.size() > 1 || oldShape->GetType() == GEOM_GROUP )
3080 int groupType = getShapeType( myNewMesh_i, newIndices[0] );
3082 GEOM::GEOM_IGroupOperations_wrap grOp = geomGen->GetIGroupOperations();
3083 newShape = grOp->CreateGroup( mainShapeNew, groupType );
3085 GEOM::ListOfLong_var newIndicesList = new GEOM::ListOfLong();
3086 newIndicesList->length( newIndices.size() );
3087 for ( size_t i = 0; i < newIndices.size(); ++i )
3088 newIndicesList[ i ] = newIndices[ i ];
3089 grOp->UnionIDs( newShape, newIndicesList );
3093 GEOM::GEOM_IShapesOperations_wrap shOp = geomGen->GetIShapesOperations();
3094 newShape = shOp->GetSubShape( mainShapeNew, newIndices[0] );
3103 if ( !newShape->_is_nil() && myToPublish )
3105 CORBA::String_var oldEntry, newEntry = newShape->GetStudyEntry();
3106 if ( !newEntry.in() || !newEntry.in()[0] )
3108 CORBA::String_var name = oldShape->GetName();
3109 SALOMEDS::SObject_wrap so = geomGen->AddInStudy( newShape, name, mainShapeNew );
3110 newEntry = newShape->GetStudyEntry();
3111 oldEntry = oldShape->GetStudyEntry();
3112 myOld2NewEntryMap.insert( std::make_pair( std::string( oldEntry.in() ),
3113 std::string( newEntry.in() )));
3117 return newShape._retn();
3120 //================================================================================
3122 * \brief Return a study entry of a new shape by study entry of the old one
3124 std::string FindNew( const std::string & oldEntry )
3129 TStr2StrMap::iterator o2nID = myOld2NewEntryMap.find( oldEntry );
3130 if ( o2nID != myOld2NewEntryMap.end() )