-// SMESH SMESH_I : idl implementation based on 'SMESH' unit's classes
+// Copyright (C) 2007-2016 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.
//
-// 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
+// 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
-// $Header$
-
-
+//
#include "SMESH_Group_i.hxx"
-#include "SMESH_Mesh_i.hxx"
-#include "SMESH_Gen_i.hxx"
-#include "SMESH_Group.hxx"
+
#include "SMESHDS_Group.hxx"
+#include "SMESHDS_GroupOnFilter.hxx"
#include "SMESHDS_GroupOnGeom.hxx"
-#include "SMDSAbs_ElementType.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 )
+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 )
{
- thePOA->activate_object( this );
+ // 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_Group_i::SMESH_Group_i( PortableServer::POA_ptr thePOA, SMESH_Mesh_i* theMeshServant, const int theLocalID )
-: 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_GroupOnGeom_i::SMESH_GroupOnGeom_i( PortableServer::POA_ptr thePOA, SMESH_Mesh_i* theMeshServant, const int theLocalID )
-: 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()
{
- MESSAGE("~SMESH_GroupBase_i;" );
- if ( myMeshServant )
- myMeshServant->removeGroup(myLocalID);
+ if ( myPreMeshInfo ) delete myPreMeshInfo; myPreMeshInfo = NULL;
}
//=======================================================================
void SMESH_GroupBase_i::SetName( const char* theName )
{
+ // Perform renaming
::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 );
+ if (!aGroup) {
+ MESSAGE("can't set name of a vague group");
return;
}
- MESSAGE("can't set name of a vague group");
+
+ 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();
+ SALOMEDS::Study_var aStudy = aGen->GetCurrentStudy();
+ SMESH::SMESH_GroupBase_var aGrp = _this();
+ SALOMEDS::SObject_var anSO = aGen->ObjectToSObject( aStudy, aGrp );
+ if ( !anSO->_is_nil() )
+ {
+ aGen->SetName( anSO, theName );
+
+ // Update Python script
+ TPythonDump() << anSO << ".SetName( '" << theName << "' )";
+ }
}
//=============================================================================
::SMESH_Group* aGroup = GetSmeshGroup();
if (aGroup)
return CORBA::string_dup (aGroup->GetName());
- MESSAGE("get name of a vague group");
return CORBA::string_dup( "NO_NAME" );
}
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;
+ 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;
}
- MESSAGE("get type of a vague group");
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();
- MESSAGE("get size of a vague group");
return 0;
}
CORBA::Boolean SMESH_GroupBase_i::IsEmpty()
{
+ if ( myPreMeshInfo )
+ return Size() == 0;
+
SMESHDS_GroupBase* aGroupDS = GetGroupDS();
if (aGroupDS)
return aGroupDS->IsEmpty();
- MESSAGE("checking IsEmpty of a vague group");
return true;
}
+//=============================================================================
+/*
+ * Returns \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;
}
- MESSAGE("attempt to clear a vague group");
+ 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);
- MESSAGE("attempt to check contents of a vague group");
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 (int i = 0; i < theIDs.length(); i++) {
+ for ( CORBA::ULong i = 0; i < theIDs.length(); i++) {
int anID = (int) theIDs[i];
- if (aGroupDS->Add(anID))
+ 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");
//=============================================================================
/*!
- *
+ *
+ */
+//=============================================================================
+
+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);
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) {
+ 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();
+
+ if ( 0 < aSize && aSize < 100 ) // for comfortable testing ;)
+ std::sort( &aRes[0], &aRes[0]+aSize );
}
- MESSAGE("get list of IDs of a vague group");
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_Group_i::Remove( const SMESH::long_array& theIDs )
+CORBA::Long SMESH_GroupBase_i::GetNumberOfNodes()
{
- 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++;
+ 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 nbDel;
}
- MESSAGE("attempt to remove elements from a vague group");
- return 0;
+ return myNbNodes;
}
-//=============================================================================
+//================================================================================
/*!
- *
+ * \brief Return true if GetNumberOfNodes() won't take a long time for computation
*/
-//=============================================================================
-SMESH::SMESH_Mesh_ptr SMESH_GroupBase_i::GetMesh()
+//================================================================================
+
+CORBA::Boolean SMESH_GroupBase_i::IsNodeInfoAvailable()
{
- SMESH::SMESH_Mesh_var aMesh;
- if ( myMeshServant )
- aMesh = SMESH::SMESH_Mesh::_narrow( myMeshServant->_this() );
- return aMesh._retn();
+ 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::long_array* SMESH_GroupBase_i::GetIDs()
+SMESH::SMESH_Mesh_ptr SMESH_GroupBase_i::GetMesh()
{
- SMESH::long_array_var aResult = GetListOfID();
- return aResult._retn();
+ SMESH::SMESH_Mesh_var aMesh;
+ if ( myMeshServant )
+ aMesh = myMeshServant->_this();
+ return aMesh._retn();
}
//=======================================================================
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 ;
+}
+
+//=============================================================================
+/*
+ * Returns 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();
+}
+
+//=============================================================================
+/*
+ * Returns 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 : Returns ids of members
+//=======================================================================
+
+SMESH::long_array* SMESH_GroupBase_i::GetIDs()
+{
+ return GetListOfID();
+}
+
+//=======================================================================
+//function : GetTypes
+//purpose : Returns 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 : * Returns 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
+ }
+
+ TPythonDump()<< SMESH::SMESH_GroupOnFilter_var(_this()) <<".SetFilter( "<<theFilter<<" )";
+}
+
+//================================================================================
+/*!
+ * \brief Returns 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();
+}
+
+//=======================================================================
+//function : GetIDs
+//purpose : Returns 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();
+}
+
+//=============================================================================
+/*!
+ * Returns 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
+}
