OCCT

// Copyright (c) 2015 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.

#ifndef _ShapePersistent_BRep_HeaderFile
#define _ShapePersistent_BRep_HeaderFile

#include <ShapePersistent_TopoDS.hxx>
#include <ShapePersistent_Geom2d.hxx>
#include <ShapePersistent_Poly.hxx>
#include <NCollection_DataMap.hxx>
#include <StdObject_Location.hxx>
#include <StdObject_gp_Vectors.hxx>

#include <BRep_PointRepresentation.hxx>
#include <NCollection_List.hxx>
#include <BRep_CurveRepresentation.hxx>

#include <gp_Pnt.hxx>
#include <gp_Pnt2d.hxx>
class Standard_Transient;
class StdObjMgt_Persistent;

class BRep_PointRepresentation;
class BRep_CurveRepresentation;
class TopoDS_Vertex;
class TopoDS_Edge;
class TopoDS_Face;

class ShapePersistent_BRep : public ShapePersistent_TopoDS
{
public:
  class PointRepresentation : public StdObjMgt_Persistent
  {
    friend class ShapePersistent_BRep;

  public:
    //! Empty constructor.
    PointRepresentation()
        : myParameter(0.0)
    {
    }

    //! Read persistent data from a file.
    Standard_EXPORT void Read(StdObjMgt_ReadData& theReadData) override;
    //! Write persistent data to a file.
    Standard_EXPORT void Write(StdObjMgt_WriteData& theWriteData) const override;
    //! Gets persistent child objects
    void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const override;

    //! Returns persistent type name
    const char* PName() const override { return "PBRep_PointRepresentation"; }

    //! Import transient object from the persistent data.
    Standard_EXPORT void Import(
      NCollection_List<occ::handle<BRep_PointRepresentation>>& thePoints) const;

  protected:
    virtual occ::handle<BRep_PointRepresentation> import() const;

  protected:
    StdObject_Location myLocation;
    double             myParameter;

  private:
    occ::handle<PointRepresentation> myNext;
  };

  class PointOnCurve : public PointRepresentation
  {
    friend class ShapePersistent_BRep;

  public:
    void Read(StdObjMgt_ReadData& theReadData) override;
    void Write(StdObjMgt_WriteData& theWriteData) const override;
    void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const override;

    const char* PName() const override { return "PBRep_PointOnCurve"; }

    occ::handle<BRep_PointRepresentation> import() const override;

  private:
    Handle(ShapePersistent_Geom::Curve) myCurve;
  };

  class PointsOnSurface : public PointRepresentation
  {
    friend class ShapePersistent_BRep;

  public:
    void Read(StdObjMgt_ReadData& theReadData) override;
    void Write(StdObjMgt_WriteData& theWriteData) const override;
    void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const override;

    const char* PName() const override { return "PBRep_PointsOnSurface"; }

  protected:
    Handle(ShapePersistent_Geom::Surface) mySurface;
  };

  class PointOnCurveOnSurface : public PointsOnSurface
  {
    friend class ShapePersistent_BRep;

  public:
    void Read(StdObjMgt_ReadData& theReadData) override;
    void Write(StdObjMgt_WriteData& theWriteData) const override;
    void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const override;

    const char* PName() const override { return "PBRep_PointOnCurveOnSurface"; }

    occ::handle<BRep_PointRepresentation> import() const override;

  private:
    Handle(ShapePersistent_Geom2d::Curve) myPCurve;
  };

  class PointOnSurface : public PointsOnSurface
  {
    friend class ShapePersistent_BRep;

  public:
    PointOnSurface()
        : myParameter2(0.0)
    {
    }

    void Read(StdObjMgt_ReadData& theReadData) override;
    void Write(StdObjMgt_WriteData& theWriteData) const override;

    const char* PName() const override { return "PBRep_PointOnSurface"; }

    occ::handle<BRep_PointRepresentation> import() const override;

