src/SMDS/SMDS_ElementHolder.cxx

   1 // Copyright (C) 2007-2022  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   for ( SMDS_ElemIteratorPtr it = getElements(); it->more(); )
  63   {
  64     const SMDS_MeshElement* e = it->next();
  65     if ( !e ) continue;
  66     if ( e->IsNull() && !dynamic_cast<const SMDS_CellOfNodes*>( e ))
  67       continue; // removed element
  68     myIsNode.push_back( e->GetType() == SMDSAbs_Node );
  69     if ( myMesh->Contains( e ))
  70     {
  71       myVtkIDs.push_back( e->GetVtkID() );
  72     }
  73     else
  74     {
  75       myExternalElems.push_back( e );
  76       myVtkIDs.push_back( -1 * (vtkIdType) myExternalElems.size() );
  77     }
  78   }
  79 }
  80
  81 //=======================================================================
  82 //function : restoreElements
  83 //purpose  : restore pointers to elements
  84 //=======================================================================
  85
  86 void SMDS_ElementHolder::restoreElements( const std::vector<smIdType>& idNodesOldToNew,
  87                                           const std::vector<smIdType>& idCellsOldToNew )
  88 {
  89   tmpClear();
  90
  91   const SMDS_MeshElement* elem;
  92
  93   std::vector< bool >::iterator isNode = myIsNode.begin();
  94   for ( size_t i = 0; i < myVtkIDs.size(); ++i, ++isNode )
  95   {
  96     vtkIdType vtkID = myVtkIDs[i];
  97     if ( vtkID < 0 )
  98     {
  99       elem = myExternalElems[ (-vtkID)-1 ];
 100     }
 101     else if ( *isNode )
 102     {
 103       if ( vtkID < (vtkIdType)idNodesOldToNew.size() )
 104         elem = myMesh->FindNodeVtk( idNodesOldToNew[ vtkID ]);
 105       else
 106         elem = myMesh->FindNodeVtk( vtkID );
 107     }
 108     else
 109     {
 110       if ( vtkID < (vtkIdType)idCellsOldToNew.size() )
 111         elem = myMesh->FindElementVtk( idCellsOldToNew[ vtkID ]);
 112       else
 113         elem = myMesh->FindElementVtk( vtkID );
 114     }
 115     if ( elem )
 116       add( elem );
 117   }
 118   clearVector( myExternalElems );
 119   clearVector( myVtkIDs );
 120   clearVector( myIsNode );
 121
 122   compact();
 123 }