src/GEOMImpl/GEOMImpl_3DSketcherDriver.cxx

   1 // Copyright (C) 2007-2016  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, or (at your option) any later version.
   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
  20 #include <Standard_Stream.hxx>
  21
  22 #include <GEOMImpl_3DSketcherDriver.hxx>
  23 #include <GEOMImpl_I3DSketcher.hxx>
  24 #include <GEOMImpl_Types.hxx>
  25 #include <GEOM_Function.hxx>
  26
  27 #include <GEOMImpl_IMeasureOperations.hxx>
  28
  29 #include <Basics_Utils.hxx>
  30
  31 // OCCT Includes
  32 #include <BRepBuilderAPI_MakePolygon.hxx>
  33 #include <BRepBuilderAPI_MakeVertex.hxx>
  34 #include <BRepBuilderAPI_MakeEdge.hxx>
  35 #include <BRepBuilderAPI_MakeWire.hxx>
  36 #include <TopoDS.hxx>
  37 #include <TopoDS_Shape.hxx>
  38 #include <TopoDS_Wire.hxx>
  39 #include <gp_Pnt.hxx>
  40
  41 #include <Standard_ConstructionError.hxx>
  42
  43 #include <utilities.h>
  44
  45 //=======================================================================
  46 //function : GetID
  47 //purpose  :
  48 //=======================================================================
  49 const Standard_GUID& GEOMImpl_3DSketcherDriver::GetID()
  50 {
  51   static Standard_GUID a3DSketcherDriver("FF2BBB54-5D24-4df3-210B-3A678263EA26");
  52   return a3DSketcherDriver;
  53 }
  54
  55
  56 //=======================================================================
  57 //function : GEOMImpl_3DSketcherDriver
  58 //purpose  :
  59 //=======================================================================
  60 GEOMImpl_3DSketcherDriver::GEOMImpl_3DSketcherDriver()
  61 {
  62 }
  63
  64 //=======================================================================
  65 //function : Execute
  66 //purpose  :
  67 //=======================================================================
  68 Standard_Integer GEOMImpl_3DSketcherDriver::Execute(LOGBOOK& log) const
  69 {
  70   if (Label().IsNull()) return 0;
  71   Handle(GEOM_Function) aFunction = GEOM_Function::GetFunction(Label());
  72
  73   GEOMImpl_I3DSketcher aCI (aFunction);
  74
  75   TopoDS_Shape aShape;
  76
  77   if (aFunction->GetType() == SKETCHER3D_COORDS) {
  78     Handle(TColStd_HArray1OfReal) aCoordsArray = aCI.GetCoordinates();
  79     int anArrayLength = aCoordsArray->Length();
  80
  81     std::list<gp_Pnt> points;
  82
  83     for (int i = 0; i <= (anArrayLength-3); i += 3) {
  84       gp_Pnt aPnt = gp_Pnt(aCoordsArray->Value(i+1), aCoordsArray->Value(i+2), aCoordsArray->Value(i+3));
  85       if (points.empty() || aPnt.Distance(points.back()) > gp::Resolution())
  86         points.push_back(aPnt);
  87     }
  88
  89     if ( points.size() == 1) { // Only Start Point
  90       BRepBuilderAPI_MakeVertex mkVertex (points.back());
  91       aShape = mkVertex.Shape();
  92     }
  93     else if ( points.size() > 1) { // Make Wire
  94       BRepBuilderAPI_MakePolygon aMakePoly;
  95       std::list<gp_Pnt>::iterator it;
  96       for (it = points.begin(); it != points.end(); ++it) {
  97         aMakePoly.Add(*it);
  98       }
  99
 100       if (points.size() > 2 &&
 101           points.back().X() == points.front().X() &&
 102           points.back().Y() == points.front().Y() &&
 103           points.back().Z() == points.front().Z())
 104         aMakePoly.Close();
 105
 106       if (aMakePoly.IsDone())
 107         aShape = aMakePoly.Wire();
 108     }
 109   }
 110   else if (aFunction->GetType() == SKETCHER3D_COMMAND) {
 111     Kernel_Utils::Localizer loc;
 112
 113     TCollection_AsciiString aCommand = aCI.GetCommand();
 114     // "3DSketcher:CMD[:CMD[:CMD...]]"
 115
 116     // Split the command string to separate CMDs
 117     int icmd = 2;
 118     TColStd_SequenceOfAsciiString aSequence;
 119     if (aCommand.Length()) {
 120       TCollection_AsciiString aToken = aCommand.Token(":", icmd);
 121       while (aToken.Length() > 0) {
 122         aSequence.Append(aToken);
 123         aToken = aCommand.Token(":", ++icmd);
 124       }
 125     }
 126
 127     int nbEdges = 0;
 128     bool isFirstPointSet = false;
 129     gp_XYZ p = gp::Origin().XYZ();
 130     BRepBuilderAPI_MakeVertex MV0 (p);
 131     TopoDS_Vertex V = TopoDS::Vertex(MV0.Shape());
 132
 133     gp_XYZ p0 = p;
 134     TopoDS_Vertex V0 = V;
 135
 136     bool doClose = false;
 137     BRepBuilderAPI_MakeWire MW;
 138
 139     int nbCMDs = aSequence.Length();
 140     for (icmd = 1; icmd <= nbCMDs; icmd++) {
 141       TCollection_AsciiString aCMD = aSequence.Value(icmd);
 142
 143       // Split the CMD into string values
 144       TColStd_SequenceOfAsciiString aStrVals;
 145       int ival = 1;
 146       TCollection_AsciiString aToken = aCMD.Token(" ", ival);
 147       while (aToken.Length() > 0) {
 148         aStrVals.Append(aToken);
 149         aToken = aCMD.Token(" ", ++ival);
 150       }
 151
 152       // "TT x y z" : Create segment by point at X & Y or set the first point
 153       // "T dx dy dz" : Create segment by point with DX & DY
 154       //
 155       // "OXY angleX angle2 length" : Create segment by two angles and length
 156       // "OYZ angleY angle2 length" : Create segment by two angles and length
 157       // "OXZ angleX angle2 length" : Create segment by two angles and length
 158       //
 159       // "WW" : Close Wire (to finish)
 160
 161       switch (aStrVals.Value(1).Value(1))
 162       {
 163       case 'T':
 164         {
 165           if (aStrVals.Length() != 4)
 166             Standard_ConstructionError::Raise("3D Sketcher error: Bad format of command.");
 167
 168           gp_XYZ vp;
 169           vp.SetX(aStrVals.Value(2).RealValue());
 170           vp.SetY(aStrVals.Value(3).RealValue());
 171           vp.SetZ(aStrVals.Value(4).RealValue());
 172
 173           if (aStrVals.Value(1) == "TT") { // absolute coordinates
 174             if (!isFirstPointSet) {
 175               p = vp;
 176               BRepBuilderAPI_MakeVertex MV (p);
 177               V = TopoDS::Vertex(MV.Shape());
 178
 179               p0 = p;
 180               V0 = V;
 181
 182               isFirstPointSet = true;
 183             }
 184             else {
 185               if ((vp - p).SquareModulus() > Precision::Confusion()) {
 186                 BRepBuilderAPI_MakeVertex MV (vp);
 187                 TopoDS_Vertex VV = TopoDS::Vertex(MV.Shape());
 188                 BRepBuilderAPI_MakeEdge ME (V, VV);
 189                 MW.Add(ME);
 190                 nbEdges++;
 191
 192                 p = vp;
 193                 V = VV;
 194               }
 195             }
 196           }
 197           else if (aStrVals.Value(1) == "T") { // relative coordinates
 198             if (vp.SquareModulus() > Precision::Confusion()) {
 199               vp = p + vp;
 200
 201               BRepBuilderAPI_MakeVertex MV (vp);
 202               TopoDS_Vertex VV = TopoDS::Vertex(MV.Shape());
 203               BRepBuilderAPI_MakeEdge ME (V, VV);
 204               MW.Add(ME);
 205               nbEdges++;
 206
 207               p = vp;
 208               V = VV;
 209             }
 210           }
 211           else
 212             Standard_ConstructionError::Raise("3D Sketcher error: Bad format of command.");
 213         }
 214         break;
 215       case 'O':
 216         {
 217           if (aStrVals.Length() != 4)
 218             Standard_ConstructionError::Raise("3D Sketcher error: Bad format of command.");
 219
 220           char type = aStrVals.Value(1).Value(4);
 221           char mode = aStrVals.Value(1).Value(5);
 222
 223           double anAngle  = aStrVals.Value(2).RealValue() * M_PI/180.0;
 224           double anAngle2 = aStrVals.Value(3).RealValue() * M_PI/180.0;
 225           double aLength  = aStrVals.Value(4).RealValue();
 226
 227           double aHeight  = aLength * sin(anAngle2);
 228           if (type == 'C')
 229           {
 230             aHeight = aStrVals.Value(3).RealValue();
 231             anAngle2 = 0.0;
 232           }
 233
 234           double aProjectedLength = aLength * cos(anAngle2);
 235
 236           gp_XYZ vp;
 237           vp.SetX(aStrVals.Value(2).RealValue());
 238           vp.SetY(aStrVals.Value(3).RealValue());
 239           vp.SetZ(aStrVals.Value(4).RealValue());
 240
 241           gp_XYZ pref = gp::Origin().XYZ();
 242           if(mode == 'R')
 243             pref = p;
 244
 245           TCollection_AsciiString aTruncatedCommand = aStrVals.Value(1);
 246           aTruncatedCommand.Trunc(3);
 247
 248           if (aTruncatedCommand == "OXY") {
 249             vp.SetX(pref.X() + aProjectedLength * cos(anAngle));
 250             vp.SetY(pref.Y() + aProjectedLength * sin(anAngle));
 251             vp.SetZ(pref.Z() + aHeight);
 252           }
 253           else if (aTruncatedCommand == "OYZ") {
 254             vp.SetX(pref.X() + aHeight);
 255             vp.SetY(pref.Y() + aProjectedLength * cos(anAngle));
 256             vp.SetZ(pref.Z() + aProjectedLength * sin(anAngle));
 257           }
 258           else if (aTruncatedCommand == "OXZ") {
 259             vp.SetX(pref.X() + aProjectedLength * cos(anAngle));
 260             vp.SetY(pref.Y() + aHeight);
 261             vp.SetZ(pref.Z() + aProjectedLength * sin(anAngle));
 262           }
 263           else
 264             Standard_ConstructionError::Raise("3D Sketcher error: Bad format of command.");
 265
 266           if ((vp - p).SquareModulus() > Precision::Confusion()) {
 267             BRepBuilderAPI_MakeVertex MV (vp);
 268             TopoDS_Vertex VV = TopoDS::Vertex(MV.Shape());
 269             BRepBuilderAPI_MakeEdge ME (V, VV);
 270             MW.Add(ME);
 271             nbEdges++;
 272
 273             p = vp;
 274             V = VV;
 275           }
 276         }
 277         break;
 278       case 'W':
 279         {
 280           if (aStrVals.Length() != 1)
 281             Standard_ConstructionError::Raise("3D Sketcher error: Bad format of command.");
 282
 283           if (aStrVals.Value(1) == "WW")
 284             doClose = true;
 285           else
 286             Standard_ConstructionError::Raise("3D Sketcher error: Bad format of command.");
 287         }
 288         break;
 289       default:
 290         {
 291           Standard_ConstructionError::Raise("3D Sketcher error: Bad format of command.");
 292         }
 293       }
 294     }
 295
 296     if (doClose &&
 297         nbEdges > 1 && // as 3D sketcher has only straight edges
 298         (p - p0).SquareModulus() > Precision::Confusion()) {
 299       BRepBuilderAPI_MakeEdge ME (V, V0);
 300       MW.Add(ME);
 301       nbEdges++;
 302     }
 303
 304     if (nbEdges > 0) {
 305       if (!MW.IsDone())
 306         Standard_ConstructionError::Raise("3D Sketcher error: Wire construction failed.");
 307
 308       aShape = MW;
 309     }
 310     else {
 311       if (isFirstPointSet) {
 312         aShape = V0;
 313       }
 314     }
 315   }
 316   else {
 317   }
 318
 319   if (aShape.IsNull()) return 0;
 320
 321   aFunction->SetValue(aShape);
 322 #if OCC_VERSION_MAJOR < 7
 323   log.SetTouched(Label());
 324 #else
 325   log->SetTouched(Label());
 326 #endif
 327   return 1;
 328 }
 329
 330 //================================================================================
 331 /*!
 332  * \brief Returns a name of creation operation and names and values of creation parameters
 333  */
 334 //================================================================================
 335
 336 bool GEOMImpl_3DSketcherDriver::
 337 GetCreationInformation(std::string&             theOperationName,
 338                        std::vector<GEOM_Param>& theParams)
 339 {
 340   if (Label().IsNull()) return 0;
 341   Handle(GEOM_Function) function = GEOM_Function::GetFunction(Label());
 342
 343   GEOMImpl_I3DSketcher aCI( function );
 344   Standard_Integer aType = function->GetType();
 345
 346   theOperationName = "3DSKETCH";
 347
 348   if (aType == SKETCHER3D_COORDS)
 349   {
 350     theParams.resize( 1 );
 351     theParams[0].Set( "(X,Y,Z)" );
 352
 353     Handle(TColStd_HArray1OfReal) aCoordsArray = aCI.GetCoordinates();
 354     int anArrayLength = aCoordsArray->Length();
 355     for (int i = 0; i <= (anArrayLength-3); i += 3)
 356       theParams[0] << "( "
 357                    << aCoordsArray->Value(i+1) << ", "
 358                    << aCoordsArray->Value(i+2) << ", "
 359                    << aCoordsArray->Value(i+3) << " ) ";
 360   }
 361   else if (aType == SKETCHER3D_COMMAND)
 362   {
 363     AddParam( theParams, "Command", aCI.GetCommand() );
 364   }
 365   else {
 366     return false;
 367   }
 368
 369   return true;
 370 }
 371
 372 OCCT_IMPLEMENT_STANDARD_RTTIEXT(GEOMImpl_3DSketcherDriver,GEOM_BaseDriver);