src/HYDRO_tests/test_HYDROData_Stream.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_Stream.h>
  20 #include <HYDROGUI_StreamDlg.h>
  21 #include <HYDROData_Document.h>
  22 #include <HYDROData_DTM.h>
  23 #include <HYDROData_Stream.h>
  24 #include <HYDROData_IPolyline.h>
  25 #include <HYDROData_Profile.h>
  26 #include <HYDROData_PolylineXY.h>
  27 #include <HYDROData_Iterator.h>
  28 #include <AIS_InteractiveContext.hxx>
  29 #include <TestViewer.h>
  30 #include <QApplication>
  31 #include <QTest>
  32
  33 extern QString REF_DATA_PATH;
  34 NCollection_Sequence<HYDROData_IPolyline::Point> points2;
  35 const double EPS = 1E-3;
  36
  37 void test_HYDROData_Stream::setUp()
  38 {
  39   points2.Clear();
  40   points2.Append( gp_XY( 0.0, 5.0 ) );
  41   points2.Append( gp_XY( 1.0, 1.0 ) );
  42   points2.Append( gp_XY( 1.5, 0.0 ) );
  43   points2.Append( gp_XY( 4.0, 4.0 ) );
  44 }
  45
  46 void test_HYDROData_Stream::tearDown()
  47 {
  48 }
  49
  50 void test_HYDROData_Stream::test_dialog()
  51 {
  52   TestViewer::eraseAll( true, true );
  53
  54   HYDROGUI_StreamDlg* aDlg = new HYDROGUI_StreamDlg( 0, "stream" );
  55   aDlg->show();
  56   qApp->processEvents();
  57
  58   QImage aStreamDlgImage = QPixmap::grabWidget( aDlg ).toImage();
  59   CPPUNIT_ASSERT_IMAGES2( &aStreamDlgImage, "StreamDlg" );
  60
  61   aDlg->setDDZ( 12.34 );
  62   CPPUNIT_ASSERT_DOUBLES_EQUAL( 12.34, aDlg->getDDZ(), EPS );
  63
  64   aDlg->setSpatialStep( 56.78 );
  65   CPPUNIT_ASSERT_DOUBLES_EQUAL( 56.78, aDlg->getSpatialStep(), EPS );
  66
  67   delete aDlg;
  68 }
  69
  70 void test_HYDROData_Stream::test_alt_object()
  71 {
  72   TestViewer::eraseAll( true, true );
  73
  74   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  75
  76   Handle(HYDROData_Stream) aStream =
  77     Handle(HYDROData_Stream)::DownCast( aDoc->CreateObject( KIND_STREAM ) );
  78
  79   CPPUNIT_ASSERT_EQUAL( false, (bool)aStream.IsNull() );
  80   CPPUNIT_ASSERT_EQUAL( true, (bool)aStream->GetAltitudeObject().IsNull() );
  81   CPPUNIT_ASSERT_EQUAL( false, (bool)aStream->DTM().IsNull() );
  82   CPPUNIT_ASSERT_EQUAL( false, (bool)aStream->GetAltitudeObject().IsNull() );
  83   CPPUNIT_ASSERT_EQUAL( KIND_DTM, aStream->getAltitudeObjectType() );
  84
  85   Handle(HYDROData_DTM) aDTM =
  86     Handle(HYDROData_DTM)::DownCast( aStream->GetAltitudeObject() );
  87   CPPUNIT_ASSERT_EQUAL( false, (bool)aDTM.IsNull() );
  88
  89   aDoc->Close();
  90 }
  91
  92 void test_HYDROData_Stream::test_params_sync()
  93 {
  94   TestViewer::eraseAll( true, true );
  95
  96   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  97
  98   Handle(HYDROData_Stream) aStream =
  99     Handle(HYDROData_Stream)::DownCast( aDoc->CreateObject( KIND_STREAM ) );
 100   Handle(HYDROData_DTM) aDTM = aStream->DTM();
 101   CPPUNIT_ASSERT_EQUAL( false, (bool)aDTM.IsNull() );
 102
 103   Handle(HYDROData_Profile) aProfile1 =
 104     Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) );
 105
 106   Handle(HYDROData_Profile) aProfile2 =
 107     Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) );
 108
 109   aProfile1->SetParametricPoints( points2 );
 110   aProfile1->GetProfileUZ()->SetSectionType( 0, HYDROData_IPolyline::SECTION_POLYLINE );
 111   aProfile1->SetLeftPoint( gp_XY( 10, 10 ) );
 112   aProfile1->SetRightPoint( gp_XY( 20, 0 ) );
 113
 114   aProfile2->SetParametricPoints( points2 );
 115   aProfile2->GetProfileUZ()->SetSectionType( 0, HYDROData_IPolyline::SECTION_POLYLINE );
 116   aProfile2->SetLeftPoint( gp_XY( 50, 0 ) );
 117   aProfile2->SetRightPoint( gp_XY( 60, 10 ) );
 118
 119   HYDROData_SequenceOfObjects profiles;
 120   profiles.Append( aProfile1 );
 121   profiles.Append( aProfile2 );
 122
 123   aStream->SetProfiles( profiles, false );
 124   aStream->SetDDZ( 3.14 );
 125   aStream->SetSpatialStep( 4.14 );
 126
 127   CPPUNIT_ASSERT_DOUBLES_EQUAL( 3.14, aStream->GetDDZ(), EPS );
 128   CPPUNIT_ASSERT_DOUBLES_EQUAL( 3.14, aDTM->GetDDZ(), EPS );
 129   CPPUNIT_ASSERT_DOUBLES_EQUAL( 4.14, aStream->GetSpatialStep(), EPS );
 130   CPPUNIT_ASSERT_DOUBLES_EQUAL( 4.14, aDTM->GetSpatialStep(), EPS );
 131
 132   HYDROData_SequenceOfObjects profiles1 = aStream->GetProfiles();
 133   CPPUNIT_ASSERT_EQUAL( 2, profiles1.Size() );
 134   CPPUNIT_ASSERT( profiles.Value(1)->Label() == profiles1.Value(1)->Label() );
 135   CPPUNIT_ASSERT( profiles.Value(2)->Label() == profiles1.Value(2)->Label() );
 136
 137   HYDROData_SequenceOfObjects profiles2 = aDTM->GetProfiles();
 138   CPPUNIT_ASSERT_EQUAL( 2, profiles2.Size() );
 139   CPPUNIT_ASSERT( profiles.Value(1)->Label() == profiles2.Value(1)->Label() );
 140   CPPUNIT_ASSERT( profiles.Value(2)->Label() == profiles2.Value(2)->Label() );
 141
 142   aDoc->Close();
 143 }
 144
 145 void test_HYDROData_Stream::test_dump()
 146 {
 147   TestViewer::eraseAll( true, true );
 148
 149   // Case 1. Without hydraulic axis
 150   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 151
 152   Handle(HYDROData_Stream) aStream1 =
 153     Handle(HYDROData_Stream)::DownCast( aDoc->CreateObject( KIND_STREAM ) );
 154   aStream1->SetName( "stream_1" );
 155
 156   Handle(HYDROData_Profile) aProfile1 =
 157     Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) );
 158   aProfile1->SetName( "p1" );
 159
 160   Handle(HYDROData_Profile) aProfile2 =
 161     Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) );
 162   aProfile2->SetName( "p2" );
 163
 164   aProfile1->SetParametricPoints( points2 );
 165   aProfile1->GetProfileUZ()->SetSectionType( 0, HYDROData_IPolyline::SECTION_POLYLINE );
 166   aProfile1->SetLeftPoint( gp_XY( 10, 10 ) );
 167   aProfile1->SetRightPoint( gp_XY( 20, 0 ) );
 168
 169   aProfile2->SetParametricPoints( points2 );
 170   aProfile2->GetProfileUZ()->SetSectionType( 0, HYDROData_IPolyline::SECTION_POLYLINE );
 171   aProfile2->SetLeftPoint( gp_XY( 50, 0 ) );
 172   aProfile2->SetRightPoint( gp_XY( 60, 10 ) );
 173
 174   HYDROData_SequenceOfObjects profiles;
 175   profiles.Append( aProfile1 );
 176   profiles.Append( aProfile2 );
 177
 178   aStream1->SetProfiles( profiles, false );
 179   aStream1->SetDDZ( 0.2 );
 180   aStream1->SetSpatialStep( 3.0 );
 181
 182   MapOfTreatedObjects objs;
 183   objs["p1"] = aProfile1;
 184   objs["p2"] = aProfile2;
 185
 186   QStringList aScript1 = aStream1->DumpToPython( "", objs );
 187   CPPUNIT_ASSERT_EQUAL( 10, aScript1.size() );
 188   CPPUNIT_ASSERT_EQUAL( std::string( "stream_1 = hydro_doc.CreateObject( KIND_STREAM )" ), aScript1[0].toStdString() );
 189   CPPUNIT_ASSERT_EQUAL( std::string( "stream_1.SetName( \"stream_1\" )" ),                 aScript1[1].toStdString() );
 190   CPPUNIT_ASSERT_EQUAL( std::string( "" ),                                                 aScript1[2].toStdString() );
 191   CPPUNIT_ASSERT_EQUAL( std::string( "stream_1.AddProfile( p1 )" ),                        aScript1[3].toStdString() );
 192   CPPUNIT_ASSERT_EQUAL( std::string( "stream_1.AddProfile( p2 )" ),                        aScript1[4].toStdString() );
 193   CPPUNIT_ASSERT_EQUAL( std::string( "stream_1.SetDDZ( 0.200 )" ),                         aScript1[5].toStdString() );
 194   CPPUNIT_ASSERT_EQUAL( std::string( "stream_1.SetSpatialStep( 3.000 )" ),                 aScript1[6].toStdString() );
 195   CPPUNIT_ASSERT_EQUAL( std::string( "" ),                                                 aScript1[7].toStdString() );
 196   CPPUNIT_ASSERT_EQUAL( std::string( "stream_1.Update()" ),                                aScript1[8].toStdString() );
 197   CPPUNIT_ASSERT_EQUAL( std::string( "" ),                                                 aScript1[9].toStdString() );
 198
 199   // Case 2. With hydraulic axis
 200
 201   Handle(HYDROData_Stream) aStream2 =
 202     Handle(HYDROData_Stream)::DownCast( aDoc->CreateObject( KIND_STREAM ) );
 203   aStream2->SetName( "stream_2" );
 204
 205   Handle(HYDROData_PolylineXY) anHAxis =
 206     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
 207   anHAxis->SetName( "axis" );
 208
 209   aStream2->SetProfiles( profiles, false );
 210   aStream2->SetDDZ( 0.2 );
 211   aStream2->SetSpatialStep( 3.0 );
 212   aStream2->SetReferenceObject( anHAxis, HYDROData_Stream::DataTag_HydraulicAxis );
 213
 214   objs.clear();
 215   objs["p1"] = aProfile1;
 216   objs["p2"] = aProfile2;
 217   objs["axis"] = anHAxis;
 218
 219   QStringList aScript2 = aStream2->DumpToPython( "", objs );
 220   CPPUNIT_ASSERT_EQUAL( 11, aScript2.size() );
 221   CPPUNIT_ASSERT_EQUAL( std::string( "stream_2 = hydro_doc.CreateObject( KIND_STREAM )" ), aScript2[0].toStdString() );
 222   CPPUNIT_ASSERT_EQUAL( std::string( "stream_2.SetName( \"stream_2\" )" ),                 aScript2[1].toStdString() );
 223   CPPUNIT_ASSERT_EQUAL( std::string( "" ),                                                 aScript2[2].toStdString() );
 224   CPPUNIT_ASSERT_EQUAL( std::string( "stream_2.SetHydraulicAxis( axis )" ),                aScript2[3].toStdString() );
 225   CPPUNIT_ASSERT_EQUAL( std::string( "stream_2.AddProfile( p1 )" ),                        aScript2[4].toStdString() );
 226   CPPUNIT_ASSERT_EQUAL( std::string( "stream_2.AddProfile( p2 )" ),                        aScript2[5].toStdString() );
 227   CPPUNIT_ASSERT_EQUAL( std::string( "stream_2.SetDDZ( 0.200 )" ),                         aScript2[6].toStdString() );
 228   CPPUNIT_ASSERT_EQUAL( std::string( "stream_2.SetSpatialStep( 3.000 )" ),                 aScript2[7].toStdString() );
 229   CPPUNIT_ASSERT_EQUAL( std::string( "" ),                                                 aScript2[8].toStdString() );
 230   CPPUNIT_ASSERT_EQUAL( std::string( "stream_2.Update()" ),                                aScript2[9].toStdString() );
 231   CPPUNIT_ASSERT_EQUAL( std::string( "" ),                                                 aScript2[10].toStdString() );
 232
 233   aDoc->Close();
 234 }
 235
 236 void test_HYDROData_Stream::test_presentation()
 237 {
 238   TestViewer::eraseAll( true, true );
 239
 240   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 241
 242   TCollection_AsciiString fname = REF_DATA_PATH.toLatin1().data();
 243   fname += "/Profiles.xyz";
 244   NCollection_Sequence<int> bad_ids;
 245
 246   int aSize = HYDROData_Profile::ImportFromFile( aDoc, fname, bad_ids, true );
 247
 248   CPPUNIT_ASSERT_EQUAL( 0, bad_ids.Size() );
 249   CPPUNIT_ASSERT_EQUAL( 46, aSize );
 250
 251   HYDROData_SequenceOfObjects profiles;
 252   HYDROData_Iterator it( aDoc, KIND_PROFILE );
 253   for( int i=0; it.More(); it.Next(), i++ )
 254   {
 255     if( i>=25 && i<=35 )
 256     {
 257       it.Current()->Update();
 258       profiles.Append( Handle(HYDROData_Profile)::DownCast( it.Current() ) );
 259     }
 260   }
 261
 262   Handle(HYDROData_Stream) aStream =
 263     Handle(HYDROData_Stream)::DownCast( aDoc->CreateObject( KIND_STREAM ) );
 264
 265   aStream->SetProfiles( profiles, false );
 266   aStream->SetDDZ( 0.2 );
 267   aStream->SetSpatialStep( 10 );
 268   aStream->Update();
 269
 270   TopoDS_Shape aPrs3d = aStream->GetShape3D();
 271   TopoDS_Shape aPrs2d = aStream->GetTopShape();
 272
 273   TestViewer::show( aPrs2d, 0, true, "stream_dtm_2d", 1, 1 );
 274   CPPUNIT_ASSERT_IMAGES;
 275
 276   TestViewer::eraseAll( true );
 277   TestViewer::show( aPrs3d, 0, true, "stream_dtm_3d" );
 278   CPPUNIT_ASSERT_IMAGES
 279
 280   aDoc->Close();
 281 }
 282