src/SMESH_I/SMESH_Gen_i.cxx

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