src/HYDRO_tests/test_HYDROData_LandCoverMap.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 <test_HYDROData_LandCoverMap.h>
  20 #include <HYDROData_Document.h>
  21 #include <HYDROData_LandCoverMap.h>
  22 #include <HYDROData_PolylineXY.h>
  23 #include <HYDROData_Tool.h>
  24 #include <TopoDS_Edge.hxx>
  25 #include <TopoDS_Wire.hxx>
  26 #include <TopoDS_Face.hxx>
  27 #include <TColgp_HArray1OfPnt.hxx>
  28 #include <GeomAPI_Interpolate.hxx>
  29 #include <BRepBuilderAPI_MakeEdge.hxx>
  30 #include <BRepBuilderAPI_MakeFace.hxx>
  31 #include <BRepBuilderAPI_MakeWire.hxx>
  32 #include <TestViewer.h>
  33 #include <AIS_DisplayMode.hxx>
  34 #include <QString>
  35 #include <QColor>
  36
  37 TopoDS_Edge Spline( const QList<double>& theXYList, bool isClosed = false )
  38 {
  39   int n = theXYList.size()/2;
  40   Handle(TColgp_HArray1OfPnt) aPointsArray = new TColgp_HArray1OfPnt( 1, n );
  41   for( int i=1; i<=n; i++ )
  42   {
  43     double x = theXYList[2*i-2];
  44     double y = theXYList[2*i-1];
  45     gp_Pnt aPnt( x, y, 0 );
  46     aPointsArray->SetValue( i, aPnt );
  47   }
  48   GeomAPI_Interpolate anInterpolator( aPointsArray, isClosed, 1E-3 );
  49   anInterpolator.Perform();
  50   bool aResult = anInterpolator.IsDone() == Standard_True;
  51   if( aResult )
  52   {
  53     Handle( Geom_BSplineCurve ) aCurve = anInterpolator.Curve();
  54     return BRepBuilderAPI_MakeEdge( aCurve ).Edge();
  55   }
  56   else
  57     return TopoDS_Edge();
  58 }
  59
  60 TopoDS_Face LandCover( const QList<double>& theXYList )
  61 {
  62   TopoDS_Edge anEdge = Spline( theXYList, true );
  63   if( anEdge.IsNull() )
  64     return TopoDS_Face();
  65
  66   TopoDS_Wire aWire = BRepBuilderAPI_MakeWire( anEdge ).Wire();
  67   TopoDS_Face aFace = BRepBuilderAPI_MakeFace( aWire, Standard_True ).Face();
  68   return aFace;
  69 }
  70
  71 void test_HYDROData_LandCoverMap::test_add_2_objects()
  72 {
  73   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  74
  75   Handle(HYDROData_LandCoverMap) aMap =
  76     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
  77
  78   CPPUNIT_ASSERT_EQUAL( KIND_LAND_COVER_MAP, aMap->GetKind() );
  79
  80   TopoDS_Face aLC1 = LandCover( QList<double>() << 10 << 10 << 50 << 20 << 30 << 50 << 15 << 30 );
  81   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC1, "test1" ) );
  82
  83   TopoDS_Face aLC2 = LandCover( QList<double>() << 30 << 20 << 60 << 10 << 70 << 35 << 40 << 40 );
  84   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC2, "test2" ) );
  85
  86   TestViewer::show( aMap->GetShape(), AIS_Shaded, true );
  87   TestViewer::AssertEqual( "LandCoverMap_Add_2_Objects" );
  88
  89   HYDROData_LandCoverMap::Iterator anIt( aMap );
  90   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
  91   CPPUNIT_ASSERT_EQUAL( QString( "test1" ), anIt.StricklerType() );
  92   anIt.Next();
  93   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
  94   CPPUNIT_ASSERT_EQUAL( QString( "test2" ), anIt.StricklerType() );
  95   anIt.Next();
  96   CPPUNIT_ASSERT_EQUAL( false, anIt.More() );
  97
  98   aDoc->Close();
  99 }
 100
 101 void test_HYDROData_LandCoverMap::test_split_by_polyline()
 102 {
 103   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 104
 105   Handle(HYDROData_LandCoverMap) aMap =
 106     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
 107
 108   CPPUNIT_ASSERT_EQUAL( KIND_LAND_COVER_MAP, aMap->GetKind() );
 109
 110   TopoDS_Face aLC = LandCover( QList<double>() << 10 << 10 << 50 << 20 << 30 << 50 << 15 << 30 );
 111   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC, "test1" ) );
 112
 113   Handle(HYDROData_PolylineXY) aPolyline =
 114     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
 115   TopoDS_Wire aWire = BRepBuilderAPI_MakeWire( Spline( QList<double>() << 10 << 40 << 30 << 10 << 40 << 10 << 60 << 10 ) ).Wire();
 116   aPolyline->SetShape( aWire );
 117
 118   CPPUNIT_ASSERT_EQUAL( true, aMap->Split( aPolyline ) );
 119
 120   TestViewer::show( aMap->GetShape(), AIS_Shaded, true );
 121   //TestViewer::show( aWire, QColor(), 0 );
 122   TestViewer::AssertEqual( "LandCoverMap_Split_1" );
 123
 124   HYDROData_LandCoverMap::Iterator anIt( aMap );
 125   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
 126   CPPUNIT_ASSERT_EQUAL( QString( "test1" ), anIt.StricklerType() );
 127   anIt.Next();
 128   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
 129   CPPUNIT_ASSERT_EQUAL( QString( "test1" ), anIt.StricklerType() );
 130   anIt.Next();
 131   CPPUNIT_ASSERT_EQUAL( false, anIt.More() );
 132
 133   aDoc->Close();
 134 }