+// 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_PolylineXY.h"
#include <BRep_Builder.hxx>
#include <BRepBuilderAPI_MakeEdge.hxx>
#include <BRepBuilderAPI_MakeWire.hxx>
+#include <BRepBuilderAPI_MakePolygon.hxx>
+#include <BRepBuilderAPI_MakeFace.hxx>
+#include <BRepOffsetAPI_NormalProjection.hxx>
+
+#include <GEOMBase.h>
#include <GeomAPI_ProjectPointOnCurve.hxx>
#include <GeomAdaptor_Curve.hxx>
#include <gp_Pnt.hxx>
#include <gp_XY.hxx>
+#include <gp_Pln.hxx>
#include <NCollection_Map.hxx>
#include <TColStd_Array1OfReal.hxx>
+#include <TDataStd_AsciiString.hxx>
#include <TDataStd_BooleanList.hxx>
#include <TDataStd_ExtStringList.hxx>
#include <TDataStd_IntegerList.hxx>
#include <TopExp_Explorer.hxx>
#include <ShapeAnalysis_FreeBounds.hxx>
#include <TopoDS.hxx>
+
#include <QColor>
#include <QPainterPath>
#include <QVariant>
static const Standard_GUID GUID_IS_UNEDITABLE("e5799736-9030-4051-91a4-2e58321fa153");
-#define PYTHON_POLYLINEXY_ID "KIND_POLYLINEXY"
-
const double LOCAL_SELECTION_TOLERANCE = 0.0001;
TCollection_AsciiString getUniqueSectionName( const NCollection_Sequence<TCollection_AsciiString>& theNamesSeq )
QStringList HYDROData_PolylineXY::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
{
- QStringList aResList;
+ QStringList aResList = dumpObjectCreation( theTreatedObjects );
+ QString aPolylineName = GetObjPyName();
- Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
- if ( aDocument.IsNull() )
- return aResList;
-
- QString aDocName = aDocument->GetDocPyName();
- QString aPolylineName = GetName();
+ // Set the wire color
+ QStringList aWireColorDef;
- aResList << QString( "%1 = %2.CreateObject( %3 );" )
- .arg( aPolylineName ).arg( aDocName ).arg( PYTHON_POLYLINEXY_ID );
- aResList << QString( "%1.SetName( \"%1\" );" ).arg( aPolylineName );
-
- // Set polilyne data
- NCollection_Sequence<TCollection_AsciiString> aSectNames;
- NCollection_Sequence<HYDROData_PolylineXY::SectionType> aSectTypes;
- NCollection_Sequence<bool> aSectClosures;
- GetSections( aSectNames, aSectTypes, aSectClosures );
+ QColor aWireColor = GetWireColor();
+ setPythonObjectColor( aWireColorDef, aWireColor, DefaultWireColor(), "SetWireColor" );
+
+ if ( !aWireColorDef.isEmpty() )
+ {
+ aResList << aWireColorDef;
+ aResList << QString( "" );
+ }
- for ( int i = 1, n = aSectNames.Size(); i <= n; ++i )
+ bool anIsEditable = IsEditable();
+ if ( !anIsEditable )
{
- const TCollection_AsciiString& aSectName = aSectNames.Value( i );
- const SectionType& aSectType = aSectTypes.Value( i );
- bool aSectClosure = aSectClosures.Value( i );
+ // If polyline is not editable we try to import the shape from geom
+ TCollection_AsciiString aGeomObjectEntry = GetGeomObjectEntry();
+ if ( !aGeomObjectEntry.IsEmpty() )
+ {
+ QString aSalomeObjName = HYDROData_Tool::GenerateNameForPython( theTreatedObjects, "polyline_sobj" );
+ aResList << QString( "%1 = salome.myStudy.FindObjectID( \"%2\" );" )
+ .arg( aSalomeObjName ).arg( aGeomObjectEntry.ToCString() );
- aResList << QString( "%1.AddSection( \"%2\", %3, %4 );" ).arg( aPolylineName )
- .arg( aSectName.ToCString() ).arg( aSectType ).arg( aSectClosure );
+ aResList << QString( "%1.ImportFromGeomIOR( %2.GetIOR() );" )
+ .arg( aPolylineName ).arg( aSalomeObjName );
- HYDROData_PolylineXY::PointsList aSectPointsList = GetPoints( i );
- for ( int k = 1, aNbPoints = aSectPointsList.Size(); k <= aNbPoints; ++k )
+ aResList << QString( "%1.SetGeomObjectEntry( \"%2\" );" )
+ .arg( aPolylineName ).arg( aGeomObjectEntry.ToCString() );
+ }
+ }
+ else
+ {
+ // Set polilyne data
+ NCollection_Sequence<TCollection_AsciiString> aSectNames;
+ NCollection_Sequence<HYDROData_PolylineXY::SectionType> aSectTypes;
+ NCollection_Sequence<bool> aSectClosures;
+ GetSections( aSectNames, aSectTypes, aSectClosures );
+
+ for ( int i = 1, n = aSectNames.Size(); i <= n; ++i )
{
- const Point& aSectPoint = aSectPointsList.Value( k );
+ const TCollection_AsciiString& aSectName = aSectNames.Value( i );
+ const SectionType& aSectType = aSectTypes.Value( i );
+ bool aSectClosure = aSectClosures.Value( i );
+
+ aResList << QString( "%1.AddSection( \"%2\", %3, %4 );" ).arg( aPolylineName )
+ .arg( aSectName.ToCString() ).arg( aSectType ).arg( aSectClosure );
- aResList << QString( "%1.AddPoint( %2, QPointF( %3, %4 ) );" ).arg( aPolylineName )
- .arg( i - 1 ).arg( aSectPoint.X() ).arg( aSectPoint.Y() );
+ HYDROData_IPolyline::PointsList aSectPointsList = GetPoints( i - 1 );
+ for ( int k = 1, aNbPoints = aSectPointsList.Size(); k <= aNbPoints; ++k )
+ {
+ const Point& aSectPoint = aSectPointsList.Value( k );
+
+ QString anXStr = QString::number( aSectPoint.X(), 'f', 2 );
+ QString anYStr = QString::number( aSectPoint.Y(), 'f', 2 );
+ aResList << QString( "%1.AddPoint( %2, gp_XY( %3, %4 ) );" ).arg( aPolylineName )
+ .arg( i - 1 ).arg( anXStr ).arg( anYStr );
+ }
}
}
+ aResList << QString( "" );
+ aResList << QString( "%1.Update();" ).arg( aPolylineName );
+ aResList << QString( "" );
return aResList;
}
return QColor( Qt::red );
}
+bool HYDROData_PolylineXY::ImportFromGeomIOR( const TCollection_AsciiString& theIOR )
+{
+ if ( theIOR.IsEmpty() )
+ return false;
+
+ TopoDS_Shape aShape = GEOMBase::GetShapeFromIOR( theIOR.ToCString() );
+ if ( aShape.IsNull() )
+ return false;
+
+ return ImportShape( aShape );
+}
+
+void HYDROData_PolylineXY::SetGeomObjectEntry( const TCollection_AsciiString& theEntry )
+{
+ TDataStd_AsciiString::Set( myLab.FindChild( DataTag_GeomObjectEntry ), theEntry );
+}
+
+TCollection_AsciiString HYDROData_PolylineXY::GetGeomObjectEntry() const
+{
+ TCollection_AsciiString aRes;
+
+ TDF_Label aLabel = myLab.FindChild( DataTag_GeomObjectEntry, false );
+ if ( !aLabel.IsNull() )
+ {
+ Handle(TDataStd_AsciiString) anAsciiStr;
+ if ( aLabel.FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
+ aRes = anAsciiStr->Get();
+ }
+
+ return aRes;
+}
+
TopoDS_Shape HYDROData_PolylineXY::GetShape() const
{
return getPolylineShape();
}
+bool HYDROData_PolylineXY::SetShape( const TopoDS_Shape& theShape )
+{
+ setPolylineShape( theShape );
+ return true;
+}
+
bool convertEdgeToSection( const TopoDS_Edge& theEdge,
NCollection_Sequence<TCollection_AsciiString>& theSectNames,
NCollection_Sequence<HYDROData_PolylineXY::SectionType>& theSectTypes,
}
else
{
- setPolylineShape( theShape );
+ TopoDS_Shape aShape = theShape;
+
+ if ( theShape.ShapeType() == TopAbs_EDGE )
+ {
+ // We make the wire from incoming edge because of other algorithms
+ // are waiting at least the wire from polyline
+ TopoDS_Edge anEdge = TopoDS::Edge( theShape );
+ BRepBuilderAPI_MakeWire aMakeWire( anEdge );
+ aMakeWire.Build();
+ if ( aMakeWire.IsDone() )
+ aShape = aMakeWire.Wire();
+ }
+
+ gp_Pln aPlane( gp_Pnt( 0, 0, 0 ), gp_Dir( 0, 0, 1 ) );
+ BRepBuilderAPI_MakeFace aMakeFace( aPlane );
+ aMakeFace.Build();
+ BRepOffsetAPI_NormalProjection aProj( aMakeFace.Face() );
+ aProj.Add( aShape );
+ aProj.Build();
+ TopoDS_Shape aResult;
+ if( aProj.IsDone() )
+ aResult = aProj.Shape();
+
+ setPolylineShape( aResult );
}
setEditable( anIsCanBeImported );
const bool& theIsClosed,
const NCollection_Sequence<gp_XYZ>& thePoints )
{
- BRepBuilderAPI_MakeWire aMakeWire;
-
+ TopoDS_Wire aWire;
if( theType == SECTION_POLYLINE )
{
- for( int i = 1, n = thePoints.Size(); i <= n; ++i )
+ int aNbPoints = thePoints.Length();
+ BRepBuilderAPI_MakePolygon aMakeWire;
+ for ( int i = 1, n = aNbPoints; i <= n ; ++i )
{
- const gp_XYZ& aFirstPoint = thePoints.Value( i );
-
- gp_XYZ aLastPoint;
- if ( i == n )
- {
- if( theIsClosed )
- aLastPoint = thePoints.Value( 1 );
- else
- break;
- }
- else
- {
- aLastPoint = thePoints.Value( i + 1 );
- }
-
- gp_Pnt aPnt1( aFirstPoint.X(), aFirstPoint.Y(), aFirstPoint.Z() );
- gp_Pnt aPnt2( aLastPoint.X(), aLastPoint.Y(), aLastPoint.Z() );
-
- if ( aPnt1.IsEqual( aPnt2, LOCAL_SELECTION_TOLERANCE ) )
- continue;
-
- TopoDS_Edge anEdge = BRepBuilderAPI_MakeEdge( aPnt1, aPnt2 ).Edge();
- aMakeWire.Add( anEdge );
+ gp_XYZ aPoint = thePoints.Value( i );
+ gp_Pnt aPnt( aPoint.X(), aPoint.Y(), aPoint.Z() );
+ aMakeWire.Add( aPnt );
}
+ if( theIsClosed && ( aNbPoints > 2 ) )
+ aMakeWire.Close();
+
+ if ( aMakeWire.IsDone() )
+ aWire = aMakeWire.Wire();
}
else //if( theType == PolylineSection::SECTION_SPLINE )
{
- HYDROData_BSplineOperation aBSpline( thePoints, theIsClosed, LOCAL_SELECTION_TOLERANCE );
+ BRepBuilderAPI_MakeWire aMakeWire;
- TopoDS_Edge anEdge = BRepBuilderAPI_MakeEdge( aBSpline.Curve() ).Edge();
- aMakeWire.Add( anEdge );
- }
+ if ( thePoints.Size() > 1 )
+ {
+ Handle(Geom_BSplineCurve) aCurve =
+ HYDROData_BSplineOperation::ComputeCurve( thePoints, theIsClosed, LOCAL_SELECTION_TOLERANCE );
- TopoDS_Wire aWire;
- aMakeWire.Build();
- if ( aMakeWire.IsDone() )
- aWire = aMakeWire;
+ TopoDS_Edge anEdge = BRepBuilderAPI_MakeEdge( aCurve ).Edge();
+ aMakeWire.Add( anEdge );
+ }
+ aMakeWire.Build();
+ if ( aMakeWire.IsDone() )
+ aWire = aMakeWire;
+ }
return aWire;
}
}
else
{
- HYDROData_BSplineOperation aBSpline( thePoints, theIsClosed, LOCAL_SELECTION_TOLERANCE );
- aBSpline.ComputePath( thePath );
+ Handle(Geom_BSplineCurve) aCurve =
+ HYDROData_BSplineOperation::ComputeCurve( thePoints, theIsClosed, LOCAL_SELECTION_TOLERANCE );
+ HYDROData_BSplineOperation::ComputePath( aCurve, thePath );
}
}
NCollection_Sequence<bool> aSectClosures;
GetSections( aSectNames, aSectTypes, aSectClosures );
- BRepBuilderAPI_MakeWire aMakeWire;
+ //BRepBuilderAPI_MakeWire aMakeWire;
TopTools_ListOfShape aSectionWiresList;
TopoDS_Wire aSectionWire = BuildWire( aSectionType, anIsSectionClosed, aPoints );
if ( !aSectionWire.IsNull() ) {
aSectionWiresList.Append( aSectionWire );
- aMakeWire.Add( aSectionWire );
+ //aMakeWire.Add( aSectionWire );
}
}
// all input wires in the <aSectionWiresList>
setPolylineShape( aResult );
}
+bool HYDROData_PolylineXY::IsHas2dPrs() const
+{
+ return true;
+}
+
bool HYDROData_PolylineXY::IsEditable() const
{
return !myLab.IsAttribute( GUID_IS_UNEDITABLE );
if ( aShape.IsNull() )
return anIsClosed;
- if ( aShape.ShapeType() == TopAbs_EDGE )
- {
- //For unrecognized curves from GEOM module
- anIsClosed = BRep_Tool::IsClosed( aShape );
- return anIsClosed;
- }
-
TopTools_SequenceOfShape aWires;
HYDROData_ShapesTool::ExploreShapeToShapes( aShape, TopAbs_WIRE, aWires );
aPointToTest = aPoint;
}
- HYDROData_BSplineOperation aBSpline( aPoints, anIsSectionClosed, LOCAL_SELECTION_TOLERANCE );
+ Handle(Geom_BSplineCurve) aCurve =
+ HYDROData_BSplineOperation::ComputeCurve( aPoints, anIsSectionClosed, LOCAL_SELECTION_TOLERANCE );
- Quantity_Parameter aFirstParam = aBSpline.Curve()->FirstParameter();
- Quantity_Parameter aSecondParam = aBSpline.Curve()->LastParameter();
+ Quantity_Parameter aFirstParam = aCurve->FirstParameter();
+ Quantity_Parameter aSecondParam = aCurve->LastParameter();
if ( thePointIndex != aSectNbPoints - 1 )
{
- GeomAPI_ProjectPointOnCurve aProject( aPointToTest, aBSpline.Curve() );
+ GeomAPI_ProjectPointOnCurve aProject( aPointToTest, aCurve );
aSecondParam = aProject.LowerDistanceParameter();
}
- GeomAdaptor_Curve anAdap( aBSpline.Curve() );
+ GeomAdaptor_Curve anAdap( aCurve );
aResDistance = GCPnts_AbscissaPoint::Length( anAdap, aFirstParam, aSecondParam );
}
{
PointsList aResList;
+ if( IsCustom() )
+ {
+ //TODO: make interpolation to fill the list
+ }
+
Handle(TDataStd_RealList) aListX, aListY;
getPointsLists( theSectionIndex, aListX, aListY, false );
if ( aListX.IsNull() || aListY.IsNull() || aListX->IsEmpty() )
return aPath;
}
+void HYDROData_PolylineXY::UpdateLocalCS( double theDx, double theDy )
+{
+ NCollection_Sequence<TCollection_AsciiString> aSectNames;
+ NCollection_Sequence<HYDROData_PolylineXY::SectionType> aSectTypes;
+ NCollection_Sequence<bool> aSectClosures;
+ GetSections( aSectNames, aSectTypes, aSectClosures );
+
+ gp_XY aDelta( theDx, theDy );
+ for ( int i = 0, aNbSects = aSectNames.Size(); i < aNbSects; i++ )
+ {
+ PointsList aPoints = GetPoints( i );
+ for( int j = 1, n = aPoints.Size(); j <= n; ++j )
+ {
+ Point& aPoint = aPoints.ChangeValue( j );
+ aPoint += aDelta;
+ }
+ SetPoints( i, aPoints );
+ }
+ SetToUpdate( true );
+}
+
+void HYDROData_PolylineXY::Transform( const QTransform& theTrsf )
+{
+ NCollection_Sequence<TCollection_AsciiString> aSectNames;
+ NCollection_Sequence<HYDROData_PolylineXY::SectionType> aSectTypes;
+ NCollection_Sequence<bool> aSectClosures;
+ GetSections( aSectNames, aSectTypes, aSectClosures );
+
+ for ( int i = 0, aNbSects = aSectNames.Size(); i < aNbSects; i++ ) {
+ PointsList aPoints = GetPoints( i );
+ for( int j = 1, n = aPoints.Size(); j <= n; ++j ) {
+ Point& aPoint = aPoints.ChangeValue( j );
+
+ QPointF aTrsfPoint = theTrsf.map( QPointF( aPoint.X(), aPoint.Y() ) );
+
+ aPoint.SetX( aTrsfPoint.x() );
+ aPoint.SetY( aTrsfPoint.y() );
+ }
+ SetPoints( i, aPoints );
+ }
+ Update();
+}
+
+bool HYDROData_PolylineXY::IsCustom() const
+{
+ bool isNull = getPolylineShape().IsNull();
+ int aNbPoints = 0;
+
+ //TODO: to check if there is no points
+ //for( int i=0, n=NbSections(); i<n; i++ )
+ // aNbPoints += NbPoints( i );
+
+ return !isNull && aNbPoints == 0;
+}
