OCCT

// Created on: 1996-12-30
// Created by: Stagiaire Mary FABIEN
// Copyright (c) 1996-1999 Matra Datavision
// Copyright (c) 1999-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 <BRep_Tool.hxx>
#include <BRepTools_GTrsfModification.hxx>
#include <Geom2d_Curve.hxx>
#include <Geom2d_TrimmedCurve.hxx>
#include <Geom_BezierCurve.hxx>
#include <Geom_BezierSurface.hxx>
#include <Geom_BSplineCurve.hxx>
#include <Geom_BSplineSurface.hxx>
#include <Geom_Curve.hxx>
#include <Geom_Surface.hxx>
#include <Geom_TrimmedCurve.hxx>
#include <GeomLib.hxx>
#include <gp_GTrsf.hxx>
#include <gp_Pnt.hxx>
#include <gp_Quaternion.hxx>
#include <gp_XYZ.hxx>
#include <Standard_NoSuchObject.hxx>
#include <Standard_Type.hxx>
#include <TopoDS_Edge.hxx>
#include <TopoDS_Face.hxx>
#include <TopoDS_Vertex.hxx>

IMPLEMENT_STANDARD_RTTIEXT(BRepTools_GTrsfModification, BRepTools_Modification)

//=================================================================================================

BRepTools_GTrsfModification::BRepTools_GTrsfModification(const gp_GTrsf& T)
    : myGTrsf(T)
{
  // use the sup norm as the maximum dilation for the tolerance
  double loc1, loc2, loc3, loc4;

  loc1 = std::max(std::abs(T.Value(1, 1)), std::abs(T.Value(1, 2)));
  loc2 = std::max(std::abs(T.Value(2, 1)), std::abs(T.Value(2, 2)));
  loc3 = std::max(std::abs(T.Value(3, 1)), std::abs(T.Value(3, 2)));
  loc4 = std::max(std::abs(T.Value(1, 3)), std::abs(T.Value(2, 3)));

  loc1 = std::max(loc1, loc2);
  loc2 = std::max(loc3, loc4);

  loc1 = std::max(loc1, loc2);

  myGScale = std::max(loc1, std::abs(T.Value(3, 3)));
}

//=================================================================================================

gp_GTrsf& BRepTools_GTrsfModification::GTrsf()
{
  return myGTrsf;
}

//=================================================================================================

bool BRepTools_GTrsfModification::NewSurface(const TopoDS_Face&         F,
                                             occ::handle<Geom_Surface>& S,
                                             TopLoc_Location&           L,
                                             double&                    Tol,
                                             bool&                      RevWires,
                                             bool&                      RevFace)
{
  gp_GTrsf gtrsf;
  gtrsf.SetVectorialPart(myGTrsf.VectorialPart());
  gtrsf.SetTranslationPart(myGTrsf.TranslationPart());
  S = BRep_Tool::Surface(F, L);
  if (S.IsNull())
  {
    // processing the case when there is no geometry
    return false;
  }
  S = occ::down_cast<Geom_Surface>(S->Copy());

  Tol = BRep_Tool::Tolerance(F);
  Tol *= myGScale;
  RevWires = false;
  RevFace  = myGTrsf.IsNegative();
  S        = occ::down_cast<Geom_Surface>(S->Transformed(L.Transformation()));

  occ::handle<Standard_Type> TheTypeS = S->DynamicType();
  if (TheTypeS == STANDARD_TYPE(Geom_BSplineSurface))
  {
    occ::handle<Geom_BSplineSurface> S2 = occ::down_cast<Geom_BSplineSurface>(S);
    for (int i = 1; i <= S2->NbUPoles(); i++)
    {
      for (int j = 1; j <= S2->NbVPoles(); j++)
      {
        gp_XYZ coor(S2->Pole(i, j).Coord());
        gtrsf.Transforms(coor);
        gp_Pnt P(coor);
        S2->SetPole(i, j, P);
      }
    }
  }
  else if (TheTypeS == STANDARD_TYPE(Geom_BezierSurface))
  {
    occ::handle<Geom_BezierSurface> S2 = occ::down_cast<Geom_BezierSurface>(S);
    for (int i = 1; i <= S2->NbUPoles(); i++)
    {
      for (int j = 1; j <= S2->NbVPoles(); j++)
      {
        gp_XYZ coor(S2->Pole(i, j).Coord());
        gtrsf.Transforms(coor);
        gp_Pnt P(coor);
        S2->SetPole(i, j, P);
      }
    }
  }
  else
  {
    throw Standard_NoSuchObject("BRepTools_GTrsfModification : Pb no BSpline/Bezier Type Surface");
  }

  L.Identity();
  return true;
}