  private:
    double myParameter2;
  };

  class CurveRepresentation : public StdObjMgt_Persistent
  {
    friend class ShapePersistent_BRep;

  public:
    //! Read persistent data from a file.
    Standard_EXPORT void Read(StdObjMgt_ReadData& theReadData) override;
    //! Write persistent data from a file.
    Standard_EXPORT void Write(StdObjMgt_WriteData& theWriteData) const override;
    //! Gets persistent child objects
    void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const override;

    //! Returns persistent type name
    const char* PName() const override { return "PBRep_CurveRepresentation"; }

    //! Import transient object from the persistent data.
    Standard_EXPORT void Import(
      NCollection_List<occ::handle<BRep_CurveRepresentation>>& theCurves) const;

  protected:
    virtual occ::handle<BRep_CurveRepresentation> import() const;

  protected:
    StdObject_Location myLocation;

  private:
    occ::handle<CurveRepresentation> myNext;
  };

  class GCurve : public CurveRepresentation
  {
    friend class ShapePersistent_BRep;

  public:
    GCurve()
        : myFirst(0.0),
          myLast(0.0)
    {
    }

    void Read(StdObjMgt_ReadData& theReadData) override;
    void Write(StdObjMgt_WriteData& theWriteData) const override;

    const char* PName() const override { return "PBRep_GCurve"; }

  protected:
    double myFirst;
    double myLast;
  };

  class Curve3D : public GCurve
  {
    friend class ShapePersistent_BRep;

  public:
    void Read(StdObjMgt_ReadData& theReadData) override;
    void Write(StdObjMgt_WriteData& theWriteData) const override;
    void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const override;

    const char* PName() const override { return "PBRep_Curve3D"; }

    occ::handle<BRep_CurveRepresentation> import() const override;

  private:
    Handle(ShapePersistent_Geom::Curve) myCurve3D;
  };

  class CurveOnSurface : public GCurve
  {
    friend class ShapePersistent_BRep;

  public:
    void Read(StdObjMgt_ReadData& theReadData) override;
    void Write(StdObjMgt_WriteData& theWriteData) const override;
    void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const override;

    const char* PName() const override { return "PBRep_CurveOnSurface"; }

    occ::handle<BRep_CurveRepresentation> import() const override;

  protected:
    Handle(ShapePersistent_Geom2d::Curve) myPCurve;
    Handle(ShapePersistent_Geom::Surface) mySurface;
    gp_Pnt2d                              myUV1;
    gp_Pnt2d                              myUV2;
  };

  class CurveOnClosedSurface : public CurveOnSurface
  {
    friend class ShapePersistent_BRep;

  public:
    CurveOnClosedSurface()
        : myContinuity(0)
    {
    }

    void Read(StdObjMgt_ReadData& theReadData) override;
    void Write(StdObjMgt_WriteData& theWriteData) const override;
    void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const override;

    const char* PName() const override { return "PBRep_CurveOnClosedSurface"; }

    occ::handle<BRep_CurveRepresentation> import() const override;

  private:
    Handle(ShapePersistent_Geom2d::Curve) myPCurve2;
    int                                   myContinuity;
    gp_Pnt2d                              myUV21;
    gp_Pnt2d                              myUV22;
  };

  class Polygon3D : public CurveRepresentation
  {
    friend class ShapePersistent_BRep;

  public:
    void Read(StdObjMgt_ReadData& theReadData) override;
    void Write(StdObjMgt_WriteData& theWriteData) const override;
    void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const override;

    const char* PName() const override { return "PBRep_Polygon3D"; }

    occ::handle<BRep_CurveRepresentation> import() const override;

  private:
    Handle(ShapePersistent_Poly::Polygon3D) myPolygon3D;
  };

  class PolygonOnTriangulation : public CurveRepresentation
  {
    friend class ShapePersistent_BRep;

