[occt.git] / src / MeshTest / MeshTest_PluginCommands.cxx

// Created on: 2008-04-11
// Created by: Peter KURNEV
// Copyright (c) 2008-2014 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 <Bnd_Box.hxx>
#include <BRep_Tool.hxx>
#include <BRepGProp.hxx>
#include <BRepMesh_DiscretFactory.hxx>
#include <BRepMesh_DiscretRoot.hxx>
#include <BRepMesh_Edge.hxx>
#include <BRepMesh_FactoryError.hxx>
#include <BRepMesh_IncrementalMesh.hxx>
#include <DBRep.hxx>
#include <Draw.hxx>
#include <Draw_Interpretor.hxx>
#include <DrawTrSurf.hxx>
#include <gp_Vec.hxx>
#include <GProp_GProps.hxx>
#include <MeshTest.hxx>
#include <MeshTest_CheckTopology.hxx>
#include <NCollection_Map.hxx>
#include <Poly_Polygon2D.hxx>
#include <Poly_Polygon3D.hxx>
#include <Poly_PolygonOnTriangulation.hxx>
#include <Poly_Triangulation.hxx>
#include <Standard.hxx>
#include <TColgp_Array1OfPnt2d.hxx>
#include <TCollection_AsciiString.hxx>
#include <TColStd_Array1OfInteger.hxx>
#include <TColStd_MapIteratorOfMapOfAsciiString.hxx>
#include <TColStd_MapOfAsciiString.hxx>
#include <TopExp.hxx>
#include <TopExp_Explorer.hxx>
#include <TopoDS.hxx>
#include <TopoDS_Face.hxx>
#include <TopTools_IndexedMapOfShape.hxx>

static Standard_Integer mpnames           (Draw_Interpretor& , Standard_Integer , const char** );
static Standard_Integer mpsetdefaultname  (Draw_Interpretor& , Standard_Integer , const char** );
static Standard_Integer mpgetdefaultname  (Draw_Interpretor& , Standard_Integer , const char** );
static Standard_Integer mpsetfunctionname (Draw_Interpretor& , Standard_Integer , const char** );
static Standard_Integer mpgetfunctionname (Draw_Interpretor& , Standard_Integer , const char** );
static Standard_Integer mperror           (Draw_Interpretor& , Standard_Integer , const char** );
static Standard_Integer mpincmesh         (Draw_Interpretor& , Standard_Integer , const char** );
static Standard_Integer mpparallel        (Draw_Interpretor& , Standard_Integer , const char** );
static Standard_Integer triarea           (Draw_Interpretor& , Standard_Integer , const char** );
static Standard_Integer tricheck          (Draw_Interpretor& , Standard_Integer , const char** );

//=======================================================================
//function : PluginCommands
//purpose  : 
//=======================================================================
void MeshTest::PluginCommands(Draw_Interpretor& theCommands)
{
  static Standard_Boolean done = Standard_False;
  if (done) {
    return;
  }
  done = Standard_True;
  //
  const char* g = "Mesh Commands";
  // Commands
  theCommands.Add("mpnames"          , "use mpnames"          , __FILE__, mpnames    , g);
  theCommands.Add("mpsetdefaultname" , "use mpsetdefaultname" , __FILE__, mpsetdefaultname     , g);
  theCommands.Add("mpgetdefaultname" , "use mpgetdefaultname" , __FILE__, mpgetdefaultname     , g);
  theCommands.Add("mpsetfunctionname", "use mpsetfunctionname", __FILE__, mpsetfunctionname     , g);
  theCommands.Add("mpgetfunctionname", "use mpgetfunctionname", __FILE__, mpgetfunctionname     , g);
  theCommands.Add("mperror"          , "use mperror"          , __FILE__, mperror     , g);
  theCommands.Add("mpincmesh"        , "use mpincmesh"        , __FILE__, mpincmesh      , g);
  theCommands.Add("mpparallel"       , "mpparallel [toTurnOn] : show / set multi-threading flag for incremental mesh",
    __FILE__, mpparallel, g);
  theCommands.Add("triarea","shape [eps]  (computes triangles and surface area)",__FILE__, triarea, g);
  theCommands.Add("tricheck", "shape   (checks triangulation of shape)", __FILE__, tricheck, g);
}

//=======================================================================
//function : mpnames
//purpose  : 
//=======================================================================
static Standard_Integer mpnames (Draw_Interpretor& , Standard_Integer n, const char** )
{
  Standard_Integer aNb;
  TColStd_MapIteratorOfMapOfAsciiString aIt;
  //
  if (n!=1) {
    printf(" use mpnames\n");
    return 0;
  }
  //
  const TColStd_MapOfAsciiString& aMN=BRepMesh_DiscretFactory::Get().Names();
  aNb=aMN.Extent();
  if (!aNb) {
    printf(" *no names found\n");
    return 0;
  }
  //
  printf(" *available names:\n");
  aIt.Initialize(aMN);
  for (; aIt.More(); aIt.Next()) {
    const TCollection_AsciiString& aName=aIt.Key();
    printf("  %s\n", aName.ToCString());
  }
  //
  return 0;
}
//=======================================================================
//function : mpsetdefaultname
//purpose  : 
//=======================================================================
static Standard_Integer mpsetdefaultname (Draw_Interpretor& , Standard_Integer n, const char**a )
{
  TCollection_AsciiString aName;
  //
  if (n!=2) {
    printf(" use mpsetdefaultname name\n");
    return 0;
  }
  //
  aName=a[1];
  //
  if (BRepMesh_DiscretFactory::Get().SetDefaultName (aName))
    printf(" *ready\n");
  else
    printf(" *fault\n");
  //
  return 0;
}
//=======================================================================
//function : mpgetdefaultname
//purpose  : 
//=======================================================================
static Standard_Integer mpgetdefaultname (Draw_Interpretor& , Standard_Integer n, const char** )
{
  if (n!=1) {
    printf(" use mpgetdefaultname\n");
    return 0;
  }
  //
  const TCollection_AsciiString& aName=BRepMesh_DiscretFactory::Get().DefaultName();
  printf(" *default name: %s\n", aName.ToCString());
  //
  return 0;
}
//=======================================================================
//function : mpsetfunctionname
//purpose  : 
//=======================================================================
static Standard_Integer mpsetfunctionname (Draw_Interpretor& , Standard_Integer n, const char**a )
{
  TCollection_AsciiString aName;
  //
  if (n!=2) {
    printf(" use mpsetfunctionname name\n");
    return 0;
  }
  //
  aName=a[1];
  //
  if (BRepMesh_DiscretFactory::Get().SetFunctionName (aName))
    printf(" *ready\n");
  else
    printf(" *fault\n");
  //
  return 0;
}
//=======================================================================
//function : mpgetdefaultname
//purpose  : 
//=======================================================================
static Standard_Integer mpgetfunctionname (Draw_Interpretor& , Standard_Integer n, const char** )
{
  if (n!=1) {
    printf(" use mpgetfunctionname\n");
    return 0;
  }
  //
  const TCollection_AsciiString& aName=BRepMesh_DiscretFactory::Get().FunctionName();
  printf(" *function name: %s\n", aName.ToCString());
  //
  return 0;
}
//=======================================================================
//function : mperror
//purpose  : 
//=======================================================================
static Standard_Integer mperror (Draw_Interpretor& , Standard_Integer n, const char** )
{
  BRepMesh_FactoryError aErr;
  //
  if (n!=1) {
    printf(" use mperror\n");
    return 0;
  }
  //
  aErr=BRepMesh_DiscretFactory::Get().ErrorStatus();
  printf(" *ErrorStatus: %d\n", (int)aErr);
  //
  return 0;
}