//=================================================================================================

bool BRepTools_GTrsfModification::NewCurve(const TopoDS_Edge&       E,
                                           occ::handle<Geom_Curve>& C,
                                           TopLoc_Location&         L,
                                           double&                  Tol)
{
  double   f, l;
  gp_GTrsf gtrsf;
  gtrsf.SetVectorialPart(myGTrsf.VectorialPart());
  gtrsf.SetTranslationPart(myGTrsf.TranslationPart());
  Tol = BRep_Tool::Tolerance(E) * myGScale;
  C   = BRep_Tool::Curve(E, L, f, l);

  if (!C.IsNull())
  {
    C = occ::down_cast<Geom_Curve>(C->Copy()->Transformed(L.Transformation()));
    occ::handle<Standard_Type> TheTypeC = C->DynamicType();
    if (TheTypeC == STANDARD_TYPE(Geom_BSplineCurve))
    {
      occ::handle<Geom_BSplineCurve> C2 = occ::down_cast<Geom_BSplineCurve>(C);
      for (int i = 1; i <= C2->NbPoles(); i++)
      {
        gp_XYZ coor(C2->Pole(i).Coord());
        gtrsf.Transforms(coor);
        gp_Pnt P(coor);
        C2->SetPole(i, P);
      }
    }
    else if (TheTypeC == STANDARD_TYPE(Geom_BezierCurve))
    {
      occ::handle<Geom_BezierCurve> C2 = occ::down_cast<Geom_BezierCurve>(C);
      for (int i = 1; i <= C2->NbPoles(); i++)
      {
        gp_XYZ coor(C2->Pole(i).Coord());
        gtrsf.Transforms(coor);
        gp_Pnt P(coor);
        C2->SetPole(i, P);
      }
    }
    else
    {
      throw Standard_NoSuchObject("BRepTools_GTrsfModification : Pb no BSpline/Bezier Type Curve");
    }
    C = new Geom_TrimmedCurve(C, f, l);
  }
  L.Identity();
  return !C.IsNull();
}

//=================================================================================================

bool BRepTools_GTrsfModification::NewPoint(const TopoDS_Vertex& V, gp_Pnt& P, double& Tol)
{
  gp_Pnt Pnt = BRep_Tool::Pnt(V);
  Tol        = BRep_Tool::Tolerance(V);
  Tol *= myGScale;
  gp_XYZ coor(Pnt.Coord());
  myGTrsf.Transforms(coor);
  P.SetXYZ(coor);

  return true;
}

//=================================================================================================

bool BRepTools_GTrsfModification::NewCurve2d(const TopoDS_Edge& E,
                                             const TopoDS_Face& F,
                                             const TopoDS_Edge&,
                                             const TopoDS_Face&,
                                             occ::handle<Geom2d_Curve>& C,
                                             double&                    Tol)
{
  TopLoc_Location loc;
  Tol = BRep_Tool::Tolerance(E);
  Tol *= myGScale;
  double f, l;
  C = BRep_Tool::CurveOnSurface(E, F, f, l);
  if (C.IsNull())
  {
    // processing the case when there is no geometry
    return false;
  }
  C = new Geom2d_TrimmedCurve(C, f, l);
  return true;
}

//=================================================================================================

bool BRepTools_GTrsfModification::NewParameter(const TopoDS_Vertex& V,
                                               const TopoDS_Edge&   E,
                                               double&              P,
                                               double&              Tol)
{
  Tol = BRep_Tool::Tolerance(V);
  Tol *= myGScale;
  P = BRep_Tool::Parameter(V, E);
  return true;
}

//=================================================================================================

GeomAbs_Shape BRepTools_GTrsfModification::Continuity(const TopoDS_Edge& E,
                                                      const TopoDS_Face& F1,
                                                      const TopoDS_Face& F2,
                                                      const TopoDS_Edge&,
                                                      const TopoDS_Face&,
                                                      const TopoDS_Face&)
{
  return BRep_Tool::Continuity(E, F1, F2);
}

