smesh/src/SMESH_I/SMESH_Group_i.cxx

437 lines
12 KiB
C++
Raw Normal View History

2004-06-18 14:34:31 +06:00
// SMESH SMESH_I : 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 : SMESH_Group_i.cxx
// Author : Sergey ANIKIN, OCC
// Module : SMESH
// $Header$
2004-12-01 15:48:31 +05:00
2004-06-18 14:34:31 +06:00
#include "SMESH_Group_i.hxx"
#include "SMESH_Mesh_i.hxx"
#include "SMESH_Gen_i.hxx"
2004-12-01 15:48:31 +05:00
#include "SMESH_Group.hxx"
#include "SMESHDS_Group.hxx"
#include "SMESHDS_GroupOnGeom.hxx"
#include "SMDSAbs_ElementType.hxx"
#include "SMESH_Filter_i.hxx"
#include "SMESH_PythonDump.hxx"
2004-12-01 15:48:31 +05:00
#include "utilities.h"
2004-06-18 14:34:31 +06:00
using namespace SMESH;
2004-06-18 14:34:31 +06:00
//=============================================================================
/*!
*
*/
//=============================================================================
2004-12-01 15:48:31 +05:00
SMESH_GroupBase_i::SMESH_GroupBase_i( PortableServer::POA_ptr thePOA, SMESH_Mesh_i* theMeshServant, const int theLocalID )
2004-06-18 14:34:31 +06:00
: SALOME::GenericObj_i( thePOA ),
myMeshServant( theMeshServant ),
myLocalID( theLocalID )
{
2005-02-02 16:09:27 +05:00
// PAL7962: san -- To ensure correct mapping of servant and correct reference counting in GenericObj_i,
// servant activation is performed by SMESH_Mesh_i::createGroup()
// thePOA->activate_object( this );
2004-06-18 14:34:31 +06:00
}
2004-12-01 15:48:31 +05:00
SMESH_Group_i::SMESH_Group_i( PortableServer::POA_ptr thePOA, SMESH_Mesh_i* theMeshServant, const int theLocalID )
2005-02-02 16:09:27 +05:00
: SALOME::GenericObj_i( thePOA ),
SMESH_GroupBase_i( thePOA, theMeshServant, theLocalID )
2004-12-01 15:48:31 +05:00
{
2005-02-02 16:09:27 +05:00
MESSAGE("SMESH_Group_i; this = "<<this );
2004-12-01 15:48:31 +05:00
}
SMESH_GroupOnGeom_i::SMESH_GroupOnGeom_i( PortableServer::POA_ptr thePOA, SMESH_Mesh_i* theMeshServant, const int theLocalID )
2005-02-02 16:09:27 +05:00
: SALOME::GenericObj_i( thePOA ),
SMESH_GroupBase_i( thePOA, theMeshServant, theLocalID )
2004-12-01 15:48:31 +05:00
{
2005-02-02 16:09:27 +05:00
MESSAGE("SMESH_GroupOnGeom_i; this = "<<this );
2004-12-01 15:48:31 +05:00
}
2004-06-18 14:34:31 +06:00
//=============================================================================
/*!
*
*/
//=============================================================================
2004-12-01 15:48:31 +05:00
SMESH_GroupBase_i::~SMESH_GroupBase_i()
2004-06-18 14:34:31 +06:00
{
2005-02-02 16:09:27 +05:00
MESSAGE("~SMESH_GroupBase_i; this = "<<this );
2004-06-18 14:34:31 +06:00
if ( myMeshServant )
myMeshServant->removeGroup(myLocalID);
}
2004-12-01 15:48:31 +05:00
//=======================================================================
//function : GetSmeshGroup
//purpose :
//=======================================================================
::SMESH_Group* SMESH_GroupBase_i::GetSmeshGroup() const
{
if ( myMeshServant ) {
::SMESH_Mesh& aMesh = myMeshServant->GetImpl();
return aMesh.GetGroup(myLocalID);
}
return 0;
}
//=======================================================================
//function : GetGroupDS
//purpose :
//=======================================================================
SMESHDS_GroupBase* SMESH_GroupBase_i::GetGroupDS() const
{
::SMESH_Group* aGroup = GetSmeshGroup();
if ( aGroup )
return aGroup->GetGroupDS();
return 0;
}
2004-06-18 14:34:31 +06:00
//=============================================================================
/*!
*
*/
//=============================================================================
2004-12-01 15:48:31 +05:00
void SMESH_GroupBase_i::SetName( const char* theName )
2004-06-18 14:34:31 +06:00
{
// Update Python script
TCollection_AsciiString aStr, aStrName ((char*)theName);
SMESH_Gen_i::AddObject(aStr, _this()) += ".SetName(\"";
aStr += aStrName + "\")";
SMESH_Gen_i::AddToCurrentPyScript(aStr);
// Perform renaming
2004-12-01 15:48:31 +05:00
::SMESH_Group* aGroup = GetSmeshGroup();
if (aGroup) {
aGroup->SetName(theName);
// Update group name in a study
SMESH_Gen_i* aGen = myMeshServant->GetGen();
aGen->SetName( aGen->ObjectToSObject( aGen->GetCurrentStudy(), _this() ), theName );
return;
2004-06-18 14:34:31 +06:00
}
MESSAGE("can't set name of a vague group");
}
//=============================================================================
/*!
*
*/
//=============================================================================
2004-12-01 15:48:31 +05:00
char* SMESH_GroupBase_i::GetName()
2004-06-18 14:34:31 +06:00
{
2004-12-01 15:48:31 +05:00
::SMESH_Group* aGroup = GetSmeshGroup();
if (aGroup)
return CORBA::string_dup (aGroup->GetName());
2004-06-18 14:34:31 +06:00
MESSAGE("get name of a vague group");
return CORBA::string_dup( "NO_NAME" );
}
//=============================================================================
/*!
*
*/
//=============================================================================
2004-12-01 15:48:31 +05:00
SMESH::ElementType SMESH_GroupBase_i::GetType()
2004-06-18 14:34:31 +06:00
{
2004-12-01 15:48:31 +05:00
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
if (aGroupDS) {
SMDSAbs_ElementType aSMDSType = aGroupDS->GetType();
SMESH::ElementType aType;
switch (aSMDSType) {
case SMDSAbs_Node: aType = SMESH::NODE; break;
case SMDSAbs_Edge: aType = SMESH::EDGE; break;
case SMDSAbs_Face: aType = SMESH::FACE; break;
case SMDSAbs_Volume: aType = SMESH::VOLUME; break;
default: aType = SMESH::ALL; break;
2004-06-18 14:34:31 +06:00
}
2004-12-01 15:48:31 +05:00
return aType;
2004-06-18 14:34:31 +06:00
}
MESSAGE("get type of a vague group");
return SMESH::ALL;
}
//=============================================================================
/*!
*
*/
//=============================================================================
2004-12-01 15:48:31 +05:00
CORBA::Long SMESH_GroupBase_i::Size()
2004-06-18 14:34:31 +06:00
{
2004-12-01 15:48:31 +05:00
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
if (aGroupDS)
return aGroupDS->Extent();
2004-06-18 14:34:31 +06:00
MESSAGE("get size of a vague group");
return 0;
}
//=============================================================================
/*!
*
*/
//=============================================================================
2004-12-01 15:48:31 +05:00
CORBA::Boolean SMESH_GroupBase_i::IsEmpty()
2004-06-18 14:34:31 +06:00
{
2004-12-01 15:48:31 +05:00
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
if (aGroupDS)
return aGroupDS->IsEmpty();
2004-06-18 14:34:31 +06:00
MESSAGE("checking IsEmpty of a vague group");
return true;
}
//=============================================================================
/*!
*
*/
//=============================================================================
void SMESH_Group_i::Clear()
{
// Update Python script
TCollection_AsciiString aStr;
SMESH_Gen_i::AddObject(aStr, _this()) += ".Clear()";
SMESH_Gen_i::AddToCurrentPyScript(aStr);
// Clear the group
2004-12-01 15:48:31 +05:00
SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
if (aGroupDS) {
aGroupDS->Clear();
return;
2004-06-18 14:34:31 +06:00
}
MESSAGE("attempt to clear a vague group");
}
//=============================================================================
/*!
*
*/
//=============================================================================
2004-12-01 15:48:31 +05:00
CORBA::Boolean SMESH_GroupBase_i::Contains( CORBA::Long theID )
2004-06-18 14:34:31 +06:00
{
2004-12-01 15:48:31 +05:00
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
if (aGroupDS)
return aGroupDS->Contains(theID);
2004-06-18 14:34:31 +06:00
MESSAGE("attempt to check contents of a vague group");
return false;
}
//=============================================================================
/*!
*
*/
//=============================================================================
CORBA::Long SMESH_Group_i::Add( const SMESH::long_array& theIDs )
{
// Update Python script
TCollection_AsciiString aStr ("nbAdd = ");
SMESH_Gen_i::AddObject(aStr, _this()) += ".Add(";
SMESH_Gen_i::AddArray(aStr, theIDs) += ")";
SMESH_Gen_i::AddToCurrentPyScript(aStr);
// Add elements to the group
2004-12-01 15:48:31 +05:00
SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
if (aGroupDS) {
int nbAdd = 0;
for (int i = 0; i < theIDs.length(); i++) {
int anID = (int) theIDs[i];
if (aGroupDS->Add(anID))
nbAdd++;
2004-06-18 14:34:31 +06:00
}
2004-12-01 15:48:31 +05:00
return nbAdd;
2004-06-18 14:34:31 +06:00
}
MESSAGE("attempt to add elements to a vague group");
return 0;
}
//=============================================================================
/*!
*
*/
//=============================================================================
CORBA::Long SMESH_Group_i::Remove( const SMESH::long_array& theIDs )
{
// Update Python script
TCollection_AsciiString aStr ("nbDel = ");
SMESH_Gen_i::AddObject(aStr, _this()) += ".Remove(";
SMESH_Gen_i::AddArray(aStr, theIDs) += ")";
SMESH_Gen_i::AddToCurrentPyScript(aStr);
// Remove elements from the group
SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
if (aGroupDS) {
int nbDel = 0;
for (int i = 0; i < theIDs.length(); i++) {
int anID = (int) theIDs[i];
if (aGroupDS->Remove(anID))
nbDel++;
}
return nbDel;
}
MESSAGE("attempt to remove elements from a vague group");
return 0;
}
//=============================================================================
/*!
*
*/
//=============================================================================
typedef bool (SMESHDS_Group::*TFunChangeGroup)(const int);
CORBA::Long
ChangeByPredicate( SMESH::Predicate_i* thePredicate,
SMESHDS_GroupBase* theGroupBase,
TFunChangeGroup theFun)
{
CORBA::Long aNb = 0;
if(SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>(theGroupBase)){
SMESH::Controls::Filter::TIdSequence aSequence;
const SMDS_Mesh* aMesh = theGroupBase->GetMesh();
SMESH::Filter_i::GetElementsId(thePredicate,aMesh,aSequence);
CORBA::Long i = 0, iEnd = aSequence.size();
for(; i < iEnd; i++)
if((aGroupDS->*theFun)(aSequence[i]))
aNb++;
return aNb;
}
return aNb;
}
CORBA::Long
SMESH_Group_i::
AddByPredicate( SMESH::Predicate_ptr thePredicate )
{
if(SMESH::Predicate_i* aPredicate = SMESH::GetPredicate(thePredicate)){
TPythonDump()<<_this()<<".AddByPredicate("<<aPredicate<<")";
return ChangeByPredicate(aPredicate,GetGroupDS(),&SMESHDS_Group::Add);
}
return 0;
}
CORBA::Long
SMESH_Group_i::
RemoveByPredicate( SMESH::Predicate_ptr thePredicate )
{
if(SMESH::Predicate_i* aPredicate = SMESH::GetPredicate(thePredicate)){
TPythonDump()<<_this()<<".RemoveByPredicate("<<aPredicate<<")";
return ChangeByPredicate(aPredicate,GetGroupDS(),&SMESHDS_Group::Remove);
}
return 0;
}
//=============================================================================
/*!
*
*/
//=============================================================================
2004-12-01 15:48:31 +05:00
CORBA::Long SMESH_GroupBase_i::GetID( CORBA::Long theIndex )
2004-06-18 14:34:31 +06:00
{
2004-12-01 15:48:31 +05:00
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
if (aGroupDS)
return aGroupDS->GetID(theIndex);
2004-06-18 14:34:31 +06:00
MESSAGE("attempt to iterate on a vague group");
return -1;
}
//=============================================================================
/*!
*
*/
//=============================================================================
2004-12-01 15:48:31 +05:00
SMESH::long_array* SMESH_GroupBase_i::GetListOfID()
2004-06-18 14:34:31 +06:00
{
SMESH::long_array_var aRes = new SMESH::long_array();
2004-12-01 15:48:31 +05:00
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
if (aGroupDS) {
int aSize = aGroupDS->Extent();
aRes->length(aSize);
for (int i = 0; i < aSize; i++)
aRes[i] = aGroupDS->GetID(i+1);
return aRes._retn();
2004-06-18 14:34:31 +06:00
}
MESSAGE("get list of IDs of a vague group");
return aRes._retn();
}
//=============================================================================
/*!
*
*/
//=============================================================================
2004-12-01 15:48:31 +05:00
SMESH::SMESH_Mesh_ptr SMESH_GroupBase_i::GetMesh()
2004-06-18 14:34:31 +06:00
{
SMESH::SMESH_Mesh_var aMesh;
if ( myMeshServant )
aMesh = SMESH::SMESH_Mesh::_narrow( myMeshServant->_this() );
return aMesh._retn();
}
//=============================================================================
/*!
*
*/
//=============================================================================
2004-12-01 15:48:31 +05:00
SMESH::long_array* SMESH_GroupBase_i::GetIDs()
{
SMESH::long_array_var aResult = GetListOfID();
return aResult._retn();
}
//=======================================================================
//function : GetShape
//purpose :
//=======================================================================
GEOM::GEOM_Object_ptr SMESH_GroupOnGeom_i::GetShape()
2004-06-18 14:34:31 +06:00
{
2004-12-01 15:48:31 +05:00
GEOM::GEOM_Object_var aGeomObj;
SMESHDS_GroupOnGeom* aGroupDS = dynamic_cast<SMESHDS_GroupOnGeom*>( GetGroupDS() );
if ( aGroupDS ) {
SMESH_Gen_i* aGen = GetMeshServant()->GetGen();
aGeomObj = aGen->ShapeToGeomObject( aGroupDS->GetShape() );
}
return aGeomObj._retn();
2004-06-18 14:34:31 +06:00
}
2004-12-01 15:48:31 +05:00