Salome HOME
untabify
[modules/smesh.git] / src / SMESH / SMESH_MesherHelper.hxx
index 733cfa4a436e0ee1870b069d74246e5bcd259d05..65c3f600d9f8e9bd29162044f3c651703b5764db 100644 (file)
 #include <SMDS_MeshNode.hxx>
 #include <SMDS_QuadraticEdge.hxx>
 
+#include <Geom_Surface.hxx>
 #include <TopoDS_Face.hxx>
 #include <TopoDS_Shape.hxx>
 #include <gp_Pnt2d.hxx>
 
 #include <map>
 
-typedef std::pair<const SMDS_MeshNode*, const SMDS_MeshNode*> NLink;
-typedef std::map<NLink, const SMDS_MeshNode*>                 NLinkNodeMap;
-typedef std::map<NLink, const SMDS_MeshNode*>::iterator       ItNLinkNode;
+typedef std::map<SMESH_TLink, const SMDS_MeshNode*>           TLinkNodeMap;
+typedef std::map<SMESH_TLink, const SMDS_MeshNode*>::iterator ItTLinkNode;
 
 /*!
  * \brief It helps meshers to add elements
@@ -91,9 +91,8 @@ public:
    * \param meshDS - mesh DS
    * \retval TopoDS_Shape - found support shape
    */
-  static const TopoDS_Shape& GetSubShapeByNode(const SMDS_MeshNode* node,
-                                               SMESHDS_Mesh*        meshDS)
-  { return meshDS->IndexToShape( node->GetPosition()->GetShapeId() ); }
+  static TopoDS_Shape GetSubShapeByNode(const SMDS_MeshNode* node,
+                                        SMESHDS_Mesh*        meshDS);
 
   /*!
    * \brief Return a valid node index, fixing the given one if necessary
@@ -126,7 +125,7 @@ public:
     
   /*!
    * Check submesh for given shape: if all elements on this shape are quadratic,
-   * quadratic elements will be created. Also fill myNLinkNodeMap
+   * quadratic elements will be created. Also fill myTLinkNodeMap
    */
   bool IsQuadraticSubMesh(const TopoDS_Shape& theShape);
   /*!
@@ -139,6 +138,12 @@ public:
    */
   bool GetIsQuadratic() const { return myCreateQuadratic; }
 
+  /*!
+   * \brief Move medium nodes of faces and volumes to fix distorted elements
+   * \param volumeOnly - fix nodes on geom faces or not if the shape is solid
+   */
+  void FixQuadraticElements(bool volumeOnly=true);
+
   /*!
    * \brief To set created elements on the shape set by IsQuadraticSubMesh()
    *        or the next methods. By defaul elements are set on the shape if
@@ -146,6 +151,15 @@ public:
    */
   void SetElementsOnShape(bool toSet) { mySetElemOnShape = toSet; }
 
+  /*!
+   * \brief Enable/disable checking of node parameters on shapes while adding elements.
+   * In case of incorrect parameters mudium node place is computed as the middle
+   * of two nodes. Default is false.
+   * NOTE that this flag is reset to "not to check" if check with non-default partameter
+   * is successful
+   */
+  void SetCheckNodePosition(bool toCheck) { myCheckNodePos = toCheck; }
+
   /*!
    * \brief Set shape to make elements on without calling IsQuadraticSubMesh()
    */
@@ -236,14 +250,22 @@ public:
    * \brief Return U of the given node on the edge
    */
   double GetNodeU(const TopoDS_Edge&   theEdge,
-                  const SMDS_MeshNode* theNode);
+                  const SMDS_MeshNode* theNode,
+                  bool*                check=0);
   /*!
    * \brief Return node UV on face
-    * \param inFaceNode - a node of element being created located inside a face
+    \param inFaceNode - a node of element being created located inside a face
    */
   gp_XY GetNodeUV(const TopoDS_Face&   F,
                   const SMDS_MeshNode* n,
-                  const SMDS_MeshNode* inFaceNode=0) const;
+                  const SMDS_MeshNode* inFaceNode=0,
+                  bool*                check=0) const;
+  /*!
+   * \brief Return middle UV taking in account surface period
+   */
+  static gp_XY GetMiddleUV(const Handle(Geom_Surface)& surface,
+                           const gp_XY&                uv1,
+                           const gp_XY&                uv2);
   /*!
    * \brief Check if inFaceNode argument is necessary for call GetNodeUV(F,..)
     * \retval bool - return true if the face is periodic
@@ -317,21 +339,21 @@ public:
                                      const SMDS_MeshNode* n2,
                                      const bool force3d);
   /*!
-   * Auxilary function for filling myNLinkNodeMap
+   * Auxilary function for filling myTLinkNodeMap
    */
-  void AddNLinkNode(const SMDS_MeshNode* n1,
+  void AddTLinkNode(const SMDS_MeshNode* n1,
                     const SMDS_MeshNode* n2,
                     const SMDS_MeshNode* n12);
   /**
-   * Auxilary function for filling myNLinkNodeMap
+   * Auxilary function for filling myTLinkNodeMap
    */
-  void AddNLinkNodeMap(const NLinkNodeMap& aMap)
-    { myNLinkNodeMap.insert(aMap.begin(), aMap.end()); }
+  void AddTLinkNodeMap(const TLinkNodeMap& aMap)
+    { myTLinkNodeMap.insert(aMap.begin(), aMap.end()); }
 
   /**
-   * Returns myNLinkNodeMap
+   * Returns myTLinkNodeMap
    */
-  const NLinkNodeMap& GetNLinkNodeMap() const { return myNLinkNodeMap; }
+  const TLinkNodeMap& GetTLinkNodeMap() const { return myTLinkNodeMap; }
 
   /**
    * Check mesh without geometry for: if all elements on this shape are quadratic,
@@ -357,7 +379,7 @@ protected:
   SMESH_MesherHelper (const SMESH_MesherHelper& theOther) {};
 
   // special map for using during creation of quadratic elements
-  NLinkNodeMap    myNLinkNodeMap;
+  TLinkNodeMap    myTLinkNodeMap;
 
   std::set< int > myDegenShapeIds;
   std::set< int > mySeamShapeIds;
@@ -371,6 +393,7 @@ protected:
   // to create quadratic elements
   bool            myCreateQuadratic;
   bool            mySetElemOnShape;
+  bool            myCheckNodePos;
 
 };