//=================================================================================================

bool BRepTools_GTrsfModification::NewTriangulation(
  const TopoDS_Face&               theFace,
  occ::handle<Poly_Triangulation>& theTriangulation)
{
  TopLoc_Location aLoc;
  theTriangulation = BRep_Tool::Triangulation(theFace, aLoc);
  if (theTriangulation.IsNull())
  {
    return false;
  }

  gp_GTrsf aGTrsf;
  aGTrsf.SetVectorialPart(myGTrsf.VectorialPart());
  aGTrsf.SetTranslationPart(myGTrsf.TranslationPart());
  aGTrsf.Multiply(aLoc.Transformation());

  theTriangulation = theTriangulation->Copy();
  theTriangulation->SetCachedMinMax(Bnd_Box()); // clear bounding box
  theTriangulation->Deflection(theTriangulation->Deflection() * std::abs(myGScale));
  // apply transformation to 3D nodes
  for (int anInd = 1; anInd <= theTriangulation->NbNodes(); ++anInd)
  {
    gp_Pnt aP = theTriangulation->Node(anInd);
    aGTrsf.Transforms(aP.ChangeCoord());
    theTriangulation->SetNode(anInd, aP);
  }
  // modify triangles orientation in case of mirror transformation
  if (myGScale < 0.0)
  {
    for (int anInd = 1; anInd <= theTriangulation->NbTriangles(); ++anInd)
    {
      Poly_Triangle aTria = theTriangulation->Triangle(anInd);
      int           aN1, aN2, aN3;
      aTria.Get(aN1, aN2, aN3);
      aTria.Set(aN1, aN3, aN2);
      theTriangulation->SetTriangle(anInd, aTria);
    }
  }
  // modify normals
  if (theTriangulation->HasNormals())
  {
    for (int anInd = 1; anInd <= theTriangulation->NbNodes(); ++anInd)
    {
      gp_Dir aNormal = theTriangulation->Normal(anInd);
      gp_Mat aMat    = aGTrsf.VectorialPart();
      aMat.SetDiagonal(1., 1., 1.);
      gp_Trsf aTrsf;
      aTrsf.SetForm(gp_Rotation);
      (gp_Mat&)aTrsf.HVectorialPart() = aMat;
      aNormal.Transform(aTrsf);
      theTriangulation->SetNormal(anInd, aNormal);
    }
  }

  return true;
}

//=================================================================================================

bool BRepTools_GTrsfModification::NewPolygon(const TopoDS_Edge&           theEdge,
                                             occ::handle<Poly_Polygon3D>& thePoly)
{
  TopLoc_Location aLoc;
  thePoly = BRep_Tool::Polygon3D(theEdge, aLoc);
  if (thePoly.IsNull())
  {
    return false;
  }

  gp_GTrsf aGTrsf;
  aGTrsf.SetVectorialPart(myGTrsf.VectorialPart());
  aGTrsf.SetTranslationPart(myGTrsf.TranslationPart());
  aGTrsf.Multiply(aLoc.Transformation());

  thePoly = thePoly->Copy();
  thePoly->Deflection(thePoly->Deflection() * std::abs(myGScale));
  // transform nodes
  NCollection_Array1<gp_Pnt>& aNodesArray = thePoly->ChangeNodes();
  for (int anId = aNodesArray.Lower(); anId <= aNodesArray.Upper(); ++anId)
  {
    gp_Pnt& aP = aNodesArray.ChangeValue(anId);
    aGTrsf.Transforms(aP.ChangeCoord());
  }
  return true;
}

//=================================================================================================

bool BRepTools_GTrsfModification::NewPolygonOnTriangulation(
  const TopoDS_Edge&                        theEdge,
  const TopoDS_Face&                        theFace,
  occ::handle<Poly_PolygonOnTriangulation>& thePoly)
{
  TopLoc_Location                 aLoc;
  occ::handle<Poly_Triangulation> aT = BRep_Tool::Triangulation(theFace, aLoc);
  if (aT.IsNull())
  {
    return false;
  }

  thePoly = BRep_Tool::PolygonOnTriangulation(theEdge, aT, aLoc);
  if (!thePoly.IsNull())
  {
    thePoly = thePoly->Copy();
  }
  return true;
}