  public:
    void Read(StdObjMgt_ReadData& theReadData) override;
    void Write(StdObjMgt_WriteData& theWriteData) const override;
    void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const override;

    const char* PName() const override { return "PBRep_PolygonOnTriangulation"; }

    occ::handle<BRep_CurveRepresentation> import() const override;

  protected:
    Handle(ShapePersistent_Poly::PolygonOnTriangulation) myPolygon;
    Handle(ShapePersistent_Poly::Triangulation)          myTriangulation;
  };

  class PolygonOnClosedTriangulation : public PolygonOnTriangulation
  {
    friend class ShapePersistent_BRep;

  public:
    void Read(StdObjMgt_ReadData& theReadData) override;
    void Write(StdObjMgt_WriteData& theWriteData) const override;
    void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const override;

    const char* PName() const override { return "PBRep_PolygonOnClosedTriangulation"; }

    occ::handle<BRep_CurveRepresentation> import() const override;

  private:
    Handle(ShapePersistent_Poly::PolygonOnTriangulation) myPolygon2;
  };

  class PolygonOnSurface : public CurveRepresentation
  {
    friend class ShapePersistent_BRep;

  public:
    void Read(StdObjMgt_ReadData& theReadData) override;
    void Write(StdObjMgt_WriteData& theWriteData) const override;
    void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const override;

    const char* PName() const override { return "PBRep_PolygonOnSurface"; }

    occ::handle<BRep_CurveRepresentation> import() const override;

  protected:
    Handle(ShapePersistent_Poly::Polygon2D) myPolygon2D;
    Handle(ShapePersistent_Geom::Surface)   mySurface;
  };

  class PolygonOnClosedSurface : public PolygonOnSurface
  {
    friend class ShapePersistent_BRep;

  public:
    void Read(StdObjMgt_ReadData& theReadData) override;
    void Write(StdObjMgt_WriteData& theWriteData) const override;
    void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const override;

    const char* PName() const override { return "PBRep_PolygonOnClosedSurface"; }

    occ::handle<BRep_CurveRepresentation> import() const override;

  private:
    Handle(ShapePersistent_Poly::Polygon2D) myPolygon2;
  };

  class CurveOn2Surfaces : public CurveRepresentation
  {
    friend class ShapePersistent_BRep;

  public:
    CurveOn2Surfaces()
        : myContinuity(0)
    {
    }

    void Read(StdObjMgt_ReadData& theReadData) override;
    void Write(StdObjMgt_WriteData& theWriteData) const override;
    void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const override;

    const char* PName() const override { return "PBRep_CurveOn2Surfaces"; }

    occ::handle<BRep_CurveRepresentation> import() const override;

  private:
    Handle(ShapePersistent_Geom::Surface) mySurface;
    Handle(ShapePersistent_Geom::Surface) mySurface2;
    StdObject_Location                    myLocation2;
    int                                   myContinuity;
  };

private:
  class pTVertex : public pTBase
  {
    friend class ShapePersistent_BRep;

  public:
    pTVertex()
        : myTolerance(0.0)
    {
    }

    inline void Read(StdObjMgt_ReadData& theReadData)
    {
      pTBase::Read(theReadData);
      theReadData >> myTolerance >> myPnt >> myPoints;
    }

    inline void Write(StdObjMgt_WriteData& theWriteData) const
    {
      pTBase::Write(theWriteData);
      theWriteData << myTolerance << myPnt << myPoints;
    }

    inline void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const
    {
      pTBase::PChildren(theChildren);
      theChildren.Append(myPoints);
    }

    inline const char* PName() const { return "PBRep_TVertex"; }

  private:
    occ::handle<TopoDS_TShape> createTShape() const override;

  private:
    double                           myTolerance;
    gp_Pnt                           myPnt;
    occ::handle<PointRepresentation> myPoints;
  };

  class pTEdge : public pTBase
  {
    friend class ShapePersistent_BRep;

