]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESHGUI/SMESHGUI_GEOMGenUtils.cxx
Salome HOME
Adding tests for netgen_runner and parallel compute
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_GEOMGenUtils.cxx
1 // Copyright (C) 2007-2022  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 SMESHGUI : GUI for SMESH component
24 // File   : SMESHGUI_GEOMGenUtils.cxx
25 // Author : Open CASCADE S.A.S.
26 // SMESH includes
27 //
28 #include "SMESHGUI_GEOMGenUtils.h"
29 #include "SMESHGUI_Utils.h"
30 #include "SMESHGUI.h"
31
32 // SALOME GEOM includes
33 #include <GeometryGUI.h>
34 #include <GEOM_wrap.hxx>
35
36 // SALOME KERNEL includes
37 #include <SALOMEDS_SObject.hxx>
38
39 // IDL includes
40 #include <SALOMEconfig.h>
41 #include CORBA_CLIENT_HEADER(SMESH_Mesh)
42
43 #include <QString>
44
45 #include <TopExp.hxx>
46 #include <TopExp_Explorer.hxx>
47 #include <TopTools_IndexedMapOfShape.hxx>
48 #include <TopoDS_Iterator.hxx>
49
50 namespace SMESH
51 {
52   GEOM::GEOM_Gen_var GetGEOMGen( GEOM::GEOM_Object_ptr go )
53   {
54     GEOM::GEOM_Gen_ptr gen = GEOM::GEOM_Gen::_nil();
55     if ( !CORBA::is_nil( go ))
56       gen = go->GetGen();
57     return gen;
58   }
59
60   GEOM::GEOM_Object_var GetShapeOnMeshOrSubMesh(_PTR(SObject) theMeshOrSubmesh,
61                                                 bool*         isMesh)
62   {
63     SALOMEDS_SObject* aMeshOrSubmesh = _CAST(SObject,theMeshOrSubmesh);
64     if(aMeshOrSubmesh) {
65       CORBA::Object_var Obj = aMeshOrSubmesh->GetObject();
66       if ( !CORBA::is_nil( Obj ) ) {
67         SMESH::SMESH_Mesh_var aMesh =
68           SObjectToInterface<SMESH::SMESH_Mesh>( theMeshOrSubmesh );
69         if ( !aMesh->_is_nil() )
70         {
71           if ( isMesh ) *isMesh = true;
72           return aMesh->GetShapeToMesh();
73         }
74         SMESH::SMESH_subMesh_var aSubmesh =
75           SObjectToInterface<SMESH::SMESH_subMesh>( theMeshOrSubmesh );
76         if ( !aSubmesh->_is_nil() )
77         {
78           if ( isMesh ) *isMesh = false;
79           return aSubmesh->GetSubShape();
80         }
81       }
82     }
83     return GEOM::GEOM_Object::_nil();
84   }
85
86   GEOM::GEOM_Object_var GetGeom (_PTR(SObject) theSO)
87   {
88     GEOM::GEOM_Object_var aMeshShape;
89     if (!theSO)
90       return aMeshShape;
91
92     CORBA::Object_var obj = _CAST( SObject,theSO )->GetObject();
93     aMeshShape = GEOM::GEOM_Object::_narrow( obj );
94     if ( !aMeshShape->_is_nil() )
95       return aMeshShape;
96
97     _PTR(ChildIterator) anIter (SMESH::getStudy()->NewChildIterator(theSO));
98     for ( ; anIter->More(); anIter->Next()) {
99       _PTR(SObject) aSObject = anIter->Value();
100       _PTR(SObject) aRefSOClient;
101
102       if (aSObject->ReferencedObject(aRefSOClient)) {
103         SALOMEDS_SObject* aRefSO = _CAST(SObject,aRefSOClient);
104         aMeshShape = GEOM::GEOM_Object::_narrow(aRefSO->GetObject());
105       } else {
106         SALOMEDS_SObject* aSO = _CAST(SObject,aSObject);
107         aMeshShape = GEOM::GEOM_Object::_narrow(aSO->GetObject());
108       }
109       if ( !aMeshShape->_is_nil() )
110         return aMeshShape;
111     }
112     return aMeshShape;
113   }
114
115   GEOM::GEOM_Object_var GetGeom( Handle(SALOME_InteractiveObject) io )
116   {
117     GEOM::GEOM_Object_var go;
118     if ( !io.IsNull() && io->hasEntry() )
119     {
120       _PTR(SObject) so = SMESH::getStudy()->FindObjectID( io->getEntry() );
121       go = GetGeom( so );
122     }
123     return go._retn();
124   }
125
126   SMESHGUI_EXPORT char* GetGeomName( _PTR(SObject) smeshSO )
127   {
128     if (!smeshSO)
129       return 0;
130
131     _PTR(ChildIterator) anIter (SMESH::getStudy()->NewChildIterator( smeshSO ));
132     for ( ; anIter->More(); anIter->Next()) {
133       _PTR(SObject) aSObject = anIter->Value();
134       _PTR(SObject) aRefSOClient;
135       GEOM::GEOM_Object_var aMeshShape;
136
137       if (aSObject->ReferencedObject(aRefSOClient)) {
138         SALOMEDS_SObject* aRefSO = _CAST(SObject,aRefSOClient);
139         aMeshShape = GEOM::GEOM_Object::_narrow(aRefSO->GetObject());
140         aSObject = aRefSOClient;
141       }
142       else {
143         SALOMEDS_SObject* aSO = _CAST(SObject,aSObject);
144         aMeshShape = GEOM::GEOM_Object::_narrow(aSO->GetObject());
145       }
146
147       if (!aMeshShape->_is_nil())
148       {
149         std::string name = aSObject->GetName();
150         return CORBA::string_dup( name.c_str() );
151       }
152     }
153     return 0;
154   }
155
156   GEOM::GEOM_Object_ptr GetSubShape (GEOM::GEOM_Object_ptr theMainShape,
157                                      long                  theID)
158   {
159     if ( CORBA::is_nil( theMainShape ))
160       return GEOM::GEOM_Object::_nil();
161
162     GEOM::GEOM_Gen_var geomGen = SMESH::GetGEOMGen( theMainShape );
163     if (geomGen->_is_nil())
164       return GEOM::GEOM_Object::_nil();
165
166     GEOM::GEOM_IShapesOperations_wrap aShapesOp = geomGen->GetIShapesOperations();
167     if (aShapesOp->_is_nil())
168       return GEOM::GEOM_Object::_nil();
169
170     GEOM::GEOM_Object_wrap subShape = aShapesOp->GetSubShape( theMainShape, theID );
171     return subShape._retn();
172   }
173
174   //================================================================================
175   /*!
176    * \brief Return entries of sub-mesh geometry and mesh geometry by an IO of assigned
177    *        hypothesis
178    *  \param [in] hypIO - IO of hyp which is a reference SO to a hyp SO
179    *  \param [out] subGeom - found entry of a sub-mesh if any
180    *  \param [out] meshGeom - found entry of a mesh
181    *  \return bool - \c true if any geometry has been found
182    */
183   //================================================================================
184
185   bool GetGeomEntries( Handle(SALOME_InteractiveObject)& hypIO,
186                        QString&                          subGeom,
187                        QString&                          meshGeom )
188   {
189     subGeom.clear();
190     meshGeom.clear();
191     if ( hypIO.IsNull() ) return false;
192
193     _PTR(SObject) hypSO = SMESH::getStudy()->FindObjectID( hypIO->getEntry() );
194     if ( !hypSO ) return false;
195
196     // Depth() is a number of fathers
197     if ( hypSO->Depth() == 4 ) // hypSO is not a reference to a hyp but a hyp it-self
198     {
199       SMESH::SMESH_Hypothesis_var hyp =
200         SMESH::SObjectToInterface< SMESH::SMESH_Hypothesis >( hypSO );
201       SMESH::SMESH_Mesh_var mesh;
202       GEOM::GEOM_Object_var geom;
203       SMESH::SMESH_Gen_var  gen = SMESHGUI::GetSMESHGUI()->GetSMESHGen();
204       if ( !gen || !gen->GetSoleSubMeshUsingHyp( hyp, mesh.out(), geom.out() ))
205         return false;
206
207       subGeom = toQStr( geom->GetStudyEntry() );
208
209       geom  = mesh->GetShapeToMesh();
210       if ( geom->_is_nil() )
211         return false;
212       meshGeom = toQStr( geom->GetStudyEntry() );
213     }
214     else
215     {
216       _PTR(SObject) appliedSO = hypSO->GetFather(); // "Applied hypotheses" folder
217       if ( !appliedSO ) return false;
218
219       _PTR(SObject) subOrMeshSO = appliedSO->GetFather(); // mesh or sub-mesh SO
220       if ( !subOrMeshSO ) return false;
221
222       bool isMesh;
223       GEOM::GEOM_Object_var geom = GetShapeOnMeshOrSubMesh( subOrMeshSO, &isMesh );
224       if ( geom->_is_nil() )
225         return false;
226
227       if ( isMesh )
228       {
229         meshGeom = toQStr( geom->GetStudyEntry() );
230         return !meshGeom.isEmpty();
231       }
232
233       subGeom = toQStr( geom->GetStudyEntry() );
234
235       _PTR(SObject) subFolderSO = subOrMeshSO->GetFather(); // "SubMeshes on ..." folder
236       if ( !subFolderSO ) return false;
237
238       _PTR(SObject) meshSO = subFolderSO->GetFather(); // mesh SO
239       if ( !meshSO ) return false;
240
241       geom = GetShapeOnMeshOrSubMesh( meshSO );
242       if ( geom->_is_nil() )
243         return false;
244
245       meshGeom = toQStr( geom->GetStudyEntry() );
246     }
247
248     return !meshGeom.isEmpty() && !subGeom.isEmpty();
249   }
250
251
252   //================================================================================
253   /*!
254    * \brief Return type of shape contained in a group
255    */
256   //================================================================================
257
258   TopAbs_ShapeEnum _getGroupType(const TopoDS_Shape& group)
259   {
260     if ( group.ShapeType() != TopAbs_COMPOUND )
261       return group.ShapeType();
262
263     // iterate on a compound
264     TopoDS_Iterator it( group );
265     if ( it.More() )
266       return _getGroupType( it.Value() );
267
268     return TopAbs_SHAPE;
269   }
270
271
272   //================================================================================
273   /*!
274    * \brief Check if a subGeom contains sub-shapes of a mainGeom
275    */
276   //================================================================================
277
278   bool ContainsSubShape( GEOM::GEOM_Object_ptr mainGeom,
279                          GEOM::GEOM_Object_ptr subGeom, bool allowMainShape )
280   {
281     if ( CORBA::is_nil( mainGeom ) ||
282          CORBA::is_nil( subGeom ))
283       return false;
284
285     if (allowMainShape && mainGeom->IsSame(subGeom))
286       return true;
287
288     GEOM::GEOM_Gen_var geomGen = mainGeom->GetGen();
289     if ( geomGen->_is_nil() ) return false;
290
291     GEOM::GEOM_IGroupOperations_wrap op = geomGen->GetIGroupOperations();
292     if ( op->_is_nil() ) return false;
293
294     GEOM::GEOM_Object_var mainObj = op->GetMainShape( subGeom ); /* _var not _wrap as
295                                                                     mainObj already exists! */
296     while ( !mainObj->_is_nil() )
297     {
298       CORBA::String_var entry1 = mainObj->GetEntry();
299       CORBA::String_var entry2 = mainGeom->GetEntry();
300       if ( std::string( entry1.in() ) == entry2.in() )
301         return true;
302       mainObj = op->GetMainShape( mainObj );
303     }
304     if ( subGeom->GetShapeType() == GEOM::COMPOUND )
305     {
306       // is subGeom a compound of sub-shapes?
307       GEOM::GEOM_IShapesOperations_wrap sop = geomGen->GetIShapesOperations();
308       if ( sop->_is_nil() ) return false;
309       GEOM::ListOfLong_var ids = sop->GetAllSubShapesIDs( subGeom,
310                                                           GEOM::SHAPE,/*sorted=*/false);
311       if ( ids->length() > 0 )
312       {
313         GEOM_Client geomClient;
314         TopoDS_Shape  subShape = geomClient.GetShape( geomGen, subGeom );
315         TopoDS_Shape mainShape = geomClient.GetShape( geomGen, mainGeom );
316         if ( subShape.IsNull() || mainShape.IsNull() )
317           return false;
318
319         TopAbs_ShapeEnum subType = _getGroupType( subShape );
320         TopTools_IndexedMapOfShape subMap;
321         TopExp::MapShapes( subShape, subType, subMap );
322         for ( TopExp_Explorer exp( mainShape, subType ); exp.More(); exp.Next() )
323           if ( subMap.Contains( exp.Current() ))
324             return true;
325
326       }
327     }
328     return false;
329   }
330
331 } // end of namespace SMESH