Salome HOME
patch for install error on Linux
[modules/hydro.git] / src / HYDROData / HYDROData_Stream.cxx
index 33515d7d7640d2e311280b29032808e40ed4c4ce..022cc30664aecc4bf3708c748afc7dab995dd0f0 100644 (file)
@@ -28,6 +28,7 @@
 #include "HYDROData_IProfilesInterpolator.h"
 #include "HYDROData_Tool.h"
 #include "HYDROData_DTM.h"
+#include <HYDROData_Bathymetry.h>
 
 #include <TDataStd_RealArray.hxx>
 
@@ -75,6 +76,7 @@
 #include <Geom_BSplineCurve.hxx>
 
 #include <TopTools_HArray1OfShape.hxx>
+#include <TopTools_IndexedMapOfOrientedShape.hxx>
 
 #include <SortTools_QuickSortOfReal.hxx>
 
@@ -104,19 +106,6 @@ HYDROData_Stream::~HYDROData_Stream()
 {
 }
 
-void HYDROData_Stream::SetLabel( const TDF_Label& theLabel )
-{
-  HYDROData_NaturalObject::SetLabel( theLabel );
-
-  if( GetAltitudeObject().IsNull() )
-  {
-    Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( myLab );
-    Handle(HYDROData_DTM) aDTM = 
-      Handle(HYDROData_DTM)::DownCast( aDoc->CreateObject( KIND_DTM ) );
-    SetAltitudeObject( aDTM );
-  }
-}
-
 QStringList HYDROData_Stream::DumpToPython( const QString&       thePyScriptPath,
                                             MapOfTreatedObjects& theTreatedObjects ) const
 {
@@ -178,7 +167,8 @@ Handle(Geom_BSplineCurve) HYDROData_Stream::buildInterpolationCurve(
 
 void HYDROData_Stream::Update()
 {
-  updateProfilesOrder();
+  if (!GetHydraulicAxis().IsNull())
+    updateProfilesOrder();
 
   // Update bottom polyline if exists
   const Handle(HYDROData_Polyline3D) aBottomPolyline = GetBottomPolyline();
@@ -192,8 +182,11 @@ void HYDROData_Stream::Update()
     }
   }
 
-  DTM()->Update();
-  UpdatePrs();
+  Handle_HYDROData_DTM dtm = DTM();
+  dtm->Update();
+  UpdatePrs( dtm );
+
+  HYDROData_NaturalObject::Update();
 }
 
 bool HYDROData_Stream::IsHas2dPrs() const
@@ -211,7 +204,23 @@ bool HYDROData_Stream::CreatePresentations( const Handle_HYDROData_DTM& theDTM,
   if( profiles.Length() < 2 )
     return false;
 
-  TopTools_ListOfShape profiles3d;
+  TopoDS_Shape Out3dPres;
+  TopoDS_Shape Out2dPres;
+  TopoDS_Shape OutLeftB;
+  TopoDS_Shape OutRightB;
+  TopoDS_Shape OutInlet;
+  TopoDS_Shape OutOutlet;
+
+  theDTM->GetPresentationShapes(Out3dPres, Out2dPres, OutLeftB, OutRightB, OutInlet, OutOutlet);
+
+  thePrs.myInlet = TopoDS::Wire(OutInlet);
+  thePrs.myOutlet = TopoDS::Wire(OutOutlet);
+  thePrs.myLeftBank = TopoDS::Wire(OutLeftB);
+  thePrs.myRightBank = TopoDS::Wire(OutRightB);
+  thePrs.myPrs2D = Out2dPres;
+  thePrs.myPrs3D = Out3dPres;
+  /*std::vector<TopoDS_Wire> profiles3d;
+  profiles3d.reserve(profiles.Length());
 
   // Pre-processing
   HYDROData_SequenceOfObjects::Iterator anIter( profiles );
@@ -224,20 +233,19 @@ bool HYDROData_Stream::CreatePresentations( const Handle_HYDROData_DTM& theDTM,
       continue;
 
     const TopoDS_Shape& aProfileShape = aProfile->GetShape3D();
-    profiles3d.Append( aProfileShape );
-  }
+    //TopExp_Explorer exp(aProfileShape, TopAbs_EDGE);
+    profiles3d.push_back( TopoDS::Wire(aProfileShape) );
+  }*/
 
-  TopoDS_Edge aLeftBank, aRightBank;
-  theDTM->CreateBankShapes( aLeftBank, aRightBank );
-  return CreatePresentations( aLeftBank, aRightBank, profiles3d, thePrs );
+  return true;
 }
 