//=======================================================================
//function :mpincmesh
//purpose  : 
//=======================================================================
static Standard_Integer mpincmesh (Draw_Interpretor& , Standard_Integer n, const char** a)
{
  Standard_Real aDeflection, aAngle;
  TopoDS_Shape aS;
  //
  if (n<3) {
    printf(" use mpincmesh s deflection [angle]\n");
    return 0;
  }
  //
  aS=DBRep::Get(a[1]);
  if (aS.IsNull()) {
    printf(" null shapes is not allowed here\n");
    return 0;
  }
  //
  aDeflection=Draw::Atof(a[2]);
  aAngle=0.5;
  if (n>3) {
    aAngle=Draw::Atof(a[3]);
  }
  //
  Handle(BRepMesh_DiscretRoot) aMeshAlgo = BRepMesh_DiscretFactory::Get().Discret (aS,
                                                                                   aDeflection,
                                                                                   aAngle);
  //
  BRepMesh_FactoryError aErr = BRepMesh_DiscretFactory::Get().ErrorStatus();
  if (aErr != BRepMesh_FE_NOERROR)
  {
    printf(" *Factory::Get().ErrorStatus()=%d\n", (int)aErr);
  }
  //
  if (aMeshAlgo.IsNull())
  {
    printf(" *Can not create the algo\n");
    return 0;
  }
  //
  aMeshAlgo->Perform();
  if (!aMeshAlgo->IsDone())
  {
    printf(" *Not done\n");
  }
  //
  return 0;
}

//#######################################################################
static Standard_Integer triarea (Draw_Interpretor& di, int n, const char ** a)
{

  if (n < 2) return 1;

  TopoDS_Shape shape = DBRep::Get(a[1]);
  if (shape.IsNull()) return 1;
  Standard_Real anEps = -1.;
  if (n > 2)
    anEps = Draw::Atof(a[2]);

  TopTools_IndexedMapOfShape aMapF;
  TopExp::MapShapes (shape, TopAbs_FACE, aMapF);

  // detect if a shape has triangulation
  Standard_Boolean hasPoly = Standard_False;
  int i;
  for (i=1; i <= aMapF.Extent(); i++) {
    const TopoDS_Face& aFace = TopoDS::Face(aMapF(i));
    TopLoc_Location aLoc;
    Handle(Poly_Triangulation) aPoly = BRep_Tool::Triangulation(aFace,aLoc);
    if (!aPoly.IsNull()) {
      hasPoly = Standard_True;
      break;
    }
  }

  // compute area by triangles
  double aTriArea=0;
  if (hasPoly) {
    for (i=1; i <= aMapF.Extent(); i++) {
      const TopoDS_Face& aFace = TopoDS::Face(aMapF(i));
      TopLoc_Location aLoc;
      Handle(Poly_Triangulation) aPoly = BRep_Tool::Triangulation(aFace,aLoc);
      if (aPoly.IsNull()) {
        cout << "face "<<i<<" has no triangulation"<<endl;
        continue;
      }
      const Poly_Array1OfTriangle& triangles = aPoly->Triangles();
      const TColgp_Array1OfPnt& nodes = aPoly->Nodes();
      for (int j=triangles.Lower(); j <= triangles.Upper(); j++) {
        const Poly_Triangle& tri = triangles(j);
        int n1, n2, n3;
        tri.Get (n1, n2, n3);
        const gp_Pnt& p1 = nodes(n1);
        const gp_Pnt& p2 = nodes(n2);
        const gp_Pnt& p3 = nodes(n3);
        gp_Vec v1(p1, p2);
        gp_Vec v2(p1, p3);
        double ar = v1.CrossMagnitude(v2);
        aTriArea += ar;
      }
    }
    aTriArea /= 2;
  }

  // compute area by geometry
  GProp_GProps props;
  if (anEps <= 0.)
    BRepGProp::SurfaceProperties(shape, props);
  else
    BRepGProp::SurfaceProperties(shape, props, anEps);
  double aGeomArea = props.Mass();

  di << aTriArea << " " << aGeomArea << "\n";
  return 0;
}

//#######################################################################
Standard_Boolean IsEqual(const BRepMesh_Edge& theFirst, const BRepMesh_Edge& theSecond) 
{
  return theFirst.IsEqual(theSecond);
}