  public:
    pTEdge()
        : myTolerance(0.0),
          myFlags(0)
    {
    }

    inline void Read(StdObjMgt_ReadData& theReadData)
    {
      pTBase::Read(theReadData);
      theReadData >> myTolerance >> myFlags >> myCurves;
    }

    inline void Write(StdObjMgt_WriteData& theWriteData) const
    {
      pTBase::Write(theWriteData);
      theWriteData << myTolerance << myFlags << myCurves;
    }

    inline void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const
    {
      pTBase::PChildren(theChildren);
      theChildren.Append(myCurves);
    }

    inline const char* PName() const { return "PBRep_TEdge"; }

  private:
    occ::handle<TopoDS_TShape> createTShape() const override;

  private:
    double                           myTolerance;
    int                              myFlags;
    occ::handle<CurveRepresentation> myCurves;
  };

  class pTFace : public pTBase
  {
    friend class ShapePersistent_BRep;

  public:
    pTFace()
        : myTolerance(0.0),
          myNaturalRestriction(false)
    {
    }

    inline void Read(StdObjMgt_ReadData& theReadData)
    {
      pTBase::Read(theReadData);
      theReadData >> mySurface >> myTriangulation >> myLocation;
      theReadData >> myTolerance >> myNaturalRestriction;
    }

    inline void Write(StdObjMgt_WriteData& theWriteData) const
    {
      pTBase::Write(theWriteData);
      theWriteData << mySurface << myTriangulation << myLocation;
      theWriteData << myTolerance << myNaturalRestriction;
    }

    inline void PChildren(StdObjMgt_Persistent::SequenceOfPersistent& theChildren) const
    {
      pTBase::PChildren(theChildren);
      theChildren.Append(mySurface);
      theChildren.Append(myTriangulation);
      myLocation.PChildren(theChildren);
    }

    inline const char* PName() const { return "PBRep_TFace"; }

  private:
    occ::handle<TopoDS_TShape> createTShape() const override;

  private:
    Handle(ShapePersistent_Geom::Surface)       mySurface;
    Handle(ShapePersistent_Poly::Triangulation) myTriangulation;
    StdObject_Location                          myLocation;
    double                                      myTolerance;
    bool                                        myNaturalRestriction;
  };

public:
  typedef tObject<pTVertex> TVertex;
  typedef tObject<pTEdge>   TEdge;
  typedef tObject<pTFace>   TFace;

