Salome HOME
23514: EDF 16031 - SMESH freezes
[modules/smesh.git] / src / DriverMED / DriverMED_Family.h
1 // Copyright (C) 2007-2016  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, or (at your option) any later version.
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 DriverMED : tool to split groups on families
24 //  File   : DriverMED_Family.hxx
25 //  Author : Julia DOROVSKIKH
26 //  Module : SMESH
27 //
28 #ifndef _INCLUDE_DRIVERMED_FAMILY
29 #define _INCLUDE_DRIVERMED_FAMILY
30
31 #include "SMESH_DriverMED.hxx"
32 #include "DriverMED.hxx"
33
34 #include "SMDS_Mesh.hxx"
35 #include "SMESHDS_GroupBase.hxx"
36 #include "SMESHDS_SubMesh.hxx"
37 #include "MED_Common.hxx"
38
39 #include <boost/container/flat_set.hpp>
40 #include <boost/shared_ptr.hpp>
41 #include <set>
42 #include <limits>
43
44 #define REST_NODES_FAMILY 1
45 #define FIRST_NODE_FAMILY 2
46
47 #define REST_EDGES_FAMILY   -1
48 #define REST_FACES_FAMILY   -2
49 #define REST_VOLUMES_FAMILY -3
50 #define REST_0DELEM_FAMILY  -4
51 #define REST_BALL_FAMILY    -5
52 #define FIRST_ELEM_FAMILY   -6
53
54 // Not In Group families
55 #define NIG_EDGES_FAMILY   INT_MAX-1
56 #define NIG_FACES_FAMILY   INT_MAX-2
57 #define NIG_VOLS_FAMILY    INT_MAX-3
58 #define NIG_0DELEM_FAMILY  INT_MAX-4
59 #define NIG_BALL_FAMILY    INT_MAX-5
60 #define NIG_GROUP_PREFIX "NOT_IN_GRP"
61
62 typedef std::list<DriverMED_FamilyPtr                   > DriverMED_FamilyPtrList;
63 typedef std::map<int,SMESHDS_SubMesh*                   > SMESHDS_SubMeshPtrMap;
64 typedef std::list<SMESHDS_GroupBase*                    > SMESHDS_GroupBasePtrList;
65 typedef std::set<const SMDS_MeshElement*,TIDCompare     > ElementsSet;
66 typedef boost::container::flat_set< SMDSAbs_ElementType > ElemTypeSet;
67
68 class MESHDRIVERMED_EXPORT DriverMED_Family
69 {
70  public:
71
72   DriverMED_Family();
73
74   //! Methods for groups storing to MED
75   /*!
76     Split each group from list <theGroups> and each sub-mesh from list <theSubMeshes>
77     on some parts (families) on the basis of the elements membership in other groups
78     from <theGroups> and other sub-meshes from <theSubMeshes>.
79     Resulting families have no common elements.
80   */
81   static 
82   DriverMED_FamilyPtrList
83   MakeFamilies (SMESHDS_SubMeshIteratorPtr      theSubMeshes,
84                 const SMESHDS_GroupBasePtrList& theGroups,
85                 const bool doGroupOfNodes,
86                 const bool doGroupOfEdges,
87                 const bool doGroupOfFaces,
88                 const bool doGroupOfVolumes,
89                 const bool doGroupOf0DElems,
90                 const bool doGroupOfBalls,
91                 const bool doAllInGroups);
92
93   //! Create TFamilyInfo for this family
94   MED::PFamilyInfo 
95   GetFamilyInfo (const MED::PWrapper& theWrapper, 
96                  const MED::PMeshInfo& theMeshInfo) const;
97
98   //! Returns elements of this family
99   const ElementsSet& GetElements () const;
100
101   //! Returns a family ID
102   int GetId () const;
103
104   //! Sets a family ID
105   void SetId (const int theId);
106
107  public:
108
109   // Methods for groups reading from MED
110
111   void AddElement(const SMDS_MeshElement* theElement);
112
113   const MED::TStringSet& GetGroupNames() const;
114   void AddGroupName(std::string theGroupName);
115
116   void SetType(const SMDSAbs_ElementType theType);
117   SMDSAbs_ElementType GetType();
118   const ElemTypeSet&  GetTypes() const;
119
120   bool MemberOf(std::string theGroupName) const;
121
122   int GetGroupAttributVal() const;
123   void SetGroupAttributVal( int theValue);
124
125  private:
126   //! Initialize the tool by SMESHDS_GroupBase
127   void Init (SMESHDS_GroupBase* group);
128
129   //! Split <theSubMesh> on some parts (families) on the basis of the elements type.
130   static
131   DriverMED_FamilyPtrList SplitByType(SMESHDS_SubMesh* theSubMesh,
132                                       const int        theId);
133
134
135   /*! Remove from <Elements> elements, common with <by>,
136     Remove from <by> elements, common with <Elements>,
137     Create family <common> from common elements, with combined groups list.
138   */
139   void Split (DriverMED_FamilyPtr by,
140               DriverMED_FamilyPtr common);
141
142   //! Check, if this family has empty list of elements
143   bool IsEmpty () const;
144
145
146  private:
147   int                 myId;
148   SMDSAbs_ElementType myType;
149   ElementsSet         myElements;
150   MED::TStringSet     myGroupNames;
151   int                 myGroupAttributVal;
152   ElemTypeSet         myTypes; // Issue 0020576
153 };
154
155 #endif