static Standard_Integer tricheck (Draw_Interpretor& di, int n, const char ** a)
{
  if (n < 2) return 1;

  TopoDS_Shape shape = DBRep::Get(a[1]);
  if (shape.IsNull()) return 1;

  TopTools_IndexedMapOfShape aMapF;
  TopExp::MapShapes (shape, TopAbs_FACE, aMapF);
  Standard_CString name = ".";

  // execute check
  MeshTest_CheckTopology aCheck(shape);
  aCheck.Perform(di);

  // dump info on free links inside the triangulation
  Standard_Integer nbFree = 0;
  Standard_Integer nbFac = aCheck.NbFacesWithFL(), i, k;
  if (nbFac > 0) {
    for (k=1; k <= nbFac; k++) {
      Standard_Integer nbEdge = aCheck.NbFreeLinks(k);
      Standard_Integer iF = aCheck.GetFaceNumWithFL(k);
      nbFree += nbEdge;
      di << "free links of face " << iF << "\n";

      const TopoDS_Shape& aShape = aMapF.FindKey(iF);
      const TopoDS_Face& aFace = TopoDS::Face(aShape);
      TopLoc_Location aLoc;
      Handle(Poly_Triangulation) aT = BRep_Tool::Triangulation(aFace, aLoc);
      const TColgp_Array1OfPnt& aPoints = aT->Nodes();
      const TColgp_Array1OfPnt2d& aPoints2d = aT->UVNodes();
      const gp_Trsf& trsf = aLoc.Transformation();

      TColgp_Array1OfPnt pnts(1,2);
      TColgp_Array1OfPnt2d pnts2d(1,2);
      for (i=1; i <= nbEdge; i++) {
        Standard_Integer n1, n2;
        aCheck.GetFreeLink(k, i, n1, n2);
        di << "{" << n1 << " " << n2 << "} ";
        pnts(1) = aPoints(n1).Transformed(trsf);
        pnts(2) = aPoints(n2).Transformed(trsf);
        Handle(Poly_Polygon3D) poly = new Poly_Polygon3D (pnts);
        DrawTrSurf::Set (name, poly);
        DrawTrSurf::Set (name, pnts(1));
        DrawTrSurf::Set (name, pnts(2));
        pnts2d(1) = aPoints2d(n1);
        pnts2d(2) = aPoints2d(n2);
        Handle(Poly_Polygon2D) poly2d = new Poly_Polygon2D (pnts2d);
        DrawTrSurf::Set (name, poly2d);
        DrawTrSurf::Set (name, pnts2d(1));
        DrawTrSurf::Set (name, pnts2d(2));
      }
      di << "\n";
    }
  }

  // dump info on cross face errors
  Standard_Integer nbErr = aCheck.NbCrossFaceErrors();
  if (nbErr > 0) {
    di << "cross face errors: {face1, node1, face2, node2, distance}" << "\n";
    for (i=1; i <= nbErr; i++) {
      Standard_Integer iF1, n1, iF2, n2;
      Standard_Real aVal;
      aCheck.GetCrossFaceError(i, iF1, n1, iF2, n2, aVal);
      di << "{" << iF1 << " " << n1 << " " << iF2 << " " << n2 << " " << aVal << "} ";
    }
    di << "\n";
  }

  // dump info on edges
  Standard_Integer nbAsync = aCheck.NbAsyncEdges();
  if (nbAsync > 0) {
    di << "async edges:" << "\n";
    for (i=1; i <= nbAsync; i++) {
      Standard_Integer ie = aCheck.GetAsyncEdgeNum(i);
      di << ie << " ";
    }
    di << "\n";
  }

  // dump info on free nodes
  Standard_Integer nbFreeNodes = aCheck.NbFreeNodes();
  if (nbFreeNodes > 0) {
    di << "free nodes (in pairs: face / node): " << "\n";
    for (i=1; i <= nbFreeNodes; i++) {
      Standard_Integer iface, inode;
      aCheck.GetFreeNodeNum(i, iface, inode);

      const TopoDS_Face& aFace = TopoDS::Face(aMapF.FindKey(iface));
      TopLoc_Location aLoc;
      Handle(Poly_Triangulation) aT = BRep_Tool::Triangulation(aFace, aLoc);
      const TColgp_Array1OfPnt& aPoints = aT->Nodes();
      const TColgp_Array1OfPnt2d& aPoints2d = aT->UVNodes();
      const gp_Trsf& trsf = aLoc.Transformation();
      DrawTrSurf::Set (name, aPoints(inode).Transformed(trsf));
      DrawTrSurf::Set (name, aPoints2d(inode));

      di << "{" << iface << " " << inode << "} ";
    }
    di << "\n";
  }

  // output errors summary to DRAW
  if ( nbFree > 0 || nbErr > 0 || nbAsync > 0 || nbFreeNodes > 0)
    di << "Free_links " << nbFree
       << " Cross_face_errors " << nbErr
       << " Async_edges " << nbAsync 
       << " Free_nodes " << nbFreeNodes << "\n";


  Standard_Integer aFaceId = 1;
  TopExp_Explorer aFaceExp(shape, TopAbs_FACE);
  for ( ; aFaceExp.More(); aFaceExp.Next(), ++aFaceId)
  {
    const TopoDS_Shape& aShape = aFaceExp.Current();
    const TopoDS_Face& aFace = TopoDS::Face(aShape);

    TopLoc_Location aLoc;
    Handle(Poly_Triangulation) aT = BRep_Tool::Triangulation(aFace, aLoc);

    // Iterate boundary edges
    NCollection_Map<BRepMesh_Edge> aBoundaryEdgeMap;
    TopExp_Explorer anExp(aShape, TopAbs_EDGE);
    for ( ; anExp.More(); anExp.Next() )
    {
      TopLoc_Location anEdgeLoc;
      const TopoDS_Edge& anEdge = TopoDS::Edge(anExp.Current());
      Handle(Poly_PolygonOnTriangulation) aPoly = BRep_Tool::PolygonOnTriangulation(anEdge, aT, aLoc);
      if (aPoly.IsNull())
      {
        continue;
      }

      const TColStd_Array1OfInteger& anIndices = aPoly->Nodes();
      Standard_Integer aLower  = anIndices.Lower(); 
      Standard_Integer anUpper = anIndices.Upper();

      Standard_Integer aPrevNode = -1;
      for (Standard_Integer i = aLower; i <= anUpper; ++i)
      {
        Standard_Integer aNodeIdx = anIndices.Value(i);
        if (i != aLower)
        {
          BRepMesh_Edge aLink(aPrevNode, aNodeIdx, BRepMesh_Frontier);
          aBoundaryEdgeMap.Add(aLink);
        }
        aPrevNode = aNodeIdx;
      }
    }

    if (aBoundaryEdgeMap.Size() == 0)
    {
      break;
    }

    const Poly_Array1OfTriangle& aTris = aT->Triangles();
    NCollection_Map<BRepMesh_Edge> aFreeEdgeMap;
    Standard_Integer aTriNum = aTris.Length();
    for ( Standard_Integer aTriIndx = 1; aTriIndx <= aTriNum; aTriIndx++ )
    {
      const Poly_Triangle& aTri = aTris(aTriIndx);
      Standard_Integer aTriNodes[3] = { aTri.Value(1), aTri.Value(2), aTri.Value(3)};

      for (Standard_Integer i = 1; i <= 3; ++i)
      {
        Standard_Integer aLastId  = aTriNodes[i % 3];
        Standard_Integer aFirstId = aTriNodes[i - 1];

        BRepMesh_Edge aLink(aFirstId, aLastId, BRepMesh_Free);
        if (!aBoundaryEdgeMap.Contains(aLink))
        {
          if (!aFreeEdgeMap.Add(aLink))
          {
            aFreeEdgeMap.Remove(aLink);
          }
        }
      }
    }

    if (aFreeEdgeMap.Size() != 0)
    {
      di << "Not connected mesh inside face " << aFaceId << "\n";

      const TColgp_Array1OfPnt& aPoints = aT->Nodes();
      const TColgp_Array1OfPnt2d& aPoints2d = aT->UVNodes();
      const gp_Trsf& trsf = aLoc.Transformation();

      TColgp_Array1OfPnt pnts(1,2);
      TColgp_Array1OfPnt2d pnts2d(1,2);
      NCollection_Map<BRepMesh_Edge>::Iterator aMapIt(aFreeEdgeMap);
      for (; aMapIt.More(); aMapIt.Next())
      {
        const BRepMesh_Edge& aLink = aMapIt.Key();
        di << "{" << aLink.FirstNode() << " " << aLink.LastNode() << "} ";
        pnts(1) = aPoints(aLink.FirstNode()).Transformed(trsf);
        pnts(2) = aPoints(aLink.LastNode()).Transformed(trsf);
        Handle(Poly_Polygon3D) poly = new Poly_Polygon3D (pnts);
        DrawTrSurf::Set (name, poly);
        DrawTrSurf::Set (name, pnts(1));
        DrawTrSurf::Set (name, pnts(2));
        pnts2d(1) = aPoints2d(aLink.FirstNode());
        pnts2d(2) = aPoints2d(aLink.LastNode());
        Handle(Poly_Polygon2D) poly2d = new Poly_Polygon2D (pnts2d);
        DrawTrSurf::Set (name, poly2d);
        DrawTrSurf::Set (name, pnts2d(1));
        DrawTrSurf::Set (name, pnts2d(2));
      }
      di << "\n";
    }
  }
  return 0;
}

