src/SMDS/SMDS_ElementHolder.cxx

   1 // Copyright (C) 2007-2016  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   myPtrInMesh = myMesh->myElemHolders.insert( this ).first;
  41 }
  42
  43 //=======================================================================
  44 //function : ~SMDS_ElementHolder
  45 //purpose  : un-register self from the mesh
  46 //=======================================================================
  47
  48 SMDS_ElementHolder::~SMDS_ElementHolder()
  49 {
  50   myMesh->myElemHolders.erase( myPtrInMesh );
  51 }
  52
  53 //=======================================================================
  54 //function : beforeCompacting
  55 //purpose  : store vtkIDs of elements
  56 //=======================================================================
  57
  58 void SMDS_ElementHolder::beforeCompacting()
  59 {
  60   int i = 0;
  61   for ( SMDS_ElemIteratorPtr it = getElements(); it->more(); ++i )
  62   {
  63     const SMDS_MeshElement* e = it->next();
  64     if ( !e ) continue;
  65     if ( e->IsNull() && !dynamic_cast<const SMDS_CellOfNodes*>( e ))
  66       continue; // removed element
  67     myIsNode.push_back( e->GetType() == SMDSAbs_Node );
  68     if ( myMesh->Contains( e ))
  69     {
  70       myVtkIDs.push_back( e->GetVtkID() );
  71     }
  72     else
  73     {
  74       myExternalElems.push_back( e );
  75       myVtkIDs.push_back( -1 * (int)myExternalElems.size() );
  76     }
  77   }
  78 }
  79
  80 //=======================================================================
  81 //function : restoreElements
  82 //purpose  : restore pointers to elements
  83 //=======================================================================
  84
  85 void SMDS_ElementHolder::restoreElements( const std::vector<int>& idNodesOldToNew,
  86                                           const std::vector<int>& idCellsOldToNew )
  87 {
  88   tmpClear();
  89
  90   const SMDS_MeshElement* elem;
  91
  92   std::vector< bool >::iterator isNode = myIsNode.begin();
  93   for ( size_t i = 0; i < myVtkIDs.size(); ++i, ++isNode )
  94   {
  95     int vtkID = myVtkIDs[i];
  96     if ( vtkID < 0 )
  97     {
  98       elem = myExternalElems[ (-vtkID)-1 ];
  99     }
 100     else if ( *isNode )
 101     {
 102       if ( vtkID < (int)idNodesOldToNew.size() )
 103         elem = myMesh->FindNodeVtk( idNodesOldToNew[ vtkID ]);
 104       else
 105         elem = myMesh->FindNodeVtk( vtkID );
 106     }
 107     else
 108     {
 109       if ( vtkID < (int)idCellsOldToNew.size() )
 110         elem = myMesh->FindElementVtk( idCellsOldToNew[ vtkID ]);
 111       else
 112         elem = myMesh->FindElementVtk( vtkID );
 113     }
 114     if ( elem )
 115       add( elem );
 116   }
 117   clearVector( myExternalElems );
 118   clearVector( myVtkIDs );
 119   clearVector( myIsNode );
 120
 121   compact();
 122 }