  typedef tObject1<pTVertex> TVertex1;
  typedef tObject1<pTEdge>   TEdge1;
  typedef tObject1<pTFace>   TFace1;

public:
  //! Create a persistent object for a vertex
  Standard_EXPORT static Handle(TVertex::pTObjectT) Translate(
    const TopoDS_Vertex& theVertex,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>&
      theMap);
  //! Create a persistent object for an edge
  Standard_EXPORT static Handle(TEdge::pTObjectT) Translate(
    const TopoDS_Edge& theEdge,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>& theMap,
    ShapePersistent_TriangleMode theTriangleMode);
  //! Create a persistent object for a face
  Standard_EXPORT static Handle(TFace::pTObjectT) Translate(
    const TopoDS_Face& theFace,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>& theMap,
    ShapePersistent_TriangleMode theTriangleMode);
  //! Create a persistent object for a point on a 3D curve
  Standard_EXPORT static occ::handle<PointOnCurve> Translate(
    double                         theParam,
    const occ::handle<Geom_Curve>& theCurve,
    const TopLoc_Location&         theLoc,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>&
      theMap);
  //! Create a persistent object for a point on a 3D curve on a surface
  Standard_EXPORT static occ::handle<PointOnCurveOnSurface> Translate(
    double                           theParam,
    const occ::handle<Geom2d_Curve>& theCurve,
    const occ::handle<Geom_Surface>& theSurf,
    const TopLoc_Location&           theLoc,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>&
      theMap);
  //! Create a persistent object for a point on a surface
  Standard_EXPORT static occ::handle<PointOnSurface> Translate(
    double                           theParam,
    double                           theParam2,
    const occ::handle<Geom_Surface>& theSurf,
    const TopLoc_Location&           theLoc,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>&
      theMap);
  //! Create a persistent object for a curve on a surface
  Standard_EXPORT static occ::handle<CurveOnSurface> Translate(
    const occ::handle<Geom2d_Curve>& theCurve,
    const double                     theFirstParam,
    const double                     theLastParam,
    const occ::handle<Geom_Surface>& theSurf,
    const TopLoc_Location&           theLoc,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>&
      theMap);
  //! Create a persistent object for a curve on a closed surface
  Standard_EXPORT static occ::handle<CurveOnClosedSurface> Translate(
    const occ::handle<Geom2d_Curve>& theCurve,
    const occ::handle<Geom2d_Curve>& theCurve2,
    const double                     theFirstParam,
    const double                     theLastParam,
    const occ::handle<Geom_Surface>& theSurf,
    const TopLoc_Location&           theLoc,
    const GeomAbs_Shape              theContinuity,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>&
      theMap);
  //! Create a persistent object for a curve on two surfaces
  Standard_EXPORT static occ::handle<CurveOn2Surfaces> Translate(
    const occ::handle<Geom_Surface>& theSurf,
    const occ::handle<Geom_Surface>& theSurf2,
    const TopLoc_Location&           theLoc,
    const TopLoc_Location&           theLoc2,
    const GeomAbs_Shape              theContinuity,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>&
      theMap);
  //! Create a persistent object for a 3D curve
  Standard_EXPORT static occ::handle<Curve3D> Translate(
    const occ::handle<Geom_Curve>& theCurve,
    const double                   theFirstParam,
    const double                   theLastParam,
    const TopLoc_Location&         theLoc,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>&
      theMap);
  //! Create a persistent object for a 3D polygon
  Standard_EXPORT static occ::handle<Polygon3D> Translate(
    const occ::handle<Poly_Polygon3D>& thePoly,
    const TopLoc_Location&             theLoc,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>&
      theMap);
  //! Create a persistent object for a polygon on a closed surface
  Standard_EXPORT static occ::handle<PolygonOnClosedSurface> Translate(
    const occ::handle<Poly_Polygon2D>& thePoly,
    const occ::handle<Poly_Polygon2D>& thePoly2,
    const occ::handle<Geom_Surface>&   theSurf,
    const TopLoc_Location&             theLoc,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>&
      theMap);
  //! Create a persistent object for a polygon on a surface
  Standard_EXPORT static occ::handle<PolygonOnSurface> Translate(
    const occ::handle<Poly_Polygon2D>& thePoly,
    const occ::handle<Geom_Surface>&   theSurf,
    const TopLoc_Location&             theLoc,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>&
      theMap);
  //! Create a persistent object for a polygon on a surface
  Standard_EXPORT static occ::handle<PolygonOnClosedTriangulation> Translate(
    const occ::handle<Poly_PolygonOnTriangulation>& thePolyOnTriang,
    const occ::handle<Poly_PolygonOnTriangulation>& thePolyOnTriang2,
    const occ::handle<Poly_Triangulation>&          thePolyTriang,
    const TopLoc_Location&                          theLoc,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>&
      theMap);
  //! Create a persistent object for a polygon on a surface
  Standard_EXPORT static occ::handle<PolygonOnTriangulation> Translate(
    const occ::handle<Poly_PolygonOnTriangulation>& thePolyOnTriang,
    const occ::handle<Poly_Triangulation>&          thePolyTriang,
    const TopLoc_Location&                          theLoc,
    NCollection_DataMap<occ::handle<Standard_Transient>, occ::handle<StdObjMgt_Persistent>>&
      theMap);
};

#endif