#include <BRepBuilderAPI_MakeFace.hxx>
#include <BRepOffsetAPI_NormalProjection.hxx>
+#include <GEOMBase.h>
+
#include <GeomAPI_ProjectPointOnCurve.hxx>
#include <GeomAdaptor_Curve.hxx>
#include <Geom_Line.hxx>
#include <TColStd_Array1OfReal.hxx>
+#include <TDataStd_AsciiString.hxx>
#include <TDataStd_BooleanList.hxx>
#include <TDataStd_ExtStringList.hxx>
#include <TDataStd_IntegerList.hxx>
QStringList HYDROData_PolylineXY::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
{
QStringList aResList = dumpObjectCreation( theTreatedObjects );
- QString aPolylineName = GetName();
+ QString aPolylineName = GetObjPyName();
- // Set polilyne data
- NCollection_Sequence<TCollection_AsciiString> aSectNames;
- NCollection_Sequence<HYDROData_PolylineXY::SectionType> aSectTypes;
- NCollection_Sequence<bool> aSectClosures;
- GetSections( aSectNames, aSectTypes, aSectClosures );
+ // Set the wire color
+ QStringList aWireColorDef;
- for ( int i = 1, n = aSectNames.Size(); i <= n; ++i )
+ QColor aWireColor = GetWireColor();
+ setPythonObjectColor( aWireColorDef, aWireColor, DefaultWireColor(), "SetWireColor" );
+
+ if ( !aWireColorDef.isEmpty() )
{
- 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 << aWireColorDef;
+ aResList << QString( "" );
+ }
- HYDROData_IPolyline::PointsList aSectPointsList = GetPoints( i - 1 );
- for ( int k = 1, aNbPoints = aSectPointsList.Size(); k <= aNbPoints; ++k )
+ bool anIsEditable = IsEditable();
+ if ( !anIsEditable )
+ {
+ // If polyline is not editable we try to import the shape from geom
+ TCollection_AsciiString aGeomObjectEntry = GetGeomObjectEntry();
+ if ( !aGeomObjectEntry.IsEmpty() )
{
- const Point& aSectPoint = aSectPointsList.Value( k );
+ QString aSalomeObjName = HYDROData_Tool::GenerateNameForPython( theTreatedObjects, "polyline_sobj" );
+ aResList << QString( "%1 = salome.myStudy.FindObjectID( \"%2\" );" )
+ .arg( aSalomeObjName ).arg( aGeomObjectEntry.ToCString() );
+
+ aResList << QString( "%1.ImportFromGeomIOR( %2.GetIOR() );" )
+ .arg( aPolylineName ).arg( aSalomeObjName );
- aResList << QString( "%1.AddPoint( %2, gp_XY( %3, %4 ) );" ).arg( aPolylineName )
- .arg( i - 1 ).arg( aSectPoint.X() ).arg( aSectPoint.Y() );
+ 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 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 );
+
+ 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 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();
TopoDS_Wire aWire;
if( theType == SECTION_POLYLINE )
{
+ int aNbPoints = thePoints.Length();
BRepBuilderAPI_MakePolygon aMakeWire;
- for ( int i = 1, n = thePoints.Length(); i <= n ; ++i )
+ for ( int i = 1, n = aNbPoints; i <= n ; ++i )
{
gp_XYZ aPoint = thePoints.Value( i );
gp_Pnt aPnt( aPoint.X(), aPoint.Y(), aPoint.Z() );
aMakeWire.Add( aPnt );
}
- if( theIsClosed )
+ if( theIsClosed && ( aNbPoints > 2 ) )
aMakeWire.Close();
if ( aMakeWire.IsDone() )
setPolylineShape( aResult );
}
+bool HYDROData_PolylineXY::IsHas2dPrs() const
+{
+ return true;
+}
+
bool HYDROData_PolylineXY::IsEditable() const
{
return !myLab.IsAttribute( GUID_IS_UNEDITABLE );
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 );
+}
+
+