// Module : SMESH
//
#include "SMESH_Group_i.hxx"
-#include "SMESH_Mesh_i.hxx"
-#include "SMESH_Gen_i.hxx"
-#include "SMESH_Group.hxx"
+
+#include "SMDSAbs_ElementType.hxx"
#include "SMESHDS_Group.hxx"
+#include "SMESHDS_GroupOnFilter.hxx"
#include "SMESHDS_GroupOnGeom.hxx"
-#include "SMDSAbs_ElementType.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 CORBA_SERVER_HEADER(SMESH_Filter)
*/
//=============================================================================
-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 ),
myMeshServant( theMeshServant ),
- myLocalID( theLocalID )
+ myLocalID( theLocalID ),
+ myNbNodes(-1),
+ myGroupDSTic(0)
{
// 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 )
+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 )
{
//MESSAGE("SMESH_Group_i; this = "<<this );
}
-SMESH_GroupOnGeom_i::SMESH_GroupOnGeom_i( PortableServer::POA_ptr thePOA, SMESH_Mesh_i* theMeshServant, const int 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 )
{
//MESSAGE("SMESH_GroupOnGeom_i; this = "<<this );
}
+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 )
+{
+ //MESSAGE("SMESH_GroupOnGeom_i; this = "<<this );
+}
+
//=============================================================================
/*!
*
CORBA::Long SMESH_Group_i::AddFrom( SMESH::SMESH_IDSource_ptr theSource )
{
+ TPythonDump pd;
long nbAdd = 0;
SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
if (aGroupDS) {
anIds = mesh->GetElementsByType( GetType() );
else if ( !submesh->_is_nil())
anIds = submesh->GetElementsByType( GetType() );
- else if ( !filter->_is_nil() )
+ else if ( !filter->_is_nil() ) {
+ filter->SetMesh( GetMeshServant()->_this() );
anIds = filter->GetElementType()==GetType() ? theSource->GetIDs() : new SMESH::long_array();
+ }
else
anIds = theSource->GetIDs();
for ( int i = 0, total = anIds->length(); i < total; i++ ) {
}
// Update Python script
- TPythonDump() << "nbAdd = " << _this() << ".AddFrom( " << theSource << " )";
+ pd << "nbAdd = " << _this() << ".AddFrom( " << theSource << " )";
return nbAdd;
}
return aRes._retn();
}
+namespace
+{
+ //================================================================================
+ /*!
+ * \brief return nodes of elements pointered by iterator
+ */
+ //================================================================================
+
+ void getNodesOfElements(SMDS_ElemIteratorPtr elemIt,
+ 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 ( SMESHDS_GroupBase* g = GetGroupDS())
+ {
+ if ( myNbNodes < 0 || g->GetTic() != myGroupDSTic )
+ {
+ 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 ( 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();
+
+ SMESH::long_array_var aRes = new SMESH::long_array();
+ if ( SMESHDS_GroupBase* g = GetGroupDS())
+ {
+ set<const SMDS_MeshNode* > nodes;
+ getNodesOfElements( g->GetElements(), nodes );
+ aRes->length( nodes.size() );
+ 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();
+}
+
//=============================================================================
/*!
*
if (aGroupDS)
{
Quantity_Color aQColor( color.R, color.G, color.B, Quantity_TOC_RGB );
- aGroupDS->SetColor(aQColor);
- TPythonDump()<<_this()<<".SetColor( SALOMEDS.Color( "<<color.R<<", "<<color.G<<", "<<color.B<<" ))";
+ Quantity_Color oldColor = aGroupDS->GetColor();
+ if ( oldColor != aQColor )
+ {
+ aGroupDS->SetColor(aQColor);
+ TPythonDump()<<_this()<<".SetColor( SALOMEDS.Color( "<<color.R<<", "<<color.G<<", "<<color.B<<" ))";
+ }
}
}
SMESH::array_of_ElementType* SMESH_GroupBase_i::GetTypes()
{
SMESH::array_of_ElementType_var types = new SMESH::array_of_ElementType;
- types->length( 1 );
- types[0] = GetType();
+ if ( SMESHDS_GroupBase* ds = GetGroupDS() )
+ if ( !ds->IsEmpty() )
+ {
+ types->length( 1 );
+ types[0] = GetType();
+ }
return types._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)
+{
+ if ( ! myFilter->_is_nil() )
+ myFilter->UnRegister();
+
+ myFilter = SMESH::Filter::_duplicate( theFilter );
+
+ if ( SMESHDS_GroupOnFilter* grDS = dynamic_cast< SMESHDS_GroupOnFilter*>( GetGroupDS() ))
+ grDS->SetPredicate( GetPredicate( myFilter ));
+
+ TPythonDump()<< _this() <<".SetFilter( "<<theFilter<<" )";
+
+ if ( myFilter )
+ {
+ myFilter->Register();
+ SMESH::DownCast< SMESH::Filter_i* >( myFilter )->AddWaiter( this );
+ }
+}
+
+//================================================================================
+/*!
+ * \brief Returns the filter defining group contents
+ */
+//================================================================================
+
+SMESH::Filter_ptr SMESH_GroupOnFilter_i::GetFilter()
+{
+ SMESH::Filter_var f = myFilter;
+ TPythonDump() << f << " = " << _this() << ".GetFilter()";
+ return f._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 )
+ {
+ // write FunctorType as string but not as number to assure correct
+ // persistence if enum FunctorType is modified by insertion in the middle
+ SMESH::Filter::Criterion& crit = criteria[ i ];
+ 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();
+}
+
+SMESH_GroupOnFilter_i::~SMESH_GroupOnFilter_i()
+{
+ if ( ! myFilter->_is_nil() )
+ {
+ SMESH::DownCast< SMESH::Filter_i* >( myFilter )->RemoveWaiter( this );
+ myFilter->UnRegister();
+ }
+}
+
+void SMESH_GroupOnFilter_i::PredicateChanged()
+{
+ if ( SMESHDS_GroupOnFilter* grDS = dynamic_cast< SMESHDS_GroupOnFilter*>( GetGroupDS() ))
+ grDS->SetPredicate( GetPredicate( myFilter ));
+}
