src/GEOMFiltersSelection/GEOM_EdgeFilter.cxx

   1 // Copyright (C) 2005  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   2 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   3 //
   4 // This library is free software; you can redistribute it and/or
   5 // modify it under the terms of the GNU Lesser General Public
   6 // License as published by the Free Software Foundation; either
   7 // version 2.1 of the License.
   8 //
   9 // This library is distributed in the hope that it will be useful
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 // Lesser General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU Lesser General Public
  15 // License along with this library; if not, write to the Free Software
  16 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17 //
  18 // See http://www.salome-platform.org/
  19 //
  20 #include "GEOM_EdgeFilter.h"
  21
  22 #include <BRepAdaptor_Curve.hxx>
  23 #include <TopoDS_Edge.hxx>
  24 #include <TopoDS.hxx>
  25 #include <StdSelect_TypeOfEdge.hxx>
  26
  27 //=======================================================================
  28 // function : GEOM_EdgeFilter
  29 // purpose  :
  30 //=======================================================================
  31 GEOM_EdgeFilter::GEOM_EdgeFilter( SalomeApp_Study* study, const int kind )
  32 : GEOM_SelectionFilter( study ),
  33 myKind( kind )
  34 {
  35   add( TopAbs_EDGE );
  36 }
  37
  38 //=======================================================================
  39 // function : ~GEOM_SelectionFilter
  40 // purpose  :
  41 //=======================================================================
  42 GEOM_EdgeFilter::~GEOM_EdgeFilter()
  43 {
  44 }
  45
  46 //=======================================================================
  47 // function : isShapeOk
  48 // purpose  :
  49 //=======================================================================
  50 bool GEOM_EdgeFilter::isShapeOk( const TopoDS_Shape& theShape ) const
  51 {
  52   if ( !theShape.IsNull() && theShape.ShapeType() == TopAbs_EDGE )
  53   {
  54     BRepAdaptor_Curve aCurve( TopoDS::Edge( theShape ) );
  55     GeomAbs_CurveType aType = aCurve.GetType();
  56
  57     switch ( myKind )
  58     {
  59     case StdSelect_AnyEdge: return Standard_True;
  60     case StdSelect_Line:    return ( aType == GeomAbs_Line );
  61     case StdSelect_Circle:  return ( aType == GeomAbs_Circle );
  62     }
  63   }
  64   return false;
  65 }
  66