]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESHDS/SMESHDS_GroupOnGeom.cxx
Salome HOME
PAL13330( When mesh generation does not success, trace where )
[modules/smesh.git] / src / SMESHDS / SMESHDS_GroupOnGeom.cxx
1 //  SMESH SMESHDS : idl implementation based on 'SMESH' unit's classes
2 //
3 //  Copyright (C) 2004  CEA
4 // 
5 //  This library is free software; you can redistribute it and/or 
6 //  modify it under the terms of the GNU Lesser General Public 
7 //  License as published by the Free Software Foundation; either 
8 //  version 2.1 of the License. 
9 // 
10 //  This library is distributed in the hope that it will be useful, 
11 //  but WITHOUT ANY WARRANTY; without even the implied warranty of 
12 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU 
13 //  Lesser General Public License for more details. 
14 // 
15 //  You should have received a copy of the GNU Lesser General Public 
16 //  License along with this library; if not, write to the Free Software 
17 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA 
18 // 
19 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 //
21 //
22 //
23 //  File   : SMESHDS_GroupOnGeom.cxx
24 //  Module : SMESH
25 //  $Header$
26
27 #include "SMESHDS_GroupOnGeom.hxx"
28 #include "SMESHDS_Mesh.hxx"
29 #include "utilities.h"
30
31 using namespace std;
32
33 //=============================================================================
34 /*!
35  *  
36  */
37 //=============================================================================
38
39 SMESHDS_GroupOnGeom::SMESHDS_GroupOnGeom (const int                 theID,
40                                           const SMESHDS_Mesh*       theMesh,
41                                           const SMDSAbs_ElementType theType,
42                                           const TopoDS_Shape&       theShape)
43      : SMESHDS_GroupBase(theID,theMesh,theType), myShape(theShape)
44 {
45   SMESHDS_Mesh* aMesh = const_cast<SMESHDS_Mesh*>(theMesh);
46   mySubMesh = aMesh->MeshElements( aMesh->AddCompoundSubmesh( theShape ));
47 }
48
49 // =====================
50 // class MyGroupIterator
51 // =====================
52
53 class MyIterator: public SMDS_ElemIterator
54 {
55   SMDSAbs_ElementType     myType;
56   SMDS_ElemIteratorPtr    myElemIt;
57   SMDS_NodeIteratorPtr    myNodeIt;
58   const SMDS_MeshElement* myElem;
59  public:
60   MyIterator(SMDSAbs_ElementType type, const SMESHDS_SubMesh* subMesh)
61     : myType(type), myElem(0)
62   {
63     if ( subMesh ) 
64       if ( myType == SMDSAbs_Node )
65         myNodeIt = subMesh->GetNodes();
66       else {
67         myElemIt = subMesh->GetElements();
68         next();
69       }
70   }
71   bool more()
72   {
73     if ( myType == SMDSAbs_Node && myNodeIt )
74       return myNodeIt->more();
75     return ( myElem != 0 );
76   }
77   const SMDS_MeshElement* next()
78   {
79     if ( myType == SMDSAbs_Node && myNodeIt )
80       return myNodeIt->next();
81     const SMDS_MeshElement* res = myElem;
82     myElem = 0;
83     while ( myElemIt && myElemIt->more() ) {
84       myElem = myElemIt->next();
85       if ( myElem && myElem->GetType() == myType )
86         break;
87       else
88         myElem = 0;
89     }
90     return res;
91   }
92 };
93
94 //=======================================================================
95 //function : GetElements
96 //purpose  : 
97 //=======================================================================
98
99 SMDS_ElemIteratorPtr SMESHDS_GroupOnGeom::GetElements()
100 {
101   return SMDS_ElemIteratorPtr( new MyIterator ( GetType(), mySubMesh ));
102 }
103
104 //=======================================================================
105 //function : Contains
106 //purpose  : 
107 //=======================================================================
108
109 bool SMESHDS_GroupOnGeom::Contains (const int theID)
110 {
111   return mySubMesh->Contains( findInMesh( theID ));
112 }