#include "HYDROData_Obstacle.h"
#include "HYDROData_Document.h"
+#include "HYDROData_ShapesGroup.h"
+#include "HYDROData_ShapesTool.h"
+#include "HYDROData_Tool.h"
#include <Basics_Utils.hxx>
#include <BRepTools.hxx>
#include <BRep_Builder.hxx>
+
+#include <GEOMBase.h>
+
#include <IGESControl_Reader.hxx>
#include <IGESData_IGESModel.hxx>
+
#include <STEPControl_Reader.hxx>
+
#include <Interface_Static.hxx>
+
+#include <TopoDS.hxx>
#include <TopoDS_Iterator.hxx>
#include <TopoDS_Shape.hxx>
+#include <TopoDS_Edge.hxx>
+
#include <TDataStd_AsciiString.hxx>
+#include <TDataStd_RealArray.hxx>
+
#include <TColStd_SequenceOfAsciiString.hxx>
+
#include <TopExp_Explorer.hxx>
+#include <QColor>
#include <QFile>
#include <QFileInfo>
#include <QStringList>
#include <Standard_ErrorHandler.hxx> // CAREFUL ! position of this file is critic
-#define PYTHON_OBSTACLE_ID "KIND_OBSTACLE"
+#include <HYDROData_Projection.h>
IMPLEMENT_STANDARD_HANDLE(HYDROData_Obstacle,HYDROData_ArtificialObject)
IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Obstacle,HYDROData_ArtificialObject)
QStringList HYDROData_Obstacle::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
{
- QStringList aResList;
+ QStringList aResList = dumpObjectCreation( theTreatedObjects );
+
+ QString anObstacleName = GetObjPyName();
- Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
- if ( aDocument.IsNull() )
- return aResList;
+ TCollection_AsciiString aGeomObjectEntry = GetGeomObjectEntry();
+ QString aFilePath = GetFilePath();
- QString aDocName = aDocument->GetDocPyName();
- QString anObstacleName = GetName();
+ if ( !aGeomObjectEntry.IsEmpty() )
+ {
+ QString aSalomeObjName = HYDROData_Tool::GenerateNameForPython( theTreatedObjects, "obstacle_sobj" );
+ aResList << QString( "%1 = theStudy.FindObjectID( \"%2\" );" )
+ .arg( aSalomeObjName ).arg( aGeomObjectEntry.ToCString() );
- aResList << QString( "%1 = %2.CreateObject( %3 );" )
- .arg( anObstacleName ).arg( aDocName ).arg( PYTHON_OBSTACLE_ID );
- aResList << QString( "%1.SetName( \"%2\" );" )
- .arg( anObstacleName ).arg( anObstacleName );
- aResList << QString( "" );
+ aResList << QString( "%1.ImportFromGeomIOR( %2.GetIOR() );" )
+ .arg( anObstacleName ).arg( aSalomeObjName );
- // TODO
+ aResList << QString( "%1.SetGeomObjectEntry( \"%2\" );" )
+ .arg( anObstacleName ).arg( aGeomObjectEntry.ToCString() );
+ aResList << QString( "" );
+ }
+ else if ( !aFilePath.isEmpty() )
+ {
+ aResList << QString( "%1.ImportFromFile( \"%2\" );" )
+ .arg( anObstacleName ).arg( aFilePath );
+ aResList << QString( "" );
+ }
+
+ // Write the translation points
+ double aDx, aDy, aDz;
+ if ( getTranslation( aDx, aDy, aDz ) )
+ {
+ aResList << QString( "%1.Translate( %2, %3, %4 );" )
+ .arg( anObstacleName ).arg( aDx ).arg( aDy ).arg( aDz );
+ aResList << QString( "" );
+ }
+
+ aResList << QString( "%1.Update();" ).arg( anObstacleName );
+ aResList << QString( "" );
return aResList;
}
+void HYDROData_Obstacle::Update()
+{
+ removeGroupObjects();
+ createGroupObjects();
+ checkAndSetAltitudeObject();
+
+ HYDROData_Entity::Update();
+}
+
TopoDS_Shape HYDROData_Obstacle::GetTopShape() const
{
- // TODO
return getTopShape();
}
TopoDS_Shape HYDROData_Obstacle::GetShape3D() const
{
- // TODO
return getShape3D();
}
+void HYDROData_Obstacle::SetShape3D( const TopoDS_Shape& theShape )
+{
+ TopoDS_Face aShape2d = HYDROData_Projection::MakeProjection( theShape );
+ HYDROData_ArtificialObject::SetShape3D( theShape );
+ HYDROData_ArtificialObject::SetTopShape( aShape2d );
+}
+
+QColor HYDROData_Obstacle::DefaultFillingColor()
+{
+ return QColor( Qt::yellow );
+}
+
+QColor HYDROData_Obstacle::DefaultBorderColor()
+{
+ return QColor( Qt::transparent );
+}
+
bool HYDROData_Obstacle::ImportFromFile( const QString& theFilePath )
{
// Check the file existence
{
QString aRes;
- Handle(TDataStd_AsciiString) anAsciiStr;
- if ( myLab.FindChild( DataTag_FilePath ).FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
- aRes = QString( anAsciiStr->Get().ToCString() );
+ TDF_Label aLabel = myLab.FindChild( DataTag_FilePath, false );
+ if ( !aLabel.IsNull() )
+ {
+ Handle(TDataStd_AsciiString) anAsciiStr;
+ if ( aLabel.FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
+ aRes = QString( anAsciiStr->Get().ToCString() );
+ }
return aRes;
}
-void HYDROData_Obstacle::SetGeomObjectEntry( const QString& theEntry )
+bool HYDROData_Obstacle::ImportFromGeomIOR( const TCollection_AsciiString& theIOR )
{
- TCollection_AsciiString anAsciiStr( theEntry.toStdString().c_str() );
- TDataStd_AsciiString::Set( myLab.FindChild( DataTag_GeomObjectEntry ), anAsciiStr );
+ if ( theIOR.IsEmpty() )
+ return false;
+
+ TopoDS_Shape aShape = GEOMBase::GetShapeFromIOR( theIOR.ToCString() );
+ if ( aShape.IsNull() )
+ return false;
+
+ SetShape3D( aShape );
+
+ return true;
}
-QString HYDROData_Obstacle::GetGeomObjectEntry() const
+void HYDROData_Obstacle::SetGeomObjectEntry( const TCollection_AsciiString& theEntry )
{
- QString aRes;
+ TDataStd_AsciiString::Set( myLab.FindChild( DataTag_GeomObjectEntry ), theEntry );
+}
- Handle(TDataStd_AsciiString) anAsciiStr;
- if ( myLab.FindChild( DataTag_GeomObjectEntry ).FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
- aRes = QString( anAsciiStr->Get().ToCString() );
+TCollection_AsciiString HYDROData_Obstacle::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;
}
+void HYDROData_Obstacle::Translate( const double theDx,
+ const double theDy,
+ const double theDz )
+{
+ TopoDS_Shape aShape3D = GetShape3D();
+
+ TopoDS_Shape aTranslatedShape3D = HYDROData_ShapesTool::Translated( aShape3D, theDx, theDy, theDz );
+ if ( aTranslatedShape3D.IsNull() )
+ return;
+
+ SetShape3D( aTranslatedShape3D );
+
+ setTranslation( theDx, theDy, theDz );
+}
+
TopoDS_Shape HYDROData_Obstacle::ImportBREP( const QString& theFilePath ) const
{
TopoDS_Shape aResShape;
return aResShape;
}
+
+QColor HYDROData_Obstacle::getDefaultFillingColor() const
+{
+ return DefaultFillingColor();
+}
+
+QColor HYDROData_Obstacle::getDefaultBorderColor() const
+{
+ return DefaultBorderColor();
+}
+
+ObjectKind HYDROData_Obstacle::getAltitudeObjectType() const
+{
+ return KIND_OBSTACLE_ALTITUDE;
+}
+
+void HYDROData_Obstacle::createGroupObjects()
+{
+ TopoDS_Shape anObstacleShape = GetTopShape();
+ if ( !anObstacleShape.IsNull() )
+ {
+ TopTools_SequenceOfShape aWireEdges;
+ HYDROData_ShapesTool::ExploreShapeToShapes( anObstacleShape, TopAbs_EDGE, aWireEdges );
+ if ( !aWireEdges.IsEmpty() )
+ {
+ QString aWireGroupName = GetName() + "_Outer_Wire";
+
+ Handle(HYDROData_ShapesGroup) anExtWireGroup = createGroupObject();
+ anExtWireGroup->SetName( aWireGroupName );
+
+ anExtWireGroup->SetShapes( aWireEdges );
+ }
+ }
+}
+
+void HYDROData_Obstacle::setTranslation( const double theDx,
+ const double theDy,
+ const double theDz )
+{
+ TDF_Label aLabel = myLab.FindChild( DataTag_Translation );
+
+ double aCurDx = theDx;
+ double aCurDy = theDy;
+ double aCurDz = theDz;
+
+ double aPrevDx, aPrevDy, aPrevDz;
+ if ( getTranslation( aPrevDx, aPrevDy, aPrevDz ) )
+ {
+ aCurDx += aPrevDx;
+ aCurDy += aPrevDy;
+ aCurDz += aPrevDz;
+ }
+
+ Handle(TDataStd_RealArray) aCoeffsArray = TDataStd_RealArray::Set( aLabel, 1, 3 );
+ aCoeffsArray->SetValue( 1, aCurDx );
+ aCoeffsArray->SetValue( 2, aCurDy );
+ aCoeffsArray->SetValue( 3, aCurDz );
+}
+
+bool HYDROData_Obstacle::getTranslation( double& theDx, double& theDy, double& theDz ) const
+{
+ theDx = theDy = theDz = 0.0;
+
+ TDF_Label aLabel = myLab.FindChild( DataTag_Translation, false );
+ if ( aLabel.IsNull() )
+ return false;
+
+ Handle(TDataStd_RealArray) aCoeffsArray;
+ if ( !aLabel.FindAttribute( TDataStd_RealArray::GetID(), aCoeffsArray ) )
+ return false;
+
+ theDx = aCoeffsArray->Value( 1 );
+ theDy = aCoeffsArray->Value( 2 );
+ theDz = aCoeffsArray->Value( 3 );
+
+ return true;
+}
+