#include <QColor>
#include <QStringList>
-#define PYTHON_PROFILE_ID "KIND_PROFILE"
-
IMPLEMENT_STANDARD_HANDLE(HYDROData_Profile, HYDROData_Object)
IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Profile, HYDROData_Object)
QStringList HYDROData_Profile::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
{
- QStringList aResList;
+ QStringList aResList = dumpObjectCreation( theTreatedObjects );
+ QString aProfileName = GetObjPyName();
+
+ //TCollection_AsciiString aFilePath = GetFilePath();
+ //if ( !aFilePath.IsEmpty() )
+ //{
+ // aResList << QString( "%1.ImportFromFile( \"%2\" );" )
+ // .arg( aName ).arg( aFilePath.ToCString() );
+ //}
+
+ bool anIsValidProfile = IsValid();
+
+ QStringList aPntsDefinition;
+ QString aPntsListName = HYDROData_Tool::GenerateNameForPython( theTreatedObjects, "profile_points" );
+
+ QString aGap = QString().fill( ' ', aPntsListName.length() + 5 );
+ if ( anIsValidProfile )
+ {
+ HYDROData_Profile::ProfilePoints aPointsList = GetProfilePoints( true );
+ for ( int k = 1, aNbPoints = aPointsList.Size(); k <= aNbPoints; ++k )
+ {
+ const ProfilePoint& aPoint = aPointsList.Value( k );
+ aPntsDefinition << QString( aGap + "gp_XYZ( %1, %2, %3 )%4" )
+ .arg( aPoint.X() ).arg( aPoint.Y() ).arg( aPoint.Z() )
+ .arg( ( k < aNbPoints ? "," : "" ) );
+ }
+ }
+ else
+ {
+ HYDROData_IPolyline::PointsList aPointsList = GetParametricPoints();
+ for ( int k = 1, aNbPoints = aPointsList.Size(); k <= aNbPoints; ++k )
+ {
+ const HYDROData_IPolyline::Point& aPoint = aPointsList.Value( k );
+ aPntsDefinition << QString( aGap + "gp_XY( %1, %2 )%3" )
+ .arg( aPoint.X() ).arg( aPoint.Y() )
+ .arg( ( k < aNbPoints ? "," : "" ) );
+ }
+ }
- Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
- if ( aDocument.IsNull() )
- return aResList;
-
- QString aDocName = aDocument->GetDocPyName();
- QString aProfileName = GetName();
+ if ( !aPntsDefinition.isEmpty() )
+ {
+ QString& aFirstStr = aPntsDefinition.first();
+ aFirstStr = aFirstStr.trimmed();
+ aFirstStr.prepend( QString( "%1 = [ " ).arg( aPntsListName ) );
+
+ aPntsDefinition.last().append( " ];" );
- aResList << QString( "%1 = %2.CreateObject( %3 );" )
- .arg( aProfileName ).arg( aDocName ).arg( PYTHON_PROFILE_ID );
- aResList << QString( "%1.SetName( \"%1\" );" ).arg( aProfileName );
+ aResList << aPntsDefinition;
+
+ aResList << QString( "%1.%3( %2 );" )
+ .arg( aProfileName ).arg( aPntsListName )
+ .arg( anIsValidProfile ? "SetProfilePoints" : "SetParametricPoints" );
+
+ aResList << QString( "" );
+ }
+
+ // Set a polyline type if it is not default
+ Handle(HYDROData_ProfileUZ) aPrf = GetProfileUZ( false );
+ if ( !aPrf.IsNull() )
+ {
+ HYDROData_IPolyline::SectionType aSecType = aPrf->GetSectionType( 0 );
+ if ( aSecType != HYDROData_IPolyline::SECTION_POLYLINE )
+ {
+ aResList << QString( "%1.GetProfileUZ().SetSectionType( 0, %2 );" )
+ .arg( aProfileName ).arg( "HYDROData_IPolyline.SECTION_SPLINE" );
+ aResList << QString( "" );
+ }
+ }
+
+ aResList << QString( "%1.Update();" ).arg( aProfileName );
+ aResList << QString( "" );
return aResList;
}
TopoDS_Wire aWire;
gp_XY aFirstPoint, aLastPoint;
- if ( !GetLeftPoint( aFirstPoint ) || !GetRightPoint( aLastPoint ) )
+ if ( !GetLeftPoint( aFirstPoint, false ) || !GetRightPoint( aLastPoint, false ) )
return aWire;
gp_Pnt aPnt1( aFirstPoint.X(), aFirstPoint.Y(), 0 );
Handle(HYDROData_ProfileUZ) aProfile = GetProfileUZ( false );
if ( !aProfile.IsNull() )
{
- ProfilePoints aProfilePoints = GetProfilePoints();
+ ProfilePoints aProfilePoints = GetProfilePoints( false );
HYDROData_IPolyline::SectionType aSectionType = aProfile->GetSectionType( 0 );
aWire = HYDROData_PolylineXY::BuildWire( aSectionType, false, aProfilePoints );
bool HYDROData_Profile::IsValid() const
{
gp_XY aFirstPoint, aLastPoint;
- if ( !GetLeftPoint( aFirstPoint ) || !GetRightPoint( aLastPoint ) )
+ if ( !GetLeftPoint( aFirstPoint, false ) || !GetRightPoint( aLastPoint, false ) )
return false;
int aNbPoints = NbPoints();
return aNbPoints > 1;
}
-void HYDROData_Profile::SetLeftPoint( const gp_XY& thePoint )
+void HYDROData_Profile::SetLeftPoint( const gp_XY& theGPoint, bool IsConvertFromGlobal )
{
TDF_Label aLabel = myLab.FindChild( DataTag_FirstPoint );
+ if ( aLabel.IsNull() )
+ return;
+
+ Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( Label() );
+ gp_XY aLPoint = theGPoint;
+ if( IsConvertFromGlobal )
+ aDoc->Transform( aLPoint, true );
Handle(TDataStd_RealArray) anArray;
if ( !aLabel.FindAttribute( TDataStd_RealArray::GetID(), anArray ) )
anArray = TDataStd_RealArray::Set( aLabel, 0, 1 );
- anArray->SetValue( 0, thePoint.X() );
- anArray->SetValue( 1, thePoint.Y() );
+ anArray->SetValue( 0, aLPoint.X() );
+ anArray->SetValue( 1, aLPoint.Y() );
SetToUpdate( true );
}
-bool HYDROData_Profile::GetLeftPoint( gp_XY& thePoint ) const
+bool HYDROData_Profile::GetLeftPoint( gp_XY& thePoint, bool IsConvertToGlobal ) const
{
TDF_Label aLabel = myLab.FindChild( DataTag_FirstPoint, false );
if ( aLabel.IsNull() )
return false;
+ Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( myLab );
Handle(TDataStd_RealArray) anArray;
if ( !aLabel.FindAttribute( TDataStd_RealArray::GetID(), anArray ) )
return false;
thePoint.SetX( anArray->Value( 0 ) );
thePoint.SetY( anArray->Value( 1 ) );
+ if( IsConvertToGlobal )
+ aDoc->Transform( thePoint, false );
+
return true;
}
-void HYDROData_Profile::SetRightPoint( const gp_XY& thePoint )
+void HYDROData_Profile::SetRightPoint( const gp_XY& theGPoint, bool IsConvertFromGlobal )
{
TDF_Label aLabel = myLab.FindChild( DataTag_LastPoint );
+ Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( Label() );
+ gp_XY aLPoint = theGPoint;
+ if( IsConvertFromGlobal )
+ aDoc->Transform( aLPoint, true );
+
Handle(TDataStd_RealArray) anArray;
if ( !aLabel.FindAttribute( TDataStd_RealArray::GetID(), anArray ) )
anArray = TDataStd_RealArray::Set( aLabel, 0, 1 );
- anArray->SetValue( 0, thePoint.X() );
- anArray->SetValue( 1, thePoint.Y() );
+ anArray->SetValue( 0, aLPoint.X() );
+ anArray->SetValue( 1, aLPoint.Y() );
SetToUpdate( true );
}
-bool HYDROData_Profile::GetRightPoint( gp_XY& thePoint ) const
+bool HYDROData_Profile::GetRightPoint( gp_XY& thePoint, bool IsConvertToGlobal ) const
{
TDF_Label aLabel = myLab.FindChild( DataTag_LastPoint, false );
if ( aLabel.IsNull() )
return false;
+ Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( myLab );
Handle(TDataStd_RealArray) anArray;
if ( !aLabel.FindAttribute( TDataStd_RealArray::GetID(), anArray ) )
return false;
thePoint.SetX( anArray->Value( 0 ) );
thePoint.SetY( anArray->Value( 1 ) );
+ if( IsConvertToGlobal )
+ aDoc->Transform( thePoint, false );
+
return true;
}
return aProfileUZ.IsNull() ? HYDROData_ProfileUZ::PointsList() : aProfileUZ->GetPoints();
}
-void HYDROData_Profile::SetProfilePoints( const ProfilePoints& thePoints )
+void HYDROData_Profile::SetProfilePoints( const ProfilePoints& thePoints, bool IsConvertFromGlobal )
{
RemovePoints();
if ( thePoints.Length() < 2 )
gp_XY aFirstPoint, aLastPoint;
+ Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( Label() );
Handle(HYDROData_ProfileUZ) aProfileUZ = GetProfileUZ();
for ( int i = 1, n = thePoints.Length(); i <= n ; ++i )
{
- const ProfilePoint& aPoint = thePoints.Value( i );
+ ProfilePoint aPoint = thePoints.Value( i );
+ if( IsConvertFromGlobal )
+ aDoc->Transform( aPoint, true );
+
gp_XY aPointXY( aPoint.X(), aPoint.Y() );
if ( i == 1 )
aProfileUZ->AddPoint( 0, aParPoint );
}
- SetLeftPoint( aFirstPoint );
- SetRightPoint( aLastPoint );
+ SetLeftPoint( aFirstPoint, false );//already converted to local CS
+ SetRightPoint( aLastPoint, false );
}
-HYDROData_Profile::ProfilePoints HYDROData_Profile::GetProfilePoints() const
+HYDROData_Profile::ProfilePoints HYDROData_Profile::GetProfilePoints( bool IsConvertToGlobal ) const
{
ProfilePoints aResPoints;
gp_XY aFirstPoint, aLastPoint;
- if ( !GetLeftPoint( aFirstPoint ) || !GetRightPoint( aLastPoint ) )
+ if ( !GetLeftPoint( aFirstPoint, IsConvertToGlobal ) ||
+ !GetRightPoint( aLastPoint, IsConvertToGlobal ) )
return aResPoints;
HYDROData_ProfileUZ::PointsList aParametricPoints = GetParametricPoints();
}
else if ( anIsGeoref )
{
- SetProfilePoints( aPointsXYZ );
+ SetProfilePoints( aPointsXYZ, true );
}
Update();
return aRes;
}
+void HYDROData_Profile::UpdateLocalCS( double theDx, double theDy )
+{
+ gp_XY aDelta( theDx, theDy );
+ gp_XY aPnt;
+
+ GetLeftPoint( aPnt, false );
+ aPnt += aDelta;
+ SetLeftPoint( aPnt, false );
+
+ GetRightPoint( aPnt, false );
+ aPnt += aDelta;
+ SetRightPoint( aPnt, false );
+}
