Salome HOME
0022098: EDF 2036 SMESH: Create groups from none conected parts of a mesh
[modules/smesh.git] / src / SMESHUtils / SMESH_OctreeNode.hxx
1 // Copyright (C) 2007-2013  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   struct Limit : public SMESH_TreeLimit
113   {
114     int myMaxNbNodes;
115     Limit(int maxLevel, double minSize, int maxNbNodes)
116       :SMESH_TreeLimit(maxLevel, minSize), myMaxNbNodes(maxNbNodes) {}
117   };
118
119   int                   getMaxNbNodes() const;
120
121   SMESH_OctreeNode();
122
123   // Compute the bounding box of the whole set of nodes myNodes
124   virtual Bnd_B3d*      buildRootBox();
125
126   // Shares the father's data with each of his child
127   virtual void          buildChildrenData();
128
129   // Construct an empty SMESH_OctreeNode used by SMESH_Octree::buildChildren()
130   virtual SMESH_Octree* newChild() const;
131
132   // Return in result a list of nodes closed to Node and remove it from SetOfNodes
133   void                  FindCoincidentNodes( const SMDS_MeshNode *            Node,
134                                              TIDSortedNodeSet*                SetOfNodes,
135                                              std::list<const SMDS_MeshNode*>* Result,
136                                              const double                     precision);
137
138   // The set of nodes inside the box of the Octree (Empty if Octree is not a leaf)
139   TIDSortedNodeSet   myNodes;
140
141 };
142
143 #endif