smesh/src/SMESHDS/SMESHDS_GroupOnGeom.cxx

113 lines
3.4 KiB
C++

// 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.com
//
//
//
// 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 ));
}