occt/src/BOPTest/BOPTest_UtilityCommands.cxx

// Created on: 2016-04-01
// Created by: Nikolai BUKHALOV
// Copyright (c) 2000-2016 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.

#include <BOPTest.hxx>

#include <BOPTools_AlgoTools.hxx>
#include <BOPTools_AlgoTools2D.hxx>
#include <DBRep.hxx>
#include <IntTools_Context.hxx>
#include <TopoDS.hxx>
#include <TopoDS_Edge.hxx>
#include <TopoDS_Face.hxx>
#include <TopoDS_Shape.hxx>
#include <TopExp_Explorer.hxx>
#include <Draw.hxx>
#include <BOPAlgo_Tools.hxx>
#include <BRepLib.hxx>

static Standard_Integer attachpcurve (Draw_Interpretor&, Standard_Integer, const char**);
static Standard_Integer edgestowire  (Draw_Interpretor&, Standard_Integer, const char**);
static Standard_Integer edgestofaces  (Draw_Interpretor&, Standard_Integer, const char**);
static Standard_Integer BuildPcurvesOnPlane(Draw_Interpretor&, Standard_Integer, const char**);

//=======================================================================
//function : BOPCommands
//purpose  : 
//=======================================================================
  void BOPTest::UtilityCommands(Draw_Interpretor& theCommands)
{
  static Standard_Boolean done = Standard_False;
  if (done) return;
  done = Standard_True;
  // Chapter's name
  const char* group = "BOPTest commands";
  // Commands
  
  theCommands.Add("attachpcurve", "attachpcurve eold enew face", __FILE__, attachpcurve, group);
  theCommands.Add("edgestowire" , "edgestowire wire edges"     , __FILE__, edgestowire , group);
  theCommands.Add("edgestofaces" , "edgestofaces faces edges [-a AngTol -s Shared(0/1)]", __FILE__, edgestofaces , group);
  theCommands.Add("buildpcurvesonplane", "buildpcurvesonplane shape", __FILE__, BuildPcurvesOnPlane, group);
  }

//=======================================================================
//function : BOPCommands
//purpose  : Attaches p-curve of the given edge to the given face.
//=======================================================================
static Standard_Integer attachpcurve(Draw_Interpretor& theDI,
                                 Standard_Integer  theNArg,
                                 const char ** theArgVal)
{
  if (theNArg != 4)
  {
    theDI << "Use: " << theArgVal[0] << " eold enew face\n";
    return 1;
  }

  TopoDS_Shape aShEOld(DBRep::Get(theArgVal[1]));
  TopoDS_Shape aShENew(DBRep::Get(theArgVal[2]));
  TopoDS_Shape aShFace(DBRep::Get(theArgVal[3]));

  if (aShEOld.IsNull()) {
    theDI << theArgVal[1] << " is null shape\n";
    return 1;
  } else if (aShEOld.ShapeType() != TopAbs_EDGE) {
    theDI << theArgVal[1] << " is not an edge\n";
    return 1;
  }

  if (aShENew.IsNull()) {
    theDI << theArgVal[2] << " is null shape\n";
    return 1;
  } else if (aShENew.ShapeType() != TopAbs_EDGE) {
    theDI << theArgVal[2] << " is not an edge\n";
    return 1;
  }

  if (aShFace.IsNull()) {
    theDI << theArgVal[3] << " is null shape\n";
    return 1;
  } else if (aShFace.ShapeType() != TopAbs_FACE) {
    theDI << theArgVal[3] << " is not a face\n";
    return 1;
  }

  TopoDS_Edge aEOld = TopoDS::Edge(aShEOld);
  TopoDS_Edge aENew = TopoDS::Edge(aShENew);
  TopoDS_Face aFace = TopoDS::Face(aShFace);

  // Try to copy PCurve from old edge to the new one.
  Handle(IntTools_Context) aCtx = new IntTools_Context;
  const Standard_Integer   iRet =
    BOPTools_AlgoTools2D::AttachExistingPCurve(aEOld, aENew, aFace, aCtx);

  if (iRet) {
    theDI << "Error! Code: " << iRet << "\n";
  } else {
    theDI << "PCurve is attached successfully\n";
  }

  return 0;
}

