Salome HOME
[GPUSPHGUI] fix errors relating to eCustomControl + offset
[modules/smesh.git] / src / SMESH_I / SMESH_Gen_i.cxx
1 // Copyright (C) 2007-2016  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
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.
10 //
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.
15 //
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
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22 //  File   : SMESH_Gen_i.cxx
23 //  Author : Paul RASCLE, EDF
24 //  Module : SMESH
25
26 #include <BRepPrimAPI_MakeBox.hxx>
27 #include <BRepPrimAPI_MakeCylinder.hxx>
28 #include <BRepPrimAPI_MakeSphere.hxx>
29 #include <BRep_Tool.hxx>
30 #include <OSD.hxx>
31 #include <TColStd_MapOfAsciiString.hxx>
32 #include <TCollection_AsciiString.hxx>
33 #include <TopExp.hxx>
34 #include <TopExp_Explorer.hxx>
35 #include <TopTools_IndexedMapOfShape.hxx>
36 #include <TopTools_ListIteratorOfListOfShape.hxx>
37 #include <TopTools_ListOfShape.hxx>
38 #include <TopTools_MapOfShape.hxx>
39 #include <TopoDS.hxx>
40 #include <TopoDS_CompSolid.hxx>
41 #include <TopoDS_Compound.hxx>
42 #include <TopoDS_Edge.hxx>
43 #include <TopoDS_Face.hxx>
44 #include <TopoDS_Iterator.hxx>
45 #include <TopoDS_Shape.hxx>
46 #include <TopoDS_Shell.hxx>
47 #include <TopoDS_Solid.hxx>
48 #include <TopoDS_Vertex.hxx>
49 #include <TopoDS_Wire.hxx>
50 #include <gp_Pnt.hxx>
51
52
53 #ifdef WIN32
54  #include <windows.h>
55  #include <process.h>
56 #else
57  #include <dlfcn.h>
58  #include <libgen.h> // for basename function
59 #endif
60
61 #ifdef WIN32
62  #define LibHandle HMODULE
63  #define LoadLib( name ) LoadLibrary( name )
64  #define GetProc GetProcAddress
65  #define UnLoadLib( handle ) FreeLibrary( handle );
66 #else // WIN32
67  #define LibHandle void*
68  #ifdef DYNLOAD_LOCAL
69   #define LoadLib( name ) dlopen( name, RTLD_LAZY | RTLD_LOCAL )
70  #else // DYNLOAD_LOCAL
71   #define LoadLib( name ) dlopen( name, RTLD_LAZY | RTLD_GLOBAL )
72  #endif // DYNLOAD_LOCAL
73  #define GetProc dlsym
74  #define UnLoadLib( handle ) dlclose( handle );
75 #endif // WIN32
76
77 #include "SMESH_Gen_i.hxx"
78 #include "SMESH_version.h"
79
80 #include "DriverMED_W_SMESHDS_Mesh.h"
81 #include "DriverMED_R_SMESHDS_Mesh.h"
82 #ifdef WITH_CGNS
83 #include "DriverCGNS_Read.hxx"
84 #endif
85 #include "MED_Factory.hxx"
86 #include "SMDS_EdgePosition.hxx"
87 #include "SMDS_FacePosition.hxx"
88 #include "SMDS_PolyhedralVolumeOfNodes.hxx"
89 #include "SMDS_SetIterator.hxx"
90 #include "SMDS_SpacePosition.hxx"
91 #include "SMDS_VertexPosition.hxx"
92 #include "SMESHDS_Document.hxx"
93 #include "SMESHDS_Group.hxx"
94 #include "SMESHDS_GroupOnGeom.hxx"
95 #include "SMESH_Algo_i.hxx"
96 #include "SMESH_File.hxx"
97 #include "SMESH_Group.hxx"
98 #include "SMESH_Group_i.hxx"
99 #include "SMESH_Hypothesis.hxx"
100 #include "SMESH_Hypothesis_i.hxx"
101 #include "SMESH_Mesh.hxx"
102 #include "SMESH_MeshEditor.hxx"
103 #include "SMESH_Mesh_i.hxx"
104 #include "SMESH_PreMeshInfo.hxx"
105 #include "SMESH_PythonDump.hxx"
106 #include "SMESH_ControlsDef.hxx"
107 #include "SMESH_TryCatch.hxx" // to include after OCC headers!
108
109 #include CORBA_SERVER_HEADER(SMESH_Group)
110 #include CORBA_SERVER_HEADER(SMESH_Filter)
111 #include CORBA_SERVER_HEADER(SMESH_MeshEditor)
112
113
114 #include <GEOM_Client.hxx>
115
116 #include <Basics_Utils.hxx>
117 #include <Basics_DirUtils.hxx>
118 #include <HDFOI.hxx>
119 #include <OpUtil.hxx>
120 #include <SALOMEDS_Tool.hxx>
121 #include <SALOME_Container_i.hxx>
122 #include <SALOME_LifeCycleCORBA.hxx>
123 #include <SALOME_NamingService.hxx>
124 #include <Utils_CorbaException.hxx>
125 #include <Utils_ExceptHandlers.hxx>
126 #include <Utils_SINGLETON.hxx>
127 #include <utilities.h>
128
129 #include CORBA_CLIENT_HEADER(SALOME_ModuleCatalog)
130 #include CORBA_CLIENT_HEADER(SALOME_Session)
131
132 // helpers about SALOME::GenericObj
133 #include <SALOMEDS_wrap.hxx>
134 #include <SALOMEDS_Attributes_wrap.hxx>
135 #include <GEOM_wrap.hxx>
136
137 #include <map>
138 #include <fstream>
139 #include <cstdio>
140 #include <cstdlib>
141
142 using namespace std;
143 using SMESH::TPythonDump;
144 using SMESH::TVar;
145
146 #define NUM_TMP_FILES 2
147
148 #ifdef _DEBUG_
149 static int MYDEBUG = 0;
150 #else
151 static int MYDEBUG = 0;
152 #endif
153
154 // Static variables definition
155 GEOM::GEOM_Gen_var      SMESH_Gen_i::myGeomGen = GEOM::GEOM_Gen::_nil();
156 CORBA::ORB_var          SMESH_Gen_i::myOrb;
157 PortableServer::POA_var SMESH_Gen_i::myPoa;
158 SALOME_NamingService*   SMESH_Gen_i::myNS  = NULL;
159 SALOME_LifeCycleCORBA*  SMESH_Gen_i::myLCC = NULL;
160 SMESH_Gen_i*            SMESH_Gen_i::mySMESHGen = NULL;
161
162
163 const int nbElemPerDiagonal = 10;
164
165 //=============================================================================
166 /*!
167  *  GetServant [ static ]
168  *
169  *  Get servant of the CORBA object
170  */
171 //=============================================================================
172
173 PortableServer::ServantBase_var SMESH_Gen_i::GetServant( CORBA::Object_ptr theObject )
174 {
175   if( CORBA::is_nil( theObject ) || CORBA::is_nil( GetPOA() ) )
176     return NULL;
177   try {
178     PortableServer::Servant aServant = GetPOA()->reference_to_servant( theObject );
179     return aServant;
180   }
181   catch (...) {
182     INFOS( "GetServant - Unknown exception was caught!!!" );
183     return NULL;
184   }
185 }
186
187 //=============================================================================
188 /*!
189  *  SObjectToObject [ static ]
190  *
191  *  Get CORBA object corresponding to the SALOMEDS::SObject
192  */
193 //=============================================================================
194
195 CORBA::Object_var SMESH_Gen_i::SObjectToObject( SALOMEDS::SObject_ptr theSObject )
196 {
197   SALOMEDS::GenericAttribute_wrap anAttr;
198   CORBA::Object_var anObj;
199   if ( !theSObject->_is_nil() ) {
200     try {
201       if( theSObject->FindAttribute( anAttr.inout(), "AttributeIOR" ) ) {
202         SALOMEDS::AttributeIOR_wrap anIOR  = anAttr;
203         CORBA::String_var aValue = anIOR->Value();
204         if( strcmp( aValue, "" ) != 0 )
205           anObj = GetORB()->string_to_object( aValue );
206       }
207     }
208     catch( ... ) {
209       INFOS( "SObjectToObject - Unknown exception was caught!!!" );
210     }
211   }
212   return anObj;
213 }
214
215 //=============================================================================
216 /*!
217  *  GetNS [ static ]
218  *
219  *  Get SALOME_NamingService object
220  */
221 //=============================================================================
222
223 SALOME_NamingService* SMESH_Gen_i::GetNS()
224 {
225   if ( myNS == NULL ) {
226     myNS = SINGLETON_<SALOME_NamingService>::Instance();
227     ASSERT(SINGLETON_<SALOME_NamingService>::IsAlreadyExisting());
228     myNS->init_orb( GetORB() );
229   }
230   return myNS;
231 }
232
233 //=============================================================================
234 /*!
235  *  GetLCC [ static ]
236  *
237  *  Get SALOME_LifeCycleCORBA object
238  */
239 //=============================================================================
240 SALOME_LifeCycleCORBA*  SMESH_Gen_i::GetLCC() {
241   if ( myLCC == NULL ) {
242     myLCC = new SALOME_LifeCycleCORBA( GetNS() );
243   }
244   return myLCC;
245 }
246
247
248 //=============================================================================
249 /*!
250  *  GetGeomEngine [ static ]
251  *
252  *  Get GEOM::GEOM_Gen reference
253  */
254 //=============================================================================
255 GEOM::GEOM_Gen_var SMESH_Gen_i::GetGeomEngine() {
256   //CCRT GEOM::GEOM_Gen_var aGeomEngine =
257   //CCRT   GEOM::GEOM_Gen::_narrow( GetLCC()->FindOrLoad_Component("FactoryServer","GEOM") );
258   //CCRT return aGeomEngine._retn();
259   if(CORBA::is_nil(myGeomGen))
260   {
261     Engines::EngineComponent_ptr temp=GetLCC()->FindOrLoad_Component("FactoryServer","GEOM");
262     myGeomGen=GEOM::GEOM_Gen::_narrow(temp);
263   }
264   return myGeomGen;
265 }
266
267 //=============================================================================
268 /*!
269  *  SMESH_Gen_i::SMESH_Gen_i
270  *
271  *  Default constructor: not for use
272  */
273 //=============================================================================
274
275 SMESH_Gen_i::SMESH_Gen_i()
276 {
277 }
278
279 //=============================================================================
280 /*!
281  *  SMESH_Gen_i::SMESH_Gen_i
282  *
283  *  Standard constructor, used with Container
284  */
285 //=============================================================================
286
287 SMESH_Gen_i::SMESH_Gen_i( CORBA::ORB_ptr            orb,
288                           PortableServer::POA_ptr   poa,
289                           PortableServer::ObjectId* contId,
290                           const char*               instanceName,
291                           const char*               interfaceName )
292   : Engines_Component_i( orb, poa, contId, instanceName, interfaceName )
293 {
294
295   myOrb = CORBA::ORB::_duplicate(orb);
296   myPoa = PortableServer::POA::_duplicate(poa);
297
298   _thisObj = this ;
299   _id = myPoa->activate_object( _thisObj );
300
301   myIsEmbeddedMode = false;
302   myShapeReader = NULL;  // shape reader
303   mySMESHGen = this;
304   myIsHistoricalPythonDump = true;
305   myToForgetMeshDataOnHypModif = false;
306
307   // set it in standalone mode only
308   //OSD::SetSignal( true );
309
310   // 0020605: EDF 1190 SMESH: Display performance. 80 seconds for 52000 cells.
311   // find out mode (embedded or standalone) here else
312   // meshes created before calling SMESH_Client::GetSMESHGen(), which calls
313   // SMESH_Gen_i::SetEmbeddedMode(), have wrong IsEmbeddedMode flag
314   if ( SALOME_NamingService* ns = GetNS() )
315   {
316     CORBA::Object_var obj = ns->Resolve( "/Kernel/Session" );
317     SALOME::Session_var session = SALOME::Session::_narrow( obj ) ;
318     if ( !session->_is_nil() )
319     {
320       CORBA::String_var str_host = session->getHostname();
321       CORBA::Long        s_pid = session->getPID();
322       string my_host = Kernel_Utils::GetHostname();
323 #ifdef WIN32
324       long    my_pid = (long)_getpid();
325 #else
326       long    my_pid = (long) getpid();
327 #endif
328       SetEmbeddedMode( s_pid == my_pid && my_host == str_host.in() );
329     }
330   }
331 }
332
333 //=============================================================================
334 /*!
335  *  SMESH_Gen_i::~SMESH_Gen_i
336  *
337  *  Destructor
338  */
339 //=============================================================================
340
341 SMESH_Gen_i::~SMESH_Gen_i()
342 {
343   // delete hypothesis creators
344   map<string, GenericHypothesisCreator_i*>::iterator itHyp, itHyp2;
345   for (itHyp = myHypCreatorMap.begin(); itHyp != myHypCreatorMap.end(); itHyp++)
346   {
347     // same creator can be mapped under different names
348     GenericHypothesisCreator_i* creator = (*itHyp).second;
349     if ( !creator )
350       continue;
351     delete creator;
352     for (itHyp2 = itHyp; itHyp2 != myHypCreatorMap.end(); itHyp2++)
353       if ( creator == (*itHyp2).second )
354         (*itHyp2).second = 0;
355   }
356   myHypCreatorMap.clear();
357
358   // Clear study contexts data
359   map<int, StudyContext*>::iterator it;
360   for ( it = myStudyContextMap.begin(); it != myStudyContextMap.end(); ++it ) {
361     delete it->second;
362   }
363   myStudyContextMap.clear();
364   // delete shape reader
365   if ( myShapeReader )
366     delete myShapeReader;
367 }
368 //=============================================================================
369 /*!
370  *  SMESH_Gen_i::getHypothesisCreator
371  *
372  *  Get hypothesis creator
373  */
374 //=============================================================================
375 GenericHypothesisCreator_i* SMESH_Gen_i::getHypothesisCreator(const char* theHypName,
376                                                               const char* theLibName,
377                                                               std::string& thePlatformLibName)
378   throw (SALOME::SALOME_Exception)
379 {
380   std::string aPlatformLibName;
381   /* It's Need to translate lib name for WIN32 or X platform */
382   if ( theLibName && theLibName[0] != '\0'  )
383   {
384     int libNameLen = strlen(theLibName);
385     //check for old format "libXXXXXXX.so"
386     if (libNameLen > 7 &&
387         !strncmp( theLibName, "lib", 3 ) &&
388         !strcmp( theLibName+libNameLen-3, ".so" ))
389     {
390       //the old format
391 #if defined(WIN32)
392       aPlatformLibName = std::string( theLibName+3, libNameLen-6 ) + ".dll";
393 #elif defined(__APPLE__)
394       aPlatformLibName = std::string( theLibName, libNameLen-3 ) + ".dylib";
395 #else
396       aPlatformLibName = theLibName;
397 #endif
398     }
399     else
400     {
401       //try to use new format
402 #if defined(WIN32)
403       aPlatformLibName = theLibName;
404       aPlatformLibName += ".dll";
405 #elif defined(__APPLE__)
406       aPlatformLibName = std::string( "lib" ) + std::string( theLibName ) + ".dylib";
407 #else
408       aPlatformLibName = std::string( "lib" ) + std::string( theLibName ) + ".so";
409 #endif
410     }
411   }
412   thePlatformLibName = aPlatformLibName;
413
414   Unexpect aCatch(SALOME_SalomeException);
415   if(MYDEBUG) MESSAGE( "Create Hypothesis <" << theHypName << "> from " << aPlatformLibName);
416
417   typedef GenericHypothesisCreator_i* (*GetHypothesisCreator)(const char* );
418   GenericHypothesisCreator_i* aCreator;
419   try
420   {
421     // check, if creator for this hypothesis type already exists
422     if (myHypCreatorMap.find(string(theHypName)) == myHypCreatorMap.end())
423     {
424       // load plugin library
425       if(MYDEBUG) MESSAGE("Loading server meshers plugin library ...");
426       LibHandle libHandle = LoadLib( aPlatformLibName.c_str() );
427       if (!libHandle)
428       {
429         // report any error, if occurred
430 #ifndef WIN32
431         const char* anError = dlerror();
432         throw(SALOME_Exception(anError));
433 #else
434         throw(SALOME_Exception(LOCALIZED( "Can't load server meshers plugin library" )));
435 #endif
436       }
437
438       // get method, returning hypothesis creator
439       if(MYDEBUG) MESSAGE("Find GetHypothesisCreator() method ...");
440       GetHypothesisCreator procHandle =
441         (GetHypothesisCreator)GetProc( libHandle, "GetHypothesisCreator" );
442       if (!procHandle)
443       {
444         throw(SALOME_Exception(LOCALIZED("bad hypothesis plugin library")));
445         UnLoadLib(libHandle);
446       }
447
448       // get hypothesis creator
449       if(MYDEBUG) MESSAGE("Get Hypothesis Creator for " << theHypName);
450       aCreator = procHandle(theHypName);
451       if (!aCreator)
452       {
453         throw(SALOME_Exception(LOCALIZED("no such a hypothesis in this plugin")));
454       }
455       // map hypothesis creator to a hypothesis name
456       myHypCreatorMap[string(theHypName)] = aCreator;
457       return aCreator;
458     }
459     else
460     {
461       return myHypCreatorMap[string(theHypName)];
462     }
463   }
464   catch (SALOME_Exception& S_ex)
465   {
466     THROW_SALOME_CORBA_EXCEPTION(S_ex.what(), SALOME::BAD_PARAM);
467   }
468   return aCreator;
469 }
470
471 //=============================================================================
472 /*!
473  *  SMESH_Gen_i::createHypothesis
474  *
475  *  Create hypothesis of given type
476  */
477 //=============================================================================
478 SMESH::SMESH_Hypothesis_ptr SMESH_Gen_i::createHypothesis(const char* theHypName,
479                                                           const char* theLibName)
480 {
481   SMESH_Hypothesis_i* myHypothesis_i = 0;
482   SMESH::SMESH_Hypothesis_var hypothesis_i;
483   std::string aPlatformLibName;
484   GenericHypothesisCreator_i* aCreator =
485     getHypothesisCreator(theHypName, theLibName, aPlatformLibName);
486
487   // create a new hypothesis object, store its ref. in studyContext
488   myHypothesis_i = aCreator->Create(myPoa, GetCurrentStudyID(), &myGen);
489   if (myHypothesis_i)
490   {
491     myHypothesis_i->SetLibName( aPlatformLibName.c_str() ); // for persistency assurance
492     CORBA::String_var hypName = myHypothesis_i->GetName();
493     myHypCreatorMap[ hypName.in() ] = aCreator;
494
495     // activate the CORBA servant of hypothesis
496     hypothesis_i = myHypothesis_i->_this();
497     int nextId = RegisterObject( hypothesis_i );
498     if(MYDEBUG) { MESSAGE( "Add hypo to map with id = "<< nextId ); }
499     else        { nextId = 0; } // avoid "unused variable" warning in release mode
500   }
501   return hypothesis_i._retn();
502 }
503
504 //=============================================================================
505 /*!
506  *  SMESH_Gen_i::createMesh
507  *
508  *  Create empty mesh on shape
509  */
510 //=============================================================================
511 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::createMesh()
512   throw ( SALOME::SALOME_Exception )
513 {
514   Unexpect aCatch(SALOME_SalomeException);
515   if(MYDEBUG) MESSAGE( "SMESH_Gen_i::createMesh" );
516
517   // Get or create the GEOM_Client instance
518   try {
519     // create a new mesh object servant, store it in a map in study context
520     SMESH_Mesh_i* meshServant = new SMESH_Mesh_i( GetPOA(), this, GetCurrentStudyID() );
521     // create a new mesh object
522     if(MYDEBUG) MESSAGE("myIsEmbeddedMode " << myIsEmbeddedMode);
523     meshServant->SetImpl( myGen.CreateMesh( GetCurrentStudyID(), myIsEmbeddedMode ));
524
525     // activate the CORBA servant of Mesh
526     SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( meshServant->_this() );
527     int nextId = RegisterObject( mesh );
528     if(MYDEBUG) { MESSAGE( "Add mesh to map with id = "<< nextId); }
529     else        { nextId = 0; } // avoid "unused variable" warning in release mode
530     return mesh._retn();
531   }
532   catch (SALOME_Exception& S_ex) {
533     THROW_SALOME_CORBA_EXCEPTION( S_ex.what(), SALOME::BAD_PARAM );
534   }
535   return SMESH::SMESH_Mesh::_nil();
536 }
537
538 //=============================================================================
539 /*!
540  *  SMESH_Gen_i::GetShapeReader
541  *
542  *  Get shape reader
543  */
544 //=============================================================================
545 GEOM_Client* SMESH_Gen_i::GetShapeReader()
546 {
547   // create shape reader if necessary
548   if ( !myShapeReader )
549     myShapeReader = new GEOM_Client(GetContainerRef());
550   ASSERT( myShapeReader );
551   return myShapeReader;
552 }
553
554 //=============================================================================
555 /*!
556  *  SMESH_Gen_i::SetGeomEngine
557  *
558  *  Set GEOM::GEOM_Gen reference
559  */
560 //=============================================================================
561 //GEOM::GEOM_Gen_ptr SMESH_Gen_i::SetGeomEngine( const char* containerLoc )
562 void SMESH_Gen_i::SetGeomEngine( GEOM::GEOM_Gen_ptr geomcompo )
563 {
564   //Engines::Component_ptr temp=GetLCC()->FindOrLoad_Component(containerLoc,"GEOM");
565   //myGeomGen=GEOM::GEOM_Gen::_narrow(temp);
566   myGeomGen=GEOM::GEOM_Gen::_duplicate(geomcompo);
567   //return myGeomGen;
568 }
569
570 //=============================================================================
571 /*!
572  *  SMESH_Gen_i::SetEmbeddedMode
573  *
574  *  Set current mode
575  */
576 //=============================================================================
577
578 void SMESH_Gen_i::SetEmbeddedMode( CORBA::Boolean theMode )
579 {
580   myIsEmbeddedMode = theMode;
581
582   if ( !myIsEmbeddedMode ) {
583     //PAL10867: disable signals catching with "noexcepthandler" option
584     char* envNoCatchSignals = getenv("NOT_INTERCEPT_SIGNALS");
585     if (!envNoCatchSignals || !atoi(envNoCatchSignals))
586     {
587       bool raiseFPE;
588 #ifdef _DEBUG_
589       raiseFPE = true;
590       char* envDisableFPE = getenv("DISABLE_FPE");
591       if (envDisableFPE && atoi(envDisableFPE))
592         raiseFPE = false;
593 #else
594       raiseFPE = false;
595 #endif
596       OSD::SetSignal( raiseFPE );
597     }
598     // else OSD::SetSignal() is called in GUI
599   }
600 }
601
602 //=============================================================================
603 /*!
604  *  SMESH_Gen_i::IsEmbeddedMode
605  *
606  *  Get current mode
607  */
608 //=============================================================================
609
610 CORBA::Boolean SMESH_Gen_i::IsEmbeddedMode()
611 {
612   return myIsEmbeddedMode;
613 }
614
615 //=============================================================================
616 /*!
617  *  SMESH_Gen_i::SetCurrentStudy
618  *
619  *  Set current study
620  */
621 //=============================================================================
622
623 void SMESH_Gen_i::SetCurrentStudy( SALOMEDS::Study_ptr theStudy )
624 {
625   setCurrentStudy( theStudy );
626 }
627
628 void SMESH_Gen_i::setCurrentStudy( SALOMEDS::Study_ptr theStudy,
629                                    bool                theStudyIsBeingClosed)
630 {
631   int curStudyId = GetCurrentStudyID();
632   myCurrentStudy = SALOMEDS::Study::_duplicate( theStudy );
633   // create study context, if it doesn't exist and set current study
634   int studyId = GetCurrentStudyID();
635   if ( myStudyContextMap.find( studyId ) == myStudyContextMap.end() )
636     myStudyContextMap[ studyId ] = new StudyContext;
637
638   // myCurrentStudy may be nil
639   if ( !theStudyIsBeingClosed && !CORBA::is_nil( myCurrentStudy ) ) {
640     SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
641     SALOMEDS::SComponent_wrap GEOM_var = myCurrentStudy->FindComponent( "GEOM" );
642     if( !GEOM_var->_is_nil() )
643       aStudyBuilder->LoadWith( GEOM_var, GetGeomEngine() );
644     // NPAL16168, issue 0020210
645     // Let meshes update their data depending on GEOM groups that could change
646     if ( curStudyId != studyId )
647     {
648       CORBA::String_var compDataType = ComponentDataType();
649       SALOMEDS::SComponent_wrap me = myCurrentStudy->FindComponent( compDataType.in() );
650       if ( !me->_is_nil() ) {
651         SALOMEDS::ChildIterator_wrap anIter = myCurrentStudy->NewChildIterator( me );
652         for ( ; anIter->More(); anIter->Next() ) {
653           SALOMEDS::SObject_wrap so = anIter->Value();
654           CORBA::Object_var     ior = SObjectToObject( so );
655           if ( SMESH_Mesh_i*   mesh = SMESH::DownCast<SMESH_Mesh_i*>( ior ))
656             mesh->CheckGeomModif();
657         }
658       }
659     }
660   }
661 }
662
663 //=============================================================================
664 /*!
665  *  SMESH_Gen_i::GetCurrentStudy
666  *
667  *  Get current study
668  */
669 //=============================================================================
670
671 SALOMEDS::Study_ptr SMESH_Gen_i::GetCurrentStudy()
672 {
673   if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetCurrentStudy: study Id = " << GetCurrentStudyID() );
674   if ( GetCurrentStudyID() < 0 )
675     return SALOMEDS::Study::_nil();
676   return SALOMEDS::Study::_duplicate( myCurrentStudy );
677 }
678
679 //=============================================================================
680 /*!
681  *  SMESH_Gen_i::GetCurrentStudyContext
682  *
683  *  Get current study context
684  */
685 //=============================================================================
686 StudyContext* SMESH_Gen_i::GetCurrentStudyContext()
687 {
688   if ( !CORBA::is_nil( myCurrentStudy ) &&
689        myStudyContextMap.find( GetCurrentStudyID() ) != myStudyContextMap.end() )
690     return myStudyContextMap[ myCurrentStudy->StudyId() ];
691   else
692     return 0;
693 }
694
695 //=============================================================================
696 /*!
697  *  SMESH_Gen_i::CreateHypothesis
698  *
699  *  Create hypothesis/algorothm of given type and publish it in the study
700  */
701 //=============================================================================
702
703 SMESH::SMESH_Hypothesis_ptr SMESH_Gen_i::CreateHypothesis( const char* theHypName,
704                                                            const char* theLibName )
705   throw ( SALOME::SALOME_Exception )
706 {
707   Unexpect aCatch(SALOME_SalomeException);
708   // Create hypothesis/algorithm
709   SMESH::SMESH_Hypothesis_var hyp = this->createHypothesis( theHypName, theLibName );
710
711   // Publish hypothesis/algorithm in the study
712   if ( CanPublishInStudy( hyp ) ) {
713     SALOMEDS::SObject_wrap aSO = PublishHypothesis( myCurrentStudy, hyp );
714     if ( !aSO->_is_nil() ) {
715       // Update Python script
716       TPythonDump() << aSO << " = " << this << ".CreateHypothesis('"
717                     << theHypName << "', '" << theLibName << "')";
718     }
719   }
720
721   return hyp._retn();
722 }
723
724 //================================================================================
725 /*!
726  * \brief Return a hypothesis holding parameter values corresponding either to the mesh
727  * existing on the given geometry or to size of the geometry.
728  *  \param theHypType - hypothesis type name
729  *  \param theLibName - plugin library name
730  *  \param theMesh - The mesh of interest
731  *  \param theGeom - The shape to get parameter values from
732  *  \retval SMESH::SMESH_Hypothesis_ptr - The returned hypothesis may be the one existing
733  *     in a study and used to compute the mesh, or a temporary one created just to pass
734  *     parameter values
735  */
736 //================================================================================
737
738 SMESH::SMESH_Hypothesis_ptr
739 SMESH_Gen_i::GetHypothesisParameterValues (const char*           theHypType,
740                                            const char*           theLibName,
741                                            SMESH::SMESH_Mesh_ptr theMesh,
742                                            GEOM::GEOM_Object_ptr theGeom,
743                                            CORBA::Boolean        byMesh)
744   throw ( SALOME::SALOME_Exception )
745 {
746   Unexpect aCatch(SALOME_SalomeException);
747   if ( byMesh && CORBA::is_nil( theMesh ) )
748     return SMESH::SMESH_Hypothesis::_nil();
749   if ( byMesh && CORBA::is_nil( theGeom ) )
750     return SMESH::SMESH_Hypothesis::_nil();
751
752   // -----------------------------------------------
753   // find hypothesis used to mesh theGeom
754   // -----------------------------------------------
755
756   // get mesh and shape
757   SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
758   TopoDS_Shape shape = GeomObjectToShape( theGeom );
759   if ( byMesh && ( !meshServant || meshServant->NbNodes()==0 || shape.IsNull() ))
760     return SMESH::SMESH_Hypothesis::_nil();
761   ::SMESH_Mesh* mesh = meshServant ? &meshServant->GetImpl() : (::SMESH_Mesh*)0;
762
763   // create a temporary hypothesis to know its dimension
764   SMESH::SMESH_Hypothesis_var tmpHyp = this->createHypothesis( theHypType, theLibName );
765   SMESH_Hypothesis_i* hypServant = SMESH::DownCast<SMESH_Hypothesis_i*>( tmpHyp );
766   if ( !hypServant )
767     return SMESH::SMESH_Hypothesis::_nil();
768   ::SMESH_Hypothesis* hyp = hypServant->GetImpl();
769
770   if ( byMesh ) {
771     // look for a hypothesis of theHypType used to mesh the shape
772     if ( myGen.GetShapeDim( shape ) == hyp->GetDim() )
773     {
774       // check local shape
775       SMESH::ListOfHypothesis_var aHypList = theMesh->GetHypothesisList( theGeom );
776       int nbLocalHyps = aHypList->length();
777       for ( int i = 0; i < nbLocalHyps; i++ ) {
778         CORBA::String_var hypName = aHypList[i]->GetName();
779         if ( strcmp( theHypType, hypName.in() ) == 0 ) // FOUND local!
780           return SMESH::SMESH_Hypothesis::_duplicate( aHypList[i] );
781       }
782       // check super shapes
783       TopTools_ListIteratorOfListOfShape itShape( mesh->GetAncestors( shape ));
784       while ( nbLocalHyps == 0 && itShape.More() ) {
785         GEOM::GEOM_Object_ptr geomObj = ShapeToGeomObject( itShape.Value() );
786         if ( ! CORBA::is_nil( geomObj )) {
787           SMESH::ListOfHypothesis_var aHypList = theMesh->GetHypothesisList( geomObj );
788           nbLocalHyps = aHypList->length();
789           for ( int i = 0; i < nbLocalHyps; i++ )
790             if ( strcmp( theHypType, aHypList[i]->GetName() ) == 0 ) // FOUND global!
791               return SMESH::SMESH_Hypothesis::_duplicate( aHypList[i] );
792         }
793         itShape.Next();
794       }
795     }
796
797     // let the temporary hypothesis find out somehow parameter values by mesh
798     if ( hyp->SetParametersByMesh( mesh, shape ))
799       return SMESH::SMESH_Hypothesis::_duplicate( tmpHyp );
800   }
801   else {
802     double diagonal = 0;
803     if ( mesh )
804       diagonal = mesh->GetShapeDiagonalSize();
805     else
806       diagonal = ::SMESH_Mesh::GetShapeDiagonalSize( shape );
807     ::SMESH_Hypothesis::TDefaults dflts;
808     dflts._elemLength = diagonal / myGen.GetBoundaryBoxSegmentation();
809     dflts._nbSegments = myGen.GetDefaultNbSegments();
810     dflts._shape      = &shape;
811     // let the temporary hypothesis initialize it's values
812     if ( hyp->SetParametersByDefaults( dflts, mesh ))
813       return SMESH::SMESH_Hypothesis::_duplicate( tmpHyp );
814   }
815
816   return SMESH::SMESH_Hypothesis::_nil();
817 }
818
819 //=============================================================================
820 /*!
821  * Returns \c True if a hypothesis is assigned to a sole sub-mesh in a current Study
822  *  \param [in] theHyp - the hypothesis of interest
823  *  \param [out] theMesh - the sole mesh using \a theHyp
824  *  \param [out] theShape - the sole geometry \a theHyp is assigned to
825  *  \return boolean - \c True if \a theMesh and \a theShape are sole using \a theHyp
826  *
827  * If two meshes on same shape have theHyp assigned to the same sub-shape, they are
828  * considered as SAME sub-mesh => result is \c true.
829  * This method ids used to initialize SMESHGUI_GenericHypothesisCreator with
830  * a shape to which an hyp being edited is assigned.
831  */
832 //=============================================================================
833
834 CORBA::Boolean SMESH_Gen_i::GetSoleSubMeshUsingHyp( SMESH::SMESH_Hypothesis_ptr theHyp,
835                                                     SMESH::SMESH_Mesh_out       theMesh,
836                                                     GEOM::GEOM_Object_out       theShape)
837 {
838   if ( GetCurrentStudyID() < 0 || CORBA::is_nil( theHyp ))
839     return false;
840
841   // get Mesh component SO
842   CORBA::String_var compDataType = ComponentDataType();
843   SALOMEDS::SComponent_wrap comp = myCurrentStudy->FindComponent( compDataType.in() );
844   if ( CORBA::is_nil( comp ))
845     return false;
846
847   // look for child SO of meshes
848   SMESH::SMESH_Mesh_var foundMesh;
849   TopoDS_Shape          foundShape;
850   bool                  isSole = true;
851   SALOMEDS::ChildIterator_wrap meshIter = myCurrentStudy->NewChildIterator( comp );
852   for ( ; meshIter->More() && isSole; meshIter->Next() )
853   {
854     SALOMEDS::SObject_wrap curSO = meshIter->Value();
855     CORBA::Object_var        obj = SObjectToObject( curSO );
856     SMESH_Mesh_i*         mesh_i = SMESH::DownCast< SMESH_Mesh_i* >( obj );
857     if ( ! mesh_i )
858       continue;
859
860     // look for a sole shape where theHyp is assigned
861     bool isHypFound = false;
862     const ShapeToHypothesis & s2hyps = mesh_i->GetImpl().GetMeshDS()->GetHypotheses();
863     ShapeToHypothesis::Iterator s2hypsIt( s2hyps );
864     for ( ; s2hypsIt.More() && isSole; s2hypsIt.Next() )
865     {
866       const THypList& hyps = s2hypsIt.Value();
867       THypList::const_iterator h = hyps.begin();
868       for ( ; h != hyps.end(); ++h )
869         if ( (*h)->GetID() == theHyp->GetId() )
870           break;
871       if ( h != hyps.end()) // theHyp found
872       {
873         isHypFound = true;
874         if ( ! foundShape.IsNull() &&
875              ! foundShape.IsSame( s2hypsIt.Key() )) // not a sole sub-shape
876         {
877           foundShape.Nullify();
878           isSole = false;
879           break;
880         }
881         foundShape = s2hypsIt.Key();
882       }
883     } // loop on assigned hyps
884
885     if ( isHypFound && !foundShape.IsNull() ) // a mesh using theHyp is found
886     {
887       if ( !foundMesh->_is_nil() ) // not a sole mesh
888       {
889         if ( !foundMesh->HasShapeToMesh() ||
890              !mesh_i   ->HasShapeToMesh() )
891         {
892           isSole = ( foundMesh->HasShapeToMesh() == mesh_i->HasShapeToMesh() );
893         }
894         else
895         {
896           GEOM::GEOM_Object_var s1 = mesh_i   ->GetShapeToMesh();
897           GEOM::GEOM_Object_var s2 = foundMesh->GetShapeToMesh();
898           isSole = s1->IsSame( s2 );
899         }
900       }
901       foundMesh = SMESH::SMESH_Mesh::_narrow( obj );
902     }
903
904   } // loop on meshes
905
906   if ( isSole &&
907        ! foundMesh->_is_nil() &&
908        ! foundShape.IsNull() )
909   {
910     theMesh  = foundMesh._retn();
911     theShape = ShapeToGeomObject( foundShape );
912     return ( !theMesh->_is_nil() && !theShape->_is_nil() );
913   }
914   return false;
915 }
916
917 //=============================================================================
918 /*!
919  * Sets number of segments per diagonal of boundary box of geometry by which
920  * default segment length of appropriate 1D hypotheses is defined
921  */
922 //=============================================================================
923
924 void SMESH_Gen_i::SetBoundaryBoxSegmentation( CORBA::Long theNbSegments )
925   throw ( SALOME::SALOME_Exception )
926 {
927   if ( theNbSegments > 0 )
928     myGen.SetBoundaryBoxSegmentation( int( theNbSegments ));
929   else
930     THROW_SALOME_CORBA_EXCEPTION( "non-positive number of segments", SALOME::BAD_PARAM );
931 }
932 //=============================================================================
933   /*!
934    * \brief Sets default number of segments per edge
935    */
936 //=============================================================================
937 void SMESH_Gen_i::SetDefaultNbSegments(CORBA::Long theNbSegments)
938   throw ( SALOME::SALOME_Exception )
939 {
940   if ( theNbSegments > 0 )
941     myGen.SetDefaultNbSegments( int(theNbSegments) );
942   else
943     THROW_SALOME_CORBA_EXCEPTION( "non-positive number of segments", SALOME::BAD_PARAM );
944 }
945
946 //=============================================================================
947 /*!
948   Set an option value
949 */
950 //=============================================================================
951
952 void SMESH_Gen_i::SetOption(const char* name, const char* value)
953 {
954   if ( name && value && strlen( value ) > 0 )
955   {
956     string msgToGUI;
957     if ( strcmp(name, "historical_python_dump") == 0 )
958     {
959       myIsHistoricalPythonDump = ( value[0] == '1' || toupper(value[0]) == 'T' ); // 1 || true
960       msgToGUI = "preferences/SMESH/historical_python_dump/";
961       msgToGUI += myIsHistoricalPythonDump ? "true" : "false";
962     }
963     else if ( strcmp(name, "forget_mesh_on_hyp_modif") == 0 )
964     {
965       myToForgetMeshDataOnHypModif = ( value[0] == '1' || toupper(value[0]) == 'T' ); // 1 || true
966       msgToGUI = "preferences/SMESH/forget_mesh_on_hyp_modif/";
967       msgToGUI += myToForgetMeshDataOnHypModif ? "true" : "false";
968     }
969     else if ( strcmp(name, "default_grp_color") == 0 )
970     {
971       vector<int> color;
972       string str = value;
973       // color must be presented as a string of following form:
974       if ( str.at(0) == '#' && str.length() == 7 ) { // hexadecimal color ("#ffaa00", for example)
975         str = str.substr(1);
976         for ( size_t i = 0; i < str.length()/2; i++ )
977           if ( str.at(i*2) >= '0' && str.at(i*2) <= 'f' && str.at(i*2+1) >= '0' && str.at(i*2+1) <= 'f' )
978             color.push_back( strtol( str.substr( i*2, 2 ).c_str(), NULL, 16 ) );
979       }
980       else if ( value ) { // rgb color ("255,170,0", for example)
981         string tempValue( value );
982         char* colorValue = strtok( &tempValue[0], "," );
983         while ( colorValue != NULL ) {
984           int c_value = atoi( colorValue );
985           if ( c_value >= 0 && c_value <= 255 )
986             color.push_back( c_value );
987           colorValue = strtok( NULL, "," );
988         }
989       }
990       if ( color.size() == 3 ) { // color must have three valid component
991         SMESHDS_GroupBase::SetDefaultColor( Quantity_Color( color[0]/255., color[1]/255., color[2]/255., Quantity_TOC_RGB ) );
992         myDefaultGroupColor = value;
993         msgToGUI = "preferences/SMESH/default_grp_color/";
994         msgToGUI += value;
995       }
996     }
997
998     // update preferences in case if SetOption() is invoked from python console
999     if ( !msgToGUI.empty() )
1000     {
1001       CORBA::Object_var obj = SMESH_Gen_i::GetNS()->Resolve( "/Kernel/Session" );
1002       SALOME::Session_var session = SALOME::Session::_narrow( obj );
1003       if ( !CORBA::is_nil( session ) )
1004         session->emitMessageOneWay(msgToGUI.c_str());
1005     }
1006   }
1007 }
1008
1009 //=============================================================================
1010 /*!
1011   Return an option value
1012 */
1013 //=============================================================================
1014
1015 char* SMESH_Gen_i::GetOption(const char* name)
1016 {
1017   if ( name )
1018   {
1019     if ( strcmp(name, "historical_python_dump") == 0 )
1020     {
1021       return CORBA::string_dup( myIsHistoricalPythonDump ? "true" : "false" );
1022     }
1023     if ( strcmp(name, "forget_mesh_on_hyp_modif") == 0 )
1024     {
1025       return CORBA::string_dup( myToForgetMeshDataOnHypModif ? "true" : "false" );
1026     }
1027     if ( strcmp(name, "default_grp_color") == 0 )
1028     {
1029       return CORBA::string_dup( myDefaultGroupColor.c_str() );
1030     }
1031   }
1032   return CORBA::string_dup( "" );
1033 }
1034
1035 //=============================================================================
1036 /*!
1037  *  SMESH_Gen_i::CreateMesh
1038  *
1039  *  Create empty mesh on a shape and publish it in the study
1040  */
1041 //=============================================================================
1042
1043 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMesh( GEOM::GEOM_Object_ptr theShapeObject )
1044      throw ( SALOME::SALOME_Exception )
1045 {
1046   Unexpect aCatch(SALOME_SalomeException);
1047   if(MYDEBUG) MESSAGE( "SMESH_Gen_i::CreateMesh" );
1048   // create mesh
1049   SMESH::SMESH_Mesh_var mesh = this->createMesh();
1050   // set shape
1051   SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( mesh );
1052   ASSERT( meshServant );
1053   meshServant->SetShape( theShapeObject );
1054
1055   // publish mesh in the study
1056   if ( CanPublishInStudy( mesh ) ) {
1057     SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1058     aStudyBuilder->NewCommand();  // There is a transaction
1059     SALOMEDS::SObject_wrap aSO = PublishMesh( myCurrentStudy, mesh.in() );
1060     aStudyBuilder->CommitCommand();
1061     if ( !aSO->_is_nil() ) {
1062       // Update Python script
1063       TPythonDump() << aSO << " = " << this << ".CreateMesh(" << theShapeObject << ")";
1064     }
1065   }
1066
1067   return mesh._retn();
1068 }
1069
1070 //=============================================================================
1071 /*!
1072  *  SMESH_Gen_i::CreateEmptyMesh
1073  *
1074  *  Create empty mesh
1075  */
1076 //=============================================================================
1077
1078 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateEmptyMesh()
1079      throw ( SALOME::SALOME_Exception )
1080 {
1081   Unexpect aCatch(SALOME_SalomeException);
1082   if(MYDEBUG) MESSAGE( "SMESH_Gen_i::CreateMesh" );
1083   // create mesh
1084   SMESH::SMESH_Mesh_var mesh = this->createMesh();
1085
1086   // publish mesh in the study
1087   if ( CanPublishInStudy( mesh ) ) {
1088     SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1089     aStudyBuilder->NewCommand();  // There is a transaction
1090     SALOMEDS::SObject_wrap aSO = PublishMesh( myCurrentStudy, mesh.in() );
1091     aStudyBuilder->CommitCommand();
1092     if ( !aSO->_is_nil() ) {
1093       // Update Python script
1094       TPythonDump() << aSO << " = " << this << ".CreateEmptyMesh()";
1095     }
1096   }
1097
1098   return mesh._retn();
1099 }
1100
1101 namespace
1102 {
1103   //================================================================================
1104   /*!
1105    * \brief Throws an exception in case if the file can't be read
1106    */
1107   //================================================================================
1108
1109   void checkFileReadable( const char* theFileName ) throw ( SALOME::SALOME_Exception )
1110   {
1111     SMESH_File f ( theFileName );
1112     if ( !f )
1113     {
1114       if ( !f.error().empty() )
1115         THROW_SALOME_CORBA_EXCEPTION( f.error().c_str(), SALOME::BAD_PARAM);
1116
1117       THROW_SALOME_CORBA_EXCEPTION
1118         (( SMESH_Comment("Can't open for reading the file ") << theFileName ).c_str(),
1119          SALOME::BAD_PARAM );
1120     }
1121   }
1122 }
1123
1124 //=============================================================================
1125 /*!
1126  *  SMESH_Gen_i::CreateMeshFromUNV
1127  *
1128  *  Create mesh and import data from UNV file
1129  */
1130 //=============================================================================
1131
1132 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMeshesFromUNV( const char* theFileName )
1133   throw ( SALOME::SALOME_Exception )
1134 {
1135   Unexpect aCatch(SALOME_SalomeException);
1136
1137   checkFileReadable( theFileName );
1138
1139   SMESH::SMESH_Mesh_var aMesh = createMesh();
1140   string aFileName;
1141   // publish mesh in the study
1142   if ( CanPublishInStudy( aMesh ) ) {
1143     SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1144     aStudyBuilder->NewCommand();  // There is a transaction
1145     SALOMEDS::SObject_wrap aSO = PublishMesh( myCurrentStudy, aMesh.in(), aFileName.c_str() );
1146     aStudyBuilder->CommitCommand();
1147     if ( !aSO->_is_nil() ) {
1148       // Update Python script
1149       TPythonDump() << aSO << " = " << this << ".CreateMeshesFromUNV(r'" << theFileName << "')";
1150     }
1151   }
1152
1153   SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
1154   ASSERT( aServant );
1155   aServant->ImportUNVFile( theFileName );
1156
1157   // Dump creation of groups
1158   SMESH::ListOfGroups_var groups = aServant->GetGroups();
1159
1160   aServant->GetImpl().GetMeshDS()->Modified();
1161   return aMesh._retn();
1162 }
1163
1164 //=============================================================================
1165 /*!
1166  *  SMESH_Gen_i::CreateMeshFromMED
1167  *
1168  *  Create mesh and import data from MED file
1169  */
1170 //=============================================================================
1171
1172 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromMEDorSAUV( const char* theFileName,
1173                                                            SMESH::DriverMED_ReadStatus& theStatus,
1174                                                            const char* theCommandNameForPython,
1175                                                            const char* theFileNameForPython)
1176 {
1177 #ifdef WIN32
1178   char bname[ _MAX_FNAME ];
1179   _splitpath( theFileNameForPython, NULL, NULL, bname, NULL );
1180   string aFileName = bname;
1181 #else
1182   string aFileName = basename( const_cast<char *>(theFileNameForPython) );
1183 #endif
1184   // Retrieve mesh names from the file
1185   DriverMED_R_SMESHDS_Mesh myReader;
1186   myReader.SetFile( theFileName );
1187   myReader.SetMeshId( -1 );
1188   Driver_Mesh::Status aStatus;
1189   list<string> aNames = myReader.GetMeshNames(aStatus);
1190   SMESH::mesh_array_var aResult = new SMESH::mesh_array();
1191   theStatus = (SMESH::DriverMED_ReadStatus)aStatus;
1192
1193   { // open a new scope to make aPythonDump die before PythonDump in SMESH_Mesh::GetGroups()
1194
1195   // Python Dump
1196   TPythonDump aPythonDump;
1197   aPythonDump << "([";
1198
1199   if (theStatus == SMESH::DRS_OK) {
1200     SALOMEDS::StudyBuilder_var aStudyBuilder;
1201     if ( GetCurrentStudyID() > -1 )
1202     {
1203       aStudyBuilder = myCurrentStudy->NewBuilder();
1204       aStudyBuilder->NewCommand();  // There is a transaction
1205     }
1206     aResult->length( aNames.size() );
1207     int i = 0;
1208
1209     // Iterate through all meshes and create mesh objects
1210     for ( list<string>::iterator it = aNames.begin(); it != aNames.end(); it++ )
1211     {
1212       // Python Dump
1213       if (i > 0) aPythonDump << ", ";
1214
1215       // create mesh
1216       SMESH::SMESH_Mesh_var mesh = createMesh();
1217
1218       // publish mesh in the study
1219       SALOMEDS::SObject_wrap aSO;
1220       if ( CanPublishInStudy( mesh ) )
1221         // little trick: for MED file theFileName and theFileNameForPython are the same, but they are different for SAUV
1222         // - as names of meshes are stored in MED file, we use them for data publishing
1223         // - as mesh name is not stored in UNV file, we use file name as name of mesh when publishing data
1224         aSO = PublishMesh( myCurrentStudy, mesh.in(), ( theFileName == theFileNameForPython ) ? (*it).c_str() : aFileName.c_str() );
1225
1226       // Python Dump
1227       if ( !aSO->_is_nil() ) {
1228         aPythonDump << aSO;
1229       } else {
1230         aPythonDump << "mesh_" << i;
1231       }
1232
1233       // Read mesh data (groups are published automatically by ImportMEDFile())
1234       SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( mesh ).in() );
1235       ASSERT( meshServant );
1236       SMESH::DriverMED_ReadStatus status1 =
1237         meshServant->ImportMEDFile( theFileName, (*it).c_str() );
1238       if (status1 > theStatus)
1239         theStatus = status1;
1240
1241       aResult[i++] = SMESH::SMESH_Mesh::_duplicate( mesh );
1242       meshServant->GetImpl().GetMeshDS()->Modified();
1243     }
1244     if ( !aStudyBuilder->_is_nil() )
1245       aStudyBuilder->CommitCommand();
1246   }
1247
1248   // Update Python script
1249   aPythonDump << "], status) = " << this << "." << theCommandNameForPython << "(r'" << theFileNameForPython << "')";
1250   }
1251   // Dump creation of groups
1252   for ( CORBA::ULong  i = 0; i < aResult->length(); ++i )
1253     SMESH::ListOfGroups_var groups = aResult[ i ]->GetGroups();
1254
1255   return aResult._retn();
1256 }
1257
1258 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromMED( const char* theFileName,
1259                                                      SMESH::DriverMED_ReadStatus& theStatus)
1260   throw ( SALOME::SALOME_Exception )
1261 {
1262   Unexpect aCatch(SALOME_SalomeException);
1263   checkFileReadable( theFileName );
1264
1265   SMESH::mesh_array* result = CreateMeshesFromMEDorSAUV(theFileName, theStatus, "CreateMeshesFromMED", theFileName);
1266   return result;
1267 }
1268
1269 //=============================================================================
1270 /*!
1271  *  SMESH_Gen_i::CreateMeshFromSAUV
1272  *
1273  *  Create mesh and import data from SAUV file
1274  */
1275 //=============================================================================
1276
1277 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromSAUV( const char* theFileName,
1278                                                       SMESH::DriverMED_ReadStatus& theStatus)
1279      throw ( SALOME::SALOME_Exception )
1280 {
1281   Unexpect aCatch(SALOME_SalomeException);
1282   checkFileReadable( theFileName );
1283
1284   std::string sauvfilename(theFileName);
1285   std::string medfilename(theFileName);
1286   medfilename += ".med";
1287   std::string cmd;
1288 #ifdef WIN32
1289   cmd = "%PYTHONBIN% ";
1290 #else
1291   cmd = "python ";
1292 #endif
1293   cmd += "-c \"";
1294   cmd += "from medutilities import convert ; convert(r'" + sauvfilename + "', 'GIBI', 'MED', 1, r'" + medfilename + "')";
1295   cmd += "\"";
1296   system(cmd.c_str());
1297   SMESH::mesh_array* result = CreateMeshesFromMEDorSAUV(medfilename.c_str(), theStatus, "CreateMeshesFromSAUV", sauvfilename.c_str());
1298 #ifdef WIN32
1299   cmd = "%PYTHONBIN% ";
1300 #else
1301   cmd = "python ";
1302 #endif
1303   cmd += "-c \"";
1304   cmd += "from medutilities import my_remove ; my_remove(r'" + medfilename + "')";
1305   cmd += "\"";
1306   system(cmd.c_str());
1307   return result;
1308 }
1309
1310 //=============================================================================
1311 /*!
1312  *  SMESH_Gen_i::CreateMeshFromSTL
1313  *
1314  *  Create mesh and import data from STL file
1315  */
1316 //=============================================================================
1317
1318 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CreateMeshesFromSTL( const char* theFileName )
1319   throw ( SALOME::SALOME_Exception )
1320 {
1321   Unexpect aCatch(SALOME_SalomeException);
1322   checkFileReadable( theFileName );
1323
1324   SMESH::SMESH_Mesh_var aMesh = createMesh();
1325   //string aFileName;
1326 #ifdef WIN32
1327   char bname[ _MAX_FNAME ];
1328   _splitpath( theFileName, NULL, NULL, bname, NULL );
1329   string aFileName = bname;
1330 #else
1331   string aFileName = basename( const_cast<char *>(theFileName) );
1332 #endif
1333   // publish mesh in the study
1334   if ( CanPublishInStudy( aMesh ) ) {
1335     SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1336     aStudyBuilder->NewCommand();  // There is a transaction
1337     SALOMEDS::SObject_wrap aSO = PublishInStudy
1338       ( myCurrentStudy, SALOMEDS::SObject::_nil(), aMesh.in(), aFileName.c_str() );
1339     aStudyBuilder->CommitCommand();
1340     if ( !aSO->_is_nil() ) {
1341       // Update Python script
1342       TPythonDump() << aSO << " = " << this << ".CreateMeshesFromSTL(r'" << theFileName << "')";
1343     }
1344   }
1345
1346   SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
1347   ASSERT( aServant );
1348   aServant->ImportSTLFile( theFileName );
1349   aServant->GetImpl().GetMeshDS()->Modified();
1350   return aMesh._retn();
1351 }
1352
1353 //================================================================================
1354 /*!
1355  * \brief Create meshes and import data from the CGSN file
1356  */
1357 //================================================================================
1358
1359 SMESH::mesh_array* SMESH_Gen_i::CreateMeshesFromCGNS( const char* theFileName,
1360                                                       SMESH::DriverMED_ReadStatus& theStatus)
1361   throw ( SALOME::SALOME_Exception )
1362 {
1363   Unexpect aCatch(SALOME_SalomeException);
1364   checkFileReadable( theFileName );
1365
1366   SMESH::mesh_array_var aResult = new SMESH::mesh_array();
1367
1368 #ifdef WITH_CGNS
1369   // Retrieve nb meshes from the file
1370   DriverCGNS_Read myReader;
1371   myReader.SetFile( theFileName );
1372   Driver_Mesh::Status aStatus;
1373   int nbMeshes = myReader.GetNbMeshes(aStatus);
1374   theStatus = (SMESH::DriverMED_ReadStatus)aStatus;
1375
1376   aResult->length( nbMeshes );
1377
1378   { // open a new scope to make aPythonDump die before PythonDump in SMESH_Mesh::GetGroups()
1379
1380     // Python Dump
1381     TPythonDump aPythonDump;
1382     aPythonDump << "([";
1383
1384     if (theStatus == SMESH::DRS_OK)
1385     {
1386       SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1387       aStudyBuilder->NewCommand();  // There is a transaction
1388
1389       int i = 0;
1390
1391       // Iterate through all meshes and create mesh objects
1392       for ( ; i < nbMeshes; ++i )
1393       {
1394         // Python Dump
1395         if (i > 0) aPythonDump << ", ";
1396
1397         // create mesh
1398         SMESH::SMESH_Mesh_var mesh = createMesh();
1399         aResult[i] = SMESH::SMESH_Mesh::_duplicate( mesh );
1400
1401         // Read mesh data (groups are published automatically by ImportMEDFile())
1402         SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( mesh ).in() );
1403         ASSERT( meshServant );
1404         string meshName;
1405         SMESH::DriverMED_ReadStatus status1 =
1406           meshServant->ImportCGNSFile( theFileName, i, meshName );
1407         if (status1 > theStatus)
1408           theStatus = status1;
1409
1410         meshServant->GetImpl().GetMeshDS()->Modified();
1411         // publish mesh in the study
1412         SALOMEDS::SObject_wrap aSO;
1413         if ( CanPublishInStudy( mesh ) )
1414           aSO = PublishMesh( myCurrentStudy, mesh.in(), meshName.c_str() );
1415
1416         // Python Dump
1417         if ( !aSO->_is_nil() ) {
1418           aPythonDump << aSO;
1419         }
1420         else {
1421           aPythonDump << "mesh_" << i;
1422         }
1423       }
1424       aStudyBuilder->CommitCommand();
1425     }
1426
1427     aPythonDump << "], status) = " << this << ".CreateMeshesFromCGNS(r'" << theFileName << "')";
1428   }
1429   // Dump creation of groups
1430   for ( CORBA::ULong i = 0; i < aResult->length(); ++i )
1431     SMESH::ListOfGroups_var groups = aResult[ i ]->GetGroups();
1432 #else
1433   THROW_SALOME_CORBA_EXCEPTION("CGNS library is unavailable", SALOME::INTERNAL_ERROR);
1434 #endif
1435
1436   return aResult._retn();
1437 }
1438
1439 //================================================================================
1440 /*!
1441  * \brief Create a mesh and import data from a GMF file
1442  */
1443 //================================================================================
1444
1445 SMESH::SMESH_Mesh_ptr
1446 SMESH_Gen_i::CreateMeshesFromGMF( const char*             theFileName,
1447                                   CORBA::Boolean          theMakeRequiredGroups,
1448                                   SMESH::ComputeError_out theError)
1449     throw ( SALOME::SALOME_Exception )
1450 {
1451   Unexpect aCatch(SALOME_SalomeException);
1452   checkFileReadable( theFileName );
1453
1454   SMESH::SMESH_Mesh_var aMesh = createMesh();
1455 #ifdef WIN32
1456   char bname[ _MAX_FNAME ];
1457   _splitpath( theFileName, NULL, NULL, bname, NULL );
1458   string aFileName = bname;
1459 #else
1460   string aFileName = basename( const_cast<char *>(theFileName) );
1461 #endif
1462   // publish mesh in the study
1463   if ( CanPublishInStudy( aMesh ) ) {
1464     SALOMEDS::StudyBuilder_var aStudyBuilder = myCurrentStudy->NewBuilder();
1465     aStudyBuilder->NewCommand();  // There is a transaction
1466     SALOMEDS::SObject_wrap aSO = PublishInStudy
1467       ( myCurrentStudy, SALOMEDS::SObject::_nil(), aMesh.in(), aFileName.c_str() );
1468     aStudyBuilder->CommitCommand();
1469     if ( !aSO->_is_nil() ) {
1470       // Update Python script
1471       TPythonDump() << "("<< aSO << ", error) = " << this << ".CreateMeshesFromGMF(r'"
1472                     << theFileName << "', "
1473                     << theMakeRequiredGroups << " )";
1474     }
1475   }
1476   SMESH_Mesh_i* aServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( aMesh ).in() );
1477   ASSERT( aServant );
1478   theError = aServant->ImportGMFFile( theFileName, theMakeRequiredGroups );
1479   aServant->GetImpl().GetMeshDS()->Modified();
1480   return aMesh._retn();
1481 }
1482
1483
1484 //=============================================================================
1485 /*!
1486  *  SMESH_Gen_i::IsReadyToCompute
1487  *
1488  *  Returns true if mesh contains enough data to be computed
1489  */
1490 //=============================================================================
1491
1492 CORBA::Boolean SMESH_Gen_i::IsReadyToCompute( SMESH::SMESH_Mesh_ptr theMesh,
1493                                               GEOM::GEOM_Object_ptr theShapeObject )
1494   throw ( SALOME::SALOME_Exception )
1495 {
1496   Unexpect aCatch(SALOME_SalomeException);
1497   if(MYDEBUG) MESSAGE( "SMESH_Gen_i::IsReadyToCompute" );
1498
1499   if ( CORBA::is_nil( theShapeObject ) )
1500     THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1501                                   SALOME::BAD_PARAM );
1502
1503   if ( CORBA::is_nil( theMesh ) )
1504     THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
1505                                   SALOME::BAD_PARAM );
1506
1507   try {
1508     // get mesh servant
1509     SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1510     ASSERT( meshServant );
1511     if ( meshServant ) {
1512       // get local TopoDS_Shape
1513       TopoDS_Shape myLocShape = GeomObjectToShape( theShapeObject );
1514       // call implementation
1515       ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1516       return myGen.CheckAlgoState( myLocMesh, myLocShape );
1517     }
1518   }
1519   catch ( SALOME_Exception& S_ex ) {
1520     INFOS( "catch exception "<< S_ex.what() );
1521   }
1522   return false;
1523 }
1524
1525 //================================================================================
1526 /*!
1527  * \brief  Find SObject for an algo
1528  */
1529 //================================================================================
1530
1531 SALOMEDS::SObject_ptr SMESH_Gen_i::GetAlgoSO(const ::SMESH_Algo* algo)
1532 {
1533   if ( algo ) {
1534     if ( !myCurrentStudy->_is_nil() ) {
1535       // find algo in the study
1536       CORBA::String_var compDataType  = ComponentDataType();
1537       SALOMEDS::SComponent_wrap father = myCurrentStudy->FindComponent( compDataType.in() );
1538       if ( !father->_is_nil() ) {
1539         SALOMEDS::ChildIterator_wrap itBig = myCurrentStudy->NewChildIterator( father );
1540         for ( ; itBig->More(); itBig->Next() ) {
1541           SALOMEDS::SObject_wrap gotBranch = itBig->Value();
1542           if ( gotBranch->Tag() == GetAlgorithmsRootTag() ) {
1543             SALOMEDS::ChildIterator_wrap algoIt = myCurrentStudy->NewChildIterator( gotBranch );
1544             for ( ; algoIt->More(); algoIt->Next() ) {
1545               SALOMEDS::SObject_wrap algoSO = algoIt->Value();
1546               CORBA::Object_var     algoIOR = SObjectToObject( algoSO );
1547               if ( !CORBA::is_nil( algoIOR )) {
1548                 SMESH_Hypothesis_i* impl = SMESH::DownCast<SMESH_Hypothesis_i*>( algoIOR );
1549                 if ( impl && impl->GetImpl() == algo )
1550                   return algoSO._retn();
1551               }
1552             } // loop on algo SO's
1553             break;
1554           } // if algo tag
1555         } // SMESH component iterator
1556       }
1557     }
1558   }
1559   return SALOMEDS::SObject::_nil();
1560 }
1561
1562 //================================================================================
1563 /*!
1564  * \brief Return errors of mesh computation
1565  */
1566 //================================================================================
1567
1568 SMESH::compute_error_array* SMESH_Gen_i::GetComputeErrors( SMESH::SMESH_Mesh_ptr theMesh,
1569                                                            GEOM::GEOM_Object_ptr theSubObject )
1570   throw ( SALOME::SALOME_Exception )
1571 {
1572   Unexpect aCatch(SALOME_SalomeException);
1573   if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetComputeErrors()" );
1574
1575   if ( CORBA::is_nil( theSubObject ) && theMesh->HasShapeToMesh())
1576     THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference", SALOME::BAD_PARAM );
1577
1578   if ( CORBA::is_nil( theMesh ) )
1579     THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1580
1581   SMESH::compute_error_array_var error_array = new SMESH::compute_error_array;
1582   try {
1583     if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1584     {
1585       TopoDS_Shape shape;
1586       if(theMesh->HasShapeToMesh())
1587         shape = GeomObjectToShape( theSubObject );
1588       else
1589         shape = SMESH_Mesh::PseudoShape();
1590
1591       ::SMESH_Mesh& mesh = meshServant->GetImpl();
1592
1593       error_array->length( mesh.GetMeshDS()->MaxShapeIndex() );
1594       int nbErr = 0;
1595
1596       SMESH_subMesh *sm = mesh.GetSubMesh(shape);
1597       const bool includeSelf = true, complexShapeFirst = true;
1598       SMESH_subMeshIteratorPtr smIt = sm->getDependsOnIterator(includeSelf,
1599                                                                complexShapeFirst);
1600       while ( smIt->more() )
1601       {
1602         sm = smIt->next();
1603         // if ( sm->GetSubShape().ShapeType() == TopAbs_VERTEX )
1604         //   break;
1605         SMESH_ComputeErrorPtr error = sm->GetComputeError();
1606         if ( error && !error->IsOK() )
1607         {
1608           if ( !( error->myAlgo ) &&
1609                !( error->myAlgo = sm->GetAlgo() ))
1610             continue;
1611           SMESH::ComputeError & errStruct = error_array[ nbErr++ ];
1612           errStruct.code       = -( error->myName < 0 ? error->myName + 1: error->myName ); // -1 -> 0
1613           errStruct.comment    = error->myComment.c_str();
1614           errStruct.subShapeID = sm->GetId();
1615           SALOMEDS::SObject_wrap algoSO = GetAlgoSO( error->myAlgo );
1616           if ( !algoSO->_is_nil() ) {
1617             CORBA::String_var algoName = algoSO->GetName();
1618             errStruct.algoName = algoName;
1619           }
1620           else {
1621             errStruct.algoName = error->myAlgo->GetName();
1622           }
1623           errStruct.hasBadMesh = !error->myBadElements.empty();
1624         }
1625       }
1626       error_array->length( nbErr );
1627     }
1628   }
1629   catch ( SALOME_Exception& S_ex ) {
1630     INFOS( "catch exception "<< S_ex.what() );
1631   }
1632
1633   return error_array._retn();
1634 }
1635
1636 //================================================================================
1637 /*!
1638  * \brief Return mesh elements preventing computation of a subshape
1639  */
1640 //================================================================================
1641
1642 SMESH::MeshPreviewStruct*
1643 SMESH_Gen_i::GetBadInputElements( SMESH::SMESH_Mesh_ptr theMesh,
1644                                   CORBA::Short          theSubShapeID )
1645   throw ( SALOME::SALOME_Exception )
1646 {
1647   Unexpect aCatch(SALOME_SalomeException);
1648   if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetBadInputElements()" );
1649
1650   if ( CORBA::is_nil( theMesh ) )
1651     THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1652
1653   SMESH::MeshPreviewStruct_var result = new SMESH::MeshPreviewStruct;
1654   try {
1655     // mesh servant
1656     if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1657     {
1658       // mesh implementation
1659       ::SMESH_Mesh& mesh = meshServant->GetImpl();
1660       // submesh by subshape id
1661       if ( SMESH_subMesh * sm = mesh.GetSubMeshContaining( theSubShapeID ))
1662       {
1663         // compute error
1664         SMESH_ComputeErrorPtr error = sm->GetComputeError();
1665         if ( error && !error->myBadElements.empty())
1666         {
1667           typedef map<const SMDS_MeshElement*, int > TNode2LocalIDMap;
1668           typedef TNode2LocalIDMap::iterator         TNodeLocalID;
1669
1670           // get nodes of elements and count elements
1671           TNode2LocalIDMap mapNode2LocalID;
1672           list< TNodeLocalID > connectivity;
1673           int i, nbElements = 0, nbConnNodes = 0;
1674
1675           list<const SMDS_MeshElement*>::iterator elemIt  = error->myBadElements.begin();
1676           list<const SMDS_MeshElement*>::iterator elemEnd = error->myBadElements.end();
1677           for ( ; elemIt != elemEnd; ++elemIt, ++nbElements )
1678           {
1679             SMDS_ElemIteratorPtr nIt = (*elemIt)->nodesIterator();
1680             while ( nIt->more() )
1681               connectivity.push_back
1682                 ( mapNode2LocalID.insert( make_pair( nIt->next(), ++nbConnNodes)).first );
1683           }
1684           // fill node coords and assign local ids to the nodes
1685           int nbNodes = mapNode2LocalID.size();
1686           result->nodesXYZ.length( nbNodes );
1687           TNodeLocalID node2ID = mapNode2LocalID.begin();
1688           for ( i = 0; i < nbNodes; ++i, ++node2ID ) {
1689             node2ID->second = i;
1690             const SMDS_MeshNode* node = (const SMDS_MeshNode*) node2ID->first;
1691             result->nodesXYZ[i].x = node->X();
1692             result->nodesXYZ[i].y = node->Y();
1693             result->nodesXYZ[i].z = node->Z();
1694           }
1695           // fill connectivity
1696           result->elementConnectivities.length( nbConnNodes );
1697           list< TNodeLocalID >::iterator connIt = connectivity.begin();
1698           for ( i = 0; i < nbConnNodes; ++i, ++connIt ) {
1699             result->elementConnectivities[i] = (*connIt)->second;
1700           }
1701           // fill element types
1702           result->elementTypes.length( nbElements );
1703           for ( i = 0, elemIt = error->myBadElements.begin(); i <nbElements; ++i, ++elemIt )
1704           {
1705             const SMDS_MeshElement* elem = *elemIt;
1706             result->elementTypes[i].SMDS_ElementType = (SMESH::ElementType) elem->GetType();
1707             result->elementTypes[i].isPoly           = elem->IsPoly();
1708             result->elementTypes[i].nbNodesInElement = elem->NbNodes();
1709           }
1710         }
1711       }
1712     }
1713   }
1714   catch ( SALOME_Exception& S_ex ) {
1715     INFOS( "catch exception "<< S_ex.what() );
1716   }
1717
1718   return result._retn();
1719 }
1720
1721 //================================================================================
1722 /*!
1723  * \brief Create a group of elements preventing computation of a sub-shape
1724  */
1725 //================================================================================
1726
1727 SMESH::ListOfGroups*
1728 SMESH_Gen_i::MakeGroupsOfBadInputElements( SMESH::SMESH_Mesh_ptr theMesh,
1729                                            CORBA::Short          theSubShapeID,
1730                                            const char*           theGroupName )
1731   throw ( SALOME::SALOME_Exception )
1732 {
1733   Unexpect aCatch(SALOME_SalomeException);
1734
1735   SMESH::ListOfGroups_var groups;
1736
1737   if ( CORBA::is_nil( theMesh ) )
1738     THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1739
1740   try {
1741     if ( SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh ))
1742     {
1743       groups = meshServant->MakeGroupsOfBadInputElements( theSubShapeID, theGroupName );
1744       TPythonDump() << groups << " = " << this
1745                     << ".MakeGroupsOfBadInputElements( "
1746                     << theMesh << ", " << theSubShapeID << ", '" << theGroupName << "' )";
1747     }
1748   }
1749   catch ( SALOME_Exception& S_ex ) {
1750     INFOS( "catch exception "<< S_ex.what() );
1751   }
1752   return groups._retn();
1753 }
1754
1755 //================================================================================
1756 /*!
1757  * \brief Returns errors of hypotheses definition
1758  * \param theMesh - the mesh
1759  * \param theSubObject - the main or sub- shape
1760  * \retval SMESH::algo_error_array* - sequence of errors
1761  */
1762 //================================================================================
1763
1764 SMESH::algo_error_array* SMESH_Gen_i::GetAlgoState( SMESH::SMESH_Mesh_ptr theMesh,
1765                                                     GEOM::GEOM_Object_ptr theSubObject )
1766       throw ( SALOME::SALOME_Exception )
1767 {
1768   Unexpect aCatch(SALOME_SalomeException);
1769   if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetAlgoState()" );
1770
1771   if ( CORBA::is_nil( theSubObject ) && theMesh->HasShapeToMesh())
1772     THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference", SALOME::BAD_PARAM );
1773
1774   if ( CORBA::is_nil( theMesh ) )
1775     THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",SALOME::BAD_PARAM );
1776
1777   SMESH::algo_error_array_var error_array = new SMESH::algo_error_array;
1778   try {
1779     SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
1780     ASSERT( meshServant );
1781     if ( meshServant ) {
1782       TopoDS_Shape myLocShape;
1783       if(theMesh->HasShapeToMesh())
1784         myLocShape = GeomObjectToShape( theSubObject );
1785       else
1786         myLocShape = SMESH_Mesh::PseudoShape();
1787
1788       ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1789       list< ::SMESH_Gen::TAlgoStateError > error_list;
1790       list< ::SMESH_Gen::TAlgoStateError >::iterator error;
1791       // call ::SMESH_Gen::GetAlgoState()
1792       myGen.GetAlgoState( myLocMesh, myLocShape, error_list );
1793       error_array->length( error_list.size() );
1794       int i = 0;
1795       for ( error = error_list.begin(); error != error_list.end(); ++error )
1796       {
1797         // fill AlgoStateError structure
1798         SMESH::AlgoStateError & errStruct = error_array[ i++ ];
1799         errStruct.state        = SMESH_Mesh_i::ConvertHypothesisStatus( error->_name );
1800         errStruct.algoDim      = error->_algoDim;
1801         errStruct.isGlobalAlgo = error->_isGlobalAlgo;
1802         errStruct.algoName     = "";
1803         SALOMEDS::SObject_wrap algoSO = GetAlgoSO( error->_algo );
1804         if ( !algoSO->_is_nil() ) {
1805           CORBA::String_var algoName = algoSO->GetName();
1806           errStruct.algoName = algoName.in();
1807         }
1808       }
1809     }
1810   }
1811   catch ( SALOME_Exception& S_ex ) {
1812     INFOS( "catch exception "<< S_ex.what() );
1813   }
1814   return error_array._retn();
1815 }
1816
1817 //=============================================================================
1818 /*!
1819  *  SMESH_Gen_i::GetSubShapesId
1820  *
1821  *  Get sub-shapes unique ID's list
1822  */
1823 //=============================================================================
1824
1825 SMESH::long_array*
1826 SMESH_Gen_i::GetSubShapesId( GEOM::GEOM_Object_ptr      theMainShapeObject,
1827                              const SMESH::object_array& theListOfSubShapeObject )
1828   throw ( SALOME::SALOME_Exception )
1829 {
1830   Unexpect aCatch(SALOME_SalomeException);
1831   if(MYDEBUG) MESSAGE( "SMESH_Gen_i::GetSubShapesId" );
1832
1833   SMESH::long_array_var shapesId = new SMESH::long_array;
1834   set<int> setId;
1835
1836   if ( CORBA::is_nil( theMainShapeObject ) )
1837     THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference", SALOME::BAD_PARAM );
1838
1839   try
1840   {
1841     TopoDS_Shape myMainShape = GeomObjectToShape(theMainShapeObject);
1842     TopTools_IndexedMapOfShape myIndexToShape;
1843     TopExp::MapShapes(myMainShape,myIndexToShape);
1844
1845     for ( CORBA::ULong i = 0; i < theListOfSubShapeObject.length(); i++ )
1846     {
1847       GEOM::GEOM_Object_var aShapeObject
1848         = GEOM::GEOM_Object::_narrow(theListOfSubShapeObject[i]);
1849       if ( CORBA::is_nil( aShapeObject ) )
1850         THROW_SALOME_CORBA_EXCEPTION ("bad shape object reference",     \
1851                                       SALOME::BAD_PARAM );
1852
1853       TopoDS_Shape locShape  = GeomObjectToShape(aShapeObject);
1854       for (TopExp_Explorer exp(locShape,TopAbs_FACE); exp.More(); exp.Next())
1855       {
1856         const TopoDS_Face& F = TopoDS::Face(exp.Current());
1857         setId.insert(myIndexToShape.FindIndex(F));
1858         if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(F));
1859       }
1860       for (TopExp_Explorer exp(locShape,TopAbs_EDGE); exp.More(); exp.Next())
1861       {
1862         const TopoDS_Edge& E = TopoDS::Edge(exp.Current());
1863         setId.insert(myIndexToShape.FindIndex(E));
1864         if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(E));
1865       }
1866       for (TopExp_Explorer exp(locShape,TopAbs_VERTEX); exp.More(); exp.Next())
1867       {
1868         const TopoDS_Vertex& V = TopoDS::Vertex(exp.Current());
1869         setId.insert(myIndexToShape.FindIndex(V));
1870         if(MYDEBUG) SCRUTE(myIndexToShape.FindIndex(V));
1871       }
1872     }
1873     shapesId->length(setId.size());
1874     set<int>::iterator iind;
1875     int i=0;
1876     for (iind = setId.begin(); iind != setId.end(); iind++)
1877     {
1878       if(MYDEBUG) SCRUTE((*iind));
1879       shapesId[i] = (*iind);
1880       if(MYDEBUG) SCRUTE(shapesId[i]);
1881       i++;
1882     }
1883   }
1884   catch (SALOME_Exception& S_ex)
1885   {
1886     THROW_SALOME_CORBA_EXCEPTION(S_ex.what(), SALOME::BAD_PARAM);
1887   }
1888
1889   return shapesId._retn();
1890 }
1891
1892 //=============================================================================
1893 /*!
1894  *  SMESH_Gen_i::Compute
1895  *
1896  *  Compute mesh on a shape
1897  */
1898 //=============================================================================
1899
1900 CORBA::Boolean SMESH_Gen_i::Compute( SMESH::SMESH_Mesh_ptr theMesh,
1901                                      GEOM::GEOM_Object_ptr theShapeObject )
1902      throw ( SALOME::SALOME_Exception )
1903 {
1904   //MEMOSTAT;
1905   Unexpect aCatch(SALOME_SalomeException);
1906   if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Compute" );
1907
1908   if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
1909     THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1910                                   SALOME::BAD_PARAM );
1911
1912   if ( CORBA::is_nil( theMesh ) )
1913     THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
1914                                   SALOME::BAD_PARAM );
1915
1916   // Update Python script
1917   TPythonDump() << "isDone = " << this << ".Compute( "
1918                 << theMesh << ", " << theShapeObject << ")";
1919
1920   try {
1921     // get mesh servant
1922     SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1923     ASSERT( meshServant );
1924     if ( meshServant ) {
1925       meshServant->Load();
1926       // NPAL16168: "geometrical group edition from a submesh don't modifiy mesh computation"
1927       meshServant->CheckGeomModif();
1928       // get local TopoDS_Shape
1929       TopoDS_Shape myLocShape;
1930       if(theMesh->HasShapeToMesh())
1931         myLocShape = GeomObjectToShape( theShapeObject );
1932       else
1933         myLocShape = SMESH_Mesh::PseudoShape();
1934       // call implementation compute
1935       ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1936       myGen.PrepareCompute( myLocMesh, myLocShape );
1937       int how = ::SMESH_Gen::COMPACT_MESH;
1938       if ( myLocShape != myLocMesh.GetShapeToMesh() ) // compute a sub-mesh
1939         how |= ::SMESH_Gen::SHAPE_ONLY;
1940       bool ok = myGen.Compute( myLocMesh, myLocShape, how );
1941       meshServant->CreateGroupServants(); // algos can create groups (issue 0020918)
1942       myLocMesh.GetMeshDS()->Modified();
1943       return ok;
1944     }
1945   }
1946   catch ( std::bad_alloc ) {
1947     INFOS( "Compute(): lack of memory" );
1948   }
1949   catch ( SALOME_Exception& S_ex ) {
1950     INFOS( "Compute(): catch exception "<< S_ex.what() );
1951   }
1952   catch ( ... ) {
1953     INFOS( "Compute(): unknown exception " );
1954   }
1955   return false;
1956 }
1957
1958 //=============================================================================
1959 /*!
1960  *  SMESH_Gen_i::CancelCompute
1961  *
1962  *  Cancel Compute mesh on a shape
1963  */
1964 //=============================================================================
1965
1966 void SMESH_Gen_i::CancelCompute( SMESH::SMESH_Mesh_ptr theMesh,
1967                                  GEOM::GEOM_Object_ptr theShapeObject )
1968 {
1969   SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
1970   ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
1971   TopoDS_Shape myLocShape;
1972   if(theMesh->HasShapeToMesh())
1973     myLocShape = GeomObjectToShape( theShapeObject );
1974   else
1975     myLocShape = SMESH_Mesh::PseudoShape();
1976   myGen.CancelCompute( myLocMesh, myLocShape);
1977 }
1978
1979 //=============================================================================
1980 /*!
1981  *  SMESH_Gen_i::Precompute
1982  *
1983  *  Compute mesh as preview till indicated dimension on shape
1984  */
1985 //=============================================================================
1986
1987 SMESH::MeshPreviewStruct* SMESH_Gen_i::Precompute( SMESH::SMESH_Mesh_ptr theMesh,
1988                                                    GEOM::GEOM_Object_ptr theShapeObject,
1989                                                    SMESH::Dimension      theDimension,
1990                                                    SMESH::long_array&    theShapesId)
1991      throw ( SALOME::SALOME_Exception )
1992 {
1993   Unexpect aCatch(SALOME_SalomeException);
1994   if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Precompute" );
1995
1996   if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
1997     THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
1998                                   SALOME::BAD_PARAM );
1999
2000   if ( CORBA::is_nil( theMesh ) )
2001     THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
2002                                   SALOME::BAD_PARAM );
2003
2004   SMESH::MeshPreviewStruct_var result = new SMESH::MeshPreviewStruct;
2005   try {
2006     // get mesh servant
2007     SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
2008     meshServant->Load();
2009     ASSERT( meshServant );
2010     if ( meshServant ) {
2011       // NPAL16168: "geometrical group edition from a submesh don't modifiy mesh computation"
2012       meshServant->CheckGeomModif();
2013       // get local TopoDS_Shape
2014       TopoDS_Shape myLocShape;
2015       if(theMesh->HasShapeToMesh())
2016         myLocShape = GeomObjectToShape( theShapeObject );
2017       else
2018         return result._retn();;
2019
2020       // call implementation compute
2021       ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
2022       TSetOfInt shapeIds;
2023       ::MeshDimension aDim = (MeshDimension)theDimension;
2024       if ( myGen.Compute( myLocMesh, myLocShape, ::SMESH_Gen::COMPACT_MESH, aDim, &shapeIds ) )
2025       {
2026         int nbShapeId = shapeIds.size();
2027         theShapesId.length( nbShapeId );
2028         // iterates on shapes and collect mesh entities into mesh preview
2029         TSetOfInt::const_iterator idIt = shapeIds.begin();
2030         TSetOfInt::const_iterator idEnd = shapeIds.end();
2031         std::map< int, int > mapOfShIdNb;
2032         std::set< SMESH_TLink > setOfEdge;
2033         std::list< SMDSAbs_ElementType > listOfElemType;
2034         typedef map<const SMDS_MeshElement*, int > TNode2LocalIDMap;
2035         typedef TNode2LocalIDMap::iterator         TNodeLocalID;
2036         TNode2LocalIDMap mapNode2LocalID;
2037         list< TNodeLocalID > connectivity;
2038         int i, nbConnNodes = 0;
2039         std::set< const SMESH_subMesh* > setOfVSubMesh;
2040         // iterates on shapes
2041         for ( ; idIt != idEnd; idIt++ )
2042         {
2043           if ( mapOfShIdNb.find( *idIt ) != mapOfShIdNb.end() )
2044             continue;
2045           SMESH_subMesh* sm = myLocMesh.GetSubMeshContaining(*idIt);
2046           if ( !sm || !sm->IsMeshComputed() )
2047             continue;
2048
2049           const TopoDS_Shape& aSh = sm->GetSubShape();
2050           const int shDim = myGen.GetShapeDim( aSh );
2051           if ( shDim < 1 || shDim > theDimension )
2052             continue;
2053
2054           mapOfShIdNb[ *idIt ] = 0;
2055           theShapesId[ mapOfShIdNb.size() - 1 ] = *idIt;
2056
2057           SMESHDS_SubMesh* smDS = sm->GetSubMeshDS();
2058           if ( !smDS ) continue;
2059
2060           if ( theDimension == SMESH::DIM_2D )
2061           {
2062             SMDS_ElemIteratorPtr faceIt = smDS->GetElements();
2063             while ( faceIt->more() )
2064             {
2065               const SMDS_MeshElement* face = faceIt->next();
2066               int aNbNode = face->NbNodes();
2067               if ( aNbNode > 4 )
2068                 aNbNode /= 2; // do not take into account additional middle nodes
2069
2070               SMDS_MeshNode* node1 = (SMDS_MeshNode*)face->GetNode( 0 );
2071               for ( int nIndx = 0; nIndx < aNbNode; nIndx++ )
2072               {
2073                 SMDS_MeshNode* node2 = (SMDS_MeshNode*)face->GetNode( nIndx+1 < aNbNode ? nIndx+1 : 0 );
2074                 if ( setOfEdge.insert( SMESH_TLink ( node1, node2 ) ).second )
2075                 {
2076                   listOfElemType.push_back( SMDSAbs_Edge );
2077                   connectivity.push_back
2078                     ( mapNode2LocalID.insert( make_pair( node1, ++nbConnNodes)).first );
2079                   connectivity.push_back
2080                     ( mapNode2LocalID.insert( make_pair( node2, ++nbConnNodes)).first );
2081                 }
2082                 node1 = node2;
2083               }
2084             }
2085           }
2086           else if ( theDimension == SMESH::DIM_1D )
2087           {
2088             SMDS_NodeIteratorPtr nodeIt = smDS->GetNodes();
2089             while ( nodeIt->more() )
2090             {
2091               listOfElemType.push_back( SMDSAbs_Node );
2092               connectivity.push_back
2093                 ( mapNode2LocalID.insert( make_pair( nodeIt->next(), ++nbConnNodes)).first );
2094             }
2095             // add corner nodes by first vertex from edge
2096             SMESH_subMeshIteratorPtr edgeSmIt =
2097               sm->getDependsOnIterator(/*includeSelf*/false,
2098                                        /*complexShapeFirst*/false);
2099             while ( edgeSmIt->more() )
2100             {
2101               SMESH_subMesh* vertexSM = edgeSmIt->next();
2102               // check that vertex is not already treated
2103               if ( !setOfVSubMesh.insert( vertexSM ).second )
2104                 continue;
2105               if ( vertexSM->GetSubShape().ShapeType() != TopAbs_VERTEX )
2106                 continue;
2107
2108               const SMESHDS_SubMesh* vertexSmDS = vertexSM->GetSubMeshDS();
2109               SMDS_NodeIteratorPtr nodeIt = vertexSmDS->GetNodes();
2110               while ( nodeIt->more() )
2111               {
2112                 listOfElemType.push_back( SMDSAbs_Node );
2113                 connectivity.push_back
2114                   ( mapNode2LocalID.insert( make_pair( nodeIt->next(), ++nbConnNodes)).first );
2115               }
2116             }
2117           }
2118         }
2119
2120         // fill node coords and assign local ids to the nodes
2121         int nbNodes = mapNode2LocalID.size();
2122         result->nodesXYZ.length( nbNodes );
2123         TNodeLocalID node2ID = mapNode2LocalID.begin();
2124         for ( i = 0; i < nbNodes; ++i, ++node2ID ) {
2125           node2ID->second = i;
2126           const SMDS_MeshNode* node = (const SMDS_MeshNode*) node2ID->first;
2127           result->nodesXYZ[i].x = node->X();
2128           result->nodesXYZ[i].y = node->Y();
2129           result->nodesXYZ[i].z = node->Z();
2130         }
2131         // fill connectivity
2132         result->elementConnectivities.length( nbConnNodes );
2133         list< TNodeLocalID >::iterator connIt = connectivity.begin();
2134         for ( i = 0; i < nbConnNodes; ++i, ++connIt ) {
2135           result->elementConnectivities[i] = (*connIt)->second;
2136         }
2137
2138         // fill element types
2139         result->elementTypes.length( listOfElemType.size() );
2140         std::list< SMDSAbs_ElementType >::const_iterator typeIt = listOfElemType.begin();
2141         std::list< SMDSAbs_ElementType >::const_iterator typeEnd = listOfElemType.end();
2142         for ( i = 0; typeIt != typeEnd; ++i, ++typeIt )
2143         {
2144           SMDSAbs_ElementType elemType = *typeIt;
2145           result->elementTypes[i].SMDS_ElementType = (SMESH::ElementType)elemType;
2146           result->elementTypes[i].isPoly           = false;
2147           result->elementTypes[i].nbNodesInElement = elemType == SMDSAbs_Edge ? 2 : 1;
2148         }
2149
2150         // correct number of shapes
2151         theShapesId.length( mapOfShIdNb.size() );
2152       }
2153     }
2154   }
2155   catch ( std::bad_alloc ) {
2156     INFOS( "Precompute(): lack of memory" );
2157   }
2158   catch ( SALOME_Exception& S_ex ) {
2159     INFOS( "Precompute(): catch exception "<< S_ex.what() );
2160   }
2161   catch ( ... ) {
2162     INFOS( "Precompute(): unknown exception " );
2163   }
2164   return result._retn();
2165 }
2166
2167
2168 //=============================================================================
2169 /*!
2170  *  SMESH_Gen_i::Evaluate
2171  *
2172  *  Evaluate mesh on a shape
2173  */
2174 //=============================================================================
2175
2176 SMESH::long_array* SMESH_Gen_i::Evaluate(SMESH::SMESH_Mesh_ptr theMesh,
2177                                          GEOM::GEOM_Object_ptr theShapeObject)
2178 //                                     SMESH::long_array& theNbElems)
2179      throw ( SALOME::SALOME_Exception )
2180 {
2181   Unexpect aCatch(SALOME_SalomeException);
2182   if(MYDEBUG) MESSAGE( "SMESH_Gen_i::Evaluate" );
2183
2184   if ( CORBA::is_nil( theShapeObject ) && theMesh->HasShapeToMesh())
2185     THROW_SALOME_CORBA_EXCEPTION( "bad shape object reference",
2186                                   SALOME::BAD_PARAM );
2187
2188   if ( CORBA::is_nil( theMesh ) )
2189     THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference",
2190                                   SALOME::BAD_PARAM );
2191
2192   SMESH::long_array_var nbels = new SMESH::long_array;
2193   nbels->length(SMESH::Entity_Last);
2194   int i = SMESH::Entity_Node;
2195   for (; i < SMESH::Entity_Last; i++)
2196     nbels[i] = 0;
2197
2198   // Update Python script
2199   TPythonDump() << "theNbElems = " << this << ".Evaluate( "
2200                 << theMesh << ", " << theShapeObject << ")";
2201
2202   try {
2203     // get mesh servant
2204     SMESH_Mesh_i* meshServant = dynamic_cast<SMESH_Mesh_i*>( GetServant( theMesh ).in() );
2205     ASSERT( meshServant );
2206     if ( meshServant ) {
2207       meshServant->Load();
2208       // NPAL16168: "geometrical group edition from a submesh don't modifiy mesh computation"
2209       meshServant->CheckGeomModif();
2210       // get local TopoDS_Shape
2211       TopoDS_Shape myLocShape;
2212       if(theMesh->HasShapeToMesh())
2213         myLocShape = GeomObjectToShape( theShapeObject );
2214       else
2215         myLocShape = SMESH_Mesh::PseudoShape();
2216       // call implementation compute
2217       ::SMESH_Mesh& myLocMesh = meshServant->GetImpl();
2218       MapShapeNbElems aResMap;
2219       /*CORBA::Boolean ret =*/ myGen.Evaluate( myLocMesh, myLocShape, aResMap);
2220       MapShapeNbElemsItr anIt = aResMap.begin();
2221       for(; anIt!=aResMap.end(); anIt++) {
2222         const vector<int>& aVec = (*anIt).second;
2223         for ( i = SMESH::Entity_Node; i < (int)aVec.size(); i++ ) {
2224           int nbElem = aVec[i];
2225           if ( nbElem < 0 ) // algo failed, check that it has reported a message
2226           {
2227             SMESH_subMesh*            sm = anIt->first;
2228             SMESH_ComputeErrorPtr& error = sm->GetComputeError();
2229             const SMESH_Algo*       algo = sm->GetAlgo();
2230             if ( (algo && !error.get()) || error->IsOK() )
2231               error.reset( new SMESH_ComputeError( COMPERR_ALGO_FAILED,"Failed to evaluate",algo));
2232           }
2233           else
2234           {
2235             nbels[i] += aVec[i];
2236           }
2237         }
2238       }
2239       return nbels._retn();
2240     }
2241   }
2242   catch ( std::bad_alloc ) {
2243     INFOS( "Evaluate(): lack of memory" );
2244   }
2245   catch ( SALOME_Exception& S_ex ) {
2246     INFOS( "Evaluate(): catch exception "<< S_ex.what() );
2247   }
2248   catch ( ... ) {
2249     INFOS( "Evaluate(): unknown exception " );
2250   }
2251
2252   return nbels._retn();
2253 }
2254
2255 //================================================================================
2256 /*!
2257  * \brief Return geometrical object the given element is built on
2258  *  \param theMesh - the mesh the element is in
2259  *  \param theElementID - the element ID
2260  *  \param theGeomName - the name of the result geom object if it is not yet published
2261  *  \retval GEOM::GEOM_Object_ptr - the found or just published geom object (no need to UnRegister())
2262  */
2263 //================================================================================
2264
2265 GEOM::GEOM_Object_ptr
2266 SMESH_Gen_i::GetGeometryByMeshElement( SMESH::SMESH_Mesh_ptr  theMesh,
2267                                        CORBA::Long            theElementID,
2268                                        const char*            theGeomName)
2269   throw ( SALOME::SALOME_Exception )
2270 {
2271   Unexpect aCatch(SALOME_SalomeException);
2272
2273   GEOM::GEOM_Object_wrap geom = FindGeometryByMeshElement(theMesh, theElementID);
2274   if ( !geom->_is_nil() ) {
2275     GEOM::GEOM_Object_var mainShape = theMesh->GetShapeToMesh();
2276     GEOM::GEOM_Gen_ptr    geomGen   = GetGeomEngine();
2277
2278     // try to find the corresponding SObject
2279     SALOMEDS::SObject_wrap SObj = ObjectToSObject( myCurrentStudy, geom.in() );
2280     if ( SObj->_is_nil() ) // submesh can be not found even if published
2281     {
2282       // try to find published submesh
2283       GEOM::ListOfLong_var list = geom->GetSubShapeIndices();
2284       if ( !geom->IsMainShape() && list->length() == 1 ) {
2285         SALOMEDS::SObject_wrap mainSO = ObjectToSObject( myCurrentStudy, mainShape );
2286         SALOMEDS::ChildIterator_wrap it;
2287         if ( !mainSO->_is_nil() ) {
2288           it = myCurrentStudy->NewChildIterator( mainSO );
2289         }
2290         if ( !it->_is_nil() ) {
2291           for ( it->InitEx(true); it->More(); it->Next() ) {
2292             SALOMEDS::SObject_wrap      so = it->Value();
2293             CORBA::Object_var         obj = SObjectToObject( so );
2294             GEOM::GEOM_Object_var subGeom = GEOM::GEOM_Object::_narrow( obj );
2295             if ( !subGeom->_is_nil() ) {
2296               GEOM::ListOfLong_var subList = subGeom->GetSubShapeIndices();
2297               if ( subList->length() == 1 && list[0] == subList[0] ) {
2298                 SObj = so;
2299                 geom = subGeom;
2300                 break;
2301               }
2302             }
2303           }
2304         }
2305       }
2306     }
2307     if ( SObj->_is_nil() ) // publish a new subshape
2308       SObj = geomGen->AddInStudy( myCurrentStudy, geom, theGeomName, mainShape );
2309
2310     // return only published geometry
2311     if ( !SObj->_is_nil() ) {
2312       //return geom._retn(); -- servant of geom must be UnRegister()ed;
2313       CORBA::Object_var    obj = SObjectToObject( SObj );
2314       GEOM::GEOM_Object_var go = GEOM::GEOM_Object::_narrow( obj );
2315       return go._retn();
2316     }
2317   }
2318   return GEOM::GEOM_Object::_nil();
2319 }
2320
2321 //================================================================================
2322 /*!
2323  * \brief Return geometrical object the given element is built on.
2324  *  \param theMesh - the mesh the element is in
2325  *  \param theElementID - the element ID
2326  *  \retval GEOM::GEOM_Object_ptr - the found or created (UnRegister()!) geom object
2327  */
2328 //================================================================================
2329
2330 GEOM::GEOM_Object_ptr
2331 SMESH_Gen_i::FindGeometryByMeshElement( SMESH::SMESH_Mesh_ptr  theMesh,
2332                                         CORBA::Long            theElementID)
2333   throw ( SALOME::SALOME_Exception )
2334 {
2335   Unexpect aCatch(SALOME_SalomeException);
2336   if ( CORBA::is_nil( theMesh ) )
2337     THROW_SALOME_CORBA_EXCEPTION( "bad Mesh reference", SALOME::BAD_PARAM );
2338
2339   GEOM::GEOM_Object_var mainShape = theMesh->GetShapeToMesh();
2340   GEOM::GEOM_Gen_ptr    geomGen   = GetGeomEngine();
2341
2342   // get a core mesh DS
2343   SMESH_Mesh_i* meshServant = SMESH::DownCast<SMESH_Mesh_i*>( theMesh );
2344   if ( meshServant && !geomGen->_is_nil() && !mainShape->_is_nil() )
2345   {
2346     ::SMESH_Mesh & mesh = meshServant->GetImpl();
2347     SMESHDS_Mesh* meshDS = mesh.GetMeshDS();
2348     // find the element in mesh
2349     if ( const SMDS_MeshElement * elem = meshDS->FindElement( theElementID ) ) {
2350       // find a shape id by the element
2351       if ( int shapeID = ::SMESH_MeshEditor( &mesh ).FindShape( elem )) {
2352         // get a geom object by the shape id
2353         GEOM::GEOM_Object_var geom = ShapeToGeomObject( meshDS->IndexToShape( shapeID ));
2354         if ( geom->_is_nil() ) {
2355           // try to find a published sub-shape
2356           SALOMEDS::SObject_wrap mainSO = ObjectToSObject( myCurrentStudy, mainShape );
2357           SALOMEDS::ChildIterator_wrap it;
2358           if ( !mainSO->_is_nil() ) {
2359             it = myCurrentStudy->NewChildIterator( mainSO );
2360           }
2361           if ( !it->_is_nil() ) {
2362             for ( it->InitEx(true); it->More(); it->Next() ) {
2363               SALOMEDS::SObject_wrap      so = it->Value();
2364               CORBA::Object_var         obj = SObjectToObject( so );
2365               GEOM::GEOM_Object_var subGeom = GEOM::GEOM_Object::_narrow( obj );
2366               if ( !subGeom->_is_nil() ) {
2367                 GEOM::ListOfLong_var subList = subGeom->GetSubShapeIndices();
2368                 if ( subList->length() == 1 && shapeID == subList[0] ) {
2369                   geom = subGeom;
2370                   break;
2371                 }
2372               }
2373             }
2374           }
2375         }
2376         if ( geom->_is_nil() ) {
2377           // explode
2378           GEOM::GEOM_IShapesOperations_wrap op =
2379             geomGen->GetIShapesOperations( GetCurrentStudyID() );
2380           if ( !op->_is_nil() )
2381             geom = op->GetSubShape( mainShape, shapeID );
2382         }
2383         else {
2384           geom->Register();
2385         }
2386         if ( !geom->_is_nil() ) {
2387           GeomObjectToShape( geom ); // let geom client remember the found shape
2388           return geom._retn();
2389         }
2390       }
2391     }
2392   }
2393   return GEOM::GEOM_Object::_nil();
2394 }
2395
2396 //================================================================================
2397 /*!
2398  *  SMESH_Gen_i::Concatenate
2399  *
2400  *  Concatenate the given meshes into one mesh
2401  */
2402 //================================================================================
2403
2404 SMESH::SMESH_Mesh_ptr
2405 SMESH_Gen_i::Concatenate(const SMESH::ListOfIDSources& theMeshesArray,
2406                          CORBA::Boolean                theUniteIdenticalGroups,
2407                          CORBA::Boolean                theMergeNodesAndElements,
2408                          CORBA::Double                 theMergeTolerance)
2409   throw ( SALOME::SALOME_Exception )
2410 {
2411   return ConcatenateCommon(theMeshesArray,
2412                            theUniteIdenticalGroups,
2413                            theMergeNodesAndElements,
2414                            theMergeTolerance,
2415                            false);
2416 }
2417
2418 //================================================================================
2419 /*!
2420  *  SMESH_Gen_i::ConcatenateWithGroups
2421  *
2422  *  Concatenate the given meshes into one mesh
2423  *  Create the groups of all elements from initial meshes
2424  */
2425 //================================================================================
2426
2427 SMESH::SMESH_Mesh_ptr
2428 SMESH_Gen_i::ConcatenateWithGroups(const SMESH::ListOfIDSources& theMeshesArray,
2429                                    CORBA::Boolean                theUniteIdenticalGroups,
2430                                    CORBA::Boolean                theMergeNodesAndElements,
2431                                    CORBA::Double                 theMergeTolerance)
2432   throw ( SALOME::SALOME_Exception )
2433 {
2434   return ConcatenateCommon(theMeshesArray,
2435                            theUniteIdenticalGroups,
2436                            theMergeNodesAndElements,
2437                            theMergeTolerance,
2438                            true);
2439 }
2440
2441 //================================================================================
2442 /*!
2443  *  SMESH_Gen_i::ConcatenateCommon
2444  *
2445  *  Concatenate the given meshes into one mesh
2446  */
2447 //================================================================================
2448
2449 SMESH::SMESH_Mesh_ptr
2450 SMESH_Gen_i::ConcatenateCommon(const SMESH::ListOfIDSources& theMeshesArray,
2451                                CORBA::Boolean                theUniteIdenticalGroups,
2452                                CORBA::Boolean                theMergeNodesAndElements,
2453                                CORBA::Double                 theMergeTolerance,
2454                                CORBA::Boolean                theCommonGroups)
2455   throw ( SALOME::SALOME_Exception )
2456 {
2457   std::unique_ptr< TPythonDump > pPythonDump( new TPythonDump );
2458   TPythonDump& pythonDump = *pPythonDump; // prevent dump of called methods
2459
2460   // create mesh
2461   SMESH::SMESH_Mesh_var newMesh = CreateEmptyMesh();
2462   SMESH_Mesh_i*         newImpl = SMESH::DownCast<SMESH_Mesh_i*>( newMesh );
2463   if ( !newImpl ) return newMesh._retn();
2464
2465   ::SMESH_Mesh&   locMesh = newImpl->GetImpl();
2466   SMESHDS_Mesh* newMeshDS = locMesh.GetMeshDS();
2467
2468   typedef std::list<SMESH::SMESH_Group_var>          TListOfNewGroups;
2469   typedef std::pair<string, SMESH::ElementType >     TNameAndType;
2470   typedef std::map< TNameAndType, TListOfNewGroups > TGroupsMap;
2471   TGroupsMap       groupsMap;
2472   TListOfNewGroups listOfNewGroups;
2473
2474   ::SMESH_MeshEditor               newEditor( &locMesh );
2475   ::SMESH_MeshEditor::ElemFeatures elemType;
2476
2477   // loop on sub-meshes
2478   for ( CORBA::ULong i = 0; i < theMeshesArray.length(); i++ )
2479   {
2480     if ( CORBA::is_nil( theMeshesArray[i] )) continue;
2481     SMESH::SMESH_Mesh_var initMesh = theMeshesArray[i]->GetMesh();
2482     SMESH_Mesh_i*         initImpl = SMESH::DownCast<SMESH_Mesh_i*>( initMesh );
2483     if ( !initImpl ) continue;
2484     initImpl->Load();
2485
2486     // assure that IDs increments by one during iteration
2487     ::SMESH_Mesh& initLocMesh = initImpl->GetImpl();
2488     SMESHDS_Mesh*  initMeshDS = initLocMesh.GetMeshDS();
2489     if ( initMeshDS->MaxNodeID()    > initMeshDS->NbNodes() ||
2490          initMeshDS->MaxElementID() > initMeshDS->NbElements() )
2491     {
2492       initMeshDS->Modified();
2493       initMeshDS->CompactMesh();
2494     }
2495
2496     // remember nb of elements before filling in
2497     SMESH::long_array_var prevState =  newMesh->GetNbElementsByType();
2498
2499     // copy nodes
2500
2501     std::vector< const SMDS_MeshElement* > newNodes( initMeshDS->NbNodes() + 1, 0 );
2502     SMDS_ElemIteratorPtr elemIt = initImpl->GetElements( theMeshesArray[i], SMESH::NODE );
2503     while ( elemIt->more() )
2504     {
2505       SMESH_NodeXYZ node = elemIt->next();
2506       newNodes[ node->GetID() ] = newMeshDS->AddNode( node.X(), node.Y(), node.Z() );
2507     }
2508
2509     // copy elements
2510
2511     std::vector< const SMDS_MeshElement* > newElems( initMeshDS->NbElements() + 1, 0 );
2512     elemIt = initImpl->GetElements( theMeshesArray[i], SMESH::ALL );
2513     while ( elemIt->more() )
2514     {
2515       const SMDS_MeshElement* elem = elemIt->next();
2516       elemType.myNodes.resize( elem->NbNodes() );
2517
2518       SMDS_NodeIteratorPtr itNodes = elem->nodeIterator();
2519       for ( int k = 0; itNodes->more(); k++)
2520       {
2521         const SMDS_MeshNode* node = itNodes->next();
2522         elemType.myNodes[ k ] = static_cast< const SMDS_MeshNode*> ( newNodes[ node->GetID() ]);
2523       }
2524
2525       // creates a corresponding element on existent nodes in new mesh
2526       newElems[ elem->GetID() ] =
2527         newEditor.AddElement( elemType.myNodes, elemType.Init( elem, /*basicOnly=*/false ));
2528     }
2529     newEditor.ClearLastCreated(); // forget the history
2530
2531
2532     // create groups of just added elements
2533     SMESH::SMESH_Group_var newGroup;
2534     SMESH::ElementType     groupType;
2535     if ( theCommonGroups )
2536     {
2537       // type names
2538       const char* typeNames[] = { "All","Nodes","Edges","Faces","Volumes","0DElems","Balls" };
2539       { // check of typeNames: compilation failure mains that NB_ELEMENT_TYPES changed:
2540         const int nbNames = sizeof(typeNames) / sizeof(const char*);
2541         int _assert[( nbNames == SMESH::NB_ELEMENT_TYPES ) ? 2 : -1 ]; _assert[0]=_assert[1]=0;
2542       }
2543
2544       SMESH::long_array_var curState = newMesh->GetNbElementsByType();
2545
2546       for( groupType = SMESH::NODE;
2547            groupType < SMESH::NB_ELEMENT_TYPES;
2548            groupType = (SMESH::ElementType)( groupType + 1 ))
2549       {
2550         if ( curState[ groupType ] <= prevState[ groupType ])
2551           continue; // no elements of groupType added from the i-th mesh
2552
2553         // make a group name
2554         std::string groupName = "Gr";
2555         SALOMEDS::SObject_wrap meshSO = ObjectToSObject( myCurrentStudy, theMeshesArray[i] );
2556         if ( meshSO ) {
2557           CORBA::String_var name = meshSO->GetName();
2558           groupName += name;
2559         }
2560         groupName += "_";
2561         groupName += typeNames[ groupType ];
2562
2563         // make and fill a group
2564         newGroup = newImpl->CreateGroup( groupType, groupName.c_str() );
2565         std::vector< const SMDS_MeshElement* > & elemVec =
2566           ( groupType == SMESH::NODE ) ? newNodes : newElems;
2567         if ( SMESH_Group_i* grp_i = SMESH::DownCast<SMESH_Group_i*>( newGroup ))
2568         {
2569           if ( SMESHDS_Group* grpDS = dynamic_cast<SMESHDS_Group*>( grp_i->GetGroupDS() ))
2570           {
2571             for ( size_t j = 0; j < elemVec.size(); ++j )
2572             {
2573               if ( elemVec[j] && elemVec[j]->GetType() == grpDS->GetType() )
2574                 grpDS->Add( elemVec[j] );
2575             }
2576           }
2577         }
2578         listOfNewGroups.clear();
2579         listOfNewGroups.push_back( newGroup );
2580         groupsMap.insert( std::make_pair( TNameAndType( groupName, groupType ),
2581                                           listOfNewGroups ));
2582       }
2583     }
2584
2585     if ( SMESH_Mesh_i* initImpl = SMESH::DownCast<SMESH_Mesh_i*>( theMeshesArray[i] ))
2586     {
2587       // copy groups
2588
2589       SMESH::SMESH_GroupBase_ptr group;
2590       CORBA::String_var          groupName;
2591       SMESH::long_array_var newIDs = new SMESH::long_array();
2592
2593       // loop on groups of a source mesh
2594       SMESH::ListOfGroups_var listOfGroups = initImpl->GetGroups();
2595       for ( CORBA::ULong iG = 0; iG < listOfGroups->length(); iG++ )
2596       {
2597         group     = listOfGroups[iG];
2598         groupType = group->GetType();
2599         groupName = group->GetName();
2600         std::string name = groupName.in();
2601
2602         // convert a list of IDs
2603         newIDs->length( group->Size() );
2604         std::vector< const SMDS_MeshElement* > & elemVec =
2605           ( groupType == SMESH::NODE ) ? newNodes : newElems;
2606         SMDS_ElemIteratorPtr itGrElems = initImpl->GetElements( group, SMESH::ALL );
2607         int nbElems = 0;
2608         while ( itGrElems->more() )
2609         {
2610           const SMDS_MeshElement*    elem = itGrElems->next();
2611           const SMDS_MeshElement* newElem = elemVec[ elem->GetID() ];
2612           if ( newElem )
2613             newIDs[ nbElems++ ] = newElem->GetID();
2614         }
2615         newIDs->length( nbElems );
2616
2617         // check that a current group name and type don't have identical ones in final mesh
2618         listOfNewGroups.clear();
2619         TNameAndType nameAndType( name, groupType );
2620         TGroupsMap::iterator anIter = groupsMap.find( nameAndType );
2621         if ( anIter == groupsMap.end() )
2622         {
2623           // add a new group in the mesh
2624           newGroup = newImpl->CreateGroup( groupType, groupName.in() );
2625           newGroup->Add( newIDs );
2626
2627           listOfNewGroups.push_back( newGroup );
2628           groupsMap.insert( std::make_pair( nameAndType, listOfNewGroups ));
2629         }
2630         else if ( theUniteIdenticalGroups )
2631         {
2632           // unite identical groups
2633           TListOfNewGroups& aNewGroups = anIter->second;
2634           aNewGroups.front()->Add( newIDs );
2635         }
2636         else
2637         {
2638           // rename identical groups
2639           newGroup = newImpl->CreateGroup( groupType, groupName );
2640           newGroup->Add( newIDs );
2641
2642           TListOfNewGroups& newGroups = anIter->second;
2643           std::string newGroupName;
2644           if ( newGroups.size() == 1 )
2645           {
2646             newGroupName = name + "_1";
2647             newGroups.front()->SetName( newGroupName.c_str() );
2648           }
2649           newGroupName = name + "_" + SMESH_Comment( newGroups.size() + 1 );
2650           newGroup->SetName( newGroupName.c_str() );
2651           newGroups.push_back( newGroup );
2652         }
2653       } // loop on groups
2654     } // if an IDSource is a mesh
2655   } //meshes loop
2656
2657   if ( theMergeNodesAndElements ) // merge nodes
2658   {
2659     TIDSortedNodeSet meshNodes; // no input nodes == treat all
2660     SMESH_MeshEditor::TListOfListOfNodes groupsOfNodes;
2661     newEditor.FindCoincidentNodes( meshNodes, theMergeTolerance, groupsOfNodes,
2662                                    /*SeparateCornersAndMedium=*/ false );
2663     newEditor.MergeNodes( groupsOfNodes );
2664     // merge elements
2665     newEditor.MergeEqualElements();
2666   }
2667
2668   // Update Python script
2669   pythonDump << newMesh << " = " << this
2670              << "." << ( theCommonGroups ? "ConcatenateWithGroups" : "Concatenate" ) << "("
2671              << theMeshesArray << ", "
2672              << theUniteIdenticalGroups << ", "
2673              << theMergeNodesAndElements << ", "
2674              << TVar( theMergeTolerance ) << ")";
2675
2676   pPythonDump.reset(); // enable python dump from GetGroups()
2677
2678   // 0020577: EDF 1164 SMESH: Bad dump of concatenate with create common groups
2679   if ( !newMesh->_is_nil() )
2680   {
2681     SMESH::ListOfGroups_var groups = newMesh->GetGroups();
2682   }
2683
2684   // IPAL21468 Change icon of compound because it need not be computed.
2685   SALOMEDS::SObject_wrap meshSO = ObjectToSObject( myCurrentStudy, newMesh );
2686   SetPixMap( meshSO, "ICON_SMESH_TREE_MESH" );
2687
2688   newMeshDS->Modified();
2689
2690   return newMesh._retn();
2691 }
2692
2693 //================================================================================
2694 /*!
2695  * \brief Create a mesh by copying a part of another mesh
2696  *  \param meshPart - a part of mesh to copy
2697  *  \param toCopyGroups - to create in the new mesh groups
2698  *                        the copied elements belongs to
2699  *  \param toKeepIDs - to preserve IDs of the copied elements or not
2700  *  \retval SMESH::SMESH_Mesh_ptr - the new mesh
2701  */
2702 //================================================================================
2703
2704 SMESH::SMESH_Mesh_ptr SMESH_Gen_i::CopyMesh(SMESH::SMESH_IDSource_ptr meshPart,
2705                                             const char*               meshName,
2706                                             CORBA::Boolean            toCopyGroups,
2707                                             CORBA::Boolean            toKeepIDs)
2708 {
2709   Unexpect aCatch(SALOME_SalomeException);
2710
2711   TPythonDump* pyDump = new TPythonDump; // prevent dump from CreateMesh()
2712
2713   // 1. Get source mesh
2714
2715   if ( CORBA::is_nil( meshPart ))
2716     THROW_SALOME_CORBA_EXCEPTION( "bad IDSource", SALOME::BAD_PARAM );
2717
2718   SMESH::SMESH_Mesh_var srcMesh = meshPart->GetMesh();
2719   SMESH_Mesh_i*       srcMesh_i = SMESH::DownCast<SMESH_Mesh_i*>( srcMesh );
2720   if ( !srcMesh_i )
2721     THROW_SALOME_CORBA_EXCEPTION( "bad mesh of IDSource", SALOME::BAD_PARAM );
2722
2723   SMESHDS_Mesh* srcMeshDS = srcMesh_i->GetImpl().GetMeshDS();
2724
2725   // 2. Make a new mesh
2726
2727   SMESH::SMESH_Mesh_var newMesh = CreateMesh(GEOM::GEOM_Object::_nil());
2728   SMESH_Mesh_i*       newMesh_i = SMESH::DownCast<SMESH_Mesh_i*>( newMesh );
2729   if ( !newMesh_i )
2730     THROW_SALOME_CORBA_EXCEPTION( "can't create a mesh", SALOME::INTERNAL_ERROR );
2731   SALOMEDS::SObject_wrap meshSO = ObjectToSObject(myCurrentStudy, newMesh );
2732   if ( !meshSO->_is_nil() )
2733   {
2734     SetName( meshSO, meshName, "Mesh" );
2735     SetPixMap( meshSO, "ICON_SMESH_TREE_MESH_IMPORTED");
2736   }
2737   SMESHDS_Mesh* newMeshDS = newMesh_i->GetImpl().GetMeshDS();
2738   ::SMESH_MeshEditor editor( &newMesh_i->GetImpl() );
2739   ::SMESH_MeshEditor::ElemFeatures elemType;
2740
2741   // 3. Get elements to copy
2742
2743   SMDS_ElemIteratorPtr srcElemIt; SMDS_NodeIteratorPtr srcNodeIt;
2744   TIDSortedElemSet srcElems;
2745   SMESH::array_of_ElementType_var srcElemTypes = meshPart->GetTypes();
2746   if ( SMESH::DownCast<SMESH_Mesh_i*>( meshPart ))
2747   {
2748     srcMesh_i->Load();
2749     srcElemIt = srcMeshDS->elementsIterator();
2750     srcNodeIt = srcMeshDS->nodesIterator();
2751   }
2752   else
2753   {
2754     SMESH::long_array_var ids = meshPart->GetIDs();
2755     if ( srcElemTypes->length() == 1 && srcElemTypes[0] == SMESH::NODE ) // group of nodes
2756     {
2757       for ( CORBA::ULong i=0; i < ids->length(); i++ )
2758         if ( const SMDS_MeshElement * elem = srcMeshDS->FindNode( ids[i] ))
2759           srcElems.insert( elem );
2760     }
2761     else
2762     {
2763       for ( CORBA::ULong i = 0; i < ids->length(); i++ )
2764         if ( const SMDS_MeshElement * elem = srcMeshDS->FindElement( ids[i] ))
2765           srcElems.insert( elem );
2766     }
2767     if ( srcElems.empty() )
2768       return newMesh._retn();
2769
2770     typedef SMDS_SetIterator< SMDS_pElement, TIDSortedElemSet::const_iterator > ElIter;
2771     srcElemIt = SMDS_ElemIteratorPtr( new ElIter( srcElems.begin(), srcElems.end() ));
2772   }
2773
2774   // 4. Copy elements
2775
2776   typedef map<SMDS_pElement, SMDS_pElement, TIDCompare> TE2EMap;
2777   TE2EMap e2eMapByType[ SMDSAbs_NbElementTypes ];
2778   TE2EMap& n2nMap = e2eMapByType[ SMDSAbs_Node ];
2779   int iN;
2780   const SMDS_MeshNode *nSrc, *nTgt;
2781   vector< const SMDS_MeshNode* > nodes;
2782   while ( srcElemIt->more() )
2783   {
2784     const SMDS_MeshElement * elem = srcElemIt->next();
2785     // find / add nodes
2786     nodes.resize( elem->NbNodes());
2787     SMDS_ElemIteratorPtr nIt = elem->nodesIterator();
2788     if ( toKeepIDs ) {
2789       for ( iN = 0; nIt->more(); ++iN )
2790       {
2791         nSrc = static_cast<const SMDS_MeshNode*>( nIt->next() );
2792         nTgt = newMeshDS->FindNode( nSrc->GetID());
2793         if ( !nTgt )
2794           nTgt = newMeshDS->AddNodeWithID( nSrc->X(), nSrc->Y(), nSrc->Z(), nSrc->GetID());
2795         nodes[ iN ] = nTgt;
2796       }
2797     }
2798     else {
2799       for ( iN = 0; nIt->more(); ++iN )
2800       {
2801         nSrc = static_cast<const SMDS_MeshNode*>( nIt->next() );
2802         TE2EMap::iterator n2n = n2nMap.insert( make_pair( nSrc, SMDS_pNode(0) )).first;
2803         if ( !n2n->second )
2804           n2n->second = newMeshDS->AddNode( nSrc->X(), nSrc->Y(), nSrc->Z() );
2805         nodes[ iN ] = (const SMDS_MeshNode*) n2n->second;
2806       }
2807     }
2808     // add elements
2809     if ( elem->GetType() != SMDSAbs_Node )
2810     {
2811       elemType.Init( elem, /*basicOnly=*/false );
2812       if ( toKeepIDs ) elemType.SetID( elem->GetID() );
2813
2814       const SMDS_MeshElement * newElem = editor.AddElement( nodes, elemType );
2815       if ( toCopyGroups && !toKeepIDs )
2816         e2eMapByType[ elem->GetType() ].insert( make_pair( elem, newElem ));
2817     }
2818   } // while ( srcElemIt->more() )
2819
2820   // 4(b). Copy free nodes
2821
2822   if ( srcNodeIt && srcMeshDS->NbNodes() != newMeshDS->NbNodes() )
2823   {
2824     while ( srcNodeIt->more() )
2825     {
2826       nSrc = srcNodeIt->next();
2827       if ( nSrc->NbInverseElements() == 0 )
2828       {
2829         if ( toKeepIDs )
2830           nTgt = newMeshDS->AddNodeWithID( nSrc->X(), nSrc->Y(), nSrc->Z(), nSrc->GetID());
2831         else
2832           n2nMap[ nSrc ] = newMeshDS->AddNode( nSrc->X(), nSrc->Y(), nSrc->Z() );
2833       }
2834     }
2835   }
2836
2837   // 5. Copy groups
2838
2839   int nbNewGroups = 0;
2840   if ( toCopyGroups )
2841   {
2842     SMESH_Mesh::GroupIteratorPtr gIt = srcMesh_i->GetImpl().GetGroups();
2843     while ( gIt->more() )
2844     {
2845       SMESH_Group* group = gIt->next();
2846       const SMESHDS_GroupBase* groupDS = group->GetGroupDS();
2847
2848       // Check group type. We copy nodal groups containing nodes of copied element
2849       SMDSAbs_ElementType groupType = groupDS->GetType();
2850       if ( groupType != SMDSAbs_Node &&
2851            newMeshDS->GetMeshInfo().NbElements( groupType ) == 0 )
2852         continue; // group type differs from types of meshPart
2853
2854       // Find copied elements in the group
2855       vector< const SMDS_MeshElement* > groupElems;
2856       SMDS_ElemIteratorPtr eIt = groupDS->GetElements();
2857       if ( toKeepIDs )
2858       {
2859         const SMDS_MeshElement* foundElem;
2860         if ( groupType == SMDSAbs_Node )
2861         {
2862           while ( eIt->more() )
2863             if (( foundElem = newMeshDS->FindNode( eIt->next()->GetID() )))
2864               groupElems.push_back( foundElem );
2865         }
2866         else
2867         {
2868           while ( eIt->more() )
2869             if (( foundElem = newMeshDS->FindElement( eIt->next()->GetID() )))
2870               groupElems.push_back( foundElem );
2871         }
2872       }
2873       else
2874       {
2875         TE2EMap & e2eMap = e2eMapByType[ groupDS->GetType() ];
2876         if ( e2eMap.empty() ) continue;
2877         int minID = e2eMap.begin()->first->GetID();
2878         int maxID = e2eMap.rbegin()->first->GetID();
2879         TE2EMap::iterator e2e;
2880         while ( eIt->more() && groupElems.size() < e2eMap.size())
2881         {
2882           const SMDS_MeshElement* e = eIt->next();
2883           if ( e->GetID() < minID || e->GetID() > maxID ) continue;
2884           if ((e2e = e2eMap.find( e )) != e2eMap.end())
2885             groupElems.push_back( e2e->second );
2886         }
2887       }
2888       // Make a new group
2889       if ( !groupElems.empty() )
2890       {
2891         SMESH::SMESH_Group_var newGroupObj =
2892           newMesh->CreateGroup( SMESH::ElementType(groupType), group->GetName() );
2893         if ( SMESH_GroupBase_i* newGroup_i = SMESH::DownCast<SMESH_GroupBase_i*>( newGroupObj))
2894         {
2895           SMESHDS_GroupBase * newGroupDS = newGroup_i->GetGroupDS();
2896           SMDS_MeshGroup& smdsGroup = ((SMESHDS_Group*)newGroupDS)->SMDSGroup();
2897           for ( unsigned i = 0; i < groupElems.size(); ++i )
2898             smdsGroup.Add( groupElems[i] );
2899
2900           nbNewGroups++;
2901         }
2902       }
2903     }
2904   }
2905
2906   newMeshDS->Modified();
2907
2908   *pyDump << newMesh << " = " << this
2909           << ".CopyMesh( " << meshPart << ", "
2910           << "'" << meshName << "', "
2911           << toCopyGroups << ", "
2912           << toKeepIDs << ")";
2913
2914   delete pyDump; pyDump = 0; // allow dump in GetGroups()
2915
2916   if ( nbNewGroups > 0 ) // dump created groups
2917     SMESH::ListOfGroups_var groups = newMesh->GetGroups();
2918
2919   return newMesh._retn();
2920 }
2921
2922 //================================================================================
2923 /*!
2924  *  SMESH_Gen_i::GetMEDVersion
2925  *
2926  *  Get MED version of the file by its name
2927  */
2928 //================================================================================
2929 CORBA::Boolean SMESH_Gen_i::GetMEDVersion(const char* theFileName,
2930                                           SMESH::MED_VERSION& theVersion)
2931 {
2932   theVersion = SMESH::MED_V2_1;
2933   MED::EVersion aVersion = MED::GetVersionId( theFileName );
2934   switch( aVersion ) {
2935     case MED::eV2_1     : theVersion = SMESH::MED_V2_1;    return true;
2936     case MED::eV2_2     : theVersion = SMESH::MED_V2_2;    return true;
2937     case MED::eLATEST   : theVersion = SMESH::MED_LATEST;  return true;
2938     case MED::eMINOR_0  : theVersion = SMESH::MED_MINOR_0; return true;
2939     case MED::eMINOR_1  : theVersion = SMESH::MED_MINOR_1; return true;
2940     case MED::eMINOR_2  : theVersion = SMESH::MED_MINOR_2; return true;
2941     case MED::eMINOR_3  : theVersion = SMESH::MED_MINOR_3; return true;
2942     case MED::eMINOR_4  : theVersion = SMESH::MED_MINOR_4; return true;
2943     case MED::eMINOR_5  : theVersion = SMESH::MED_MINOR_5; return true;
2944     case MED::eMINOR_6  : theVersion = SMESH::MED_MINOR_6; return true;
2945     case MED::eMINOR_7  : theVersion = SMESH::MED_MINOR_7; return true;
2946     case MED::eMINOR_8  : theVersion = SMESH::MED_MINOR_8; return true;
2947     case MED::eMINOR_9  : theVersion = SMESH::MED_MINOR_9; return true;
2948     case MED::eVUnknown : return false;
2949   }
2950   return false;
2951 }
2952
2953 //================================================================================
2954 /*!
2955  *  SMESH_Gen_i::GetMeshNames
2956  *
2957  *  Get names of meshes defined in file with the specified name
2958  */
2959 //================================================================================
2960 SMESH::string_array* SMESH_Gen_i::GetMeshNames(const char* theFileName)
2961 {
2962   SMESH::string_array_var aResult = new SMESH::string_array();
2963   MED::PWrapper aMed = MED::CrWrapper( theFileName );
2964   MED::TErr anErr;
2965   MED::TInt aNbMeshes = aMed->GetNbMeshes( &anErr );
2966   if( anErr >= 0 ) {
2967     aResult->length( aNbMeshes );
2968     for( MED::TInt i = 0; i < aNbMeshes; i++ ) {
2969       MED::PMeshInfo aMeshInfo = aMed->GetPMeshInfo( i+1 );
2970       aResult[i] = CORBA::string_dup( aMeshInfo->GetName().c_str() );
2971     }
2972   }
2973   return aResult._retn();
2974 }
2975
2976 //=============================================================================
2977 /*!
2978  *  SMESH_Gen_i::Save
2979  *
2980  *  Save SMESH module's data
2981  */
2982 //=============================================================================
2983 SALOMEDS::TMPFile* SMESH_Gen_i::Save( SALOMEDS::SComponent_ptr theComponent,
2984                                       const char*              theURL,
2985                                       bool                     isMultiFile )
2986 {
2987   //  ASSERT( theComponent->GetStudy()->StudyId() == myCurrentStudy->StudyId() )
2988   // san -- in case <myCurrentStudy> differs from theComponent's study,
2989   // use that of the component
2990   if ( theComponent->GetStudy()->StudyId() != GetCurrentStudyID() )
2991     SetCurrentStudy( theComponent->GetStudy() );
2992
2993   // Store study contents as a set of python commands
2994   SavePython(myCurrentStudy);
2995
2996   StudyContext* myStudyContext = GetCurrentStudyContext();
2997
2998   // Declare a byte stream
2999   SALOMEDS::TMPFile_var aStreamFile;
3000
3001   // Obtain a temporary dir
3002   TCollection_AsciiString tmpDir =
3003     ( isMultiFile ) ? TCollection_AsciiString( ( char* )theURL ) : ( char* )SALOMEDS_Tool::GetTmpDir().c_str();
3004
3005   // Create a sequence of files processed
3006   SALOMEDS::ListOfFileNames_var aFileSeq = new SALOMEDS::ListOfFileNames;
3007   aFileSeq->length( NUM_TMP_FILES );
3008
3009   TCollection_AsciiString aStudyName( "" );
3010   if ( isMultiFile )
3011     aStudyName = ( (char*)SALOMEDS_Tool::GetNameFromPath( myCurrentStudy->URL() ).c_str() );
3012
3013   // Set names of temporary files
3014   TCollection_AsciiString filename =
3015     aStudyName + TCollection_AsciiString( "_SMESH.hdf" );        // for SMESH data itself
3016   TCollection_AsciiString meshfile =
3017     aStudyName + TCollection_AsciiString( "_SMESH_Mesh.med" );   // for mesh data to be stored in MED file
3018   aFileSeq[ 0 ] = CORBA::string_dup( filename.ToCString() );
3019   aFileSeq[ 1 ] = CORBA::string_dup( meshfile.ToCString() );
3020   filename = tmpDir + filename;
3021   meshfile = tmpDir + meshfile;
3022
3023   HDFfile*    aFile;
3024   HDFdataset* aDataset;
3025   HDFgroup*   aTopGroup;
3026   HDFgroup*   aGroup;
3027   HDFgroup*   aSubGroup;
3028   HDFgroup*   aSubSubGroup;
3029   hdf_size    aSize[ 1 ];
3030
3031
3032   //Remove the files if they exist: BugID: 11225
3033 #ifndef WIN32 /* unix functionality */
3034   TCollection_AsciiString cmd("rm -f \"");
3035 #else /* windows */
3036   TCollection_AsciiString cmd("del /F \"");
3037 #endif
3038
3039   cmd+=filename;
3040   cmd+="\" \"";
3041   cmd+=meshfile;
3042   cmd+="\"";
3043   system(cmd.ToCString());
3044
3045   // MED writer to be used by storage process
3046   DriverMED_W_SMESHDS_Mesh myWriter;
3047   myWriter.SetFile( meshfile.ToCString() );
3048
3049   // IMP issue 20918
3050   // SetStoreName() to groups before storing hypotheses to let them refer to
3051   // groups using "store name", which is "Group <group_persistent_id>"
3052   {
3053     SALOMEDS::ChildIterator_wrap itBig = myCurrentStudy->NewChildIterator( theComponent );
3054     for ( ; itBig->More(); itBig->Next() ) {
3055       SALOMEDS::SObject_wrap gotBranch = itBig->Value();
3056       if ( gotBranch->Tag() > GetAlgorithmsRootTag() ) {
3057         CORBA::Object_var anObject = SObjectToObject( gotBranch );
3058         if ( !CORBA::is_nil( anObject ) ) {
3059           SMESH::SMESH_Mesh_var myMesh = SMESH::SMESH_Mesh::_narrow( anObject ) ;
3060           if ( !myMesh->_is_nil() ) {
3061             myMesh->Load(); // load from study file if not yet done
3062             TPythonDump pd; // not to dump GetGroups()
3063             SMESH::ListOfGroups_var groups = myMesh->GetGroups();
3064             pd << ""; // to avoid optimizing pd out
3065             for ( CORBA::ULong i = 0; i < groups->length(); ++i )
3066             {
3067               SMESH_GroupBase_i* grImpl = SMESH::DownCast<SMESH_GroupBase_i*>( groups[i]);
3068               if ( grImpl )
3069               {
3070                 CORBA::String_var objStr = GetORB()->object_to_string( grImpl->_this() );
3071                 int anId = myStudyContext->findId( string( objStr.in() ) );
3072                 char grpName[ 30 ];
3073                 sprintf( grpName, "Group %d", anId );
3074                 SMESHDS_GroupBase* aGrpBaseDS = grImpl->GetGroupDS();
3075                 aGrpBaseDS->SetStoreName( grpName );
3076               }
3077             }
3078           }
3079         }
3080       }
3081     }
3082   }
3083
3084   // Write data
3085   // ---> create HDF file
3086   aFile = new HDFfile( (char*) filename.ToCString() );
3087   aFile->CreateOnDisk();
3088
3089   // --> iterator for top-level objects
3090   SALOMEDS::ChildIterator_wrap itBig = myCurrentStudy->NewChildIterator( theComponent );
3091   for ( ; itBig->More(); itBig->Next() ) {
3092     SALOMEDS::SObject_wrap gotBranch = itBig->Value();
3093
3094     // --> hypotheses root branch (only one for the study)
3095     if ( gotBranch->Tag() == GetHypothesisRootTag() ) {
3096       // create hypotheses root HDF group
3097       aTopGroup = new HDFgroup( "Hypotheses", aFile );
3098       aTopGroup->CreateOnDisk();
3099
3100       // iterator for all hypotheses
3101       SALOMEDS::ChildIterator_wrap it = myCurrentStudy->NewChildIterator( gotBranch );
3102       for ( ; it->More(); it->Next() ) {
3103         SALOMEDS::SObject_wrap mySObject = it->Value();
3104         CORBA::Object_var anObject = SObjectToObject( mySObject );
3105         if ( !CORBA::is_nil( anObject ) ) {
3106           SMESH::SMESH_Hypothesis_var myHyp = SMESH::SMESH_Hypothesis::_narrow( anObject );
3107           if ( !myHyp->_is_nil() ) {
3108             SMESH_Hypothesis_i* myImpl = dynamic_cast<SMESH_Hypothesis_i*>( GetServant( myHyp ).in() );
3109             if ( myImpl ) {
3110               string hypname = string( myHyp->GetName() );
3111               string libname = string( myHyp->GetLibName() );
3112               // BUG SWP13062
3113               // Needs for save crossplatform libname, i.e. parth of name ( ".dll" for
3114               // WIN32 and ".so" for X-system) must be deleted
3115               int libname_len = libname.length();
3116 #ifdef WIN32
3117               if( libname_len > 4 )
3118                 libname.resize( libname_len - 4 );
3119 #else
3120               // PAL17753 (Regression: missing hypothesis in restored study)
3121               // "lib" also should be removed from the beginning
3122               //if( libname_len > 3 )
3123                 //libname.resize( libname_len - 3 );
3124               if( libname_len > 6 )
3125                 libname = libname.substr( 3, libname_len - 3&