#include <HYDROData_PolylineOperator.h>
#include <HYDROData_Document.h>
+#include <HYDROData_TopoCurve.h>
+
+#include <CurveCreator_Utils.hxx>
+
+#include <BRepAdaptor_Curve.hxx>
+#include <BRep_Builder.hxx>
+#include <BRep_Tool.hxx>
#include <BRepBuilderAPI_MakeEdge2d.hxx>
+#include <BRepBuilderAPI_MakeEdge.hxx>
#include <BRepBuilderAPI_MakeWire.hxx>
+#include <Extrema_ExtCC.hxx>
+#include <Extrema_ExtPC.hxx>
+#include <GeomAPI_Interpolate.hxx>
+#include <NCollection_Vector.hxx>
+#include <Precision.hxx>
+#include <ShapeAnalysis_TransferParametersProj.hxx>
+#include <ShapeBuild_Edge.hxx>
+#include <TColgp_Array1OfVec.hxx>
+#include <TColgp_HArray1OfPnt.hxx>
+#include <TColStd_HArray1OfBoolean.hxx>
+#include <TopoDS.hxx>
#include <TopoDS_Edge.hxx>
#include <TopoDS_Wire.hxx>
+#include <TopExp.hxx>
+#include <TopExp_Explorer.hxx>
+#include <TopTools_IndexedDataMapOfShapeListOfShape.hxx>
+#include <QString>
template<class T> void append( std::vector<T>& theList, const std::vector<T>& theList2 )
{
theList[i] = theList2[j];
}
-
bool HYDROData_PolylineOperator::Split( const Handle( HYDROData_Document )& theDoc,
- const TCollection_AsciiString& theNamePrefix,
const Handle( HYDROData_PolylineXY )& thePolyline,
- const gp_Pnt2d& thePoint ) const
+ const gp_Pnt2d& thePoint,
+ double theTolerance ) const
{
+ if (thePolyline.IsNull())
+ {
+ return false;
+ }
+
std::vector<gp_Pnt2d> aPointsList( 1 );
aPointsList[0] = thePoint;
- std::vector<Handle( Geom2d_Curve )> aCurves = GetCurves( thePolyline );
+ std::vector<TopoDS_Wire> aCurves;
+ GetWires(thePolyline, aCurves);
bool isOK = true;
for( int i=0, n=aCurves.size(); i<n; i++ )
{
- std::vector<Handle( Geom2d_Curve )> aCurvesList = Split( aCurves[i], aPointsList );
- bool isLocalOK = CreatePolylines( theDoc, theNamePrefix, aCurvesList );
+ std::vector<TopoDS_Shape> aCurvesList;
+ Split(aCurves[i], thePoint, theTolerance, aCurvesList);
+ bool isLocalOK = CreatePolylines( theDoc, thePolyline->GetName(), aCurvesList, true );
isOK = isOK && isLocalOK;
}
return isOK;
}
bool HYDROData_PolylineOperator::Split( const Handle( HYDROData_Document )& theDoc,
- const TCollection_AsciiString& theNamePrefix,
- const Handle( HYDROData_PolylineXY )& thePolyline,
- const Handle( HYDROData_PolylineXY )& theTool ) const
+ const Handle( HYDROData_PolylineXY )& thePolyline,
+ const Handle( HYDROData_PolylineXY )& theTool,
+ double theTolerance,
+ bool& theIsIntersected) const
{
- std::vector<Handle( Geom2d_Curve )> aCurves = GetCurves( thePolyline );
- std::vector<Handle( Geom2d_Curve )> aToolCurves = GetCurves( theTool );
- bool isOK = true;
- for( int i=0, n=aCurves.size(); i<n; i++ )
- for( int j=0, m=aToolCurves.size(); j<m; j++ )
- {
- std::vector<gp_Pnt2d> aPointsList = Intersection( aCurves[i], aToolCurves[j] );
- std::vector<Handle( Geom2d_Curve )> aCurvesList = Split( aCurves[i], aPointsList );
- bool isLocalOK = CreatePolylines( theDoc, theNamePrefix, aCurvesList );
- isOK = isOK && isLocalOK;
- }
- return isOK;
+ if (thePolyline.IsNull() || theTool.IsNull())
+ {
+ return false;
+ }
+
+ HYDROData_SequenceOfObjects aSeq;
+ aSeq.Append( theTool );
+ return split( theDoc, thePolyline, aSeq, theTolerance, -1, theIsIntersected);
}
bool HYDROData_PolylineOperator::Split( const Handle( HYDROData_Document )& theDoc,
- const TCollection_AsciiString& theNamePrefix,
- const HYDROData_SequenceOfObjects& thePolylines )
+ const HYDROData_SequenceOfObjects& thePolylines,
+ double theTolerance )
{
int f = thePolylines.Lower(), l = thePolylines.Upper();
- bool isOK = true;
- std::vector<Handle( Geom2d_Curve )> anAllCurves;
for( int i=f; i<=l; i++ )
{
Handle( HYDROData_PolylineXY ) aPolyline = Handle( HYDROData_PolylineXY )::DownCast( thePolylines.Value( i ) );
- std::vector<Handle( Geom2d_Curve )> aCurves = GetCurves( aPolyline );
- append( anAllCurves, aCurves );
+ bool isIntersected;
+ if( !split( theDoc, aPolyline, thePolylines, theTolerance, i, isIntersected) )
+ return false;
}
+ return true;
+}
- for( int i=0, n=anAllCurves.size(); i<n; i++ )
+bool HYDROData_PolylineOperator::Merge( const Handle( HYDROData_Document )& theDoc,
+ const QString& theName,
+ const HYDROData_SequenceOfObjects& thePolylines,
+ bool isConnectByNewSegment,
+ double theTolerance )
+{
+ std::deque<HYDROData_TopoCurve> aMergedCurves;
+ HYDROData_SequenceOfObjects::Iterator aPIt(thePolylines);
+ for (; aPIt.More(); aPIt.Next())
{
- std::vector<gp_Pnt2d> aCompletePointsList;
- for( int j=0; j<n; j++ )
+ Handle(HYDROData_PolylineXY) aPolyline =
+ Handle(HYDROData_PolylineXY)::DownCast(aPIt.Value());
+ std::vector<TopoDS_Wire> aWires;
+ GetWires(aPolyline, aWires);
+ for (std::vector<TopoDS_Wire>::const_iterator aWIt = aWires.begin(),
+ aLastWIt = aWires.end(); aWIt != aLastWIt; ++aWIt)
{
- if( i==j )
- continue;
- std::vector<gp_Pnt2d> aPointsList = Intersection( anAllCurves[i], anAllCurves[j] );
- append( aCompletePointsList, aPointsList );
+ const Standard_Boolean aResult = !isConnectByNewSegment ?
+ HYDROData_TopoCurve::Merge(theTolerance, *aWIt, aMergedCurves) :
+ HYDROData_TopoCurve::Connect(theTolerance, *aWIt, aMergedCurves);
+ if (!aResult)
+ {
+ return false;
+ }
}
- std::vector<Handle( Geom2d_Curve )> aCurvesList = Split( anAllCurves[i], aCompletePointsList );
- bool isLocalOK = CreatePolylines( theDoc, theNamePrefix, aCurvesList );
- isOK = isOK && isLocalOK;
}
- return isOK;
-}
-bool HYDROData_PolylineOperator::Merge( const Handle( HYDROData_Document )& theDoc,
- const TCollection_AsciiString& theName,
- const HYDROData_SequenceOfObjects& thePolylines )
-{
- //TODO
+ TopoDS_Compound aWireSet;
+ BRep_Builder aBuilder;
+ aBuilder.MakeCompound(aWireSet);
+ std::deque<HYDROData_TopoCurve>::iterator aCIt = aMergedCurves.begin();
+ std::deque<HYDROData_TopoCurve>::iterator aLastCIt = aMergedCurves.end();
+ for (; aCIt != aLastCIt; ++aCIt)
+ {
+ if (!aCIt->IsEmpty())
+ {
+ aBuilder.Add(aWireSet, aCIt->Wire());
+ }
+ }
+
+ std::vector<TopoDS_Shape> aPolylines(1);
+ aPolylines[0] = aWireSet;
+ CreatePolylines(theDoc, theName, aPolylines, false);
return true;
}
-std::vector<Handle( Geom2d_Curve )> HYDROData_PolylineOperator::GetCurves( const Handle( HYDROData_PolylineXY )& thePolyline )
+bool HYDROData_PolylineOperator::split( const Handle( HYDROData_Document )& theDoc,
+ const Handle( HYDROData_PolylineXY )& thePolyline,
+ const HYDROData_SequenceOfObjects& theTools,
+ double theTolerance,
+ int theIgnoreIndex,
+ bool& theIsIntersected) const
{
- std::vector<Handle( Geom2d_Curve )> aResult;
- //TODO
- return aResult;
+ theIsIntersected = false;
+
+ if (thePolyline.IsNull())
+ {
+ return false;
+ }
+
+ std::vector<TopoDS_Wire> aCurves;
+ GetWires(thePolyline, aCurves);
+ std::vector<TopoDS_Wire> aToolCurves;
+ for( int i=theTools.Lower(), n=theTools.Upper(); i<=n; i++ )
+ if( i!=theIgnoreIndex )
+ {
+ Handle( HYDROData_PolylineXY ) aToolPolyline =
+ Handle( HYDROData_PolylineXY )::DownCast( theTools.Value( i ) );
+ if (!aToolPolyline.IsNull())
+ {
+ std::vector<TopoDS_Wire> aTCurves;
+ GetWires(aToolPolyline, aTCurves);
+ append( aToolCurves, aTCurves);
+ }
+ }
+
+ if (aToolCurves.empty())
+ {
+ return false;
+ }
+
+ const int aPSCount = aCurves.size();
+ const int aTSCount = aToolCurves.size();
+ std::vector<TopoDS_Shape> aResult;
+ for (int aPSI = 0; aPSI < aPSCount; ++aPSI)
+ {
+ HYDROData_TopoCurve aCurve;
+ if (!aCurve.Initialize(aCurves[aPSI]))
+ {
+ continue;
+ }
+
+ std::deque<std::list<double> > aParams;
+ for (int aTSI = 0; aTSI < aTSCount; ++aTSI)
+ {
+ aCurve.Intersect(aToolCurves[aTSI], aParams);
+ }
+
+ std::deque<HYDROData_TopoCurve> aSplittedCurves;
+ theIsIntersected |= aCurve.Cut(aParams, aSplittedCurves);
+ std::deque<HYDROData_TopoCurve>::const_iterator aCIt =
+ aSplittedCurves.begin();
+ std::deque<HYDROData_TopoCurve>::const_iterator aLastCIt =
+ aSplittedCurves.end();
+ for (; aCIt != aLastCIt; ++aCIt)
+ {
+ aResult.push_back(aCIt->Wire());
+ }
+ }
+
+ CreatePolylines(theDoc, thePolyline->GetName(), aResult, true);
+ return true;
}
-std::vector<gp_Pnt2d> HYDROData_PolylineOperator::Intersection( const Handle( Geom2d_Curve )& theCurve,
- const Handle( Geom2d_Curve )& theTool )
+void HYDROData_PolylineOperator::GetWires(
+ const Handle( HYDROData_PolylineXY )& thePolyline,
+ std::vector<TopoDS_Wire>& theWires)
{
- std::vector<gp_Pnt2d> aResult;
- //TODO
- return aResult;
+ TopoDS_Shape aShape = thePolyline->GetShape();
+ if( aShape.ShapeType()==TopAbs_WIRE )
+ {
+ theWires.push_back( TopoDS::Wire( aShape ) );
+ }
+ else
+ {
+ TopExp_Explorer anExp( aShape, TopAbs_WIRE );
+ for( ; anExp.More(); anExp.Next() )
+ {
+ theWires.push_back( TopoDS::Wire( anExp.Current() ) );
+ }
+ }
}
-std::vector<Handle( Geom2d_Curve )> HYDROData_PolylineOperator::Split( const Handle( Geom2d_Curve )& theCurve,
- const std::vector<gp_Pnt2d>& thePoints )
+void HYDROData_PolylineOperator::Split(
+ const TopoDS_Wire& theWire,
+ const gp_Pnt2d& thePoint,
+ double theTolerance,
+ std::vector<TopoDS_Shape>& theWires)
{
- std::vector<Handle( Geom2d_Curve )> aResult;
- //TODO
- return aResult;
+ HYDROData_TopoCurve aCurve;
+ if (!aCurve.Initialize(theWire))
+ {
+ theWires.push_back(theWire);
+ return;
+ }
+
+ const gp_XYZ aP(thePoint.X(), thePoint.Y(), 0);
+ std::list<TopoDS_Edge>::const_iterator aEPos;
+ double aParam;
+ aCurve.Project(aP, aEPos, aParam);
+ HYDROData_TopoCurve aCurve1, aCurve2;
+ aCurve.Cut(aEPos, aParam, aCurve1, aCurve2);
+ theWires.push_back(aCurve1.Wire());
+ if (!aCurve2.IsEmpty())
+ {
+ theWires.push_back(aCurve2.Wire());
+ }
}
bool HYDROData_PolylineOperator::CreatePolylines( const Handle( HYDROData_Document )& theDoc,
- const TCollection_AsciiString& theNamePrefix,
- const std::vector<Handle( Geom2d_Curve )>& theCurves )
+ const QString& theNamePrefix,
+ const std::vector<TopoDS_Shape>& theShapes,
+ bool isUseIndices )
{
if( theDoc.IsNull() )
return false;
- int n = theCurves.size();
+ int n = theShapes.size();
+ int anIndex = 1;
for( int i=0; i<n; i++ )
{
- TopoDS_Edge anEdge = BRepBuilderAPI_MakeEdge2d( theCurves[i] ).Edge();
- BRepBuilderAPI_MakeWire aMakeWire;
- aMakeWire.Add( anEdge );
-
Handle( HYDROData_PolylineXY ) aPolyline =
Handle( HYDROData_PolylineXY )::DownCast( theDoc->CreateObject( KIND_POLYLINEXY ) );
if( aPolyline.IsNull() )
return false;
- aPolyline->SetShape( aMakeWire.Wire() );
- //TODO: set name
+ aPolyline->SetShape( theShapes[i] );
+
+ if( isUseIndices )
+ {
+ QString aNewName = theNamePrefix + "_" + QString::number( anIndex );
+ if( theDoc->FindObjectByName( aNewName ).IsNull() ) // the object with such a name is not found
+ aPolyline->SetName( aNewName );
+ anIndex++;
+ }
+ else
+ {
+ aPolyline->SetName( theNamePrefix );
+ }
}
return true;
}
+
+double HYDROData_PolylineOperator::ReduceDeflection(
+ const double theDeflection,
+ HYDROData_TopoCurve& theCurve,
+ int& thePieceCount)
+{
+ // Construct the approximating B-spline.
+ std::list<gp_XYZ> aPs;
+ if (!theCurve.ValuesInKnots(aPs))
+ {
+ return -1;
+ }
+
+ Handle(TColgp_HArray1OfPnt) aPs2 = new TColgp_HArray1OfPnt(1, aPs.size());
+ {
+ std::list<gp_XYZ>::const_iterator aLastPIt = aPs.end();
+ std::list<gp_XYZ>::const_iterator aPIt = aPs.begin();
+ for (int aPN = 1; aPIt != aLastPIt; ++aPN, ++aPIt)
+ {
+ aPs2->SetValue(aPN, *aPIt);
+ }
+ }
+ Handle(Geom_BSplineCurve) aBSpline2;
+ const bool isClosed = theCurve.IsClosed();
+ if (!CurveCreator_Utils::constructBSpline(aPs2, isClosed, aBSpline2))
+ {
+ return -1;
+ }
+
+ // Calculate the piece deflections.
+ std::deque<double> aSqDefls;
+ double aMaxSqDefl = 0;
+ std::list<TopoDS_Edge>& aEdges = theCurve.Edges();
+ std::list<TopoDS_Edge>::const_iterator aLastEIt = aEdges.end();
+ {
+ std::list<TopoDS_Edge>::const_iterator aEIt = aEdges.begin();
+ for (int aPrevKCount = 0; aEIt != aLastEIt; ++aEIt)
+ {
+ TopLoc_Location aLoc;
+ double aParams[2];
+ Handle(Geom_BSplineCurve) aBSpline = Handle(Geom_BSplineCurve)::DownCast(
+ BRep_Tool::Curve(*aEIt, aLoc, aParams[0], aParams[1]));
+ const int aKCount = aBSpline->NbKnots();
+ for (int aKN = 1; aKN < aKCount; ++aKN)
+ {
+ const double aParam =
+ (aBSpline->Knot(aKN) + aBSpline->Knot(aKN + 1)) * 0.5;
+ const double aParam2 = (aBSpline2->Knot(aPrevKCount + aKN) +
+ aBSpline2->Knot(aPrevKCount + aKN + 1)) * 0.5;
+ const double aSqDefl = Abs(aBSpline->Value(aParam).
+ SquareDistance(aBSpline2->Value(aParam2)));
+ aSqDefls.push_back(aSqDefl);
+ if (aMaxSqDefl < aSqDefl)
+ {
+ aMaxSqDefl = aSqDefl;
+ }
+ }
+ aPrevKCount += aKCount - 1;
+ }
+ }
+
+ // Check whether the reducing is necessary.
+ const double aMaxDefl = Sqrt(aMaxSqDefl);
+ if (aMaxDefl <= theDeflection)
+ {
+ return aMaxDefl;
+ }
+
+ // Reduce the deflections.
+ const double aThresSqDefl =
+ Max(aMaxSqDefl * 0.25, theDeflection * theDeflection);
+ std::list<TopoDS_Edge>::iterator aEIt = aEdges.begin();
+ std::deque<double>::const_iterator aSqDIt = aSqDefls.begin();
+ thePieceCount = 0;
+ for (; aEIt != aLastEIt; ++aEIt)
+ {
+ TopLoc_Location aLoc;
+ double aParams[2];
+ Handle(Geom_BSplineCurve) aBSpline = Handle(Geom_BSplineCurve)::DownCast(
+ BRep_Tool::Curve(*aEIt, aLoc, aParams[0], aParams[1]));
+ Handle(Geom_BSplineCurve) aBSpline2 =
+ Handle(Geom_BSplineCurve)::DownCast(aBSpline->Copy());
+ const int aKCount = aBSpline->NbKnots();
+ for (int aKN = 1; aKN < aKCount; ++aSqDIt, ++aKN)
+ {
+ if (*aSqDIt > aThresSqDefl)
+ {
+ aBSpline2->InsertKnot(
+ (aBSpline->Knot(aKN) + aBSpline->Knot(aKN + 1)) * 0.5);
+ }
+ }
+ TopoDS_Edge aEdge;
+ BRep_Builder().MakeEdge(aEdge, aBSpline2, Precision::Confusion());
+ BRep_Builder().Add(aEdge, TopExp::FirstVertex(*aEIt));
+ BRep_Builder().Add(aEdge, TopExp::LastVertex(*aEIt));
+ thePieceCount += aBSpline2->NbKnots() - 1;
+ *aEIt = aEdge;
+ }
+ return aMaxDefl;
+}