//=======================================================================
//function : mpparallel
//purpose  :
//=======================================================================
static int mpparallel (Draw_Interpretor& /*di*/, Standard_Integer argc, const char** argv)
{
  if (argc == 2)
  {
    Standard_Boolean isParallelOn = Draw::Atoi (argv[1]) == 1;
    BRepMesh_IncrementalMesh::SetParallelDefault (isParallelOn);
  }
  std::cout << "Incremental Mesh, multi-threading "
            << (BRepMesh_IncrementalMesh::IsParallelDefault() ? "ON\n" : "OFF\n");
  return 0;
}
Commit	Line	Data
b311480e	1	// Created on: 2008-04-11
b311480e	2	// Created by: Peter KURNEV
973c2be1	3	// Copyright (c) 2008-2014 OPEN CASCADE SAS
b311480e	4	//
973c2be1	5	// This file is part of Open CASCADE Technology software library.
b311480e	6	//
d5f74e42	7	// This library is free software; you can redistribute it and/or modify it under
d5f74e42	8	// the terms of the GNU Lesser General Public License version 2.1 as published
973c2be1	9	// by the Free Software Foundation, with special exception defined in the file
	10	// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
	11	// distribution for complete text of the license and disclaimer of any warranty.
b311480e	12	//
973c2be1	13	// Alternatively, this file may be used under the terms of Open CASCADE
973c2be1	14	// commercial license or contractual agreement.
7fd59977	15
42cf5bc1	16
	17	#include <Bnd_Box.hxx>
	18	#include <BRep_Tool.hxx>
	19	#include <BRepGProp.hxx>
7fd59977	20	#include <BRepMesh_DiscretFactory.hxx>
7fd59977	21	#include <BRepMesh_DiscretRoot.hxx>
42cf5bc1	22	#include <BRepMesh_Edge.hxx>
42cf5bc1	23	#include <BRepMesh_FactoryError.hxx>
0b97567d	24	#include <BRepMesh_IncrementalMesh.hxx>
7fd59977	25	#include <DBRep.hxx>
42cf5bc1	26	#include <Draw.hxx>
	27	#include <Draw_Interpretor.hxx>
	28	#include <DrawTrSurf.hxx>
7fd59977	29	#include <gp_Vec.hxx>
7fd59977	30	#include <GProp_GProps.hxx>
42cf5bc1	31	#include <MeshTest.hxx>
7fd59977	32	#include <MeshTest_CheckTopology.hxx>
42cf5bc1	33	#include <NCollection_Map.hxx>
7fd59977	34	#include <Poly_Polygon2D.hxx>
42cf5bc1	35	#include <Poly_Polygon3D.hxx>
	36	#include <Poly_PolygonOnTriangulation.hxx>
	37	#include <Poly_Triangulation.hxx>
ca0db031	38	#include <Standard.hxx>
42cf5bc1	39	#include <TColgp_Array1OfPnt2d.hxx>
42cf5bc1	40	#include <TCollection_AsciiString.hxx>
2e1a4dae	41	#include <TColStd_Array1OfInteger.hxx>
42cf5bc1	42	#include <TColStd_MapIteratorOfMapOfAsciiString.hxx>
	43	#include <TColStd_MapOfAsciiString.hxx>
	44	#include <TopExp.hxx>
	45	#include <TopExp_Explorer.hxx>
	46	#include <TopoDS.hxx>