-void HYDROData_Stream::UpdatePrs()
+void HYDROData_Stream::UpdatePrs( const Handle_HYDROData_DTM& theDTM )
 {
   HYDROData_NaturalObject::Update();
-
+  
   PrsDefinition aResultPrs;
-  if ( !CreatePresentations( DTM(), aResultPrs ) )
+  if ( !CreatePresentations( theDTM, aResultPrs ) )
     return;
 
   SetShape3D( aResultPrs.myPrs3D );
@@ -282,7 +290,7 @@ QColor HYDROData_Stream::DefaultBorderColor() const
 bool HYDROData_Stream::IsValidAsAxis( const Handle(HYDROData_PolylineXY)& theHydAxis )
 {
   if ( theHydAxis.IsNull() )
-    return false;
+    return true;
 
   TopoDS_Shape aHydraulicShape = theHydAxis->GetShape();
   if ( aHydraulicShape.IsNull() || 
@@ -319,6 +327,7 @@ TopoDS_Shape HYDROData_Stream::GetOutletShape() const
 
 Handle_HYDROData_DTM HYDROData_Stream::DTM() const
 {
+  const_cast<HYDROData_Stream*>( this )->checkAndSetAltitudeObject();
   return Handle(HYDROData_DTM)::DownCast( GetAltitudeObject() );
 }
 
@@ -330,6 +339,7 @@ double HYDROData_Stream::GetDDZ() const
 void HYDROData_Stream::SetDDZ( double theDDZ )
 {
   DTM()->SetDDZ( theDDZ );
+  Changed( Geom_3d );
 }
   
 double HYDROData_Stream::GetSpatialStep() const
@@ -340,6 +350,7 @@ double HYDROData_Stream::GetSpatialStep() const
 void HYDROData_Stream::SetSpatialStep( double theSpatialStep )
 {
   DTM()->SetSpatialStep( theSpatialStep );
+  Changed( Geom_3d );
 }
 
 bool HYDROData_Stream::SetHydraulicAxis( const Handle(HYDROData_PolylineXY)& theAxis )
@@ -391,23 +402,32 @@ bool HYDROData_Stream::HasIntersection( const Handle(HYDROData_Profile)& theProf
   return HasIntersection( aHydAxis, theProfile, thePlane, theOutPar );
 }
 
+#include <BRepAlgo_NormalProjection.hxx>
+
 bool HYDROData_Stream::HasIntersection( const Handle(HYDROData_PolylineXY)& theHydAxis, 
                                         const Handle(HYDROData_Profile)&    theProfile, 
                                         const TopoDS_Face&                  thePlane,
                                         Standard_Real&                      theOutPar )
 {
-  if ( theProfile.IsNull() || !IsValidAsAxis( theHydAxis ) )
+  if ( theProfile.IsNull() /*|| !IsValidAsAxis( theHydAxis )*/ )
     return false; 
 
+  if (theHydAxis.IsNull())
+    return true; //empty h_axis; its's OK
+
   TopoDS_Wire aHydraulicWire = TopoDS::Wire( theHydAxis->GetShape() ); //guide line
   TopoDS_Wire aProfileWire = TopoDS::Wire( theProfile->GetTopShape() );
-  if ( aHydraulicWire.IsNull() || aProfileWire.IsNull() )
+  if ( aProfileWire.IsNull() )
     return false;
 
-  BRepProj_Projection aProjector (aProfileWire, thePlane, gp::OZ().Direction());
-  if(!aProjector.IsDone())
+  //BRepProj_Projection aProjector (aProfileWire, thePlane, gp::OZ().Direction());
+  BRepAlgo_NormalProjection nproj(thePlane);
+  nproj.Add(aProfileWire);
+  nproj.SetDefaultParams();
+  nproj.Build();
+  if(!nproj.IsDone())
     return false;
-  TopoDS_Shape aPrjProfile = aProjector.Shape();
+  TopoDS_Shape aPrjProfile = nproj.Projection();
   if(aPrjProfile.IsNull())
     return false;
   TopoDS_Vertex aV1, aV2;
@@ -498,13 +518,12 @@ bool HYDROData_Stream::AddProfile( const Handle(HYDROData_Profile)& theProfile )
   if ( theProfile.IsNull() )
     return false;
 
-  Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
-  if ( aHydAxis.IsNull() )
-    return false;
// Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
// if ( aHydAxis.IsNull() )
//   return false;
 
   TopoDS_Face aPlane;
-  if(!BuildFace(aHydAxis, aPlane))
-    return false;
+  BuildRefFace( aPlane );
 
   Standard_Real aPar(.0);
   if ( HasReference( theProfile, DataTag_Profile ) || !HasIntersection( theProfile, aPlane, aPar ) )
@@ -625,13 +644,13 @@ void HYDROData_Stream::RemoveProfiles()
 void HYDROData_Stream::insertProfileInToOrder( const Handle(HYDROData_Profile)& theProfile,
                                                const int                        theBeforeIndex )
 {
-  Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
-  if ( theProfile.IsNull() || aHydAxis.IsNull() )
+  //Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
+  if ( theProfile.IsNull() )
     return; 
 
-  TopoDS_Wire aHydraulicWire = TopoDS::Wire( aHydAxis->GetShape() );
+  //TopoDS_Wire aHydraulicWire = TopoDS::Wire( aHydAxis->GetShape() );
   TopoDS_Wire aProfileWire = TopoDS::Wire( theProfile->GetTopShape() );
-  if ( aHydraulicWire.IsNull() || aProfileWire.IsNull() )
+  if ( aProfileWire.IsNull() )
     return;
 
   if ( theBeforeIndex == -1 )
@@ -640,25 +659,9 @@ void HYDROData_Stream::insertProfileInToOrder( const Handle(HYDROData_Profile)&
     InsertReferenceObject( theProfile, DataTag_Profile, theBeforeIndex );
 }
 
-bool HYDROData_Stream::BuildFace( const Handle(HYDROData_PolylineXY)& theHydAxis,
-                                  TopoDS_Face&                        thePlane )
+void HYDROData_Stream::BuildRefFace( TopoDS_Face& thePlane )
 {
-  if ( !IsValidAsAxis( theHydAxis ) )
-    return false;
-
-  TopoDS_Wire aHydraulicWire = TopoDS::Wire( theHydAxis->GetShape() );
-
-  gp_Ax2 aX2(gp::XOY());
-  gp_Ax3 aX3(aX2);
-  gp_Pln aPln(aX3);   
-  Bnd_Box B;
-  BRepBndLib::Add(aHydraulicWire,B);
-  Standard_Real axmin,aymin,azmin,axmax,aymax,azmax;
-  B.Get(axmin,aymin,azmin,axmax,aymax,azmax);
-  BRepBuilderAPI_MakeFace aMkr(aPln, axmin-500., axmax+500., aymin-500., aymax+500.); // to be tuned later according max/ Profile deviation
-  if(!aMkr.IsDone() || aMkr.Shape().IsNull()) return false;
-  thePlane = TopoDS::Face(aMkr.Shape());
-  return true;
+  thePlane = BRepBuilderAPI_MakeFace(gp_Pln(gp_Pnt(0,0,0),gp_Dir(0,0,1))).Face();
 }
 
 void HYDROData_Stream::updateProfilesOrder()
@@ -675,8 +678,7 @@ void HYDROData_Stream::updateProfilesOrder()
     return; 
 
   TopoDS_Face aPlane;
-  if ( !BuildFace( aHydAxis, aPlane ) )
-    return;
+  BuildRefFace( aPlane );
 
   Standard_Real aPar( .0 );
 
@@ -1044,154 +1046,51 @@ void HYDROData_Stream::CopyTo( const Handle(HYDROData_Entity)& theDestination,
   }
 }
 
-bool HYDROData_Stream::CreatePresentations( const TopoDS_Edge&          theLeftBank,
-                                            const TopoDS_Edge&          theRightBank,
-                                            const TopTools_ListOfShape& theProfiles3d,
-                                            PrsDefinition&              thePrs )
+void HYDROData_Stream::CreatePresentations( const Handle(TColgp_HArray1OfPnt)     theArrayOfFPnt,
+                                            const Handle(TColgp_HArray1OfPnt)     theArrayOfLPnt,
+                                            const Handle(TopTools_HArray1OfShape) theArrOfProfiles,
+                                            PrsDefinition&                        thePrs )
 {
-  return true;
-  /*if ( theArrayOfFPnt.IsNull() || theArrayOfLPnt.IsNull() || theArrOfProfiles.IsNull() ) {
-    return false;
-  }
 
-  if ( theArrayOfFPnt->Length() != theArrayOfLPnt->Length() ) {
-    return false;
-  }
-  
-  // Construct of the 3D presentation
-  Handle(Geom_BSplineCurve) aBSpline = buildInterpolationCurve (theArrayOfFPnt);
-  if(aBSpline.IsNull())
-    return false;
-
-  TopoDS_Edge anEdgLeft, anEdgRight;
-  
-  BRepBuilderAPI_MakeEdge aMakeEdge(aBSpline);
-  if(aMakeEdge.IsDone()) 
-    anEdgLeft = aMakeEdge.Edge();
-
-  if(anEdgLeft.IsNull())
-    return false;
-
-  aBSpline.Nullify();
-  aBSpline = buildInterpolationCurve (theArrayOfLPnt);  
-  if(aBSpline.IsNull())
-    return false;
-
-  aMakeEdge.Init(aBSpline);
-  if(aMakeEdge.IsDone()) 
-    anEdgRight = aMakeEdge.Edge();
-
-  if(anEdgRight.IsNull())
-    return false;
-
-  BRep_Builder aBB;
-  TopoDS_Compound aCmp;
-  aBB.MakeCompound(aCmp);
-  for (int i=1 ; i < theArrOfProfiles->Length() +1; i++ )  
-    aBB.Add(aCmp, theArrOfProfiles->Value(i));
-
-  aBB.Add(aCmp,anEdgLeft);
-  aBB.Add(aCmp,anEdgRight);
-  BRepCheck_Analyzer aCh(aCmp);
-  if(aCh.IsValid())
-    thePrs.myPrs3D = aCmp;
-#ifdef DEB_UPDATE
-  else {
-    BRepTools::Write(aCmp, "str3d.brep");
-    thePrs.myPrs3D = aCmp;
+  HYDROData_Bathymetry::AltitudePoints left;
+  for (int i = theArrayOfLPnt->Lower(); i <= theArrayOfLPnt->Upper(); i++)
+  {
+    left.push_back(HYDROData_Bathymetry::AltitudePoint(theArrayOfLPnt->Value(i).X(), 
+      theArrayOfLPnt->Value(i).Y(),
+      theArrayOfLPnt->Value(i).Z()));
   }
-#endif
 
-  // Construct the top presentation
-  int aNbPoints = theArrayOfFPnt->Length();
-  Handle(TColgp_HArray1OfPnt) anArrayOfFPnt = new TColgp_HArray1OfPnt(1, aNbPoints);
-  Handle(TColgp_HArray1OfPnt) anArrayOfLPnt = new TColgp_HArray1OfPnt(1, aNbPoints);  
-  for( int i=1; i <= aNbPoints; i++ ) {
-      gp_Pnt aPnt = theArrayOfFPnt->Value(i);
-      aPnt.SetZ(.0); // make 2d
-      anArrayOfFPnt->SetValue(i, aPnt);
-      aPnt = theArrayOfLPnt->Value(i);
-      aPnt.SetZ(.0);
-      anArrayOfLPnt->SetValue(i, aPnt);
+  HYDROData_Bathymetry::AltitudePoints right;
+  for (int i = theArrayOfFPnt->Lower(); i <= theArrayOfFPnt->Upper(); i++)
+  {
+    right.push_back(HYDROData_Bathymetry::AltitudePoint(theArrayOfFPnt->Value(i).X(), 
+      theArrayOfFPnt->Value(i).Y(),
+      theArrayOfFPnt->Value(i).Z()));
   }
 
-  aBSpline.Nullify();
-  aBSpline = buildInterpolationCurve (anArrayOfFPnt);  
-  if(aBSpline.IsNull())
-    return false; 
+  std::vector<HYDROData_Bathymetry::AltitudePoints> dummy;
+  TopTools_IndexedMapOfOrientedShape ll = HYDROData_DTM::Create3DShape(left, right, dummy);
 
-  aMakeEdge.Init(aBSpline);
-  if(aMakeEdge.IsDone()) 
-      anEdgLeft = aMakeEdge.Edge();
-
-  aBSpline.Nullify();
-  aBSpline = buildInterpolationCurve (anArrayOfLPnt);  
-  if(aBSpline.IsNull())
-    return false; 
-
-  aMakeEdge.Init(aBSpline);
-  if(aMakeEdge.IsDone()) 
-    anEdgRight = aMakeEdge.Edge();
-  if(anEdgRight.IsNull())
-    return false;
-
-  BRepBuilderAPI_MakeEdge aMakeEdge2(anArrayOfFPnt->Value(1),anArrayOfLPnt->Value(1));
-  TopoDS_Edge aBotEdge, aTopEdge;
-  if(aMakeEdge2.IsDone()) 
-    aBotEdge = aMakeEdge2.Edge();
-
-  BRepBuilderAPI_MakeEdge aMakeEdge3(anArrayOfFPnt->Value(anArrayOfFPnt->Length()),anArrayOfLPnt->Value(anArrayOfLPnt->Length()));
-  if(aMakeEdge3.IsDone()) 
-    aTopEdge = aMakeEdge3.Edge();
-
-  // Make wire for 2D presentation with updating of corresponding edges
-  BRepBuilderAPI_MakeWire aMakeWire;
-  
-  aMakeWire.Add( aBotEdge );
-  thePrs.myInlet = aMakeWire.Edge();
-
-  aMakeWire.Add( anEdgLeft );
-  thePrs.myLeftBank = aMakeWire.Edge();
-
-  aMakeWire.Add( aTopEdge );
-  thePrs.myOutlet = aMakeWire.Edge();
-
-  aMakeWire.Add( anEdgRight );
-  thePrs.myRightBank = aMakeWire.Edge();
-
-  TopoDS_Wire aSectProfileWire;
-  if(aMakeWire.IsDone())
-    aSectProfileWire = aMakeWire.Wire();
-
-  BRepBuilderAPI_MakeFace aMakeFace( aSectProfileWire, Standard_True );
-  TopoDS_Face aFace;
-  aMakeFace.Build();
-  if( aMakeFace.IsDone() )
-    aFace = aMakeFace.Face();
+  if (!ll.IsEmpty())
+  {
+    thePrs.myLeftBank = TopoDS::Wire(ll(1));
+    thePrs.myRightBank = TopoDS::Wire(ll(2));
+  }
 
-  TopoDS_Shape aPrs2D;
+  thePrs.myInlet = TopoDS::Wire(theArrOfProfiles->Value(theArrOfProfiles->Lower())); //TODO check that
+  thePrs.myOutlet = TopoDS::Wire(theArrOfProfiles->Value(theArrOfProfiles->Upper()));
 
-  if ( !theArrOf2DProfiles.IsNull() ) {
-    aCmp.Nullify();
-    aBB.MakeCompound(aCmp);
-    aBB.Add(aCmp,aFace);
-    for(int i=1;i <= theArrOf2DProfiles->Length(); i++)
-      aBB.Add(aCmp, theArrOf2DProfiles->Value(i));
+  //make new compound so it's shapes will be in known order to build correct projection
+  BRep_Builder BB;
+  TopoDS_Compound newCmp;
+  BB.MakeCompound(newCmp);
+  BB.Add(newCmp, thePrs.myLeftBank);
+  BB.Add(newCmp, thePrs.myInlet);
+  BB.Add(newCmp, thePrs.myOutlet);
+  BB.Add(newCmp, thePrs.myRightBank);
 
-    aPrs2D = aCmp;
-  } else {
-    aPrs2D = aFace;
-  }
+  thePrs.myPrs3D = newCmp;
 
-  aCh.Init(aPrs2D);
-  if(aCh.IsValid())
-   thePrs.myPrs2D = aPrs2D;
-#ifdef DEB_UPDATE
-  else {
-    BRepTools::Write(aPrs2D, "str2d.brep");
-    thePrs.myPrs2D = aPrs2D;
-  }
-#endif
+  HYDROData_DTM::Get2dFaceFrom3dPres( newCmp, TopoDS::Face(thePrs.myPrs2D) ); //__TODO
 
-  return true;*/
 }
\ No newline at end of file