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