2e1a4dae	47	#include <TopoDS_Face.hxx>
42cf5bc1	48	#include <TopTools_IndexedMapOfShape.hxx>
7fd59977	49
	50	static Standard_Integer mpnames (Draw_Interpretor& , Standard_Integer , const char** );
	51	static Standard_Integer mpsetdefaultname (Draw_Interpretor& , Standard_Integer , const char** );
	52	static Standard_Integer mpgetdefaultname (Draw_Interpretor& , Standard_Integer , const char** );
	53	static Standard_Integer mpsetfunctionname (Draw_Interpretor& , Standard_Integer , const char** );
	54	static Standard_Integer mpgetfunctionname (Draw_Interpretor& , Standard_Integer , const char** );
	55	static Standard_Integer mperror (Draw_Interpretor& , Standard_Integer , const char** );
	56	static Standard_Integer mpincmesh (Draw_Interpretor& , Standard_Integer , const char** );
0b97567d	57	static Standard_Integer mpparallel (Draw_Interpretor& , Standard_Integer , const char** );
7fd59977	58	static Standard_Integer triarea (Draw_Interpretor& , Standard_Integer , const char** );
e1bcbb52	59	static Standard_Integer tricheck (Draw_Interpretor& , Standard_Integer , const char** );
7fd59977	60
	61	//=======================================================================
	62	//function : PluginCommands
	63	//purpose :
	64	//=======================================================================
	65	void MeshTest::PluginCommands(Draw_Interpretor& theCommands)
	66	{
	67	static Standard_Boolean done = Standard_False;
	68	if (done) {
	69	return;
	70	}
	71	done = Standard_True;
	72	//
	73	const char* g = "Mesh Commands";
	74	// Commands
	75	theCommands.Add("mpnames" , "use mpnames" , __FILE__, mpnames , g);
	76	theCommands.Add("mpsetdefaultname" , "use mpsetdefaultname" , __FILE__, mpsetdefaultname , g);
	77	theCommands.Add("mpgetdefaultname" , "use mpgetdefaultname" , __FILE__, mpgetdefaultname , g);
	78	theCommands.Add("mpsetfunctionname", "use mpsetfunctionname", __FILE__, mpsetfunctionname , g);
	79	theCommands.Add("mpgetfunctionname", "use mpgetfunctionname", __FILE__, mpgetfunctionname , g);
	80	theCommands.Add("mperror" , "use mperror" , __FILE__, mperror , g);
	81	theCommands.Add("mpincmesh" , "use mpincmesh" , __FILE__, mpincmesh , g);
0b97567d K	82	theCommands.Add("mpparallel" , "mpparallel [toTurnOn] : show / set multi-threading flag for incremental mesh",
0b97567d K	83	__FILE__, mpparallel, g);
7fd59977	84	theCommands.Add("triarea","shape [eps] (computes triangles and surface area)",__FILE__, triarea, g);
e1bcbb52	85	theCommands.Add("tricheck", "shape (checks triangulation of shape)", __FILE__, tricheck, g);
7fd59977	86	}
	87
	88	//=======================================================================
	89	//function : mpnames
	90	//purpose :
	91	//=======================================================================
	92	static Standard_Integer mpnames (Draw_Interpretor& , Standard_Integer n, const char** )
	93	{
	94	Standard_Integer aNb;
	95	TColStd_MapIteratorOfMapOfAsciiString aIt;
	96	//
	97	if (n!=1) {
	98	printf(" use mpnames\n");
	99	return 0;
	100	}
	101	//
	102	const TColStd_MapOfAsciiString& aMN=BRepMesh_DiscretFactory::Get().Names();
	103	aNb=aMN.Extent();
	104	if (!aNb) {
	105	printf(" *no names found\n");
	106	return 0;
	107	}
	108	//
	109	printf(" *available names:\n");
	110	aIt.Initialize(aMN);
	111	for (; aIt.More(); aIt.Next()) {
	112	const TCollection_AsciiString& aName=aIt.Key();
	113	printf(" %s\n", aName.ToCString());
	114	}
	115	//
	116	return 0;
	117	}
	118	//=======================================================================
	119	//function : mpsetdefaultname
	120	//purpose :
	121	//=======================================================================
	122	static Standard_Integer mpsetdefaultname (Draw_Interpretor& , Standard_Integer n, const char**a )
	123	{
	124	TCollection_AsciiString aName;
	125	//
	126	if (n!=2) {
	127	printf(" use mpsetdefaultname name\n");
	128	return 0;
	129	}
	130	//
	131	aName=a[1];
	132	//
0b97567d K	133	if (BRepMesh_DiscretFactory::Get().SetDefaultName (aName))
	134	printf(" *ready\n");
	135	else
	136	printf(" *fault\n");
7fd59977	137	//
	138	return 0;
	139	}
	140	//=======================================================================
	141	//function : mpgetdefaultname
	142	//purpose :
	143	//=======================================================================
	144	static Standard_Integer mpgetdefaultname (Draw_Interpretor& , Standard_Integer n, const char** )
	145	{
	146	if (n!=1) {
	147	printf(" use mpgetdefaultname\n");
	148	return 0;
	149	}
	150	//
	151	const TCollection_AsciiString& aName=BRepMesh_DiscretFactory::Get().DefaultName();
	152	printf(" *default name: %s\n", aName.ToCString());
	153	//
	154	return 0;
	155	}
	156	//=======================================================================
	157	//function : mpsetfunctionname
	158	//purpose :
	159	//=======================================================================
	160	static Standard_Integer mpsetfunctionname (Draw_Interpretor& , Standard_Integer n, const char**a )
	161	{
	162	TCollection_AsciiString aName;
	163	//
	164	if (n!=2) {
	165	printf(" use mpsetfunctionname name\n");
	166	return 0;
	167	}
	168	//
	169	aName=a[1];
	170	//
0b97567d K	171	if (BRepMesh_DiscretFactory::Get().SetFunctionName (aName))
	172	printf(" *ready\n");
	173	else
	174	printf(" *fault\n");
7fd59977	175	//
	176	return 0;
	177	}
	178	//=======================================================================
	179	//function : mpgetdefaultname
	180	//purpose :
	181	//=======================================================================
	182	static Standard_Integer mpgetfunctionname (Draw_Interpretor& , Standard_Integer n, const char** )
	183	{
	184	if (n!=1) {
	185	printf(" use mpgetfunctionname\n");
	186	return 0;
	187	}
	188	//
	189	const TCollection_AsciiString& aName=BRepMesh_DiscretFactory::Get().FunctionName();
	190	printf(" *function name: %s\n", aName.ToCString());
	191	//
	192	return 0;
	193	}
	194	//=======================================================================
	195	//function : mperror
	196	//purpose :
	197	//=======================================================================
	198	static Standard_Integer mperror (Draw_Interpretor& , Standard_Integer n, const char** )
	199	{
	200	BRepMesh_FactoryError aErr;
	201	//
	202	if (n!=1) {
	203	printf(" use mperror\n");
	204	return 0;
	205	}
	206	//
	207	aErr=BRepMesh_DiscretFactory::Get().ErrorStatus();
	208	printf(" *ErrorStatus: %d\n", (int)aErr);
	209	//
	210	return 0;
	211	}
	212
	213	//=======================================================================
	214	//function :mpincmesh
	215	//purpose :
	216	//=======================================================================
	217	static Standard_Integer mpincmesh (Draw_Interpretor& , Standard_Integer n, const char** a)
	218	{
7fd59977	219	Standard_Real aDeflection, aAngle;
7fd59977	220	TopoDS_Shape aS;
7fd59977	221	//
	222	if (n<3) {
	223	printf(" use mpincmesh s deflection [angle]\n");
	224	return 0;
	225	}
	226	//
	227	aS=DBRep::Get(a[1]);
	228	if (aS.IsNull()) {
	229	printf(" null shapes is not allowed here\n");
	230	return 0;
	231	}
	232	//
91322f44	233	aDeflection=Draw::Atof(a[2]);
7fd59977	234	aAngle=0.5;
7fd59977	235	if (n>3) {
91322f44	236	aAngle=Draw::Atof(a[3]);
7fd59977	237	}
7fd59977	238	//
0b97567d K	239	Handle(BRepMesh_DiscretRoot) aMeshAlgo = BRepMesh_DiscretFactory::Get().Discret (aS,
	240	aDeflection,
	241	aAngle);
7fd59977	242	//
0b97567d K	243	BRepMesh_FactoryError aErr = BRepMesh_DiscretFactory::Get().ErrorStatus();
	244	if (aErr != BRepMesh_FE_NOERROR)
	245	{
7fd59977	246	printf(" *Factory::Get().ErrorStatus()=%d\n", (int)aErr);
	247	}
	248	//
0b97567d K	249	if (aMeshAlgo.IsNull())
0b97567d K	250	{
7fd59977	251	printf(" *Can not create the algo\n");
	252	return 0;
	253	}
	254	//
0b97567d K	255	aMeshAlgo->Perform();
	256	if (!aMeshAlgo->IsDone())
	257	{
7fd59977	258	printf(" *Not done\n");
	259	}
	260	//
	261	return 0;
	262	}
	263
	264	//#######################################################################
	265	static Standard_Integer triarea (Draw_Interpretor& di, int n, const char ** a)
	266	{
	267
	268	if (n < 2) return 1;
	269
	270	TopoDS_Shape shape = DBRep::Get(a[1]);
	271	if (shape.IsNull()) return 1;
	272	Standard_Real anEps = -1.;
	273	if (n > 2)
91322f44	274	anEps = Draw::Atof(a[2]);
7fd59977	275
	276	TopTools_IndexedMapOfShape aMapF;
	277	TopExp::MapShapes (shape, TopAbs_FACE, aMapF);
	278
	279	// detect if a shape has triangulation
	280	Standard_Boolean hasPoly = Standard_False;
	281	int i;
	282	for (i=1; i <= aMapF.Extent(); i++) {
	283	const TopoDS_Face& aFace = TopoDS::Face(aMapF(i));
	284	TopLoc_Location aLoc;
	285	Handle(Poly_Triangulation) aPoly = BRep_Tool::Triangulation(aFace,aLoc);
	286	if (!aPoly.IsNull()) {
	287	hasPoly = Standard_True;
	288	break;
	289	}
	290	}
	291
	292	// compute area by triangles
	293	double aTriArea=0;
	294	if (hasPoly) {
	295	for (i=1; i <= aMapF.Extent(); i++) {
	296	const TopoDS_Face& aFace = TopoDS::Face(aMapF(i));
	297	TopLoc_Location aLoc;
	298	Handle(Poly_Triangulation) aPoly = BRep_Tool::Triangulation(aFace,aLoc);
	299	if (aPoly.IsNull()) {
e1bcbb52 O	300	cout << "face "<<i<<" has no triangulation"<<endl;
e1bcbb52 O	301	continue;
7fd59977	302	}
	303	const Poly_Array1OfTriangle& triangles = aPoly->Triangles();
	304	const TColgp_Array1OfPnt& nodes = aPoly->Nodes();
	305	for (int j=triangles.Lower(); j <= triangles.Upper(); j++) {
e1bcbb52 O	306	const Poly_Triangle& tri = triangles(j);
	307	int n1, n2, n3;
	308	tri.Get (n1, n2, n3);
	309	const gp_Pnt& p1 = nodes(n1);
	310	const gp_Pnt& p2 = nodes(n2);
	311	const gp_Pnt& p3 = nodes(n3);
	312	gp_Vec v1(p1, p2);
	313	gp_Vec v2(p1, p3);
	314	double ar = v1.CrossMagnitude(v2);
	315	aTriArea += ar;
7fd59977	316	}
	317	}
	318	aTriArea /= 2;
	319	}
	320
	321	// compute area by geometry
	322	GProp_GProps props;
	323	if (anEps <= 0.)
	324	BRepGProp::SurfaceProperties(shape, props);
	325	else
	326	BRepGProp::SurfaceProperties(shape, props, anEps);
	327	double aGeomArea = props.Mass();
	328
d2c43192	329	di << aTriArea << " " << aGeomArea << "\n";
7fd59977	330	return 0;
	331	}
	332
	333	//#######################################################################
