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/
  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