]> SALOME platform Git repositories - modules/geom.git/blob - src/GEOMImpl/GEOMImpl_SplineDriver.cxx
Salome HOME
Fix error with GetExistingSubObjects
[modules/geom.git] / src / GEOMImpl / GEOMImpl_SplineDriver.cxx
1 //  Copyright (C) 2007-2010  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 //  This library is free software; you can redistribute it and/or
7 //  modify it under the terms of the GNU Lesser General Public
8 //  License as published by the Free Software Foundation; either
9 //  version 2.1 of the License.
10 //
11 //  This library is distributed in the hope that it will be useful,
12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 //  Lesser General Public License for more details.
15 //
16 //  You should have received a copy of the GNU Lesser General Public
17 //  License along with this library; if not, write to the Free Software
18 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21
22 #include <Standard_Stream.hxx>
23
24 #include <GEOMImpl_SplineDriver.hxx>
25 #include <GEOMImpl_ISpline.hxx>
26 #include <GEOMImpl_Types.hxx>
27 #include <GEOM_Function.hxx>
28
29 #include <BRepBuilderAPI_MakeEdge.hxx>
30 #include <BRep_Tool.hxx>
31
32 #include <TopAbs.hxx>
33 #include <TopExp.hxx>
34 #include <TopoDS.hxx>
35 #include <TopoDS_Shape.hxx>
36 #include <TopoDS_Edge.hxx>
37 #include <TopoDS_Vertex.hxx>
38
39 #include <Geom_BezierCurve.hxx>
40 //#include <GeomAPI_PointsToBSpline.hxx>
41 #include <GeomAPI_Interpolate.hxx>
42
43 #include <gp.hxx>
44 #include <gp_Pnt.hxx>
45 #include <gp_Circ.hxx>
46 #include <Precision.hxx>
47 #include <TColgp_Array1OfPnt.hxx>
48 #include <TColgp_HArray1OfPnt.hxx>
49
50 //=======================================================================
51 //function : GetID
52 //purpose  :
53 //=======================================================================
54 const Standard_GUID& GEOMImpl_SplineDriver::GetID()
55 {
56   static Standard_GUID aSplineDriver("FF1BBB33-5D14-4df2-980B-3A668264EA16");
57   return aSplineDriver;
58 }
59
60
61 //=======================================================================
62 //function : GEOMImpl_SplineDriver
63 //purpose  :
64 //=======================================================================
65 GEOMImpl_SplineDriver::GEOMImpl_SplineDriver()
66 {
67 }
68
69 //=======================================================================
70 //function : Execute
71 //purpose  :
72 //=======================================================================
73 Standard_Integer GEOMImpl_SplineDriver::Execute(TFunction_Logbook& log) const
74 {
75   if (Label().IsNull()) return 0;
76   Handle(GEOM_Function) aFunction = GEOM_Function::GetFunction(Label());
77
78   GEOMImpl_ISpline aCI (aFunction);
79   Standard_Integer aType = aFunction->GetType();
80
81   TopoDS_Shape aShape;
82
83   if (aType == SPLINE_BEZIER || aType == SPLINE_INTERPOLATION) {
84     int ind, aLen = aCI.GetLength();
85     if (aLen < 2) return 0;
86     Standard_Boolean isSeveral = Standard_False;
87     gp_Pnt aPrevP;
88     int aRealLen = aLen;
89     if (aType == SPLINE_BEZIER && aCI.GetIsClosed()) {
90       Handle(GEOM_Function) aFPoint = aCI.GetPoint(1);
91       TopoDS_Shape aFirstPnt = aFPoint->GetValue();
92       TopoDS_Vertex aV1 = TopoDS::Vertex(aFirstPnt);
93
94       Handle(GEOM_Function) aLPoint = aCI.GetPoint(aLen);
95       TopoDS_Shape aLastPnt = aLPoint->GetValue();
96       TopoDS_Vertex aV2 = TopoDS::Vertex(aLastPnt);
97
98       if (!aV1.IsNull() && !aV2.IsNull() && !aV1.IsSame(aV2)) {
99         aRealLen++;
100       }
101     }
102     TColgp_Array1OfPnt CurvePoints (1, aRealLen);
103     for (ind = 1; ind <= aLen; ind++) {
104       Handle(GEOM_Function) aRefPoint = aCI.GetPoint(ind);
105       TopoDS_Shape aShapePnt = aRefPoint->GetValue();
106       if (aShapePnt.ShapeType() == TopAbs_VERTEX) {
107         gp_Pnt aP = BRep_Tool::Pnt(TopoDS::Vertex(aShapePnt));
108         if (!isSeveral && ind > 1) {
109           if (aP.Distance(aPrevP) > Precision::Confusion()) {
110             isSeveral = Standard_True;
111           }
112         }
113         CurvePoints.SetValue(ind, aP);
114         aPrevP = aP;
115       }
116     }
117     if (aType == SPLINE_BEZIER) {
118       if (!isSeveral) {
119         Standard_ConstructionError::Raise("Points for Bezier Curve are too close");
120       }
121       if (aRealLen > aLen) { // set last point equal to first for the closed curve
122         CurvePoints.SetValue(aRealLen, CurvePoints.Value(1));
123       }
124       Handle(Geom_BezierCurve) GBC = new Geom_BezierCurve(CurvePoints);
125       aShape = BRepBuilderAPI_MakeEdge(GBC).Edge();
126     } else {
127       //GeomAPI_PointsToBSpline GBC (CurvePoints);
128       //aShape = BRepBuilderAPI_MakeEdge(GBC).Edge();
129
130       Handle(TColgp_HArray1OfPnt) aHCurvePoints = new TColgp_HArray1OfPnt(1, aLen);
131
132       if (aCI.GetDoReordering()) {
133         for (int curInd = 1; curInd < aLen - 1; curInd++) {
134           gp_Pnt curPnt = CurvePoints.Value(curInd);
135           int nearInd = 0;
136           double nearDist = RealLast();
137           for (ind = curInd + 1; ind <= aLen; ind++) {
138             double dist = curPnt.SquareDistance(CurvePoints.Value(ind));
139             if (dist < nearDist && (nearDist - dist) > Precision::Confusion()) {
140               nearInd = ind;
141               nearDist = dist;
142             }
143           }
144           if (nearInd > 0 && nearInd != curInd + 1) {
145             // Keep given order of points to use it in case of equidistant candidates
146             //               .-<---<-.
147             //              /         \
148             // o  o  o  c  o->o->o->o->n  o  o
149             //          |  |           |
150             //     curInd  curInd+1    nearInd
151             gp_Pnt nearPnt = CurvePoints.Value(nearInd);
152             for (ind = nearInd; ind > curInd + 1; ind--) {
153               CurvePoints.SetValue(ind, CurvePoints.Value(ind - 1));
154             }
155             CurvePoints.SetValue(curInd + 1, nearPnt);
156           }
157         }
158         for (ind = 1; ind <= aLen; ind++) {
159           aHCurvePoints->SetValue(ind, CurvePoints.Value(ind));
160         }
161       }
162       else {
163         for (ind = 1; ind <= aLen; ind++) {
164           aHCurvePoints->SetValue(ind, CurvePoints.Value(ind));
165         }
166       }
167
168       bool isClosed = aCI.GetIsClosed();
169       GeomAPI_Interpolate GBC (aHCurvePoints, isClosed, gp::Resolution());
170       GBC.Perform();
171       if (GBC.IsDone())
172         aShape = BRepBuilderAPI_MakeEdge(GBC.Curve()).Edge();
173       else
174         return 0;
175     }
176   }
177   else {
178   }
179
180   if (aShape.IsNull()) return 0;
181
182   aFunction->SetValue(aShape);
183
184   log.SetTouched(Label());
185
186   return 1;
187 }
188
189
190 //=======================================================================
191 //function :  GEOMImpl_SplineDriver_Type_
192 //purpose  :
193 //=======================================================================
194 Standard_EXPORT Handle_Standard_Type& GEOMImpl_SplineDriver_Type_()
195 {
196
197   static Handle_Standard_Type aType1 = STANDARD_TYPE(TFunction_Driver);
198   if ( aType1.IsNull()) aType1 = STANDARD_TYPE(TFunction_Driver);
199   static Handle_Standard_Type aType2 = STANDARD_TYPE(MMgt_TShared);
200   if ( aType2.IsNull()) aType2 = STANDARD_TYPE(MMgt_TShared);
201   static Handle_Standard_Type aType3 = STANDARD_TYPE(Standard_Transient);
202   if ( aType3.IsNull()) aType3 = STANDARD_TYPE(Standard_Transient);
203
204
205   static Handle_Standard_Transient _Ancestors[]= {aType1,aType2,aType3,NULL};
206   static Handle_Standard_Type _aType = new Standard_Type("GEOMImpl_SplineDriver",
207                                                          sizeof(GEOMImpl_SplineDriver),
208                                                          1,
209                                                          (Standard_Address)_Ancestors,
210                                                          (Standard_Address)NULL);
211
212   return _aType;
213 }
214
215 //=======================================================================
216 //function : DownCast
217 //purpose  :
218 //=======================================================================
219 const Handle(GEOMImpl_SplineDriver) Handle(GEOMImpl_SplineDriver)::DownCast(const Handle(Standard_Transient)& AnObject)
220 {
221   Handle(GEOMImpl_SplineDriver) _anOtherObject;
222
223   if (!AnObject.IsNull()) {
224      if (AnObject->IsKind(STANDARD_TYPE(GEOMImpl_SplineDriver))) {
225        _anOtherObject = Handle(GEOMImpl_SplineDriver)((Handle(GEOMImpl_SplineDriver)&)AnObject);
226      }
227   }
228
229   return _anOtherObject ;
230 }