1 // Copyright (C) 2007-2020 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 // File : SMESH_NumberFilter.cxx
26 #include "SMESH_NumberFilter.hxx"
28 #include <SALOME_LifeCycleCORBA.hxx>
30 #include "GEOM_Client.hxx"
32 #include "SUIT_Application.h"
33 #include "SUIT_Session.h"
35 #include "SalomeApp_Application.h"
36 #include "SalomeApp_Study.h"
37 #include "LightApp_DataOwner.h"
39 #include "SALOME_InteractiveObject.hxx"
40 #include "SALOMEDSClient_SObject.hxx"
41 #include "SALOMEDS_SObject.hxx"
43 #include <TopExp_Explorer.hxx>
44 #include <TopTools_IndexedMapOfShape.hxx>
48 * Class : SMESH_NumberFilter
49 * Description : Filter for geom objects.
50 * Filter geom objects by number of subshapes of the given type
53 //=======================================================================
54 // name : SMESH_NumberFilter::SMESH_NumberFilter
55 // Purpose : Constructor
56 //=======================================================================
57 SMESH_NumberFilter::SMESH_NumberFilter (const char* theKind,
58 const TopAbs_ShapeEnum theSubShapeType,
60 const TopAbs_ShapeEnum theShapeType,
61 GEOM::GEOM_Object_ptr theMainObj,
62 const bool theIsClosedOnly)
64 myKind = (char*)theKind;
65 mySubShapeType = theSubShapeType;
67 myIsClosedOnly = theIsClosedOnly;
68 myShapeTypes.Add(theShapeType);
69 myMainObj = GEOM::GEOM_Object::_duplicate(theMainObj);
72 //=======================================================================
73 // name : SMESH_NumberFilter::SMESH_NumberFilter
74 // Purpose : Constructor
75 //=======================================================================
76 SMESH_NumberFilter::SMESH_NumberFilter (const char* theKind,
77 const TopAbs_ShapeEnum theSubShapeType,
79 const TColStd_MapOfInteger& theShapeTypes,
80 GEOM::GEOM_Object_ptr theMainObj,
81 const bool theIsClosedOnly )
83 myKind = (char*)theKind;
84 mySubShapeType = theSubShapeType;
86 myIsClosedOnly = theIsClosedOnly;
87 myShapeTypes = theShapeTypes;
88 myMainObj = GEOM::GEOM_Object::_duplicate(theMainObj);
91 SMESH_NumberFilter::~SMESH_NumberFilter()
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
104 // Get geom object from IO
105 GEOM::GEOM_Object_var aGeomObj = getGeom(theDataOwner);
106 if (aGeomObj->_is_nil())
110 GEOM::GEOM_Gen_var geomEngine = aGeomObj->GetGen();
111 if ( CORBA::is_nil( geomEngine ))
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()))
120 if (myIsClosedOnly && aShape.ShapeType() == TopAbs_SHELL && !aShape.Closed())
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())
129 TopExp_Explorer anExp (aMainShape, aShape.ShapeType());
130 for (; anExp.More(); anExp.Next())
131 if (anExp.Current() == aShape)
137 // Verify number of sub-shapes
138 if (mySubShapeType == TopAbs_SHAPE);
141 TopTools_IndexedMapOfShape aMap;
142 TopExp::MapShapes(aShape, mySubShapeType, aMap);
145 return myNumber == aMap.Extent(); // given number
147 return aMap.Extent(); // at least one?
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
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());
162 GEOM::GEOM_Object_var anObj;
164 if (!owner || !appStudy)
165 return GEOM::GEOM_Object::_nil();
167 _PTR(Study) study = appStudy->studyDS();
168 QString entry = owner->entry();
170 _PTR(SObject) aSO( study->FindObjectID( entry.toUtf8().data() ) ), aRefSO;
171 if( extractReference && aSO && aSO->ReferencedObject( aRefSO ) )
175 return GEOM::GEOM_Object::_nil();
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();
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();
189 _PTR(SObject) aRefSO;
191 if (aSO->ReferencedObject(aRefSO))
197 anObject = _CAST(SObject,anObj)->GetObject();
198 GEOM::GEOM_Object_var aMeshShape = GEOM::GEOM_Object::_narrow(anObject);
200 if (!aMeshShape->_is_nil())
201 return aMeshShape._retn();
205 return GEOM::GEOM_Object::_nil();
208 void SMESH_NumberFilter::SetSubShapeType (const TopAbs_ShapeEnum theSubShapeType)
210 mySubShapeType = theSubShapeType;
213 void SMESH_NumberFilter::SetNumber (const int theNumber)
215 myNumber = theNumber;
218 void SMESH_NumberFilter::SetClosedOnly (const bool theIsClosedOnly)
220 myIsClosedOnly = theIsClosedOnly;
223 void SMESH_NumberFilter::SetShapeType (const TopAbs_ShapeEnum theShapeType)
225 myShapeTypes.Add( theShapeType );
228 void SMESH_NumberFilter::SetMainShape (GEOM::GEOM_Object_ptr theMainObj)
230 myMainObj = GEOM::GEOM_Object::_duplicate(theMainObj);