Salome HOME
Merge branch 'BR_H2018_4' into BR_PY3
[modules/hydro.git] / src / HYDRO_tests / test_HYDROData_PolylineXY.cxx
index abc72f651b6c4a3d99c16ceef37e5768c27e96e4..cc07bbf7256c2ea5c560d3c21b16ceb21a109213 100644 (file)
@@ -29,6 +29,8 @@
 #include <HYDROData_Tool.h>
 #include <HYDROGUI_Shape.h>
 #include <HYDROGUI_Polyline.h>
+#include <HYDROData_SinusX.h>
+#include <BRep_Builder.hxx>
 
 #include <AIS_DisplayMode.hxx>
 #include <AIS_PointCloud.hxx>
 #include <TopoDS_Vertex.hxx>
 #include <TopoDS_Wire.hxx>
 #include <gp_XY.hxx>
+#include <HYDROData_Tool.h>
+
+extern QString REF_DATA_PATH;
 
 void test_HYDROData_PolylineXY::test_polyline()
 {
-  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document();
 
   Handle(HYDROData_PolylineXY) aPolyline = 
     Handle(HYDROData_PolylineXY)::DownCast(aDoc->CreateObject(KIND_POLYLINEXY));
@@ -79,7 +84,7 @@ void test_HYDROData_PolylineXY::test_polyline()
 
 void test_HYDROData_PolylineXY::test_copy()
 {
-  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document();
   Handle(HYDROData_PolylineXY) aPolyline1 = 
     Handle(HYDROData_PolylineXY)::DownCast(aDoc->CreateObject(KIND_POLYLINEXY));
 
@@ -97,7 +102,7 @@ void test_HYDROData_PolylineXY::test_copy()
 
 void test_HYDROData_PolylineXY::test_split_refs_624()
 {
-  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document();
 
   Handle(HYDROData_PolylineXY) aPolyline = 
     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
@@ -130,7 +135,7 @@ void test_HYDROData_PolylineXY::test_split_refs_624()
 
 void test_HYDROData_PolylineXY::test_extraction_immersible_zone()
 {
-  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document();
 
   Handle(HYDROData_PolylineXY) aPolyline = 
     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
@@ -164,7 +169,7 @@ void test_HYDROData_PolylineXY::test_extraction_immersible_zone()
 
 void test_HYDROData_PolylineXY::test_extraction_channel_refs_611()
 {
-  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document();
 
   Handle(HYDROData_PolylineXY) aPolyline2d = 
     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
@@ -221,7 +226,7 @@ void test_HYDROData_PolylineXY::test_extraction_channel_refs_611()
 
 void test_HYDROData_PolylineXY::test_presentation()
 {
-  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document();
 
   Handle(HYDROData_PolylineXY) aPolyline2d = 
     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
@@ -294,15 +299,27 @@ void test_HYDROData_PolylineXY::test_presentation()
 
 void test_HYDROData_PolylineXY::test_split_refs_627()
 {
-  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document();
 
   Handle(HYDROData_PolylineXY) aPolyline = 
     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
   aPolyline->SetName( "test" );
 
-  QList<double> aCoords = QList<double>() << 10 << 10 << 20 << 10 << 20 << 20 << 10 << 20;
-  TopoDS_Wire aWire = Wire2d( aCoords, false );
-  aPolyline->SetShape( aWire );
+
+  //QList<double> aCoords = QList<double>() << 10 << 10 << 20 << 10 << 20 << 20 << 10 << 20;
+  aPolyline->AddSection( "", HYDROData_PolylineXY::SECTION_POLYLINE, false );
+  aPolyline->AddPoint( 0, gp_XY( 10, 10  ) );
+  aPolyline->AddPoint( 0, gp_XY( 20, 10 ) );
+  aPolyline->AddPoint( 0, gp_XY( 20, 20 ) );
+  aPolyline->AddPoint( 0, gp_XY( 10, 20 ) );
+  aPolyline->Update();
+
+
+  //TopoDS_Wire aWire = Wire2d( aCoords, false );
+  //aPolyline->SetShape( aWire );
+  TopoDS_Shape aW = aPolyline->GetShape();
+  CPPUNIT_ASSERT (aW.ShapeType() == TopAbs_WIRE);
+
   aPolyline->SetWireColor( Qt::red );
 
   gp_Pnt2d aPnt( 20, 20 );
@@ -316,25 +333,34 @@ void test_HYDROData_PolylineXY::test_split_refs_627()
   {
     Handle(HYDROData_PolylineXY) anObj = Handle(HYDROData_PolylineXY)::DownCast( anIt.Current() );
     if( aPolyline->Label() != anObj->Label() )
-      TestViewer::show( anObj->GetShape(), 0, true, anObj->GetWireColor() );
+    {
+      QColor color;
+      anObj->GetSectionColor(0, color);
+      TestViewer::show( anObj->GetShape(), 0, true, color );
+    }
   }
   CPPUNIT_ASSERT_IMAGES
     
   HYDROData_Iterator anIt2( aDoc, KIND_POLYLINEXY );
   CPPUNIT_ASSERT_EQUAL( true, anIt2.More() );
   CPPUNIT_ASSERT_EQUAL( QString( "test" ), anIt2.Current()->GetName() );
-  CPPUNIT_ASSERT_EQUAL( QColor( Qt::red ),
-    Handle(HYDROData_PolylineXY)::DownCast( anIt2.Current() )->GetWireColor() );
+
+  QColor color1, color2, color3;
+  Handle(HYDROData_PolylineXY)::DownCast( anIt2.Current() )->GetSectionColor(0, color1);
+
+  CPPUNIT_ASSERT_EQUAL( QColor( Qt::red ), color1);
   anIt2.Next();
   CPPUNIT_ASSERT_EQUAL( true, anIt2.More() );
   CPPUNIT_ASSERT_EQUAL( QString( "test_1" ), anIt2.Current()->GetName() );
-  CPPUNIT_ASSERT_EQUAL( QColor( Qt::red ), 
-    Handle(HYDROData_PolylineXY)::DownCast( anIt2.Current() )->GetWireColor() );
+
+  Handle(HYDROData_PolylineXY)::DownCast( anIt2.Current() )->GetSectionColor(0, color2);
+  CPPUNIT_ASSERT_EQUAL( QColor( Qt::red ), color2 );
+
   anIt2.Next();
   CPPUNIT_ASSERT_EQUAL( true, anIt2.More() );
   CPPUNIT_ASSERT_EQUAL( QString( "test_2" ), anIt2.Current()->GetName() );
-  CPPUNIT_ASSERT_EQUAL( QColor( Qt::red ), 
-    Handle(HYDROData_PolylineXY)::DownCast( anIt2.Current() )->GetWireColor() );
+  Handle(HYDROData_PolylineXY)::DownCast( anIt2.Current() )->GetSectionColor(0, color3);
+  CPPUNIT_ASSERT_EQUAL( QColor( Qt::red ), color3 );
   anIt2.Next();
   CPPUNIT_ASSERT_EQUAL( false, anIt2.More() );
   anIt2.Next();
@@ -344,7 +370,7 @@ void test_HYDROData_PolylineXY::test_split_refs_627()
 
 void test_HYDROData_PolylineXY::test_custom_polylines()
 {
-  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document();
 
   Handle(HYDROData_PolylineXY) aPolyline1 = 
     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
@@ -400,7 +426,7 @@ void test_HYDROData_PolylineXY::test_custom_polylines()
 
 void test_HYDROData_PolylineXY::test_merge_refs_630()
 {
-  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document();
 
   Handle(HYDROData_PolylineXY) aPolyline1 = 
     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
@@ -450,7 +476,7 @@ void test_HYDROData_PolylineXY::test_merge_refs_630()
 
 void test_HYDROData_PolylineXY::test_split_straight_refs_634()
 {
-  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document();
 
   Handle(HYDROData_PolylineXY) aPolyline1 = 
     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
@@ -518,3 +544,135 @@ void test_HYDROData_PolylineXY::test_split_straight_refs_634()
 
   aDoc->Close();
 }
+
+void test_HYDROData_PolylineXY::test_import_from_xyz()
+{
+  NCollection_Sequence<Handle(HYDROData_Entity)> ents;
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document();
+  TCollection_AsciiString fname = REF_DATA_PATH.toLatin1().data();
+  fname += "/profiles1.xyz";
+
+  NCollection_Sequence<Handle(HYDROData_Entity)> importedEntities;
+  bool stat = HYDROData_Tool::importPolylineFromXYZ(QString(fname.ToCString()), aDoc, true, importedEntities);
+  CPPUNIT_ASSERT (stat);
+  Handle(HYDROData_PolylineXY) aPolyXY = Handle(HYDROData_PolylineXY)::DownCast(importedEntities.First());   
+  CPPUNIT_ASSERT_EQUAL (importedEntities.Size(), 1);
+  CPPUNIT_ASSERT (!aPolyXY.IsNull());
+  TestViewer::show( aPolyXY->GetShape(), 0, true, "Polyline_import_XY" );
+  CPPUNIT_ASSERT_IMAGES
+  importedEntities.Clear();
+
+  stat = HYDROData_Tool::importPolylineFromXYZ(QString(fname.ToCString()), aDoc, false, importedEntities);
+  CPPUNIT_ASSERT (stat);
+  CPPUNIT_ASSERT_EQUAL (importedEntities.Size(), 2);
+  Handle(HYDROData_Polyline3D) aPoly3D;
+  aPolyXY = Handle(HYDROData_PolylineXY)::DownCast(importedEntities.First());   
+  aPoly3D = Handle(HYDROData_Polyline3D)::DownCast(importedEntities.Last());
+  CPPUNIT_ASSERT (!aPolyXY.IsNull());
+  CPPUNIT_ASSERT (!aPoly3D.IsNull());
+  TestViewer::show( aPolyXY->GetShape(), 0, false, "Polyline_import_XYZ" );
+  TestViewer::show( aPoly3D->GetShape3D(), 0, false, "Polyline_import_XYZ_3D" );
+  CPPUNIT_ASSERT_IMAGES
+}
+
+
+void test_HYDROData_PolylineXY::test_import_from_sx()
+{
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document();
+  TCollection_AsciiString fname = REF_DATA_PATH.toLatin1().data();
+  fname += "/polylines_sx.sx";
+
+  HYDROData_SinusX aSinusXImporter;
+  NCollection_Sequence<Handle(HYDROData_Entity)> importedEntities;
+  //bool stat = aSinusXImporter.Import(QString(fname.ToCString()), aDoc, importedEntities);
+  bool ParseStat = aSinusXImporter.OpenAndParse(QString(fname.ToCString()));
+  CPPUNIT_ASSERT (ParseStat);
+  aSinusXImporter.Import(aDoc, importedEntities, NULL);
+  CPPUNIT_ASSERT_EQUAL (importedEntities.Size(), 9);
+
+  Handle(HYDROData_PolylineXY) aPolyXY1 = Handle(HYDROData_PolylineXY)::DownCast(importedEntities(1));   
+  Handle(HYDROData_PolylineXY) aPolyXY2 = Handle(HYDROData_PolylineXY)::DownCast(importedEntities(4));  
+  Handle(HYDROData_PolylineXY) aPolyXY3 = Handle(HYDROData_PolylineXY)::DownCast(importedEntities(7));   
+
+  CPPUNIT_ASSERT (!aPolyXY1.IsNull());
+  CPPUNIT_ASSERT (!aPolyXY2.IsNull());
+  CPPUNIT_ASSERT (!aPolyXY3.IsNull());
+
+  aPolyXY1->Update();
+  aPolyXY2->Update();
+  aPolyXY3->Update();
+
+  CPPUNIT_ASSERT_EQUAL (aPolyXY1->GetName(), QString("AXIS"));
+  CPPUNIT_ASSERT_EQUAL (aPolyXY2->GetName(), QString("LB"));
+  CPPUNIT_ASSERT_EQUAL (aPolyXY3->GetName(), QString("RB"));
+
+  TopoDS_Shape sh1 = aPolyXY1->GetShape();
+  TopoDS_Shape sh2 = aPolyXY2->GetShape();
+  TopoDS_Shape sh3 = aPolyXY3->GetShape();
+  CPPUNIT_ASSERT (!sh1.IsNull());
+  CPPUNIT_ASSERT (!sh2.IsNull());
+  CPPUNIT_ASSERT (!sh3.IsNull());
+
+  BRep_Builder BB;
+  TopoDS_Compound cmp;
+  BB.MakeCompound(cmp);
+  BB.Add(cmp, sh1);
+  BB.Add(cmp, sh2);
+  BB.Add(cmp, sh3);
+
+  TestViewer::show( cmp, 0, true, "Polylines_import_SX" );
+  CPPUNIT_ASSERT_IMAGES
+}
+
+void test_HYDROData_PolylineXY::test_import_from_sx_options()
+{
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
+  TCollection_AsciiString fname = REF_DATA_PATH.toLatin1().data();
+  fname += "/trait_cote_lambert1N.sx";
+
+  HYDROData_SinusX aSinusXImporter;
+  NCollection_Sequence<Handle(HYDROData_Entity)> importedEntities;
+  //bool stat = aSinusXImporter.Import(QString(fname.ToCString()), aDoc, importedEntities);
+  bool ParseStat = aSinusXImporter.OpenAndParse(QString(fname.ToCString()));
+  CPPUNIT_ASSERT (ParseStat);
+
+  std::vector<HYDROData_SinusX::ImportOptions> options; 
+  int size = aSinusXImporter.GetCurveBlocks().size();
+  CPPUNIT_ASSERT_EQUAL(33, size);
+  for (int i = 0; i < 33; i++)
+  {
+    HYDROData_SinusX::ImportOptions option;
+    option.ImportAsBathy = false;
+    if (i>12)
+    {
+      option.ImportAsPolyXY = true;
+      option.ImportAsProfile = true;
+    }
+    else
+    {
+      option.ImportAsPolyXY = false;
+      option.ImportAsProfile = false;
+    }
+    options.push_back(option);
+  }  
+
+  aSinusXImporter.Import(aDoc, importedEntities, &options);
+  CPPUNIT_ASSERT_EQUAL (importedEntities.Size(), 60);
+}
+
+void test_HYDROData_PolylineXY::test_polyline_dbf_info_simple()
+{
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
+  Handle(HYDROData_PolylineXY) aPolyline = Handle(HYDROData_PolylineXY)::DownCast(aDoc->CreateObject(KIND_POLYLINEXY));
+  QStringList dbf_dummy, dbf_out;
+  dbf_dummy << "n1" << "n2" << "n3" << "n4";
+  aPolyline->SetDBFInfo(dbf_dummy);
+  CPPUNIT_ASSERT (aPolyline->GetDBFInfo(dbf_out));
+  CPPUNIT_ASSERT_EQUAL (dbf_out.size(), 4);
+  CPPUNIT_ASSERT ("n1" == dbf_out[0]);
+  CPPUNIT_ASSERT ("n2" == dbf_out[1]);
+  CPPUNIT_ASSERT ("n3" == dbf_out[2]);
+  CPPUNIT_ASSERT ("n4" == dbf_out[3]);
+}
+
+