Salome HOME
bos #26515 [CEA 24622] Failing SMESH tests
[modules/smesh.git] / src / SMESHFiltersSelection / SMESH_NumberFilter.cxx
1 // Copyright (C) 2007-2021  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 //  File   : SMESH_NumberFilter.cxx
24 //  Module : SMESH
25 //
26 #include "SMESH_NumberFilter.hxx"
27
28 #include <SALOME_LifeCycleCORBA.hxx>
29
30 #include "GEOM_Client.hxx"
31
32 #include "SUIT_Application.h"
33 #include "SUIT_Session.h"
34
35 #include "SalomeApp_Application.h"
36 #include "SalomeApp_Study.h"
37 #include "LightApp_DataOwner.h"
38
39 #include "SALOME_InteractiveObject.hxx"
40 #include "SALOMEDSClient_SObject.hxx"
41 #include "SALOMEDS_SObject.hxx"
42
43 #include <TopExp_Explorer.hxx>
44 #include <TopTools_IndexedMapOfShape.hxx>
45 #include <TopExp.hxx>
46
47 /*!
48  *  Class       : SMESH_NumberFilter
49  *  Description : Filter for geom objects.
50  *                Filter geom objects by number of subshapes of the given type
51  */
52
53 //=======================================================================
54 // name    : SMESH_NumberFilter::SMESH_NumberFilter
55 // Purpose : Constructor
56 //=======================================================================
57 SMESH_NumberFilter::SMESH_NumberFilter (const char*            theKind,
58                                         const TopAbs_ShapeEnum theSubShapeType,
59                                         const int              theNumber,
60                                         const TopAbs_ShapeEnum theShapeType,
61                                         GEOM::GEOM_Object_ptr  theMainObj,
62                                         const bool             theIsClosedOnly)
63 {
64   myKind = (char*)theKind;
65   mySubShapeType = theSubShapeType;
66   myNumber = theNumber;
67   myIsClosedOnly = theIsClosedOnly;
68   myShapeTypes.Add(theShapeType);
69   myMainObj = GEOM::GEOM_Object::_duplicate(theMainObj);
70 }
71
72 //=======================================================================
73 // name    : SMESH_NumberFilter::SMESH_NumberFilter
74 // Purpose : Constructor
75 //=======================================================================
76 SMESH_NumberFilter::SMESH_NumberFilter (const char*                 theKind,
77                                         const TopAbs_ShapeEnum      theSubShapeType,
78                                         const int                   theNumber,
79                                         const TColStd_MapOfInteger& theShapeTypes,
80                                         GEOM::GEOM_Object_ptr       theMainObj,
81                                         const bool                  theIsClosedOnly )
82 {
83   myKind = (char*)theKind;
84   mySubShapeType = theSubShapeType;
85   myNumber = theNumber;
86   myIsClosedOnly = theIsClosedOnly;
87   myShapeTypes = theShapeTypes;
88   myMainObj = GEOM::GEOM_Object::_duplicate(theMainObj);
89 }
90
91 SMESH_NumberFilter::~SMESH_NumberFilter()
92 {
93 }
94
95 //=======================================================================
96 // name    : SMESH_NumberFilter::SMESH_NumberFilter
97 // Purpose : Verify validity of entry object
98 //=======================================================================
99 bool SMESH_NumberFilter::isOk (const SUIT_DataOwner* theDataOwner) const
100 {
101   if (!theDataOwner)
102     return false;
103
104   // Get geom object from IO
105   GEOM::GEOM_Object_var aGeomObj = getGeom(theDataOwner);
106   if (aGeomObj->_is_nil())
107     return false;
108
109   // Get GEOM engine
110   GEOM::GEOM_Gen_var geomEngine = aGeomObj->GetGen();
111   if ( CORBA::is_nil( geomEngine ))
112     return false;
113
114   // Get shape from geom object and verify its parameters
115   GEOM_Client aGeomClient;
116   TopoDS_Shape aShape = aGeomClient.GetShape(geomEngine.in(), aGeomObj);
117   if (aShape.IsNull() || !myShapeTypes.Contains(aShape.ShapeType()))
118     return false;
119
120   if (myIsClosedOnly && aShape.ShapeType() == TopAbs_SHELL && !aShape.Closed())
121     return false;
122
123   // Verify whether shape of entry object is sub-shape of myMainObj
124   if (!myMainObj->_is_nil()) {
125     TopoDS_Shape aMainShape = aGeomClient.GetShape(geomEngine.in(), myMainObj);
126     if (aMainShape.IsNull())
127       return false;
128
129     TopExp_Explorer anExp (aMainShape, aShape.ShapeType());
130     for (; anExp.More(); anExp.Next())
131       if (anExp.Current() == aShape)
132         break;
133     if (!anExp.More())
134       return false;
135   }
136
137   // Verify number of sub-shapes
138   if (mySubShapeType == TopAbs_SHAPE)
139     return true;
140
141   TopTools_IndexedMapOfShape aMap;
142   TopExp::MapShapes(aShape, mySubShapeType, aMap);
143
144   if ( myNumber )
145     return myNumber == aMap.Extent(); // given number
146
147   return aMap.Extent(); // at least one?
148 }
149
150 //=======================================================================
151 // name    : SMESH_NumberFilter::getGeom
152 // Purpose : Retrieve geom object from SALOME_InteractiveObject
153 //=======================================================================
154 GEOM::GEOM_Object_ptr SMESH_NumberFilter::getGeom
155   (const SUIT_DataOwner* theDataOwner, const bool extractReference ) const
156 {
157   const LightApp_DataOwner* owner =
158     dynamic_cast<const LightApp_DataOwner*>(theDataOwner);
159   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>
160     (SUIT_Session::session()->activeApplication()->activeStudy());
161
162   GEOM::GEOM_Object_var anObj;
163
164   if (!owner || !appStudy)
165     return GEOM::GEOM_Object::_nil();
166
167   _PTR(Study) study = appStudy->studyDS();
168   QString entry = owner->entry();
169
170   _PTR(SObject) aSO( study->FindObjectID( entry.toUtf8().data() ) ), aRefSO;
171   if( extractReference && aSO && aSO->ReferencedObject( aRefSO ) )
172     aSO = aRefSO;
173
174   if (!aSO)
175     return GEOM::GEOM_Object::_nil();
176
177   CORBA::Object_var anObject = _CAST(SObject,aSO)->GetObject();
178   anObj = GEOM::GEOM_Object::_narrow(anObject);
179   if (!CORBA::is_nil(anObj))
180     return anObj._retn();
181
182   // Get geom object corresponding to the mesh
183   if ( myKind == "SMESH" ) {
184     _PTR(ChildIterator) anIter = study->NewChildIterator(aSO);
185     for (; anIter->More(); anIter->Next()) {
186       _PTR(SObject) aSO = anIter->Value();
187       if (!aSO)
188         continue;
189       _PTR(SObject) aRefSO;
190       _PTR(SObject) anObj;
191       if (aSO->ReferencedObject(aRefSO))
192         anObj = aRefSO;
193
194       if (!anObj)
195         anObj = aSO;
196
197       anObject = _CAST(SObject,anObj)->GetObject();
198       GEOM::GEOM_Object_var aMeshShape = GEOM::GEOM_Object::_narrow(anObject);
199
200       if (!aMeshShape->_is_nil())
201         return aMeshShape._retn();
202     }
203   }
204
205   return GEOM::GEOM_Object::_nil();
206 }
207
208 void SMESH_NumberFilter::SetSubShapeType (const TopAbs_ShapeEnum theSubShapeType)
209 {
210   mySubShapeType = theSubShapeType;
211 }
212
213 void SMESH_NumberFilter::SetNumber (const int theNumber)
214 {
215   myNumber = theNumber;
216 }
217
218 void SMESH_NumberFilter::SetClosedOnly (const bool theIsClosedOnly)
219 {
220   myIsClosedOnly = theIsClosedOnly;
221 }
222
223 void SMESH_NumberFilter::SetShapeType (const TopAbs_ShapeEnum theShapeType)
224 {
225   myShapeTypes.Add( theShapeType );
226 }
227
228 void SMESH_NumberFilter::SetMainShape (GEOM::GEOM_Object_ptr theMainObj)
229 {
230   myMainObj = GEOM::GEOM_Object::_duplicate(theMainObj);
231 }