1 // Copyright (C) 2007-2014 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License, or (at your option) any later version.
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 // Lesser General Public License for more details.
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
22 // File : SMESH_Gen_i.cxx
23 // Author : Paul RASCLE, EDF
27 #include <TopExp_Explorer.hxx>
29 #include <TopoDS_Iterator.hxx>
30 #include <TopoDS_Compound.hxx>
31 #include <TopoDS_CompSolid.hxx>
32 #include <TopoDS_Solid.hxx>
33 #include <TopoDS_Shell.hxx>
34 #include <TopoDS_Face.hxx>
35 #include <TopoDS_Wire.hxx>
36 #include <TopoDS_Edge.hxx>
37 #include <TopoDS_Vertex.hxx>
38 #include <TopoDS_Shape.hxx>
39 #include <TopTools_MapOfShape.hxx>
40 #include <TopTools_IndexedMapOfShape.hxx>
41 #include <TopTools_ListOfShape.hxx>
42 #include <TopTools_ListIteratorOfListOfShape.hxx>
44 #include <BRep_Tool.hxx>
45 #include <TCollection_AsciiString.hxx>
56 #define LibHandle HMODULE
57 #define LoadLib( name ) LoadLibrary( name )
58 #define GetProc GetProcAddress
59 #define UnLoadLib( handle ) FreeLibrary( handle );
61 #define LibHandle void*
62 #define LoadLib( name ) dlopen( name, RTLD_LAZY )
64 #define UnLoadLib( handle ) dlclose( handle );
67 #include "SMESH_Gen_i.hxx"
68 #include "SMESH_version.h"
70 #include "DriverMED_W_SMESHDS_Mesh.h"
71 #include "DriverMED_R_SMESHDS_Mesh.h"
73 #include "DriverCGNS_Read.hxx"
75 #include "MED_Factory.hxx"
76 #include "SMDS_EdgePosition.hxx"
77 #include "SMDS_FacePosition.hxx"
78 #include "SMDS_PolyhedralVolumeOfNodes.hxx"
79 #include "SMDS_SetIterator.hxx"
80 #include "SMDS_SpacePosition.hxx"
81 #include "SMDS_VertexPosition.hxx"
82 #include "SMESHDS_Document.hxx"
83 #include "SMESHDS_Group.hxx"
84 #include "SMESHDS_GroupOnGeom.hxx"
85 #include "SMESH_Algo_i.hxx"
86 #include "SMESH_File.hxx"
87 #include "SMESH_Group.hxx"
88 #include "SMESH_Group_i.hxx"
89 #include "SMESH_Hypothesis.hxx"
90 #include "SMESH_Hypothesis_i.hxx"
91 #include "SMESH_Mesh.hxx"
92 #include "SMESH_MeshEditor.hxx"
93 #include "SMESH_Mesh_i.hxx"
94 #include "SMESH_PreMeshInfo.hxx"
95 #include "SMESH_PythonDump.hxx"
96 #include "SMESH_TryCatch.hxx" // to include after OCC headers!
98 #include CORBA_SERVER_HEADER(SMESH_Group)
99 #include CORBA_SERVER_HEADER(SMESH_Filter)
100 #include CORBA_SERVER_HEADER(SMESH_MeshEditor)
103 #include <GEOM_Client.hxx>
105 #include <Basics_Utils.hxx>
106 #include <Basics_DirUtils.hxx>
108 #include <OpUtil.hxx>
109 #include <SALOMEDS_Tool.hxx>
110 #include <SALOME_Container_i.hxx>
111 #include <SALOME_DataContainer_i.hxx>
112 #include <SALOME_LifeCycleCORBA.hxx>
113 #include <SALOME_NamingService.hxx>
114 #include <Utils_CorbaException.hxx>
115 #include <Utils_ExceptHandlers.hxx>
116 #include <Utils_SINGLETON.hxx>
117 #include <utilities.h>
119 #include CORBA_CLIENT_HEADER(SALOME_ModuleCatalog)
120 #include CORBA_CLIENT_HEADER(SALOME_Session)
122 // helpers about SALOME::GenericObj
123 #include <SALOMEDS_wrap.hxx>
124 #include <SALOMEDS_Attributes_wrap.hxx>
125 #include <GEOM_wrap.hxx>
133 using SMESH::TPythonDump;
136 #define NUM_TMP_FILES 2
139 static int MYDEBUG = 0;
141 static int MYDEBUG = 0;
144 // Static variables definition
145 GEOM::GEOM_Gen_var SMESH_Gen_i::myGeomGen = GEOM::GEOM_Gen::_nil();
146 CORBA::ORB_var SMESH_Gen_i::myOrb;
147 PortableServer::POA_var SMESH_Gen_i::myPoa;
148 SALOME_NamingService* SMESH_Gen_i::myNS = NULL;
149 SALOME_LifeCycleCORBA* SMESH_Gen_i::myLCC = NULL;
150 SMESH_Gen_i* SMESH_Gen_i::mySMESHGen = NULL;
153 const int nbElemPerDiagonal = 10;
155 //=============================================================================
157 * GetServant [ static ]
159 * Get servant of the CORBA object
161 //=============================================================================
163 PortableServer::ServantBase_var SMESH_Gen_i::GetServant( CORBA::Object_ptr theObject )
165 if( CORBA::is_nil( theObject ) || CORBA::is_nil( GetPOA() ) )
168 PortableServer::Servant aServant = GetPOA()->reference_to_servant( theObject );
172 INFOS( "GetServant - Unknown exception was caught!!!" );
177 //=============================================================================
179 * SObjectToObject [ static ]
181 * Get CORBA object corresponding to the SALOMEDS::SObject
183 //=============================================================================
185 CORBA::Object_var SMESH_Gen_i::SObjectToObject( SALOMEDS::SObject_ptr theSObject )
187 SALOMEDS::GenericAttribute_wrap anAttr;
188 CORBA::Object_var anObj;
189 if ( !theSObject->_is_nil() ) {
191 if( theSObject->FindAttribute( anAttr.inout(), "AttributeIOR" ) ) {
192 SALOMEDS::AttributeIOR_wrap anIOR = anAttr;
193 CORBA::String_var aValue = anIOR->Value();
194 if( strcmp( aValue, "" ) != 0 )
195 anObj = GetORB()->string_to_object( aValue );
199 INFOS( "SObjectToObject - Unknown exception was caught!!!" );
205 //=============================================================================
209 * Get SALOME_NamingService object
211 //=============================================================================
213 SALOME_NamingService* SMESH_Gen_i::GetNS()
215 if ( myNS == NULL ) {
216 myNS = SINGLETON_<SALOME_NamingService>::Instance();
217 ASSERT(SINGLETON_<SALOME_NamingService>::IsAlreadyExisting());
218 myNS->init_orb( GetORB() );
223 //=============================================================================
227 * Get SALOME_LifeCycleCORBA object
229 //=============================================================================
230 SALOME_LifeCycleCORBA* SMESH_Gen_i::GetLCC() {
231 if ( myLCC == NULL ) {
232 myLCC = new SALOME_LifeCycleCORBA( GetNS() );
238 //=============================================================================
240 * GetGeomEngine [ static ]
242 * Get GEOM::GEOM_Gen reference
244 //=============================================================================
245 GEOM::GEOM_Gen_var SMESH_Gen_i::GetGeomEngine() {
246 //CCRT GEOM::GEOM_Gen_var aGeomEngine =
247 //CCRT GEOM::GEOM_Gen::_narrow( GetLCC()->FindOrLoad_Component("FactoryServer","GEOM") );
248 //CCRT return aGeomEngine._retn();
249 if(CORBA::is_nil(myGeomGen))
251 Engines::EngineComponent_ptr temp=GetLCC()->FindOrLoad_Component("FactoryServer","GEOM");
252 myGeomGen=GEOM::GEOM_Gen::_narrow(temp);
257 //=============================================================================
259 * SMESH_Gen_i::SMESH_Gen_i
261 * Default constructor: not for use
263 //=============================================================================
265 SMESH_Gen_i::SMESH_Gen_i()
267 INFOS( "SMESH_Gen_i::SMESH_Gen_i : default constructor" );
270 //=============================================================================
272 * SMESH_Gen_i::SMESH_Gen_i
274 * Standard constructor, used with Container
276 //=============================================================================
278 SMESH_Gen_i::SMESH_Gen_i( CORBA::ORB_ptr orb,
279 PortableServer::POA_ptr poa,
280 PortableServer::ObjectId* contId,
281 const char* instanceName,
282 const char* interfaceName )
283 : Engines_Component_i( orb, poa, contId, instanceName, interfaceName )
285 MESSAGE( "SMESH_Gen_i::SMESH_Gen_i : standard constructor" );
287 myOrb = CORBA::ORB::_duplicate(orb);
288 myPoa = PortableServer::POA::_duplicate(poa);
291 _id = myPoa->activate_object( _thisObj );
293 myIsEmbeddedMode = false;
294 myShapeReader = NULL; // shape reader
296 myIsHistoricalPythonDump = true;
297 myToForgetMeshDataOnHypModif = false;
299 myImportedStudyChanged = true;
300 myImportedStudyId = 0;
302 // set it in standalone mode only
303 //OSD::SetSignal( true );
305 // 0020605: EDF 1190 SMESH: Display performance. 80 seconds for 52000 cells.
306 // find out mode (embedded or standalone) here else
307 // meshes created before calling SMESH_Client::GetSMESHGen(), which calls
308 // SMESH_Gen_i::SetEmbeddedMode(), have wrong IsEmbeddedMode flag
309 if ( SALOME_NamingService* ns = GetNS() )
311 CORBA::Object_var obj = ns->Resolve( "/Kernel/Session" );
312 SALOME::Session_var session = SALOME::Session::_narrow( obj ) ;
313 if ( !session->_is_nil() )
315 CORBA::String_var str_host = session->getHostname();
316 CORBA::Long s_pid = session->getPID();
317 string my_host = Kernel_Utils::GetHostname();
319 long my_pid = (long)_getpid();
321 long my_pid = (long) getpid();
323 SetEmbeddedMode( s_pid == my_pid && my_host == str_host.in() );
328 //=============================================================================
330 * SMESH_Gen_i::~SMESH_Gen_i
334 //=============================================================================
336 SMESH_Gen_i::~SMESH_Gen_i()
338 MESSAGE( "SMESH_Gen_i::~SMESH_Gen_i" );
340 // delete hypothesis creators
341 map<string, GenericHypothesisCreator_i*>::iterator itHyp;
342 for (itHyp = myHypCreatorMap.begin(); itHyp != myHypCreatorMap.end(); itHyp++)
344 delete (*itHyp).second;
346 myHypCreatorMap.clear();
348 // Clear study contexts data
349 map<int, StudyContext*>::iterator it;
350 for ( it = myStudyContextMap.begin(); it != myStudyContextMap.end(); ++it ) {
353 myStudyContextMap.clear();
354 // delete shape reader
356 delete myShapeReader;
358 //=============================================================================
360 * SMESH_Gen_i::getHypothesisCreator
362 * Get hypothesis creator
364 //=============================================================================
365 GenericHypothesisCreator_i* SMESH_Gen_i::getHypothesisCreator(const char* theHypName,
366 const char* theLibName,
367 std::string& thePlatformLibName)
368 throw (SALOME::SALOME_Exception)
370 std::string aPlatformLibName;
371 /* It's Need to tranlate lib name for WIN32 or X platform */
372 if ( theLibName && theLibName[0] != '\0' )
374 int libNameLen = strlen(theLibName);
375 //check for old format "libXXXXXXX.so"
376 if (libNameLen > 7 &&
377 !strncmp( theLibName, "lib", 3 ) &&
378 !strcmp( theLibName+libNameLen-3, ".so" ))
382 aPlatformLibName = std::string( theLibName+3, libNameLen-6 ) + ".dll";
384 aPlatformLibName = theLibName;
389 //try to use new format
391 aPlatformLibName = theLibName;
392 aPlatformLibName += ".dll";
394 aPlatformLibName = "lib" + std::string( theLibName ) + ".so";
398 thePlatformLibName = aPlatformLibName;
400 Unexpect aCatch(SALOME_SalomeException);
401 if(MYDEBUG) MESSAGE( "Create Hypothesis <" << theHypName << "> from " << aPlatformLibName);
403 typedef GenericHypothesisCreator_i* (*GetHypothesisCreator)(const char* );
404 GenericHypothesisCreator_i* aCreator;
407 // check, if creator for this hypothesis type already exists
408 if (myHypCreatorMap.find(string(theHypName)) == myHypCreatorMap.end())
410 // load plugin library
411 if(MYDEBUG) MESSAGE("Loading server meshers plugin library ...");
412 LibHandle libHandle = LoadLib( aPlatformLibName.c_str() );
415 // report any error, if occured
417 const char* anError = dlerror();
418 throw(SALOME_Exception(anError));
420 throw(SALOME_Exception(LOCALIZED( "Can't load server meshers plugin library" )));
424 // get method, returning hypothesis creator
425 if(MYDEBUG) MESSAGE("Find GetHypothesisCreator() method ...");
426 GetHypothesisCreator procHandle =
427 (GetHypothesisCreator)GetProc( libHandle, "GetHypothesisCreator" );
430 throw(SALOME_Exception(LOCALIZED("bad hypothesis plugin library")));
431 UnLoadLib(libHandle);
434 // get hypothesis creator
435 if(MYDEBUG) MESSAGE("Get Hypothesis Creator for " << theHypName);
436 aCreator = procHandle(theHypName);
439 throw(SALOME_Exception(LOCALIZED("no such a hypothesis in this plugin")));
441 // map hypothesis creator to a hypothesis name
442 myHypCreatorMap[string(theHypName)] = aCreator;
447 return myHypCreatorMap[string(theHypName)];
450 catch (SALOME_Exception& S_ex)
452 THROW_SALOME_CORBA_EXCEPTION(S_ex.what(), SALOME::BAD_PARAM);
457 //=============================================================================
459 * SMESH_Gen_i::createHypothesis
461 * Create hypothesis of given type
463 //=============================================================================
464 SMESH::SMESH_Hypothesis_ptr SMESH_Gen_i::createHypothesis(const char* theHypName,
465 const char* theLibName)
467 SMESH_Hypothesis_i* myHypothesis_i = 0;
468 SMESH::SMESH_Hypothesis_var hypothesis_i;
469 std::string aPlatformLibName;
470 typedef GenericHypothesisCreator_i* (*GetHypothesisCreator)(const char* );
471 GenericHypothesisCreator_i* aCreator = getHypothesisCreator(theHypName, theLibName, aPlatformLibName);
472 // create a new hypothesis object, store its ref. in studyContext
473 if(MYDEBUG) MESSAGE("Create Hypothesis " << theHypName);
475 myHypCreatorMap[string(theHypName)]->Create(myPoa, GetCurrentStudyID(), &myGen);
476 myHypothesis_i->SetLibName(aPlatformLibName.c_str()); // for persistency assurance
479 return hypothesis_i._retn();
481 // activate the CORBA servant of hypothesis
482 hypothesis_i = myHypothesis_i->_this();
483 int nextId = RegisterObject( hypothesis_i );
484 if(MYDEBUG) { MESSAGE( "Add hypo to map with id = "<< nextId ); }
485 else { nextId = 0; } // avoid "unused variable" warning in release mode
487 return hypothesis_i._retn();
490 //=============================================================================
492 * SMESH_Gen_i::createMesh
494 * Create empty mesh on shape
496 //=============================================================================
497 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::createMesh()
498 throw ( SALOME::SALOME_Exception )
500 Unexpect aCatch(SALOME_SalomeException);
501 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::createMesh" );
503 // Get or create the GEOM_Client instance
505 // create a new mesh object servant, store it in a map in study context
506 SMESH_Mesh_i* meshServant = new SMESH_Mesh_i( GetPOA(), this, GetCurrentStudyID() );
507 // create a new mesh object
508 MESSAGE("myIsEmbeddedMode " << myIsEmbeddedMode);
509 meshServant->SetImpl( myGen.CreateMesh( GetCurrentStudyID(), myIsEmbeddedMode ));
511 // activate the CORBA servant of Mesh
512 SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( meshServant->_this() );
513 int nextId = RegisterObject( mesh );
514 if(MYDEBUG) { MESSAGE( "Add mesh to map with id = "<< nextId); }
515 else { nextId = 0; } // avoid "unused variable" warning in release mode
518 catch (SALOME_Exception& S_ex) {
519 THROW_SALOME_CORBA_EXCEPTION( S_ex.what(), SALOME::BAD_PARAM );
521 return SMESH::SMESH_Mesh::_nil();
524 //=============================================================================
526 * SMESH_Gen_i::GetShapeReader
530 //=============================================================================
531 GEOM_Client* SMESH_Gen_i::GetShapeReader()
533 // create shape reader if necessary
534 if ( !myShapeReader )
535 myShapeReader = new GEOM_Client(GetContainerRef());
536 ASSERT( myShapeReader );
537 return myShapeReader;
540 //=============================================================================
542 * SMESH_Gen_i::SetGeomEngine
544 * Set GEOM::GEOM_Gen reference
546 //=============================================================================
547 //GEOM::GEOM_Gen_ptr SMESH_Gen_i::SetGeomEngine( const char* containerLoc )
548 void SMESH_Gen_i::SetGeomEngine( GEOM::GEOM_Gen_ptr geomcompo )
550 //Engines::Component_ptr temp=GetLCC()->FindOrLoad_Component(containerLoc,"GEOM");
551 //myGeomGen=GEOM::GEOM_Gen::_narrow(temp);
552 myGeomGen=GEOM::GEOM_Gen::_duplicate(geomcompo);
556 //=============================================================================
558 * SMESH_Gen_i::SetEmbeddedMode
562 //=============================================================================
564 void SMESH_Gen_i::SetEmbeddedMode( CORBA::Boolean theMode )
566 myIsEmbeddedMode = theMode;
567 MESSAGE("myIsEmbeddedMode " << myIsEmbeddedMode);
569 if ( !myIsEmbeddedMode ) {
570 //PAL10867: disable signals catching with "noexcepthandler" option
571 char* envNoCatchSignals = getenv("NOT_INTERCEPT_SIGNALS");
572 if (!envNoCatchSignals || !atoi(envNoCatchSignals))
577 char* envDisableFPE = getenv("DISABLE_FPE");
578 if (envDisableFPE && atoi(envDisableFPE))
583 OSD::SetSignal( raiseFPE );
585 // else OSD::SetSignal() is called in GUI
589 //=============================================================================
591 * SMESH_Gen_i::IsEmbeddedMode
595 //=============================================================================
597 CORBA::Boolean SMESH_Gen_i::IsEmbeddedMode()
599 return myIsEmbeddedMode;
602 //=============================================================================
604 * SMESH_Gen_i::SetCurrentStudy
608 //=============================================================================
610 void SMESH_Gen_i::SetCurrentStudy( SALOMEDS::Study_ptr theStudy )
612 setCurrentStudy( theStudy );
615 void SMESH_Gen_i::setCurrentStudy( SALOMEDS::Study_ptr theStudy,
616 bool theStudyIsBeingClosed)
618 int curStudyId = GetCurrentStudyID();
619 myCurrentStudy = SALOMEDS::Study::_duplicate( theStudy );
620 // create study context, if it doesn't exist and set current study
621 int studyId = GetCurrentStudyID();
622 if ( myStudyContextMap.find( studyId ) == myStudyContextMap.end() )
623 myStudyContextMap[ studyId ] = new StudyContext;
625 // myCurrentStudy may be nil
626 if ( !theStudyIsBeingClosed && !CORBA::is_nil( myCurrentStudy ) ) {
627 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
628 SALOMEDS::SComponent_wrap GEOM_var = myCurrentStudy->FindComponent( "GEOM" );
629 if( !GEOM_var->_is_nil() )
630 aStudyBuilder->LoadWith( GEOM_var, GetGeomEngine() );
631 // NPAL16168, issue 0020210
632 // Let meshes update their data depending on GEOM groups that could change
633 if ( curStudyId != studyId )
635 CORBA::String_var compDataType = ComponentDataType();
636 SALOMEDS::SComponent_wrap me = myCurrentStudy->FindComponent( compDataType.in() );
637 if ( !me->_is_nil() ) {
638 SALOMEDS::ChildIterator_wrap anIter = myCurrentStudy->NewChildIterator( me );
639 for ( ; anIter->More(); anIter->Next() ) {
640 SALOMEDS::SObject_wrap so = anIter->Value();
641 CORBA::Object_var ior = SObjectToObject( so );
642 if ( SMESH_Mesh_i* mesh = SMESH::DownCast<SMESH_Mesh_i*>( ior ))
643 mesh->CheckGeomModif();
650 //=============================================================================
652 * SMESH_Gen_i::GetCurrentStudy
656 //=============================================================================
658 SALOMEDS::Study_ptr SMESH_Gen_i::GetCurrentStudy()
660 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetCurrentStudy: study Id = " << GetCurrentStudyID() );
661 return SALOMEDS::Study::_duplicate( myCurrentStudy );
664 //=============================================================================
666 * SMESH_Gen_i::GetCurrentStudyContext
668 * Get current study context
670 //=============================================================================
671 StudyContext* SMESH_Gen_i::GetCurrentStudyContext()
673 if ( !CORBA::is_nil( myCurrentStudy ) &&
674 myStudyContextMap.find( GetCurrentStudyID() ) != myStudyContextMap.end() )
675 return myStudyContextMap[ myCurrentStudy->StudyId() ];
680 //=============================================================================
682 * SMESH_Gen_i::CreateHypothesis
684 * Create hypothesis/algorothm of given type and publish it in the study
686 //=============================================================================
688 SMESH::SMESH_Hypothesis_ptr SMESH_Gen_i::CreateHypothesis( const char* theHypName,
689 const char* theLibName )
690 throw ( SALOME::SALOME_Exception )
692 Unexpect aCatch(SALOME_SalomeException);
693 // Create hypothesis/algorithm
694 SMESH::SMESH_Hypothesis_var hyp = this->createHypothesis( theHypName, theLibName );
696 // Publish hypothesis/algorithm in the study
697 if ( CanPublishInStudy( hyp ) ) {
698 SALOMEDS::SObject_wrap aSO = PublishHypothesis( myCurrentStudy, hyp );
699 if ( !aSO->_is_nil() ) {
700 // Update Python script
701 TPythonDump() << aSO << " = " << this << ".CreateHypothesis('"
702 << theHypName << "', '" << theLibName << "')";
709 //================================================================================
711 * \brief Return a hypothesis holding parameter values corresponding either to the mesh
712 * existing on the given geometry or to size of the geometry.
713 * \param theHypType - hypothesis type name
714 * \param theLibName - plugin library name
715 * \param theMesh - The mesh of interest
716 * \param theGeom - The shape to get parameter values from
717 * \retval SMESH::SMESH_Hypothesis_ptr - The returned hypothesis may be the one existing
718 * in a study and used to compute the mesh, or a temporary one created just to pass
721 //================================================================================
723 SMESH::SMESH_Hypothesis_ptr
724 SMESH_Gen_i::GetHypothesisParameterValues (const char* theHypType,
725 const char* theLibName,
726 SMESH::SMESH_Mesh_ptr theMesh,
727 GEOM::GEOM_Object_ptr theGeom,
728 CORBA::Boolean byMesh)
729 throw ( SALOME::SALOME_Exception )
731 Unexpect aCatch(SALOME_SalomeException);
732 if ( byMesh && CORBA::is_nil( theMesh ) )
733 return SMESH::SMESH_Hypothesis::_nil();
734 if ( byMesh && CORBA::is_nil( theGeom ) )
735 return SMESH::SMESH_Hypothesis::_nil();
737 // -----------------------------------------------
738 // find hypothesis used to mesh theGeom
739 // -----------------------------------------------
741 // get mesh and shape
742 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
743 TopoDS_Shape shape = GeomObjectToShape( theGeom );
744 if ( byMesh && ( !meshServant || meshServant->NbNodes()==0 || shape.IsNull() ))
745 return SMESH::SMESH_Hypothesis::_nil();
746 ::SMESH_Mesh* mesh = meshServant ? &meshServant->GetImpl() : (::SMESH_Mesh*)0;
748 // create a temporary hypothesis to know its dimention
749 SMESH::SMESH_Hypothesis_var tmpHyp = this->createHypothesis( theHypType, theLibName );
750 SMESH_Hypothesis_i* hypServant = SMESH::DownCast<SMESH_Hypothesis_i*>( tmpHyp );
752 return SMESH::SMESH_Hypothesis::_nil();
753 ::SMESH_Hypothesis* hyp = hypServant->GetImpl();
756 // look for a hypothesis of theHypType used to mesh the shape
757 if ( myGen.GetShapeDim( shape ) == hyp->GetDim() )
760 SMESH::ListOfHypothesis_var aHypList = theMesh->GetHypothesisList( theGeom );
761 int nbLocalHyps = aHypList->length();
762 for ( int i = 0; i < nbLocalHyps; i++ ) {
763 CORBA::String_var hypName = aHypList[i]->GetName();
764 if ( strcmp( theHypType, hypName.in() ) == 0 ) // FOUND local!
765 return SMESH::SMESH_Hypothesis::_duplicate( aHypList[i] );
767 // check super shapes
768 TopTools_ListIteratorOfListOfShape itShape( mesh->GetAncestors( shape ));
769 while ( nbLocalHyps == 0 && itShape.More() ) {
770 GEOM::GEOM_Object_ptr geomObj = ShapeToGeomObject( itShape.Value() );
771 if ( ! CORBA::is_nil( geomObj )) {
772 SMESH::ListOfHypothesis_var aHypList = theMesh->GetHypothesisList( geomObj );
773 nbLocalHyps = aHypList->length();
774 for ( int i = 0; i < nbLocalHyps; i++ )
775 if ( strcmp( theHypType, aHypList[i]->GetName() ) == 0 ) // FOUND global!
776 return SMESH::SMESH_Hypothesis::_duplicate( aHypList[i] );
782 // let the temporary hypothesis find out somehow parameter values by mesh
783 if ( hyp->SetParametersByMesh( mesh, shape ))
784 return SMESH::SMESH_Hypothesis::_duplicate( tmpHyp );
789 diagonal = mesh->GetShapeDiagonalSize();
791 diagonal = ::SMESH_Mesh::GetShapeDiagonalSize( shape );
792 ::SMESH_Hypothesis::TDefaults dflts;
793 dflts._elemLength = diagonal / myGen.GetBoundaryBoxSegmentation();
794 dflts._nbSegments = myGen.GetDefaultNbSegments();
795 dflts._shape = &shape;
796 // let the temporary hypothesis initialize it's values
797 if ( hyp->SetParametersByDefaults( dflts, mesh ))
798 return SMESH::SMESH_Hypothesis::_duplicate( tmpHyp );
801 return SMESH::SMESH_Hypothesis::_nil();
804 //=============================================================================
806 * Sets number of segments per diagonal of boundary box of geometry by which
807 * default segment length of appropriate 1D hypotheses is defined
809 //=============================================================================
811 void SMESH_Gen_i::SetBoundaryBoxSegmentation( CORBA::Long theNbSegments )
812 throw ( SALOME::SALOME_Exception )
814 if ( theNbSegments > 0 )
815 myGen.SetBoundaryBoxSegmentation( int( theNbSegments ));
817 THROW_SALOME_CORBA_EXCEPTION( "non-positive number of segments", SALOME::BAD_PARAM );
819 //=============================================================================
821 * \brief Sets default number of segments per edge
823 //=============================================================================
824 void SMESH_Gen_i::SetDefaultNbSegments(CORBA::Long theNbSegments)
825 throw ( SALOME::SALOME_Exception )
827 if ( theNbSegments > 0 )
828 myGen.SetDefaultNbSegments( int(theNbSegments) );
830 THROW_SALOME_CORBA_EXCEPTION( "non-positive number of segments", SALOME::BAD_PARAM );
833 //=============================================================================
837 //=============================================================================
839 void SMESH_Gen_i::SetOption(const char* name, const char* value)
841 if ( name && value && strlen( value ) > 0 )
844 if ( strcmp(name, "historical_python_dump") == 0 )
846 myIsHistoricalPythonDump = ( value[0] == '1' || toupper(value[0]) == 'T' ); // 1 || true
847 msgToGUI = "preferences/SMESH/historical_python_dump/";
848 msgToGUI += myIsHistoricalPythonDump ? "true" : "false";
850 else if ( strcmp(name, "forget_mesh_on_hyp_modif") == 0 )
852 myToForgetMeshDataOnHypModif = ( value[0] == '1' || toupper(value[0]) == 'T' ); // 1 || true
853 msgToGUI = "preferences/SMESH/forget_mesh_on_hyp_modif/";
854 msgToGUI += myToForgetMeshDataOnHypModif ? "true" : "false";
857 // update preferences in case if SetOption() is invoked from python console
858 if ( !msgToGUI.empty() )
860 CORBA::Object_var obj = SMESH_Gen_i::GetNS()->Resolve( "/Kernel/Session" );
861 SALOME::Session_var session = SALOME::Session::_narrow( obj );
862 if ( !CORBA::is_nil( session ) )
863 session->emitMessageOneWay(msgToGUI.c_str());
868 //=============================================================================
870 Return an option value
872 //=============================================================================
874 char* SMESH_Gen_i::GetOption(const char* name)
878 if ( strcmp(name, "historical_python_dump") == 0 )
880 return CORBA::string_dup( myIsHistoricalPythonDump ? "true" : "false" );
882 if ( strcmp(name, "forget_mesh_on_hyp_modif") == 0 )
884 return CORBA::string_dup( myToForgetMeshDataOnHypModif ? "true" : "false" );
887 return CORBA::string_dup( "" );
890 //=============================================================================
892 * SMESH_Gen_i::CreateMesh
894 * Create empty mesh on a shape and publish it in the study
896 //=============================================================================
898 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMesh( GEOM::GEOM_Object_ptr theShapeObject )
899 throw ( SALOME::SALOME_Exception )
901 Unexpect aCatch(SALOME_SalomeException);
902 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::CreateMesh" );
904 SMESH::SMESH_Mesh_var mesh = this->createMesh();
906 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( mesh );
907 ASSERT( meshServant );
908 meshServant->SetShape( theShapeObject );
910 // publish mesh in the study
911 if ( CanPublishInStudy( mesh ) ) {
912 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
913 aStudyBuilder->NewCommand(); // There is a transaction
914 SALOMEDS::SObject_wrap aSO = PublishMesh( myCurrentStudy, mesh.in() );
915 aStudyBuilder->CommitCommand();
916 if ( !aSO->_is_nil() ) {
917 // Update Python script
918 TPythonDump() << aSO << " = " << this << ".CreateMesh(" << theShapeObject << ")";
925 //=============================================================================
927 * SMESH_Gen_i::CreateEmptyMesh
931 //=============================================================================
933 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateEmptyMesh()
934 throw ( SALOME::SALOME_Exception )
936 Unexpect aCatch(SALOME_SalomeException);
937 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::CreateMesh" );
939 SMESH::SMESH_Mesh_var mesh = this->createMesh();
941 // publish mesh in the study
942 if ( CanPublishInStudy( mesh ) ) {
943 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
944 aStudyBuilder->NewCommand(); // There is a transaction
945 SALOMEDS::SObject_wrap aSO = PublishMesh( myCurrentStudy, mesh.in() );
946 aStudyBuilder->CommitCommand();
947 if ( !aSO->_is_nil() ) {
948 // Update Python script
949 TPythonDump() << aSO << " = " << this << ".CreateEmptyMesh()";
956 //=============================================================================
958 * SMESH_Gen_i::CreateMeshFromUNV
960 * Create mesh and import data from UNV file
962 //=============================================================================
964 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMeshesFromUNV( const char* theFileName )
965 throw ( SALOME::SALOME_Exception )
967 Unexpect aCatch(SALOME_SalomeException);
968 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::CreateMeshesFromUNV" );
970 SMESH::SMESH_Mesh_var aMesh = createMesh();
972 // publish mesh in the study
973 if ( CanPublishInStudy( aMesh ) ) {
974 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
975 aStudyBuilder->NewCommand(); // There is a transaction
976 SALOMEDS::SObject_wrap aSO = PublishMesh( myCurrentStudy, aMesh.in(), aFileName.c_str() );
977 aStudyBuilder->CommitCommand();
978 if ( !aSO->_is_nil() ) {
979 // Update Python script
980 TPythonDump() << aSO << " = " << this << ".CreateMeshesFromUNV(r'" << theFileName << "')";
984 SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
986 aServant->ImportUNVFile( theFileName );
988 // Dump creation of groups
989 SMESH::ListOfGroups_var groups = aServant->GetGroups();
991 aServant->GetImpl().GetMeshDS()->Modified();
992 return aMesh._retn();
995 //=============================================================================
997 * SMESH_Gen_i::CreateMeshFromMED
999 * Create mesh and import data from MED file
1001 //=============================================================================
1003 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromMEDorSAUV( const char* theFileName,
1004 SMESH::DriverMED_ReadStatus& theStatus,
1005 const char* theCommandNameForPython,
1006 const char* theFileNameForPython)
1009 char bname[ _MAX_FNAME ];
1010 _splitpath( theFileNameForPython, NULL, NULL, bname, NULL );
1011 string aFileName = bname;
1013 string aFileName = basename( theFileNameForPython );
1015 // Retrieve mesh names from the file
1016 DriverMED_R_SMESHDS_Mesh myReader;
1017 myReader.SetFile( theFileName );
1018 myReader.SetMeshId( -1 );
1019 Driver_Mesh::Status aStatus;
1020 list<string> aNames = myReader.GetMeshNames(aStatus);
1021 SMESH::mesh_array_var aResult = new SMESH::mesh_array();
1022 theStatus = (SMESH::DriverMED_ReadStatus)aStatus;
1024 { // open a new scope to make aPythonDump die before PythonDump in SMESH_Mesh::GetGroups()
1027 TPythonDump aPythonDump;
1028 aPythonDump << "([";
1030 if (theStatus == SMESH::DRS_OK) {
1031 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1032 aStudyBuilder->NewCommand(); // There is a transaction
1033 aResult->length( aNames.size() );
1036 // Iterate through all meshes and create mesh objects
1037 for ( list<string>::iterator it = aNames.begin(); it != aNames.end(); it++ ) {
1039 if (i > 0) aPythonDump << ", ";
1042 SMESH::SMESH_Mesh_var mesh = createMesh();
1044 // publish mesh in the study
1045 SALOMEDS::SObject_wrap aSO;
1046 if ( CanPublishInStudy( mesh ) )
1047 // little trick: for MED file theFileName and theFileNameForPython are the same, but they are different for SAUV
1048 // - as names of meshes are stored in MED file, we use them for data publishing
1049 // - as mesh name is not stored in UNV file, we use file name as name of mesh when publishing data
1050 aSO = PublishMesh( myCurrentStudy, mesh.in(), ( theFileName == theFileNameForPython ) ? (*it).c_str() : aFileName.c_str() );
1051 if ( !aSO->_is_nil() ) {
1056 aPythonDump << "mesh_" << i;
1059 // Read mesh data (groups are published automatically by ImportMEDFile())
1060 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( mesh ).in() );
1061 ASSERT( meshServant );
1062 SMESH::DriverMED_ReadStatus status1 =
1063 meshServant->ImportMEDFile( theFileName, (*it).c_str() );
1064 if (status1 > theStatus)
1065 theStatus = status1;
1067 aResult[i++] = SMESH::SMESH_Mesh::_duplicate( mesh );
1068 meshServant->GetImpl().GetMeshDS()->Modified();
1070 aStudyBuilder->CommitCommand();
1073 // Update Python script
1074 aPythonDump << "], status) = " << this << "." << theCommandNameForPython << "(r'" << theFileNameForPython << "')";
1076 // Dump creation of groups
1077 for ( int i = 0; i < aResult->length(); ++i )
1078 SMESH::ListOfGroups_var groups = aResult[ i ]->GetGroups();
1080 return aResult._retn();
1083 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromMED( const char* theFileName,
1084 SMESH::DriverMED_ReadStatus& theStatus)
1085 throw ( SALOME::SALOME_Exception )
1087 Unexpect aCatch(SALOME_SalomeException);
1088 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::CreateMeshFromMED" );
1089 SMESH::mesh_array* result = CreateMeshesFromMEDorSAUV(theFileName, theStatus, "CreateMeshesFromMED", theFileName);
1093 //=============================================================================
1095 * SMESH_Gen_i::CreateMeshFromSAUV
1097 * Create mesh and import data from SAUV file
1099 //=============================================================================
1101 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromSAUV( const char* theFileName,
1102 SMESH::DriverMED_ReadStatus& theStatus)
1103 throw ( SALOME::SALOME_Exception )
1105 Unexpect aCatch(SALOME_SalomeException);
1106 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::CreateMeshFromSAUV" );
1107 std::string sauvfilename(theFileName);
1108 std::string medfilename(theFileName);
1109 medfilename += ".med";
1112 cmd = "%PYTHONBIN% ";
1117 cmd += "from medutilities import convert ; convert(r'" + sauvfilename + "', 'GIBI', 'MED', 1, r'" + medfilename + "')";
1119 system(cmd.c_str());
1120 SMESH::mesh_array* result = CreateMeshesFromMEDorSAUV(medfilename.c_str(), theStatus, "CreateMeshesFromSAUV", sauvfilename.c_str());
1122 cmd = "%PYTHONBIN% ";
1127 cmd += "from medutilities import my_remove ; my_remove(r'" + medfilename + "')";
1129 system(cmd.c_str());
1133 //=============================================================================
1135 * SMESH_Gen_i::CreateMeshFromSTL
1137 * Create mesh and import data from STL file
1139 //=============================================================================
1141 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMeshesFromSTL( const char* theFileName )
1142 throw ( SALOME::SALOME_Exception )
1144 Unexpect aCatch(SALOME_SalomeException);
1145 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::CreateMeshesFromSTL" );
1147 SMESH::SMESH_Mesh_var aMesh = createMesh();
1150 char bname[ _MAX_FNAME ];
1151 _splitpath( theFileName, NULL, NULL, bname, NULL );
1152 string aFileName = bname;
1154 string aFileName = basename( theFileName );
1156 // publish mesh in the study
1157 if ( CanPublishInStudy( aMesh ) ) {
1158 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1159 aStudyBuilder->NewCommand(); // There is a transaction
1160 SALOMEDS::SObject_wrap aSO = PublishInStudy
1161 ( myCurrentStudy, SALOMEDS::SObject::_nil(), aMesh.in(), aFileName.c_str() );
1162 aStudyBuilder->CommitCommand();
1163 if ( !aSO->_is_nil() ) {
1164 // Update Python script
1165 TPythonDump() << aSO << " = " << this << ".CreateMeshesFromSTL(r'" << theFileName << "')";
1169 SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
1171 aServant->ImportSTLFile( theFileName );
1172 aServant->GetImpl().GetMeshDS()->Modified();
1173 return aMesh._retn();
1176 //================================================================================
1178 * \brief Create meshes and import data from the CGSN file
1180 //================================================================================
1182 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromCGNS( const char* theFileName,
1183 SMESH::DriverMED_ReadStatus& theStatus)
1184 throw ( SALOME::SALOME_Exception )
1186 Unexpect aCatch(SALOME_SalomeException);
1188 SMESH::mesh_array_var aResult = new SMESH::mesh_array();
1191 // Retrieve nb meshes from the file
1192 DriverCGNS_Read myReader;
1193 myReader.SetFile( theFileName );
1194 Driver_Mesh::Status aStatus;
1195 int nbMeshes = myReader.GetNbMeshes(aStatus);
1196 theStatus = (SMESH::DriverMED_ReadStatus)aStatus;
1198 aResult->length( nbMeshes );
1200 { // open a new scope to make aPythonDump die before PythonDump in SMESH_Mesh::GetGroups()
1203 TPythonDump aPythonDump;
1204 aPythonDump << "([";
1206 if (theStatus == SMESH::DRS_OK)
1208 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1209 aStudyBuilder->NewCommand(); // There is a transaction
1213 // Iterate through all meshes and create mesh objects
1214 for ( ; i < nbMeshes; ++i )
1217 if (i > 0) aPythonDump << ", ";
1220 SMESH::SMESH_Mesh_var mesh = createMesh();
1221 aResult[i] = SMESH::SMESH_Mesh::_duplicate( mesh );
1223 // Read mesh data (groups are published automatically by ImportMEDFile())
1224 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( mesh ).in() );
1225 ASSERT( meshServant );
1227 SMESH::DriverMED_ReadStatus status1 =
1228 meshServant->ImportCGNSFile( theFileName, i, meshName );
1229 if (status1 > theStatus)
1230 theStatus = status1;
1232 meshServant->GetImpl().GetMeshDS()->Modified();
1233 // publish mesh in the study
1234 SALOMEDS::SObject_wrap aSO;
1235 if ( CanPublishInStudy( mesh ) )
1236 aSO = PublishMesh( myCurrentStudy, mesh.in(), meshName.c_str() );
1239 if ( !aSO->_is_nil() ) {
1243 aPythonDump << "mesh_" << i;
1246 aStudyBuilder->CommitCommand();
1249 aPythonDump << "], status) = " << this << ".CreateMeshesFromCGNS(r'" << theFileName << "')";
1251 // Dump creation of groups
1252 for ( int i = 0; i < aResult->length(); ++i )
1253 SMESH::ListOfGroups_var groups = aResult[ i ]->GetGroups();
1255 THROW_SALOME_CORBA_EXCEPTION("CGNS library is unavailable", SALOME::INTERNAL_ERROR);
1258 return aResult._retn();
1261 //================================================================================
1263 * \brief Create a mesh and import data from a GMF file
1265 //================================================================================
1267 SMESH::SMESH_Mesh_ptr
1268 SMESH_Gen_i::CreateMeshesFromGMF( const char* theFileName,
1269 CORBA::Boolean theMakeRequiredGroups,
1270 SMESH::ComputeError_out theError)
1271 throw ( SALOME::SALOME_Exception )
1273 Unexpect aCatch(SALOME_SalomeException);
1275 SMESH::SMESH_Mesh_var aMesh = createMesh();
1277 char bname[ _MAX_FNAME ];
1278 _splitpath( theFileName, NULL, NULL, bname, NULL );
1279 string aFileName = bname;
1281 string aFileName = basename( theFileName );
1283 // publish mesh in the study
1284 if ( CanPublishInStudy( aMesh ) ) {
1285 SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1286 aStudyBuilder->NewCommand(); // There is a transaction
1287 SALOMEDS::SObject_wrap aSO = PublishInStudy
1288 ( myCurrentStudy, SALOMEDS::SObject::_nil(), aMesh.in(), aFileName.c_str() );
1289 aStudyBuilder->CommitCommand();
1290 if ( !aSO->_is_nil() ) {
1291 // Update Python script
1292 TPythonDump() << "("<< aSO << ", error) = " << this << ".CreateMeshesFromGMF(r'"
1293 << theFileName << "', "
1294 << theMakeRequiredGroups << " )";
1297 SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
1299 theError = aServant->ImportGMFFile( theFileName, theMakeRequiredGroups );
1300 aServant->GetImpl().GetMeshDS()->Modified();
1301 return aMesh._retn();
1305 //=============================================================================
1307 * SMESH_Gen_i::IsReadyToCompute
1309 * Returns true if mesh contains enough data to be computed
1311 //=============================================================================
1313 CORBA::Boolean SMESH_Gen_i::IsReadyToCompute( SMESH::SMESH_Mesh_ptr theMesh,
1314 GEOM::GEOM_Object_ptr theShapeObject )
1315 throw ( SALOME::SALOME_Exception )
1317 Unexpect aCatch(SALOME_SalomeException);
1318 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::IsReadyToCompute" );
1320 if ( CORBA::is_nil( theShapeObject ) )
1321 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1322 SALOME::BAD_PARAM );
1324 if ( CORBA::is_nil( theMesh ) )
1325 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
1326 SALOME::BAD_PARAM );
1330 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1331 ASSERT( meshServant );
1332 if ( meshServant ) {
1333 // get local TopoDS_Shape
1334 TopoDS_Shape myLocShape = GeomObjectToShape( theShapeObject );
1335 // call implementation
1336 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1337 return myGen.CheckAlgoState( myLocMesh, myLocShape );
1340 catch ( SALOME_Exception& S_ex ) {
1341 INFOS( "catch exception "<< S_ex.what() );
1346 //================================================================================
1348 * \brief Find SObject for an algo
1350 //================================================================================
1352 SALOMEDS::SObject_ptr SMESH_Gen_i::GetAlgoSO(const ::SMESH_Algo* algo)
1355 if ( !myCurrentStudy->_is_nil() ) {
1356 // find algo in the study
1357 CORBA::String_var compDataType = ComponentDataType();
1358 SALOMEDS::SComponent_wrap father = myCurrentStudy->FindComponent( compDataType.in() );
1359 if ( !father->_is_nil() ) {
1360 SALOMEDS::ChildIterator_wrap itBig = myCurrentStudy->NewChildIterator( father );
1361 for ( ; itBig->More(); itBig->Next() ) {
1362 SALOMEDS::SObject_wrap gotBranch = itBig->Value();
1363 if ( gotBranch->Tag() == GetAlgorithmsRootTag() ) {
1364 SALOMEDS::ChildIterator_wrap algoIt = myCurrentStudy->NewChildIterator( gotBranch );
1365 for ( ; algoIt->More(); algoIt->Next() ) {
1366 SALOMEDS::SObject_wrap algoSO = algoIt->Value();
1367 CORBA::Object_var algoIOR = SObjectToObject( algoSO );
1368 if ( !CORBA::is_nil( algoIOR )) {
1369 SMESH_Hypothesis_i* impl = SMESH::DownCast<SMESH_Hypothesis_i*>( algoIOR );
1370 if ( impl && impl->GetImpl() == algo )
1371 return algoSO._retn();
1373 } // loop on algo SO's
1376 } // SMESH component iterator
1380 return SALOMEDS::SObject::_nil();
1383 //================================================================================
1385 * \brief Return errors of mesh computation
1387 //================================================================================
1389 SMESH::compute_error_array* SMESH_Gen_i::GetComputeErrors( SMESH::SMESH_Mesh_ptr theMesh,
1390 GEOM::GEOM_Object_ptr theSubObject )
1391 throw ( SALOME::SALOME_Exception )
1393 Unexpect aCatch(SALOME_SalomeException);
1394 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetComputeErrors()" );
1396 if ( CORBA::is_nil( theSubObject ) && theMesh->HasShapeToMesh())
1397 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference", SALOME::BAD_PARAM );
1399 if ( CORBA::is_nil( theMesh ) )
1400 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1402 SMESH::compute_error_array_var error_array = new SMESH::compute_error_array;
1404 if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1407 if(theMesh->HasShapeToMesh())
1408 shape = GeomObjectToShape( theSubObject );
1410 shape = SMESH_Mesh::PseudoShape();
1412 ::SMESH_Mesh& mesh = meshServant->GetImpl();
1414 error_array->length( mesh.GetMeshDS()->MaxShapeIndex() );
1417 SMESH_subMesh *sm = mesh.GetSubMesh(shape);
1418 const bool includeSelf = true, complexShapeFirst = true;
1419 SMESH_subMeshIteratorPtr smIt = sm->getDependsOnIterator(includeSelf,
1421 while ( smIt->more() )
1424 // if ( sm->GetSubShape().ShapeType() == TopAbs_VERTEX )
1426 SMESH_ComputeErrorPtr error = sm->GetComputeError();
1427 if ( error && !error->IsOK() && error->myAlgo )
1429 SMESH::ComputeError & errStruct = error_array[ nbErr++ ];
1430 errStruct.code = -( error->myName < 0 ? error->myName + 1: error->myName ); // -1 -> 0
1431 errStruct.comment = error->myComment.c_str();
1432 errStruct.subShapeID = sm->GetId();
1433 SALOMEDS::SObject_wrap algoSO = GetAlgoSO( error->myAlgo );
1434 if ( !algoSO->_is_nil() ) {
1435 CORBA::String_var algoName = algoSO->GetName();
1436 errStruct.algoName = algoName;
1439 errStruct.algoName = error->myAlgo->GetName();
1441 errStruct.hasBadMesh = !error->myBadElements.empty();
1444 error_array->length( nbErr );
1447 catch ( SALOME_Exception& S_ex ) {
1448 INFOS( "catch exception "<< S_ex.what() );
1451 return error_array._retn();
1454 //================================================================================
1456 * \brief Return mesh elements preventing computation of a subshape
1458 //================================================================================
1460 SMESH::MeshPreviewStruct*
1461 SMESH_Gen_i::GetBadInputElements( SMESH::SMESH_Mesh_ptr theMesh,
1462 CORBA::Short theSubShapeID )
1463 throw ( SALOME::SALOME_Exception )
1465 Unexpect aCatch(SALOME_SalomeException);
1466 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetBadInputElements()" );
1468 if ( CORBA::is_nil( theMesh ) )
1469 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1471 SMESH::MeshPreviewStruct_var result = new SMESH::MeshPreviewStruct;
1474 if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1476 // mesh implementation
1477 ::SMESH_Mesh& mesh = meshServant->GetImpl();
1478 // submesh by subshape id
1479 if ( SMESH_subMesh * sm = mesh.GetSubMeshContaining( theSubShapeID ))
1482 SMESH_ComputeErrorPtr error = sm->GetComputeError();
1483 if ( error && !error->myBadElements.empty())
1485 typedef map<const SMDS_MeshElement*, int > TNode2LocalIDMap;
1486 typedef TNode2LocalIDMap::iterator TNodeLocalID;
1488 // get nodes of elements and count elements
1489 TNode2LocalIDMap mapNode2LocalID;
1490 list< TNodeLocalID > connectivity;
1491 int i, nbElements = 0, nbConnNodes = 0;
1493 list<const SMDS_MeshElement*>::iterator elemIt = error->myBadElements.begin();
1494 list<const SMDS_MeshElement*>::iterator elemEnd = error->myBadElements.end();
1495 for ( ; elemIt != elemEnd; ++elemIt, ++nbElements )
1497 SMDS_ElemIteratorPtr nIt = (*elemIt)->nodesIterator();
1498 while ( nIt->more() )
1499 connectivity.push_back
1500 ( mapNode2LocalID.insert( make_pair( nIt->next(), ++nbConnNodes)).first );
1502 // fill node coords and assign local ids to the nodes
1503 int nbNodes = mapNode2LocalID.size();
1504 result->nodesXYZ.length( nbNodes );
1505 TNodeLocalID node2ID = mapNode2LocalID.begin();
1506 for ( i = 0; i < nbNodes; ++i, ++node2ID ) {
1507 node2ID->second = i;
1508 const SMDS_MeshNode* node = (const SMDS_MeshNode*) node2ID->first;
1509 result->nodesXYZ[i].x = node->X();
1510 result->nodesXYZ[i].y = node->Y();
1511 result->nodesXYZ[i].z = node->Z();
1513 // fill connectivity
1514 result->elementConnectivities.length( nbConnNodes );
1515 list< TNodeLocalID >::iterator connIt = connectivity.begin();
1516 for ( i = 0; i < nbConnNodes; ++i, ++connIt ) {
1517 result->elementConnectivities[i] = (*connIt)->second;
1519 // fill element types
1520 result->elementTypes.length( nbElements );
1521 for ( i = 0, elemIt = error->myBadElements.begin(); i <nbElements; ++i, ++elemIt )
1523 const SMDS_MeshElement* elem = *elemIt;
1524 result->elementTypes[i].SMDS_ElementType = (SMESH::ElementType) elem->GetType();
1525 result->elementTypes[i].isPoly = elem->IsPoly();
1526 result->elementTypes[i].nbNodesInElement = elem->NbNodes();
1532 catch ( SALOME_Exception& S_ex ) {
1533 INFOS( "catch exception "<< S_ex.what() );
1536 return result._retn();
1539 //================================================================================
1541 * \brief Create a group of elements preventing computation of a sub-shape
1543 //================================================================================
1545 SMESH::ListOfGroups*
1546 SMESH_Gen_i::MakeGroupsOfBadInputElements( SMESH::SMESH_Mesh_ptr theMesh,
1547 CORBA::Short theSubShapeID,
1548 const char* theGroupName )
1549 throw ( SALOME::SALOME_Exception )
1551 Unexpect aCatch(SALOME_SalomeException);
1553 SMESH::ListOfGroups_var groups;
1555 if ( CORBA::is_nil( theMesh ) )
1556 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1559 if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1561 groups = meshServant->MakeGroupsOfBadInputElements( theSubShapeID, theGroupName );
1562 TPythonDump() << groups << " = " << this
1563 << ".MakeGroupsOfBadInputElements( "
1564 << theMesh << ", " << theSubShapeID << ", '" << theGroupName << "' )";
1567 catch ( SALOME_Exception& S_ex ) {
1568 INFOS( "catch exception "<< S_ex.what() );
1570 return groups._retn();
1573 //================================================================================
1575 * \brief Returns errors of hypotheses definintion
1576 * \param theMesh - the mesh
1577 * \param theSubObject - the main or sub- shape
1578 * \retval SMESH::algo_error_array* - sequence of errors
1580 //================================================================================
1582 SMESH::algo_error_array* SMESH_Gen_i::GetAlgoState( SMESH::SMESH_Mesh_ptr theMesh,
1583 GEOM::GEOM_Object_ptr theSubObject )
1584 throw ( SALOME::SALOME_Exception )
1586 Unexpect aCatch(SALOME_SalomeException);
1587 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetAlgoState()" );
1589 if ( CORBA::is_nil( theSubObject ) && theMesh->HasShapeToMesh())
1590 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference", SALOME::BAD_PARAM );
1592 if ( CORBA::is_nil( theMesh ) )
1593 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1595 SMESH::algo_error_array_var error_array = new SMESH::algo_error_array;
1597 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
1598 ASSERT( meshServant );
1599 if ( meshServant ) {
1600 TopoDS_Shape myLocShape;
1601 if(theMesh->HasShapeToMesh())
1602 myLocShape = GeomObjectToShape( theSubObject );
1604 myLocShape = SMESH_Mesh::PseudoShape();
1606 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1607 list< ::SMESH_Gen::TAlgoStateError > error_list;
1608 list< ::SMESH_Gen::TAlgoStateError >::iterator error;
1609 // call ::SMESH_Gen::GetAlgoState()
1610 myGen.GetAlgoState( myLocMesh, myLocShape, error_list );
1611 error_array->length( error_list.size() );
1613 for ( error = error_list.begin(); error != error_list.end(); ++error )
1615 // fill AlgoStateError structure
1616 SMESH::AlgoStateError & errStruct = error_array[ i++ ];
1617 errStruct.state = SMESH_Mesh_i::ConvertHypothesisStatus( error->_name );
1618 errStruct.algoDim = error->_algoDim;
1619 errStruct.isGlobalAlgo = error->_isGlobalAlgo;
1620 errStruct.algoName = "";
1621 SALOMEDS::SObject_wrap algoSO = GetAlgoSO( error->_algo );
1622 if ( !algoSO->_is_nil() ) {
1623 CORBA::String_var algoName = algoSO->GetName();
1624 errStruct.algoName = algoName.in();
1629 catch ( SALOME_Exception& S_ex ) {
1630 INFOS( "catch exception "<< S_ex.what() );
1632 return error_array._retn();
1635 //=============================================================================
1637 * SMESH_Gen_i::GetSubShapesId
1639 * Get sub-shapes unique ID's list
1641 //=============================================================================
1643 SMESH::long_array* SMESH_Gen_i::GetSubShapesId( GEOM::GEOM_Object_ptr theMainShapeObject,
1644 const SMESH::object_array& theListOfSubShapeObject )
1645 throw ( SALOME::SALOME_Exception )
1647 Unexpect aCatch(SALOME_SalomeException);
1648 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetSubShapesId" );
1650 SMESH::long_array_var shapesId = new SMESH::long_array;
1653 if ( CORBA::is_nil( theMainShapeObject ) )
1654 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1655 SALOME::BAD_PARAM );
1659 TopoDS_Shape myMainShape = GeomObjectToShape(theMainShapeObject);
1660 TopTools_IndexedMapOfShape myIndexToShape;
1661 TopExp::MapShapes(myMainShape,myIndexToShape);
1663 for ( int i = 0; i < theListOfSubShapeObject.length(); i++ )
1665 GEOM::GEOM_Object_var aShapeObject
1666 = GEOM::GEOM_Object::_narrow(theListOfSubShapeObject[i]);
1667 if ( CORBA::is_nil( aShapeObject ) )
1668 THROW_SALOME_CORBA_EXCEPTION ("bad shape object reference", \
1669 SALOME::BAD_PARAM );
1671 TopoDS_Shape locShape = GeomObjectToShape(aShapeObject);
1672 for (TopExp_Explorer exp(locShape,TopAbs_FACE); exp.More(); exp.Next())
1674 const TopoDS_Face& F = TopoDS::Face(exp.Current());
1675 setId.insert(myIndexToShape.FindIndex(F));
1676 if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(F));
1678 for (TopExp_Explorer exp(locShape,TopAbs_EDGE); exp.More(); exp.Next())
1680 const TopoDS_Edge& E = TopoDS::Edge(exp.Current());
1681 setId.insert(myIndexToShape.FindIndex(E));
1682 if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(E));
1684 for (TopExp_Explorer exp(locShape,TopAbs_VERTEX); exp.More(); exp.Next())
1686 const TopoDS_Vertex& V = TopoDS::Vertex(exp.Current());
1687 setId.insert(myIndexToShape.FindIndex(V));
1688 if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(V));
1691 shapesId->length(setId.size());
1692 set<int>::iterator iind;
1694 for (iind = setId.begin(); iind != setId.end(); iind++)
1696 if(MYDEBUG) SCRUTE((*iind));
1697 shapesId[i] = (*iind);
1698 if(MYDEBUG) SCRUTE(shapesId[i]);
1702 catch (SALOME_Exception& S_ex)
1704 THROW_SALOME_CORBA_EXCEPTION(S_ex.what(), SALOME::BAD_PARAM);
1707 return shapesId._retn();
1710 //=============================================================================
1712 * SMESH_Gen_i::Compute
1714 * Compute mesh on a shape
1716 //=============================================================================
1718 CORBA::Boolean SMESH_Gen_i::Compute( SMESH::SMESH_Mesh_ptr theMesh,
1719 GEOM::GEOM_Object_ptr theShapeObject )
1720 throw ( SALOME::SALOME_Exception )
1723 Unexpect aCatch(SALOME_SalomeException);
1724 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Compute" );
1726 if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
1727 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1728 SALOME::BAD_PARAM );
1730 if ( CORBA::is_nil( theMesh ) )
1731 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
1732 SALOME::BAD_PARAM );
1734 // Update Python script
1735 TPythonDump() << "isDone = " << this << ".Compute( "
1736 << theMesh << ", " << theShapeObject << ")";
1740 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1741 meshServant->Load();
1742 ASSERT( meshServant );
1743 if ( meshServant ) {
1744 // NPAL16168: "geometrical group edition from a submesh don't modifiy mesh computation"
1745 meshServant->CheckGeomModif();
1746 // get local TopoDS_Shape
1747 TopoDS_Shape myLocShape;
1748 if(theMesh->HasShapeToMesh())
1749 myLocShape = GeomObjectToShape( theShapeObject );
1751 myLocShape = SMESH_Mesh::PseudoShape();
1752 // call implementation compute
1753 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1754 myGen.PrepareCompute( myLocMesh, myLocShape);
1755 bool ok = myGen.Compute( myLocMesh, myLocShape);
1756 meshServant->CreateGroupServants(); // algos can create groups (issue 0020918)
1757 myLocMesh.GetMeshDS()->Modified();
1761 catch ( std::bad_alloc ) {
1762 INFOS( "Compute(): lack of memory" );
1764 catch ( SALOME_Exception& S_ex ) {
1765 INFOS( "Compute(): catch exception "<< S_ex.what() );
1768 INFOS( "Compute(): unknown exception " );
1773 //=============================================================================
1775 * SMESH_Gen_i::CancelCompute
1777 * Cancel Compute mesh on a shape
1779 //=============================================================================
1781 void SMESH_Gen_i::CancelCompute( SMESH::SMESH_Mesh_ptr theMesh,
1782 GEOM::GEOM_Object_ptr theShapeObject )
1784 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1785 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1786 TopoDS_Shape myLocShape;
1787 if(theMesh->HasShapeToMesh())
1788 myLocShape = GeomObjectToShape( theShapeObject );
1790 myLocShape = SMESH_Mesh::PseudoShape();
1791 myGen.CancelCompute( myLocMesh, myLocShape);
1794 //=============================================================================
1796 * SMESH_Gen_i::Precompute
1798 * Compute mesh as preview till indicated dimension on shape
1800 //=============================================================================
1802 SMESH::MeshPreviewStruct* SMESH_Gen_i::Precompute( SMESH::SMESH_Mesh_ptr theMesh,
1803 GEOM::GEOM_Object_ptr theShapeObject,
1804 SMESH::Dimension theDimension,
1805 SMESH::long_array& theShapesId)
1806 throw ( SALOME::SALOME_Exception )
1808 Unexpect aCatch(SALOME_SalomeException);
1809 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Precompute" );
1811 if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
1812 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1813 SALOME::BAD_PARAM );
1815 if ( CORBA::is_nil( theMesh ) )
1816 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
1817 SALOME::BAD_PARAM );
1819 SMESH::MeshPreviewStruct_var result = new SMESH::MeshPreviewStruct;
1822 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1823 meshServant->Load();
1824 ASSERT( meshServant );
1825 if ( meshServant ) {
1826 // NPAL16168: "geometrical group edition from a submesh don't modifiy mesh computation"
1827 meshServant->CheckGeomModif();
1828 // get local TopoDS_Shape
1829 TopoDS_Shape myLocShape;
1830 if(theMesh->HasShapeToMesh())
1831 myLocShape = GeomObjectToShape( theShapeObject );
1833 return result._retn();;
1835 // call implementation compute
1836 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1838 ::MeshDimension aDim = (MeshDimension)theDimension;
1839 if ( myGen.Compute( myLocMesh, myLocShape, false, false, aDim, &shapeIds ) )
1841 int nbShapeId = shapeIds.size();
1842 theShapesId.length( nbShapeId );
1843 // iterates on shapes and collect mesh entities into mesh preview
1844 TSetOfInt::const_iterator idIt = shapeIds.begin();
1845 TSetOfInt::const_iterator idEnd = shapeIds.end();
1846 std::map< int, int > mapOfShIdNb;
1847 std::set< SMESH_TLink > setOfEdge;
1848 std::list< SMDSAbs_ElementType > listOfElemType;
1849 typedef map<const SMDS_MeshElement*, int > TNode2LocalIDMap;
1850 typedef TNode2LocalIDMap::iterator TNodeLocalID;
1851 TNode2LocalIDMap mapNode2LocalID;
1852 list< TNodeLocalID > connectivity;
1853 int i, nbConnNodes = 0;
1854 std::set< const SMESH_subMesh* > setOfVSubMesh;
1855 // iterates on shapes
1856 for ( ; idIt != idEnd; idIt++ )
1858 if ( mapOfShIdNb.find( *idIt ) != mapOfShIdNb.end() )
1860 SMESH_subMesh* sm = myLocMesh.GetSubMeshContaining(*idIt);
1861 if ( !sm || !sm->IsMeshComputed() )
1864 const TopoDS_Shape& aSh = sm->GetSubShape();
1865 const int shDim = myGen.GetShapeDim( aSh );
1866 if ( shDim < 1 || shDim > theDimension )
1869 mapOfShIdNb[ *idIt ] = 0;
1870 theShapesId[ mapOfShIdNb.size() - 1 ] = *idIt;
1872 SMESHDS_SubMesh* smDS = sm->GetSubMeshDS();
1873 if ( !smDS ) continue;
1875 if ( theDimension == SMESH::DIM_2D )
1877 SMDS_ElemIteratorPtr faceIt = smDS->GetElements();
1878 while ( faceIt->more() )
1880 const SMDS_MeshElement* face = faceIt->next();
1881 int aNbNode = face->NbNodes();
1883 aNbNode /= 2; // do not take into account additional middle nodes
1885 SMDS_MeshNode* node1 = (SMDS_MeshNode*)face->GetNode( 0 );
1886 for ( int nIndx = 0; nIndx < aNbNode; nIndx++ )
1888 SMDS_MeshNode* node2 = (SMDS_MeshNode*)face->GetNode( nIndx+1 < aNbNode ? nIndx+1 : 0 );
1889 if ( setOfEdge.insert( SMESH_TLink ( node1, node2 ) ).second )
1891 listOfElemType.push_back( SMDSAbs_Edge );
1892 connectivity.push_back
1893 ( mapNode2LocalID.insert( make_pair( node1, ++nbConnNodes)).first );
1894 connectivity.push_back
1895 ( mapNode2LocalID.insert( make_pair( node2, ++nbConnNodes)).first );
1901 else if ( theDimension == SMESH::DIM_1D )
1903 SMDS_NodeIteratorPtr nodeIt = smDS->GetNodes();
1904 while ( nodeIt->more() )
1906 listOfElemType.push_back( SMDSAbs_Node );
1907 connectivity.push_back
1908 ( mapNode2LocalID.insert( make_pair( nodeIt->next(), ++nbConnNodes)).first );
1910 // add corner nodes by first vertex from edge
1911 SMESH_subMeshIteratorPtr edgeSmIt =
1912 sm->getDependsOnIterator(/*includeSelf*/false,
1913 /*complexShapeFirst*/false);
1914 while ( edgeSmIt->more() )
1916 SMESH_subMesh* vertexSM = edgeSmIt->next();
1917 // check that vertex is not already treated
1918 if ( !setOfVSubMesh.insert( vertexSM ).second )
1920 if ( vertexSM->GetSubShape().ShapeType() != TopAbs_VERTEX )
1923 const SMESHDS_SubMesh* vertexSmDS = vertexSM->GetSubMeshDS();
1924 SMDS_NodeIteratorPtr nodeIt = vertexSmDS->GetNodes();
1925 while ( nodeIt->more() )
1927 listOfElemType.push_back( SMDSAbs_Node );
1928 connectivity.push_back
1929 ( mapNode2LocalID.insert( make_pair( nodeIt->next(), ++nbConnNodes)).first );
1935 // fill node coords and assign local ids to the nodes
1936 int nbNodes = mapNode2LocalID.size();
1937 result->nodesXYZ.length( nbNodes );
1938 TNodeLocalID node2ID = mapNode2LocalID.begin();
1939 for ( i = 0; i < nbNodes; ++i, ++node2ID ) {
1940 node2ID->second = i;
1941 const SMDS_MeshNode* node = (const SMDS_MeshNode*) node2ID->first;
1942 result->nodesXYZ[i].x = node->X();
1943 result->nodesXYZ[i].y = node->Y();
1944 result->nodesXYZ[i].z = node->Z();
1946 // fill connectivity
1947 result->elementConnectivities.length( nbConnNodes );
1948 list< TNodeLocalID >::iterator connIt = connectivity.begin();
1949 for ( i = 0; i < nbConnNodes; ++i, ++connIt ) {
1950 result->elementConnectivities[i] = (*connIt)->second;
1953 // fill element types
1954 result->elementTypes.length( listOfElemType.size() );
1955 std::list< SMDSAbs_ElementType >::const_iterator typeIt = listOfElemType.begin();
1956 std::list< SMDSAbs_ElementType >::const_iterator typeEnd = listOfElemType.end();
1957 for ( i = 0; typeIt != typeEnd; ++i, ++typeIt )
1959 SMDSAbs_ElementType elemType = *typeIt;
1960 result->elementTypes[i].SMDS_ElementType = (SMESH::ElementType)elemType;
1961 result->elementTypes[i].isPoly = false;
1962 result->elementTypes[i].nbNodesInElement = elemType == SMDSAbs_Edge ? 2 : 1;
1965 // correct number of shapes
1966 theShapesId.length( mapOfShIdNb.size() );
1970 catch ( std::bad_alloc ) {
1971 INFOS( "Precompute(): lack of memory" );
1973 catch ( SALOME_Exception& S_ex ) {
1974 INFOS( "Precompute(): catch exception "<< S_ex.what() );
1977 INFOS( "Precompute(): unknown exception " );
1979 return result._retn();
1983 //=============================================================================
1985 * SMESH_Gen_i::Evaluate
1987 * Evaluate mesh on a shape
1989 //=============================================================================
1991 SMESH::long_array* SMESH_Gen_i::Evaluate(SMESH::SMESH_Mesh_ptr theMesh,
1992 GEOM::GEOM_Object_ptr theShapeObject)
1993 // SMESH::long_array& theNbElems)
1994 throw ( SALOME::SALOME_Exception )
1996 Unexpect aCatch(SALOME_SalomeException);
1997 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Evaluate" );
1999 if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
2000 THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
2001 SALOME::BAD_PARAM );
2003 if ( CORBA::is_nil( theMesh ) )
2004 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
2005 SALOME::BAD_PARAM );
2007 SMESH::long_array_var nbels = new SMESH::long_array;
2008 nbels->length(SMESH::Entity_Last);
2009 int i = SMESH::Entity_Node;
2010 for (; i < SMESH::Entity_Last; i++)
2013 // Update Python script
2014 TPythonDump() << "theNbElems = " << this << ".Evaluate( "
2015 << theMesh << ", " << theShapeObject << ")";
2019 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
2020 ASSERT( meshServant );
2021 if ( meshServant ) {
2022 // NPAL16168: "geometrical group edition from a submesh don't modifiy mesh computation"
2023 meshServant->CheckGeomModif();
2024 // get local TopoDS_Shape
2025 TopoDS_Shape myLocShape;
2026 if(theMesh->HasShapeToMesh())
2027 myLocShape = GeomObjectToShape( theShapeObject );
2029 myLocShape = SMESH_Mesh::PseudoShape();
2030 // call implementation compute
2031 ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
2032 MapShapeNbElems aResMap;
2033 /*CORBA::Boolean ret =*/ myGen.Evaluate( myLocMesh, myLocShape, aResMap);
2034 MapShapeNbElemsItr anIt = aResMap.begin();
2035 for(; anIt!=aResMap.end(); anIt++) {
2036 const vector<int>& aVec = (*anIt).second;
2037 for(i = SMESH::Entity_Node; i < aVec.size(); i++) {
2038 int nbElem = aVec[i];
2039 if ( nbElem < 0 ) // algo failed, check that it has reported a message
2041 SMESH_subMesh* sm = anIt->first;
2042 SMESH_ComputeErrorPtr& error = sm->GetComputeError();
2043 const SMESH_Algo* algo = myGen.GetAlgo( myLocMesh, sm->GetSubShape());
2044 if ( (algo && !error.get()) || error->IsOK() )
2045 error.reset( new SMESH_ComputeError( COMPERR_ALGO_FAILED,"Failed to evaluate",algo));
2049 nbels[i] += aVec[i];
2053 return nbels._retn();
2056 catch ( std::bad_alloc ) {
2057 INFOS( "Evaluate(): lack of memory" );
2059 catch ( SALOME_Exception& S_ex ) {
2060 INFOS( "Evaluate(): catch exception "<< S_ex.what() );
2063 INFOS( "Evaluate(): unknown exception " );
2066 return nbels._retn();
2069 //================================================================================
2071 * \brief Return geometrical object the given element is built on
2072 * \param theMesh - the mesh the element is in
2073 * \param theElementID - the element ID
2074 * \param theGeomName - the name of the result geom object if it is not yet published
2075 * \retval GEOM::GEOM_Object_ptr - the found or just published geom object (no need to UnRegister())
2077 //================================================================================
2079 GEOM::GEOM_Object_ptr
2080 SMESH_Gen_i::GetGeometryByMeshElement( SMESH::SMESH_Mesh_ptr theMesh,
2081 CORBA::Long theElementID,
2082 const char* theGeomName)
2083 throw ( SALOME::SALOME_Exception )
2085 Unexpect aCatch(SALOME_SalomeException);
2087 GEOM::GEOM_Object_wrap geom = FindGeometryByMeshElement(theMesh, theElementID);
2088 if ( !geom->_is_nil() ) {
2089 GEOM::GEOM_Object_var mainShape = theMesh->GetShapeToMesh();
2090 GEOM::GEOM_Gen_ptr geomGen = GetGeomEngine();
2092 // try to find the corresponding SObject
2093 SALOMEDS::SObject_wrap SObj = ObjectToSObject( myCurrentStudy, geom.in() );
2094 if ( SObj->_is_nil() ) // submesh can be not found even if published
2096 // try to find published submesh
2097 GEOM::ListOfLong_var list = geom->GetSubShapeIndices();
2098 if ( !geom->IsMainShape() && list->length() == 1 ) {
2099 SALOMEDS::SObject_wrap mainSO = ObjectToSObject( myCurrentStudy, mainShape );
2100 SALOMEDS::ChildIterator_wrap it;
2101 if ( !mainSO->_is_nil() ) {
2102 it = myCurrentStudy->NewChildIterator( mainSO );
2104 if ( !it->_is_nil() ) {
2105 for ( it->InitEx(true); it->More(); it->Next() ) {
2106 SALOMEDS::SObject_wrap so = it->Value();
2107 CORBA::Object_var obj = SObjectToObject( so );
2108 GEOM::GEOM_Object_var subGeom = GEOM::GEOM_Object::_narrow( obj );
2109 if ( !subGeom->_is_nil() ) {
2110 GEOM::ListOfLong_var subList = subGeom->GetSubShapeIndices();
2111 if ( subList->length() == 1 && list[0] == subList[0] ) {
2121 if ( SObj->_is_nil() ) // publish a new subshape
2122 SObj = geomGen->AddInStudy( myCurrentStudy, geom, theGeomName, mainShape );
2124 // return only published geometry
2125 if ( !SObj->_is_nil() ) {
2126 //return geom._retn(); -- servant of geom must be UnRegister()ed;
2127 CORBA::Object_var obj = SObjectToObject( SObj );
2128 GEOM::GEOM_Object_var go = GEOM::GEOM_Object::_narrow( obj );
2132 return GEOM::GEOM_Object::_nil();
2135 //================================================================================
2137 * \brief Return geometrical object the given element is built on.
2138 * \param theMesh - the mesh the element is in
2139 * \param theElementID - the element ID
2140 * \retval GEOM::GEOM_Object_ptr - the found or created (UnRegister()!) geom object
2142 //================================================================================
2144 GEOM::GEOM_Object_ptr
2145 SMESH_Gen_i::FindGeometryByMeshElement( SMESH::SMESH_Mesh_ptr theMesh,
2146 CORBA::Long theElementID)
2147 throw ( SALOME::SALOME_Exception )
2149 Unexpect aCatch(SALOME_SalomeException);
2150 if ( CORBA::is_nil( theMesh ) )
2151 THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference", SALOME::BAD_PARAM );
2153 GEOM::GEOM_Object_var mainShape = theMesh->GetShapeToMesh();
2154 GEOM::GEOM_Gen_ptr geomGen = GetGeomEngine();
2156 // get a core mesh DS
2157 SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
2158 if ( meshServant && !geomGen->_is_nil() && !mainShape->_is_nil() )
2160 ::SMESH_Mesh & mesh = meshServant->GetImpl();
2161 SMESHDS_Mesh* meshDS = mesh.GetMeshDS();
2162 // find the element in mesh
2163 if ( const SMDS_MeshElement * elem = meshDS->FindElement( theElementID ) ) {
2164 // find a shape id by the element
2165 if ( int shapeID = ::SMESH_MeshEditor( &mesh ).FindShape( elem )) {
2166 // get a geom object by the shape id
2167 GEOM::GEOM_Object_var geom = ShapeToGeomObject( meshDS->IndexToShape( shapeID ));
2168 if ( geom->_is_nil() ) {
2169 // try to find a published sub-shape
2170 SALOMEDS::SObject_wrap mainSO = ObjectToSObject( myCurrentStudy, mainShape );
2171 SALOMEDS::ChildIterator_wrap it;
2172 if ( !mainSO->_is_nil() ) {
2173 it = myCurrentStudy->NewChildIterator( mainSO );
2175 if ( !it->_is_nil() ) {
2176 for ( it->InitEx(true); it->More(); it->Next() ) {
2177 SALOMEDS::SObject_wrap so = it->Value();
2178 CORBA::Object_var obj = SObjectToObject( so );
2179 GEOM::GEOM_Object_var subGeom = GEOM::GEOM_Object::_narrow( obj );
2180 if ( !subGeom->_is_nil() ) {
2181 GEOM::ListOfLong_var subList = subGeom->GetSubShapeIndices();
2182 if ( subList->length() == 1 && shapeID == subList[0] ) {
2190 if ( geom->_is_nil() ) {
2192 GEOM::GEOM_IShapesOperations_wrap op =
2193 geomGen->GetIShapesOperations( GetCurrentStudyID() );
2194 if ( !op->_is_nil() )
2195 geom = op->GetSubShape( mainShape, shapeID );
2200 if ( !geom->_is_nil() ) {
2201 GeomObjectToShape( geom ); // let geom client remember the found shape
2202 return geom._retn();
2207 return GEOM::GEOM_Object::_nil();
2210 //================================================================================
2212 * SMESH_Gen_i::Concatenate
2214 * Concatenate the given meshes into one mesh
2216 //================================================================================
2218 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::Concatenate(const SMESH::mesh_array& theMeshesArray,
2219 CORBA::Boolean theUniteIdenticalGroups,
2220 CORBA::Boolean theMergeNodesAndElements,
2221 CORBA::Double theMergeTolerance)
2222 throw ( SALOME::SALOME_Exception )
2224 return ConcatenateCommon(theMeshesArray,
2225 theUniteIdenticalGroups,
2226 theMergeNodesAndElements,
2231 //================================================================================
2233 * SMESH_Gen_i::ConcatenateWithGroups
2235 * Concatenate the given meshes into one mesh
2236 * Create the groups of all elements from initial meshes
2238 //================================================================================
2240 SMESH::SMESH_Mesh_ptr
2241 SMESH_Gen_i::ConcatenateWithGroups(const SMESH::mesh_array& theMeshesArray,
2242 CORBA::Boolean theUniteIdenticalGroups,
2243 CORBA::Boolean theMergeNodesAndElements,
2244 CORBA::Double theMergeTolerance)
2245 throw ( SALOME::SALOME_Exception )
2247 return ConcatenateCommon(theMeshesArray,
2248 theUniteIdenticalGroups,
2249 theMergeNodesAndElements,
2254 //================================================================================
2256 * SMESH_Gen_i::ConcatenateCommon
2258 * Concatenate the given meshes into one mesh
2260 //================================================================================
2262 SMESH::SMESH_Mesh_ptr
2263 SMESH_Gen_i::ConcatenateCommon(const SMESH::mesh_array& theMeshesArray,
2264 CORBA::Boolean theUniteIdenticalGroups,
2265 CORBA::Boolean theMergeNodesAndElements,
2266 CORBA::Double theMergeTolerance,
2267 CORBA::Boolean theCommonGroups)
2268 throw ( SALOME::SALOME_Exception )
2270 typedef map<int, int> TIDsMap;
2271 typedef list<SMESH::SMESH_Group_var> TListOfNewGroups;
2272 typedef map< pair<string, SMESH::ElementType>, TListOfNewGroups > TGroupsMap;
2273 typedef std::set<SMESHDS_GroupBase*> TGroups;
2275 TPythonDump* pPythonDump = new TPythonDump;
2276 TPythonDump& aPythonDump = *pPythonDump; // prevent dump of called methods
2279 SMESH::SMESH_Mesh_var aNewMesh = CreateEmptyMesh();
2281 SMESHDS_Mesh* aNewMeshDS = 0;
2282 if ( !aNewMesh->_is_nil() ) {
2283 SMESH_Mesh_i* aNewImpl = dynamic_cast<SMESH_Mesh_i*>( GetServant( aNewMesh ).in() );
2285 ::SMESH_Mesh& aLocMesh = aNewImpl->GetImpl();
2286 aNewMeshDS = aLocMesh.GetMeshDS();
2288 TGroupsMap aGroupsMap;
2289 TListOfNewGroups aListOfNewGroups;
2290 SMESH_MeshEditor aNewEditor = ::SMESH_MeshEditor(&aLocMesh);
2291 SMESH::ListOfGroups_var aListOfGroups = new SMESH::ListOfGroups();
2294 for ( int i = 0; i < theMeshesArray.length(); i++) {
2295 SMESH::SMESH_Mesh_var anInitMesh = theMeshesArray[i];
2296 if ( !anInitMesh->_is_nil() ) {
2297 SMESH_Mesh_i* anInitImpl = dynamic_cast<SMESH_Mesh_i*>( GetServant( anInitMesh ).in() );
2299 ::SMESH_Mesh& aInitLocMesh = anInitImpl->GetImpl();
2300 aInitLocMesh.Load();
2301 SMESHDS_Mesh* anInitMeshDS = aInitLocMesh.GetMeshDS();
2306 // loop on elements of mesh
2307 SMDS_ElemIteratorPtr itElems = anInitMeshDS->elementsIterator();
2308 const SMDS_MeshElement* anElem = 0;
2309 const SMDS_MeshElement* aNewElem = 0;
2310 int anElemNbNodes = 0;
2315 int anNbVolumes = 0;
2318 SMESH::long_array_var anIDsNodes = new SMESH::long_array();
2319 SMESH::long_array_var anIDsEdges = new SMESH::long_array();
2320 SMESH::long_array_var anIDsFaces = new SMESH::long_array();
2321 SMESH::long_array_var anIDsVolumes = new SMESH::long_array();
2322 SMESH::long_array_var anIDsBalls = new SMESH::long_array();
2324 if( theCommonGroups ) {
2325 anIDsNodes->length( anInitMeshDS->NbNodes() );
2326 anIDsEdges->length( anInitMeshDS->NbEdges() );
2327 anIDsFaces->length( anInitMeshDS->NbFaces() );
2328 anIDsVolumes->length( anInitMeshDS->NbVolumes() );
2329 anIDsBalls->length( anInitMeshDS->NbBalls() );
2332 for ( int j = 0; itElems->more(); j++) {
2333 anElem = itElems->next();
2334 SMDSAbs_ElementType anElemType = anElem->GetType();
2335 anElemNbNodes = anElem->NbNodes();
2336 std::vector<const SMDS_MeshNode*> aNodesArray (anElemNbNodes);
2338 // loop on nodes of element
2339 const SMDS_MeshNode* aNode = 0;
2340 const SMDS_MeshNode* aNewNode = 0;
2341 SMDS_ElemIteratorPtr itNodes = anElem->nodesIterator();
2343 for ( int k = 0; itNodes->more(); k++) {
2344 aNode = static_cast<const SMDS_MeshNode*>(itNodes->next());
2345 if ( nodesMap.find(aNode->GetID()) == nodesMap.end() ) {
2346 aNewNode = aNewMeshDS->AddNode(aNode->X(), aNode->Y(), aNode->Z());
2347 nodesMap.insert( make_pair(aNode->GetID(), aNewNode->GetID()) );
2348 if( theCommonGroups )
2349 anIDsNodes[anNbNodes++] = aNewNode->GetID();
2352 aNewNode = aNewMeshDS->FindNode( nodesMap.find(aNode->GetID())->second );
2353 aNodesArray[k] = aNewNode;
2356 // creates a corresponding element on existent nodes in new mesh
2357 switch ( anElem->GetEntityType() ) {
2358 case SMDSEntity_Polyhedra:
2359 if ( const SMDS_VtkVolume* aVolume =
2360 dynamic_cast<const SMDS_VtkVolume*> (anElem))
2362 aNewElem = aNewMeshDS->AddPolyhedralVolume(aNodesArray,
2363 aVolume->GetQuantities());
2364 elemsMap.insert(make_pair(anElem->GetID(), aNewElem->GetID()));
2365 if( theCommonGroups )
2366 anIDsVolumes[anNbVolumes++] = aNewElem->GetID();
2369 case SMDSEntity_Ball:
2370 if ( const SMDS_BallElement* aBall =
2371 dynamic_cast<const SMDS_BallElement*> (anElem))
2373 aNewElem = aNewEditor.AddElement(aNodesArray, SMDSAbs_Ball,
2374 /*isPoly=*/false, /*id=*/0,
2375 aBall->GetDiameter() );
2376 elemsMap.insert(make_pair(anElem->GetID(), aNewElem->GetID()));
2377 if( theCommonGroups )
2378 anIDsBalls[aNbBalls++] = aNewElem->GetID();
2383 aNewElem = aNewEditor.AddElement(aNodesArray,
2386 elemsMap.insert(make_pair(anElem->GetID(), aNewElem->GetID()));
2387 if( theCommonGroups ) {
2388 if( anElemType == SMDSAbs_Edge )
2389 anIDsEdges[anNbEdges++] = aNewElem->GetID();
2390 else if( anElemType == SMDSAbs_Face )
2391 anIDsFaces[anNbFaces++] = aNewElem->GetID();
2392 else if( anElemType == SMDSAbs_Volume )
2393 anIDsVolumes[anNbVolumes++] = aNewElem->GetID();
2399 // copy orphan nodes
2400 SMDS_NodeIteratorPtr itNodes = anInitMeshDS->nodesIterator();
2401 while ( itNodes->more() )
2403 const SMDS_MeshNode* aNode = itNodes->next();
2404 if ( aNode->NbInverseElements() == 0 )
2406 const SMDS_MeshNode* aNewNode =
2407 aNewMeshDS->AddNode(aNode->X(), aNode->Y(), aNode->Z());
2408 nodesMap.insert( make_pair(aNode->GetID(), aNewNode->GetID()) );
2409 if( theCommonGroups )
2410 anIDsNodes[anNbNodes++] = aNewNode->GetID();
2415 aListOfGroups = anInitImpl->GetGroups();
2416 SMESH::SMESH_GroupBase_ptr aGroup;
2418 // loop on groups of mesh
2419 SMESH::long_array_var anInitIDs = new SMESH::long_array();
2420 SMESH::long_array_var anNewIDs = new SMESH::long_array();
2421 SMESH::SMESH_Group_var aNewGroup;
2423 SMESH::ElementType aGroupType;
2424 CORBA::String_var aGroupName;
2425 if ( theCommonGroups ) {
2426 for(aGroupType=SMESH::NODE;aGroupType<=SMESH::BALL;aGroupType=(SMESH::ElementType)(aGroupType+1)) {
2428 SALOMEDS::SObject_wrap aMeshSObj = ObjectToSObject( myCurrentStudy, anInitMesh );
2430 str += aMeshSObj->GetName();
2435 switch(aGroupType) {
2438 anIDsNodes->length(anNbNodes);
2443 anIDsEdges->length(anNbEdges);
2448 anIDsFaces->length(anNbFaces);
2453 anIDsVolumes->length(anNbVolumes);
2454 anLen = anNbVolumes;
2458 anIDsBalls->length(aNbBalls);
2466 aGroupName = str.c_str();
2468 // add a new group in the mesh
2469 aNewGroup = aNewImpl->CreateGroup(aGroupType, aGroupName);
2471 switch(aGroupType) {
2473 aNewGroup->Add( anIDsNodes );
2476 aNewGroup->Add( anIDsEdges );
2479 aNewGroup->Add( anIDsFaces );
2482 aNewGroup->Add( anIDsVolumes );
2485 aNewGroup->Add( anIDsBalls );
2491 aListOfNewGroups.clear();
2492 aListOfNewGroups.push_back(aNewGroup);
2493 aGroupsMap.insert(make_pair( make_pair(aGroupName, aGroupType), aListOfNewGroups ));
2498 // check that current group name and type don't have identical ones in union mesh
2499 for (int iG = 0; iG < aListOfGroups->length(); iG++) {
2500 aGroup = aListOfGroups[iG];
2501 aListOfNewGroups.clear();
2502 aGroupType = aGroup->GetType();
2503 aGroupName = aGroup->GetName();
2505 TGroupsMap::iterator anIter = aGroupsMap.find(make_pair(aGroupName, aGroupType));
2507 // convert a list of IDs
2508 anInitIDs = aGroup->GetListOfID();
2509 anNewIDs->length(anInitIDs->length());
2510 if ( aGroupType == SMESH::NODE )
2511 for (int j = 0; j < anInitIDs->length(); j++) {
2512 anNewIDs[j] = nodesMap.find(anInitIDs[j])->second;
2515 for (int j = 0; j < anInitIDs->length(); j++) {
2516 anNewIDs[j] = elemsMap.find(anInitIDs[j])->second;
2519 // check that current group name and type don't have identical ones in union mesh
2520 if ( anIter == aGroupsMap.end() ) {
2521 // add a new group in the mesh
2522 aNewGroup = aNewImpl->CreateGroup(aGroupType, aGroupName);
2523 // add elements into new group
2524 aNewGroup->Add( anNewIDs );
2526 aListOfNewGroups.push_back(aNewGroup);
2527 aGroupsMap.insert(make_pair( make_pair(aGroupName, aGroupType), aListOfNewGroups ));
2530 else if ( theUniteIdenticalGroups ) {
2531 // unite identical groups
2532 TListOfNewGroups& aNewGroups = anIter->second;
2533 aNewGroups.front()->Add( anNewIDs );
2537 // rename identical groups
2538 aNewGroup = aNewImpl->CreateGroup(aGroupType, aGroupName);
2539 aNewGroup->Add( anNewIDs );
2541 TListOfNewGroups& aNewGroups = anIter->second;
2542 string aNewGroupName;
2543 if (aNewGroups.size() == 1) {
2544 aNewGroupName = string(aGroupName) + "_1";
2545 aNewGroups.front()->SetName(aNewGroupName.c_str());
2547 char aGroupNum[128];
2548 sprintf(aGroupNum, "%u", aNewGroups.size()+1);
2549 aNewGroupName = string(aGroupName) + "_" + string(aGroupNum);
2550 aNewGroup->SetName(aNewGroupName.c_str());
2551 aNewGroups.push_back(aNewGroup);
2558 if (theMergeNodesAndElements) {
2560 TIDSortedNodeSet aMeshNodes; // no input nodes
2561 SMESH_MeshEditor::TListOfListOfNodes aGroupsOfNodes;
2562 aNewEditor.FindCoincidentNodes( aMeshNodes, theMergeTolerance, aGroupsOfNodes );
2563 aNewEditor.MergeNodes( aGroupsOfNodes );
2565 aNewEditor.MergeEqualElements();
2570 // Update Python script
2571 aPythonDump << aNewMesh << " = " << this;
2572 if( !theCommonGroups )
2573 aPythonDump << ".Concatenate(";
2575 aPythonDump << ".ConcatenateWithGroups(";
2577 for ( int i = 0; i < theMeshesArray.length(); i++) {
2578 if (i > 0) aPythonDump << ", ";
2579 aPythonDump << theMeshesArray[i];
2581 aPythonDump << "], ";
2582 aPythonDump << theUniteIdenticalGroups << ", "
2583 << theMergeNodesAndElements << ", "
2584 << TVar( theMergeTolerance ) << ")";
2586 delete pPythonDump; // enable python dump from GetGroups()
2588 // 0020577: EDF 1164 SMESH: Bad dump of concatenate with create common groups
2589 if ( !aNewMesh->_is_nil() )
2591 SMESH::ListOfGroups_var groups = aNewMesh->GetGroups();
2594 // IPAL21468 Change icon of compound because it need not be computed.
2595 SALOMEDS::SObject_wrap aMeshSObj = ObjectToSObject( myCurrentStudy, aNewMesh );
2596 SetPixMap( aMeshSObj, "ICON_SMESH_TREE_MESH" );
2599 aNewMeshDS->Modified();
2600 return aNewMesh._retn();
2603 //================================================================================
2605 * \brief Create a mesh by copying a part of another mesh
2606 * \param meshPart - a part of mesh to copy
2607 * \param toCopyGroups - to create in the new mesh groups
2608 * the copied elements belongs to
2609 * \param toKeepIDs - to preserve IDs of the copied elements or not
2610 * \retval SMESH::SMESH_Mesh_ptr - the new mesh
2612 //================================================================================
2614 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CopyMesh(SMESH::SMESH_IDSource_ptr meshPart,
2615 const char* meshName,
2616 CORBA::Boolean toCopyGroups,
2617 CORBA::Boolean toKeepIDs)
2619 Unexpect aCatch(SALOME_SalomeException);
2621 TPythonDump* pyDump = new TPythonDump; // prevent dump from CreateMesh()
2623 // 1. Get source mesh
2625 if ( CORBA::is_nil( meshPart ))
2626 THROW_SALOME_CORBA_EXCEPTION( "bad IDSource", SALOME::BAD_PARAM );
2628 SMESH::SMESH_Mesh_var srcMesh = meshPart->GetMesh();
2629 SMESH_Mesh_i* srcMesh_i = SMESH::DownCast<SMESH_Mesh_i*>( srcMesh );
2631 THROW_SALOME_CORBA_EXCEPTION( "bad mesh of IDSource", SALOME::BAD_PARAM );
2633 SMESHDS_Mesh* srcMeshDS = srcMesh_i->GetImpl().GetMeshDS();
2635 // 2. Make a new mesh
2637 SMESH::SMESH_Mesh_var newMesh = CreateMesh(GEOM::GEOM_Object::_nil());
2638 SMESH_Mesh_i* newMesh_i = SMESH::DownCast<SMESH_Mesh_i*>( newMesh );
2640 THROW_SALOME_CORBA_EXCEPTION( "can't create a mesh", SALOME::INTERNAL_ERROR );
2641 SALOMEDS::SObject_wrap meshSO = ObjectToSObject(myCurrentStudy, newMesh );
2642 if ( !meshSO->_is_nil() )
2644 SetName( meshSO, meshName, "Mesh" );
2645 SetPixMap( meshSO, "ICON_SMESH_TREE_MESH_IMPORTED");
2647 SMESHDS_Mesh* newMeshDS = newMesh_i->GetImpl().GetMeshDS();
2648 ::SMESH_MeshEditor editor( &newMesh_i->GetImpl() );
2650 // 3. Get elements to copy
2652 SMDS_ElemIteratorPtr srcElemIt; SMDS_NodeIteratorPtr srcNodeIt;
2653 TIDSortedElemSet srcElems;
2654 SMESH::array_of_ElementType_var srcElemTypes = meshPart->GetTypes();
2655 if ( SMESH::DownCast<SMESH_Mesh_i*>( meshPart ))
2658 srcElemIt = srcMeshDS->elementsIterator();
2659 srcNodeIt = srcMeshDS->nodesIterator();
2663 SMESH::long_array_var ids = meshPart->GetIDs();
2664 if ( srcElemTypes->length() == 1 && srcElemTypes[0] == SMESH::NODE ) // group of nodes
2666 for (int i=0; i < ids->length(); i++)
2667 if ( const SMDS_MeshElement * elem = srcMeshDS->FindNode( ids[i] ))
2668 srcElems.insert( elem );
2672 for (int i=0; i < ids->length(); i++)
2673 if ( const SMDS_MeshElement * elem = srcMeshDS->FindElement( ids[i] ))
2674 srcElems.insert( elem );
2676 if ( srcElems.empty() )
2677 return newMesh._retn();
2679 typedef SMDS_SetIterator< SMDS_pElement, TIDSortedElemSet::const_iterator > ElIter;
2680 srcElemIt = SMDS_ElemIteratorPtr( new ElIter( srcElems.begin(), srcElems.end() ));
2685 typedef map<SMDS_pElement, SMDS_pElement, TIDCompare> TE2EMap;
2686 TE2EMap e2eMapByType[ SMDSAbs_NbElementTypes ];
2687 TE2EMap& n2nMap = e2eMapByType[ SMDSAbs_Node ];
2689 const SMDS_MeshNode *nSrc, *nTgt;
2690 vector< const SMDS_MeshNode* > nodes;
2691 while ( srcElemIt->more() )
2693 const SMDS_MeshElement * elem = srcElemIt->next();
2695 nodes.resize( elem->NbNodes());
2696 SMDS_ElemIteratorPtr nIt = elem->nodesIterator();
2698 for ( iN = 0; nIt->more(); ++iN )
2700 nSrc = static_cast<const SMDS_MeshNode*>( nIt->next() );
2701 nTgt = newMeshDS->FindNode( nSrc->GetID());
2703 nTgt = newMeshDS->AddNodeWithID( nSrc->X(), nSrc->Y(), nSrc->Z(), nSrc->GetID());
2708 for ( iN = 0; nIt->more(); ++iN )
2710 nSrc = static_cast<const SMDS_MeshNode*>( nIt->next() );
2711 TE2EMap::iterator n2n = n2nMap.insert( make_pair( nSrc, SMDS_pNode(0) )).first;
2713 n2n->second = newMeshDS->AddNode( nSrc->X(), nSrc->Y(), nSrc->Z() );
2714 nodes[ iN ] = (const SMDS_MeshNode*) n2n->second;
2718 if ( elem->GetType() != SMDSAbs_Node )
2720 int ID = toKeepIDs ? elem->GetID() : 0;
2721 const SMDS_MeshElement * newElem;
2722 switch ( elem->GetEntityType() ) {
2723 case SMDSEntity_Polyhedra:
2725 newElem = editor.GetMeshDS()->
2726 AddPolyhedralVolumeWithID( nodes,
2727 static_cast<const SMDS_VtkVolume*>(elem)->GetQuantities(),
2730 newElem = editor.GetMeshDS()->
2731 AddPolyhedralVolume( nodes,
2732 static_cast<const SMDS_VtkVolume*>(elem)->GetQuantities());
2734 case SMDSEntity_Ball:
2735 newElem = editor.AddElement( nodes, SMDSAbs_Ball, false, ID,
2736 static_cast<const SMDS_BallElement*>(elem)->GetDiameter());
2739 newElem = editor.AddElement( nodes,elem->GetType(),elem->IsPoly(),ID);
2741 if ( toCopyGroups && !toKeepIDs )
2742 e2eMapByType[ elem->GetType() ].insert( make_pair( elem, newElem ));
2745 } // while ( srcElemIt->more() )
2747 // 4(b). Copy free nodes
2749 if ( srcNodeIt && srcMeshDS->NbNodes() != newMeshDS->NbNodes() )
2751 while ( srcNodeIt->more() )
2753 nSrc = srcNodeIt->next();
2754 if ( nSrc->NbInverseElements() == 0 )
2757 nTgt = newMeshDS->AddNodeWithID( nSrc->X(), nSrc->Y(), nSrc->Z(), nSrc->GetID());
2759 n2nMap[ nSrc ] = newMeshDS->AddNode( nSrc->X(), nSrc->Y(), nSrc->Z() );
2766 int nbNewGroups = 0;
2769 SMESH_Mesh::GroupIteratorPtr gIt = srcMesh_i->GetImpl().GetGroups();
2770 while ( gIt->more() )
2772 SMESH_Group* group = gIt->next();
2773 const SMESHDS_GroupBase* groupDS = group->GetGroupDS();
2775 // Check group type. We copy nodal groups containing nodes of copied element
2776 SMDSAbs_ElementType groupType = groupDS->GetType();
2777 if ( groupType != SMDSAbs_Node &&
2778 newMeshDS->GetMeshInfo().NbElements( groupType ) == 0 )
2779 continue; // group type differs from types of meshPart
2781 // Find copied elements in the group
2782 vector< const SMDS_MeshElement* > groupElems;
2783 SMDS_ElemIteratorPtr eIt = groupDS->GetElements();
2786 const SMDS_MeshElement* foundElem;
2787 if ( groupType == SMDSAbs_Node )
2789 while ( eIt->more() )
2790 if (( foundElem = newMeshDS->FindNode( eIt->next()->GetID() )))
2791 groupElems.push_back( foundElem );
2795 while ( eIt->more() )
2796 if (( foundElem = newMeshDS->FindElement( eIt->next()->GetID() )))
2797 groupElems.push_back( foundElem );
2802 TE2EMap & e2eMap = e2eMapByType[ groupDS->GetType() ];
2803 if ( e2eMap.empty() ) continue;
2804 int minID = e2eMap.begin()->first->GetID();
2805 int maxID = e2eMap.rbegin()->first->GetID();
2806 TE2EMap::iterator e2e;
2807 while ( eIt->more() && groupElems.size() < e2eMap.size())
2809 const SMDS_MeshElement* e = eIt->next();
2810 if ( e->GetID() < minID || e->GetID() > maxID ) continue;
2811 if ((e2e = e2eMap.find( e )) != e2eMap.end())
2812 groupElems.push_back( e2e->second );
2816 if ( !groupElems.empty() )
2818 SMESH::SMESH_Group_var newGroupObj =
2819 newMesh->CreateGroup( SMESH::ElementType(groupType), group->GetName() );
2820 if ( SMESH_GroupBase_i* newGroup_i = SMESH::DownCast<SMESH_GroupBase_i*>( newGroupObj))
2822 SMESHDS_GroupBase * newGroupDS = newGroup_i->GetGroupDS();
2823 SMDS_MeshGroup& smdsGroup = ((SMESHDS_Group*)newGroupDS)->SMDSGroup();
2824 for ( unsigned i = 0; i < groupElems.size(); ++i )
2825 smdsGroup.Add( groupElems[i] );
2833 newMeshDS->Modified();
2835 *pyDump << newMesh << " = " << this
2836 << ".CopyMesh( " << meshPart << ", "
2837 << "'" << meshName << "', "
2838 << toCopyGroups << ", "
2839 << toKeepIDs << ")";
2841 delete pyDump; pyDump = 0; // allow dump in GetGroups()
2843 if ( nbNewGroups > 0 ) // dump created groups
2844 SMESH::ListOfGroups_var groups = newMesh->GetGroups();
2846 return newMesh._retn();
2849 //================================================================================
2851 * SMESH_Gen_i::GetMEDVersion
2853 * Get MED version of the file by its name
2855 //================================================================================
2856 CORBA::Boolean SMESH_Gen_i::GetMEDVersion(const char* theFileName,
2857 SMESH::MED_VERSION& theVersion)
2859 theVersion = SMESH::MED_V2_1;
2860 MED::EVersion aVersion = MED::GetVersionId( theFileName );
2861 switch( aVersion ) {
2862 case MED::eV2_1 : theVersion = SMESH::MED_V2_1; return true;
2863 case MED::eV2_2 : theVersion = SMESH::MED_V2_2; return true;
2864 case MED::eVUnknown : return false;
2869 //================================================================================
2871 * SMESH_Gen_i::GetMeshNames
2873 * Get names of meshes defined in file with the specified name
2875 //================================================================================
2876 SMESH::string_array* SMESH_Gen_i::GetMeshNames(const char* theFileName)
2878 SMESH::string_array_var aResult = new SMESH::string_array();
2879 MED::PWrapper aMed = MED::CrWrapper( theFileName );
2881 MED::TInt aNbMeshes = aMed->GetNbMeshes( &anErr );
2883 aResult->length( aNbMeshes );
2884 for( MED::TInt i = 0; i < aNbMeshes; i++ ) {
2885 MED::PMeshInfo aMeshInfo = aMed->GetPMeshInfo( i+1 );
2886 aResult[i] = CORBA::string_dup( aMeshInfo->GetName().c_str() );
2889 return aResult._retn();
2892 //=============================================================================
2896 * Save SMESH module's data
2898 //=============================================================================
2899 SALOMEDS::TMPFile* SMESH_Gen_i::Save( SALOMEDS::SComponent_ptr theComponent,
2903 INFOS( "SMESH_Gen_i::Save" );
2905 // ASSERT( theComponent->GetStudy()->StudyId() == myCurrentStudy->StudyId() )
2906 // san -- in case <myCurrentStudy> differs from theComponent's study,
2907 // use that of the component
2908 if ( myCurrentStudy->_is_nil() ||
2909 theComponent->GetStudy()->StudyId() != myCurrentStudy->StudyId() )
2910 SetCurrentStudy( theComponent->GetStudy() );
2912 // Store study contents as a set of python commands
2913 SavePython(myCurrentStudy);
2915 StudyContext* myStudyContext = GetCurrentStudyContext();
2917 // Declare a byte stream
2918 SALOMEDS::TMPFile_var aStreamFile;
2920 // Obtain a temporary dir
2921 TCollection_AsciiString tmpDir =
2922 ( isMultiFile ) ? TCollection_AsciiString( ( char* )theURL ) : ( char* )SALOMEDS_Tool::GetTmpDir().c_str();
2924 // Create a sequence of files processed
2925 SALOMEDS::ListOfFileNames_var aFileSeq = new SALOMEDS::ListOfFileNames;
2926 aFileSeq->length( NUM_TMP_FILES );
2928 TCollection_AsciiString aStudyName( "" );
2930 aStudyName = ( (char*)SALOMEDS_Tool::GetNameFromPath( myCurrentStudy->URL() ).c_str() );
2932 // Set names of temporary files
2933 TCollection_AsciiString filename =
2934 aStudyName + TCollection_AsciiString( "_SMESH.hdf" ); // for SMESH data itself
2935 TCollection_AsciiString meshfile =
2936 aStudyName + TCollection_AsciiString( "_SMESH_Mesh.med" ); // for mesh data to be stored in MED file
2937 aFileSeq[ 0 ] = CORBA::string_dup( filename.ToCString() );
2938 aFileSeq[ 1 ] = CORBA::string_dup( meshfile.ToCString() );
2939 filename = tmpDir + filename;
2940 meshfile = tmpDir + meshfile;
2943 HDFdataset* aDataset;
2944 HDFgroup* aTopGroup;
2946 HDFgroup* aSubGroup;
2947 HDFgroup* aSubSubGroup;
2948 hdf_size aSize[ 1 ];
2951 //Remove the files if they exist: BugID: 11225
2952 #ifndef WIN32 /* unix functionality */
2953 TCollection_AsciiString cmd("rm -f \"");
2955 TCollection_AsciiString cmd("del /F \"");
2962 system(cmd.ToCString());
2964 // MED writer to be used by storage process
2965 DriverMED_W_SMESHDS_Mesh myWriter;
2966 myWriter.SetFile( meshfile.ToCString() );
2969 // SetStoreName() to groups before storing hypotheses to let them refer to
2970 // groups using "store name", which is "Group <group_persistent_id>"
2972 SALOMEDS::ChildIterator_wrap itBig = myCurrentStudy->NewChildIterator( theComponent );
2973 for ( ; itBig->More(); itBig->Next() ) {
2974 SALOMEDS::SObject_wrap gotBranch = itBig->Value();
2975 if ( gotBranch->Tag() > GetAlgorithmsRootTag() ) {
2976 CORBA::Object_var anObject = SObjectToObject( gotBranch );
2977 if ( !CORBA::is_nil( anObject ) ) {
2978 SMESH::SMESH_Mesh_var myMesh = SMESH::SMESH_Mesh::_narrow( anObject ) ;
2979 if ( !myMesh->_is_nil() ) {
2980 myMesh->Load(); // load from study file if not yet done
2981 TPythonDump pd; // not to dump GetGroups()
2982 SMESH::ListOfGroups_var groups = myMesh->GetGroups();
2983 pd << ""; // to avoid optimizing pd out
2984 for ( int i = 0; i < groups->length(); ++i )
2986 SMESH_GroupBase_i* grImpl = SMESH::DownCast<SMESH_GroupBase_i*>( groups[i]);
2989 CORBA::String_var objStr = GetORB()->object_to_string( grImpl->_this() );
2990 int anId = myStudyContext->findId( string( objStr.in() ) );
2992 sprintf( grpName, "Group %d", anId );
2993 SMESHDS_GroupBase* aGrpBaseDS = grImpl->GetGroupDS();
2994 aGrpBaseDS->SetStoreName( grpName );
3004 // ---> create HDF file
3005 aFile = new HDFfile( (char*) filename.ToCString() );
3006 aFile->CreateOnDisk();
3008 // --> iterator for top-level objects
3009 SALOMEDS::ChildIterator_wrap itBig = myCurrentStudy->NewChildIterator( theComponent );
3010 for ( ; itBig->More(); itBig->Next() ) {
3011 SALOMEDS::SObject_wrap gotBranch = itBig->Value();
3013 // --> hypotheses root branch (only one for the study)
3014 if ( gotBranch->Tag() == GetHypothesisRootTag() ) {
3015 // create hypotheses root HDF group
3016 aTopGroup = new HDFgroup( "Hypotheses", aFile );
3017 aTopGroup->CreateOnDisk();
3019 // iterator for all hypotheses
3020 SALOMEDS::ChildIterator_wrap it = myCurrentStudy->NewChildIterator( gotBranch );
3021 for ( ; it->More(); it->Next() ) {
3022 SALOMEDS::SObject_wrap mySObject = it->Value();
3023 CORBA::Object_var anObject = SObjectToObject( mySObject );
3024 if ( !CORBA::is_nil( anObject ) ) {
3025 SMESH::SMESH_Hypothesis_var myHyp = SMESH::SMESH_Hypothesis::_narrow( anObject );
3026 if ( !myHyp->_is_nil() ) {
3027 SMESH_Hypothesis_i* myImpl = dynamic_cast<SMESH_Hypothesis_i*>( GetServant( myHyp ).in() );
3029 string hypname = string( myHyp->GetName() );
3030 string libname = string( myHyp->GetLibName() );
3032 // Needs for save crossplatform libname, i.e. parth of name ( ".dll" for
3033 // WIN32 and ".so" for X-system) must be deleted
3034 int libname_len = libname.length();
3036 if( libname_len > 4 )
3037 libname.resize( libname_len - 4 );
3039 // PAL17753 (Regresion: missing hypothesis in restored study)
3040 // "lib" also should be removed from the beginning
3041 //if( libname_len > 3 )
3042 //libname.resize( libname_len - 3 );
3043 if( libname_len > 6 )
3044 libname = libname.substr( 3, libname_len - 3 - 3 );
3046 CORBA::String_var objStr = GetORB()->object_to_string( anObject );
3047 int id = myStudyContext->findId( string( objStr.in() ) );
3048 string hypdata = string( myImpl->SaveTo() );
3050 // for each hypothesis create HDF group basing on its id
3051 char hypGrpName[30];
3052 sprintf( hypGrpName, "Hypothesis %d", id );
3053 aGroup = new HDFgroup( hypGrpName, aTopGroup );
3054 aGroup->CreateOnDisk();
3055 // --> type name of hypothesis
3056 aSize[ 0 ] = hypname.length() + 1;
3057 aDataset = new HDFdataset( "Name", aGroup, HDF_STRING, aSize, 1 );
3058 aDataset->CreateOnDisk();
3059 aDataset->WriteOnDisk( ( char* )( hypname.c_str() ) );
3060 aDataset->CloseOnDisk();
3061 // --> server plugin library name of hypothesis
3062 aSize[ 0 ] = libname.length() + 1;
3063 aDataset = new HDFdataset( "LibName", aGroup, HDF_STRING, aSize, 1 );
3064 aDataset->CreateOnDisk();
3065 aDataset->WriteOnDisk( ( char* )( libname.c_str() ) );
3066 aDataset->CloseOnDisk();
3067 // --> persistent data of hypothesis
3068 aSize[ 0 ] = hypdata.length() + 1;
3069 aDataset = new HDFdataset( "Data", aGroup, HDF_STRING, aSize, 1 );
3070 aDataset->CreateOnDisk();
3071 aDataset->WriteOnDisk( ( char* )( hypdata.c_str() ) );
3072 aDataset->CloseOnDisk();
3073 // close hypothesis HDF group
3074 aGroup->CloseOnDisk();
3079 // close hypotheses root HDF group
3080 aTopGroup->CloseOnDisk();
3082 // --> algorithms root branch (only one for the study)
3083 else if ( gotBranch->Tag() == GetAlgorithmsRootTag() ) {
3084 // create algorithms root HDF group
3085 aTopGroup = new HDFgroup( "Algorithms", aFile );
3086 aTopGroup->CreateOnDisk();
3088 // iterator for all algorithms
3089 SALOMEDS::ChildIterator_wrap it = myCurrentStudy->NewChildIterator( gotBranch );
3090 for ( ; it->More(); it->Next() ) {
3091 SALOMEDS::SObject_wrap mySObject = it->Value();
3092 CORBA::Object_var anObject = SObjectToObject( mySObject );
3093 if ( !CORBA::is_nil( anObject ) ) {
3094 SMESH::SMESH_Hypothesis_var myHyp = SMESH::SMESH_Hypothesis::_narrow( anObject );
3095 if ( !myHyp->_is_nil() ) {
3096 SMESH_Hypothesis_i* myImpl = dynamic_cast<SMESH_Hypothesis_i*>( GetServant( myHyp ).in() );
3098 string hypname = string( myHyp->GetName() );
3099 string libname = string( myHyp->GetLibName() );
3101 // Needs for save crossplatform libname, i.e. parth of name ( ".dll" for
3102 // WIN32 and ".so" for X-system) must be deleted
3103 int libname_len = libname.length();
3105 if( libname_len > 4 )
3106 libname.resize( libname_len - 4 );
3108 // PAL17753 (Regresion: missing hypothesis in restored study)
3109 // "lib" also should be removed from the beginning
3110 //if( libname_len > 3 )
3111 //libname.resize( libname_len - 3 );
3112 if( libname_len > 6 )
3113 libname = libname.substr( 3, libname_len - 3 - 3 );
3115 CORBA::String_var objStr = GetORB()->object_to_string( anObject );
3116 int id = myStudyContext->findId( string( objStr.in() ) );
3117 string hypdata = string( myImpl->SaveTo() );
3119 // for each algorithm create HDF group basing on its id
3120 char hypGrpName[30];
3121 sprintf( hypGrpName, "Algorithm %d", id );
3122 aGroup = new HDFgroup( hypGrpName, aTopGroup );
3123 aGroup->CreateOnDisk();
3124 // --> type name of algorithm
3125 aSize[0] = hypname.length() + 1;
3126 aDataset = new HDFdataset( "Name", aGroup, HDF_STRING, aSize, 1 );
3127 aDataset->CreateOnDisk();
3128 aDataset->WriteOnDisk( ( char* )( hypname.c_str() ) );
3129 aDataset->CloseOnDisk();
3130 // --> server plugin library name of hypothesis
3131 aSize[0] = libname.length() + 1;
3132 aDataset = new HDFdataset( "LibName", aGroup, HDF_STRING, aSize, 1 );
3133 aDataset->CreateOnDisk();
3134 aDataset->WriteOnDisk( ( char* )( libname.c_str() ) );
3135 aDataset->CloseOnDisk();
3136 // --> persistent data of algorithm
3137 aSize[0] = hypdata.length() + 1;
3138 aDataset = new HDFdataset( "Data", aGroup, HDF_STRING, aSize, 1 );
3139 aDataset->CreateOnDisk();
3140 aDataset->WriteOnDisk( ( char* )( hypdata.c_str() ) );
3141 aDataset->CloseOnDisk();
3142 // close algorithm HDF group
3143 aGroup->CloseOnDisk();
3148 // close algorithms root HDF group
3149 aTopGroup->CloseOnDisk();
3151 // --> mesh objects roots branches
3152 else if ( gotBranch->Tag() > GetAlgorithmsRootTag() ) {
3153 CORBA::Object_var anObject = SObjectToObject( gotBranch );
3154 if ( !CORBA::is_nil( anObject ) ) {
3155 SMESH::SMESH_Mesh_var myMesh = SMESH::SMESH_Mesh::_narrow( anObject ) ;
3156 if ( !myMesh->_is_nil() ) {
3157 SMESH_Mesh_i* myImpl = dynamic_cast<SMESH_Mesh_i*>( GetServant( myMesh ).in() );
3159 CORBA::String_var objStr = GetORB()->object_to_string( anObject );
3160 int id = myStudyContext->findId( string( objStr.in() ) );
3161 ::SMESH_Mesh& myLocMesh = myImpl->GetImpl();
3162 SMESHDS_Mesh* mySMESHDSMesh = myLocMesh.GetMeshDS();
3163 bool hasShape = myLocMesh.HasShapeToMesh();
3165 // for each mesh open the HDF group basing on its id
3166 char meshGrpName[ 30 ];
3167 sprintf( meshGrpName, "Mesh %d", id );
3168 aTopGroup = new HDFgroup( meshGrpName, aFile );
3169 aTopGroup->CreateOnDisk();
3171 // --> put dataset to hdf file which is a flag that mesh has data
3172 string strHasData = "0";
3173 // check if the mesh is not empty
3174 if ( mySMESHDSMesh->NbNodes() > 0 ) {
3175 // write mesh data to med file
3176 myWriter.SetMesh( mySMESHDSMesh );
3177 myWriter.SetMeshId( id );
3180 aSize[ 0 ] = strHasData.length() + 1;
3181 aDataset = new HDFdataset( "Has data", aTopGroup, HDF_STRING, aSize, 1 );
3182 aDataset->CreateOnDisk();
3183 aDataset->WriteOnDisk( ( char* )( strHasData.c_str() ) );
3184 aDataset->CloseOnDisk();
3187 // for each mesh open the HDF group basing on its auto color parameter
3188 char meshAutoColorName[ 30 ];
3189 sprintf( meshAutoColorName, "AutoColorMesh %d", id );
3191 anAutoColor[0] = myImpl->GetAutoColor();
3193 aDataset = new HDFdataset( meshAutoColorName, aTopGroup, HDF_INT32, aSize, 1 );
3194 aDataset->CreateOnDisk();
3195 aDataset->WriteOnDisk( anAutoColor );
3196 aDataset->CloseOnDisk();
3198 // issue 0020693. Store _isModified flag
3199 int isModified = myLocMesh.GetIsModified();
3201 aDataset = new HDFdataset( "_isModified", aTopGroup, HDF_INT32, aSize, 1 );
3202 aDataset->CreateOnDisk();
3203 aDataset->WriteOnDisk( &isModified );
3204 aDataset->CloseOnDisk();
3206 // issue 20918. Store Persistent Id of SMESHDS_Mesh
3207 int meshPersistentId = mySMESHDSMesh->GetPersistentId();
3209 aDataset = new HDFdataset( "meshPersistentId", aTopGroup, HDF_INT32, aSize, 1 );
3210 aDataset->CreateOnDisk();
3211 aDataset->WriteOnDisk( &meshPersistentId );
3212 aDataset->CloseOnDisk();
3214 // write reference on a shape if exists
3215 SALOMEDS::SObject_wrap myRef;
3216 bool shapeRefFound = false;
3217 bool found = gotBranch->FindSubObject( GetRefOnShapeTag(), myRef.inout() );
3219 SALOMEDS::SObject_wrap myShape;
3220 bool ok = myRef->ReferencedObject( myShape.inout() );
3222 shapeRefFound = (! CORBA::is_nil( myShape->GetObject() ));
3223 string myRefOnObject = myShape->GetID();
3224 if ( shapeRefFound && myRefOnObject.length() > 0 ) {
3225 aSize[ 0 ] = myRefOnObject.length() + 1;
3226 aDataset = new HDFdataset( "Ref on shape", aTopGroup, HDF_STRING, aSize, 1 );
3227 aDataset->CreateOnDisk();
3228 aDataset->WriteOnDisk( ( char* )( myRefOnObject.c_str() ) );
3229 aDataset->CloseOnDisk();
3234 // write applied hypotheses if exist
3235 SALOMEDS::SObject_wrap myHypBranch;
3236 found = gotBranch->FindSubObject( GetRefOnAppliedHypothesisTag(), myHypBranch.inout() );
3237 if ( found && !shapeRefFound && hasShape) { // remove applied hyps
3238 myCurrentStudy->NewBuilder()->RemoveObjectWithChildren( myHypBranch );
3240 if ( found && (shapeRefFound || !hasShape) ) {
3241 aGroup = new HDFgroup( "Applied Hypotheses", aTopGroup );
3242 aGroup->CreateOnDisk();
3244 SALOMEDS::ChildIterator_wrap it = myCurrentStudy->NewChildIterator( myHypBranch );
3246 for ( ; it->More(); it->Next() ) {
3247 SALOMEDS::SObject_wrap mySObject = it->Value();
3248 SALOMEDS::SObject_wrap myRefOnHyp;
3249 bool ok = mySObject->ReferencedObject( myRefOnHyp.inout() );
3251 // san - it is impossible to recover applied hypotheses
3252 // using their entries within Load() method,
3253 // for there are no AttributeIORs in the study when Load() is working.
3254 // Hence, it is better to store persistent IDs of hypotheses as references to them
3256 //string myRefOnObject = myRefOnHyp->GetID();
3257 CORBA::Object_var anObject = SObjectToObject( myRefOnHyp );
3258 CORBA::String_var objStr = GetORB()->object_to_string( anObject );
3259 int id = myStudyContext->findId( string( objStr.in() ) );
3260 //if ( myRefOnObject.length() > 0 ) {
3261 //aSize[ 0 ] = myRefOnObject.length() + 1;
3262 char hypName[ 30 ], hypId[ 30 ];
3263 sprintf( hypName, "Hyp %d", ++hypNb );
3264 sprintf( hypId, "%d", id );
3265 aSize[ 0 ] = strlen( hypId ) + 1;
3266 aDataset = new HDFdataset( hypName, aGroup, HDF_STRING, aSize, 1 );
3267 aDataset->CreateOnDisk();
3268 //aDataset->WriteOnDisk( ( char* )( myRefOnObject.c_str() ) );
3269 aDataset->WriteOnDisk( hypId );
3270 aDataset->CloseOnDisk();
3274 aGroup->CloseOnDisk();
3277 // write applied algorithms if exist
3278 SALOMEDS::SObject_wrap myAlgoBranch;
3279 found = gotBranch->FindSubObject( GetRefOnAppliedAlgorithmsTag(),
3280 myAlgoBranch.inout() );
3281 if ( found && !shapeRefFound && hasShape) { // remove applied algos
3282 myCurrentStudy->NewBuilder()->RemoveObjectWithChildren( myAlgoBranch );
3284 if ( found && (shapeRefFound || !hasShape)) {
3285 aGroup = new HDFgroup( "Applied Algorithms", aTopGroup );
3286 aGroup->CreateOnDisk();
3288 SALOMEDS::ChildIterator_wrap it = myCurrentStudy->NewChildIterator( myAlgoBranch );
3290 for ( ; it->More(); it->Next() ) {
3291 SALOMEDS::SObject_wrap mySObject = it->Value();
3292 SALOMEDS::SObject_wrap myRefOnAlgo;
3293 bool ok = mySObject->ReferencedObject( myRefOnAlgo.inout() );
3295 // san - it is impossible to recover applied algorithms
3296 // using their entries within Load() method,
3297 // for there are no AttributeIORs in the study when Load() is working.
3298 // Hence, it is better to store persistent IDs of algorithms as references to them
3300 //string myRefOnObject = myRefOnAlgo->GetID();
3301 CORBA::Object_var anObject = SObjectToObject( myRefOnAlgo );
3302 CORBA::String_var objStr = GetORB()->object_to_string( anObject );
3303 int id = myStudyContext->findId( string( objStr.in() ) );
3304 //if ( myRefOnObject.length() > 0 ) {
3305 //aSize[ 0 ] = myRefOnObject.length() + 1;
3306 char algoName[ 30 ], algoId[ 30 ];
3307 sprintf( algoName, "Algo %d", ++algoNb );
3308 sprintf( algoId, "%d", id );
3309 aSize[ 0 ] = strlen( algoId ) + 1;
3310 aDataset = new HDFdataset( algoName, aGroup, HDF_STRING, aSize, 1 );
3311 aDataset->CreateOnDisk();
3312 //aDataset->WriteOnDisk( ( char* )( myRefOnObject.c_str() ) );
3313 aDataset->WriteOnDisk( algoId );
3314 aDataset->CloseOnDisk();
3318 aGroup->CloseOnDisk();
3321 // --> submesh objects sub-branches
3323 for ( int i = GetSubMeshOnVertexTag(); i <= GetSubMeshOnCompoundTag(); i++ ) {
3324 SALOMEDS::SObject_wrap mySubmeshBranch;
3325 found = gotBranch->FindSubObject( i, mySubmeshBranch.inout() );
3327 if ( found ) // check if there is shape reference in submeshes
3329 bool hasShapeRef = false;
3330 SALOMEDS::ChildIterator_wrap itSM =
3331 myCurrentStudy->NewChildIterator( mySubmeshBranch );
3332 for ( ; itSM->More(); itSM->Next() ) {
3333 SALOMEDS::SObject_wrap mySubRef, myShape, mySObject = itSM->Value();
3334 if ( mySObject->FindSubObject( GetRefOnShapeTag(), mySubRef.inout() ))
3335 mySubRef->ReferencedObject( myShape.inout() );
3336 if ( !CORBA::is_nil( myShape ) && !CORBA::is_nil( myShape->GetObject() ))
3339 { // remove one submesh
3340 if ( shapeRefFound )
3341 { // unassign hypothesis
3342 SMESH::SMESH_subMesh_var mySubMesh =
3343 SMESH::SMESH_subMesh::_narrow( SObjectToObject( mySObject ));
3344 if ( !mySubMesh->_is_nil() ) {
3345 int shapeID = mySubMesh->GetId();
3346 TopoDS_Shape S = mySMESHDSMesh->IndexToShape( shapeID );
3347 const list<const SMESHDS_Hypothesis*>& hypList =
3348 mySMESHDSMesh->GetHypothesis( S );
3349 list<const SMESHDS_Hypothesis*>::const_iterator hyp = hypList.begin();
3350 while ( hyp != hypList.end() ) {
3351 int hypID = (*hyp++)->GetID(); // goto next hyp here because
3352 myLocMesh.RemoveHypothesis( S, hypID ); // hypList changes here
3356 myCurrentStudy->NewBuilder()->RemoveObjectWithChildren( mySObject );
3358 } // loop on submeshes of a type
3359 if ( !shapeRefFound || !hasShapeRef ) { // remove the whole submeshes branch
3360 myCurrentStudy->NewBuilder()->RemoveObjectWithChildren( mySubmeshBranch );
3363 } // end check if there is shape reference in submeshes
3365 char name_meshgroup[ 30 ];
3366 if ( i == GetSubMeshOnVertexTag() )
3367 strcpy( name_meshgroup, "SubMeshes On Vertex" );
3368 else if ( i == GetSubMeshOnEdgeTag() )
3369 strcpy( name_meshgroup, "SubMeshes On Edge" );
3370 else if ( i == GetSubMeshOnWireTag() )
3371 strcpy( name_meshgroup, "SubMeshes On Wire" );
3372 else if ( i == GetSubMeshOnFaceTag() )
3373 strcpy( name_meshgroup, "SubMeshes On Face" );
3374 else if ( i == GetSubMeshOnShellTag() )
3375 strcpy( name_meshgroup, "SubMeshes On Shell" );
3376 else if ( i == GetSubMeshOnSolidTag() )
3377 strcpy( name_meshgroup, "SubMeshes On Solid" );
3378 else if ( i == GetSubMeshOnCompoundTag() )
3379 strcpy( name_meshgroup, "SubMeshes On Compound" );
3381 // for each type of submeshes create container HDF group
3382 aGroup = new HDFgroup( name_meshgroup, aTopGroup );
3383 aGroup->CreateOnDisk();
3385 // iterator for all submeshes of given type
3386 SALOMEDS::ChildIterator_wrap itSM = myCurrentStudy->NewChildIterator( mySubmeshBranch );
3387 for ( ; itSM->More(); itSM->Next() ) {
3388 SALOMEDS::SObject_wrap mySObject = itSM->Value();
3389 CORBA::Object_var anSubObject = SObjectToObject( mySObject );
3390 if ( !CORBA::is_nil( anSubObject ))
3392 SMESH::SMESH_subMesh_var mySubMesh = SMESH::SMESH_subMesh::_narrow( anSubObject ) ;
3393 CORBA::String_var objStr = GetORB()->object_to_string( anSubObject );
3394 int subid = myStudyContext->findId( string( objStr.in() ) );
3396 // for each mesh open the HDF group basing on its id
3397 char submeshGrpName[ 30 ];
3398 sprintf( submeshGrpName, "SubMesh %d", subid );
3399 aSubGroup = new HDFgroup( submeshGrpName, aGroup );
3400 aSubGroup->CreateOnDisk();
3402 // write reference on a shape, already checked if it exists
3403 SALOMEDS::SObject_wrap mySubRef, myShape;
3404 if ( mySObject->FindSubObject( GetRefOnShapeTag(), mySubRef.inout() ))
3405 mySubRef->ReferencedObject( myShape.inout() );
3406 string myRefOnObject = myShape->GetID();
3407 if ( myRefOnObject.length() > 0 ) {
3408 aSize[ 0 ] = myRefOnObject.length() + 1;
3409 aDataset = new HDFdataset( "Ref on shape", aSubGroup, HDF_STRING, aSize, 1 );
3410 aDataset->CreateOnDisk();
3411 aDataset->WriteOnDisk( ( char* )( myRefOnObject.c_str() ) );
3412 aDataset->CloseOnDisk();
3415 // write applied hypotheses if exist
3416 SALOMEDS::SObject_wrap mySubHypBranch;
3417 found = mySObject->FindSubObject( GetRefOnAppliedHypothesisTag(),
3418 mySubHypBranch.inout() );
3420 aSubSubGroup = new HDFgroup( "Applied Hypotheses", aSubGroup );
3421 aSubSubGroup->CreateOnDisk();
3423 SALOMEDS::ChildIterator_wrap it = myCurrentStudy->NewChildIterator( mySubHypBranch );
3425 for ( ; it->More(); it->Next() ) {
3426 SALOMEDS::SObject_wrap mySubSObject = it->Value();
3427 SALOMEDS::SObject_wrap myRefOnHyp;
3428 bool ok = mySubSObject->ReferencedObject( myRefOnHyp.inout() );
3430 //string myRefOnObject = myRefOnHyp->GetID();
3431 CORBA::Object_var anObject = SObjectToObject( myRefOnHyp );
3432 CORBA::String_var objStr = GetORB()->object_to_string( anObject );
3433 int id = myStudyContext->findId( string( objStr.in() ) );
3434 //if ( myRefOnObject.length() > 0 ) {
3435 //aSize[ 0 ] = myRefOnObject.length() + 1;
3436 char hypName[ 30 ], hypId[ 30 ];
3437 sprintf( hypName, "Hyp %d", ++hypNb );
3438 sprintf( hypId, "%d", id );
3439 aSize[ 0 ] = strlen( hypId ) + 1;
3440 aDataset = new HDFdataset( hypName, aSubSubGroup, HDF_STRING, aSize, 1 );
3441 aDataset->CreateOnDisk();
3442 //aDataset->WriteOnDisk( ( char* )( myRefOnObject.c_str() ) );
3443 aDataset->WriteOnDisk( hypId );
3444 aDataset->CloseOnDisk();
3448 aSubSubGroup->CloseOnDisk();
3451 // write applied algorithms if exist
3452 SALOMEDS::SObject_wrap mySubAlgoBranch;
3453 found = mySObject->FindSubObject( GetRefOnAppliedAlgorithmsTag(),
3454 mySubAlgoBranch.inout() );
3456 aSubSubGroup = new HDFgroup( "Applied Algorithms", aSubGroup );
3457 aSubSubGroup->CreateOnDisk();
3459 SALOMEDS::ChildIterator_wrap it =
3460 myCurrentStudy->NewChildIterator( mySubAlgoBranch );
3462 for ( ; it->More(); it->Next() ) {
3463 SALOMEDS::SObject_wrap mySubSObject = it->Value();
3464 SALOMEDS::SObject_wrap myRefOnAlgo;
3465 bool ok = mySubSObject->ReferencedObject( myRefOnAlgo.inout() );
3467 //string myRefOnObject = myRefOnAlgo->GetID();
3468 CORBA::Object_var anObject = SObjectToObject( myRefOnAlgo );
3469 CORBA::String_var objStr = GetORB()->object_to_string( anObject );
3470 int id = myStudyContext->findId( string( objStr.in() ) );
3471 //if ( myRefOnObject.length() > 0 ) {
3472 //aSize[ 0 ] = myRefOnObject.length() + 1;
3473 char algoName[ 30 ], algoId[ 30 ];
3474 sprintf( algoName, "Algo %d", ++algoNb );
3475 sprintf( algoId, "%d", id );
3476 aSize[ 0 ] = strlen( algoId ) + 1;
3477 aDataset = new HDFdataset( algoName, aSubSubGroup, HDF_STRING, aSize, 1 );
3478 aDataset->CreateOnDisk();
3479 //aDataset->WriteOnDisk( ( char* )( myRefOnObject.c_str() ) );
3480 aDataset->WriteOnDisk( algoId );
3481 aDataset->CloseOnDisk();
3485 aSubSubGroup->CloseOnDisk();
3487 // close submesh HDF group
3488 aSubGroup->CloseOnDisk();
3491 // close container of submeshes by type HDF group
3492 aGroup->CloseOnDisk();
3495 // All sub-meshes will be stored in MED file
3496 // .. will NOT (PAL 12992)
3497 //if ( shapeRefFound )
3498 //myWriter.AddAllSubMeshes();
3500 // store submesh order if any
3501 const TListOfListOfInt& theOrderIds = myLocMesh.GetMeshOrder();
3502 if ( theOrderIds.size() ) {
3503 char order_list[ 30 ];
3504 strcpy( order_list, "Mesh Order" );
3505 // count number of submesh ids
3507 TListOfListOfInt::const_iterator idIt = theOrderIds.begin();
3508 for ( ; idIt != theOrderIds.end(); idIt++ )
3509 nbIDs += (*idIt).size();
3510 // number of values = number of IDs +
3511 // number of lists (for separators) - 1
3512 int* smIDs = new int [ nbIDs + theOrderIds.size() - 1 ];
3513 idIt = theOrderIds.begin();
3514 for ( int i = 0; idIt != theOrderIds.end(); idIt++ ) {
3515 const TListOfInt& idList = *idIt;
3516 if (idIt != theOrderIds.begin()) // not first list
3517 smIDs[ i++ ] = -1/* *idList.size()*/; // separator between lists
3518 // dump submesh ids from current list
3519 TListOfInt::const_iterator id_smId = idList.begin();
3520 for( ; id_smId != idList.end(); id_smId++ )
3521 smIDs[ i++ ] = *id_smId;
3524 aSize[ 0 ] = nbIDs + theOrderIds.size() - 1;
3526 aDataset = new HDFdataset( order_list, aTopGroup, HDF_INT32, aSize, 1 );
3527 aDataset->CreateOnDisk();
3528 aDataset->WriteOnDisk( smIDs );
3529 aDataset->CloseOnDisk();
3534 // groups root sub-branch
3535 SALOMEDS::SObject_wrap myGroupsBranch;
3536 for ( int i = GetNodeGroupsTag(); i <= GetBallElementsGroupsTag(); i++ ) {
3537 found = gotBranch->FindSubObject( i, myGroupsBranch.inout() );
3539 char name_group[ 30 ];
3540 if ( i == GetNodeGroupsTag() )
3541 strcpy( name_group, "Groups of Nodes" );
3542 else if ( i == GetEdgeGroupsTag() )
3543 strcpy( name_group, "Groups of Edges" );
3544 else if ( i == GetFaceGroupsTag() )
3545 strcpy( name_group, "Groups of Faces" );
3546 else if ( i == GetVolumeGroupsTag() )
3547 strcpy( name_group, "Groups of Volumes" );
3548 else if ( i == Get0DElementsGroupsTag() )
3549 strcpy( name_group, "Groups of 0D Elements" );
3550 else if ( i == GetBallElementsGroupsTag() )
3551 strcpy( name_group, "Groups of Balls" );
3553 aGroup = new HDFgroup( name_group, aTopGroup );
3554 aGroup->CreateOnDisk();
3556 SALOMEDS::ChildIterator_wrap it = myCurrentStudy->NewChildIterator( myGroupsBranch );
3557 for ( ; it->More(); it->Next() ) {
3558 SALOMEDS::SObject_wrap mySObject = it->Value();
3559 CORBA::Object_var aSubObject = SObjectToObject( mySObject );
3560 if ( !CORBA::is_nil( aSubObject ) ) {
3561 SMESH_GroupBase_i* myGroupImpl =
3562 dynamic_cast<SMESH_GroupBase_i*>( GetServant( aSubObject ).in() );
3565 SMESHDS_GroupBase* aGrpBaseDS = myGroupImpl->GetGroupDS();
3569 CORBA::String_var objStr = GetORB()->object_to_string( aSubObject );
3570 int anId = myStudyContext->findId( string( objStr.in() ) );
3572 // For each group, create a dataset named "Group <group_persistent_id>"
3573 // and store the group's user name into it
3574 const char* grpName = aGrpBaseDS->GetStoreName();
3575 char* aUserName = myGroupImpl->GetName();
3576 aSize[ 0 ] = strlen( aUserName ) + 1;
3578 aDataset = new HDFdataset( grpName, aGroup, HDF_STRING, aSize, 1 );
3579 aDataset->CreateOnDisk();
3580 aDataset->WriteOnDisk( aUserName );
3581 aDataset->CloseOnDisk();
3584 // For each group, create a dataset named "Group <group_persistent_id> Color"
3585 // and store the group's color into it
3586 char grpColorName[ 30 ];
3587 sprintf( grpColorName, "ColorGroup %d", anId );
3588 SALOMEDS::Color aColor = myGroupImpl->GetColor();
3590 anRGB[ 0 ] = aColor.R;
3591 anRGB[ 1 ] = aColor.G;
3592 anRGB[ 2 ] = aColor.B;
3594 aDataset = new HDFdataset( grpColorName, aGroup, HDF_FLOAT64, aSize, 1 );
3595 aDataset->CreateOnDisk();
3596 aDataset->WriteOnDisk( anRGB );
3597 aDataset->CloseOnDisk();
3599 // Pass SMESHDS_Group to MED writer
3600 SMESHDS_Group* aGrpDS = dynamic_cast<SMESHDS_Group*>( aGrpBaseDS );
3602 myWriter.AddGroup( aGrpDS );
3604 // write reference on a shape if exists
3605 SMESHDS_GroupOnGeom* aGeomGrp =
3606 dynamic_cast<SMESHDS_GroupOnGeom*>( aGrpBaseDS );
3608 SALOMEDS::SObject_wrap mySubRef, myShape;
3609 if (mySObject->FindSubObject( GetRefOnShapeTag(), mySubRef.inout() ) &&
3610 mySubRef->ReferencedObject( myShape.inout() ) &&
3611 !CORBA::is_nil( myShape->GetObject() ))
3613 string myRefOnObject = myShape->GetID();
3614 if ( myRefOnObject.length() > 0 ) {
3615 char aRefName[ 30 ];
3616 sprintf( aRefName, "Ref on shape %d", anId);
3617 aSize[ 0 ] = myRefOnObject.length() + 1;
3618 aDataset = new HDFdataset(aRefName, aGroup, HDF_STRING, aSize, 1);
3619 aDataset->CreateOnDisk();
3620 aDataset->WriteOnDisk( ( char* )( myRefOnObject.c_str() ) );
3621 aDataset->CloseOnDisk();
3624 else // shape ref is invalid:
3626 // save a group on geometry as ordinary group
3627 myWriter.AddGroup( aGeomGrp );
3630 else if ( SMESH_GroupOnFilter_i* aFilterGrp_i =
3631 dynamic_cast<SMESH_GroupOnFilter_i*>( myGroupImpl ))
3633 std::string str = aFilterGrp_i->FilterToString();
3634 std::string hdfGrpName = "Filter " + SMESH_Comment(anId);
3635 aSize[ 0 ] = str.length() + 1;
3636 aDataset = new HDFdataset( hdfGrpName.c_str(), aGroup, HDF_STRING, aSize, 1);
3637 aDataset->CreateOnDisk();
3638 aDataset->WriteOnDisk( ( char* )( str.c_str() ) );
3639 aDataset->CloseOnDisk();
3643 aGroup->CloseOnDisk();
3647 if ( strcmp( strHasData.c_str(), "1" ) == 0 )
3649 // Flush current mesh information into MED file
3652 // save info on nb of elements
3653 SMESH_PreMeshInfo::SaveToFile( myImpl, id, aFile );
3655 // maybe a shape was deleted in the study
3656 if ( !shapeRefFound && !mySMESHDSMesh->ShapeToMesh().IsNull() && hasShape) {
3657 TopoDS_Shape nullShape;
3658 myLocMesh.ShapeToMesh( nullShape ); // remove shape referring data
3661 if ( !mySMESHDSMesh->SubMeshes().empty() )
3665 aGroup = new HDFgroup( "Submeshes", aTopGroup );
3666 aGroup->CreateOnDisk();
3668 // each element belongs to one or none submesh,
3669 // so for each node/element, we store a submesh ID
3671 // Make maps of submesh IDs of elements sorted by element IDs
3672 // typedef int TElemID;
3673 // typedef int TSubMID;
3674 // map< TElemID, TSubMID > eId2smId, nId2smId;
3675 const map<int,SMESHDS_SubMesh*>& aSubMeshes = mySMESHDSMesh->SubMeshes();
3676 map<int,SMESHDS_SubMesh*>::const_iterator itSubM ( aSubMeshes.begin() );
3677 // SMDS_NodeIteratorPtr itNode;
3678 // SMDS_ElemIteratorPtr itElem;
3679 // for ( itSubM = aSubMeshes.begin(); itSubM != aSubMeshes.end() ; itSubM++ )
3681 // TSubMID aSubMeID = itSubM->first;
3682 // SMESHDS_SubMesh* aSubMesh = itSubM->second;
3683 // if ( aSubMesh->IsComplexSubmesh() )
3684 // continue; // sub-mesh containing other sub-meshes
3686 // for ( itNode = aSubMesh->GetNodes(); itNode->more(); ++hint)
3687 // nId2smId.insert( nId2smId.back(), make_pair( itNode->next()->GetID(), aSubMeID ));
3689 // for ( itElem = aSubMesh->GetElements(); itElem->more(); ++hint)
3690 // hint = eId2smId.insert( eId2smId.back(), make_pair( itElem->next()->GetID(), aSubMeID ));
3693 // // Care of elements that are not on submeshes
3694 // if ( mySMESHDSMesh->NbNodes() != nId2smId.size() ) {
3695 // for ( itNode = mySMESHDSMesh->nodesIterator(); itNode->more(); )
3696 // /* --- stl_map.h says : */
3697 // /* A %map relies on unique keys and thus a %pair is only inserted if its */
3698 // /* first element (the key) is not already present in the %map. */
3699 // nId2smId.insert( make_pair( itNode->next()->GetID(), 0 ));
3701 // int nbElems = mySMESHDSMesh->GetMeshInfo().NbElements();
3702 // if ( nbElems != eId2smId.size() ) {
3703 // for ( itElem = mySMESHDSMesh->elementsIterator(); itElem->more(); )
3704 // eId2smId.insert( make_pair( itElem->next()->GetID(), 0 ));
3707 // Store submesh IDs
3708 for ( int isNode = 0; isNode < 2; ++isNode )
3710 // map< TElemID, TSubMID >& id2smId = isNode ? nId2smId : eId2smId;
3711 // if ( id2smId.empty() ) continue;
3712 // map< TElemID, TSubMID >::const_iterator id_smId = id2smId.begin();
3713 // // make and fill array of submesh IDs
3714 // int* smIDs = new int [ id2smId.size() ];
3715 // for ( int i = 0; id_smId != id2smId.end(); ++id_smId, ++i )
3716 // smIDs[ i ] = id_smId->second;
3717 SMDS_ElemIteratorPtr eIt =
3718 mySMESHDSMesh->elementsIterator( isNode ? SMDSAbs_Node : SMDSAbs_All );
3719 int nbElems = isNode ? mySMESHDSMesh->NbNodes() : mySMESHDSMesh->GetMeshInfo().NbElements();
3722 std::vector<int> smIDs; smIDs.reserve( nbElems );
3723 while ( eIt->more() )
3724 if ( const SMDS_MeshElement* e = eIt->next())
3725 smIDs.push_back( e->getshapeId() );
3727 aSize[ 0 ] = nbElems;
3728 string aDSName( isNode ? "Node Submeshes" : "Element Submeshes");
3729 aDataset = new HDFdataset( (char*)aDSName.c_str(), aGroup, HDF_INT32, aSize, 1 );
3730 aDataset->CreateOnDisk();
3731 aDataset->WriteOnDisk( & smIDs[0] );
3732 aDataset->CloseOnDisk();
3735 aGroup->CloseOnDisk();
3737 // Store node positions on sub-shapes (SMDS_Position):
3738 // ----------------------------------------------------
3740 aGroup = new HDFgroup( "Node Positions", aTopGroup );
3741 aGroup->CreateOnDisk();
3743 // in aGroup, create 5 datasets to contain:
3744 // "Nodes on Edges" - ID of node on edge
3745 // "Edge positions" - U parameter on node on edge
3746 // "Nodes on Faces" - ID of node on face
3747 // "Face U positions" - U parameter of node on face
3748 // "Face V positions" - V parameter of node on face
3750 // Find out nb of nodes on edges and faces
3751 // Collect corresponing sub-meshes
3752 int nbEdgeNodes = 0, nbFaceNodes = 0;
3753 list<SMESHDS_SubMesh*> aEdgeSM, aFaceSM;
3754 // loop on SMESHDS_SubMesh'es
3755 for ( itSubM = aSubMeshes.begin(); itSubM != aSubMeshes.end() ; itSubM++ )
3757 SMESHDS_SubMesh* aSubMesh = (*itSubM).second;
3758 if ( aSubMesh->IsComplexSubmesh() )
3759 continue; // submesh containing other submeshs
3760 int nbNodes = aSubMesh->NbNodes();
3761 if ( nbNodes == 0 ) continue;
3763 int aShapeID = (*itSubM).first;
3764 if ( aShapeID < 1 || aShapeID > mySMESHDSMesh->MaxShapeIndex() )
3766 int aShapeType = mySMESHDSMesh->IndexToShape( aShapeID ).ShapeType();
3767 // write only SMDS_FacePosition and SMDS_EdgePosition
3768 switch ( aShapeType ) {
3770 nbFaceNodes += nbNodes;
3771 aFaceSM.push_back( aSubMesh );
3774 nbEdgeNodes += nbNodes;
3775 aEdgeSM.push_back( aSubMesh );
3781 // Treat positions on edges or faces
3782 for ( int onFace = 0; onFace < 2; onFace++ )
3784 // Create arrays to store in datasets
3785 int iNode = 0, nbNodes = ( onFace ? nbFaceNodes : nbEdgeNodes );
3786 if (!nbNodes) continue;
3787 int* aNodeIDs = new int [ nbNodes ];
3788 double* aUPos = new double [ nbNodes ];
3789 double* aVPos = ( onFace ? new double[ nbNodes ] : 0 );
3792 // loop on sub-meshes
3793 list<SMESHDS_SubMesh*> * pListSM = ( onFace ? &aFaceSM : &aEdgeSM );
3794 list<SMESHDS_SubMesh*>::iterator itSM = pListSM->begin();
3795 for ( ; itSM != pListSM->end(); itSM++ )
3797 SMESHDS_SubMesh* aSubMesh = (*itSM);
3799 SMDS_NodeIteratorPtr itNode = aSubMesh->GetNodes();
3800 // loop on nodes in aSubMesh
3801 while ( itNode->more() )
3804 const SMDS_MeshNode* node = itNode->next();
3805 aNodeIDs [ iNode ] = node->GetID();
3808 const SMDS_PositionPtr pos = node->GetPosition();
3809 if ( onFace ) { // on FACE
3810 const SMDS_FacePosition* fPos =
3811 dynamic_cast<const SMDS_FacePosition*>( pos );
3813 aUPos[ iNode ] = fPos->GetUParameter();
3814 aVPos[ iNode ] = fPos->GetVParameter();
3821 const SMDS_EdgePosition* ePos =
3822 dynamic_cast<const SMDS_EdgePosition*>( pos );
3824 aUPos[ iNode ] = ePos->GetUParameter();
3830 } // loop on nodes in aSubMesh
3831 } // loop on sub-meshes
3836 aSize[ 0 ] = nbNodes;
3838 string aDSName( onFace ? "Nodes on Faces" : "Nodes on Edges");
3839 aDataset = new HDFdataset( (char*)aDSName.c_str(), aGroup, HDF_INT32, aSize, 1 );
3840 aDataset->CreateOnDisk();
3841 aDataset->WriteOnDisk( aNodeIDs );
3842 aDataset->CloseOnDisk();
3845 aDSName = ( onFace ? "Face U positions" : "Edge positions");
3846 aDataset = new HDFdataset( (char*)aDSName.c_str(), aGroup, HDF_FLOAT64, aSize, 1);
3847 aDataset->CreateOnDisk();
3848 aDataset->WriteOnDisk( aUPos );
3849 aDataset->CloseOnDisk();
3852 aDataset = new HDFdataset( "Face V positions", aGroup, HDF_FLOAT64, aSize, 1);
3853 aDataset->CreateOnDisk();
3854 aDataset->WriteOnDisk( aVPos );
3855 aDataset->CloseOnDisk();
3860 if ( aVPos ) delete [] aVPos;
3862 } // treat positions on edges or faces
3864 // close "Node Positions" group
3865 aGroup->CloseOnDisk();
3867 } // if ( there are submeshes in SMESHDS_Mesh )
3870 // close mesh HDF group
3871 aTopGroup->CloseOnDisk();
3879 aFile->CloseOnDisk();
3882 // Convert temporary files to stream
3883 aStreamFile = SALOMEDS_Tool::PutFilesToStream( tmpDir.ToCString(), aFileSeq.in(), isMultiFile );
3885 // Remove temporary files and directory
3887 SALOMEDS_Tool::RemoveTemporaryFiles( tmpDir.ToCString(), aFileSeq.in(), true );
3889 INFOS( "SMESH_Gen_i::Save() completed" );
3890 return aStreamFile._retn();
3893 //=============================================================================
3895 * SMESH_Gen_i::SaveASCII
3897 * Save SMESH module's data in ASCII format
3899 //=============================================================================
3901 SALOMEDS::TMPFile* SMESH_Gen_i::SaveASCII( SALOMEDS::SComponent_ptr theComponent,
3903 bool isMultiFile ) {
3904 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::SaveASCII" );
3905 SALOMEDS::TMPFile_var aStreamFile = Save( theComponent, theURL, isMultiFile );
3906 return aStreamFile._retn();
3908 //after usual saving needs to encipher binary to text string
3909 //Any binary symbol will be represent as "|xx" () hexadecimal format number
3910 int size = aStreamFile.in().length();
3911 _CORBA_Octet* buffer = new _CORBA_Octet[size*3+1];
3912 for ( int i = 0; i < size; i++ )
3913 sprintf( (char*)&(buffer[i*3]), "|%02x", aStreamFile[i] );
3915 buffer[size * 3] = '\0';
3917 SALOMEDS::TMPFile_var anAsciiStreamFile = new SALOMEDS::TMPFile(size*3, size*3, buffer, 1);
3919 return anAsciiStreamFile._retn();
3922 //=============================================================================
3924 * SMESH_Gen_i::loadGeomData
3926 * Load GEOM module data
3928 //=============================================================================
3930 void SMESH_Gen_i::loadGeomData( SALOMEDS::SComponent_ptr theCompRoot )
3932 if ( theCompRoot->_is_nil() )
3935 SALOMEDS::Study_var aStudy = SALOMEDS::Study::_narrow( theCompRoot->GetStudy() );
3936 if ( aStudy->_is_nil() )
3939 SALOMEDS::StudyBuilder_var aStudyBuilder = aStudy->NewBuilder();
3940 aStudyBuilder->LoadWith( theCompRoot, GetGeomEngine() );
3943 //=============================================================================
3947 * Load SMESH module's data
3949 //=============================================================================
3951 bool SMESH_Gen_i::Load( SALOMEDS::SComponent_ptr theComponent,
3952 const SALOMEDS::TMPFile& theStream,
3956 INFOS( "SMESH_Gen_i::Load" );
3958 if ( myCurrentStudy->_is_nil() ||
3959 theComponent->GetStudy()->StudyId() != myCurrentStudy->StudyId() )
3960 SetCurrentStudy( theComponent->GetStudy() );
3962 /* if( !theComponent->_is_nil() )
3964 //SALOMEDS::Study_var aStudy = SALOMEDS::Study::_narrow( theComponent->GetStudy() );
3965 if( !myCurrentStudy->FindComponent( "GEOM" )->_is_nil() )
3966 loadGeomData( myCurrentStudy->FindComponent( "GEOM" ) );
3969 StudyContext* myStudyContext = GetCurrentStudyContext();
3971 // Get temporary files location
3972 TCollection_AsciiString tmpDir =
3973 ( char* )( isMultiFile ? theURL : SALOMEDS_Tool::GetTmpDir().c_str() );
3975 INFOS( "THE URL++++++++++++++" );
3977 INFOS( "THE TMP PATH+++++++++" );
3980 // Convert the stream into sequence of files to process
3981 SALOMEDS::ListOfFileNames_var aFileSeq = SALOMEDS_Tool::PutStreamToFiles( theStream,
3984 TCollection_AsciiString aStudyName( "" );
3985 if ( isMultiFile ) {
3986 CORBA::String_var url = myCurrentStudy->URL();
3987 aStudyName = (char*)SALOMEDS_Tool::GetNameFromPath( url.in() ).c_str();
3989 // Set names of temporary files
3990 TCollection_AsciiString filename = tmpDir + aStudyName + "_SMESH.hdf";
3991 TCollection_AsciiString meshfile = tmpDir + aStudyName + "_SMESH_Mesh.med";
3995 HDFdataset* aDataset;
3996 HDFgroup* aTopGroup;
3998 HDFgroup* aSubGroup;
3999 HDFgroup* aSubSubGroup;
4002 // ---> open HDF file
4003 aFile = new HDFfile( (char*) filename.ToCString() );
4005 aFile->OpenOnDisk( HDF_RDONLY );
4007 catch ( HDFexception ) {
4008 INFOS( "Load(): " << filename << " not found!" );
4012 TPythonDump pd; // prevent dump during loading
4014 // DriverMED_R_SMESHDS_Mesh myReader;
4015 // myReader.SetFile( meshfile.ToCString() );
4017 // For PAL13473 ("Repetitive mesh") implementation.
4018 // New dependencies between SMESH objects are established:
4019 // now hypotheses can refer to meshes, shapes and other hypotheses.
4020 // To keep data consistent, the following order of data restoration
4022 // 1. Create hypotheses
4023 // 2. Create all meshes
4024 // 3. Load hypotheses' data
4027 list< pair< SMESH_Hypothesis_i*, string > > hypDataList;
4028 list< pair< SMESH_Mesh_i*, HDFgroup* > > meshGroupList;
4030 // get total number of top-level groups
4031 int aNbGroups = aFile->nInternalObjects();
4032 if ( aNbGroups > 0 ) {
4033 // --> in first turn we should read&create hypotheses
4034 if ( aFile->ExistInternalObject( "Hypotheses" ) ) {
4035 // open hypotheses root HDF group
4036 aTopGroup = new HDFgroup( "Hypotheses", aFile );
4037 aTopGroup->OpenOnDisk();
4039 // get number of hypotheses
4040 int aNbObjects = aTopGroup->nInternalObjects();
4041 for ( int j = 0; j < aNbObjects; j++ ) {
4042 // try to identify hypothesis
4043 char hypGrpName[ HDF_NAME_MAX_LEN+1 ];
4044 aTopGroup->InternalObjectIndentify( j, hypGrpName );
4046 if ( string( hypGrpName ).substr( 0, 10 ) == string( "Hypothesis" ) ) {
4047 // open hypothesis group
4048 aGroup = new HDFgroup( hypGrpName, aTopGroup );
4049 aGroup->OpenOnDisk();
4051 // --> get hypothesis id
4052 int id = atoi( string( hypGrpName ).substr( 10 ).c_str() );
4057 // get number of datasets
4058 int aNbSubObjects = aGroup->nInternalObjects();
4059 for ( int k = 0; k < aNbSubObjects; k++ ) {
4061 char name_of_subgroup[ HDF_NAME_MAX_LEN+1 ];
4062 aGroup->InternalObjectIndentify( k, name_of_subgroup );
4063 // --> get hypothesis name
4064 if ( strcmp( name_of_subgroup, "Name" ) == 0 ) {
4065 aDataset = new HDFdataset( name_of_subgroup, aGroup );
4066 aDataset->OpenOnDisk();
4067 size = aDataset->GetSize();
4068 char* hypname_str = new char[ size ];
4069 aDataset->ReadFromDisk( hypname_str );
4070 hypname = string( hypname_str );
4071 delete [] hypname_str;
4072 aDataset->CloseOnDisk();
4074 // --> get hypothesis plugin library name
4075 if ( strcmp( name_of_subgroup, "LibName" ) == 0 ) {
4076 aDataset = new HDFdataset( name_of_subgroup, aGroup );
4077 aDataset->OpenOnDisk();
4078 size = aDataset->GetSize();
4079 char* libname_str = new char[ size ];
4080 aDataset->ReadFromDisk( libname_str );
4081 if(MYDEBUG) SCRUTE( libname_str );
4082 libname = string( libname_str );
4083 delete [] libname_str;
4084 aDataset->CloseOnDisk();
4086 // --> get hypothesis data
4087 if ( strcmp( name_of_subgroup, "Data" ) == 0 ) {
4088 aDataset = new HDFdataset( name_of_subgroup, aGroup );
4089 aDataset->OpenOnDisk();
4090 size = aDataset->GetSize();
4091 char* hypdata_str = new char[ size ];
4092 aDataset->ReadFromDisk( hypdata_str );
4093 hypdata = string( hypdata_str );
4094 delete [] hypdata_str;
4095 aDataset->CloseOnDisk();
4098 // close hypothesis HDF group
4099 aGroup->CloseOnDisk();
4101 // --> restore hypothesis from data
4102 if ( id > 0 && !hypname.empty()/* && !hypdata.empty()*/ ) { // VSR : persistent data can be empty
4103 if(MYDEBUG) MESSAGE("VSR - load hypothesis : id = " << id <<
4104 ", name = " << hypname.c_str() << ", persistent string = " << hypdata.c_str());
4105 SMESH::SMESH_Hypothesis_var myHyp;
4107 try { // protect persistence mechanism against exceptions
4108 myHyp = this->createHypothesis( hypname.c_str(), libname.c_str() );
4111 INFOS( "Exception during hypothesis creation" );
4114 SMESH_Hypothesis_i* myImpl = dynamic_cast<SMESH_Hypothesis_i*>( GetServant( myHyp ).in() );
4116 // myImpl->LoadFrom( hypdata.c_str() );
4117 hypDataList.push_back( make_pair( myImpl, hypdata ));
4118 CORBA::String_var iorString = GetORB()->object_to_string( myHyp );
4119 int newId = myStudyContext->findId( iorString.in() );
4120 myStudyContext->mapOldToNew( id, newId );
4123 if(MYDEBUG) MESSAGE( "VSR - SMESH_Gen::Load - can't get servant" );
4127 // close hypotheses root HDF group
4128 aTopGroup->CloseOnDisk();
4132 // --> then we should read&create algorithms
4133 if ( aFile->ExistInternalObject( "Algorithms" ) ) {
4134 // open algorithms root HDF group
4135 aTopGroup = new HDFgroup( "Algorithms", aFile );
4136 aTopGroup->OpenOnDisk();
4138 // get number of algorithms
4139 int aNbObjects = aTopGroup->nInternalObjects();
4140 for ( int j = 0; j < aNbObjects; j++ ) {
4141 // try to identify algorithm
4142 char hypGrpName[ HDF_NAME_MAX_LEN+1 ];
4143 aTopGroup->InternalObjectIndentify( j, hypGrpName );
4145 if ( string( hypGrpName ).substr( 0, 9 ) == string( "Algorithm" ) ) {
4146 // open algorithm group
4147 aGroup = new HDFgroup( hypGrpName, aTopGroup );
4148 aGroup->OpenOnDisk();
4150 // --> get algorithm id
4151 int id = atoi( string( hypGrpName ).substr( 9 ).c_str() );
4156 // get number of datasets
4157 int aNbSubObjects = aGroup->nInternalObjects();
4158 for ( int k = 0; k < aNbSubObjects; k++ ) {
4160 char name_of_subgroup[ HDF_NAME_MAX_LEN+1 ];
4161 aGroup->InternalObjectIndentify( k, name_of_subgroup );
4162 // --> get algorithm name
4163 if ( strcmp( name_of_subgroup, "Name" ) == 0 ) {
4164 aDataset = new HDFdataset( name_of_subgroup, aGroup );
4165 aDataset->OpenOnDisk();
4166 size = aDataset->GetSize();
4167 char* hypname_str = new char[ size ];
4168 aDataset->ReadFromDisk( hypname_str );
4169 hypname = string( hypname_str );
4170 delete [] hypname_str;
4171 aDataset->CloseOnDisk();
4173 // --> get algorithm plugin library name
4174 if ( strcmp( name_of_subgroup, "LibName" ) == 0 ) {
4175 aDataset = new HDFdataset( name_of_subgroup, aGroup );
4176 aDataset->OpenOnDisk();
4177 size = aDataset->GetSize();
4178 char* libname_str = new char[ size ];
4179 aDataset->ReadFromDisk( libname_str );
4180 if(MYDEBUG) SCRUTE( libname_str );
4181 libname = string( libname_str );
4182 delete [] libname_str;
4183 aDataset->CloseOnDisk();
4185 // --> get algorithm data
4186 if ( strcmp( name_of_subgroup, "Data" ) == 0 ) {
4187 aDataset = new HDFdataset( name_of_subgroup, aGroup );
4188 aDataset->OpenOnDisk();
4189 size = aDataset->GetSize();
4190 char* hypdata_str = new char[ size ];
4191 aDataset->ReadFromDisk( hypdata_str );
4192 if(MYDEBUG) SCRUTE( hypdata_str );
4193 hypdata = string( hypdata_str );
4194 delete [] hypdata_str;
4195 aDataset->CloseOnDisk();
4198 // close algorithm HDF group
4199 aGroup->CloseOnDisk();
4201 // --> restore algorithm from data
4202 if ( id > 0 && !hypname.empty()/* && !hypdata.empty()*/ ) { // VSR : persistent data can be empty
4203 if(MYDEBUG) MESSAGE("VSR - load algo : id = " << id <<
4204 ", name = " << hypname.c_str() << ", persistent string = " << hypdata.c_str());
4205 SMESH::SMESH_Hypothesis_var myHyp;
4207 try { // protect persistence mechanism against exceptions
4208 myHyp = this->createHypothesis( hypname.c_str(), libname.c_str() );
4211 INFOS( "Exception during hypothesis creation" );
4214 SMESH_Hypothesis_i* myImpl = dynamic_cast<SMESH_Hypothesis_i*>( GetServant( myHyp ).in() );
4216 //myImpl->LoadFrom( hypdata.c_str() );
4217 hypDataList.push_back( make_pair( myImpl, hypdata ));
4218 CORBA::String_var iorString = GetORB()->object_to_string( myHyp );
4219 int newId = myStudyContext->findId( iorString.in() );
4220 myStudyContext->mapOldToNew( id, newId );
4223 if(MYDEBUG) MESSAGE( "VSR - SMESH_Gen::Load - can't get servant" );
4227 // close algorithms root HDF group
4228 aTopGroup->CloseOnDisk();
4232 // --> the rest groups should be meshes
4233 for ( int i = 0; i < aNbGroups; i++ ) {
4234 // identify next group
4235 char meshName[ HDF_NAME_MAX_LEN+1 ];
4236 aFile->InternalObjectIndentify( i, meshName );
4238 if ( string( meshName ).substr( 0, 4 ) == string( "Mesh" ) ) {
4240 int id = atoi( string( meshName ).substr( 4 ).c_str() );
4244 // open mesh HDF group
4245 aTopGroup = new HDFgroup( meshName, aFile );
4246 aTopGroup->OpenOnDisk();
4248 // get number of child HDF objects
4249 int aNbObjects = aTopGroup->nInternalObjects();
4250 if ( aNbObjects > 0 ) {
4252 if(MYDEBUG) MESSAGE( "VSR - load mesh : id = " << id );
4253 SMESH::SMESH_Mesh_var myNewMesh = this->createMesh();
4254 SMESH_Mesh_i* myNewMeshImpl = dynamic_cast<SMESH_Mesh_i*>( GetServant( myNewMesh ).in() );
4255 if ( !myNewMeshImpl )
4257 meshGroupList.push_back( make_pair( myNewMeshImpl, aTopGroup ));
4259 CORBA::String_var iorString = GetORB()->object_to_string( myNewMesh );
4260 int newId = myStudyContext->findId( iorString.in() );
4261 myStudyContext->mapOldToNew( id, newId );
4264 // try to read and set auto color flag
4265 char aMeshAutoColorName[ 30 ];
4266 sprintf( aMeshAutoColorName, "AutoColorMesh %d", id);
4267 if( aTopGroup->ExistInternalObject( aMeshAutoColorName ) )
4269 aDataset = new HDFdataset( aMeshAutoColorName, aTopGroup );
4270 aDataset->OpenOnDisk();
4271 size = aDataset->GetSize();
4272 int* anAutoColor = new int[ size ];
4273 aDataset->ReadFromDisk( anAutoColor );
4274 aDataset->CloseOnDisk();
4275 myNewMeshImpl->GetImpl().SetAutoColor( (bool)anAutoColor[0] );
4276 delete [] anAutoColor;
4279 // try to read and set reference to shape
4280 GEOM::GEOM_Object_var aShapeObject;
4281 if ( aTopGroup->ExistInternalObject( "Ref on shape" ) ) {
4282 // load mesh "Ref on shape" - it's an entry to SObject
4283 aDataset = new HDFdataset( "Ref on shape", aTopGroup );
4284 aDataset->OpenOnDisk();
4285 size = aDataset->GetSize();
4286 char* refFromFile = new char[ size ];
4287 aDataset->ReadFromDisk( refFromFile );
4288 aDataset->CloseOnDisk();
4289 if ( strlen( refFromFile ) > 0 ) {
4290 SALOMEDS::SObject_wrap shapeSO = myCurrentStudy->FindObjectID( refFromFile );
4292 // Make sure GEOM data are loaded first
4293 //loadGeomData( shapeSO->GetFatherComponent() );
4295 CORBA::Object_var shapeObject = SObjectToObject( shapeSO );
4296 if ( !CORBA::is_nil( shapeObject ) ) {
4297 aShapeObject = GEOM::GEOM_Object::_narrow( shapeObject );
4298 if ( !aShapeObject->_is_nil() )
4299 myNewMeshImpl->SetShape( aShapeObject );
4302 delete [] refFromFile;
4305 // issue 20918. Restore Persistent Id of SMESHDS_Mesh
4306 if( aTopGroup->ExistInternalObject( "meshPersistentId" ) )
4308 aDataset = new HDFdataset( "meshPersistentId", aTopGroup );
4309 aDataset->OpenOnDisk();
4310 size = aDataset->GetSize();
4311 int* meshPersistentId = new int[ size ];
4312 aDataset->ReadFromDisk( meshPersistentId );
4313 aDataset->CloseOnDisk();
4314 myNewMeshImpl->GetImpl().GetMeshDS()->SetPersistentId( *meshPersistentId );
4315 delete [] meshPersistentId;
4321 // As all object that can be referred by hypothesis are created,
4322 // we can restore hypothesis data
4324 list< pair< SMESH_Hypothesis_i*, string > >::iterator hyp_data;
4325 for ( hyp_data = hypDataList.begin(); hyp_data != hypDataList.end(); ++hyp_data )
4327 SMESH_Hypothesis_i* hyp = hyp_data->first;
4328 string & data = hyp_data->second;
4329 hyp->LoadFrom( data.c_str() );
4332 // Restore the rest mesh data
4334 list< pair< SMESH_Mesh_i*, HDFgroup* > >::iterator meshi_group;
4335 for ( meshi_group = meshGroupList.begin(); meshi_group != meshGroupList.end(); ++meshi_group )
4337 aTopGroup = meshi_group->second;
4338 SMESH_Mesh_i* myNewMeshImpl = meshi_group->first;
4339 //::SMESH_Mesh& myLocMesh = myNewMeshImpl->GetImpl();
4340 //SMESHDS_Mesh* mySMESHDSMesh = myLocMesh.GetMeshDS();
4342 GEOM::GEOM_Object_var aShapeObject = myNewMeshImpl->GetShapeToMesh();
4343 bool hasData = false;
4346 CORBA::String_var iorString = GetORB()->object_to_string( myNewMeshImpl->_this() );
4347 int newId = myStudyContext->findId( iorString.in() );
4348 int id = myStudyContext->getOldId( newId );
4350 // try to find mesh data dataset
4351 if ( aTopGroup->ExistInternalObject( "Has data" ) ) {
4352 // load mesh "has data" flag
4353 aDataset = new HDFdataset( "Has data", aTopGroup );
4354 aDataset->OpenOnDisk();
4355 size = aDataset->GetSize();
4356 char* strHasData = new char[ size ];
4357 aDataset->ReadFromDisk( strHasData );
4358 aDataset->CloseOnDisk();
4359 if ( strcmp( strHasData, "1") == 0 ) {
4360 // read mesh data from MED file
4361 // myReader.SetMesh( mySMESHDSMesh );
4362 // myReader.SetMeshId( id );
4363 // myReader.Perform();
4366 delete [] strHasData;
4369 // Try to get applied ALGORITHMS (mesh is not cleared by algo addition because
4370 // nodes and elements are not yet put into sub-meshes)
4371 if ( aTopGroup->ExistInternalObject( "Applied Algorithms" ) ) {
4372 aGroup = new HDFgroup( "Applied Algorithms", aTopGroup );
4373 aGroup->OpenOnDisk();
4374 // get number of applied algorithms
4375 int aNbSubObjects = aGroup->nInternalObjects();
4376 if(MYDEBUG) MESSAGE( "VSR - number of applied algos " << aNbSubObjects );
4377 for ( int j = 0; j < aNbSubObjects; j++ ) {
4378 char name_dataset[ HDF_NAME_MAX_LEN+1 ];
4379 aGroup->InternalObjectIndentify( j, name_dataset );
4380 // check if it is an algorithm
4381 if ( string( name_dataset ).substr( 0, 4 ) == string( "Algo" ) ) {
4382 aDataset = new HDFdataset( name_dataset, aGroup );
4383 aDataset->OpenOnDisk();
4384 size = aDataset->GetSize();
4385 char* refFromFile = new char[ size ];
4386 aDataset->ReadFromDisk( refFromFile );
4387 aDataset->CloseOnDisk();
4388 // san - it is impossible to recover applied algorithms using their entries within Load() method
4389 //SALOMEDS::SObject_wrap hypSO = myCurrentStudy->FindObjectID( refFromFile );
4390 //CORBA::Object_var hypObject = SObjectToObject( hypSO );
4391 int id = atoi( refFromFile );
4392 delete [] refFromFile;
4393 string anIOR = myStudyContext->getIORbyOldId( id );
4394 if ( !anIOR.empty() ) {
4395 CORBA::Object_var hypObject = GetORB()->string_to_object( anIOR.c_str() );
4396 if ( !CORBA::is_nil( hypObject ) ) {
4397 SMESH::SMESH_Hypothesis_var anHyp = SMESH::SMESH_Hypothesis::_narrow( hypObject );
4398 if ( !anHyp->_is_nil() && (!aShapeObject->_is_nil()
4399 || !myNewMeshImpl->HasShapeToMesh()) )
4400 myNewMeshImpl->addHypothesis( aShapeObject, anHyp );
4405 aGroup->CloseOnDisk();
4408 // try to get applied hypotheses
4409 if ( aTopGroup->ExistInternalObject( "Applied Hypotheses" ) ) {
4410 aGroup = new HDFgroup( "Applied Hypotheses", aTopGroup );
4411 aGroup->OpenOnDisk();
4412 // get number of applied hypotheses
4413 int aNbSubObjects = aGroup->nInternalObjects();
4414 for ( int j = 0; j < aNbSubObjects; j++ ) {
4415 char name_dataset[ HDF_NAME_MAX_LEN+1 ];
4416 aGroup->InternalObjectIndentify( j, name_dataset );
4417 // check if it is a hypothesis
4418 if ( string( name_dataset ).substr( 0, 3 ) == string( "Hyp" ) ) {
4419 aDataset = new HDFdataset( name_dataset, aGroup );
4420 aDataset->OpenOnDisk();
4421 size = aDataset->GetSize();
4422 char* refFromFile = new char[ size ];
4423 aDataset->ReadFromDisk( refFromFile );
4424 aDataset->CloseOnDisk();
4425 // san - it is impossible to recover applied hypotheses using their entries within Load() method
4426 //SALOMEDS::SObject_wrap hypSO = myCurrentStudy->FindObjectID( refFromFile );
4427 //CORBA::Object_var hypObject = SObjectToObject( hypSO );
4428 int id = atoi( refFromFile );
4429 delete [] refFromFile;
4430 string anIOR = myStudyContext->getIORbyOldId( id );
4431 if ( !anIOR.empty() ) {
4432 CORBA::Object_var hypObject = GetORB()->string_to_object( anIOR.c_str() );
4433 if ( !CORBA::is_nil( hypObject ) ) {
4434 SMESH::SMESH_Hypothesis_var anHyp = SMESH::SMESH_Hypothesis::_narrow( hypObject );
4435 if ( !anHyp->_is_nil() && (!aShapeObject->_is_nil()
4436 || !myNewMeshImpl->HasShapeToMesh()) )
4437 myNewMeshImpl->addHypothesis( aShapeObject, anHyp );
4442 aGroup->CloseOnDisk();
4445 // --> try to find SUB-MESHES containers for each type of submesh
4446 for ( int j = GetSubMeshOnVertexTag(); j <= GetSubMeshOnCompoundTag(); j++ ) {
4447 const char* name_meshgroup;
4448 if ( j == GetSubMeshOnVertexTag() )
4449 name_meshgroup = "SubMeshes On Vertex";
4450 else if ( j == GetSubMeshOnEdgeTag() )
4451 name_meshgroup = "SubMeshes On Edge";
4452 else if ( j == GetSubMeshOnWireTag() )
4453 name_meshgroup = "SubMeshes On Wire";
4454 else if ( j == GetSubMeshOnFaceTag() )
4455 name_meshgroup = "SubMeshes On Face";
4456 else if ( j == GetSubMeshOnShellTag() )
4457 name_meshgroup = "SubMeshes On Shell";
4458 else if ( j == GetSubMeshOnSolidTag() )
4459 name_meshgroup = "SubMeshes On Solid";
4460 else if ( j == GetSubMeshOnCompoundTag() )
4461 name_meshgroup = "SubMeshes On Compound";
4463 // try to get submeshes container HDF group
4464 if ( aTopGroup->ExistInternalObject( name_meshgroup ) ) {
4465 // open submeshes containers HDF group
4466 aGroup = new HDFgroup( name_meshgroup, aTopGroup );
4467 aGroup->OpenOnDisk();
4469 // get number of submeshes
4470 int aNbSubMeshes = aGroup->nInternalObjects();
4471 for ( int k = 0; k < aNbSubMeshes; k++ ) {
4473 char name_submeshgroup[ HDF_NAME_MAX_LEN+1 ];
4474 aGroup->InternalObjectIndentify( k, name_submeshgroup );
4475 if ( strncmp( name_submeshgroup, "SubMesh", 7 ) == 0 ) {
4476 // --> get submesh id
4477 int subid = atoi( name_submeshgroup + 7 );
4480 // open submesh HDF group
4481 aSubGroup = new HDFgroup( name_submeshgroup, aGroup );
4482 aSubGroup->OpenOnDisk();
4484 // try to read and set reference to subshape
4485 GEOM::GEOM_Object_var aSubShapeObject;
4486 SMESH::SMESH_subMesh_var aSubMesh;
4488 if ( aSubGroup->ExistInternalObject( "Ref on shape" ) ) {
4489 // load submesh "Ref on shape" - it's an entry to SObject
4490 aDataset = new HDFdataset( "Ref on shape", aSubGroup );
4491 aDataset->OpenOnDisk();
4492 size = aDataset->GetSize();
4493 char* refFromFile = new char[ size ];
4494 aDataset->ReadFromDisk( refFromFile );
4495 aDataset->CloseOnDisk();
4496 if ( strlen( refFromFile ) > 0 ) {
4497 SALOMEDS::SObject_wrap subShapeSO = myCurrentStudy->FindObjectID( refFromFile );
4498 CORBA::Object_var subShapeObject = SObjectToObject( subShapeSO );
4499 if ( !CORBA::is_nil( subShapeObject ) ) {
4500 aSubShapeObject = GEOM::GEOM_Object::_narrow( subShapeObject );
4501 if ( !aSubShapeObject->_is_nil() )
4502 aSubMesh = SMESH::SMESH_subMesh::_duplicate
4503 ( myNewMeshImpl->createSubMesh( aSubShapeObject ) );
4504 if ( aSubMesh->_is_nil() )
4506 string iorSubString = GetORB()->object_to_string( aSubMesh );
4507 int newSubId = myStudyContext->findId( iorSubString );
4508 myStudyContext->mapOldToNew( subid, newSubId );
4513 if ( aSubMesh->_is_nil() )
4516 // try to get applied algorithms
4517 if ( aSubGroup->ExistInternalObject( "Applied Algorithms" ) ) {
4518 // open "applied algorithms" HDF group
4519 aSubSubGroup = new HDFgroup( "Applied Algorithms", aSubGroup );
4520 aSubSubGroup->OpenOnDisk();
4521 // get number of applied algorithms
4522 int aNbSubObjects = aSubSubGroup->nInternalObjects();
4523 for ( int l = 0; l < aNbSubObjects; l++ ) {
4524 char name_dataset[ HDF_NAME_MAX_LEN+1 ];
4525 aSubSubGroup->InternalObjectIndentify( l, name_dataset );
4526 // check if it is an algorithm
4527 if ( strncmp( name_dataset, "Algo", 4 ) == 0 ) {
4528 aDataset = new HDFdataset( name_dataset, aSubSubGroup );
4529 aDataset->OpenOnDisk();
4530 size = aDataset->GetSize();
4531 char* refFromFile = new char[ size ];
4532 aDataset->ReadFromDisk( refFromFile );
4533 aDataset->CloseOnDisk();
4535 int id = atoi( refFromFile );
4536 string anIOR = myStudyContext->getIORbyOldId( id );
4537 if ( !anIOR.empty() ) {
4538 CORBA::Object_var hypObject = GetORB()->string_to_object( anIOR.c_str() );
4539 if ( !CORBA::is_nil( hypObject ) ) {
4540 SMESH::SMESH_Hypothesis_var anHyp = SMESH::SMESH_Hypothesis::_narrow( hypObject );
4541 if ( !anHyp->_is_nil() && !aShapeObject->_is_nil() )
4542 myNewMeshImpl->addHypothesis( aSubShapeObject, anHyp );
4547 // close "applied algorithms" HDF group
4548 aSubSubGroup->CloseOnDisk();
4551 // try to get applied hypotheses
4552 if ( aSubGroup->ExistInternalObject( "Applied Hypotheses" ) ) {
4553 // open "applied hypotheses" HDF group
4554 aSubSubGroup = new HDFgroup( "Applied Hypotheses", aSubGroup );
4555 aSubSubGroup->OpenOnDisk();
4556 // get number of applied hypotheses
4557 int aNbSubObjects = aSubSubGroup->nInternalObjects();
4558 for ( int l = 0; l < aNbSubObjects; l++ ) {
4559 char name_dataset[ HDF_NAME_MAX_LEN+1 ];
4560 aSubSubGroup->InternalObjectIndentify( l, name_dataset );
4561 // check if it is a hypothesis
4562 if ( string( name_dataset ).substr( 0, 3 ) == string( "Hyp" ) ) {
4563 aDataset = new HDFdataset( name_dataset, aSubSubGroup );
4564 aDataset->OpenOnDisk();
4565 size = aDataset->GetSize();
4566 char* refFromFile = new char[ size ];
4567 aDataset->ReadFromDisk( refFromFile );
4568 aDataset->CloseOnDisk();
4570 int id = atoi( refFromFile );
4571 string anIOR = myStudyContext->getIORbyOldId( id );
4572 if ( !anIOR.empty() ) {
4573 CORBA::Object_var hypObject = GetORB()->string_to_object( anIOR.c_str() );
4574 if ( !CORBA::is_nil( hypObject ) ) {
4575 SMESH::SMESH_Hypothesis_var anHyp = SMESH::SMESH_Hypothesis::_narrow( hypObject );
4576 if ( !anHyp->_is_nil() && !aShapeObject->_is_nil() )
4577 myNewMeshImpl->addHypothesis( aSubShapeObject, anHyp );
4582 // close "APPLIED HYPOTHESES" hdf group
4583 aSubSubGroup->CloseOnDisk();
4586 // close SUB-MESH hdf group
4587 aSubGroup->CloseOnDisk();
4590 // close SUB-MESHES containers hdf group
4591 aGroup->CloseOnDisk();
4595 // try to get GROUPS
4596 for ( int ii = GetNodeGroupsTag(); ii <= GetBallElementsGroupsTag(); ii++ ) {
4597 char name_group[ 30 ];
4598 if ( ii == GetNodeGroupsTag() )
4599 strcpy( name_group, "Groups of Nodes" );
4600 else if ( ii == GetEdgeGroupsTag() )
4601 strcpy( name_group, "Groups of Edges" );
4602 else if ( ii == GetFaceGroupsTag() )
4603 strcpy( name_group, "Groups of Faces" );
4604 else if ( ii == GetVolumeGroupsTag() )
4605 strcpy( name_group, "Groups of Volumes" );
4606 else if ( ii == Get0DElementsGroupsTag() )
4607 strcpy( name_group, "Groups of 0D Elements" );
4608 else if ( ii == GetBallElementsGroupsTag() )
4609 strcpy( name_group, "Groups of Balls" );
4611 if ( aTopGroup->ExistInternalObject( name_group ) ) {
4612 aGroup = new HDFgroup( name_group, aTopGroup );
4613 aGroup->OpenOnDisk();
4614 // get number of groups
4615 int aNbSubObjects = aGroup->nInternalObjects();
4616 for ( int j = 0; j < aNbSubObjects; j++ ) {
4617 char name_dataset[ HDF_NAME_MAX_LEN+1 ];
4618 aGroup->InternalObjectIndentify( j, name_dataset );
4619 // check if it is an group
4620 if ( string( name_dataset ).substr( 0, 5 ) == string( "Group" ) ) {
4622 int subid = atoi( string( name_dataset ).substr( 5 ).c_str() );
4625 aDataset = new HDFdataset( name_dataset, aGroup );
4626 aDataset->OpenOnDisk();
4628 // Retrieve actual group name
4629 size = aDataset->GetSize();
4630 char* nameFromFile = new char[ size ];
4631 aDataset->ReadFromDisk( nameFromFile );
4632 aDataset->CloseOnDisk();
4634 // Try to find a shape reference
4635 TopoDS_Shape aShape;
4636 char aRefName[ 30 ];
4637 sprintf( aRefName, "Ref on shape %d", subid);
4638 if ( aGroup->ExistInternalObject( aRefName ) ) {
4639 // load mesh "Ref on shape" - it's an entry to SObject
4640 aDataset = new HDFdataset( aRefName, aGroup );
4641 aDataset->OpenOnDisk();
4642 size = aDataset->GetSize();
4643 char* refFromFile = new char[ size ];
4644 aDataset->ReadFromDisk( refFromFile );
4645 aDataset->CloseOnDisk();
4646 if ( strlen( refFromFile ) > 0 ) {
4647 SALOMEDS::SObject_wrap shapeSO = myCurrentStudy->FindObjectID( refFromFile );
4648 CORBA::Object_var shapeObject = SObjectToObject( shapeSO );
4649 if ( !CORBA::is_nil( shapeObject ) ) {
4650 aShapeObject = GEOM::GEOM_Object::_narrow( shapeObject );
4651 if ( !aShapeObject->_is_nil() )
4652 aShape = GeomObjectToShape( aShapeObject );
4656 // Try to read a filter of SMESH_GroupOnFilter
4657 SMESH::Filter_var filter;
4658 SMESH_PredicatePtr predicate;
4659 std::string hdfGrpName = "Filter " + SMESH_Comment(subid);
4660 if ( aGroup->ExistInternalObject( hdfGrpName.c_str() ))
4662 aDataset = new HDFdataset( hdfGrpName.c_str(), aGroup );
4663 aDataset->OpenOnDisk();
4664 size = aDataset->GetSize();
4665 char* persistStr = new char[ size ];
4666 aDataset->ReadFromDisk( persistStr );
4667 aDataset->CloseOnDisk();
4668 if ( strlen( persistStr ) > 0 ) {
4669 filter = SMESH_GroupOnFilter_i::StringToFilter( persistStr );
4670 predicate = SMESH_GroupOnFilter_i::GetPredicate( filter );
4674 // Create group servant
4675 SMESH::ElementType type = (SMESH::ElementType)(ii - GetNodeGroupsTag() + 1);
4676 SMESH::SMESH_GroupBase_var aNewGroup = SMESH::SMESH_GroupBase::_duplicate
4677 ( myNewMeshImpl->createGroup( type, nameFromFile, aShape, predicate ) );
4678 // Obtain a SMESHDS_Group object
4679 if ( aNewGroup->_is_nil() )
4682 string iorSubString = GetORB()->object_to_string( aNewGroup );
4683 int newSubId = myStudyContext->findId( iorSubString );
4684 myStudyContext->mapOldToNew( subid, newSubId );
4686 SMESH_GroupBase_i* aGroupImpl = SMESH::DownCast< SMESH_GroupBase_i*>( aNewGroup );
4690 if ( SMESH_GroupOnFilter_i* aFilterGroup =
4691 dynamic_cast< SMESH_GroupOnFilter_i*>( aGroupImpl ))
4693 aFilterGroup->SetFilter( filter );
4694 filter->UnRegister();
4696 SMESHDS_GroupBase* aGroupBaseDS = aGroupImpl->GetGroupDS();
4697 if ( !aGroupBaseDS )
4700 aGroupBaseDS->SetStoreName( name_dataset );
4703 // Read color of the group
4704 char aGroupColorName[ 30 ];
4705 sprintf( aGroupColorName, "ColorGroup %d", subid);
4706 if ( aGroup->ExistInternalObject( aGroupColorName ) )
4708 aDataset = new HDFdataset( aGroupColorName, aGroup );
4709 aDataset->OpenOnDisk();
4710 size = aDataset->GetSize();
4711 double* anRGB = new double[ size ];
4712 aDataset->ReadFromDisk( anRGB );
4713 aDataset->CloseOnDisk();
4714 Quantity_Color aColor( anRGB[0], anRGB[1], anRGB[2], Quantity_TOC_RGB );
4715 aGroupBaseDS->SetColor( aColor );
4718 // Fill group with contents from MED file
4719 // SMESHDS_Group* aGrp = dynamic_cast<SMESHDS_Group*>( aGroupBaseDS );
4721 // myReader.GetGroup( aGrp );
4724 aGroup->CloseOnDisk();
4728 // instead of reading mesh data, we read only brief information of all
4729 // objects: mesh, groups, sub-meshes (issue 0021208 )
4732 SMESH_PreMeshInfo::LoadFromFile( myNewMeshImpl, id,
4733 meshfile.ToCString(), filename.ToCString(),
4737 // read Sub-Mesh ORDER if any
4738 if( aTopGroup->ExistInternalObject( "Mesh Order" ) ) {
4739 aDataset = new HDFdataset( "Mesh Order", aTopGroup );
4740 aDataset->OpenOnDisk();
4741 size = aDataset->GetSize();
4742 int* smIDs = new int[ size ];
4743 aDataset->ReadFromDisk( smIDs );
4744 aDataset->CloseOnDisk();
4745 TListOfListOfInt anOrderIds;
4746 anOrderIds.push_back( TListOfInt() );
4747 for ( int i = 0; i < size; i++ )
4748 if ( smIDs[ i ] < 0 ) // is separator
4749 anOrderIds.push_back( TListOfInt() );
4751 anOrderIds.back().push_back(smIDs[ i ]);
4753 myNewMeshImpl->GetImpl().SetMeshOrder( anOrderIds );
4757 // update hyps needing full mesh data restored (issue 20918)
4758 for ( hyp_data = hypDataList.begin(); hyp_data != hypDataList.end(); ++hyp_data )
4760 SMESH_Hypothesis_i* hyp = hyp_data->first;
4761 hyp->UpdateAsMeshesRestored();
4764 // notify algos on completed restoration to set sub-mesh event listeners
4765 for ( meshi_group = meshGroupList.begin(); meshi_group != meshGroupList.end(); ++meshi_group )
4767 SMESH_Mesh_i* myNewMeshImpl = meshi_group->first;
4768 ::SMESH_Mesh& myLocMesh = myNewMeshImpl->GetImpl();
4770 TopoDS_Shape myLocShape;
4771 if(myLocMesh.HasShapeToMesh())
4772 myLocShape = myLocMesh.GetShapeToMesh();
4774 myLocShape = SMESH_Mesh::PseudoShape();
4776 myLocMesh.GetSubMesh(myLocShape)->
4777 ComputeStateEngine (SMESH_subMesh::SUBMESH_RESTORED);
4782 aTopGroup->CloseOnDisk();
4785 aFile->CloseOnDisk();
4788 // Remove temporary files created from the stream
4791 SMESH_File meshFile( meshfile.ToCString() );
4792 if ( !meshFile ) // no meshfile exists
4794 SALOMEDS_Tool::RemoveTemporaryFiles( tmpDir.ToCString(), aFileSeq.in(), true );
4798 Engines::Container_var container = GetContainerRef();
4799 if ( Engines_Container_i* container_i = SMESH::DownCast<Engines_Container_i*>( container ))
4801 container_i->registerTemporaryFile( filename.ToCString() );
4802 container_i->registerTemporaryFile( meshfile.ToCString() );
4803 container_i->registerTemporaryFile( tmpDir.ToCString() );
4807 pd << ""; // prevent optimizing pd out
4809 // creation of tree nodes for all data objects in the study
4810 // to support tree representation customization and drag-n-drop:
4811 SALOMEDS::Study_var study = theComponent->GetStudy();
4812 SALOMEDS::UseCaseBuilder_wrap useCaseBuilder = study->GetUseCaseBuilder();
4813 if ( !useCaseBuilder->IsUseCaseNode( theComponent ) ) {
4814 useCaseBuilder->SetRootCurrent();
4815 useCaseBuilder->Append( theComponent ); // component object is added as the top level item
4816 SALOMEDS::ChildIterator_wrap it = study->NewChildIterator( theComponent );
4817 for (it->InitEx(true); it->More(); it->Next()) {
4818 useCaseBuilder->AppendTo( it->Value()->GetFather(), it->Value() );
4822 INFOS( "SMESH_Gen_i::Load completed" );
4826 //=============================================================================
4828 * SMESH_Gen_i::LoadASCII
4830 * Load SMESH module's data in ASCII format
4832 //=============================================================================
4834 bool SMESH_Gen_i::LoadASCII( SALOMEDS::SComponent_ptr theComponent,
4835 const SALOMEDS::TMPFile& theStream,
4837 bool isMultiFile ) {
4838 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::LoadASCII" );
4839 return Load( theComponent, theStream, theURL, isMultiFile );
4841 //before call main ::Load method it's need for decipher text format to
4842 //binary ( "|xx" => x' )
4843 int size = theStream.length();
4844 if ( int((size / 3 )*3) != size ) //error size of buffer
4847 int real_size = int(size / 3);
4849 _CORBA_Octet* buffer = new _CORBA_Octet[real_size];
4853 for ( int i = 0; i < real_size; i++ )
4855 memcpy( &(tmp[0]), &(theStream[i*3+1]), 2 );
4856 sscanf( tmp, "%x", &c );
4857 sprintf( (char*)&(buffer[i]), "%c", (char)c );
4860 SALOMEDS::TMPFile_var aRealStreamFile = new SALOMEDS::TMPFile(real_size, real_size, buffer, 1);
4862 return Load( theComponent, *(aRealStreamFile._retn()), theURL, isMultiFile );
4865 //=============================================================================
4867 * SMESH_Gen_i::Close
4869 * Clears study-connected data when it is closed
4871 //=============================================================================
4873 void SMESH_Gen_i::Close( SALOMEDS::SComponent_ptr theComponent )
4875 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Close" );
4877 // set correct current study
4878 SALOMEDS::Study_var study = theComponent->GetStudy();
4879 if ( study->StudyId() != GetCurrentStudyID())
4880 setCurrentStudy( study, /*IsBeingClosed=*/true );
4882 // Clear study contexts data
4883 int studyId = GetCurrentStudyID();
4884 if ( myStudyContextMap.find( studyId ) != myStudyContextMap.end() ) {
4885 delete myStudyContextMap[ studyId ];
4886 myStudyContextMap.erase( studyId );
4889 // remove the tmp files meshes are loaded from
4890 SMESH_PreMeshInfo::RemoveStudyFiles_TMP_METHOD( theComponent );
4892 myCurrentStudy = SALOMEDS::Study::_nil();
4896 //=============================================================================
4898 * SMESH_Gen_i::ComponentDataType
4900 * Get component data type
4902 //=============================================================================
4904 char* SMESH_Gen_i::ComponentDataType()
4906 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::ComponentDataType" );
4907 return CORBA::string_dup( "SMESH" );
4911 //=============================================================================
4913 * SMESH_Gen_i::IORToLocalPersistentID
4915 * Transform data from transient form to persistent
4917 //=============================================================================
4919 char* SMESH_Gen_i::IORToLocalPersistentID( SALOMEDS::SObject_ptr /*theSObject*/,
4920 const char* IORString,
4921 CORBA::Boolean /*isMultiFile*/,
4922 CORBA::Boolean /*isASCII*/ )
4924 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::IORToLocalPersistentID" );
4925 StudyContext* myStudyContext = GetCurrentStudyContext();
4927 if ( myStudyContext && strcmp( IORString, "" ) != 0 ) {
4928 int anId = myStudyContext->findId( IORString );
4930 if(MYDEBUG) MESSAGE( "VSR " << anId )
4932 sprintf( strId, "%d", anId );
4933 return CORBA::string_dup( strId );
4936 return CORBA::string_dup( "" );
4939 //=============================================================================
4941 * SMESH_Gen_i::LocalPersistentIDToIOR
4943 * Transform data from persistent form to transient
4945 //=============================================================================
4947 char* SMESH_Gen_i::LocalPersistentIDToIOR( SALOMEDS::SObject_ptr /*theSObject*/,
4948 const char* aLocalPersistentID,
4949 CORBA::Boolean /*isMultiFile*/,
4950 CORBA::Boolean /*isASCII*/ )
4952 if(MYDEBUG) MESSAGE( "SMESH_Gen_i::LocalPersistentIDToIOR(): id = " << aLocalPersistentID );
4953 StudyContext* myStudyContext = GetCurrentStudyContext();
4955 if ( myStudyContext && strcmp( aLocalPersistentID, "" ) != 0 ) {
4956 int anId = atoi( aLocalPersistentID );
4957 return CORBA::string_dup( myStudyContext->getIORbyOldId( anId ).c_str() );
4959 return CORBA::string_dup( "" );
4962 //=======================================================================
4963 //function : RegisterObject
4965 //=======================================================================
4967 int SMESH_Gen_i::RegisterObject(CORBA::Object_ptr theObject)
4969 StudyContext* myStudyContext = GetCurrentStudyContext();
4970 if ( myStudyContext && !CORBA::is_nil( theObject )) {
4971 if (GetCurrentStudyID() == myImportedStudyId)
4972 myImportedStudyChanged = true;
4973 CORBA::String_var iorString = GetORB()->object_to_string( theObject );
4974 return myStudyContext->addObject( string( iorString.in() ) );
4979 //================================================================================
4981 * \brief Return id of registered object
4982 * \param theObject - the Object
4983 * \retval int - Object id
4985 //================================================================================
4987 CORBA::Long SMESH_Gen_i::GetObjectId(CORBA::Object_ptr theObject)
4989 StudyContext* myStudyContext = GetCurrentStudyContext();
4990 if ( myStudyContext && !CORBA::is_nil( theObject )) {
4991 string iorString = GetORB()->object_to_string( theObject );
4992 return myStudyContext->findId( iorString );
4997 //=============================================================================
4999 * SMESH_Gen_i::SetName
5001 * Set a new object name
5003 //=============================================================================
5004 void SMESH_Gen_i::SetName(const char* theIOR,
5005 const char* theName)
5007 if ( theIOR && strcmp( theIOR, "" ) ) {
5008 CORBA::Object_var anObject = GetORB()->string_to_object( theIOR );
5009 SALOMEDS::SObject_wrap aSO = ObjectToSObject( myCurrentStudy, anObject );
5010 if ( !aSO->_is_nil() ) {
5011 SetName( aSO, theName );
5016 int SMESH_Gen_i::GetCurrentStudyID()
5018 return myCurrentStudy->_is_nil() || myCurrentStudy->_non_existent() ? -1 : myCurrentStudy->StudyId();
5021 // Version information
5022 char* SMESH_Gen_i::getVersion()
5024 #if SMESH_DEVELOPMENT
5025 return CORBA::string_dup(SMESH_VERSION_STR"dev");
5027 return CORBA::string_dup(SMESH_VERSION_STR);
5031 //=================================================================================
5032 // function : Move()
5033 // purpose : Moves objects to the specified position.
5034 // Is used in the drag-n-drop functionality.
5035 //=================================================================================
5036 void SMESH_Gen_i::Move( const SMESH::sobject_list& what,
5037 SALOMEDS::SObject_ptr where,
5040 if ( CORBA::is_nil( where ) ) return;
5042 SALOMEDS::Study_var study = where->GetStudy();
5043 SALOMEDS::StudyBuilder_var studyBuilder = study->NewBuilder();
5044 SALOMEDS::UseCaseBuilder_var useCaseBuilder = study->GetUseCaseBuilder();
5045 SALOMEDS::SComponent_var father = where->GetFatherComponent();
5046 std::string dataType = father->ComponentDataType();
5047 if ( dataType != "SMESH" ) return; // not a SMESH component
5049 SALOMEDS::SObject_var objAfter;
5050 if ( row >= 0 && useCaseBuilder->HasChildren( where ) ) {
5051 // insert at given row -> find insertion position
5052 SALOMEDS::UseCaseIterator_var useCaseIt = useCaseBuilder->GetUseCaseIterator( where );
5054 for ( i = 0; i < row && useCaseIt->More(); i++, useCaseIt->Next() );
5055 if ( i == row && useCaseIt->More() ) {
5056 objAfter = useCaseIt->Value();
5060 for ( int i = 0; i < what.length(); i++ ) {
5061 SALOMEDS::SObject_var sobj = what[i];
5062 if ( CORBA::is_nil( sobj ) ) continue; // skip bad object
5063 // insert the object to the use case tree
5064 if ( !CORBA::is_nil( objAfter ) )
5065 useCaseBuilder->InsertBefore( sobj, objAfter ); // insert at given row
5067 useCaseBuilder->AppendTo( where, sobj ); // append to the end of list
5070 //=================================================================================
5071 // function : IsApplicable
5072 // purpose : Return true if algorithm can be applied
5073 //=================================================================================
5074 CORBA::Boolean SMESH_Gen_i::IsApplicable ( const char* theAlgoType,
5075 const char* theLibName,
5076 GEOM::GEOM_Object_ptr theGeomObject,
5077 CORBA::Boolean toCheckAll)
5081 std::string aPlatformLibName;
5082 typedef GenericHypothesisCreator_i* (*GetHypothesisCreator)(const char*);
5083 GenericHypothesisCreator_i* aCreator = getHypothesisCreator(theAlgoType, theLibName, aPlatformLibName);
5086 TopoDS_Shape shape = GeomObjectToShape( theGeomObject );
5087 if ( !shape.IsNull() )
5088 return aCreator->IsApplicable( shape, toCheckAll );
5095 SMESH_CATCH( SMESH::doNothing );
5099 //=================================================================================
5100 // function : importData
5101 // purpose : imports mesh data file (the med one) into the SMESH internal data structure
5102 //=================================================================================
5103 Engines::ListOfIdentifiers* SMESH_Gen_i::importData(CORBA::Long studyId,
5104 Engines::DataContainer_ptr data,
5105 const Engines::ListOfOptions& options)
5107 Engines::ListOfIdentifiers_var aResultIds = new Engines::ListOfIdentifiers;
5108 list<string> aResultList;
5110 CORBA::Object_var aSMObject = myNS->Resolve( "/myStudyManager" );
5111 SALOMEDS::StudyManager_var aStudyManager = SALOMEDS::StudyManager::_narrow( aSMObject );
5112 SALOMEDS::Study_var aStudy = aStudyManager->GetStudyByID( studyId );
5113 SetCurrentStudy(aStudy);
5115 // load and store temporary imported file
5116 string aFileName = Kernel_Utils::GetTmpFileName();
5117 aFileName += string(".") + data->extension();
5118 Engines::TMPFile* aFileStream = data->get();
5119 const char *aBuffer = (const char*)aFileStream->NP_data();
5121 std::ofstream aFile(aFileName.c_str(), std::ios::binary);
5123 std::ofstream aFile(aFileName.c_str());
5125 aFile.write(aBuffer, aFileStream->length());
5128 // Retrieve mesh names from the file
5129 DriverMED_R_SMESHDS_Mesh aReader;
5130 aReader.SetFile( aFileName );
5131 aReader.SetMeshId(-1);
5132 Driver_Mesh::Status aStatus;
5133 list<string> aNames = aReader.GetMeshNames(aStatus);
5134 SMESH::mesh_array_var aResult = new SMESH::mesh_array();
5135 SMESH::DriverMED_ReadStatus aStatus2 = (SMESH::DriverMED_ReadStatus)aStatus;
5136 if (aStatus2 == SMESH::DRS_OK) {
5137 // Iterate through all meshes and create mesh objects
5138 for ( list<string>::iterator it = aNames.begin(); it != aNames.end(); it++ ) {
5140 SMESH::SMESH_Mesh_var mesh = createMesh();
5142 // publish mesh in the study
5143 SALOMEDS::SObject_var aSO;
5144 if (CanPublishInStudy(mesh)) {
5145 aSO = PublishMesh(aStudy, mesh.in(), (*it).c_str());
5146 aResultList.push_back(aSO->GetID());
5148 // Read mesh data (groups are published automatically by ImportMEDFile())
5149 SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( mesh ).in() );
5150 ASSERT( meshServant );
5151 meshServant->ImportMEDFile( aFileName.c_str(), (*it).c_str() );
5152 //meshServant->GetImpl().GetMeshDS()->Modified();
5155 MESSAGE("Opening MED file problems "<<aFileName.c_str())
5156 return aResultIds._retn();
5159 // remove temporary file
5161 DeleteFileA(aFileName.c_str());
5163 unlink(aFileName.c_str());
5166 if (!aResultList.empty()) {
5167 aResultIds->length(aResultList.size());
5168 list<string>::iterator aListIter = aResultList.begin();
5169 for(int a = 0; aListIter != aResultList.end(); aListIter++, a++)
5170 aResultIds[a] = aListIter->c_str();
5173 myImportedStudyId = studyId;
5174 myImportedStudyChanged = false;
5176 return aResultIds._retn();
5179 //=================================================================================
5180 // function : getModifiedData
5181 // purpose : exports all geometry of this GEOM module into one BRep file
5182 //=================================================================================
5183 Engines::ListOfData* SMESH_Gen_i::getModifiedData(CORBA::Long studyId)
5185 Engines::ListOfData_var aResult = new Engines::ListOfData;
5187 if (!myImportedStudyChanged) {
5188 INFOS("SMESH module data was not changed")
5189 return aResult._retn();
5192 CORBA::Object_var aSMObject = myNS->Resolve("/myStudyManager");
5193 SALOMEDS::StudyManager_var aStudyManager = SALOMEDS::StudyManager::_narrow(aSMObject);
5194 SALOMEDS::Study_var aStudy = aStudyManager->GetStudyByID(studyId);
5195 SetCurrentStudy(aStudy);
5196 SALOMEDS::SComponent_var aComponent = aStudy->FindComponent("SMESH");
5198 if (CORBA::is_nil(aComponent))
5199 return aResult._retn();
5201 std::string aFullPath(Kernel_Utils::GetTmpFileName());
5202 aFullPath += ".med";
5203 StudyContext* myStudyContext = GetCurrentStudyContext();
5205 SALOMEDS::ChildIterator_var anIter = aStudy->NewChildIterator(aComponent); // check only published meshes
5206 int aNumMeshes = 0; // number of meshes in result
5207 for(; anIter->More(); anIter->Next()) {
5208 SALOMEDS::SObject_var aSO = anIter->Value();
5209 CORBA::Object_var anObj = aSO->GetObject();
5210 if (!CORBA::is_nil(anObj)) {
5211 SMESH::SMESH_Mesh_var aCORBAMesh = SMESH::SMESH_Mesh::_narrow(anObj);
5212 if(!aCORBAMesh->_is_nil()) {
5213 SMESH_Mesh_i* myImpl = dynamic_cast<SMESH_Mesh_i*>(GetServant(aCORBAMesh).in());
5216 SMESH_Mesh& aMesh = myImpl->GetImpl();
5217 CORBA::String_var objName = aSO->GetName();
5218 aMesh.ExportMED(aFullPath.c_str(), objName.in(), false, MED::eV2_2, 0);
5224 if (aNumMeshes > 0) { // prepare a container to store files
5225 INFOS("Write "<<aNumMeshes<<" meshes to "<<aFullPath.c_str());
5227 Engines::DataContainer_var aData = (new Engines_DataContainer_i(
5228 aFullPath.c_str(), "", "", true))->_this();
5231 return aResult._retn();
5234 //=============================================================================
5236 * SMESHEngine_factory
5238 * C factory, accessible with dlsym, after dlopen
5240 //=============================================================================
5244 PortableServer::ObjectId* SMESHEngine_factory( CORBA::ORB_ptr orb,
5245 PortableServer::POA_ptr poa,
5246 PortableServer::ObjectId* contId,
5247 const char* instanceName,
5248 const char* interfaceName )
5250 if(MYDEBUG) MESSAGE( "PortableServer::ObjectId* SMESHEngine_factory()" );
5251 if(MYDEBUG) SCRUTE(interfaceName);
5252 SMESH_Gen_i* aSMESHGen = new SMESH_Gen_i(orb, poa, contId, instanceName, interfaceName);
5253 return aSMESHGen->getId() ;