src/GEOMImpl/GEOMImpl_3DSketcherDriver.cxx

   1 // Copyright (C) 2007-2012  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18
  19 #include <Standard_Stream.hxx>
  20
  21 #include <GEOMImpl_3DSketcherDriver.hxx>
  22 #include <GEOMImpl_I3DSketcher.hxx>
  23 #include <GEOMImpl_Types.hxx>
  24 #include <GEOM_Function.hxx>
  25
  26 #include <GEOMImpl_IMeasureOperations.hxx>
  27
  28 #include <Basics_Utils.hxx>
  29
  30 // OCCT Includes
  31 #include <BRepBuilderAPI_MakePolygon.hxx>
  32 #include <BRepBuilderAPI_MakeVertex.hxx>
  33 #include <BRepBuilderAPI_MakeEdge.hxx>
  34 #include <BRepBuilderAPI_MakeWire.hxx>
  35 #include <TopoDS.hxx>
  36 #include <TopoDS_Shape.hxx>
  37 #include <TopoDS_Wire.hxx>
  38 #include <gp_Pnt.hxx>
  39
  40 #include <Standard_ConstructionError.hxx>
  41
  42 //=======================================================================
  43 //function : GetID
  44 //purpose  :
  45 //=======================================================================
  46 const Standard_GUID& GEOMImpl_3DSketcherDriver::GetID()
  47 {
  48   static Standard_GUID a3DSketcherDriver("FF2BBB54-5D24-4df3-210B-3A678263EA26");
  49   return a3DSketcherDriver;
  50 }
  51
  52
  53 //=======================================================================
  54 //function : GEOMImpl_3DSketcherDriver
  55 //purpose  :
  56 //=======================================================================
  57 GEOMImpl_3DSketcherDriver::GEOMImpl_3DSketcherDriver()
  58 {
  59 }
  60
  61 //=======================================================================
  62 //function : Execute
  63 //purpose  :
  64 //=======================================================================
  65 Standard_Integer GEOMImpl_3DSketcherDriver::Execute(TFunction_Logbook& log) const
  66 {
  67   if (Label().IsNull()) return 0;
  68   Handle(GEOM_Function) aFunction = GEOM_Function::GetFunction(Label());
  69
  70   GEOMImpl_I3DSketcher aCI (aFunction);
  71
  72   TopoDS_Shape aShape;
  73
  74   if (aFunction->GetType() == SKETCHER3D_COORDS) {
  75     Handle(TColStd_HArray1OfReal) aCoordsArray = aCI.GetCoordinates();
  76     int anArrayLength = aCoordsArray->Length();
  77
  78     std::list<gp_Pnt> points;
  79
  80     for (int i = 0; i <= (anArrayLength-3); i += 3) {
  81       gp_Pnt aPnt = gp_Pnt(aCoordsArray->Value(i+1), aCoordsArray->Value(i+2), aCoordsArray->Value(i+3));
  82       if (points.empty() || aPnt.Distance(points.back()) > gp::Resolution())
  83         points.push_back(aPnt);
  84     }
  85
  86     if ( points.size() == 1) { // Only Start Point
  87       BRepBuilderAPI_MakeVertex mkVertex (points.back());
  88       aShape = mkVertex.Shape();
  89     }
  90     else if ( points.size() > 1) { // Make Wire
  91       BRepBuilderAPI_MakePolygon aMakePoly;
  92       std::list<gp_Pnt>::iterator it;
  93       for (it = points.begin(); it != points.end(); ++it) {
  94         aMakePoly.Add(*it);
  95       }
  96
  97       if (points.size() > 2 &&
  98           points.back().X() == points.front().X() &&
  99           points.back().Y() == points.front().Y() &&
 100           points.back().Z() == points.front().Z())
 101         aMakePoly.Close();
 102
 103       if (aMakePoly.IsDone())
 104         aShape = aMakePoly.Wire();
 105     }
 106   }
 107   else if (aFunction->GetType() == SKETCHER3D_COMMAND) {
 108     Kernel_Utils::Localizer loc;
 109
 110     TCollection_AsciiString aCommand = aCI.GetCommand();
 111     // "3DSketcher:CMD[:CMD[:CMD...]]"
 112
 113     // Split the command string to separate CMDs
 114     int icmd = 2;
 115     TColStd_SequenceOfAsciiString aSequence;
 116     if (aCommand.Length()) {
 117       TCollection_AsciiString aToken = aCommand.Token(":", icmd);
 118       while (aToken.Length() > 0) {
 119         aSequence.Append(aToken);
 120         aToken = aCommand.Token(":", ++icmd);
 121       }
 122     }
 123
 124     int nbEdges = 0;
 125     bool isFirstPointSet = false;
 126     gp_XYZ p = gp::Origin().XYZ();
 127     BRepBuilderAPI_MakeVertex MV0 (p);
 128     TopoDS_Vertex V = TopoDS::Vertex(MV0.Shape());
 129
 130     gp_XYZ p0 = p;
 131     TopoDS_Vertex V0 = V;
 132
 133     bool doClose = false;
 134     BRepBuilderAPI_MakeWire MW;
 135
 136     int nbCMDs = aSequence.Length();
 137     for (icmd = 1; icmd <= nbCMDs; icmd++) {
 138       TCollection_AsciiString aCMD = aSequence.Value(icmd);
 139
 140       // Split the CMD into string values
 141       TColStd_SequenceOfAsciiString aStrVals;
 142       int ival = 1;
 143       TCollection_AsciiString aToken = aCMD.Token(" ", ival);
 144       while (aToken.Length() > 0) {
 145         aStrVals.Append(aToken);
 146         aToken = aCMD.Token(" ", ++ival);
 147       }
 148
 149       // "TT x y z" : Create segment by point at X & Y or set the first point
 150       // "T dx dy dz" : Create segment by point with DX & DY
 151       //
 152       // "OXY angleX angle2 length" : Create segment by two angles and length
 153       // "OYZ angleY angle2 length" : Create segment by two angles and length
 154       // "OXZ angleX angle2 length" : Create segment by two angles and length
 155       //
 156       // "WW" : Close Wire (to finish)
 157
 158       switch (aStrVals.Value(1).Value(1))
 159       {
 160       case 'T':
 161         {
 162           if (aStrVals.Length() != 4)
 163             Standard_ConstructionError::Raise("3D Sketcher error: Bad format of command.");
 164
 165           gp_XYZ vp;
 166           vp.SetX(aStrVals.Value(2).RealValue());
 167           vp.SetY(aStrVals.Value(3).RealValue());
 168           vp.SetZ(aStrVals.Value(4).RealValue());
 169
 170           if (aStrVals.Value(1) == "TT") { // absolute coordinates
 171             if (!isFirstPointSet) {
 172               p = vp;
 173               BRepBuilderAPI_MakeVertex MV (p);
 174               V = TopoDS::Vertex(MV.Shape());
 175
 176               p0 = p;
 177               V0 = V;
 178
 179               isFirstPointSet = true;
 180             }
 181             else {
 182               if ((vp - p).SquareModulus() > Precision::Confusion()) {
 183                 BRepBuilderAPI_MakeVertex MV (vp);
 184                 TopoDS_Vertex VV = TopoDS::Vertex(MV.Shape());
 185                 BRepBuilderAPI_MakeEdge ME (V, VV);
 186                 MW.Add(ME);
 187                 nbEdges++;
 188
 189                 p = vp;
 190                 V = VV;
 191               }
 192             }
 193           }
 194           else if (aStrVals.Value(1) == "T") { // relative coordinates
 195             if (vp.SquareModulus() > Precision::Confusion()) {
 196               vp = p + vp;
 197
 198               BRepBuilderAPI_MakeVertex MV (vp);
 199               TopoDS_Vertex VV = TopoDS::Vertex(MV.Shape());
 200               BRepBuilderAPI_MakeEdge ME (V, VV);
 201               MW.Add(ME);
 202               nbEdges++;
 203
 204               p = vp;
 205               V = VV;
 206             }
 207           }
 208           else
 209             Standard_ConstructionError::Raise("3D Sketcher error: Bad format of command.");
 210         }
 211         break;
 212       case 'O':
 213         {
 214           if (aStrVals.Length() != 4)
 215             Standard_ConstructionError::Raise("3D Sketcher error: Bad format of command.");
 216
 217           double anAngle  = aStrVals.Value(2).RealValue() * M_PI/180.0;
 218           double anAngle2 = aStrVals.Value(3).RealValue() * M_PI/180.0;
 219           double aLength  = aStrVals.Value(4).RealValue();
 220
 221           double aProjectedLength = aLength * cos(anAngle2);
 222
 223           gp_XYZ vp;
 224           vp.SetX(aStrVals.Value(2).RealValue());
 225           vp.SetY(aStrVals.Value(3).RealValue());
 226           vp.SetZ(aStrVals.Value(4).RealValue());
 227
 228           if (aStrVals.Value(1) == "OXY") {
 229             vp.SetX(p.X() + aProjectedLength * cos(anAngle));
 230             vp.SetY(p.Y() + aProjectedLength * sin(anAngle));
 231             vp.SetZ(p.Z() + aLength * sin(anAngle2));
 232           }
 233           else if (aStrVals.Value(1) == "OYZ") {
 234             vp.SetX(p.X() + aLength * sin(anAngle2));
 235             vp.SetY(p.Y() + aProjectedLength * cos(anAngle));
 236             vp.SetZ(p.Z() + aProjectedLength * sin(anAngle));
 237           }
 238           else if (aStrVals.Value(1) == "OXZ") {
 239             vp.SetX(p.X() + aProjectedLength * cos(anAngle));
 240             vp.SetY(p.Y() + aLength * sin(anAngle2));
 241             vp.SetZ(p.Z() + aProjectedLength * sin(anAngle));
 242           }
 243           else
 244             Standard_ConstructionError::Raise("3D Sketcher error: Bad format of command.");
 245
 246           if ((vp - p).SquareModulus() > Precision::Confusion()) {
 247             BRepBuilderAPI_MakeVertex MV (vp);
 248             TopoDS_Vertex VV = TopoDS::Vertex(MV.Shape());
 249             BRepBuilderAPI_MakeEdge ME (V, VV);
 250             MW.Add(ME);
 251             nbEdges++;
 252
 253             p = vp;
 254             V = VV;
 255           }
 256         }
 257         break;
 258       case 'W':
 259         {
 260           if (aStrVals.Length() != 1)
 261             Standard_ConstructionError::Raise("3D Sketcher error: Bad format of command.");
 262
 263           if (aStrVals.Value(1) == "WW")
 264             doClose = true;
 265           else
 266             Standard_ConstructionError::Raise("3D Sketcher error: Bad format of command.");
 267         }
 268         break;
 269       default:
 270         {
 271           Standard_ConstructionError::Raise("3D Sketcher error: Bad format of command.");
 272         }
 273       }
 274     }
 275
 276     if (doClose &&
 277         nbEdges > 1 && // as 3D sketcher has only straight edges
 278         (p - p0).SquareModulus() > Precision::Confusion()) {
 279       BRepBuilderAPI_MakeEdge ME (V, V0);
 280       MW.Add(ME);
 281       nbEdges++;
 282     }
 283
 284     if (nbEdges > 0) {
 285       if (!MW.IsDone())
 286         Standard_ConstructionError::Raise("3D Sketcher error: Wire construction failed.");
 287
 288       aShape = MW;
 289     }
 290     else {
 291       if (isFirstPointSet) {
 292         aShape = V0;
 293       }
 294     }
 295   }
 296   else {
 297   }
 298
 299   if (aShape.IsNull()) return 0;
 300
 301   aFunction->SetValue(aShape);
 302   log.SetTouched(Label());
 303   return 1;
 304 }
 305
 306
 307 //=======================================================================
 308 //function :  GEOMImpl_3DSketcherDriver_Type_
 309 //purpose  :
 310 //=======================================================================
 311 Standard_EXPORT Handle_Standard_Type& GEOMImpl_3DSketcherDriver_Type_()
 312 {
 313
 314   static Handle_Standard_Type aType1 = STANDARD_TYPE(TFunction_Driver);
 315   if ( aType1.IsNull()) aType1 = STANDARD_TYPE(TFunction_Driver);
 316   static Handle_Standard_Type aType2 = STANDARD_TYPE(MMgt_TShared);
 317   if ( aType2.IsNull()) aType2 = STANDARD_TYPE(MMgt_TShared);
 318   static Handle_Standard_Type aType3 = STANDARD_TYPE(Standard_Transient);
 319   if ( aType3.IsNull()) aType3 = STANDARD_TYPE(Standard_Transient);
 320
 321
 322   static Handle_Standard_Transient _Ancestors[]= {aType1,aType2,aType3,NULL};
 323   static Handle_Standard_Type _aType = new Standard_Type("GEOMImpl_3DSketcherDriver",
 324                                                          sizeof(GEOMImpl_3DSketcherDriver),
 325                                                          1,
 326                                                          (Standard_Address)_Ancestors,
 327                                                          (Standard_Address)NULL);
 328
 329   return _aType;
 330 }
 331
 332 //=======================================================================
 333 //function : DownCast
 334 //purpose  :
 335 //=======================================================================
 336 const Handle(GEOMImpl_3DSketcherDriver) Handle(GEOMImpl_3DSketcherDriver)::DownCast(const Handle(Standard_Transient)& AnObject)
 337 {
 338   Handle(GEOMImpl_3DSketcherDriver) _anOtherObject;
 339
 340   if (!AnObject.IsNull()) {
 341      if (AnObject->IsKind(STANDARD_TYPE(GEOMImpl_3DSketcherDriver))) {
 342        _anOtherObject = Handle(GEOMImpl_3DSketcherDriver)((Handle(GEOMImpl_3DSketcherDriver)&)AnObject);
 343      }
 344   }
 345
 346   return _anOtherObject ;
 347 }