Salome HOME
44e68d973188c85c89c4607b73ff6eae9ca9f536
[modules/geom.git] / src / GEOMImpl / GEOMImpl_RevolutionDriver.cxx
1 // Copyright (C) 2007-2012  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_RevolutionDriver.hxx>
25
26 #include <GEOMImpl_IRevolution.hxx>
27 #include <GEOMImpl_Types.hxx>
28
29 #include <GEOM_Function.hxx>
30
31 #include <GEOMUtils.hxx>
32
33 #include <BRepPrimAPI_MakeRevol.hxx>
34 #include <BRepBuilderAPI_Transform.hxx>
35 #include <BRep_Tool.hxx>
36 #include <TopoDS.hxx>
37 #include <TopoDS_Shape.hxx>
38 #include <TopoDS_Edge.hxx>
39 #include <TopoDS_Vertex.hxx>
40 #include <TopAbs.hxx>
41 #include <TopExp.hxx>
42 #include <gp_Trsf.hxx>
43 #include <gp_Pnt.hxx>
44 #include <gp_Lin.hxx>
45 #include <gp_Dir.hxx>
46 #include <Precision.hxx>
47 #include <StdFail_NotDone.hxx>
48 #include <Standard_TypeMismatch.hxx>
49 #include <Standard_ConstructionError.hxx>
50
51 //=======================================================================
52 //function : GetID
53 //purpose  :
54 //======================================================================= 
55 const Standard_GUID& GEOMImpl_RevolutionDriver::GetID()
56 {
57   static Standard_GUID aRevolutionDriver("FF1BBB18-5D14-4df2-980B-3A668264EA16");
58   return aRevolutionDriver; 
59 }
60
61
62 //=======================================================================
63 //function : GEOMImpl_RevolutionDriver
64 //purpose  : 
65 //=======================================================================
66 GEOMImpl_RevolutionDriver::GEOMImpl_RevolutionDriver() 
67 {
68 }
69
70 //=======================================================================
71 //function : Execute
72 //purpose  :
73 //======================================================================= 
74 Standard_Integer GEOMImpl_RevolutionDriver::Execute(TFunction_Logbook& log) const
75 {
76   if (Label().IsNull()) return 0;    
77   Handle(GEOM_Function) aFunction = GEOM_Function::GetFunction(Label());
78
79   GEOMImpl_IRevolution aCI (aFunction);
80   Standard_Integer aType = aFunction->GetType();
81
82   TopoDS_Shape aShape;
83
84   if (aType == REVOLUTION_BASE_AXIS_ANGLE || aType == REVOLUTION_BASE_AXIS_ANGLE_2WAYS) {
85     Handle(GEOM_Function) aRefBase = aCI.GetBase();
86     Handle(GEOM_Function) aRefAxis = aCI.GetAxis();
87     TopoDS_Shape aShapeBase = aRefBase->GetValue();
88     TopoDS_Shape aShapeAxis = aRefAxis->GetValue();
89     if (aShapeAxis.ShapeType() != TopAbs_EDGE) {
90       Standard_TypeMismatch::Raise("Revolution Axis must be an edge");
91     }
92
93     TopoDS_Edge anE = TopoDS::Edge(aShapeAxis);
94     TopoDS_Vertex V1, V2;
95     TopExp::Vertices(anE, V1, V2, Standard_True);
96     if (V1.IsNull() || V2.IsNull()) {
97       Standard_ConstructionError::Raise("Bad edge for the Revolution Axis given");
98     }
99
100     gp_Vec aV (BRep_Tool::Pnt(V1), BRep_Tool::Pnt(V2));
101     if (aV.Magnitude() < Precision::Confusion()) {
102       Standard_ConstructionError::Raise
103         ("End vertices of edge, defining the Revolution Axis, are too close");
104     }
105
106     if (aShapeBase.ShapeType() == TopAbs_VERTEX) {
107       gp_Lin aL(BRep_Tool::Pnt(V1), gp_Dir(aV));
108       Standard_Real d = aL.Distance(BRep_Tool::Pnt(TopoDS::Vertex(aShapeBase)));
109       if (d < Precision::Confusion()) {
110         Standard_ConstructionError::Raise("Vertex to be rotated is too close to Revolution Axis");
111       }
112     }
113     double anAngle = aCI.GetAngle();
114     gp_Ax1 anAxis (BRep_Tool::Pnt(V1), aV);
115     if (aType == REVOLUTION_BASE_AXIS_ANGLE_2WAYS)
116       {
117         gp_Trsf aTrsf;
118         aTrsf.SetRotation(anAxis, ( -anAngle ));
119         BRepBuilderAPI_Transform aTransformation(aShapeBase, aTrsf, Standard_False);
120         aShapeBase = aTransformation.Shape();
121         anAngle = anAngle * 2;
122       }
123     BRepPrimAPI_MakeRevol MR (aShapeBase, anAxis, anAngle, Standard_False);
124     if (!MR.IsDone()) MR.Build();
125     if (!MR.IsDone()) StdFail_NotDone::Raise("Revolution algorithm has failed");
126     aShape = MR.Shape();
127   } else {
128   }
129
130   if (aShape.IsNull()) return 0;
131
132   TopoDS_Shape aRes = GEOMUtils::CompsolidToCompound(aShape);
133   aFunction->SetValue(aRes);
134
135   log.SetTouched(Label());
136
137   return 1;
138 }
139
140
141 //=======================================================================
142 //function :  GEOMImpl_RevolutionDriver_Type_
143 //purpose  :
144 //======================================================================= 
145 Standard_EXPORT Handle_Standard_Type& GEOMImpl_RevolutionDriver_Type_()
146 {
147
148   static Handle_Standard_Type aType1 = STANDARD_TYPE(TFunction_Driver);
149   if ( aType1.IsNull()) aType1 = STANDARD_TYPE(TFunction_Driver);
150   static Handle_Standard_Type aType2 = STANDARD_TYPE(MMgt_TShared);
151   if ( aType2.IsNull()) aType2 = STANDARD_TYPE(MMgt_TShared); 
152   static Handle_Standard_Type aType3 = STANDARD_TYPE(Standard_Transient);
153   if ( aType3.IsNull()) aType3 = STANDARD_TYPE(Standard_Transient);
154  
155
156   static Handle_Standard_Transient _Ancestors[]= {aType1,aType2,aType3,NULL};
157   static Handle_Standard_Type _aType = new Standard_Type("GEOMImpl_RevolutionDriver",
158                                                          sizeof(GEOMImpl_RevolutionDriver),
159                                                          1,
160                                                          (Standard_Address)_Ancestors,
161                                                          (Standard_Address)NULL);
162
163   return _aType;
164 }
165
166 //=======================================================================
167 //function : DownCast
168 //purpose  :
169 //======================================================================= 
170 const Handle(GEOMImpl_RevolutionDriver) Handle(GEOMImpl_RevolutionDriver)::DownCast(const Handle(Standard_Transient)& AnObject)
171 {
172   Handle(GEOMImpl_RevolutionDriver) _anOtherObject;
173
174   if (!AnObject.IsNull()) {
175      if (AnObject->IsKind(STANDARD_TYPE(GEOMImpl_RevolutionDriver))) {
176        _anOtherObject = Handle(GEOMImpl_RevolutionDriver)((Handle(GEOMImpl_RevolutionDriver)&)AnObject);
177      }
178   }
179
180   return _anOtherObject ;
181 }