]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMDS/SMDS_ElementHolder.cxx
Salome HOME
Merge branch 'master' into gni/adaptation
[modules/smesh.git] / src / SMDS / SMDS_ElementHolder.cxx
1 // Copyright (C) 2007-2020  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 }