mirror of
https://git.salome-platform.org/gitpub/modules/smesh.git
synced 2024-12-29 10:50:34 +05:00
1084 lines
33 KiB
C++
1084 lines
33 KiB
C++
// Copyright (C) 2007-2019 CEA/DEN, EDF R&D, OPEN CASCADE
|
|
//
|
|
// Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
|
|
// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
|
|
//
|
|
// 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, or (at your option) any later version.
|
|
//
|
|
// 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
|
|
//
|
|
|
|
// SMESH SMESH_I : idl implementation based on 'SMESH' unit's classes
|
|
// File : SMESH_Group_i.cxx
|
|
// Author : Sergey ANIKIN, OCC
|
|
// Module : SMESH
|
|
//
|
|
#include "SMESH_Group_i.hxx"
|
|
|
|
#include "SMESHDS_Group.hxx"
|
|
#include "SMESHDS_GroupOnFilter.hxx"
|
|
#include "SMESHDS_GroupOnGeom.hxx"
|
|
#include "SMESHDS_Mesh.hxx"
|
|
#include "SMESH_Comment.hxx"
|
|
#include "SMESH_Filter_i.hxx"
|
|
#include "SMESH_Gen_i.hxx"
|
|
#include "SMESH_Group.hxx"
|
|
#include "SMESH_Mesh_i.hxx"
|
|
#include "SMESH_PythonDump.hxx"
|
|
#include "SMESH_PreMeshInfo.hxx"
|
|
|
|
#include CORBA_SERVER_HEADER(SMESH_Filter)
|
|
|
|
#include "utilities.h"
|
|
|
|
using namespace SMESH;
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
|
|
SMESH_GroupBase_i::SMESH_GroupBase_i( PortableServer::POA_ptr thePOA,
|
|
SMESH_Mesh_i* theMeshServant,
|
|
const int theLocalID )
|
|
: SALOME::GenericObj_i( thePOA ),
|
|
myPreMeshInfo(NULL),
|
|
myNbNodes(-1),
|
|
myGroupDSTic(0),
|
|
myMeshServant( theMeshServant ),
|
|
myLocalID( theLocalID )
|
|
{
|
|
// 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 );
|
|
}
|
|
|
|
SMESH_Group_i::SMESH_Group_i( PortableServer::POA_ptr thePOA,
|
|
SMESH_Mesh_i* theMeshServant,
|
|
const int theLocalID )
|
|
: SALOME::GenericObj_i( thePOA ),
|
|
SMESH_GroupBase_i( thePOA, theMeshServant, theLocalID )
|
|
{
|
|
}
|
|
|
|
SMESH_GroupOnGeom_i::SMESH_GroupOnGeom_i( PortableServer::POA_ptr thePOA,
|
|
SMESH_Mesh_i* theMeshServant,
|
|
const int theLocalID )
|
|
: SALOME::GenericObj_i( thePOA ),
|
|
SMESH_GroupBase_i( thePOA, theMeshServant, theLocalID )
|
|
{
|
|
}
|
|
|
|
SMESH_GroupOnFilter_i::SMESH_GroupOnFilter_i( PortableServer::POA_ptr thePOA,
|
|
SMESH_Mesh_i* theMeshServant,
|
|
const int theLocalID )
|
|
: SALOME::GenericObj_i( thePOA ),
|
|
SMESH_GroupBase_i( thePOA, theMeshServant, theLocalID )
|
|
{
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
|
|
SMESH_GroupBase_i::~SMESH_GroupBase_i()
|
|
{
|
|
if ( myPreMeshInfo ) delete myPreMeshInfo; myPreMeshInfo = NULL;
|
|
}
|
|
|
|
//=======================================================================
|
|
//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;
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
|
|
void SMESH_GroupBase_i::SetName( const char* theName )
|
|
{
|
|
// Perform renaming
|
|
::SMESH_Group* aGroup = GetSmeshGroup();
|
|
if (!aGroup) {
|
|
MESSAGE("can't set name of a vague group");
|
|
return;
|
|
}
|
|
|
|
if ( aGroup->GetName() && !strcmp( aGroup->GetName(), theName ) )
|
|
return; // nothing to rename
|
|
|
|
aGroup->SetName(theName);
|
|
|
|
// Update group name in a study
|
|
SMESH_Gen_i* aGen = myMeshServant->GetGen();
|
|
SMESH::SMESH_GroupBase_var aGrp = _this();
|
|
SALOMEDS::SObject_var anSO = aGen->ObjectToSObject( aGrp );
|
|
if ( !anSO->_is_nil() )
|
|
{
|
|
aGen->SetName( anSO, theName );
|
|
|
|
// Update Python script
|
|
TPythonDump() << anSO << ".SetName( '" << theName << "' )";
|
|
}
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
|
|
char* SMESH_GroupBase_i::GetName()
|
|
{
|
|
::SMESH_Group* aGroup = GetSmeshGroup();
|
|
if (aGroup)
|
|
return CORBA::string_dup (aGroup->GetName());
|
|
return CORBA::string_dup( "NO_NAME" );
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
|
|
SMESH::ElementType SMESH_GroupBase_i::GetType()
|
|
{
|
|
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;
|
|
case SMDSAbs_0DElement: aType = SMESH::ELEM0D; break;
|
|
case SMDSAbs_Ball: aType = SMESH::BALL; break;
|
|
default: aType = SMESH::ALL; break;
|
|
}
|
|
return aType;
|
|
}
|
|
return SMESH::ALL;
|
|
}
|
|
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
|
|
CORBA::Long SMESH_GroupBase_i::Size()
|
|
{
|
|
if ( myPreMeshInfo )
|
|
return GetType() == SMESH::NODE ? myPreMeshInfo->NbNodes() : myPreMeshInfo->NbElements();
|
|
|
|
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
|
|
if (aGroupDS)
|
|
return aGroupDS->Extent();
|
|
return 0;
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
|
|
CORBA::Boolean SMESH_GroupBase_i::IsEmpty()
|
|
{
|
|
if ( myPreMeshInfo )
|
|
return Size() == 0;
|
|
|
|
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
|
|
if (aGroupDS)
|
|
return aGroupDS->IsEmpty();
|
|
return true;
|
|
}
|
|
|
|
//=============================================================================
|
|
/*
|
|
* Return \c true if \c this group depends on the \a other via
|
|
* FT_BelongToMeshGroup predicate or vice versa
|
|
*/
|
|
//=============================================================================
|
|
|
|
bool SMESH_GroupBase_i::IsInDependency( SMESH::SMESH_GroupBase_ptr other )
|
|
{
|
|
if ( NotifyerAndWaiter* nw = SMESH::DownCast< NotifyerAndWaiter* >( other ))
|
|
return ( nw->ContainModifWaiter( this ) || this->ContainModifWaiter( nw ));
|
|
|
|
return false;
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
|
|
void SMESH_Group_i::Clear()
|
|
{
|
|
if ( myPreMeshInfo )
|
|
myPreMeshInfo->FullLoadFromFile();
|
|
|
|
// Update Python script
|
|
TPythonDump() << SMESH::SMESH_Group_var(_this()) << ".Clear()";
|
|
|
|
// Clear the group
|
|
SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
|
|
if (aGroupDS) {
|
|
aGroupDS->Clear();
|
|
return;
|
|
}
|
|
Modified(); // notify dependent Filter with FT_BelongToMeshGroup criterion
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
|
|
CORBA::Boolean SMESH_GroupBase_i::Contains( CORBA::Long theID )
|
|
{
|
|
if ( myPreMeshInfo )
|
|
myPreMeshInfo->FullLoadFromFile();
|
|
|
|
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
|
|
if (aGroupDS)
|
|
return aGroupDS->Contains(theID);
|
|
return false;
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
|
|
CORBA::Long SMESH_Group_i::Add( const SMESH::long_array& theIDs )
|
|
{
|
|
if ( myPreMeshInfo )
|
|
myPreMeshInfo->FullLoadFromFile();
|
|
|
|
// Update Python script
|
|
TPythonDump() << "nbAdd = " << SMESH::SMESH_Group_var(_this()) << ".Add( " << theIDs << " )";
|
|
|
|
// Add elements to the group
|
|
SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
|
|
if (aGroupDS) {
|
|
int nbAdd = 0;
|
|
for ( CORBA::ULong i = 0; i < theIDs.length(); i++) {
|
|
int anID = (int) theIDs[i];
|
|
if ( aGroupDS->Add( anID ))
|
|
nbAdd++;
|
|
}
|
|
if ( nbAdd )
|
|
Modified(); // notify dependent Filter with FT_BelongToMeshGroup criterion
|
|
return nbAdd;
|
|
}
|
|
MESSAGE("attempt to add elements to a vague group");
|
|
return 0;
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
|
|
CORBA::Long SMESH_Group_i::Remove( const SMESH::long_array& theIDs )
|
|
{
|
|
if ( myPreMeshInfo )
|
|
myPreMeshInfo->FullLoadFromFile();
|
|
|
|
// Update Python script
|
|
TPythonDump() << "nbDel = " << SMESH::SMESH_Group_var(_this())
|
|
<< ".Remove( " << theIDs << " )";
|
|
|
|
// Remove elements from the group
|
|
SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
|
|
if (aGroupDS) {
|
|
int nbDel = 0;
|
|
for ( CORBA::ULong i = 0; i < theIDs.length(); i++ ) {
|
|
int anID = (int) theIDs[i];
|
|
if ( aGroupDS->Remove( anID ))
|
|
nbDel++;
|
|
}
|
|
if ( nbDel )
|
|
Modified(); // notify dependent Filter with FT_BelongToMeshGroup criterion
|
|
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,
|
|
NotifyerAndWaiter* theGroupImpl,
|
|
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++;
|
|
if ( aNb )
|
|
theGroupImpl->Modified();
|
|
return aNb;
|
|
}
|
|
return aNb;
|
|
}
|
|
|
|
CORBA::Long
|
|
SMESH_Group_i::
|
|
AddByPredicate( SMESH::Predicate_ptr thePredicate )
|
|
{
|
|
if ( myPreMeshInfo )
|
|
myPreMeshInfo->FullLoadFromFile();
|
|
|
|
if(SMESH::Predicate_i* aPredicate = SMESH::GetPredicate(thePredicate)){
|
|
TPythonDump() << SMESH::SMESH_Group_var(_this())
|
|
<< ".AddByPredicate( " << aPredicate << " )";
|
|
return ChangeByPredicate( aPredicate, GetGroupDS(), this, &SMESHDS_Group::Add );
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
CORBA::Long
|
|
SMESH_Group_i::
|
|
RemoveByPredicate( SMESH::Predicate_ptr thePredicate )
|
|
{
|
|
if ( myPreMeshInfo )
|
|
myPreMeshInfo->FullLoadFromFile();
|
|
|
|
if(SMESH::Predicate_i* aPredicate = SMESH::GetPredicate(thePredicate)){
|
|
TPythonDump() << SMESH::SMESH_Group_var(_this())
|
|
<< ".RemoveByPredicate( " << aPredicate << " )";
|
|
return ChangeByPredicate(aPredicate,GetGroupDS(),this, &SMESHDS_Group::Remove);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
CORBA::Long SMESH_Group_i::AddFrom( SMESH::SMESH_IDSource_ptr theSource )
|
|
{
|
|
if ( myPreMeshInfo )
|
|
myPreMeshInfo->FullLoadFromFile();
|
|
|
|
TPythonDump pd;
|
|
long prevNb = Size();
|
|
SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
|
|
if (aGroupDS) {
|
|
if ( SMDS_ElemIteratorPtr elemIt = SMESH_Mesh_i::GetElements( theSource, GetType() ))
|
|
while ( elemIt->more() )
|
|
aGroupDS->SMDSGroup().Add( elemIt->next() );
|
|
}
|
|
|
|
// Update Python script
|
|
pd << "nbAdd = " << SMESH::SMESH_Group_var(_this()) << ".AddFrom( " << theSource << " )";
|
|
|
|
if ( prevNb != Size() )
|
|
Modified(); // notify dependent Filter with FT_BelongToMeshGroup criterion
|
|
|
|
return Size() - prevNb;
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
|
|
CORBA::Long SMESH_GroupBase_i::GetID( CORBA::Long theIndex )
|
|
{
|
|
if ( myPreMeshInfo )
|
|
myPreMeshInfo->FullLoadFromFile();
|
|
|
|
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
|
|
if (aGroupDS)
|
|
return aGroupDS->GetID(theIndex);
|
|
MESSAGE("attempt to iterate on a vague group");
|
|
return -1;
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
|
|
SMESH::long_array* SMESH_GroupBase_i::GetListOfID()
|
|
{
|
|
if ( myPreMeshInfo )
|
|
myPreMeshInfo->FullLoadFromFile();
|
|
|
|
SMESH::long_array_var aRes = new SMESH::long_array();
|
|
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
|
|
if (aGroupDS)
|
|
{
|
|
int aSize = aGroupDS->Extent();
|
|
aRes->length(aSize);
|
|
SMDS_ElemIteratorPtr it = aGroupDS->GetElements();
|
|
for (int i = 0; it->more(); i++)
|
|
aRes[i] = it->next()->GetID();
|
|
|
|
if ( 0 < aSize && aSize < 100 ) // for comfortable testing ;)
|
|
std::sort( &aRes[0], &aRes[0]+aSize );
|
|
}
|
|
return aRes._retn();
|
|
}
|
|
|
|
namespace
|
|
{
|
|
//================================================================================
|
|
/*!
|
|
* \brief return nodes of elements pointered by iterator
|
|
*/
|
|
//================================================================================
|
|
|
|
void getNodesOfElements(SMDS_ElemIteratorPtr elemIt,
|
|
std::set<const SMDS_MeshNode* >& nodes)
|
|
{
|
|
while ( elemIt->more() )
|
|
{
|
|
const SMDS_MeshElement* e = elemIt->next();
|
|
nodes.insert( e->begin_nodes(), e->end_nodes() );
|
|
}
|
|
}
|
|
}
|
|
|
|
//================================================================================
|
|
/*!
|
|
* \brief return the number of nodes of cells included to the group
|
|
*/
|
|
//================================================================================
|
|
|
|
CORBA::Long SMESH_GroupBase_i::GetNumberOfNodes()
|
|
{
|
|
if ( GetType() == SMESH::NODE )
|
|
return Size();
|
|
|
|
if ( myPreMeshInfo )
|
|
myPreMeshInfo->FullLoadFromFile();
|
|
|
|
if ( SMESHDS_GroupBase* g = GetGroupDS())
|
|
{
|
|
if ( myNbNodes < 0 || g->GetTic() != myGroupDSTic )
|
|
{
|
|
std::set<const SMDS_MeshNode* > nodes;
|
|
getNodesOfElements( g->GetElements(), nodes );
|
|
myNbNodes = nodes.size();
|
|
myGroupDSTic = g->GetTic();
|
|
}
|
|
}
|
|
return myNbNodes;
|
|
}
|
|
|
|
//================================================================================
|
|
/*!
|
|
* \brief Return true if GetNumberOfNodes() won't take a long time for computation
|
|
*/
|
|
//================================================================================
|
|
|
|
CORBA::Boolean SMESH_GroupBase_i::IsNodeInfoAvailable()
|
|
{
|
|
if ( GetType() == SMESH::NODE/* || Size() < 100000 */)
|
|
return true;
|
|
if ( myPreMeshInfo )
|
|
return false;
|
|
if ( SMESHDS_GroupBase* g = GetGroupDS())
|
|
return ( myNbNodes > -1 && g->GetTic() == myGroupDSTic);
|
|
return false;
|
|
}
|
|
|
|
//================================================================================
|
|
/*!
|
|
* \brief Return IDs of nodes of cells included to the group
|
|
*/
|
|
//================================================================================
|
|
|
|
SMESH::long_array* SMESH_GroupBase_i::GetNodeIDs()
|
|
{
|
|
if ( GetType() == SMESH::NODE )
|
|
return GetListOfID();
|
|
|
|
if ( myPreMeshInfo )
|
|
myPreMeshInfo->FullLoadFromFile();
|
|
|
|
SMESH::long_array_var aRes = new SMESH::long_array();
|
|
if ( SMESHDS_GroupBase* g = GetGroupDS())
|
|
{
|
|
std::set<const SMDS_MeshNode* > nodes;
|
|
getNodesOfElements( g->GetElements(), nodes );
|
|
aRes->length( nodes.size() );
|
|
std::set<const SMDS_MeshNode*>::iterator nIt = nodes.begin(), nEnd = nodes.end();
|
|
for ( int i = 0; nIt != nEnd; ++nIt, ++i )
|
|
aRes[i] = (*nIt)->GetID();
|
|
}
|
|
return aRes._retn();
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
SMESH::SMESH_Mesh_ptr SMESH_GroupBase_i::GetMesh()
|
|
{
|
|
SMESH::SMESH_Mesh_var aMesh;
|
|
if ( myMeshServant )
|
|
aMesh = myMeshServant->_this();
|
|
return aMesh._retn();
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
GEOM::GEOM_Object_ptr SMESH_GroupOnGeom_i::GetShape()
|
|
{
|
|
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();
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
SALOMEDS::Color SMESH_GroupBase_i::GetColor()
|
|
{
|
|
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
|
|
if (aGroupDS)
|
|
{
|
|
Quantity_Color aQColor = aGroupDS->GetColor();
|
|
SALOMEDS::Color aColor;
|
|
aColor.R = aQColor.Red();
|
|
aColor.G = aQColor.Green();
|
|
aColor.B = aQColor.Blue();
|
|
|
|
return aColor;
|
|
}
|
|
return SALOMEDS::Color();
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
void SMESH_GroupBase_i::SetColor(const SALOMEDS::Color& color)
|
|
{
|
|
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
|
|
if (aGroupDS)
|
|
{
|
|
Quantity_Color aQColor( color.R, color.G, color.B, Quantity_TOC_RGB );
|
|
Quantity_Color oldColor = aGroupDS->GetColor();
|
|
if ( oldColor != aQColor )
|
|
{
|
|
aGroupDS->SetColor(aQColor);
|
|
TPythonDump()<< SMESH::SMESH_GroupBase_var(_this())
|
|
<< ".SetColor( SALOMEDS.Color( "
|
|
<<color.R<<", "<<color.G<<", "<<color.B<<" ))";
|
|
}
|
|
}
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
CORBA::Long SMESH_GroupBase_i::GetColorNumber()
|
|
{
|
|
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
|
|
if (aGroupDS)
|
|
return aGroupDS->GetColorGroup();
|
|
MESSAGE("get color number of a group");
|
|
return 0;
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
*
|
|
*/
|
|
//=============================================================================
|
|
void SMESH_GroupBase_i::SetColorNumber(CORBA::Long color)
|
|
{
|
|
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
|
|
if (aGroupDS)
|
|
{
|
|
aGroupDS->SetColorGroup(color);
|
|
TPythonDump()<<SMESH::SMESH_GroupBase_var(_this())<<".SetColorNumber( "<<color<<" )";
|
|
}
|
|
return ;
|
|
}
|
|
|
|
//=============================================================================
|
|
/*
|
|
* Return number of mesh elements of each \a SMESH::EntityType
|
|
* Result array of number of elements per \a SMESH::EntityType
|
|
* Inherited from SMESH_IDSource
|
|
*/
|
|
//=============================================================================
|
|
|
|
SMESH::long_array* SMESH_GroupBase_i::GetMeshInfo()
|
|
{
|
|
if ( myPreMeshInfo )
|
|
return myPreMeshInfo->GetMeshInfo();
|
|
|
|
SMESH::long_array_var aRes = new SMESH::long_array();
|
|
aRes->length(SMESH::Entity_Last);
|
|
for (int i = SMESH::Entity_Node; i < SMESH::Entity_Last; i++)
|
|
aRes[i] = 0;
|
|
|
|
if ( SMESHDS_GroupBase* g = GetGroupDS())
|
|
{
|
|
if ( g->GetType() == SMDSAbs_Node /*|| ( myNbNodes > -1 && g->GetTic() == myGroupDSTic)*/)
|
|
aRes[ SMDSEntity_Node ] = GetNumberOfNodes();
|
|
|
|
if ( g->GetType() != SMDSAbs_Node )
|
|
SMESH_Mesh_i::CollectMeshInfo( g->GetElements(), aRes);
|
|
}
|
|
|
|
return aRes._retn();
|
|
}
|
|
|
|
//=============================================================================
|
|
/*
|
|
* Return number of mesh elements of each \a ElementType
|
|
*/
|
|
//=============================================================================
|
|
|
|
SMESH::long_array* SMESH_GroupBase_i::GetNbElementsByType()
|
|
{
|
|
SMESH::long_array_var aRes = new SMESH::long_array();
|
|
aRes->length(SMESH::NB_ELEMENT_TYPES);
|
|
for (int i = 0; i < SMESH::NB_ELEMENT_TYPES; i++)
|
|
aRes[ i ] = 0;
|
|
|
|
if ( myPreMeshInfo )
|
|
aRes[ GetType() ] = myPreMeshInfo->NbElements( SMDSAbs_ElementType( GetType() ));
|
|
else
|
|
aRes[ GetType() ] = Size();
|
|
|
|
return aRes._retn();
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetIDs
|
|
//purpose : Return ids of members
|
|
//=======================================================================
|
|
|
|
SMESH::long_array* SMESH_GroupBase_i::GetIDs()
|
|
{
|
|
return GetListOfID();
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetTypes
|
|
//purpose : Return types of elements it contains
|
|
//=======================================================================
|
|
|
|
SMESH::array_of_ElementType* SMESH_GroupBase_i::GetTypes()
|
|
{
|
|
SMESH::array_of_ElementType_var types = new SMESH::array_of_ElementType;
|
|
if ( !IsEmpty() )
|
|
{
|
|
types->length( 1 );
|
|
types[0] = GetType();
|
|
}
|
|
return types._retn();
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : IsMeshInfoCorrect
|
|
//purpose : * Return false if GetMeshInfo() returns incorrect information that may
|
|
// * happen if mesh data is not yet fully loaded from the file of study.
|
|
//=======================================================================
|
|
|
|
bool SMESH_GroupBase_i::IsMeshInfoCorrect()
|
|
{
|
|
return myPreMeshInfo ? myPreMeshInfo->IsMeshInfoCorrect() : true;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetVtkUgStream
|
|
//purpose : Return data vtk unstructured grid (not implemented)
|
|
//=======================================================================
|
|
|
|
SALOMEDS::TMPFile* SMESH_GroupBase_i::GetVtkUgStream()
|
|
{
|
|
SALOMEDS::TMPFile_var SeqFile;
|
|
return SeqFile._retn();
|
|
}
|
|
|
|
//================================================================================
|
|
/*!
|
|
* \brief Retrieves the predicate from the filter
|
|
*/
|
|
//================================================================================
|
|
|
|
SMESH_PredicatePtr SMESH_GroupOnFilter_i::GetPredicate( SMESH::Filter_ptr filter )
|
|
{
|
|
SMESH_PredicatePtr predicate;
|
|
|
|
if ( SMESH::Filter_i* filt_i = SMESH::DownCast< SMESH::Filter_i* >( filter ))
|
|
if ( SMESH::Predicate_i* predic_i= filt_i->GetPredicate_i() )
|
|
predicate = predic_i->GetPredicate();
|
|
|
|
return predicate;
|
|
}
|
|
|
|
//================================================================================
|
|
/*!
|
|
* \brief Sets the filter defining group contents
|
|
*/
|
|
//================================================================================
|
|
|
|
void SMESH_GroupOnFilter_i::SetFilter(SMESH::Filter_ptr theFilter)
|
|
throw (SALOME::SALOME_Exception)
|
|
{
|
|
if ( myFilter->_is_equivalent( theFilter ))
|
|
return;
|
|
|
|
if ( myPreMeshInfo )
|
|
myPreMeshInfo->FullLoadFromFile();
|
|
|
|
if ( ! myFilter->_is_nil() )
|
|
myFilter->UnRegister();
|
|
|
|
myFilter = SMESH::Filter::_duplicate( theFilter );
|
|
|
|
if ( !myFilter->_is_nil() )
|
|
{
|
|
myFilter->Register();
|
|
|
|
if ( SMESH::Filter_i* f = SMESH::DownCast< SMESH::Filter_i* >( myFilter ))
|
|
{
|
|
// make filter notify me about change of either a predicate or a base group
|
|
f->FindBaseObjects();
|
|
|
|
if ( f->ContainModifWaiter( this ) ||
|
|
this->ContainModifWaiter( f ))
|
|
{
|
|
SetFilter( SMESH::Filter::_nil() );
|
|
THROW_SALOME_CORBA_EXCEPTION( "Cyclic dependency between Groups on Filter",
|
|
SALOME::BAD_PARAM );
|
|
}
|
|
f->AddModifWaiter( this );
|
|
}
|
|
myFilter->SetMesh( SMESH::SMESH_Mesh::_nil() ); // to UnRegister() the mesh
|
|
}
|
|
|
|
if ( SMESHDS_GroupOnFilter* grDS = dynamic_cast< SMESHDS_GroupOnFilter*>( GetGroupDS() ))
|
|
{
|
|
grDS->SetPredicate( GetPredicate( myFilter ));
|
|
Modified(); // notify dependent Filter with FT_BelongToMeshGroup criterion
|
|
}
|
|
|
|
SMESH::SMESH_GroupOnFilter_var me = _this();
|
|
|
|
// mark the group valid after edition
|
|
GetMeshServant()->GetGen()->HighLightInvalid( me, false );
|
|
|
|
|
|
TPythonDump()<< me <<".SetFilter( "<< theFilter <<" )";
|
|
}
|
|
|
|
//================================================================================
|
|
/*!
|
|
* \brief Return the filter defining group contents
|
|
*/
|
|
//================================================================================
|
|
|
|
SMESH::Filter_ptr SMESH_GroupOnFilter_i::GetFilter()
|
|
{
|
|
SMESH::Filter_var f = myFilter;
|
|
TPythonDump() << f << " = " << SMESH::SMESH_GroupOnFilter_var(_this()) << ".GetFilter()";
|
|
return f._retn();
|
|
}
|
|
|
|
//================================================================================
|
|
/*!
|
|
* @return true if group contents is computed
|
|
*/
|
|
//================================================================================
|
|
|
|
CORBA::Boolean SMESH_GroupOnFilter_i::IsUpToDate()
|
|
{
|
|
if ( myPreMeshInfo )
|
|
return false;
|
|
|
|
if ( SMESHDS_GroupOnFilter* grDS = dynamic_cast< SMESHDS_GroupOnFilter*>( GetGroupDS() ))
|
|
return grDS->IsUpToDate();
|
|
|
|
return false;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : IsMeshInfoCorrect
|
|
//purpose : Return false in two cases: 1) if mesh not loaded and GetMeshInfo() returns
|
|
// incorrect information 2) mesh loaded but group contents is not computed
|
|
//=======================================================================
|
|
|
|
bool SMESH_GroupOnFilter_i::IsMeshInfoCorrect()
|
|
{
|
|
return myPreMeshInfo ? myPreMeshInfo->IsMeshInfoCorrect() : IsUpToDate();
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetIDs
|
|
//purpose : Return ids of members
|
|
//=======================================================================
|
|
|
|
SMESH::long_array* SMESH_GroupOnFilter_i::GetListOfID()
|
|
{
|
|
if ( myPreMeshInfo )
|
|
myPreMeshInfo->FullLoadFromFile();
|
|
|
|
SMESH::long_array_var aRes = new SMESH::long_array();
|
|
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
|
|
if ( SMESHDS_GroupOnFilter* grDS = dynamic_cast< SMESHDS_GroupOnFilter*>( GetGroupDS() ))
|
|
{
|
|
const SMDS_MeshInfo& meshInfo = aGroupDS->GetMesh()->GetMeshInfo();
|
|
aRes->length( meshInfo.NbElements( aGroupDS->GetType() ));
|
|
if ( aRes->length() ) // else aRes[0] -> SIGSEGV
|
|
aRes->length( grDS->GetElementIds( &aRes[0] ));
|
|
|
|
if ( 0 < aRes->length() && aRes->length() < 100 ) // for comfortable testing ;)
|
|
std::sort( &aRes[0], &aRes[0] + aRes->length() );
|
|
}
|
|
return aRes._retn();
|
|
}
|
|
|
|
//=============================================================================
|
|
/*!
|
|
* Return statistic of mesh elements
|
|
* Result array of number enityties
|
|
* Inherited from SMESH_IDSource
|
|
*/
|
|
//=============================================================================
|
|
|
|
SMESH::long_array* SMESH_GroupOnFilter_i::GetMeshInfo()
|
|
{
|
|
if ( myPreMeshInfo )
|
|
return myPreMeshInfo->GetMeshInfo();
|
|
|
|
SMESH::long_array_var aRes = new SMESH::long_array();
|
|
aRes->length(SMESH::Entity_Last);
|
|
for (int i = SMESH::Entity_Node; i < SMESH::Entity_Last; i++)
|
|
aRes[i] = 0;
|
|
|
|
if ( SMESHDS_GroupBase* g = GetGroupDS())
|
|
{
|
|
if ( g->GetType() == SMDSAbs_Node /*|| ( myNbNodes > -1 && g->GetTic() == myGroupDSTic)*/)
|
|
aRes[ SMDSEntity_Node ] = GetNumberOfNodes();
|
|
|
|
if ( g->GetType() != SMDSAbs_Node )
|
|
{
|
|
std::vector< int > nbElems = static_cast< SMESHDS_GroupOnFilter* >( g )->GetMeshInfo();
|
|
for ( size_t i = SMESH::Entity_Node; i < SMESH::Entity_Last; i++)
|
|
if ( i < nbElems.size() )
|
|
aRes[i] = nbElems[ i ];
|
|
}
|
|
}
|
|
|
|
return aRes._retn();
|
|
}
|
|
|
|
#define SEPAR '^'
|
|
|
|
//================================================================================
|
|
/*!
|
|
* \brief Return a string to be used to store group definition in the study
|
|
*/
|
|
//================================================================================
|
|
|
|
std::string SMESH_GroupOnFilter_i::FilterToString() const
|
|
{
|
|
SMESH_Comment result;
|
|
SMESH::Filter::Criteria_var criteria;
|
|
if ( !myFilter->_is_nil() && myFilter->GetCriteria( criteria.out() ))
|
|
{
|
|
result << criteria->length() << SEPAR;
|
|
for ( unsigned i = 0; i < criteria->length(); ++i )
|
|
{
|
|
SMESH::Filter::Criterion& crit = criteria[ i ];
|
|
|
|
if ( SMESH::FunctorType( crit.Type ) == SMESH::FT_BelongToMeshGroup &&
|
|
crit.ThresholdID.in() && crit.ThresholdID.in()[0] )
|
|
{
|
|
CORBA::Object_var obj = SMESH_Gen_i::GetORB()->string_to_object( crit.ThresholdID );
|
|
if ( SMESH_GroupBase_i * g = SMESH::DownCast< SMESH_GroupBase_i*>( obj ))
|
|
if ( SMESHDS_GroupBase* gDS = g->GetGroupDS() )
|
|
crit.ThresholdID = gDS->GetStoreName();
|
|
}
|
|
// write FunctorType as string but not as number to assure correct
|
|
// persistence if enum FunctorType is modified by insertion in the middle
|
|
result << SMESH::FunctorTypeToString( SMESH::FunctorType( crit.Type )) << SEPAR;
|
|
result << SMESH::FunctorTypeToString( SMESH::FunctorType( crit.Compare )) << SEPAR;
|
|
result << crit.Threshold << SEPAR;
|
|
result << crit.ThresholdStr << SEPAR;
|
|
result << crit.ThresholdID << SEPAR;
|
|
result << SMESH::FunctorTypeToString( SMESH::FunctorType( crit.UnaryOp )) << SEPAR;
|
|
result << SMESH::FunctorTypeToString( SMESH::FunctorType( crit.BinaryOp ))<< SEPAR;
|
|
result << crit.Tolerance << SEPAR;
|
|
result << crit.TypeOfElement << SEPAR;
|
|
result << crit.Precision << SEPAR;
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
//================================================================================
|
|
/*!
|
|
* \brief Restore the filter by the persistent string
|
|
*/
|
|
//================================================================================
|
|
|
|
SMESH::Filter_ptr SMESH_GroupOnFilter_i::StringToFilter(const std::string& thePersistStr )
|
|
{
|
|
SMESH::Filter_var filter;
|
|
|
|
// divide thePersistStr into sub-strings
|
|
std::vector< std::string > strVec;
|
|
std::string::size_type from = 0, to;
|
|
while ( from < thePersistStr.size() )
|
|
{
|
|
to = thePersistStr.find( SEPAR, from );
|
|
if ( to == std::string::npos )
|
|
break;
|
|
strVec.push_back( thePersistStr.substr( from, to-from ));
|
|
from = to+1;
|
|
}
|
|
if ( strVec.empty() || strVec[0] == "0" )
|
|
return filter._retn();
|
|
#undef SEPAR
|
|
|
|
// create Criteria
|
|
int nbCrit = atoi( strVec[0].c_str() );
|
|
SMESH::Filter::Criteria_var criteria = new SMESH::Filter::Criteria;
|
|
criteria->length( nbCrit );
|
|
int nbStrPerCrit = ( strVec.size() - 1 ) / nbCrit;
|
|
for ( int i = 0; i < nbCrit; ++i )
|
|
{
|
|
SMESH::Filter::Criterion& crit = criteria[ i ];
|
|
int iStr = 1 + i * nbStrPerCrit;
|
|
crit.Type = SMESH::StringToFunctorType( strVec[ iStr++ ].c_str() );
|
|
crit.Compare = SMESH::StringToFunctorType( strVec[ iStr++ ].c_str() );
|
|
crit.Threshold = atof( strVec[ iStr++ ].c_str() );
|
|
crit.ThresholdStr = strVec[ iStr++ ].c_str();
|
|
crit.ThresholdID = strVec[ iStr++ ].c_str();
|
|
crit.UnaryOp = SMESH::StringToFunctorType( strVec[ iStr++ ].c_str() );
|
|
crit.BinaryOp = SMESH::StringToFunctorType( strVec[ iStr++ ].c_str() );
|
|
crit.Tolerance = atof( strVec[ iStr++ ].c_str() );
|
|
crit.TypeOfElement= SMESH::ElementType( atoi( strVec[ iStr++ ].c_str() ));
|
|
crit.Precision = atoi( strVec[ iStr++ ].c_str() );
|
|
}
|
|
|
|
// create a filter
|
|
TPythonDump pd;
|
|
SMESH::FilterManager_i* aFilterMgr = new SMESH::FilterManager_i();
|
|
filter = aFilterMgr->CreateFilter();
|
|
filter->SetCriteria( criteria.inout() );
|
|
|
|
aFilterMgr->UnRegister();
|
|
|
|
pd << ""; // to avoid optimizing pd out
|
|
|
|
return filter._retn();
|
|
}
|
|
|
|
//================================================================================
|
|
/*!
|
|
* \brief Destructor of SMESH_GroupOnFilter_i
|
|
*/
|
|
//================================================================================
|
|
|
|
SMESH_GroupOnFilter_i::~SMESH_GroupOnFilter_i()
|
|
{
|
|
if ( ! myFilter->_is_nil() )
|
|
{
|
|
SMESH::DownCast< SMESH::Filter_i* >( myFilter )->RemoveModifWaiter( this );
|
|
myFilter->UnRegister();
|
|
}
|
|
}
|
|
|
|
//================================================================================
|
|
/*!
|
|
* \brief Method called when a predicate of myFilter changes
|
|
*/
|
|
//================================================================================
|
|
|
|
void SMESH_GroupOnFilter_i::OnBaseObjModified(NotifyerAndWaiter* filter, bool /*removed*/)
|
|
{
|
|
if ( myPreMeshInfo )
|
|
myPreMeshInfo->FullLoadFromFile();
|
|
|
|
if ( SMESHDS_GroupOnFilter* grDS = dynamic_cast< SMESHDS_GroupOnFilter*>( GetGroupDS() ))
|
|
grDS->SetPredicate( GetPredicate( myFilter )); // group resets its cache
|
|
}
|
|
|