//=======================================================================
//function : edgestowire
//purpose  : Orients the edges to make wire
//=======================================================================
static Standard_Integer edgestowire(Draw_Interpretor& theDI,
                                    Standard_Integer  theNArg,
                                    const char ** theArgVal)
{
  if (theNArg != 3) {
    theDI << "Use: edgestowire wire edges\n";
    return 1;
  }
  //
  TopoDS_Shape anEdges = DBRep::Get(theArgVal[2]);
  if (anEdges.IsNull()) {
    theDI << "no edges\n";
    return 1;
  }
  //
  BOPTools_AlgoTools::OrientEdgesOnWire(anEdges);
  DBRep::Set(theArgVal[1], anEdges);
  return 0;
}

//=======================================================================
//function : edgestofaces
//purpose  : Creates planar faces from linear edges
//=======================================================================
static Standard_Integer edgestofaces(Draw_Interpretor& theDI,
                                     Standard_Integer  theNArg,
                                     const char ** theArgVal)
{
  if (theNArg < 3) {
    theDI << "Use: edgestofaces faces edges [-a AngTol -s Shared(0/1)]\n";
    theDI << " AngTol - angular tolerance for comparing the planes;\n";
    theDI << " Shared - boolean flag which defines whether the input\n";
    theDI << "          edges are already shared or have to be intersected.\n";
    return 1;
  }
  //
  TopoDS_Shape anEdges = DBRep::Get(theArgVal[2]);
  if (anEdges.IsNull()) {
    theDI << "no edges\n";
    return 1;
  }
  //
  Standard_Real anAngTol = 1.e-8;
  Standard_Boolean bShared = Standard_False;
  //
  for (Standard_Integer i = 3; i < theNArg; ++i) {
    if (!strcmp(theArgVal[i], "-a") && (i+1 < theNArg)) {
      anAngTol = Draw::Atof(theArgVal[i+1]);
    }
    if (!strcmp(theArgVal[i], "-s") && (i+1 < theNArg)) {
      bShared = (Draw::Atoi(theArgVal[i+1]) == 1);
    }
  }
  //
  TopoDS_Shape aWires;
  Standard_Integer iErr = BOPAlgo_Tools::EdgesToWires(anEdges, aWires, bShared, anAngTol);
  if (iErr) {
    theDI << "Unable to build wires from given edges\n";
    return 0;
  }
  //
  TopoDS_Shape aFaces;
  Standard_Boolean bDone = BOPAlgo_Tools::WiresToFaces(aWires, aFaces, anAngTol);
  if (!bDone) {
    theDI << "Unable to build faces from wires\n";
    return 0;
  }
  //
  DBRep::Set(theArgVal[1], aFaces);
  return 0;
}

//=======================================================================
//function : BuildPcurvesOnPlane
//purpose  : Build and store pcurves of edges on planes
//=======================================================================
static Standard_Integer BuildPcurvesOnPlane(Draw_Interpretor& theDI,
                                 Standard_Integer  theNArg,
                                 const char ** theArgVal)
{
  if (theNArg != 2)
  {
    theDI << "Use: " << theArgVal[0] << " shape\n";
    return 1;
  }

  TopoDS_Shape aShape(DBRep::Get(theArgVal[1]));
  if (aShape.IsNull()) {
    theDI << theArgVal[1] << " is null shape\n";
    return 1;
  }

  TopExp_Explorer exp(aShape, TopAbs_FACE);
  for (; exp.More(); exp.Next())
  {
    const TopoDS_Face& aF = TopoDS::Face(exp.Current());
    BRepAdaptor_Surface aS(aF, Standard_False);
    if (aS.GetType() == GeomAbs_Plane)
    {
      TopTools_ListOfShape aLE;
      TopExp_Explorer exp1(aF, TopAbs_EDGE);
      for (; exp1.More(); exp1.Next())
        aLE.Append(exp1.Current());
      BRepLib::BuildPCurveForEdgesOnPlane(aLE, aF);
    }
  }
  return 0;
}