src/SMDS/SMDS_ElementHolder.cxx

   1 // Copyright (C) 2007-2019  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License, or (at your option) any later version.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22 //  File   : SMDS_ElementHolder.cxx
  23 //  Module : SMESH
  24 //
  25
  26 #include "SMDS_ElementHolder.hxx"
  27
  28 #include "ObjectPool.hxx"
  29 #include "SMDS_CellOfNodes.hxx"
  30 #include "SMDS_Mesh.hxx"
  31
  32 //=======================================================================
  33 //function : SMDS_ElementHolder
  34 //purpose  : register self in the mesh
  35 //=======================================================================
  36
  37 SMDS_ElementHolder::SMDS_ElementHolder( const SMDS_Mesh* mesh )
  38   : myMesh( const_cast< SMDS_Mesh* >( mesh ))
  39 {
  40   if ( myMesh )
  41     myPtrInMesh = myMesh->myElemHolders.insert( this ).first;
  42 }
  43
  44 //=======================================================================
  45 //function : ~SMDS_ElementHolder
  46 //purpose  : un-register self from the mesh
  47 //=======================================================================
  48
  49 SMDS_ElementHolder::~SMDS_ElementHolder()
  50 {
  51   if ( myMesh )
  52     myMesh->myElemHolders.erase( myPtrInMesh );
  53 }
  54
  55 //=======================================================================
  56 //function : beforeCompacting
  57 //purpose  : store vtkIDs of elements
  58 //=======================================================================
  59
  60 void SMDS_ElementHolder::beforeCompacting()
  61 {
  62   int i = 0;
  63   for ( SMDS_ElemIteratorPtr it = getElements(); it->more(); ++i )
  64   {
  65     const SMDS_MeshElement* e = it->next();
  66     if ( !e ) continue;
  67     if ( e->IsNull() && !dynamic_cast<const SMDS_CellOfNodes*>( e ))
  68       continue; // removed element
  69     myIsNode.push_back( e->GetType() == SMDSAbs_Node );
  70     if ( myMesh->Contains( e ))
  71     {
  72       myVtkIDs.push_back( e->GetVtkID() );
  73     }
  74     else
  75     {
  76       myExternalElems.push_back( e );
  77       myVtkIDs.push_back( -1 * (int)myExternalElems.size() );
  78     }
  79   }
  80 }
  81
  82 //=======================================================================
  83 //function : restoreElements
  84 //purpose  : restore pointers to elements
  85 //=======================================================================
  86
  87 void SMDS_ElementHolder::restoreElements( const std::vector<int>& idNodesOldToNew,
  88                                           const std::vector<int>& idCellsOldToNew )
  89 {
  90   tmpClear();
  91
  92   const SMDS_MeshElement* elem;
  93
  94   std::vector< bool >::iterator isNode = myIsNode.begin();
  95   for ( size_t i = 0; i < myVtkIDs.size(); ++i, ++isNode )
  96   {
  97     int vtkID = myVtkIDs[i];
  98     if ( vtkID < 0 )
  99     {
 100       elem = myExternalElems[ (-vtkID)-1 ];
 101     }
 102     else if ( *isNode )
 103     {
 104       if ( vtkID < (int)idNodesOldToNew.size() )
 105         elem = myMesh->FindNodeVtk( idNodesOldToNew[ vtkID ]);
 106       else
 107         elem = myMesh->FindNodeVtk( vtkID );
 108     }
 109     else
 110     {
 111       if ( vtkID < (int)idCellsOldToNew.size() )
 112         elem = myMesh->FindElementVtk( idCellsOldToNew[ vtkID ]);
 113       else
 114         elem = myMesh->FindElementVtk( vtkID );
 115     }
 116     if ( elem )
 117       add( elem );
 118   }
 119   clearVector( myExternalElems );
 120   clearVector( myVtkIDs );
 121   clearVector( myIsNode );
 122
 123   compact();
 124 }