+ 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
+}