src/SMESH_I/SMESH_subMesh_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
  23 //  SMESH SMESH_I : idl implementation based on 'SMESH' unit's calsses
  24 //  File   : SMESH_subMesh_i.cxx
  25 //  Author : Paul RASCLE, EDF
  26 //  Module : SMESH
  27 //
  28 #include "SMESH_subMesh_i.hxx"
  29 #include "SMESH_Gen_i.hxx"
  30 #include "SMESH_Mesh_i.hxx"
  31 #include "SMESH_PreMeshInfo.hxx"
  32
  33 #include "Utils_CorbaException.hxx"
  34 #include "utilities.h"
  35 #include "OpUtil.hxx"
  36 #include "Utils_ExceptHandlers.hxx"
  37
  38 #include <TopoDS_Iterator.hxx>
  39 #include <TopExp_Explorer.hxx>
  40
  41 using namespace std;
  42
  43 //=============================================================================
  44 /*!
  45  *
  46  */
  47 //=============================================================================
  48
  49 SMESH_subMesh_i::SMESH_subMesh_i()
  50      : SALOME::GenericObj_i( PortableServer::POA::_nil() )
  51 {
  52   MESSAGE("SMESH_subMesh_i::SMESH_subMesh_i default, not for use");
  53   ASSERT(0);
  54 }
  55
  56 //=============================================================================
  57 /*!
  58  *
  59  */
  60 //=============================================================================
  61
  62 SMESH_subMesh_i::SMESH_subMesh_i( PortableServer::POA_ptr thePOA,
  63                                   SMESH_Gen_i*            gen_i,
  64                                   SMESH_Mesh_i*           mesh_i,
  65                                   int                     localId )
  66      : SALOME::GenericObj_i( thePOA )
  67 {
  68   _gen_i = gen_i;
  69   _mesh_i = mesh_i;
  70   _localId = localId;
  71   _preMeshInfo = NULL;
  72 }
  73 //=============================================================================
  74 /*!
  75  *
  76  */
  77 //=============================================================================
  78
  79 SMESH_subMesh_i::~SMESH_subMesh_i()
  80 {
  81   MESSAGE("SMESH_subMesh_i::~SMESH_subMesh_i");
  82   if ( _preMeshInfo ) delete _preMeshInfo;
  83   _preMeshInfo = NULL;
  84 }
  85
  86 //=======================================================================
  87 //function : getSubMeshes
  88 //purpose  : for a submesh on shape to which elements are not bound directly,
  89 //           return submeshes containing elements
  90 //=======================================================================
  91
  92 typedef list<SMESHDS_SubMesh*> TListOfSubMeshes;
  93
  94 bool getSubMeshes(::SMESH_subMesh*  theSubMesh,
  95                   TListOfSubMeshes& theSubMeshList)
  96 {
  97   int size = theSubMeshList.size();
  98
  99   SMESH_Mesh*      aMesh      = theSubMesh->GetFather();
 100   SMESHDS_Mesh*    aMeshDS    = aMesh->GetMeshDS();
 101   SMESHDS_SubMesh* aSubMeshDS = theSubMesh->GetSubMeshDS();
 102
 103   // nodes can be bound to either vertex, edge, face or solid_or_shell
 104   TopoDS_Shape aShape = theSubMesh->GetSubShape();
 105   switch ( aShape.ShapeType() )
 106   {
 107   case TopAbs_SOLID: {
 108     // add submesh of solid itself
 109     aSubMeshDS = aMeshDS->MeshElements( aShape );
 110     if ( aSubMeshDS )
 111       theSubMeshList.push_back( aSubMeshDS );
 112     // and of the first shell
 113     TopExp_Explorer exp( aShape, TopAbs_SHELL );
 114     if ( exp.More() ) {
 115       aSubMeshDS = aMeshDS->MeshElements( exp.Current() );
 116       if ( aSubMeshDS )
 117         theSubMeshList.push_back( aSubMeshDS );
 118     }
 119     break;
 120   }
 121   case TopAbs_WIRE:
 122   case TopAbs_COMPOUND:
 123   case TopAbs_COMPSOLID: {
 124     // call getSubMeshes() for sub-shapes
 125     list<TopoDS_Shape> shapeList;
 126     shapeList.push_back( aShape );
 127     list<TopoDS_Shape>::iterator sh = shapeList.begin();
 128     for ( ; sh != shapeList.end(); ++sh ) {
 129       for ( TopoDS_Iterator it( *sh ); it.More(); it.Next() ) {
 130         if ( ::SMESH_subMesh* aSubMesh = aMesh->GetSubMeshContaining( it.Value() ))
 131           getSubMeshes( aSubMesh, theSubMeshList ); // add found submesh or explore deeper
 132         else
 133           // no submesh for a compound inside compound
 134           shapeList.push_back( it.Value() );
 135       }
 136     }
 137     // return only unique submeshes
 138     set<SMESHDS_SubMesh*> smSet;
 139     TListOfSubMeshes::iterator sm = theSubMeshList.begin();
 140     while ( sm != theSubMeshList.end() ) {
 141       if ( !smSet.insert( *sm ).second )
 142         sm = theSubMeshList.erase( sm );
 143       else
 144         ++sm;
 145     }
 146     break;
 147   }
 148   default:
 149     if ( aSubMeshDS )
 150       theSubMeshList.push_back( aSubMeshDS );
 151   }
 152   return size < theSubMeshList.size();
 153 }
 154
 155 //=============================================================================
 156 /*!
 157  *
 158  */
 159 //=============================================================================
 160
 161 CORBA::Long SMESH_subMesh_i::GetNumberOfElements()
 162   throw (SALOME::SALOME_Exception)
 163 {
 164   Unexpect aCatch(SALOME_SalomeException);
 165
 166   if ( _preMeshInfo )
 167     return _preMeshInfo->NbElements();
 168
 169   if ( _mesh_i->_mapSubMesh.find( _localId ) == _mesh_i->_mapSubMesh.end() )
 170     return 0;
 171
 172   ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
 173   SMESHDS_SubMesh* aSubMeshDS = aSubMesh->GetSubMeshDS();
 174
 175   int nbElems = aSubMeshDS ? aSubMeshDS->NbElements() : 0;
 176
 177   // volumes are bound to shell
 178   TListOfSubMeshes smList;
 179   if ( nbElems == 0 && getSubMeshes( aSubMesh, smList ))
 180   {
 181     TListOfSubMeshes::iterator sm = smList.begin();
 182     for ( ; sm != smList.end(); ++sm )
 183       nbElems += (*sm)->NbElements();
 184   }
 185   return nbElems;
 186 }
 187
 188 //=============================================================================
 189 /*!
 190  *
 191  */
 192 //=============================================================================
 193
 194 CORBA::Long SMESH_subMesh_i::GetNumberOfNodes(CORBA::Boolean all)
 195   throw (SALOME::SALOME_Exception)
 196 {
 197   Unexpect aCatch(SALOME_SalomeException);
 198
 199   if ( _mesh_i->_mapSubMesh.find( _localId ) == _mesh_i->_mapSubMesh.end() )
 200     return 0;
 201
 202   if ( _preMeshInfo )
 203   {
 204     if ( all ) return _preMeshInfo->NbNodes();
 205     else _preMeshInfo->FullLoadFromFile();
 206   }
 207   ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
 208   SMESHDS_SubMesh* aSubMeshDS = aSubMesh->GetSubMeshDS();
 209
 210   if ( aSubMeshDS && aSubMeshDS->IsComplexSubmesh() )
 211   {
 212     // sub-mesh on a geom group, always return all nodes
 213     return aSubMeshDS->NbNodes();
 214   }
 215   if ( aSubMeshDS && !all )
 216   {
 217     // return anything we have
 218     return aSubMeshDS->NbNodes();
 219   }
 220   if ( all ) // get nodes from aSubMesh and all child sub-meshes
 221   {
 222     int nbNodes = 0;
 223     SMESH_subMeshIteratorPtr smIt = aSubMesh->getDependsOnIterator( /*includeSelf=*/true );
 224     while ( smIt->more() )
 225     {
 226       aSubMesh = smIt->next();
 227       if (( aSubMeshDS = aSubMesh->GetSubMeshDS() ))
 228         nbNodes += aSubMeshDS->NbNodes();
 229     }
 230     return nbNodes;
 231   }
 232
 233   return aSubMeshDS ? aSubMeshDS->NbNodes() : 0;
 234 }
 235
 236 //=============================================================================
 237 /*!
 238  *
 239  */
 240 //=============================================================================
 241
 242 SMESH::long_array* SMESH_subMesh_i::GetElementsId()
 243   throw (SALOME::SALOME_Exception)
 244 {
 245   Unexpect aCatch(SALOME_SalomeException);
 246
 247   SMESH::long_array_var aResult = new SMESH::long_array();
 248
 249   if ( _mesh_i->_mapSubMesh.find( _localId ) == _mesh_i->_mapSubMesh.end() )
 250     return aResult._retn();
 251
 252   if ( _preMeshInfo )
 253     _preMeshInfo->FullLoadFromFile();
 254
 255   ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
 256   SMESHDS_SubMesh* aSubMeshDS = aSubMesh->GetSubMeshDS();
 257
 258   int nbElems = aSubMeshDS ? aSubMeshDS->NbElements() : 0;
 259   TListOfSubMeshes smList;
 260   if ( nbElems )
 261     smList.push_back( aSubMeshDS );
 262
 263   // volumes are bound to shell
 264   if ( nbElems == 0 && getSubMeshes( aSubMesh, smList ))
 265   {
 266     TListOfSubMeshes::iterator sm = smList.begin();
 267     for ( ; sm != smList.end(); ++sm )
 268       nbElems += (*sm)->NbElements();
 269   }
 270
 271   aResult->length( nbElems );
 272   if ( nbElems )
 273   {
 274     TListOfSubMeshes::iterator sm = smList.begin();
 275     for ( int i = 0; sm != smList.end(); sm++ )
 276     {
 277       SMDS_ElemIteratorPtr anIt = (*sm)->GetElements();
 278       for ( ; i < nbElems && anIt->more(); i++ )
 279         aResult[i] = anIt->next()->GetID();
 280     }
 281   }
 282   return aResult._retn();
 283 }
 284
 285
 286 //=============================================================================
 287 /*!
 288  *
 289  */
 290 //=============================================================================
 291
 292 SMESH::long_array* SMESH_subMesh_i::GetElementsByType( SMESH::ElementType theElemType )
 293     throw (SALOME::SALOME_Exception)
 294 {
 295   Unexpect aCatch(SALOME_SalomeException);
 296
 297   SMESH::long_array_var aResult = new SMESH::long_array();
 298
 299   if ( _mesh_i->_mapSubMesh.find( _localId ) == _mesh_i->_mapSubMesh.end() )
 300     return aResult._retn();
 301
 302   if ( _preMeshInfo )
 303     _preMeshInfo->FullLoadFromFile();
 304
 305   ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
 306   SMESHDS_SubMesh* aSubMeshDS = aSubMesh->GetSubMeshDS();
 307
 308   // PAL5440, return all nodes belonging to elements of submesh
 309   set<int> nodeIds;
 310   int nbElems = aSubMeshDS ? aSubMeshDS->NbElements() : 0;
 311
 312   // volumes may be bound to shell instead of solid
 313   TListOfSubMeshes smList;
 314   if ( nbElems == 0 && getSubMeshes( aSubMesh, smList ))
 315   {
 316     TListOfSubMeshes::iterator sm = smList.begin();
 317     for ( ; sm != smList.end(); ++sm )
 318     {
 319       if ( theElemType == SMESH::NODE )
 320       {
 321         SMDS_ElemIteratorPtr eIt = (*sm)->GetElements();
 322         if ( eIt->more() ) {
 323           while ( eIt->more() ) {
 324             const SMDS_MeshElement* anElem = eIt->next();
 325             SMDS_ElemIteratorPtr nIt = anElem->nodesIterator();
 326             while ( nIt->more() )
 327               nodeIds.insert( nIt->next()->GetID() );
 328           }
 329         } else {
 330           SMDS_NodeIteratorPtr nIt = (*sm)->GetNodes();
 331           while ( nIt->more() )
 332             nodeIds.insert( nIt->next()->GetID() );
 333         }
 334       }
 335       else
 336       {
 337         nbElems += (*sm)->NbElements();
 338       }
 339     }
 340     aSubMeshDS = 0;
 341   }
 342   else
 343   {
 344     if ( nbElems )
 345       smList.push_back( aSubMeshDS );
 346   }
 347
 348   if ( theElemType == SMESH::NODE && aSubMeshDS )
 349   {
 350     SMDS_ElemIteratorPtr eIt = aSubMeshDS->GetElements();
 351     if ( eIt->more() ) {
 352       while ( eIt->more() ) {
 353         const SMDS_MeshElement* anElem = eIt->next();
 354         SMDS_ElemIteratorPtr nIt = anElem->nodesIterator();
 355         while ( nIt->more() )
 356           nodeIds.insert( nIt->next()->GetID() );
 357       }
 358     } else {
 359       SMDS_NodeIteratorPtr nIt = aSubMeshDS->GetNodes();
 360       while ( nIt->more() )
 361         nodeIds.insert( nIt->next()->GetID() );
 362     }
 363   }
 364
 365   if ( theElemType == SMESH::NODE )
 366     aResult->length( nodeIds.size() );
 367   else
 368     aResult->length( nbElems );
 369
 370   int i = 0, n = aResult->length();
 371
 372   if ( theElemType == SMESH::NODE && !nodeIds.empty() ) {
 373     set<int>::iterator idIt = nodeIds.begin();
 374     for ( ; i < n && idIt != nodeIds.end() ; i++, idIt++ )
 375       aResult[i] = *idIt;
 376   }
 377
 378   if ( theElemType != SMESH::NODE ) {
 379     TListOfSubMeshes::iterator sm = smList.begin();
 380     for ( i = 0; sm != smList.end(); sm++ )
 381     {
 382       aSubMeshDS = *sm;
 383       SMDS_ElemIteratorPtr anIt = aSubMeshDS->GetElements();
 384       while ( i < n && anIt->more() ) {
 385         const SMDS_MeshElement* anElem = anIt->next();
 386         if ( theElemType == SMESH::ALL || anElem->GetType() == (SMDSAbs_ElementType)theElemType )
 387           aResult[i++] = anElem->GetID();
 388       }
 389     }
 390   }
 391
 392   aResult->length( i );
 393
 394   return aResult._retn();
 395 }
 396
 397 //=============================================================================
 398 /*!
 399  *
 400  */
 401 //=============================================================================
 402
 403 SMESH::long_array* SMESH_subMesh_i::GetNodesId()
 404   throw (SALOME::SALOME_Exception)
 405 {
 406   Unexpect aCatch(SALOME_SalomeException);
 407
 408   SMESH::long_array_var aResult = GetElementsByType( SMESH::NODE );
 409   return aResult._retn();
 410 }
 411
 412 //=============================================================================
 413 /*!
 414  *
 415  */
 416 //=============================================================================
 417
 418 SMESH::SMESH_Mesh_ptr SMESH_subMesh_i::GetFather()
 419   throw (SALOME::SALOME_Exception)
 420 {
 421   Unexpect aCatch(SALOME_SalomeException);
 422   return _mesh_i->_this();
 423 }
 424
 425 //=============================================================================
 426 /*!
 427  *
 428  */
 429 //=============================================================================
 430
 431 CORBA::Long SMESH_subMesh_i::GetId()
 432 {
 433   return _localId;
 434 }
 435
 436 //=======================================================================
 437 //function : GetSubShape
 438 //purpose  :
 439 //=======================================================================
 440
 441 GEOM::GEOM_Object_ptr SMESH_subMesh_i::GetSubShape()
 442      throw (SALOME::SALOME_Exception)
 443 {
 444   Unexpect aCatch(SALOME_SalomeException);
 445   GEOM::GEOM_Object_var aShapeObj;
 446   try {
 447     if ( _mesh_i->_mapSubMesh.find( _localId ) != _mesh_i->_mapSubMesh.end()) {
 448       TopoDS_Shape S = _mesh_i->_mapSubMesh[ _localId ]->GetSubShape();
 449       if ( !S.IsNull() ) {
 450         aShapeObj = _gen_i->ShapeToGeomObject( S );
 451         //mzn: N7PAL16232, N7PAL16233
 452         //In some cases it's possible that GEOM_Client contains the shape same to S, but
 453         //with another orientation.
 454         if (aShapeObj->_is_nil())
 455           aShapeObj = _gen_i->ShapeToGeomObject( S.Reversed() );
 456       }
 457     }
 458   }
 459   catch(SALOME_Exception & S_ex) {
 460     THROW_SALOME_CORBA_EXCEPTION(S_ex.what(), SALOME::BAD_PARAM);
 461   }
 462   return aShapeObj._retn();
 463 }
 464
 465 //=============================================================================
 466 /*!
 467  *
 468  */
 469 //=============================================================================
 470 SMESH::long_array* SMESH_subMesh_i::GetIDs()
 471 {
 472   return GetElementsId();
 473 }
 474
 475 //=============================================================================
 476 /*!
 477  *
 478  */
 479 //=============================================================================
 480 SMESH::ElementType SMESH_subMesh_i::GetElementType( const CORBA::Long id, const bool iselem )
 481   throw (SALOME::SALOME_Exception)
 482 {
 483   if ( _preMeshInfo )
 484     _preMeshInfo->FullLoadFromFile();
 485   return GetFather()->GetElementType( id, iselem );
 486 }
 487
 488 //=============================================================================
 489 /*
 490  * Returns number of mesh elements of each \a EntityType
 491  * @return array of number of elements per \a EntityType
 492  */
 493 //=============================================================================
 494
 495 SMESH::long_array* SMESH_subMesh_i::GetMeshInfo()
 496 {
 497   if ( _preMeshInfo )
 498     return _preMeshInfo->GetMeshInfo();
 499
 500   SMESH::long_array_var aRes = new SMESH::long_array();
 501   aRes->length(SMESH::Entity_Last);
 502   for (int i = SMESH::Entity_Node; i < SMESH::Entity_Last; i++)
 503     aRes[i] = 0;
 504
 505   // get number of nodes
 506   aRes[ SMESH::Entity_Node ] = GetNumberOfNodes(true);
 507
 508   ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
 509
 510   // get statistic from child sub-meshes
 511   TListOfSubMeshes smList;
 512   if ( getSubMeshes( aSubMesh, smList ) )
 513     for ( TListOfSubMeshes::iterator sm = smList.begin(); sm != smList.end(); ++sm )
 514       SMESH_Mesh_i::CollectMeshInfo( (*sm)->GetElements(), aRes );
 515
 516   return aRes._retn();
 517 }
 518
 519 //=======================================================================
 520 /*
 521  * Returns number of mesh elements of each \a ElementType
 522  */
 523 //=======================================================================
 524
 525 SMESH::long_array* SMESH_subMesh_i::GetNbElementsByType()
 526 {
 527   SMESH::long_array_var aRes = new SMESH::long_array();
 528   aRes->length(SMESH::NB_ELEMENT_TYPES);
 529   for (int i = 0; i < SMESH::NB_ELEMENT_TYPES; i++)
 530     if ( _preMeshInfo )
 531       aRes[ i ] = _preMeshInfo->NbElements( SMDSAbs_ElementType( i ));
 532     else
 533       aRes[ i ] = 0;
 534
 535   if ( !_preMeshInfo )
 536   {
 537     aRes[ SMESH::NODE ] = GetNumberOfNodes(true);
 538
 539     ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
 540     if ( SMESHDS_SubMesh* smDS = aSubMesh->GetSubMeshDS() )
 541     {
 542       SMDS_ElemIteratorPtr eIt = smDS->GetElements();
 543       if ( eIt->more() )
 544         aRes[ eIt->next()->GetType() ] = smDS->NbElements();
 545     }
 546   }
 547   return aRes._retn();
 548 }
 549
 550
 551 //=======================================================================
 552 //function : GetTypes
 553 //purpose  : Returns types of elements it contains
 554 //=======================================================================
 555
 556 SMESH::array_of_ElementType* SMESH_subMesh_i::GetTypes()
 557 {
 558   if ( _preMeshInfo )
 559     return _preMeshInfo->GetTypes();
 560
 561   SMESH::array_of_ElementType_var types = new SMESH::array_of_ElementType;
 562
 563   ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
 564   if ( SMESHDS_SubMesh* smDS = aSubMesh->GetSubMeshDS() )
 565   {
 566     SMDS_ElemIteratorPtr eIt = smDS->GetElements();
 567     if ( eIt->more() )
 568     {
 569       types->length( 1 );
 570       types[0] = SMESH::ElementType( eIt->next()->GetType());
 571     }
 572     else if ( smDS->GetNodes()->more() )
 573     {
 574       TopoDS_Shape shape = aSubMesh->GetSubShape();
 575       while ( !shape.IsNull() && shape.ShapeType() == TopAbs_COMPOUND )
 576       {
 577         TopoDS_Iterator it( shape );
 578         shape = it.More() ? it.Value() : TopoDS_Shape();
 579       }
 580       if ( !shape.IsNull() && shape.ShapeType() == TopAbs_VERTEX )
 581       {
 582         types->length( 1 );
 583         types[0] = SMESH::NODE;
 584       }
 585     }
 586   }
 587   return types._retn();
 588 }
 589
 590 //=======================================================================
 591 //function : GetMesh
 592 //purpose  : interface SMESH_IDSource
 593 //=======================================================================
 594
 595 SMESH::SMESH_Mesh_ptr SMESH_subMesh_i::GetMesh()
 596 {
 597   return GetFather();
 598 }
 599
 600 //=======================================================================
 601 //function : IsMeshInfoCorrect
 602 //purpose  : * Returns false if GetMeshInfo() returns incorrect information that may
 603 //           * happen if mesh data is not yet fully loaded from the file of study.
 604 //=======================================================================
 605
 606 bool SMESH_subMesh_i::IsMeshInfoCorrect()
 607 {
 608   return _preMeshInfo ? _preMeshInfo->IsMeshInfoCorrect() : true;
 609 }