In AddFrom() preview a case of filter

diff --git a/src/SMESH_I/SMESH_Group_i.cxx b/src/SMESH_I/SMESH_Group_i.cxx
index e600c37f07a57dba8557f4b62299afc8813e8a54..592d172bacbdd4b61c8da769af0630132ab76541 100644 (file)
--- a/src/SMESH_I/SMESH_Group_i.cxx
+++ b/src/SMESH_I/SMESH_Group_i.cxx
@@ -36,6 +36,8 @@
 #include "SMESH_Filter_i.hxx"
 #include "SMESH_PythonDump.hxx"
 
+#include CORBA_SERVER_HEADER(SMESH_Filter)
+
 #include "utilities.h"
 
 using namespace SMESH;
@@ -353,19 +355,20 @@ CORBA::Long SMESH_Group_i::AddFrom( SMESH::SMESH_IDSource_ptr theSource )
   SMESHDS_Group* aGroupDS = dynamic_cast<SMESHDS_Group*>( GetGroupDS() );
   if (aGroupDS) {
     SMESH::long_array_var anIds;
-    if ( !CORBA::is_nil(SMESH::SMESH_GroupBase::_narrow(theSource)) &&
-         SMESH::SMESH_GroupBase::_narrow(theSource)->GetType() == GetType() ) {
+    SMESH::SMESH_GroupBase_var group = SMESH::SMESH_GroupBase::_narrow(theSource);
+    SMESH::SMESH_Mesh_var mesh       = SMESH::SMESH_Mesh::_narrow(theSource);
+    SMESH::SMESH_subMesh_var submesh = SMESH::SMESH_subMesh::_narrow(theSource);
+    SMESH::Filter_var filter         = SMESH::Filter::_narrow(theSource);
+    if ( !group->_is_nil())
+      anIds = group->GetType()==GetType() ? theSource->GetIDs() :  new SMESH::long_array();
+    else if ( !mesh->_is_nil() )
+      anIds = mesh->GetElementsByType( GetType() );
+    else if ( !submesh->_is_nil())
+      anIds = submesh->GetElementsByType( GetType() );
+    else if ( !filter->_is_nil() )
+      anIds = filter->GetElementType()==GetType() ? theSource->GetIDs() : new SMESH::long_array();
+    else 
       anIds = theSource->GetIDs();
-    }
-    else if ( !CORBA::is_nil(SMESH::SMESH_Mesh::_narrow(theSource)) ) {
-      anIds = SMESH::SMESH_Mesh::_narrow(theSource)->GetElementsByType( GetType() );
-    }
-    else if ( !CORBA::is_nil(SMESH::SMESH_subMesh::_narrow(theSource)) ) {
-      anIds = SMESH::SMESH_subMesh::_narrow(theSource)->GetElementsByType( GetType() );
-    }
-    else {
-      anIds->length( 0 );
-    }
     for ( int i = 0, total = anIds->length(); i < total; i++ ) {
       if ( aGroupDS->Add((int)anIds[i]) ) nbAdd++;
     }