]> SALOME platform Git repositories - modules/geom.git/blob - src/GEOMImpl/GEOMImpl_TranslateDriver.cxx
Salome HOME
Add new command GetShapesOnBox.
[modules/geom.git] / src / GEOMImpl / GEOMImpl_TranslateDriver.cxx
1 // Copyright (C) 2005  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
2 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
3 // 
4 // This library is free software; you can redistribute it and/or
5 // modify it under the terms of the GNU Lesser General Public
6 // License as published by the Free Software Foundation; either 
7 // version 2.1 of the License.
8 // 
9 // This library is distributed in the hope that it will be useful 
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of 
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU 
12 // Lesser General Public License for more details.
13 //
14 // You should have received a copy of the GNU Lesser General Public  
15 // License along with this library; if not, write to the Free Software 
16 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
17 //
18 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 //
20
21 #include <Standard_Stream.hxx>
22
23 #include <GEOMImpl_TranslateDriver.hxx>
24 #include <GEOMImpl_ITranslate.hxx>
25 #include <GEOMImpl_Types.hxx>
26 #include <GEOM_Function.hxx>
27 #include <gp_Trsf.hxx>
28 #include <gp_Pnt.hxx>
29 #include <gp_Vec.hxx>
30 #include <BRepBuilderAPI_Transform.hxx>
31 #include <BRep_Builder.hxx>
32 #include <TopoDS.hxx>
33 #include <TopoDS_Shape.hxx>
34 #include <TopoDS_Edge.hxx>
35 #include <TopoDS_Compound.hxx>
36 #include <TopAbs.hxx>
37 #include <TopExp.hxx>
38 #include <TopoDS_Vertex.hxx>
39 #include <TopoDS_Edge.hxx>
40 #include <BRep_Tool.hxx>
41
42 //=======================================================================
43 //function : GetID
44 //purpose  :
45 //======================================================================= 
46 const Standard_GUID& GEOMImpl_TranslateDriver::GetID()
47 {
48   static Standard_GUID aTranslateDriver("FF1BBB03-5D14-4df2-980B-3A668264EA16");
49   return aTranslateDriver; 
50 }
51
52
53 //=======================================================================
54 //function : GEOMImpl_TranslateDriver
55 //purpose  : 
56 //=======================================================================
57
58 GEOMImpl_TranslateDriver::GEOMImpl_TranslateDriver() 
59 {
60 }
61
62 //=======================================================================
63 //function : Execute
64 //purpose  :
65 //======================================================================= 
66 Standard_Integer GEOMImpl_TranslateDriver::Execute(TFunction_Logbook& log) const
67 {
68   if(Label().IsNull())  return 0;    
69   Handle(GEOM_Function) aFunction = GEOM_Function::GetFunction(Label());
70
71   if(aFunction.IsNull()) return 0;
72
73   GEOMImpl_ITranslate TI(aFunction);
74   gp_Trsf aTrsf;
75   gp_Pnt aP1, aP2;
76   Standard_Integer aType = aFunction->GetType();
77
78   Handle(GEOM_Function) anOriginalFunction = TI.GetOriginal();
79   if(anOriginalFunction.IsNull()) return 0;
80   TopoDS_Shape aShape, anOriginal = anOriginalFunction->GetValue();
81   if(anOriginal.IsNull()) return 0;
82
83   if(aType == TRANSLATE_TWO_POINTS || aType == TRANSLATE_TWO_POINTS_COPY) {
84     Handle(GEOM_Function) aPoint1 = TI.GetPoint1();
85     Handle(GEOM_Function) aPoint2 = TI.GetPoint2();
86     if(aPoint1.IsNull() || aPoint2.IsNull()) return 0;
87     TopoDS_Shape aV1 = aPoint1->GetValue();
88     TopoDS_Shape aV2 = aPoint2->GetValue();
89     if(aV1.IsNull() || aV1.ShapeType() != TopAbs_VERTEX) return 0;
90     if(aV2.IsNull() || aV2.ShapeType() != TopAbs_VERTEX) return 0;
91
92     aP1 = BRep_Tool::Pnt(TopoDS::Vertex(aV1));
93     aP2 = BRep_Tool::Pnt(TopoDS::Vertex(aV2));
94
95     aTrsf.SetTranslation(aP1, aP2);
96     BRepBuilderAPI_Transform aTransformation(anOriginal, aTrsf, Standard_False);
97     aShape = aTransformation.Shape();
98
99   } else if(aType == TRANSLATE_VECTOR || aType == TRANSLATE_VECTOR_COPY) {
100     Handle(GEOM_Function) aVector = TI.GetVector();
101     if(aVector.IsNull()) return 0;
102     TopoDS_Shape aV = aVector->GetValue();
103     if(aV.IsNull() || aV.ShapeType() != TopAbs_EDGE) return 0;
104     TopoDS_Edge anEdge = TopoDS::Edge(aV);
105
106     aP1 = BRep_Tool::Pnt(TopExp::FirstVertex(anEdge));
107     aP2 = BRep_Tool::Pnt(TopExp::LastVertex(anEdge));
108
109     aTrsf.SetTranslation(aP1, aP2);
110     BRepBuilderAPI_Transform aTransformation(anOriginal, aTrsf, Standard_False);
111     aShape = aTransformation.Shape();
112   }
113   else if(aType == TRANSLATE_XYZ || aType == TRANSLATE_XYZ_COPY) {
114     gp_Vec aVec(TI.GetDX(), TI.GetDY(), TI.GetDZ());
115     aTrsf.SetTranslation(aVec);
116     BRepBuilderAPI_Transform aTransformation(anOriginal, aTrsf, Standard_False);
117     aShape = aTransformation.Shape();
118   }
119   else if(aType == TRANSLATE_1D) {
120     Standard_Real DX, DY, DZ, step = TI.GetStep1();
121     Standard_Integer nbtimes = TI.GetNbIter1();
122     gp_Vec myVec ;
123     TopoDS_Compound aCompound;
124     BRep_Builder B;
125     B.MakeCompound( aCompound );
126
127     Handle(GEOM_Function) aVector = TI.GetVector();
128     if(aVector.IsNull()) return 0;
129     TopoDS_Shape aV = aVector->GetValue();
130     if(aV.IsNull() || aV.ShapeType() != TopAbs_EDGE) return 0;
131     TopoDS_Edge anEdge = TopoDS::Edge(aV);
132
133     gp_Vec Vec(BRep_Tool::Pnt(TopExp::FirstVertex(anEdge)), BRep_Tool::Pnt(TopExp::LastVertex(anEdge)));
134     Vec.Normalize();
135
136     for (int i = 0; i < nbtimes; i++ ) {
137       DX = i * step * Vec.X() ;
138       DY = i * step * Vec.Y() ;
139       DZ = i * step * Vec.Z() ;
140       myVec.SetCoord( DX, DY, DZ ) ;
141       aTrsf.SetTranslation(myVec) ;
142       BRepBuilderAPI_Transform aTransformation(anOriginal, aTrsf, Standard_False) ;
143       B.Add(aCompound , aTransformation.Shape() );
144     }
145     aShape = aCompound;   
146   }
147   else if(aType == TRANSLATE_2D) {
148     Standard_Integer nbtimes1 = TI.GetNbIter1(), nbtimes2 = TI.GetNbIter2();
149     Standard_Real DX, DY, DZ,  step1 = TI.GetStep1(),  step2 = TI.GetStep2();
150     gp_Vec myVec ;
151     Handle(GEOM_Function) aVector = TI.GetVector();
152     if(aVector.IsNull()) return 0;
153     TopoDS_Shape aV = aVector->GetValue();
154     if(aV.IsNull() || aV.ShapeType() != TopAbs_EDGE) return 0;
155     TopoDS_Edge anEdge = TopoDS::Edge(aV);
156
157     gp_Vec Vec1(BRep_Tool::Pnt(TopExp::FirstVertex(anEdge)), BRep_Tool::Pnt(TopExp::LastVertex(anEdge)));
158     Vec1.Normalize();
159
160     Handle(GEOM_Function) aVector2 = TI.GetVector2();
161     if(aVector2.IsNull()) return 0;
162     aV = aVector2->GetValue();
163     if(aV.IsNull() || aV.ShapeType() != TopAbs_EDGE) return 0;
164     anEdge = TopoDS::Edge(aV);
165
166     gp_Vec Vec2(BRep_Tool::Pnt(TopExp::FirstVertex(anEdge)), BRep_Tool::Pnt(TopExp::LastVertex(anEdge)));
167     Vec2.Normalize();
168
169     TopoDS_Compound aCompound;
170     BRep_Builder B;
171     B.MakeCompound( aCompound );
172     
173     for (int i = 0; i < nbtimes1; i++ ) {
174       for (int j = 0; j < nbtimes2; j++ ) {
175         DX = i * step1 * Vec1.X() + j * step2 * Vec2.X() ;
176         DY = i * step1 * Vec1.Y() + j * step2 * Vec2.Y() ;
177         DZ = i * step1 * Vec1.Z() + j * step2 * Vec2.Z() ;
178         myVec.SetCoord( DX, DY, DZ ) ;
179         aTrsf.SetTranslation(myVec) ;
180         BRepBuilderAPI_Transform myBRepTransformation(anOriginal, aTrsf, Standard_False) ;
181         B.Add(aCompound , myBRepTransformation.Shape() );
182       }
183     }
184    aShape = aCompound;  
185   }
186   else return 0;
187
188  
189
190   if (aShape.IsNull()) return 0;
191
192   aFunction->SetValue(aShape);
193
194   log.SetTouched(Label()); 
195
196   return 1;
197 }
198
199
200 //=======================================================================
201 //function :  GEOMImpl_TranslateDriver_Type_
202 //purpose  :
203 //======================================================================= 
204 Standard_EXPORT Handle_Standard_Type& GEOMImpl_TranslateDriver_Type_()
205 {
206
207   static Handle_Standard_Type aType1 = STANDARD_TYPE(TFunction_Driver);
208   if ( aType1.IsNull()) aType1 = STANDARD_TYPE(TFunction_Driver);
209   static Handle_Standard_Type aType2 = STANDARD_TYPE(MMgt_TShared);
210   if ( aType2.IsNull()) aType2 = STANDARD_TYPE(MMgt_TShared); 
211   static Handle_Standard_Type aType3 = STANDARD_TYPE(Standard_Transient);
212   if ( aType3.IsNull()) aType3 = STANDARD_TYPE(Standard_Transient);
213  
214
215   static Handle_Standard_Transient _Ancestors[]= {aType1,aType2,aType3,NULL};
216   static Handle_Standard_Type _aType = new Standard_Type("GEOMImpl_TranslateDriver",
217                                                          sizeof(GEOMImpl_TranslateDriver),
218                                                          1,
219                                                          (Standard_Address)_Ancestors,
220                                                          (Standard_Address)NULL);
221
222   return _aType;
223 }
224
225 //=======================================================================
226 //function : DownCast
227 //purpose  :
228 //======================================================================= 
229
230 const Handle(GEOMImpl_TranslateDriver) Handle(GEOMImpl_TranslateDriver)::DownCast(const Handle(Standard_Transient)& AnObject)
231 {
232   Handle(GEOMImpl_TranslateDriver) _anOtherObject;
233
234   if (!AnObject.IsNull()) {
235      if (AnObject->IsKind(STANDARD_TYPE(GEOMImpl_TranslateDriver))) {
236        _anOtherObject = Handle(GEOMImpl_TranslateDriver)((Handle(GEOMImpl_TranslateDriver)&)AnObject);
237      }
238   }
239
240   return _anOtherObject ;
241 }
242
243