X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROData%2FHYDROData_Channel.cxx;h=14ac7fb6e94bda367fe9f2617240d7c84a0aadf7;hb=d398a8be8e0b0259b476b358d53d234ce4c82379;hp=9499aea834fd372eb43cfc4eaea9cfdbc9185d4b;hpb=f9b8b591033deed2dbf24dceaa860faf1d323749;p=modules%2Fhydro.git diff --git a/src/HYDROData/HYDROData_Channel.cxx b/src/HYDROData/HYDROData_Channel.cxx index 9499aea8..14ac7fb6 100644 --- a/src/HYDROData/HYDROData_Channel.cxx +++ b/src/HYDROData/HYDROData_Channel.cxx @@ -1,3 +1,20 @@ +// Copyright (C) 2014-2015 EDF-R&D +// This library is free software; you can redistribute it and/or +// modify it under the terms of the GNU Lesser General Public +// License as published by the Free Software Foundation; either +// version 2.1 of the License, or (at your option) any later version. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public +// License along with this library; if not, write to the Free Software +// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +// +// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com +// #include "HYDROData_Channel.h" @@ -8,19 +25,41 @@ #include "HYDROData_Projection.h" #include "HYDROData_ShapesGroup.h" #include "HYDROData_ShapesTool.h" +#include "HYDROData_Stream.h" +#include "HYDROData_Tool.h" + +#include #include #include #include +#include + +#include + +#include +#include + +#include + +#include + #include +#include + +#include +#include + +#include + +#include #include #include #include -#include -//#define DEB_CHANNEL 1 +#define DEB_CHANNEL 1 #ifdef DEB_CHANNEL #include #endif @@ -28,14 +67,15 @@ #include #include -#define PYTHON_CHANNEL_ID "KIND_CHANNEL" +#define _DEVDEBUG_ +#include "HYDRO_trace.hxx" IMPLEMENT_STANDARD_HANDLE(HYDROData_Channel,HYDROData_ArtificialObject) IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Channel,HYDROData_ArtificialObject) HYDROData_Channel::HYDROData_Channel() -: HYDROData_ArtificialObject() +: HYDROData_ArtificialObject( Geom_3d ) { } @@ -43,24 +83,23 @@ HYDROData_Channel::~HYDROData_Channel() { } -QStringList HYDROData_Channel::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const +QStringList HYDROData_Channel::DumpToPython( const QString& thePyScriptPath, + MapOfTreatedObjects& theTreatedObjects ) const { - QStringList aResList; + QStringList aResList = dumpObjectCreation( theTreatedObjects ); + QString aName = GetObjPyName(); - Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab ); - if ( aDocument.IsNull() ) - return aResList; + Handle(HYDROData_Polyline3D) aRefGideLine = GetGuideLine(); + setPythonReferenceObject( thePyScriptPath, theTreatedObjects, aResList, aRefGideLine, "SetGuideLine" ); - QString aDocName = aDocument->GetDocPyName(); - QString aChannelName = GetName(); + Handle(HYDROData_Profile) aRefProfile = GetProfile(); + setPythonReferenceObject( thePyScriptPath, theTreatedObjects, aResList, aRefProfile, "SetProfile" ); - aResList << QString( "%1 = %2.CreateObject( %3 );" ) - .arg( aChannelName ).arg( aDocName ).arg( PYTHON_CHANNEL_ID ); - aResList << QString( "%1.SetName( \"%2\" );" ) - .arg( aChannelName ).arg( aChannelName ); - aResList << QString( "" ); + aResList << QString( "%1.SetEquiDistance( %2 )" ).arg( aName ).arg( GetEquiDistance() ); - // TODO + aResList << QString( "" ); + aResList << QString( "%1.Update()" ).arg( aName ); + aResList << QString( "" ); return aResList; } @@ -80,137 +119,251 @@ HYDROData_SequenceOfObjects HYDROData_Channel::GetAllReferenceObjects() const return aResSeq; } -TopoDS_Shape HYDROData_Channel::GetTopShape() const +bool HYDROData_Channel::CreatePresentations( const Handle(HYDROData_Polyline3D)& theGuideLine, + const Handle(HYDROData_Profile)& theProfile, + PrsDefinition& thePrs, + double theEquiDistance ) { - return getTopShape(); -} + // Check input parameters + if ( theGuideLine.IsNull() || theProfile.IsNull() ) { + return false; + } -TopoDS_Shape HYDROData_Channel::GetShape3D() const -{ - return getShape3D(); -} + TopoDS_Wire aPathWire = TopoDS::Wire( theGuideLine->GetShape3D() ); + TopoDS_Wire aProfileWire = TopoDS::Wire( theProfile->GetShape3D() ); + if ( aPathWire.IsNull() || aProfileWire.IsNull() ) { + return false; + } -void HYDROData_Channel::Update() -{ - HYDROData_ArtificialObject::Update(); +#ifdef DEB_CHANNEL + std::string brepName = "guideline_"; + brepName += theGuideLine->GetName().toStdString(); + brepName += ".brep"; + BRepTools::Write( aPathWire, brepName.c_str() ); + brepName = "profile_"; + brepName += theGuideLine->GetName().toStdString(); + brepName += ".brep"; + BRepTools::Write( aProfileWire, brepName.c_str() ); +#endif - Handle(HYDROData_Polyline3D) aGuideLine = GetGuideLine(); - Handle(HYDROData_Profile) aProfile = GetProfile(); - if ( aGuideLine.IsNull() || aProfile.IsNull() ) - return; + // Pre-processing + Handle(HYDROData_PolylineXY) aPolylineXY = theGuideLine->GetPolylineXY(); + if ( aPolylineXY.IsNull() ) { + return false; + } - // build 3d shape - TopoDS_Wire aPathWire = TopoDS::Wire(aGuideLine->GetShape3D()); - if(aPathWire.IsNull()) - return; - TopoDS_Wire aProfileWire = TopoDS::Wire( aProfile->GetShape3D() ); - if(aProfileWire.IsNull()) - return; + /* + HYDROData_IPolyline::SectionType aSectionType = aPolylineXY->GetSectionType( 0 ); + HYDROData_IPolyline::PointsList aPolylinePoints = aPolylineXY->GetPoints( 0 ); + int aNbPoints = aPolylinePoints.Length(); + */ - BRepOffsetAPI_MakePipeShell aMkSweep(aPathWire); - aMkSweep.Add(aProfileWire,Standard_True, Standard_True); - aMkSweep.SetTransitionMode(BRepBuilderAPI_RightCorner); - //aMkSweep.SetMode(Standard_True); - aMkSweep.Build(); - if(aMkSweep.IsDone()) { - const TopoDS_Shape& aChannel = aMkSweep.Shape(); - BRepCheck_Analyzer aCheck(aChannel); - if(aCheck.IsValid()) - { - //BRepTools::Write(aChannel, "ChanV.brep"); - SetShape3D( aMkSweep.Shape()); - } else { -#ifdef DEB_CHANNEL - cout <<"NOT VALID" <GetPoints( theEquiDistance ); + int aNbPoints = aPolylinePoints3D.Length(); + if ( aNbPoints < 2 ) + return false; + + // Get tangent in each point of the guide line ( 2D ) + TColgp_Array1OfDir aTangents( 1, aNbPoints ); + + HYDROData_IPolyline::SectionType aSectionType = aPolylineXY->GetSectionType( 0 ); + + if( aSectionType == HYDROData_IPolyline::SECTION_POLYLINE ) + { + for ( int i = 1; i <= aNbPoints; ++i ) { + gp_XYZ aPnt = aPolylinePoints3D.Value( i ); + aPnt.SetZ( 0. ); + gp_XYZ aPrevPnt; + if ( i > 1 ) { + aPrevPnt = aPolylinePoints3D.Value( i - 1 ); + aPrevPnt.SetZ( 0. ); + } + + gp_Vec aDir; + if ( i < aNbPoints ) { + gp_XYZ aNextPnt = aPolylinePoints3D.Value( i + 1 ); + aNextPnt.SetZ( 0. ); + + gp_Vec anEdgeVec( aPnt, aNextPnt ); + + if ( i == 1 ) { + aDir = anEdgeVec; + } else { + gp_Vec aPrevVec( aPrevPnt, aPnt ); + aDir = aPrevVec.Normalized() + anEdgeVec.Normalized(); + } + } else { + aDir = gp_Vec( aPrevPnt, aPnt ); + } + + aTangents.SetValue( i, aDir ); + } + } else { + // Get curve from the first edge ( 2D ) + TopTools_SequenceOfShape anEdges; + HYDROData_ShapesTool::ExploreShapeToShapes( aPolylineXY->GetShape(), TopAbs_EDGE, anEdges ); + Standard_Real aStart, anEnd; + + Handle(Geom_Curve) aCurve = BRep_Tool::Curve( TopoDS::Edge( anEdges.First() ), aStart, anEnd ); + GeomAPI_ProjectPointOnCurve aProject; + + // Get tangents + for ( int i = 1; i <= aNbPoints; ++i ) { + gp_XYZ aPointToTest = aPolylinePoints3D.Value( i ); + aPointToTest.SetZ( 0. ); + + aProject.Init( aPointToTest, aCurve ); + Quantity_Parameter aParam = aProject.LowerDistanceParameter(); + gp_Pnt aPnt; + gp_Vec aDir; + aCurve->D1( aParam, aPnt, aDir); + + aTangents.SetValue( i, aDir ); } } - TopoDS_Shape a3dShape = GetShape3D(); - if ( a3dShape.IsNull() ) - return; + // Get the profile middle point ( 3D ) + gp_Pnt aMiddlePoint( theProfile->GetMiddlePoint( true ) ); + + // Translate the profile to each point on the guide line ( 3D ) + Handle(TColgp_HArray1OfPnt) anArrayOfFPnt = new TColgp_HArray1OfPnt(1, aNbPoints ); + Handle(TColgp_HArray1OfPnt) anArrayOfLPnt = new TColgp_HArray1OfPnt(1, aNbPoints ); + Handle(TopTools_HArray1OfShape) anArrOfProfiles = new TopTools_HArray1OfShape( 1, aNbPoints ); + + for ( int i = 1; i <= aNbPoints; ++i ) { + // Get point on the guide line + gp_Pnt aPointOnGuide( aPolylinePoints3D.Value( i ) ); + + // Define translation and rotation: + gp_Trsf Translation, Rotation; + + // Translation + Translation.SetTranslation( aMiddlePoint, aPointOnGuide ); + TopoDS_Wire aTransformedProfile = + TopoDS::Wire( BRepBuilderAPI_Transform( aProfileWire, Translation, Standard_True ) ); + + // Rotation + gp_Vec aVertical( 0., 0., 1. ); + TopoDS_Vertex aLeftVertex, aRightVertex; + TopExp::Vertices( aTransformedProfile, aLeftVertex, aRightVertex ); + gp_Pnt aLeftPoint = BRep_Tool::Pnt( aLeftVertex ); + gp_Pnt aRightPoint = BRep_Tool::Pnt( aRightVertex ); + gp_Vec aLeftToRight( aLeftPoint, aRightPoint); + gp_Vec NormalToProfile = aVertical ^ aLeftToRight; + + gp_Vec aDir = aTangents.Value( i ); + gp_Vec AxisOfRotation = NormalToProfile ^ aDir; + if (AxisOfRotation.Magnitude() <= gp::Resolution()) { + if ( aVertical * aLeftToRight < 0. ) { + gp_Ax1 theVertical(aPointOnGuide, gp::DZ() ); + Rotation.SetRotation(theVertical, M_PI); + } + } else { + gp_Ax1 theAxis(aPointOnGuide, AxisOfRotation); + Standard_Real theAngle = NormalToProfile.AngleWithRef(aDir, AxisOfRotation); + Rotation.SetRotation(theAxis, theAngle); + } - // build 2d shape -- temporary solution!! - TopoDS_Face aProj = HYDROData_Projection::MakeProjection( a3dShape ); - SetTopShape( aProj ); - if ( aProj.IsNull() ) - return; + aTransformedProfile = TopoDS::Wire(BRepBuilderAPI_Transform( aTransformedProfile, Rotation, Standard_True) ); - // Create the channel groups - TopoDS_Shape aFirstShape = aMkSweep.FirstShape(); - TopoDS_Shape aLastShape = aMkSweep.LastShape(); - if ( aFirstShape.IsNull() || aFirstShape.ShapeType() != TopAbs_WIRE || - aLastShape.IsNull() || aLastShape.ShapeType() != TopAbs_WIRE ) - return; + // Get the first and the last points of the transformed profile + TopoDS_Vertex V1, V2; + TopExp::Vertices( aTransformedProfile, V1, V2 ); + + // Fill the data + anArrayOfFPnt->SetValue( i, BRep_Tool::Pnt( V1 ) ); + anArrayOfLPnt->SetValue( i, BRep_Tool::Pnt( V2 ) ); + + anArrOfProfiles->SetValue( i, aTransformedProfile ); + } - TopoDS_Wire anInletWire = TopoDS::Wire( aFirstShape ); - TopoDS_Wire anOutletWire = TopoDS::Wire( aLastShape ); + // Create presentation + HYDROData_Stream::PrsDefinition aPrs; + Handle(TopTools_HArray1OfShape) anArrOf2DProfiles; // we don't need 2D profiles for channel/digue presentation + bool aRes = HYDROData_Stream::CreatePresentations( anArrayOfFPnt, anArrayOfLPnt, + anArrOfProfiles, anArrOf2DProfiles, aPrs ); + if ( aRes ) { + thePrs.myPrs3D = aPrs.myPrs3D; + thePrs.myPrs2D = TopoDS::Face( aPrs.myPrs2D ); + BRepBuilderAPI_MakeWire aMakeWire( aPrs.myLeftBank ) ; + thePrs.myLeftBank = aMakeWire.Wire(); + aMakeWire = BRepBuilderAPI_MakeWire( aPrs.myRightBank ); + thePrs.myRightBank = aMakeWire.Wire(); + aMakeWire = BRepBuilderAPI_MakeWire( aPrs.myInlet ); + thePrs.myInlet = aMakeWire.Wire(); + aMakeWire = BRepBuilderAPI_MakeWire( aPrs.myOutlet ); + thePrs.myOutlet = aMakeWire.Wire(); + } + + return aRes; +} + +void HYDROData_Channel::Update() +{ + HYDROData_ArtificialObject::Update(); - TopoDS_Vertex anInletFirstVert, anInletLastVert; - TopExp::Vertices( anInletWire, anInletFirstVert, anInletLastVert ); + Handle(HYDROData_Polyline3D) aGuideLine = GetGuideLine(); + Handle(HYDROData_Profile) aProfile = GetProfile(); - const TopTools_ListOfShape& aGeneratedLeftEdges = aMkSweep.Generated( anInletFirstVert ); - const TopTools_ListOfShape& aGeneratedRightEdges = aMkSweep.Generated( anInletLastVert ); + PrsDefinition aResultPrs; + double anEquiDistance = GetEquiDistance(); + if ( !CreatePresentations( aGuideLine, aProfile, aResultPrs, anEquiDistance ) ) + return; - TopTools_SequenceOfShape aProjEdges; - HYDROData_ShapesTool::ExploreShapeToShapes( aProj, TopAbs_EDGE, aProjEdges ); + SetShape3D( aResultPrs.myPrs3D ); + SetTopShape( aResultPrs.myPrs2D ); - TopTools_SequenceOfShape aProjLeftEdges; - findEdges( aProjEdges, aGeneratedLeftEdges, aProjLeftEdges ); + // Create groups for channel + TopTools_SequenceOfShape aLeftBankEdges; + HYDROData_ShapesTool::ExploreShapeToShapes( aResultPrs.myLeftBank, TopAbs_EDGE, aLeftBankEdges ); - TopTools_SequenceOfShape aProjRightEdges; - findEdges( aProjEdges, aGeneratedRightEdges, aProjRightEdges ); + TopTools_SequenceOfShape aRightBankEdges; + HYDROData_ShapesTool::ExploreShapeToShapes( aResultPrs.myRightBank, TopAbs_EDGE, aRightBankEdges ); - TopTools_SequenceOfShape aProjInletEdges; - findEdges( aProjEdges, anInletWire, aProjInletEdges ); + TopTools_SequenceOfShape anInletEdges; + HYDROData_ShapesTool::ExploreShapeToShapes( aResultPrs.myInlet, TopAbs_EDGE, anInletEdges ); - TopTools_SequenceOfShape aProjOutletEdges; - findEdges( aProjEdges, anOutletWire, aProjOutletEdges ); + TopTools_SequenceOfShape anOutletEdges; + HYDROData_ShapesTool::ExploreShapeToShapes( aResultPrs.myOutlet, TopAbs_EDGE, anOutletEdges ); QString aLeftGroupName = GetName() + "_Left_Bank"; Handle(HYDROData_ShapesGroup) aLeftGroup = createGroupObject(); aLeftGroup->SetName( aLeftGroupName ); - aLeftGroup->SetShapes( aProjLeftEdges ); + aLeftGroup->SetShapes( aLeftBankEdges ); QString aRightGroupName = GetName() + "_Right_Bank"; Handle(HYDROData_ShapesGroup) aRightGroup = createGroupObject(); aRightGroup->SetName( aRightGroupName ); - aRightGroup->SetShapes( aProjRightEdges ); + aRightGroup->SetShapes( aRightBankEdges ); QString anInGroupName = GetName() + "_Inlet"; Handle(HYDROData_ShapesGroup) anInGroup = createGroupObject(); anInGroup->SetName( anInGroupName ); - anInGroup->SetShapes( aProjInletEdges ); + anInGroup->SetShapes( anInletEdges ); QString anOutGroupName = GetName() + "_Outlet"; Handle(HYDROData_ShapesGroup) anOutGroup = createGroupObject(); anOutGroup->SetName( anOutGroupName ); - anOutGroup->SetShapes( aProjOutletEdges ); + anOutGroup->SetShapes( anOutletEdges ); } -QColor HYDROData_Channel::DefaultFillingColor() +bool HYDROData_Channel::IsHas2dPrs() const { - return QColor( Qt::blue ); -} - -QColor HYDROData_Channel::DefaultBorderColor() -{ - return QColor( Qt::transparent ); + return true; } -QColor HYDROData_Channel::getDefaultFillingColor() const +QColor HYDROData_Channel::DefaultFillingColor() const { - return DefaultFillingColor(); + return QColor( Qt::blue ); } -QColor HYDROData_Channel::getDefaultBorderColor() const +QColor HYDROData_Channel::DefaultBorderColor() const { - return DefaultBorderColor(); + return QColor( Qt::transparent ); } bool HYDROData_Channel::SetGuideLine( const Handle(HYDROData_Polyline3D)& theGuideLine ) @@ -233,7 +386,7 @@ bool HYDROData_Channel::SetGuideLine( const Handle(HYDROData_Polyline3D)& theGui SetReferenceObject( theGuideLine, DataTag_GuideLine ); // Indicate model of the need to update the chanel presentation - SetToUpdate( true ); + Changed( Geom_3d ); return true; } @@ -253,7 +406,7 @@ void HYDROData_Channel::RemoveGuideLine() ClearReferenceObjects( DataTag_GuideLine ); // Indicate model of the need to update the chanel presentation - SetToUpdate( true ); + Changed( Geom_3d ); } bool HYDROData_Channel::SetProfile( const Handle(HYDROData_Profile)& theProfile ) @@ -272,7 +425,7 @@ bool HYDROData_Channel::SetProfile( const Handle(HYDROData_Profile)& theProfile SetReferenceObject( theProfile, DataTag_Profile ); // Indicate model of the need to update the chanel presentation - SetToUpdate( true ); + Changed( Geom_3d ); return true; } @@ -292,60 +445,35 @@ void HYDROData_Channel::RemoveProfile() ClearReferenceObjects( DataTag_Profile ); // Indicate model of the need to update the chanel presentation - SetToUpdate( true ); + Changed( Geom_3d ); } ObjectKind HYDROData_Channel::getAltitudeObjectType() const { - return KIND_OBSTACLE_ALTITUDE; + DEBTRACE("HYDROData_Channel::getAltitudeObjectType"); + return KIND_CHANNEL_ALTITUDE; + //return KIND_STREAM_ALTITUDE; } -void HYDROData_Channel::findEdges( const TopTools_SequenceOfShape& theInShapes, - const TopTools_ListOfShape& theEdges3D, - TopTools_SequenceOfShape& theOutShapes ) const +TopoDS_Shape HYDROData_Channel::GetLeftShape() const { - theOutShapes.Clear(); - if ( theEdges3D.IsEmpty() || theInShapes.IsEmpty() ) - return; - - const TopoDS_Shape& aFirstShape3D = theEdges3D.First(); - const TopoDS_Shape& aLastShape3D = theEdges3D.Last(); - - TopoDS_Vertex aFirstVert, aLastVert, aDummyVert; - if ( !HYDROData_ShapesTool::Vertices( aFirstShape3D, aFirstVert, aDummyVert ) || - !HYDROData_ShapesTool::Vertices( aLastShape3D, aDummyVert, aLastVert ) ) - return; - - bool isStarted = false; - for ( int i = 1, n = theInShapes.Length(); i <= n; ++i ) - { - const TopoDS_Shape& anInShape = theInShapes.Value( i ); - - TopoDS_Vertex aShapeFirstVert, aShapeLastVert; - if ( !HYDROData_ShapesTool::Vertices( anInShape, aShapeFirstVert, aShapeLastVert ) ) - continue; - - if ( !isStarted ) - { - isStarted = HYDROData_ShapesTool::IsVerticesEquals( aFirstVert, aShapeFirstVert, true ); - } - - if ( isStarted ) - { - theOutShapes.Append( anInShape ); + HYDROData_SequenceOfObjects aGroups = GetGroups(); + return HYDROData_Tool::getFirstShapeFromGroup( aGroups, 1); +} - if ( HYDROData_ShapesTool::IsVerticesEquals( aLastVert, aShapeLastVert, true ) ) - break; - } - } +TopoDS_Shape HYDROData_Channel::GetRightShape() const +{ + HYDROData_SequenceOfObjects aGroups = GetGroups(); + return HYDROData_Tool::getFirstShapeFromGroup( aGroups, 2); } -void HYDROData_Channel::findEdges( const TopTools_SequenceOfShape& theProjShapes, - const TopoDS_Shape& theEdge3D, - TopTools_SequenceOfShape& theOutShapes ) const +void HYDROData_Channel::SetEquiDistance( double theEquiDistance ) { - TopTools_ListOfShape aTmpList; - aTmpList.Append( theEdge3D ); - findEdges( theProjShapes, aTmpList, theOutShapes ); + double anEquiDistance = theEquiDistance > 0 ? theEquiDistance : 1E-3; + SetDouble( DataTag_EquiDistance, theEquiDistance ); } +double HYDROData_Channel::GetEquiDistance() const +{ + return GetDouble( DataTag_EquiDistance, 1.0 ); +}