src/SMESHUtils/SMESH_OctreeNode.hxx

   1 // Copyright (C) 2007-2012  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 //  SMESH SMESH_OctreeNode : Octree with Nodes set
  24 //  inherites global class SMESH_Octree
  25 //  File      : SMESH_OctreeNode.hxx
  26 //  Created   : Tue Jan 16 16:00:00 2007
  27 //  Author    : Nicolas Geimer & Aurelien Motteux  (OCC)
  28 //  Module    : SMESH
  29 //
  30 #ifndef _SMESH_OCTREENODE_HXX_
  31 #define _SMESH_OCTREENODE_HXX_
  32
  33 #include "SMESH_Utils.hxx"
  34 #include "SMESH_Octree.hxx"
  35 #include <gp_Pnt.hxx>
  36 #include "SMDS_MeshNode.hxx"
  37
  38 #include <list>
  39 #include <set>
  40 #include <map>
  41
  42 #include "SMDS_ElemIterator.hxx"
  43
  44 //forward declaration
  45 class SMDS_MeshNode;
  46 class SMESH_OctreeNode;
  47
  48 typedef SMDS_Iterator<SMESH_OctreeNode*>              SMESH_OctreeNodeIterator;
  49 typedef boost::shared_ptr<SMESH_OctreeNodeIterator>   SMESH_OctreeNodeIteratorPtr;
  50 typedef std::set< const SMDS_MeshNode*, TIDCompare >  TIDSortedNodeSet;
  51
  52 class SMESHUtils_EXPORT SMESH_OctreeNode : public SMESH_Octree {
  53
  54 public:
  55
  56   // Constructor
  57   SMESH_OctreeNode (const TIDSortedNodeSet& theNodes, const int maxLevel = 8,
  58                     const int maxNbNodes = 5, const double minBoxSize = 0.);
  59
  60 //=============================
  61 /*!
  62  * \brief Empty destructor
  63  */
  64 //=============================
  65   virtual ~SMESH_OctreeNode () {};
  66
  67   // Tells us if Node is inside the current box with the precision "precision"
  68   virtual const bool isInside(const gp_XYZ& p, const double precision = 0.);
  69
  70   // Return in Result a list of Nodes potentials to be near Node
  71   void               NodesAround(const SMDS_MeshNode *            Node,
  72                                  std::list<const SMDS_MeshNode*>* Result,
  73                                  const double                     precision = 0.);
  74
  75   // Return in dist2Nodes nodes mapped to their square distance from Node
  76   bool               NodesAround(const gp_XYZ& node,
  77                                  std::map<double, const SMDS_MeshNode*>& dist2Nodes,
  78                                  double                                  precision);
  79
  80   // Return in theGroupsOfNodes a list of group of nodes close to each other within theTolerance
  81   // Search for all the nodes in nodes
  82   void               FindCoincidentNodes ( TIDSortedNodeSet*           nodes,
  83                                            const double                theTolerance,
  84                                            std::list< std::list< const SMDS_MeshNode*> >* theGroupsOfNodes);
  85
  86   // Static method that return in theGroupsOfNodes a list of group of nodes close to each other within
  87   // theTolerance search for all the nodes in nodes
  88   static void        FindCoincidentNodes ( TIDSortedNodeSet&                              nodes,
  89                                            std::list< std::list< const SMDS_MeshNode*> >* theGroupsOfNodes,
  90                                            const double theTolerance = 0.00001,
  91                                            const int maxLevel = -1,
  92                                            const int maxNbNodes = 5);
  93   /*!
  94    * \brief Update data according to node movement
  95    */
  96   void                        UpdateByMoveNode( const SMDS_MeshNode* node, const gp_Pnt& toPnt );
  97   /*!
  98    * \brief Return iterator over children
  99    */
 100   SMESH_OctreeNodeIteratorPtr GetChildrenIterator();
 101   /*!
 102    * \brief Return nodes iterator
 103    */
 104   SMDS_NodeIteratorPtr        GetNodeIterator();
 105   /*!
 106    * \brief Return nb nodes in a tree
 107    */
 108   int                         NbNodes() const { return myNodes.size(); }
 109
 110 protected:
 111
 112   SMESH_OctreeNode (int maxNbNodes );
 113
 114   // Compute the bounding box of the whole set of nodes myNodes
 115   virtual Bnd_B3d*      buildRootBox();
 116
 117   // Shares the father's data with each of his child
 118   virtual void          buildChildrenData();
 119
 120   // Construct an empty SMESH_OctreeNode used by SMESH_Octree::buildChildren()
 121   virtual SMESH_Octree* allocateOctreeChild() const;
 122
 123   // Return in result a list of nodes closed to Node and remove it from SetOfNodes
 124   void                  FindCoincidentNodes( const SMDS_MeshNode *            Node,
 125                                              TIDSortedNodeSet*                SetOfNodes,
 126                                              std::list<const SMDS_MeshNode*>* Result,
 127                                              const double                     precision);
 128
 129   // The max number of nodes a leaf box can contain
 130   int                myMaxNbNodes;
 131
 132   // The set of nodes inside the box of the Octree (Empty if Octree is not a leaf)
 133   TIDSortedNodeSet   myNodes;
 134
 135 };
 136
 137 #endif