#include "HYDROData_BSplineOperation.h"
#include "HYDROData_Document.h"
+#include "HYDROData_ShapesTool.h"
#include "HYDROData_Tool.h"
#include <BRep_Builder.hxx>
#include <GeomAPI_ProjectPointOnCurve.hxx>
#include <GeomAdaptor_Curve.hxx>
+#include <Geom_Line.hxx>
+#include <Geom_BSplineCurve.hxx>
#include <GCPnts_AbscissaPoint.hxx>
#include <TColStd_ListIteratorOfListOfInteger.hxx>
#include <TColStd_ListIteratorOfListOfReal.hxx>
+#include <TColStd_Array1OfReal.hxx>
+
#include <TDataStd_BooleanList.hxx>
#include <TDataStd_ExtStringList.hxx>
#include <TDataStd_IntegerList.hxx>
#include <TDataStd_ListIteratorOfListOfExtendedString.hxx>
#include <TDataStd_RealList.hxx>
+#include <TDataStd_UAttribute.hxx>
#include <TopoDS_Iterator.hxx>
#include <TopTools_ListIteratorOfListOfShape.hxx>
#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 )
+{
+ NCollection_Map<TCollection_AsciiString> aNamesMap;
+
+ for ( int i = 1, n = theNamesSeq.Size(); i <= n; ++i )
+ {
+ const TCollection_AsciiString& aSectName = theNamesSeq.Value( i );
+ aNamesMap.Add( aSectName );
+ }
+
+ TCollection_AsciiString aResName;
+
+ int aPrefIdx = 1;
+ do
+ {
+ aResName = TCollection_AsciiString( "Section_" ) + aPrefIdx;
+ ++aPrefIdx;
+ }
+ while ( aNamesMap.Contains( aResName ) );
+
+ return aResName;
+}
+
+TCollection_AsciiString getUniqueSectionName( const Handle(TDataStd_ExtStringList)& theNamesList )
+{
+ NCollection_Sequence<TCollection_AsciiString> aNamesSeq;
+
+ TDataStd_ListIteratorOfListOfExtendedString aNamesIter( theNamesList->List() );
+ for ( ; aNamesIter.More(); aNamesIter.Next() )
+ aNamesSeq.Append( aNamesIter.Value() );
+
+ return getUniqueSectionName( aNamesSeq );
+}
+
IMPLEMENT_STANDARD_HANDLE(HYDROData_PolylineXY, HYDROData_IPolyline)
IMPLEMENT_STANDARD_RTTIEXT(HYDROData_PolylineXY, HYDROData_IPolyline)
return QColor( Qt::red );
}
-TopoDS_Shape HYDROData_PolylineXY::GetShape()
+TopoDS_Shape HYDROData_PolylineXY::GetShape() const
{
return getPolylineShape();
}
+bool convertEdgeToSection( const TopoDS_Edge& theEdge,
+ NCollection_Sequence<TCollection_AsciiString>& theSectNames,
+ NCollection_Sequence<HYDROData_PolylineXY::SectionType>& theSectTypes,
+ NCollection_Sequence<bool>& theSectClosures,
+ NCollection_Sequence<HYDROData_PolylineXY::PointsList>& theSectPoints,
+ const bool theIsCanBeClosed )
+{
+ Standard_Real aFirst = 0.0, aLast = 0.0;
+ Handle(Geom_Curve) anEdgeGeomCurve = BRep_Tool::Curve( theEdge, aFirst, aLast );
+ if ( anEdgeGeomCurve.IsNull() )
+ return false;
+
+ TCollection_AsciiString aSectName = getUniqueSectionName( theSectNames );
+ bool anIsEdgeClosed = anEdgeGeomCurve->IsClosed();
+
+ HYDROData_PolylineXY::SectionType aSectionType = HYDROData_PolylineXY::SECTION_POLYLINE;
+ HYDROData_PolylineXY::PointsList aPointsList;
+
+ if ( anEdgeGeomCurve->IsKind( STANDARD_TYPE(Geom_Line) ) )
+ {
+ Handle(Geom_Line) aGeomLine = Handle(Geom_Line)::DownCast( anEdgeGeomCurve );
+
+ gp_Pnt aFirstPoint, aLastPoint;
+ aGeomLine->D0( aFirst, aFirstPoint );
+ aGeomLine->D0( aLast, aLastPoint );
+
+ HYDROData_PolylineXY::Point aSectFirstPoint( aFirstPoint.X(), aFirstPoint.Y() );
+ aPointsList.Append( aSectFirstPoint );
+
+ HYDROData_PolylineXY::Point aSectLastPoint( aLastPoint.X(), aLastPoint.Y() );
+ aPointsList.Append( aSectLastPoint );
+ }
+ else if ( anEdgeGeomCurve->IsKind( STANDARD_TYPE(Geom_BSplineCurve) ) )
+ {
+ aSectionType = HYDROData_PolylineXY::SECTION_SPLINE;
+
+ Handle(Geom_BSplineCurve) aGeomSpline =
+ Handle(Geom_BSplineCurve)::DownCast( anEdgeGeomCurve );
+
+ int aNbKnots = aGeomSpline->NbKnots();
+
+ TColStd_Array1OfReal aSplineKnots( 1, aNbKnots );
+ aGeomSpline->Knots( aSplineKnots );
+
+ // Decrease the number of imported knots because of last one
+ // knot is the closing point which are the start point
+ if ( anIsEdgeClosed ) aNbKnots--;
+
+ for ( int i = 1; i <= aNbKnots; ++i )
+ {
+ const Standard_Real& aKnot = aSplineKnots.Value( i );
+
+ gp_Pnt aPoint;
+ aGeomSpline->D0( aKnot, aPoint );
+
+ HYDROData_PolylineXY::Point aSectPoint( aPoint.X(), aPoint.Y() );
+ aPointsList.Append( aSectPoint );
+ }
+ }
+ else
+ {
+ // Other curve types are not supported
+ return false;
+ }
+
+ if ( aPointsList.IsEmpty() )
+ return false;
+
+ theSectNames.Append( aSectName );
+ theSectTypes.Append( aSectionType );
+ theSectClosures.Append( anIsEdgeClosed );
+ theSectPoints.Append( aPointsList );
+
+ return true;
+}
+
+bool HYDROData_PolylineXY::ImportShape( const TopoDS_Shape& theShape )
+{
+ if ( theShape.IsNull() )
+ return false;
+
+ RemoveSections();
+
+ bool anIsCanBeImported = false;
+
+ NCollection_Sequence<TCollection_AsciiString> aSectNames;
+ NCollection_Sequence<SectionType> aSectTypes;
+ NCollection_Sequence<bool> aSectClosures;
+ NCollection_Sequence<PointsList> aSectPoints;
+
+ if ( theShape.ShapeType() == TopAbs_EDGE )
+ {
+ TopoDS_Edge anEdge = TopoDS::Edge( theShape );
+ anIsCanBeImported = convertEdgeToSection( anEdge, aSectNames, aSectTypes, aSectClosures, aSectPoints, true );
+ }
+ else if ( theShape.ShapeType() == TopAbs_WIRE )
+ {
+ TopTools_SequenceOfShape anEdges;
+ HYDROData_ShapesTool::ExploreShapeToShapes( theShape, TopAbs_EDGE, anEdges );
+
+ anIsCanBeImported = !anEdges.IsEmpty();
+ for ( int i = 1, n = anEdges.Length(); i <= n && anIsCanBeImported; ++i )
+ {
+ TopoDS_Edge aWireEdge = TopoDS::Edge( anEdges.Value( i ) );
+ anIsCanBeImported = convertEdgeToSection( aWireEdge, aSectNames, aSectTypes, aSectClosures, aSectPoints, false );
+ }
+ }
+
+ if ( anIsCanBeImported )
+ {
+ for ( int i = 1, n = aSectNames.Length(); i <= n; ++i )
+ {
+ const TCollection_AsciiString& aSectName = aSectNames.Value( i );
+ const SectionType& aSectType = aSectTypes.Value( i );
+ bool anIsSectionClosed = aSectClosures.Value( i );
+ const PointsList& aSectPointsList = aSectPoints( i );
+
+ AddSection( aSectName, aSectType, anIsSectionClosed );
+ SetPoints( i - 1, aSectPointsList );
+ }
+ }
+ else
+ {
+ setPolylineShape( theShape );
+ }
+
+ setEditable( anIsCanBeImported );
+
+ return true;
+}
+
TopoDS_Wire HYDROData_PolylineXY::BuildWire( const SectionType& theType,
const bool& theIsClosed,
const NCollection_Sequence<gp_XYZ>& thePoints )
void HYDROData_PolylineXY::Update()
{
+ if ( !IsEditable() )
+ {
+ // If polyline is not editable we no need to update it wire
+ SetToUpdate( false );
+ return;
+ }
+
HYDROData_IPolyline::Update();
NCollection_Sequence<TCollection_AsciiString> aSectNames;
TopTools_ListIteratorOfListOfShape it(aSectionWiresList);
for(;it.More();it.Next())
{
- TopExp_Explorer it2(it.Value(), TopAbs_EDGE);
- for(;it2.More();it2.Next())
- aSeqEdges->Append(it2.Current());
- }
- BRep_Builder aBB;
- TopoDS_Compound aCmp;
- TopoDS_Shape aResult;
- aBB.MakeCompound(aCmp);
- if(aSeqEdges->Length() >1)
- {
- ShapeAnalysis_FreeBounds::ConnectEdgesToWires(aSeqEdges,1E-5,Standard_False,aSeqWires);
-
- if( aSeqWires->Length()==1 )
- aResult = aSeqWires->Value( 1 );
- else
- {
- for (Standard_Integer i = 1; i <= aSeqWires->Length();i++)
- {
- const TopoDS_Shape& aS1 = aSeqWires->Value(i);
- aBB.Add(aCmp, aS1);
- }
- aResult = aCmp;
- }
- }
- else if (aSeqEdges->Length() == 1)
- {
- BRepBuilderAPI_MakeWire mkWire (TopoDS::Edge(aSeqEdges->Value(1)));
- if (mkWire.IsDone())
- aResult = mkWire.Wire();
- }
+ TopExp_Explorer it2(it.Value(), TopAbs_EDGE);
+ for(;it2.More();it2.Next())
+ aSeqEdges->Append(it2.Current());
+ }
+
+ BRep_Builder aBB;
+ TopoDS_Compound aCmp;
+ TopoDS_Shape aResult;
+ aBB.MakeCompound(aCmp);
+ if(aSeqEdges->Length() >1)
+ {
+ ShapeAnalysis_FreeBounds::ConnectEdgesToWires(aSeqEdges,1E-5,Standard_False,aSeqWires);
+
+ if( aSeqWires->Length()==1 )
+ aResult = aSeqWires->Value( 1 );
+ else
+ {
+ for (Standard_Integer i = 1; i <= aSeqWires->Length();i++)
+ {
+ const TopoDS_Shape& aS1 = aSeqWires->Value(i);
+ aBB.Add(aCmp, aS1);
+ }
+ aResult = aCmp;
+ }
+ }
+ else if (aSeqEdges->Length() == 1)
+ {
+ BRepBuilderAPI_MakeWire mkWire (TopoDS::Edge(aSeqEdges->Value(1)));
+ if (mkWire.IsDone())
+ aResult = mkWire.Wire();
+ }
setPolylineShape( aResult );
}
+bool HYDROData_PolylineXY::IsEditable() const
+{
+ return !myLab.IsAttribute( GUID_IS_UNEDITABLE );
+}
+
+void HYDROData_PolylineXY::setEditable( const bool theIsEditable )
+{
+ if ( !theIsEditable )
+ TDataStd_UAttribute::Set( myLab, GUID_IS_UNEDITABLE );
+ else
+ myLab.ForgetAttribute( GUID_IS_UNEDITABLE );
+}
+
/**
* Returns true if polyline is closed
*/
-bool HYDROData_PolylineXY::IsClosed() const
+bool HYDROData_PolylineXY::IsClosed( const bool theIsSimpleCheck ) const
{
- NCollection_Sequence<TCollection_AsciiString> aSectNames;
- NCollection_Sequence<HYDROData_PolylineXY::SectionType> aSectTypes;
- NCollection_Sequence<bool> aSectClosures;
- GetSections( aSectNames, aSectTypes, aSectClosures );
- if( aSectNames.IsEmpty() )
- return false;
+ bool anIsClosed = false;
- bool anIsClosed = true;
- for( int i = 1, n = aSectClosures.Size(); i <= n && anIsClosed; ++i )
- anIsClosed = anIsClosed && aSectClosures.Value( i );
+ TopoDS_Shape aShape = GetShape();
+ 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 );
+
+ int aNbWires = aWires.Length();
+ if ( theIsSimpleCheck )
+ {
+ anIsClosed = aNbWires > 0;
+ for ( int i = 1; i <= aNbWires && anIsClosed; ++i )
+ {
+ const TopoDS_Shape& aWire = aWires.Value( i );
+ anIsClosed = BRep_Tool::IsClosed( aWire );
+ }
+ }
+ else
+ {
+ anIsClosed = aNbWires == 1 && BRep_Tool::IsClosed( aWires.First() );
+ }
return anIsClosed;
}
return !aClosuresList.IsNull() ? aClosuresList->Extent() : 0;
}
-TCollection_ExtendedString getUniqueSectionName( const Handle(TDataStd_ExtStringList)& theNamesList )
-{
- NCollection_Map<TCollection_ExtendedString> aNamesMap;
-
- TDataStd_ListIteratorOfListOfExtendedString aNamesIter( theNamesList->List() );
- for ( ; aNamesIter.More(); aNamesIter.Next() )
- aNamesMap.Add( aNamesIter.Value() );
-
- TCollection_ExtendedString aResName;
-
- int aPrefIdx = 1;
- do
- {
- aResName = "Section_" + aPrefIdx;
- ++aPrefIdx;
- }
- while ( aNamesMap.Contains( aResName ) );
-
- return aResName;
-}
-
void HYDROData_PolylineXY::AddSection( const TCollection_AsciiString& theSectName,
const SectionType theSectionType,
const bool theIsClosed )
SetToUpdate( true );
}
+void HYDROData_PolylineXY::SetPoints( const int theSectionIndex,
+ const PointsList& thePoints )
+{
+ Handle(TDataStd_RealList) aListX, aListY;
+ getPointsLists( theSectionIndex, aListX, aListY );
+
+ aListX->Clear();
+ aListY->Clear();
+
+ for ( int i = 1, n = thePoints.Length(); i <= n; ++i )
+ {
+ const Point& aPoint = thePoints.Value( i );
+ aListX->Append( aPoint.X() );
+ aListY->Append( aPoint.Y() );
+ }
+}
+
void HYDROData_PolylineXY::RemovePoint( const int theSectionIndex,
const int thePointIndex )
{
Q_OBJECT
public:
+
HYDROGUI_Operation( HYDROGUI_Module* theModule );
virtual ~HYDROGUI_Operation();
- void setName( const QString& theName );
- const QString& getName() const;
+public:
+
+ void setName( const QString& theName );
+ const QString& getName() const;
- HYDROGUI_InputPanel* inputPanel() const;
- SUIT_SelectionMgr* selectionMgr() const;
- HYDROGUI_Module* module() const;
+ HYDROGUI_InputPanel* inputPanel() const;
+ SUIT_SelectionMgr* selectionMgr() const;
+ HYDROGUI_Module* module() const;
signals:
- void helpContextModule( const QString&, const QString&, const QString& );
+ void helpContextModule( const QString&,
+ const QString&,
+ const QString& );
+
protected:
- virtual void startOperation();
- virtual void abortOperation();
- virtual void commitOperation();
- virtual void setDialogActive( const bool );
+ virtual void startOperation();
+ virtual void abortOperation();
+ virtual void commitOperation();
+ virtual void setDialogActive( const bool );
- virtual HYDROGUI_InputPanel* createInputPanel() const;
- virtual void closeInputPanel();
+ virtual HYDROGUI_InputPanel* createInputPanel() const;
+ virtual void closeInputPanel();
- virtual bool processApply( int& theUpdateFlags, QString& theErrorMsg );
- virtual void processCancel();
+ virtual bool processApply( int& theUpdateFlags, QString& theErrorMsg );
+ virtual void processCancel();
- void startDocOperation();
- void abortDocOperation();
- void commitDocOperation();
+ void startDocOperation();
+ void abortDocOperation();
+ void commitDocOperation();
- Handle_HYDROData_Document doc() const;
+ Handle_HYDROData_Document doc() const;
+
+ void printErrorMessage( const QString& theErrorMsg );
+ void setPrintErrorMessage( const bool theIsPrint );
protected slots:
- virtual void onApply();
- virtual void onCancel();
- virtual void onHelp();
+
+ virtual void onApply();
+ virtual void onCancel();
+ virtual void onHelp();
protected:
- QString getHelpComponent() const;
- QString getHelpFile() const;
- QString getHelpContext() const;
+
+ QString getHelpComponent() const;
+ QString getHelpFile() const;
+ QString getHelpContext() const;
private:
- HYDROGUI_Module* myModule;
- HYDROGUI_InputPanel* myPanel;
- QString myName;
+
+ HYDROGUI_Module* myModule;
+ HYDROGUI_InputPanel* myPanel;
+ QString myName;
+ bool myIsPrintErrorMessage;
};
#endif