Salome HOME
Merge branch 'master' of https://git.salome-platform.org/git/modules/hydro
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_LandCoverArgsFilter.cxx
1 // Copyright (C) 2014-2015  EDF-R&D
2 // This library is free software; you can redistribute it and/or
3 // modify it under the terms of the GNU Lesser General Public
4 // License as published by the Free Software Foundation; either
5 // version 2.1 of the License, or (at your option) any later version.
6 //
7 // This library is distributed in the hope that it will be useful,
8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
10 // Lesser General Public License for more details.
11 //
12 // You should have received a copy of the GNU Lesser General Public
13 // License along with this library; if not, write to the Free Software
14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
15 //
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
17 //
18
19 #include <HYDROGUI_LandCoverArgsFilter.h>
20 #include <HYDROGUI_Operations.h>
21 #include <HYDROData_ArtificialObject.h>
22 #include <HYDROData_NaturalObject.h>
23 #include <HYDROData_PolylineXY.h>
24
25 HYDROGUI_LandCoverArgsFilter::HYDROGUI_LandCoverArgsFilter( int theOperationId )
26   : myOperationId( theOperationId )
27 {
28 }
29
30 HYDROGUI_LandCoverArgsFilter::~HYDROGUI_LandCoverArgsFilter()
31 {
32 }
33
34 bool HYDROGUI_LandCoverArgsFilter::isOk( const Handle(HYDROData_Entity)& theEntity ) const
35 {
36   if( theEntity.IsNull() )
37     return false;
38
39   ObjectKind aKind = theEntity->GetKind();
40   if( aKind==KIND_POLYLINEXY )
41   {
42     Handle(HYDROData_PolylineXY) aPolylineXY = Handle(HYDROData_PolylineXY)::DownCast( theEntity );
43     switch( myOperationId )
44     {
45     case CreateLandCoverMapId:
46     case AddLandCoverId:
47       return aPolylineXY->IsClosed();  // only closed polylines are allowed
48     case SplitLandCoverId:
49       return true;  // any polylines are allowed
50     default:
51       return false;
52     }
53   }
54
55   Handle(HYDROData_NaturalObject) aNaturalObject = Handle(HYDROData_NaturalObject)::DownCast( theEntity );
56   Handle(HYDROData_ArtificialObject) anArtificialObject = Handle(HYDROData_ArtificialObject)::DownCast( theEntity );
57   bool isOK = !aNaturalObject.IsNull() || !anArtificialObject.IsNull();
58   return isOK;
59 }