Salome HOME
Merge branch 'BR_v14_rc' of ssh://git.salome-platform.org/modules/hydro into BR_v14_rc
[modules/hydro.git] / src / HYDROData / HYDROData_PolylineOperator.cxx
1 // Copyright (C) 2014-2015  EDF-R&D
2 // This library is free software; you can redistribute it and/or
3 // modify it under the terms of the GNU Lesser General Public
4 // License as published by the Free Software Foundation; either
5 // version 2.1 of the License, or (at your option) any later version.
6 //
7 // This library is distributed in the hope that it will be useful,
8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
10 // Lesser General Public License for more details.
11 //
12 // You should have received a copy of the GNU Lesser General Public
13 // License along with this library; if not, write to the Free Software
14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
15 //
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
17 //
18
19 #include <HYDROData_PolylineOperator.h>
20 #include <HYDROData_Document.h>
21 #include <BRepBuilderAPI_MakeEdge2d.hxx>
22 #include <BRepBuilderAPI_MakeWire.hxx>
23 #include <TopoDS_Edge.hxx>
24 #include <TopoDS_Wire.hxx>
25
26 template<class T> void append( std::vector<T>& theList, const std::vector<T>& theList2 )
27 {
28   int aSize = theList.size();
29   int aNewSize = aSize + theList2.size();
30
31   if( aSize==aNewSize )
32     return;
33
34   theList.resize( aNewSize );
35   for( int i=aSize, j=0; i<aNewSize; i++, j++ )
36     theList[i] = theList2[j];
37 }
38
39
40 bool HYDROData_PolylineOperator::Split( const Handle( HYDROData_Document )& theDoc,
41                                         const TCollection_AsciiString& theNamePrefix,
42                                         const Handle( HYDROData_PolylineXY )& thePolyline,
43                                         const gp_Pnt2d& thePoint ) const
44 {
45   std::vector<gp_Pnt2d> aPointsList( 1 );
46   aPointsList[0] = thePoint;
47   std::vector<Handle( Geom2d_Curve )> aCurves = GetCurves( thePolyline );
48   bool isOK = true;
49   for( int i=0, n=aCurves.size(); i<n; i++ )
50   {
51     std::vector<Handle( Geom2d_Curve )> aCurvesList = Split( aCurves[i], aPointsList );
52     bool isLocalOK = CreatePolylines( theDoc, theNamePrefix, aCurvesList );
53     isOK = isOK && isLocalOK;
54   }
55   return isOK;
56 }
57
58 bool HYDROData_PolylineOperator::Split( const Handle( HYDROData_Document )& theDoc,
59                                         const TCollection_AsciiString& theNamePrefix,
60                                         const Handle( HYDROData_PolylineXY )& thePolyline,
61                                         const Handle( HYDROData_PolylineXY )& theTool ) const
62 {
63   std::vector<Handle( Geom2d_Curve )> aCurves = GetCurves( thePolyline );
64   std::vector<Handle( Geom2d_Curve )> aToolCurves = GetCurves( theTool );
65   bool isOK = true;
66   for( int i=0, n=aCurves.size(); i<n; i++ )
67     for( int j=0, m=aToolCurves.size(); j<m; j++ )
68     {
69       std::vector<gp_Pnt2d> aPointsList = Intersection( aCurves[i], aToolCurves[j] );
70       std::vector<Handle( Geom2d_Curve )> aCurvesList = Split( aCurves[i], aPointsList );
71       bool isLocalOK = CreatePolylines( theDoc, theNamePrefix, aCurvesList );
72       isOK = isOK && isLocalOK;
73     }
74   return isOK;
75 }
76
77 bool HYDROData_PolylineOperator::Split( const Handle( HYDROData_Document )& theDoc,
78                                         const TCollection_AsciiString& theNamePrefix,
79                                         const HYDROData_SequenceOfObjects& thePolylines )
80 {
81   int f = thePolylines.Lower(), l = thePolylines.Upper();
82   bool isOK = true;
83   std::vector<Handle( Geom2d_Curve )> anAllCurves;
84   for( int i=f; i<=l; i++ )
85   {
86     Handle( HYDROData_PolylineXY ) aPolyline = Handle( HYDROData_PolylineXY )::DownCast( thePolylines.Value( i ) );
87     std::vector<Handle( Geom2d_Curve )> aCurves = GetCurves( aPolyline );
88     append( anAllCurves, aCurves );
89   }
90
91   for( int i=0, n=anAllCurves.size(); i<n; i++ )
92   {
93     std::vector<gp_Pnt2d> aCompletePointsList;
94     for( int j=0; j<n; j++ )
95     {
96       if( i==j )
97         continue;
98       std::vector<gp_Pnt2d> aPointsList = Intersection( anAllCurves[i], anAllCurves[j] );
99       append( aCompletePointsList, aPointsList );
100     }
101     std::vector<Handle( Geom2d_Curve )> aCurvesList = Split( anAllCurves[i], aCompletePointsList );
102     bool isLocalOK = CreatePolylines( theDoc, theNamePrefix, aCurvesList );
103     isOK = isOK && isLocalOK;
104   }
105   return isOK;
106 }
107
108 bool HYDROData_PolylineOperator::Merge( const Handle( HYDROData_Document )& theDoc,
109                                         const TCollection_AsciiString& theName,
110                                         const HYDROData_SequenceOfObjects& thePolylines )
111 {
112   //TODO
113   return true;
114 }
115
116 std::vector<Handle( Geom2d_Curve )> HYDROData_PolylineOperator::GetCurves( const Handle( HYDROData_PolylineXY )& thePolyline )
117 {
118   std::vector<Handle( Geom2d_Curve )> aResult;
119   //TODO
120   return aResult;
121 }
122
123 std::vector<gp_Pnt2d> HYDROData_PolylineOperator::Intersection( const Handle( Geom2d_Curve )& theCurve,
124                                                                 const Handle( Geom2d_Curve )& theTool )
125 {
126   std::vector<gp_Pnt2d> aResult;
127   //TODO
128   return aResult;
129 }
130
131 std::vector<Handle( Geom2d_Curve )> HYDROData_PolylineOperator::Split( const Handle( Geom2d_Curve )& theCurve,
132                                                                        const std::vector<gp_Pnt2d>& thePoints )
133 {
134   std::vector<Handle( Geom2d_Curve )> aResult;
135   //TODO
136   return aResult;
137 }
138
139 bool HYDROData_PolylineOperator::CreatePolylines( const Handle( HYDROData_Document )& theDoc,
140                                                   const TCollection_AsciiString& theNamePrefix,
141                                                   const std::vector<Handle( Geom2d_Curve )>& theCurves )
142 {
143   if( theDoc.IsNull() )
144     return false;
145
146   int n = theCurves.size();
147   for( int i=0; i<n; i++ )
148   {
149     TopoDS_Edge anEdge = BRepBuilderAPI_MakeEdge2d( theCurves[i] ).Edge();
150     BRepBuilderAPI_MakeWire aMakeWire;
151     aMakeWire.Add( anEdge );
152
153     Handle( HYDROData_PolylineXY ) aPolyline = 
154       Handle( HYDROData_PolylineXY )::DownCast( theDoc->CreateObject( KIND_POLYLINEXY ) );
155     if( aPolyline.IsNull() )
156       return false;
157
158     aPolyline->SetShape( aMakeWire.Wire() );
159     //TODO: set name
160   }
161   return true;
162 }