Salome HOME
Nerge with PAL/SALOME 2.1.0d
[modules/smesh.git] / src / SMESHDS / SMESHDS_GroupOnGeom.cxx
diff --git a/src/SMESHDS/SMESHDS_GroupOnGeom.cxx b/src/SMESHDS/SMESHDS_GroupOnGeom.cxx
new file mode 100644 (file)
index 0000000..187357a
--- /dev/null
@@ -0,0 +1,112 @@
+//  SMESH SMESHDS : idl implementation based on 'SMESH' unit's classes
+//
+//  Copyright (C) 2004  CEA
+// 
+//  This library is free software; you can redistribute it and/or 
+//  modify it under the terms of the GNU Lesser General Public 
+//  License as published by the Free Software Foundation; either 
+//  version 2.1 of the License. 
+// 
+//  This library is distributed in the hope that it will be useful, 
+//  but WITHOUT ANY WARRANTY; without even the implied warranty of 
+//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU 
+//  Lesser General Public License for more details. 
+// 
+//  You should have received a copy of the GNU Lesser General Public 
+//  License along with this library; if not, write to the Free Software 
+//  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA 
+// 
+//  See http://www.salome-platform.org or email : webmaster.salome@opencascade.org 
+//
+//
+//
+//  File   : SMESHDS_GroupOnGeom.cxx
+//  Module : SMESH
+//  $Header$
+
+#include "SMESHDS_GroupOnGeom.hxx"
+#include "SMESHDS_Mesh.hxx"
+#include "utilities.h"
+
+using namespace std;
+
+//=============================================================================
+/*!
+ *  
+ */
+//=============================================================================
+
+SMESHDS_GroupOnGeom::SMESHDS_GroupOnGeom (const int                 theID,
+                                          const SMESHDS_Mesh*       theMesh,
+                                          const SMDSAbs_ElementType theType,
+                                          const TopoDS_Shape&       theShape)
+     : SMESHDS_GroupBase(theID,theMesh,theType), myShape(theShape)
+{
+  SMESHDS_Mesh* aMesh = const_cast<SMESHDS_Mesh*>(theMesh);
+  mySubMesh = aMesh->MeshElements( aMesh->AddCompoundSubmesh( theShape ));
+}
+
+// =====================
+// class MyGroupIterator
+// =====================
+
+class MyIterator: public SMDS_ElemIterator
+{
+  SMDSAbs_ElementType     myType;
+  SMDS_ElemIteratorPtr    myElemIt;
+  SMDS_NodeIteratorPtr    myNodeIt;
+  const SMDS_MeshElement* myElem;
+ public:
+  MyIterator(SMDSAbs_ElementType type, const SMESHDS_SubMesh* subMesh)
+    : myType(type), myElem(0)
+  {
+    if ( subMesh ) 
+      if ( myType == SMDSAbs_Node )
+        myNodeIt = subMesh->GetNodes();
+      else {
+        myElemIt = subMesh->GetElements();
+        next();
+      }
+  }
+  bool more()
+  {
+    if ( myType == SMDSAbs_Node && myNodeIt )
+      return myNodeIt->more();
+    return ( myElem != 0 );
+  }
+  const SMDS_MeshElement* next()
+  {
+    if ( myType == SMDSAbs_Node && myNodeIt )
+      return myNodeIt->next();
+    const SMDS_MeshElement* res = myElem;
+    myElem = 0;
+    while ( myElemIt && myElemIt->more() ) {
+      myElem = myElemIt->next();
+      if ( myElem && myElem->GetType() == myType )
+        break;
+      else
+        myElem = 0;
+    }
+    return res;
+  }
+};
+
+//=======================================================================
+//function : GetElements
+//purpose  : 
+//=======================================================================
+
+SMDS_ElemIteratorPtr SMESHDS_GroupOnGeom::GetElements()
+{
+  return SMDS_ElemIteratorPtr( new MyIterator ( GetType(), mySubMesh ));
+}
+
+//=======================================================================
+//function : Contains
+//purpose  : 
+//=======================================================================
+
+bool SMESHDS_GroupOnGeom::Contains (const int theID)
+{
+  return mySubMesh->Contains( findInMesh( theID ));
+}