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.
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.
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
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 #include "HYDROData_Channel.h"
21 #include "HYDROData_Document.h"
22 #include "HYDROData_Polyline3D.h"
23 #include "HYDROData_Profile.h"
24 #include "HYDROData_PolylineXY.h"
25 #include "HYDROData_Projection.h"
26 #include "HYDROData_ShapesGroup.h"
27 #include "HYDROData_ShapesTool.h"
28 #include "HYDROData_Stream.h"
29 #include "HYDROData_Tool.h"
30 #include "HYDROData_ChannelAltitude.h"
32 #include <BRepBuilderAPI_MakeWire.hxx>
33 #include <BRepLib_MakePolygon.hxx>
35 #include <BRepOffsetAPI_MakePipeShell.hxx>
36 #include <BRepOffsetAPI_MakePipe.hxx>
37 #include <BRepCheck_Analyzer.hxx>
39 #include <BRep_Tool.hxx>
41 #include <BRepBuilderAPI_Transform.hxx>
43 #include <BRepLib_MakeEdge.hxx>
44 #include <BRepLib_MakeWire.hxx>
46 #include <GeomAPI_ProjectPointOnCurve.hxx>
51 #include <TopExp_Explorer.hxx>
53 #include <TColgp_HArray1OfPnt.hxx>
54 #include <TColgp_Array1OfDir.hxx>
56 #include <TColStd_Array1OfReal.hxx>
58 #include <TopTools_HArray1OfShape.hxx>
59 #include <TopTools_SequenceOfShape.hxx>
62 #include <TopoDS_Wire.hxx>
63 #include <TopoDS_Vertex.hxx>
65 #include <Quantity_Parameter.hxx>
69 #include <BRepTools.hxx>
73 #include <QStringList>
76 #include "HYDRO_trace.hxx"
78 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Channel,HYDROData_ArtificialObject)
81 HYDROData_Channel::HYDROData_Channel()
82 : HYDROData_ArtificialObject( Geom_3d )
86 HYDROData_Channel::~HYDROData_Channel()
90 QStringList HYDROData_Channel::DumpToPython( const QString& thePyScriptPath,
91 MapOfTreatedObjects& theTreatedObjects ) const
93 QStringList aResList = dumpObjectCreation( theTreatedObjects );
94 QString aName = GetObjPyName();
96 Handle(HYDROData_Polyline3D) aRefGideLine = GetGuideLine();
97 setPythonReferenceObject( thePyScriptPath, theTreatedObjects, aResList, aRefGideLine, "SetGuideLine" );
99 Handle(HYDROData_Profile) aRefProfile = GetProfile();
100 setPythonReferenceObject( thePyScriptPath, theTreatedObjects, aResList, aRefProfile, "SetProfile" );
102 aResList << QString( "%1.SetEquiDistance( %2 )" ).arg( aName ).arg( GetEquiDistance() );
104 aResList << QString( "" );
105 aResList << QString( "%1.Update()" ).arg( aName );
106 aResList << QString( "" );
111 HYDROData_SequenceOfObjects HYDROData_Channel::GetAllReferenceObjects() const
113 HYDROData_SequenceOfObjects aResSeq = HYDROData_ArtificialObject::GetAllReferenceObjects();
115 Handle(HYDROData_Polyline3D) aGuideLine = GetGuideLine();
116 if ( !aGuideLine.IsNull() )
117 aResSeq.Append( aGuideLine );
119 Handle(HYDROData_Profile) aProfile = GetProfile();
120 if ( !aProfile.IsNull() )
121 aResSeq.Append( aProfile );
126 bool HYDROData_Channel::CreatePresentations( const Handle(HYDROData_Polyline3D)& theGuideLine,
127 const Handle(HYDROData_Profile)& theProfile,
128 PrsDefinition& thePrs,
129 double theEquiDistance,
132 return internalCreatePresentations( true, theGuideLine, theProfile, TopoDS_Wire(), gp_Pnt(), thePrs, theEquiDistance, ReverseXCoord);
135 bool HYDROData_Channel::CreatePresentations( const Handle(HYDROData_Polyline3D)& theGuideLine,
139 PrsDefinition& thePrs,
140 double theEquiDistance,
143 Handle(HYDROData_Profile) theProfile;
147 gp_Pnt2d A1(-0.75*LC, coteZ-LC/2.0), A2(0.75*LC, coteZ -LC/2.0), B1(-LC/2.0, coteZ), B2(LC/2.0, coteZ), C(0, coteZ+deltaZ);
148 gp_Pnt A1_3d(A1.X(), 0, A1.Y()), A2_3d(A2.X(), 0, A2.Y()), B1_3d(B1.X(), 0, B1.Y()), B2_3d(B2.X(), 0, B2.Y()), C_3d(C.X(), 0, C.Y());
149 BRepLib_MakePolygon PM;
156 return internalCreatePresentations( false, theGuideLine, theProfile, W, MP, thePrs, theEquiDistance, ReverseXCoord);
159 bool HYDROData_Channel::internalCreatePresentations( bool mode,
160 const Handle(HYDROData_Polyline3D)& theGuideLine,
161 const Handle(HYDROData_Profile)& theProfile,
162 const TopoDS_Wire& theProfWire,
163 const gp_Pnt& theMiddlePnt,
164 PrsDefinition& thePrs,
165 double theEquiDistance,
168 // Check input parameters
169 if ( theGuideLine.IsNull() )
173 if (mode && theProfile.IsNull() || !mode && theProfWire.IsNull() )
176 TopoDS_Wire aPathWire = TopoDS::Wire( theGuideLine->GetShape3D() );
177 TopoDS_Wire aProfileWire = theProfWire;
178 //ignore ReverseXCoord if mode is false
182 aProfileWire = TopoDS::Wire( theProfile->GetShape3D(false, false) ); //temp force rebuild
184 aProfileWire = TopoDS::Wire( theProfile->GetShape3D(true, true));
186 if ( aPathWire.IsNull() || aProfileWire.IsNull() ) {
191 std::string brepName = "guideline_";
192 brepName += theGuideLine->GetName().toStdString();
194 BRepTools::Write( aPathWire, brepName.c_str() );
195 brepName = "profile_";
196 brepName += theGuideLine->GetName().toStdString();
198 BRepTools::Write( aProfileWire, brepName.c_str() );
202 Handle(HYDROData_PolylineXY) aPolylineXY = theGuideLine->GetPolylineXY();
203 if ( aPolylineXY.IsNull() ) {
208 HYDROData_IPolyline::SectionType aSectionType = aPolylineXY->GetSectionType( 0 );
209 HYDROData_IPolyline::PointsList aPolylinePoints = aPolylineXY->GetPoints( 0 );
210 int aNbPoints = aPolylinePoints.Length();
213 HYDROData_Polyline3D::Polyline3DPoints aPolylinePoints3D = theGuideLine->GetPoints( theEquiDistance );
214 int aNbPoints = aPolylinePoints3D.Length();
218 // Get tangent in each point of the guide line ( 2D )
219 TColgp_Array1OfDir aTangents( 1, aNbPoints );
221 HYDROData_IPolyline::SectionType aSectionType = aPolylineXY->GetSectionType( 0 );
223 if( aSectionType == HYDROData_IPolyline::SECTION_POLYLINE )
225 for ( int i = 1; i <= aNbPoints; ++i ) {
226 gp_XYZ aPnt = aPolylinePoints3D.Value( i );
230 aPrevPnt = aPolylinePoints3D.Value( i - 1 );
235 if ( i < aNbPoints ) {
236 gp_XYZ aNextPnt = aPolylinePoints3D.Value( i + 1 );
239 gp_Vec anEdgeVec( aPnt, aNextPnt );
244 gp_Vec aPrevVec( aPrevPnt, aPnt );
245 aDir = aPrevVec.Normalized() + anEdgeVec.Normalized();
248 aDir = gp_Vec( aPrevPnt, aPnt );
251 aTangents.SetValue( i, aDir );
254 // Get curve from the first edge ( 2D )
255 TopTools_SequenceOfShape anEdges;
256 HYDROData_ShapesTool::ExploreShapeToShapes( aPolylineXY->GetShape(), TopAbs_EDGE, anEdges );
257 Standard_Real aStart, anEnd;
259 Handle(Geom_Curve) aCurve = BRep_Tool::Curve( TopoDS::Edge( anEdges.First() ), aStart, anEnd );
260 GeomAPI_ProjectPointOnCurve aProject;
263 for ( int i = 1; i <= aNbPoints; ++i ) {
264 gp_XYZ aPointToTest = aPolylinePoints3D.Value( i );
265 aPointToTest.SetZ( 0. );
267 aProject.Init( aPointToTest, aCurve );
268 Quantity_Parameter aParam = aProject.LowerDistanceParameter();
271 aCurve->D1( aParam, aPnt, aDir);
273 aTangents.SetValue( i, aDir );
277 // Get the profile middle point ( 3D )
278 gp_Pnt aMiddlePoint = theMiddlePnt;
281 aMiddlePoint = theProfile->GetMiddlePoint( true );
284 // Translate the profile to each point on the guide line ( 3D )
285 Handle(TColgp_HArray1OfPnt) anArrayOfFPnt = new TColgp_HArray1OfPnt(1, aNbPoints );
286 Handle(TColgp_HArray1OfPnt) anArrayOfLPnt = new TColgp_HArray1OfPnt(1, aNbPoints );
287 Handle(TopTools_HArray1OfShape) anArrOfProfiles = new TopTools_HArray1OfShape( 1, aNbPoints );
289 for ( int i = 1; i <= aNbPoints; ++i ) {
290 // Get point on the guide line
291 gp_Pnt aPointOnGuide( aPolylinePoints3D.Value( i ) );
293 // Define translation and rotation:
294 gp_Trsf Translation, Rotation;
297 Translation.SetTranslation( aMiddlePoint, aPointOnGuide );
298 TopoDS_Wire aTransformedProfile =
299 TopoDS::Wire( BRepBuilderAPI_Transform( aProfileWire, Translation, Standard_True ) );
302 gp_Vec aVertical( 0., 0., 1. );
303 TopoDS_Vertex aLeftVertex, aRightVertex;
304 TopExp::Vertices( aTransformedProfile, aLeftVertex, aRightVertex );
305 gp_Pnt aLeftPoint = BRep_Tool::Pnt( aLeftVertex );
306 gp_Pnt aRightPoint = BRep_Tool::Pnt( aRightVertex );
307 gp_Vec aLeftToRight( aLeftPoint, aRightPoint);
308 gp_Vec NormalToProfile = aVertical ^ aLeftToRight;
310 gp_Vec aDir = aTangents.Value( i );
311 gp_Vec AxisOfRotation = NormalToProfile ^ aDir;
312 if (AxisOfRotation.Magnitude() <= gp::Resolution()) {
313 if ( aVertical * aLeftToRight < 0. ) {
314 gp_Ax1 theVertical(aPointOnGuide, gp::DZ() );
315 Rotation.SetRotation(theVertical, M_PI);
318 gp_Ax1 theAxis(aPointOnGuide, AxisOfRotation);
319 Standard_Real theAngle = NormalToProfile.AngleWithRef(aDir, AxisOfRotation);
320 Rotation.SetRotation(theAxis, theAngle);
323 aTransformedProfile = TopoDS::Wire(BRepBuilderAPI_Transform( aTransformedProfile, Rotation, Standard_True) );
325 // Get the first and the last points of the transformed profile
326 TopoDS_Vertex V1, V2;
327 TopExp::Vertices( aTransformedProfile, V1, V2 );
330 anArrayOfFPnt->SetValue( i, BRep_Tool::Pnt( V1 ) );
331 anArrayOfLPnt->SetValue( i, BRep_Tool::Pnt( V2 ) );
333 anArrOfProfiles->SetValue( i, aTransformedProfile );
336 // Create presentation
337 HYDROData_Stream::PrsDefinition aPrs;
338 Handle(TopTools_HArray1OfShape) anArrOf2DProfiles; // we don't need 2D profiles for channel/digue presentation
340 HYDROData_Stream::CreatePresentations( anArrayOfFPnt, anArrayOfLPnt, anArrOfProfiles, aPrs );
341 thePrs.myInlet = aPrs.myInlet;
342 thePrs.myOutlet = aPrs.myOutlet;
343 thePrs.myLeftBank = aPrs.myLeftBank;
344 thePrs.myRightBank = aPrs.myRightBank;
345 thePrs.myPrs2D = TopoDS::Face(aPrs.myPrs2D);
346 thePrs.myPrs3D = aPrs.myPrs3D;
348 //thePrs.myPrs2D = TopoDS::Face( aPrs.myPrs2D );
349 //BRepBuilderAPI_MakeWire aMakeWire( aPrs.myLeftBank ) ;
350 //thePrs.myLeftBank = aMakeWire.Wire();
351 //aMakeWire = BRepBuilderAPI_MakeWire( aPrs.myRightBank );
352 //thePrs.myRightBank = aMakeWire.Wire();
353 //aMakeWire = BRepBuilderAPI_MakeWire( aPrs.myInlet );
354 //thePrs.myInlet = aMakeWire.Wire();
355 //aMakeWire = BRepBuilderAPI_MakeWire( aPrs.myOutlet );
356 //thePrs.myOutlet = aMakeWire.Wire();
361 void HYDROData_Channel::Update()
363 HYDROData_ArtificialObject::Update();
365 Handle(HYDROData_Polyline3D) aGuideLine = GetGuideLine();
367 PrsDefinition aResultPrs;
368 double anEquiDistance = GetEquiDistance();
370 bool invDirection = false;
371 Handle(HYDROData_IAltitudeObject) anObjAltitude = GetAltitudeObject();
372 Handle(HYDROData_ChannelAltitude) aChannelAlt = Handle(HYDROData_ChannelAltitude)::DownCast(anObjAltitude);
373 if (!aChannelAlt.IsNull())
374 invDirection = aChannelAlt->GetInvertDirection();
376 if (GetProfileMode())
378 Handle(HYDROData_Profile) aProfile = GetProfile();
379 if ( !CreatePresentations( aGuideLine, aProfile, aResultPrs, anEquiDistance, invDirection ) )
384 double lc = GetLCValue();
385 double deltaz = GetDeltaZValue();
386 double cotez = GetCoteZValue();
387 if ( !CreatePresentations( aGuideLine, lc, deltaz, cotez, aResultPrs, anEquiDistance, invDirection ) )
391 SetShape3D( aResultPrs.myPrs3D );
392 SetTopShape( aResultPrs.myPrs2D );
394 // Create groups for channel
395 TopTools_SequenceOfShape aLeftBankEdges;
396 HYDROData_ShapesTool::ExploreShapeToShapes( aResultPrs.myLeftBank, TopAbs_EDGE, aLeftBankEdges );
398 TopTools_SequenceOfShape aRightBankEdges;
399 HYDROData_ShapesTool::ExploreShapeToShapes( aResultPrs.myRightBank, TopAbs_EDGE, aRightBankEdges );
401 TopTools_SequenceOfShape anInletEdges;
402 HYDROData_ShapesTool::ExploreShapeToShapes( aResultPrs.myInlet, TopAbs_EDGE, anInletEdges );
404 TopTools_SequenceOfShape anOutletEdges;
405 HYDROData_ShapesTool::ExploreShapeToShapes( aResultPrs.myOutlet, TopAbs_EDGE, anOutletEdges );
407 RemoveGroupObjects();
408 QString aLeftGroupName = GetName() + "_Left_Bank";
409 Handle(HYDROData_ShapesGroup) aLeftGroup = createGroupObject();
410 aLeftGroup->SetName( aLeftGroupName );
411 aLeftGroup->SetShapes( aLeftBankEdges );
413 QString aRightGroupName = GetName() + "_Right_Bank";
415 Handle(HYDROData_ShapesGroup) aRightGroup = createGroupObject();
416 aRightGroup->SetName( aRightGroupName );
417 aRightGroup->SetShapes( aRightBankEdges );
419 QString anInGroupName = GetName() + "_Inlet";
421 Handle(HYDROData_ShapesGroup) anInGroup = createGroupObject();
422 anInGroup->SetName( anInGroupName );
423 anInGroup->SetShapes( anInletEdges );
425 QString anOutGroupName = GetName() + "_Outlet";
427 Handle(HYDROData_ShapesGroup) anOutGroup = createGroupObject();
428 anOutGroup->SetName( anOutGroupName );
429 anOutGroup->SetShapes( anOutletEdges );
432 bool HYDROData_Channel::IsHas2dPrs() const
437 QColor HYDROData_Channel::DefaultFillingColor() const
439 return QColor( Qt::blue );
442 QColor HYDROData_Channel::DefaultBorderColor() const
444 return QColor( Qt::transparent );
447 bool HYDROData_Channel::SetGuideLine( const Handle(HYDROData_Polyline3D)& theGuideLine )
449 Handle(HYDROData_Polyline3D) aPrevGuideLine = GetGuideLine();
451 if ( theGuideLine.IsNull() )
454 return !aPrevGuideLine.IsNull();
457 if ( IsEqual( aPrevGuideLine, theGuideLine ) )
460 TopoDS_Wire aHydraulicWire = TopoDS::Wire( theGuideLine->GetTopShape() );
461 if ( aHydraulicWire.IsNull() )
462 return false; // The polyline must be a single wire
464 SetReferenceObject( theGuideLine, DataTag_GuideLine );
466 // Indicate model of the need to update the chanel presentation
472 Handle(HYDROData_Polyline3D) HYDROData_Channel::GetGuideLine() const
474 return Handle(HYDROData_Polyline3D)::DownCast(
475 GetReferenceObject( DataTag_GuideLine ) );
478 void HYDROData_Channel::RemoveGuideLine()
480 Handle(HYDROData_Polyline3D) aPrevGuideLine = GetGuideLine();
481 if ( aPrevGuideLine.IsNull() )
484 ClearReferenceObjects( DataTag_GuideLine );
486 // Indicate model of the need to update the chanel presentation
490 bool HYDROData_Channel::SetProfile( const Handle(HYDROData_Profile)& theProfile )
492 Handle(HYDROData_Profile) aPrevProfile = GetProfile();
494 if ( theProfile.IsNull() )
497 return !aPrevProfile.IsNull();
500 if ( IsEqual( aPrevProfile, theProfile ) )
503 SetReferenceObject( theProfile, DataTag_Profile );
505 // Indicate model of the need to update the chanel presentation
511 Handle(HYDROData_Profile) HYDROData_Channel::GetProfile() const
513 return Handle(HYDROData_Profile)::DownCast(
514 GetReferenceObject( DataTag_Profile ) );
517 void HYDROData_Channel::RemoveProfile()
519 Handle(HYDROData_Profile) aPrevProfile = GetProfile();
520 if ( aPrevProfile.IsNull() )
523 ClearReferenceObjects( DataTag_Profile );
525 // Indicate model of the need to update the chanel presentation
529 ObjectKind HYDROData_Channel::getAltitudeObjectType() const
531 //DEBTRACE("HYDROData_Channel::getAltitudeObjectType");
532 return KIND_CHANNEL_ALTITUDE;
533 //return KIND_STREAM_ALTITUDE;
536 TopoDS_Shape HYDROData_Channel::GetLeftShape() const
538 HYDROData_SequenceOfObjects aGroups = GetGroups();
539 return HYDROData_Tool::getFirstShapeFromGroup( aGroups, 1);
542 TopoDS_Shape HYDROData_Channel::GetRightShape() const
544 HYDROData_SequenceOfObjects aGroups = GetGroups();
545 return HYDROData_Tool::getFirstShapeFromGroup( aGroups, 2);
548 void HYDROData_Channel::SetEquiDistance( double theEquiDistance )
550 double anEquiDistance = theEquiDistance > 0 ? theEquiDistance : 1E-3;
551 SetDouble( DataTag_EquiDistance, theEquiDistance );
554 double HYDROData_Channel::GetEquiDistance() const
556 return GetDouble( DataTag_EquiDistance, 1.0 );
559 void HYDROData_Channel::SetLCValue( double val )
561 SetDouble( DataTag_LC, val );
564 double HYDROData_Channel::GetLCValue() const
566 return GetDouble( DataTag_LC, 1.0 );
569 void HYDROData_Channel::SetDeltaZValue( double val )
571 SetDouble( DataTag_DeltaZ, val );
574 double HYDROData_Channel::GetDeltaZValue() const
576 return GetDouble( DataTag_DeltaZ, 1.0 );
579 void HYDROData_Channel::SetCoteZValue( double val )
581 SetDouble( DataTag_CoteZ, val );
584 double HYDROData_Channel::GetCoteZValue() const
586 return GetDouble( DataTag_CoteZ, 1.0 );
589 void HYDROData_Channel::SetProfileMode( bool mode )
591 SetInteger( DataTag_ProfileMode, (bool)mode );
594 bool HYDROData_Channel::GetProfileMode() const
596 return (bool)GetInteger( DataTag_ProfileMode, 1 );