2e1a4dae	334	Standard_Boolean IsEqual(const BRepMesh_Edge& theFirst, const BRepMesh_Edge& theSecond)
	335	{
	336	return theFirst.IsEqual(theSecond);
	337	}
	338
e1bcbb52	339	static Standard_Integer tricheck (Draw_Interpretor& di, int n, const char ** a)
7fd59977	340	{
	341	if (n < 2) return 1;
	342
	343	TopoDS_Shape shape = DBRep::Get(a[1]);
	344	if (shape.IsNull()) return 1;
	345
	346	TopTools_IndexedMapOfShape aMapF;
	347	TopExp::MapShapes (shape, TopAbs_FACE, aMapF);
	348	Standard_CString name = ".";
	349
446e11f3	350	// execute check
7fd59977	351	MeshTest_CheckTopology aCheck(shape);
d2c43192	352	aCheck.Perform(di);
446e11f3 A	353
446e11f3 A	354	// dump info on free links inside the triangulation
7fd59977	355	Standard_Integer nbFree = 0;
	356	Standard_Integer nbFac = aCheck.NbFacesWithFL(), i, k;
	357	if (nbFac > 0) {
	358	for (k=1; k <= nbFac; k++) {
	359	Standard_Integer nbEdge = aCheck.NbFreeLinks(k);
	360	Standard_Integer iF = aCheck.GetFaceNumWithFL(k);
	361	nbFree += nbEdge;
d2c43192	362	di << "free links of face " << iF << "\n";
2e1a4dae	363
	364	const TopoDS_Shape& aShape = aMapF.FindKey(iF);
	365	const TopoDS_Face& aFace = TopoDS::Face(aShape);
7fd59977	366	TopLoc_Location aLoc;
	367	Handle(Poly_Triangulation) aT = BRep_Tool::Triangulation(aFace, aLoc);
	368	const TColgp_Array1OfPnt& aPoints = aT->Nodes();
	369	const TColgp_Array1OfPnt2d& aPoints2d = aT->UVNodes();
	370	const gp_Trsf& trsf = aLoc.Transformation();
2e1a4dae	371
7fd59977	372	TColgp_Array1OfPnt pnts(1,2);
	373	TColgp_Array1OfPnt2d pnts2d(1,2);
	374	for (i=1; i <= nbEdge; i++) {
e1bcbb52 O	375	Standard_Integer n1, n2;
e1bcbb52 O	376	aCheck.GetFreeLink(k, i, n1, n2);
d2c43192	377	di << "{" << n1 << " " << n2 << "} ";
e1bcbb52 O	378	pnts(1) = aPoints(n1).Transformed(trsf);
	379	pnts(2) = aPoints(n2).Transformed(trsf);
	380	Handle(Poly_Polygon3D) poly = new Poly_Polygon3D (pnts);
	381	DrawTrSurf::Set (name, poly);
	382	DrawTrSurf::Set (name, pnts(1));
	383	DrawTrSurf::Set (name, pnts(2));
	384	pnts2d(1) = aPoints2d(n1);
	385	pnts2d(2) = aPoints2d(n2);
	386	Handle(Poly_Polygon2D) poly2d = new Poly_Polygon2D (pnts2d);
	387	DrawTrSurf::Set (name, poly2d);
	388	DrawTrSurf::Set (name, pnts2d(1));
	389	DrawTrSurf::Set (name, pnts2d(2));
7fd59977	390	}
d2c43192	391	di << "\n";
7fd59977	392	}
	393	}
	394
446e11f3	395	// dump info on cross face errors
7fd59977	396	Standard_Integer nbErr = aCheck.NbCrossFaceErrors();
7fd59977	397	if (nbErr > 0) {
d2c43192	398	di << "cross face errors: {face1, node1, face2, node2, distance}" << "\n";
7fd59977	399	for (i=1; i <= nbErr; i++) {
	400	Standard_Integer iF1, n1, iF2, n2;
	401	Standard_Real aVal;
	402	aCheck.GetCrossFaceError(i, iF1, n1, iF2, n2, aVal);
d2c43192	403	di << "{" << iF1 << " " << n1 << " " << iF2 << " " << n2 << " " << aVal << "} ";
7fd59977	404	}
d2c43192	405	di << "\n";
7fd59977	406	}
7fd59977	407
446e11f3	408	// dump info on edges
7fd59977	409	Standard_Integer nbAsync = aCheck.NbAsyncEdges();
7fd59977	410	if (nbAsync > 0) {
d2c43192	411	di << "async edges:" << "\n";
7fd59977	412	for (i=1; i <= nbAsync; i++) {
7fd59977	413	Standard_Integer ie = aCheck.GetAsyncEdgeNum(i);
d2c43192	414	di << ie << " ";
7fd59977	415	}
d2c43192	416	di << "\n";
7fd59977	417	}
7fd59977	418
446e11f3 A	419	// dump info on free nodes
	420	Standard_Integer nbFreeNodes = aCheck.NbFreeNodes();
	421	if (nbFreeNodes > 0) {
d2c43192	422	di << "free nodes (in pairs: face / node): " << "\n";
446e11f3 A	423	for (i=1; i <= nbFreeNodes; i++) {
	424	Standard_Integer iface, inode;
	425	aCheck.GetFreeNodeNum(i, iface, inode);
dc66be92	426
	427	const TopoDS_Face& aFace = TopoDS::Face(aMapF.FindKey(iface));
	428	TopLoc_Location aLoc;
	429	Handle(Poly_Triangulation) aT = BRep_Tool::Triangulation(aFace, aLoc);
	430	const TColgp_Array1OfPnt& aPoints = aT->Nodes();
	431	const TColgp_Array1OfPnt2d& aPoints2d = aT->UVNodes();
	432	const gp_Trsf& trsf = aLoc.Transformation();
	433	DrawTrSurf::Set (name, aPoints(inode).Transformed(trsf));
	434	DrawTrSurf::Set (name, aPoints2d(inode));
	435
d2c43192	436	di << "{" << iface << " " << inode << "} ";
446e11f3	437	}
d2c43192	438	di << "\n";
446e11f3 A	439	}
	440
	441	// output errors summary to DRAW
2e1a4dae	442	if ( nbFree > 0 \|\| nbErr > 0 \|\| nbAsync > 0 \|\| nbFreeNodes > 0)
446e11f3 A	443	di << "Free_links " << nbFree
	444	<< " Cross_face_errors " << nbErr
	445	<< " Async_edges " << nbAsync
d2c43192	446	<< " Free_nodes " << nbFreeNodes << "\n";
2e1a4dae	447
	448
	449	Standard_Integer aFaceId = 1;
	450	TopExp_Explorer aFaceExp(shape, TopAbs_FACE);
	451	for ( ; aFaceExp.More(); aFaceExp.Next(), ++aFaceId)
	452	{
	453	const TopoDS_Shape& aShape = aFaceExp.Current();
	454	const TopoDS_Face& aFace = TopoDS::Face(aShape);
	455
	456	TopLoc_Location aLoc;
	457	Handle(Poly_Triangulation) aT = BRep_Tool::Triangulation(aFace, aLoc);
	458
	459	// Iterate boundary edges
	460	NCollection_Map<BRepMesh_Edge> aBoundaryEdgeMap;
	461	TopExp_Explorer anExp(aShape, TopAbs_EDGE);
	462	for ( ; anExp.More(); anExp.Next() )
	463	{
	464	TopLoc_Location anEdgeLoc;
	465	const TopoDS_Edge& anEdge = TopoDS::Edge(anExp.Current());
	466	Handle(Poly_PolygonOnTriangulation) aPoly = BRep_Tool::PolygonOnTriangulation(anEdge, aT, aLoc);
	467	if (aPoly.IsNull())
	468	{
	469	continue;
	470	}
	471
	472	const TColStd_Array1OfInteger& anIndices = aPoly->Nodes();
	473	Standard_Integer aLower = anIndices.Lower();
	474	Standard_Integer anUpper = anIndices.Upper();
	475
	476	Standard_Integer aPrevNode = -1;
	477	for (Standard_Integer i = aLower; i <= anUpper; ++i)
	478	{
	479	Standard_Integer aNodeIdx = anIndices.Value(i);
	480	if (i != aLower)
	481	{
	482	BRepMesh_Edge aLink(aPrevNode, aNodeIdx, BRepMesh_Frontier);
	483	aBoundaryEdgeMap.Add(aLink);
	484	}
	485	aPrevNode = aNodeIdx;
	486	}
	487	}
	488
	489	if (aBoundaryEdgeMap.Size() == 0)
	490	{
	491	break;
	492	}
	493
	494	const Poly_Array1OfTriangle& aTris = aT->Triangles();
	495	NCollection_Map<BRepMesh_Edge> aFreeEdgeMap;
	496	Standard_Integer aTriNum = aTris.Length();
	497	for ( Standard_Integer aTriIndx = 1; aTriIndx <= aTriNum; aTriIndx++ )
	498	{
	499	const Poly_Triangle& aTri = aTris(aTriIndx);
	500	Standard_Integer aTriNodes[3] = { aTri.Value(1), aTri.Value(2), aTri.Value(3)};
	501
	502	for (Standard_Integer i = 1; i <= 3; ++i)
	503	{
	504	Standard_Integer aLastId = aTriNodes[i % 3];
	505	Standard_Integer aFirstId = aTriNodes[i - 1];
	506
	507	BRepMesh_Edge aLink(aFirstId, aLastId, BRepMesh_Free);
	508	if (!aBoundaryEdgeMap.Contains(aLink))
	509	{
	510	if (!aFreeEdgeMap.Add(aLink))
511	{
512	aFreeEdgeMap.Remove(aLink);
513	}
514	}
515	}
516	}
517
518	if (aFreeEdgeMap.Size() != 0)
519	{
520	di << "Not connected mesh inside face " << aFaceId << "\n";
dc66be92	521
	522	const TColgp_Array1OfPnt& aPoints = aT->Nodes();
	523	const TColgp_Array1OfPnt2d& aPoints2d = aT->UVNodes();
	524	const gp_Trsf& trsf = aLoc.Transformation();
	525
	526	TColgp_Array1OfPnt pnts(1,2);
	527	TColgp_Array1OfPnt2d pnts2d(1,2);
	528	NCollection_Map<BRepMesh_Edge>::Iterator aMapIt(aFreeEdgeMap);
	529	for (; aMapIt.More(); aMapIt.Next())
	530	{
	531	const BRepMesh_Edge& aLink = aMapIt.Key();
	532	di << "{" << aLink.FirstNode() << " " << aLink.LastNode() << "} ";
	533	pnts(1) = aPoints(aLink.FirstNode()).Transformed(trsf);
	534	pnts(2) = aPoints(aLink.LastNode()).Transformed(trsf);
	535	Handle(Poly_Polygon3D) poly = new Poly_Polygon3D (pnts);
	536	DrawTrSurf::Set (name, poly);
	537	DrawTrSurf::Set (name, pnts(1));
	538	DrawTrSurf::Set (name, pnts(2));
	539	pnts2d(1) = aPoints2d(aLink.FirstNode());
	540	pnts2d(2) = aPoints2d(aLink.LastNode());
	541	Handle(Poly_Polygon2D) poly2d = new Poly_Polygon2D (pnts2d);
	542	DrawTrSurf::Set (name, poly2d);
	543	DrawTrSurf::Set (name, pnts2d(1));
	544	DrawTrSurf::Set (name, pnts2d(2));
	545	}
	546	di << "\n";
2e1a4dae	547	}
2e1a4dae	548	}
7fd59977	549	return 0;
7fd59977	550	}
0b97567d K	551
	552	//=======================================================================
	553	//function : mpparallel
	554	//purpose :
	555	//=======================================================================
35e08fe8	556	static int mpparallel (Draw_Interpretor& /di/, Standard_Integer argc, const char** argv)
0b97567d K	557	{
	558	if (argc == 2)
	559	{
91322f44	560	Standard_Boolean isParallelOn = Draw::Atoi (argv[1]) == 1;
0b97567d K	561	BRepMesh_IncrementalMesh::SetParallelDefault (isParallelOn);
	562	}
	563	std::cout << "Incremental Mesh, multi-threading "
	564	<< (BRepMesh_IncrementalMesh::IsParallelDefault() ? "ON\n" : "OFF\n");
	565	return 0;
	566	}