X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDRO_tests%2Ftest_HYDROData_DTM.cxx;h=453a315f6430750f93eea1a8a6aca68aca6b4b3d;hb=3bece499e83cc9ca95536228c7f6740030e9d5d8;hp=96e779c7046d65a64c2bf3fe8328cc96f942f822;hpb=0ef7a1c0faeafd936f486c117211adf59aa1e1de;p=modules%2Fhydro.git diff --git a/src/HYDRO_tests/test_HYDROData_DTM.cxx b/src/HYDRO_tests/test_HYDROData_DTM.cxx index 96e779c7..453a315f 100644 --- a/src/HYDRO_tests/test_HYDROData_DTM.cxx +++ b/src/HYDRO_tests/test_HYDROData_DTM.cxx @@ -20,6 +20,8 @@ #include #include #include +#include +#include #include #include #include @@ -29,13 +31,124 @@ #include #include #include -#include +#include +#include +#include +#include +#include +#include #include +#include +#include const double EPS = 1E-3; +extern QString REF_DATA_PATH; NCollection_Sequence points; +class DTM_item : public QGraphicsItem +{ +public: + DTM_item( HYDROGUI_ShapeBathymetry* theDTM, double d ) + { + Handle(AIS_PointCloud) pc = Handle(AIS_PointCloud)::DownCast( theDTM->getAISObjects()[0] ); + Handle(Graphic3d_ArrayOfPoints) pp = pc->GetPoints(); + myBB = QRectF(); + double xmin, xmax, ymin, ymax; + myD = d; + + int n = pp->VertexNumber(); + + for( int i=1; i<=n; i++ ) + { + gp_Pnt pnt = pp->Vertice( i ); + Quantity_Color col = pp->VertexColor( i ); + + int r = col.Red()*255; + int g = col.Green()*255; + int b = col.Blue()*255; + int val = ( r << 16 ) + ( g << 8 ) + b; + double x = pnt.X(); + double y = -pnt.Y(); + QPointF aPnt( x, y ); + myPoints[val].append( aPnt ); + if( i==1 ) + { + xmin = x; + xmax = x; + ymin = y; + ymax = y; + } + else + { + if( x>xmax ) + xmax = x; + if( xymax ) + ymax = y; + if( y >::const_iterator it = myPoints.begin(), last = myPoints.end(); + for( ; it!=last; it++ ) + { + int r = ( it.key() >> 16 ) % 256; + int g = ( it.key() >> 8 ) % 256; + int b = ( it.key() >> 0 ) % 256; + QColor aColor( r, g, b ); + QBrush aBrush( aColor ); + foreach( QPointF pnt, it.value() ) + { + QRectF r( pnt.x()-myD/2, pnt.y()-myD/2, myD, myD ); + p->fillRect( r, aBrush ); + } + } + } + +private: + QRectF myBB; + double myD; + QMap > myPoints; +}; + +QImage draw_DTM( HYDROGUI_ShapeBathymetry* theDTM, double theD, int theWidth, int theHeight ) +{ + QGraphicsScene aScene; + QGraphicsView aView; + DTM_item anItem( theDTM, theD ); + + aView.setScene( &aScene ); + aView.setHorizontalScrollBarPolicy( Qt::ScrollBarAlwaysOff ); + aView.setVerticalScrollBarPolicy( Qt::ScrollBarAlwaysOff ); + aScene.addItem( &anItem ); + + aView.resize( theWidth, theHeight ); + QRectF bb = anItem.boundingRect(); + aView.fitInView( bb, Qt::KeepAspectRatio ); + QApplication::processEvents(); + + QPixmap aPixmap = QPixmap::grabWidget( &aView ); + return aPixmap.toImage(); +} + + void test_HYDROData_DTM::setUp() { points.Clear(); @@ -47,27 +160,27 @@ void test_HYDROData_DTM::setUp() void test_HYDROData_DTM::test_creation() { - Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1); + Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(); - Handle(HYDROData_DTM) DTM = + Handle(HYDROData_DTM) DTM = Handle(HYDROData_DTM)::DownCast( aDoc->CreateObject( KIND_DTM ) ); CPPUNIT_ASSERT_EQUAL( false, (bool)DTM.IsNull() ); - + aDoc->Close(); } void test_HYDROData_DTM::test_hydraulic_axis() { - Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1); + Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(); - Handle(HYDROData_Profile) aProfile1 = + Handle(HYDROData_Profile) aProfile1 = Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); - Handle(HYDROData_Profile) aProfile2 = + Handle(HYDROData_Profile) aProfile2 = Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); - Handle(HYDROData_Profile) aProfile3 = + Handle(HYDROData_Profile) aProfile3 = Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); aProfile1->SetParametricPoints( points ); @@ -91,25 +204,33 @@ void test_HYDROData_DTM::test_hydraulic_axis() profiles.push_back( aProfile2 ); profiles.push_back( aProfile3 ); - Handle_Geom2d_BSplineCurve HA = HYDROData_DTM::CreateHydraulicAxis( profiles, distances ); + Handle(Geom2d_BSplineCurve) HA = HYDROData_DTM::CreateHydraulicAxis( profiles, distances ); CPPUNIT_ASSERT_EQUAL( false, (bool)HA.IsNull() ); CPPUNIT_ASSERT_EQUAL( 3, (int)distances.size() ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.0, distances[0], EPS ); - CPPUNIT_ASSERT_DOUBLES_EQUAL( 43.499, distances[1], EPS ); - CPPUNIT_ASSERT_DOUBLES_EQUAL( 211.474, distances[2], EPS ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 44.859, distances[1], EPS ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 207.661, distances[2], EPS ); + + gp_Pnt2d p; + gp_Vec2d q; + HA->D1( 0, p, q ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 13.75, p.X(), EPS ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 6.25, p.Y(), EPS ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.532, q.X(), EPS ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.532, q.Y(), EPS ); aDoc->Close(); } void test_HYDROData_DTM::test_profile_conversion_to_2d() { - Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1); + Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(); - Handle(HYDROData_Profile) aProfile1 = + Handle(HYDROData_Profile) aProfile1 = Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); - Handle(HYDROData_Profile) aProfile2 = + Handle(HYDROData_Profile) aProfile2 = Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); aProfile1->SetParametricPoints( points ); @@ -122,12 +243,13 @@ void test_HYDROData_DTM::test_profile_conversion_to_2d() aProfile2->SetLeftPoint( gp_XY( 10, 10 ) ); aProfile2->SetRightPoint( gp_XY( 20, 20 ) ); - double aUMin1 = std::numeric_limits::max(), + double aUMin1 = DBL_MAX, aUMax1 = -aUMin1, aUMin2 = aUMin1, aUMax2 = aUMax1; - std::vector curves1 = HYDROData_DTM::ProfileToParametric( aProfile1, aUMin1, aUMax1 ); - std::vector curves2 = HYDROData_DTM::ProfileToParametric( aProfile2, aUMin2, aUMax2 ); + gp_Vec2d aProfileDir; + std::vector curves1 = HYDROData_DTM::ProfileToParametric( aProfile1, aUMin1, aUMax1, aProfileDir ); + std::vector curves2 = HYDROData_DTM::ProfileToParametric( aProfile2, aUMin2, aUMax2, aProfileDir ); gp_Pnt2d aFirst, aLast; CPPUNIT_ASSERT_EQUAL( 3, (int)curves1.size() ); @@ -174,9 +296,9 @@ void test_HYDROData_DTM::test_profile_conversion_to_2d() void test_HYDROData_DTM::test_profile_properties() { - Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1); + Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(); - Handle(HYDROData_Profile) aProfile = + Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); aProfile->SetParametricPoints( points ); @@ -187,7 +309,7 @@ void test_HYDROData_DTM::test_profile_properties() gp_Pnt lp; gp_Vec2d dd; double zmin, zmax; - HYDROData_DTM::GetProperties( aProfile, lp, dd, false, zmin, zmax ); + HYDROData_DTM::GetProperties( aProfile, lp, dd, zmin, zmax ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 13.75, lp.X(), EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 15.625, lp.Y(), EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 0, lp.Z(), EPS ); @@ -196,23 +318,23 @@ void test_HYDROData_DTM::test_profile_properties() CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.0, zmin, EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 5.0, zmax, EPS ); - HYDROData_DTM::GetProperties( aProfile, lp, dd, true, zmin, zmax ); + /* HYDROData_DTM::GetProperties( aProfile, lp, dd, true, zmin, zmax ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 13.75, lp.X(), EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 15.625, lp.Y(), EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 0, lp.Z(), EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( -15, dd.X(), EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 10, dd.Y(), EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.0, zmin, EPS ); - CPPUNIT_ASSERT_DOUBLES_EQUAL( 5.0, zmax, EPS ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 5.0, zmax, EPS );*/ aDoc->Close(); } void test_HYDROData_DTM::test_profile_discretization_polyline() { - Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1); + Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(); - Handle(HYDROData_Profile) aProfile = + Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); aProfile->SetParametricPoints( points ); @@ -220,8 +342,10 @@ void test_HYDROData_DTM::test_profile_discretization_polyline() aProfile->SetLeftPoint( gp_XY( 10, 10 ) ); aProfile->SetRightPoint( gp_XY( 20, 20 ) ); - HYDROData_DTM::CurveUZ aMid( 0.0 ), aWid( 0.0 ); - HYDROData_DTM::ProfileDiscretization( aProfile, 0.0, 0.0, 5.0, 0.5, aMid, aWid ); + HYDROData_DTM::CurveUZ aMid( 0.0, gp_Vec2d(), 0, 0 ), aWid( 0.0, gp_Vec2d(), 0, 0 ); + int dummy = 0; + QSet warnings; + HYDROData_DTM::ProfileDiscretization( aProfile, 0.0, 0.0, 5.0, 5.0, 0.5, aMid, aWid, dummy, 1E-6, warnings ); CPPUNIT_ASSERT_EQUAL( 11, (int)aMid.size() ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.0, aMid[0].U, EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.0, aMid[0].Z, EPS ); @@ -229,7 +353,7 @@ void test_HYDROData_DTM::test_profile_discretization_polyline() CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.5, aMid[1].Z, EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 1.215, aMid[5].U, EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 2.5, aMid[5].Z, EPS ); - CPPUNIT_ASSERT_DOUBLES_EQUAL( -0.589, aMid[10].U, EPS ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 1.768, aMid[10].U, EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 5.0, aMid[10].Z, EPS ); CPPUNIT_ASSERT_EQUAL( 11, (int)aWid.size() ); @@ -247,9 +371,9 @@ void test_HYDROData_DTM::test_profile_discretization_polyline() void test_HYDROData_DTM::test_profile_discretization_spline() { -Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1); +Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(); - Handle(HYDROData_Profile) aProfile = + Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); aProfile->SetParametricPoints( points ); @@ -257,8 +381,10 @@ Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1); aProfile->SetLeftPoint( gp_XY( 10, 10 ) ); aProfile->SetRightPoint( gp_XY( 20, 20 ) ); - HYDROData_DTM::CurveUZ aMid( 0.0 ), aWid( 0.0 ); - HYDROData_DTM::ProfileDiscretization( aProfile, 0.0, 0.0, 5.0, 0.5, aMid, aWid ); + HYDROData_DTM::CurveUZ aMid( 0.0, gp_Vec2d(), 0, 0 ), aWid( 0.0, gp_Vec2d(), 0, 0 ); + int dummy = 0 ; + QSet warnings; + HYDROData_DTM::ProfileDiscretization( aProfile, 0.0, 0.0, 5.0, 5.0, 0.5, aMid, aWid, dummy, 1E-6, warnings ); CPPUNIT_ASSERT_EQUAL( 11, (int)aMid.size() ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.242, aMid[0].U, EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.0, aMid[0].Z, EPS ); @@ -266,7 +392,7 @@ Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1); CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.5, aMid[1].Z, EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 1.473, aMid[5].U, EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 2.5, aMid[5].Z, EPS ); - CPPUNIT_ASSERT_DOUBLES_EQUAL( -0.589, aMid[10].U, EPS ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 1.768, aMid[10].U, EPS ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 5.0, aMid[10].Z, EPS ); CPPUNIT_ASSERT_EQUAL( 11, (int)aWid.size() ); @@ -311,7 +437,7 @@ void operator << ( std::ostream& s, const HYDROData_DTM::CurveUZ& c ) void test_HYDROData_DTM::test_curves_interpolation() { - HYDROData_DTM::CurveUZ A(1.0), B(2.0); + HYDROData_DTM::CurveUZ A(1.0, gp_Vec2d(), 0, 0), B(2.0, gp_Vec2d(), 0, 0); A.push_back( HYDROData_DTM::PointUZ( 0, 0 ) ); A.push_back( HYDROData_DTM::PointUZ( 1, 1 ) ); A.push_back( HYDROData_DTM::PointUZ( 2, 2 ) ); @@ -353,12 +479,12 @@ void test_HYDROData_DTM::test_curves_interpolation() void test_HYDROData_DTM::test_curve_to_3d() { - Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1); + Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(); - Handle(HYDROData_Profile) aProfile1 = + Handle(HYDROData_Profile) aProfile1 = Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); - Handle(HYDROData_Profile) aProfile2 = + Handle(HYDROData_Profile) aProfile2 = Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); aProfile1->SetParametricPoints( points ); @@ -376,46 +502,46 @@ void test_HYDROData_DTM::test_curve_to_3d() profiles.push_back( aProfile1 ); profiles.push_back( aProfile2 ); - Handle_Geom2d_BSplineCurve HA = HYDROData_DTM::CreateHydraulicAxis( profiles, distances ); + Handle(Geom2d_BSplineCurve) HA = HYDROData_DTM::CreateHydraulicAxis( profiles, distances ); HYDROData_DTM::AltitudePoints points; - HYDROData_DTM::CurveUZ mid( 5.0 ); + HYDROData_DTM::CurveUZ mid( 5.0, gp_Vec2d(-10,10), 0, 10 ); mid.push_back( HYDROData_DTM::PointUZ( 0, 5 ) ); mid.push_back( HYDROData_DTM::PointUZ( 1, 6 ) ); - HYDROData_DTM::CurveUZ wid( 5.0 ); + HYDROData_DTM::CurveUZ wid( 5.0, gp_Vec2d(-10,10), 0, 10 ); wid.push_back( HYDROData_DTM::PointUZ( 2, 5 ) ); wid.push_back( HYDROData_DTM::PointUZ( 6, 6 ) ); HYDROData_DTM::CurveTo3D( HA, mid, wid, points ); CPPUNIT_ASSERT_EQUAL( 4, (int)points.size() ); - CPPUNIT_ASSERT_EQUAL( HYDROData_DTM::AltitudePoint( 15.434, -0.598, 5.0 ), points[0] ); - CPPUNIT_ASSERT_EQUAL( HYDROData_DTM::AltitudePoint( 14.497, -0.947, 5.0 ), points[1] ); - CPPUNIT_ASSERT_EQUAL( HYDROData_DTM::AltitudePoint( 15.903, -0.423, 6.0 ), points[2] ); - CPPUNIT_ASSERT_EQUAL( HYDROData_DTM::AltitudePoint( 13.092, -1.471, 6.0 ), points[3] ); + CPPUNIT_ASSERT_EQUAL( HYDROData_DTM::AltitudePoint( 21.588, 5.419, 6.0 ), points[0] ); + CPPUNIT_ASSERT_EQUAL( HYDROData_DTM::AltitudePoint( 20.881, 6.126, 5.0 ), points[1] ); + CPPUNIT_ASSERT_EQUAL( HYDROData_DTM::AltitudePoint( 19.466, 7.541, 5.0 ), points[2] ); + CPPUNIT_ASSERT_EQUAL( HYDROData_DTM::AltitudePoint( 17.345, 9.662, 6.0 ), points[3] ); aDoc->Close(); } void test_HYDROData_DTM::test_presentation() { - Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1); + Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(); Handle(HYDROData_DTM) DTM = Handle(HYDROData_DTM)::DownCast( aDoc->CreateObject( KIND_DTM ) ); - Handle(HYDROData_Profile) aProfile1 = + Handle(HYDROData_Profile) aProfile1 = Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); - Handle(HYDROData_Profile) aProfile2 = + Handle(HYDROData_Profile) aProfile2 = Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); aProfile1->SetParametricPoints( points ); aProfile1->GetProfileUZ()->SetSectionType( 0, HYDROData_IPolyline::SECTION_SPLINE ); - aProfile1->SetLeftPoint( gp_XY( 20, 0 ) ); - aProfile1->SetRightPoint( gp_XY( 10, 10 ) ); + aProfile1->SetLeftPoint( gp_XY( 10, 10 ) ); + aProfile1->SetRightPoint( gp_XY( 20, 0 ) ); aProfile2->SetParametricPoints( points ); aProfile2->GetProfileUZ()->SetSectionType( 0, HYDROData_IPolyline::SECTION_SPLINE ); - aProfile2->SetLeftPoint( gp_XY( 100, 0 ) ); - aProfile2->SetRightPoint( gp_XY( 110, 0 ) ); + aProfile2->SetLeftPoint( gp_XY( 110, 10 ) ); + aProfile2->SetRightPoint( gp_XY( 100, 0 ) ); HYDROData_SequenceOfObjects seq; seq.Append( aProfile1 ); @@ -426,28 +552,200 @@ void test_HYDROData_DTM::test_presentation() DTM->SetSpatialStep( 1.0 ); CPPUNIT_ASSERT_DOUBLES_EQUAL( 1.0, DTM->GetSpatialStep(), EPS ); DTM->Update(); - - CPPUNIT_ASSERT_EQUAL( 10404, (int)DTM->GetAltitudePoints().size() ); - Handle_AIS_InteractiveContext aContext = TestViewer::context(); + CPPUNIT_ASSERT_EQUAL( 9108, (int)DTM->GetAltitudePoints().size() ); + + Handle(AIS_InteractiveContext) aContext = TestViewer::context(); HYDROGUI_ShapeBathymetry* aBathPrs = new HYDROGUI_ShapeBathymetry( 0, aContext, DTM ); aBathPrs->update( true, false ); + aBathPrs->RescaleDefault(); + bool ColorScaleIsDisp = TestViewer::ColorScaleIsDisplayed(); TestViewer::showColorScale( true ); - Handle_Aspect_ColorScale aCS = TestViewer::colorScale(); + Handle(AIS_ColorScale) aCS = TestViewer::colorScale(); aCS->SetMin( 0.0 ); aCS->SetMax( 5.0 ); aCS->SetNumberOfIntervals( 10 ); aBathPrs->UpdateWithColorScale( aCS ); - - TestViewer::show( aBathPrs->getAISObject(), AIS_PointCloud::DM_Points, 0, true, "DTM_1" ); - CPPUNIT_ASSERT_IMAGES + //QTest::qWait(50000); + QImage aDTMPrs = draw_DTM( aBathPrs, 0.5, 600, 600 ); + CPPUNIT_ASSERT_IMAGES3( &aDTMPrs, "DTM_1", false ); delete aBathPrs; + TestViewer::showColorScale( ColorScaleIsDisp ); aDoc->Close(); } void test_HYDROData_DTM::test_garonne() { - //TODO + Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(); + + TCollection_AsciiString fname = REF_DATA_PATH.toLatin1().data(); + fname += "/Profiles.xyz"; + NCollection_Sequence bad_ids; + + int aSize = HYDROData_Profile::ImportFromFile( aDoc, fname, bad_ids, true ); + + CPPUNIT_ASSERT_EQUAL( 0, bad_ids.Size() ); + CPPUNIT_ASSERT_EQUAL( 46, aSize ); + + HYDROData_SequenceOfObjects profiles; + HYDROData_Iterator it( aDoc, KIND_PROFILE ); + for( int i=0; it.More(); it.Next(), i++ ) + { + if( i>=25 && i<=35 ) + profiles.Append( Handle(HYDROData_Profile)::DownCast( it.Current() ) ); + } + + CPPUNIT_ASSERT_EQUAL( 11, (int)profiles.Size() ); + + Handle(HYDROData_DTM) DTM = Handle(HYDROData_DTM)::DownCast( aDoc->CreateObject( KIND_DTM ) ); + DTM->SetProfiles( profiles ); + DTM->SetDDZ( 0.1 ); + DTM->SetSpatialStep( 1.0 ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.1, DTM->GetDDZ(), EPS ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 1.0, DTM->GetSpatialStep(), EPS ); + DTM->Update(); + + CPPUNIT_ASSERT_EQUAL( 281898, (int)DTM->GetAltitudePoints().size() ); + + Handle(AIS_InteractiveContext) aContext = TestViewer::context(); + HYDROGUI_ShapeBathymetry* aBathPrs = new HYDROGUI_ShapeBathymetry( 0, aContext, DTM ); + aBathPrs->update( true, false ); + aBathPrs->RescaleDefault(); + + bool ColorScaleIsDisp = TestViewer::ColorScaleIsDisplayed(); + + TestViewer::showColorScale( true ); + Handle(AIS_ColorScale) aCS = TestViewer::colorScale(); + aCS->SetMin( 0.0 ); + aCS->SetMax( 25.0 ); + aCS->SetNumberOfIntervals( 30 ); + aBathPrs->UpdateWithColorScale( aCS ); + + QImage aDTMPrs = draw_DTM( aBathPrs, 0.5, 600, 600 ); + CPPUNIT_ASSERT_IMAGES3( &aDTMPrs, "DTM_2", false ); + TestViewer::showColorScale( ColorScaleIsDisp ); + delete aBathPrs; + aDoc->Close(); +} + +void test_HYDROData_DTM::test_classifier_1() +{ + TCollection_AsciiString fname = REF_DATA_PATH.toLatin1().data(); + fname += "/pb_1066.cbf"; + CPPUNIT_ASSERT_EQUAL( (int)DocError_OK, (int)HYDROData_Document::Load( fname.ToCString() ) ); + + Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(); + + Handle(HYDROData_CalculationCase) aCase = + Handle(HYDROData_CalculationCase)::DownCast( aDoc->FindObjectByName( "Case_1" ) ); + CPPUNIT_ASSERT_EQUAL( false, aCase.IsNull() ); + std::vector points; + points.push_back( gp_XY( 43.4842, 3.33176 ) ); + points.push_back( gp_XY( -125.777, 2.24728 ) ); + points.push_back( gp_XY( -60.1628, 168.262 ) ); + points.push_back( gp_XY( 21.8055587645, 154.699344457 ) ); + points.push_back( gp_XY( -84.4764138524, 79.2606012276 ) ); + points.push_back( gp_XY( -73.4132070504, 69.7096313266 ) ); + + std::vector values = aCase->GetStricklerCoefficientForPoints( points, 0.0, true ); + CPPUNIT_ASSERT_EQUAL( 6, (int)values.size() ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 1.0123, values[0], EPS ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 1.0123, values[1], EPS ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 1.0221, values[2], EPS ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 1.0221, values[3], EPS ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 1.0221, values[4], EPS ); + CPPUNIT_ASSERT_DOUBLES_EQUAL( 1.0221, values[5], EPS ); + + std::vector types = aCase->GetStricklerTypeForPoints( points ); + CPPUNIT_ASSERT_EQUAL( 6, (int)types.size() ); + CPPUNIT_ASSERT_EQUAL( 123, types[0] ); + CPPUNIT_ASSERT_EQUAL( 123, types[1] ); + CPPUNIT_ASSERT_EQUAL( 221, types[2] ); + CPPUNIT_ASSERT_EQUAL( 221, types[3] ); + CPPUNIT_ASSERT_EQUAL( 221, types[4] ); + CPPUNIT_ASSERT_EQUAL( 221, types[5] ); + + aDoc->Close(); +} + +void test_HYDROData_DTM::test_profile_discretization_warnings() +{ + Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(); + + { + Handle(HYDROData_Profile) aProfile = + Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); + + NCollection_Sequence pnts; + pnts.Append( gp_XY( 0.0, 1.0 ) ); + pnts.Append( gp_XY( 1.0, 0.0 ) ); + pnts.Append( gp_XY( 2.0, 4.0 ) ); + pnts.Append( gp_XY( 3.0, 5.0 ) ); + + aProfile->SetParametricPoints( pnts ); + aProfile->GetProfileUZ()->SetSectionType( 0, HYDROData_IPolyline::SECTION_POLYLINE ); + aProfile->SetLeftPoint( gp_XY( 10, 10 ) ); + aProfile->SetRightPoint( gp_XY( 10, 20 ) ); + + HYDROData_DTM::CurveUZ aMid( 0.0, gp_Vec2d(), 0, 0 ), aWid( 0.0, gp_Vec2d(), 0, 0 ); + int dummy = 0; + QSet warnings; + HYDROData_DTM::ProfileDiscretization( aProfile, 0.0, 0.0, 5.0, 5.0, 0.5, aMid, aWid, dummy, 1E-6, warnings ); + + //CPPUNIT_ASSERT_EQUAL(1, warnings.size()); + CPPUNIT_ASSERT( warnings.contains("One of the extreme points is higher than another")); + } + { + Handle(HYDROData_Profile) aProfile = + Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); + + NCollection_Sequence pnts; + pnts.Append( gp_XY( 0.0, 1.0 ) ); + pnts.Append( gp_XY( 1.0, 0.0 ) ); + pnts.Append( gp_XY( 2.0, 5.0 ) ); + pnts.Append( gp_XY( 3.0, 0.0 ) ); + pnts.Append( gp_XY( 4.0, 1.0 ) ); + + aProfile->SetParametricPoints( pnts ); + aProfile->GetProfileUZ()->SetSectionType( 0, HYDROData_IPolyline::SECTION_POLYLINE ); + aProfile->SetLeftPoint( gp_XY( 10, 10 ) ); + aProfile->SetRightPoint( gp_XY( 10, 20 ) ); + + HYDROData_DTM::CurveUZ aMid( 0.0, gp_Vec2d(), 0, 0 ), aWid( 0.0, gp_Vec2d(), 0, 0 ); + int dummy = 0; + QSet warnings; + HYDROData_DTM::ProfileDiscretization( aProfile, 0.0, 0.0, 5.0, 5.0, 0.5, aMid, aWid, dummy, 1E-6, warnings ); + + CPPUNIT_ASSERT_EQUAL(1, warnings.size()); + CPPUNIT_ASSERT( warnings.contains("More than 2 intersections between profile & altitude Z-lines found")); + } + { + //plato case + Handle(HYDROData_Profile) aProfile = + Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) ); + + NCollection_Sequence pnts; + pnts.Append( gp_XY( 0.0, 5.0 ) ); + pnts.Append( gp_XY( 1.0, 5.0 ) ); + pnts.Append( gp_XY( 2.0, 0.5 ) ); + pnts.Append( gp_XY( 3.0, 5.0 ) ); + pnts.Append( gp_XY( 4.0, 5.0 ) ); + + aProfile->SetParametricPoints( pnts ); + aProfile->GetProfileUZ()->SetSectionType( 0, HYDROData_IPolyline::SECTION_POLYLINE ); + aProfile->SetLeftPoint( gp_XY( 10, 10 ) ); + aProfile->SetRightPoint( gp_XY( 10, 20 ) ); + + HYDROData_DTM::CurveUZ aMid( 0.0, gp_Vec2d(), 0, 0 ), aWid( 0.0, gp_Vec2d(), 0, 0 ); + int dummy = 0; + QSet warnings; + HYDROData_DTM::ProfileDiscretization( aProfile, 0.0, 0.0, 5.0, 5.0, 0.5, aMid, aWid, dummy, 1E-6, warnings ); + + CPPUNIT_ASSERT_EQUAL(2, warnings.size()); + CPPUNIT_ASSERT( warnings.contains("Plato case on extremes")); + } + + aDoc->Close(); }