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