smesh/src/SMDS/SMDS_MeshGroup.cxx

//  SMESH SMDS : implementaion of Salome mesh data structure
//
//  Copyright (C) 2003  OPEN CASCADE
// 
//  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.opencascade.org or email : webmaster@opencascade.org 
//
//
//
//  File   : SMDS_MeshGroup.cxx
//  Author : Jean-Michel BOULCOURT
//  Module : SMESH

#ifdef _MSC_VER
#pragma warning(disable:4786)
#endif

#include "SMDS_MeshGroup.hxx"
#include "utilities.h"

using namespace std;

//=======================================================================
//function : SMDS_MeshGroup
//purpose  : 
//=======================================================================

SMDS_MeshGroup::SMDS_MeshGroup(const SMDS_Mesh * theMesh,
                               const SMDSAbs_ElementType theType)
	:myMesh(theMesh),myType(theType), myParent(NULL)
{
}

//=======================================================================
//function : SMDS_MeshGroup
//purpose  : 
//=======================================================================

SMDS_MeshGroup::SMDS_MeshGroup(SMDS_MeshGroup * theParent,
                               const SMDSAbs_ElementType theType)
	:myMesh(theParent->myMesh),myType(theType), myParent(theParent)
{
}

//=======================================================================
//function : AddSubGroup
//purpose  : 
//=======================================================================

const SMDS_MeshGroup *SMDS_MeshGroup::AddSubGroup
                (const SMDSAbs_ElementType theType)
{
	const SMDS_MeshGroup * subgroup = new SMDS_MeshGroup(this,theType);
	myChildren.insert(myChildren.end(),subgroup);
	return subgroup;
}

//=======================================================================
//function : RemoveSubGroup
//purpose  : 
//=======================================================================

bool SMDS_MeshGroup::RemoveSubGroup(const SMDS_MeshGroup * theGroup)
{
	bool found = false;	
	list<const SMDS_MeshGroup*>::iterator itgroup;
	for(itgroup=myChildren.begin(); itgroup!=myChildren.end(); itgroup++)
	{
		const SMDS_MeshGroup* subgroup=*itgroup;
		if (subgroup == theGroup)
		{
			found = true;
			myChildren.erase(itgroup);
		}
	}

	return found;
}

//=======================================================================
//function : RemoveFromParent
//purpose  : 
//=======================================================================

bool SMDS_MeshGroup::RemoveFromParent()
{
	
	if (myParent==NULL) return false;
	else
	{
		return (myParent->RemoveSubGroup(this));
	}
}
//=======================================================================
//function : Clear
//purpose  : 
//=======================================================================

void SMDS_MeshGroup::Clear()
{
	myElements.clear();
	myType = SMDSAbs_All;
}

//=======================================================================
//function : Add
//purpose  : 
//=======================================================================

void SMDS_MeshGroup::Add(const SMDS_MeshElement * theElem)
{
	// the type of the group is determined by the first element added
	if (myElements.empty()) myType = theElem->GetType();
	else if (theElem->GetType() != myType)
		MESSAGE("SMDS_MeshGroup::Add : Type Mismatch");
	
	myElements.insert(theElem);
}

//=======================================================================
//function : Remove
//purpose  : 
//=======================================================================

void SMDS_MeshGroup::Remove(const SMDS_MeshElement * theElem)
{
	myElements.erase(theElem);
	if (myElements.empty()) myType = SMDSAbs_All;
}

//=======================================================================
//function : Contains
//purpose  : 
//=======================================================================

bool SMDS_MeshGroup::Contains(const SMDS_MeshElement * theElem) const
{
	return myElements.find(theElem)!=myElements.end();
}

//=======================================================================
//function : SetType
//purpose  : 
//=======================================================================

void SMDS_MeshGroup::SetType(const SMDSAbs_ElementType theType)
{
  if (IsEmpty())
    myType = theType;
}