0028956: Coding Rules - eliminate several trivial compiler warnings
[occt.git] / src / DNaming / DNaming_Line3DDriver.cxx
1 // Created on: 2010-02-26
2 // Created by: Sergey ZARITCHNY
3 // Copyright (c) 2010-2014 OPEN CASCADE SAS
4 //
5 // This file is part of Open CASCADE Technology software library.
6 //
7 // This library is free software; you can redistribute it and/or modify it under
8 // the terms of the GNU Lesser General Public License version 2.1 as published
9 // by the Free Software Foundation, with special exception defined in the file
10 // OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
11 // distribution for complete text of the license and disclaimer of any warranty.
12 //
13 // Alternatively, this file may be used under the terms of Open CASCADE
14 // commercial license or contractual agreement.
15
16
17 #include <BRep_Builder.hxx>
18 #include <BRep_Tool.hxx>
19 #include <BRepAlgo.hxx>
20 #include <BRepBuilderAPI_MakeEdge.hxx>
21 #include <BRepBuilderAPI_MakeWire.hxx>
22 #include <DNaming.hxx>
23 #include <DNaming_Line3DDriver.hxx>
24 #include <Geom_BSplineCurve.hxx>
25 #include <GeomAPI_Interpolate.hxx>
26 #include <gp_Pnt.hxx>
27 #include <gp_Trsf.hxx>
28 #include <gp_Vec.hxx>
29 #include <ModelDefinitions.hxx>
30 #include <Precision.hxx>
31 #include <Standard_GUID.hxx>
32 #include <Standard_Type.hxx>
33 #include <TColgp_HArray1OfPnt.hxx>
34 #include <TDataStd_Integer.hxx>
35 #include <TDataStd_Name.hxx>
36 #include <TDataStd_Real.hxx>
37 #include <TDF_AttributeMap.hxx>
38 #include <TDF_Label.hxx>
39 #include <TDF_MapIteratorOfAttributeMap.hxx>
40 #include <TFunction_Function.hxx>
41 #include <TFunction_Logbook.hxx>
42 #include <TNaming.hxx>
43 #include <TNaming_Builder.hxx>
44 #include <TNaming_NamedShape.hxx>
45 #include <TopAbs.hxx>
46 #include <TopExp.hxx>
47 #include <TopExp_Explorer.hxx>
48 #include <TopLoc_Location.hxx>
49 #include <TopoDS.hxx>
50 #include <TopoDS_Shape.hxx>
51 #include <TopoDS_Vertex.hxx>
52 #include <TopoDS_Wire.hxx>
53 #include <TopTools_Array1OfShape.hxx>
54 #include <TopTools_HSequenceOfShape.hxx>
55
56 IMPLEMENT_STANDARD_RTTIEXT(DNaming_Line3DDriver,TFunction_Driver)
57
58 //=======================================================================
59 //function : DNaming_Line3DDriver
60 //purpose  : Constructor
61 //=======================================================================
62 DNaming_Line3DDriver::DNaming_Line3DDriver()
63 {}
64
65 //=======================================================================
66 //function : Validate
67 //purpose  : Validates labels of a function in <theLog>.
68 //=======================================================================
69 void DNaming_Line3DDriver::Validate(Handle(TFunction_Logbook)&) const
70 {}
71
72 //=======================================================================
73 //function : MustExecute
74 //purpose  : Analyse in <theLog> if the loaded function must be executed
75 //=======================================================================
76 Standard_Boolean DNaming_Line3DDriver::MustExecute(const Handle(TFunction_Logbook)&) const
77 {
78   return Standard_True;
79 }
80
81 //=======================================================================
82 //function : Execute
83 //purpose  : Execute the function
84 //=======================================================================
85 Standard_Integer DNaming_Line3DDriver::Execute(Handle(TFunction_Logbook)& theLog) const
86 {
87   Handle(TFunction_Function) aFunction;
88   Label().FindAttribute(TFunction_Function::GetID(),aFunction);
89   if(aFunction.IsNull()) return -1;
90
91   aFunction->SetFailure(NOTDONE);  
92   Handle(TNaming_NamedShape) aPrevILine3D = DNaming::GetFunctionResult(aFunction);
93 // Save location
94   TopLoc_Location aLocation;
95   if (!aPrevILine3D.IsNull() && !aPrevILine3D->IsEmpty()) {
96     aLocation = aPrevILine3D->Get().Location();
97   }
98
99   const Standard_Integer aType = DNaming::GetInteger(aFunction, LINE3D_TYPE)->Get();
100   Standard_Boolean isClosed = (aType != 0);
101   Standard_Integer aCounter(0), aLength  = DNaming::GetInteger(aFunction, LINE3D_PNTNB)->Get();
102   if(aLength < 2) {
103     aFunction->SetFailure(WRONG_ARGUMENT);
104     return -1;
105   }
106
107   Handle(TNaming_NamedShape) aNS1, aNS2;
108   BRepBuilderAPI_MakeWire aMakeWire;
109   TopoDS_Wire aWire;
110   TopoDS_Shape aShape1, aShape2;
111   
112   TopTools_Array1OfShape anArV(1,aLength); //aLength - number of points
113   for(aCounter = 1; aCounter <= aLength-1; aCounter++) {       
114     Handle(TDataStd_UAttribute) aRefP1 = DNaming::GetObjectArg(aFunction, (LINE3D_TYPE + aCounter));
115     aNS1 = DNaming::GetObjectValue(aRefP1);
116     Handle(TDataStd_UAttribute) aRefP2 = DNaming::GetObjectArg(aFunction, (LINE3D_TYPE + aCounter +1));
117     aNS2 =  DNaming::GetObjectValue(aRefP2);
118 #ifdef OCCT_DEBUG
119     if(!aNS1->IsEmpty()) {
120       aShape1 = aNS1->Get();
121       const gp_Pnt aDebPoint = BRep_Tool::Pnt(TopoDS::Vertex(aShape1));
122       (void )aDebPoint;
123       //        cout << aCounter << " X = " <<  aDebPoint.X() << " Y = " <<  aDebPoint.Y() << " Z = " <<  aDebPoint.Z() << endl;
124     } else
125       cout << " Line3DDriver:: NS1 is empty" << endl;
126     if(!aNS2->IsEmpty()) {
127       aShape2 = aNS2->Get();
128       const gp_Pnt aDebPoint = BRep_Tool::Pnt(TopoDS::Vertex(aShape2));
129       (void )aDebPoint;
130       //      cout << aCounter+1 << " X = " <<  aDebPoint.X() << " Y = " <<  aDebPoint.Y() << " Z = " <<  aDebPoint.Z() << endl;
131     } else
132       cout << " Line3DDriver:: NS2 is empty" << endl;
133 #endif
134     if(aNS1->IsEmpty() || aNS2->IsEmpty() || 
135        aNS1->Get().IsNull() || aNS2->Get().IsNull()) {
136       aFunction->SetFailure(WRONG_ARGUMENT);
137       return -1;
138     }
139     aShape1 = aNS1->Get();
140     aShape2 = aNS2->Get();
141 #ifdef OCCT_DEBUG
142     const gp_Pnt aDebPoint1 = BRep_Tool::Pnt(TopoDS::Vertex(aShape1));
143     const gp_Pnt aDebPoint2 = BRep_Tool::Pnt(TopoDS::Vertex(aShape2));
144     //      cout << aCounter   << " X = " <<  aDebPoint1.X() << " Y = " <<  aDebPoint1.Y() << " Z = " <<  aDebPoint1.Z() << endl;
145     //      cout << aCounter+1 << " X = " <<  aDebPoint2.X() << " Y = " <<  aDebPoint2.Y() << " Z = " <<  aDebPoint2.Z() << endl;
146     (void )aDebPoint1;
147     (void )aDebPoint2;
148 #endif 
149     if(aShape1.ShapeType() != TopAbs_VERTEX || aShape2.ShapeType() != TopAbs_VERTEX) {
150       aFunction->SetFailure(WRONG_ARGUMENT);
151       return -1;
152     }
153     BRepBuilderAPI_MakeEdge aMakeEdge(TopoDS::Vertex(aShape1), TopoDS::Vertex(aShape2));
154     if(aMakeEdge.IsDone()) {
155       anArV.SetValue(aCounter,   aShape1);
156       anArV.SetValue(aCounter+1, aShape2);
157       aMakeWire.Add(aMakeEdge.Edge());
158     }
159   }
160 //  } else // closed
161   if(isClosed) { 
162     Handle(TDataStd_UAttribute) aRefP1 = DNaming::GetObjectArg(aFunction, (LINE3D_TYPE + 1));
163     aNS1 = DNaming::GetObjectValue(aRefP1);
164     aShape1 = aNS1->Get();
165     BRepBuilderAPI_MakeEdge aMakeEdge(TopoDS::Vertex(aShape2), TopoDS::Vertex(aShape1));
166     if(aMakeEdge.IsDone()) aMakeWire.Add(aMakeEdge.Edge());
167   }
168   if(aMakeWire.IsDone()) aWire = aMakeWire.Wire(); 
169   
170
171   if(aWire.IsNull()) {
172     aFunction->SetFailure(ALGO_FAILED);
173     return -1;
174   }
175
176   if (!BRepAlgo::IsValid(aWire)) {
177     aFunction->SetFailure(RESULT_NOT_VALID);
178     return -1;
179   }
180
181   TDF_Label aResultLabel = RESPOSITION(aFunction); //aFunction->GetResult()->Label();
182   try {  
183     LoadNamingDS(aResultLabel, aWire, anArV, isClosed);
184
185   } catch (Standard_Failure) {
186     aFunction->SetFailure(NAMING_FAILED);
187     return -1;
188   }
189
190   // restore location
191   if(!aLocation.IsIdentity())
192     TNaming::Displace(aResultLabel, aLocation, Standard_True);
193
194   theLog->SetValid(aResultLabel, Standard_True);  
195
196   aFunction->SetFailure(DONE);
197   return 0;
198 }
199
200 //=======================================================================
201 //function : LoadNamingDS
202 //purpose  : Loads a Line3D in a data framework
203 //=======================================================================
204 void DNaming_Line3DDriver::LoadNamingDS (const TDF_Label& theResultLabel, 
205                                          const TopoDS_Wire& theWire, 
206                                          const TopTools_Array1OfShape& theArV,
207                                          const Standard_Boolean isClosed) const
208 {
209   if(theWire.IsNull()) return;
210 //Wire
211   TNaming_Builder aWBuilder(theResultLabel);
212   aWBuilder.Generated(theWire);
213 #ifdef OCCT_DEBUG
214   TDataStd_Name::Set(theResultLabel, "Line3DCurve");
215 #endif
216   Standard_Integer aLength = theArV.Length();
217   if(aLength < 2) return;
218   TopoDS_Shape aShape;
219   TopTools_Array1OfShape anArE(1,aLength);
220   TopoDS_Vertex aFirst, aLast;
221   for(Standard_Integer i = 1;i < aLength;i++) {
222     gp_Pnt aP1  = BRep_Tool::Pnt(TopoDS::Vertex(theArV.Value(i)));
223     gp_Pnt aP2  = BRep_Tool::Pnt(TopoDS::Vertex(theArV.Value(i+1)));
224     Standard_Boolean aFound = Standard_False;
225     TopExp_Explorer anExp(theWire,TopAbs_EDGE);
226     for(;anExp.More();anExp.Next()) {
227       const TopoDS_Edge& anE = TopoDS::Edge(anExp.Current());
228       TopoDS_Vertex aV, aV1, aV2;
229       TopExp::Vertices(anE, aV1, aV2);
230       gp_Pnt aPE1 = BRep_Tool::Pnt(aV1);
231       gp_Pnt aPE2 = BRep_Tool::Pnt(aV2);
232       if(aP1.IsEqual(aPE1, Precision::Confusion()) && aP2.IsEqual(aPE2, Precision::Confusion())) {
233         anArE.SetValue(i, anE);
234         aFound = Standard_True;
235         break;
236       }
237     }
238     if(!aFound)
239       anArE.SetValue(i, aShape);
240     else aFound = Standard_False;
241   }
242   if(isClosed) {
243     Standard_Boolean aFound = Standard_False;
244     gp_Pnt aP1  = BRep_Tool::Pnt(TopoDS::Vertex(theArV.Value(aLength)));
245     gp_Pnt aP2  = BRep_Tool::Pnt(TopoDS::Vertex(theArV.Value(1)));
246     TopExp_Explorer anExp(theWire,TopAbs_EDGE);
247     for(;anExp.More();anExp.Next()) {
248       const TopoDS_Edge& anE = TopoDS::Edge(anExp.Current());
249       TopoDS_Vertex aV, aV1, aV2;
250       TopExp::Vertices(anE, aV1, aV2);
251       gp_Pnt aPE1 = BRep_Tool::Pnt(aV1);
252       gp_Pnt aPE2 = BRep_Tool::Pnt(aV2);
253       if(aP1.IsEqual(aPE1, Precision::Confusion()) && aP2.IsEqual(aPE2, Precision::Confusion())) {
254         anArE.SetValue(aLength, anE);
255         aFound = Standard_True;
256      // closed case
257         aFirst = aV2;
258         aLast  = aV1;
259         break;
260       }
261     }
262     if(!aFound)
263       anArE.SetValue(aLength, aShape);
264   } else {
265     // open
266     anArE.SetValue(aLength, aShape); // last edge
267     TopExp::Vertices(theWire, aFirst, aLast);
268   }
269
270   // put edges
271   // from 1:aLength - edges
272   for(Standard_Integer i1 = 1;i1 <= aLength ;i1++) {
273     TDF_Label aLab = theResultLabel.FindChild(i1);
274     if(!anArE.Value(i1).IsNull()) {
275       TNaming_Builder aBuilder(aLab);
276       aBuilder.Generated(anArE.Value(i1));
277     } else {
278       Handle(TNaming_NamedShape) aNS;
279       if(aLab.FindAttribute(TNaming_NamedShape::GetID(), aNS))
280         TNaming_Builder aB(aLab);
281     }
282   }
283   //put vertexes
284   // aLength +1 - first vertex
285   // aLength +2 - last vertex
286   TDF_Label aLab1 = theResultLabel.FindChild(aLength+1);
287   TDF_Label aLab2 = theResultLabel.FindChild(aLength+2);
288   if(!aFirst.IsNull()) {
289     TNaming_Builder aBuilder(aLab1);
290     aBuilder.Generated(aFirst);
291   } else {
292     Handle(TNaming_NamedShape) aNS;
293     if(aLab1.FindAttribute(TNaming_NamedShape::GetID(), aNS))
294       TNaming_Builder aB(aLab1);
295   }
296   if(!aLast.IsNull()) {
297     TNaming_Builder aBuilder(aLab2);
298     aBuilder.Generated(aLast);
299   } else {
300     Handle(TNaming_NamedShape) aNS;
301     if(aLab2.FindAttribute(TNaming_NamedShape::GetID(), aNS))
302       TNaming_Builder aB(aLab2);
303   }
304 }