[HYDRO module - Feature #523] river, channel, embankment meshing

[modules/smesh.git] / src / StdMeshers / StdMeshers_Quadrangle_2D.hxx

diff --git a/src/StdMeshers/StdMeshers_Quadrangle_2D.hxx b/src/StdMeshers/StdMeshers_Quadrangle_2D.hxx
index 9fe6e120a9b659da4b54ae693e3e5ef6f6713a2a..6a06c9a46af65cccf7d52a7d132a2585e8784cb6 100644 (file)
--- a/src/StdMeshers/StdMeshers_Quadrangle_2D.hxx
+++ b/src/StdMeshers/StdMeshers_Quadrangle_2D.hxx
@@ -1,4 +1,4 @@
-// Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
+// Copyright (C) 2007-2015  CEA/DEN, EDF R&D, OPEN CASCADE
 //
 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
@@ -6,7 +6,7 @@
 // This library is free software; you can redistribute it and/or
 // modify it under the terms of the GNU Lesser General Public
 // License as published by the Free Software Foundation; either
-// version 2.1 of the License.
+// version 2.1 of the License, or (at your option) any later version.
 //
 // This library is distributed in the hope that it will be useful,
 // but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ -58,6 +58,7 @@ struct FaceQuadStruct
     };
     StdMeshers_FaceSidePtr grid;
     int                    from, to;     // indices of grid points used by the quad
+    int                    di;           // +1 or -1 depending on IsReversed()
     std::set<int>          forced_nodes; // indices of forced grid points
     std::vector<Contact>   contacts;     // contacts with sides of other quads
     int                    nbNodeOut;    // nb of missing nodes on an opposite shorter side
@@ -71,10 +72,15 @@ struct FaceQuadStruct
     int  ToQuadIndex( int sideNodeIndex ) const;
     bool IsForced( int nodeIndex ) const;
     bool IsReversed() const { return nbNodeOut ? false : to < from; }
+    bool Reverse(bool keepGrid);
     int  NbPoints() const { return Abs( to - from ); }
     double Param( int nodeIndex ) const;
     double Length( int from=-1, int to=-1) const;
     gp_XY Value2d( double x ) const;
+    const UVPtStruct& First() const { return GetUVPtStruct()[ from ]; }
+    const UVPtStruct& Last()  const {
+      return GetUVPtStruct()[ to-nbNodeOut-(IsReversed() ? -1 : +1)];
+    }
     // some sortcuts
     const vector<UVPtStruct>& GetUVPtStruct(bool isXConst=0, double constValue=0) const
     { return nbNodeOut ?
@@ -82,16 +88,40 @@ struct FaceQuadStruct
         grid->GetUVPtStruct( isXConst, constValue );
     }
   };
+  struct SideIterator // iterator on UVPtStruct of a Side
+  {
+    const UVPtStruct *uvPtr, *uvEnd;
+    int               dPtr, counter;
+    SideIterator(): uvPtr(0), uvEnd(0), dPtr(0), counter(0) {}
+    void Init( const Side& side ) {
+      dPtr  = counter = 0;
+      uvPtr = uvEnd = 0;
+      if ( side.NbPoints() > 0 ) {
+        uvPtr = & side.First();
+        uvEnd = & side.Last();
+        dPtr  = ( uvEnd > uvPtr ) ? +1 : -1;
+        uvEnd += dPtr;
+      }
+    }
+    bool More() const { return uvPtr != uvEnd; }
+    void Next() { uvPtr += dPtr; ++counter; }
+    UVPtStruct& UVPt() const { return (UVPtStruct&) *uvPtr; }
+    UVPtStruct& operator[](int i) { return (UVPtStruct&) uvPtr[ i*dPtr]; }
+    int  Count() const { return counter; }
+  };
 
   std::vector< Side >      side;
   std::vector< UVPtStruct> uv_grid;
   int                      iSize, jSize;
   TopoDS_Face              face;
   Bnd_B2d                  uv_box;
+  std::string              name; // to ease debugging
 
-  FaceQuadStruct ( const TopoDS_Face& F = TopoDS_Face() );
+  FaceQuadStruct ( const TopoDS_Face& F = TopoDS_Face(), const std::string& nm="main" );
   UVPtStruct& UVPt( int i, int j ) { return uv_grid[ i + j * iSize ]; }
-  void  shift    ( size_t nb, bool keepUnitOri );
+  double&        U( int i, int j ) { return UVPt( i, j ).u; }
+  double&        V( int i, int j ) { return UVPt( i, j ).v; }
+  void  shift    ( size_t nb, bool keepUnitOri, bool keepGrid=false );
   int & nbNodeOut( int iSide ) { return side[ iSide ].nbNodeOut; }
   bool  findCell ( const gp_XY& uv, int & i, int & j );
   bool  isNear   ( const gp_XY& uv, int & i, int & j, int nbLoops=1 );
@@ -104,7 +134,7 @@ struct FaceQuadStruct
 
 class STDMESHERS_EXPORT StdMeshers_Quadrangle_2D: public SMESH_2D_Algo
 {
-public:
+ public:
   StdMeshers_Quadrangle_2D(int hypId, int studyId, SMESH_Gen* gen);
   virtual ~StdMeshers_Quadrangle_2D();
 
@@ -127,7 +157,9 @@ public:
                                    const TopoDS_Shape& aShape,
                                    const bool          considerMesh=false);
 
-protected:
+  static bool IsApplicable(const TopoDS_Shape & aShape, bool toCheckAll);
+
+ protected:
 
   bool checkNbEdgesForEvaluate(SMESH_Mesh& aMesh,
                                const TopoDS_Shape & aShape,
@@ -136,7 +168,7 @@ protected:
                                bool& IsQuadratic);
 
   bool setNormalizedGrid(FaceQuadStruct::Ptr quad);
-  
+
   void splitQuadFace(SMESHDS_Mesh *       theMeshDS,
                      const int            theFaceID,
                      const SMDS_MeshNode* theNode1,
@@ -173,6 +205,8 @@ protected:
 
   void smooth (FaceQuadStruct::Ptr quad);
 
+  bool check();
+
   int getCorners(const TopoDS_Face&          theFace,
                  SMESH_Mesh &                theMesh,
                  std::list<TopoDS_Edge>&     theWire,
@@ -186,6 +220,8 @@ protected:
 
   int splitQuad(FaceQuadStruct::Ptr quad, int i, int j);
 
+  void shiftQuad(FaceQuadStruct::Ptr& quad, const int num );
+
   typedef std::map< StdMeshers_FaceSidePtr, std::vector< FaceQuadStruct::Ptr > > TQuadsBySide;
   void updateSideUV( FaceQuadStruct::Side&  side,
                      int                    iForced,
@@ -193,12 +229,12 @@ protected:
                      int *                  iNext=NULL);
 
 
-  // Fields
+ protected: // Fields
 
   bool myQuadranglePreference;
   bool myTrianglePreference;
   int  myTriaVertexID;
-  bool myNeedSmooth;
+  bool myNeedSmooth, myCheckOri;
   const StdMeshers_QuadrangleParams* myParams;
   StdMeshers_QuadType                myQuadType;