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