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