Copyright update 2022

[modules/geom.git] / src / GEOMFiltersSelection / GEOM_TypeFilter.cxx

diff --git a/src/GEOMFiltersSelection/GEOM_TypeFilter.cxx b/src/GEOMFiltersSelection/GEOM_TypeFilter.cxx
index d5785f513f39b15cec38170da4ab8317da6e3666..c61751495d6b27aad15eef6d8b8482cb6468ffef 100644 (file)
--- a/src/GEOMFiltersSelection/GEOM_TypeFilter.cxx
+++ b/src/GEOMFiltersSelection/GEOM_TypeFilter.cxx
@@ -1,12 +1,35 @@
+// Copyright (C) 2007-2022  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
+//
+
 #include "GEOM_TypeFilter.h"
 
 //=======================================================================
 // function : GEOM_TypeFilter
 // purpose  : 
 //=======================================================================
-GEOM_TypeFilter::GEOM_TypeFilter( SalomeApp_Study* study, const int type )
+GEOM_TypeFilter::GEOM_TypeFilter( SalomeApp_Study* study, const int type, const bool isShapeType )
 :GEOM_SelectionFilter( study ),
-myType( type )
+ myType( type ),
+ myIsShapeType(isShapeType)
 {
 }
 
@@ -26,7 +49,12 @@ bool GEOM_TypeFilter::isOk( const SUIT_DataOwner* sOwner ) const
 {
   GEOM::GEOM_Object_var obj = getObject( sOwner );
   if ( !CORBA::is_nil( obj ) )
-    return obj->GetType() == type();
+    {
+      if (!myIsShapeType)
+        return obj->GetType() == type();
+      else
+        return obj->GetShapeType() == type();
+    }
   
   return false;
 }