]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESHGUI/SMESHGUI_GEOMGenUtils.cxx
Salome HOME
fix vector definition by 2 points
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_GEOMGenUtils.cxx
1 //  Copyright (C) 2003  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
2 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
3 //
4 //  This library is free software; you can redistribute it and/or
5 //  modify it under the terms of the GNU Lesser General Public
6 //  License as published by the Free Software Foundation; either
7 //  version 2.1 of the License.
8 //
9 //  This library is distributed in the hope that it will be useful,
10 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
11 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12 //  Lesser General Public License for more details.
13 //
14 //  You should have received a copy of the GNU Lesser General Public
15 //  License along with this library; if not, write to the Free Software
16 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
17 //
18 //  See http://www.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org
19
20 #include <boost/shared_ptr.hpp>
21
22 #include "SMESHGUI_GEOMGenUtils.h"
23 #include "SMESHGUI_Utils.h"
24
25 #include <GeometryGUI.h>
26
27 #include <SALOMEDSClient_SObject.hxx>
28 #include <SALOMEDSClient_ChildIterator.hxx>
29 #include <SALOMEDS_SObject.hxx>
30
31 #include CORBA_CLIENT_HEADER(SMESH_Mesh)
32
33
34 namespace SMESH {
35
36   GEOM::GEOM_Gen_var GetGEOMGen()
37   {
38     static GEOM::GEOM_Gen_var aGEOMGen;
39
40     if(CORBA::is_nil(aGEOMGen)) {
41       if ( GeometryGUI::GetGeomGen()->_is_nil() )
42         GeometryGUI::InitGeomGen();
43       aGEOMGen = GeometryGUI::GetGeomGen();
44     }
45     return aGEOMGen;
46   }
47
48   GEOM::GEOM_Object_var GetShapeOnMeshOrSubMesh(_PTR(SObject) theMeshOrSubmesh)
49   {
50     SALOMEDS_SObject* aMeshOrSubmesh = _CAST(SObject,theMeshOrSubmesh);
51     if(aMeshOrSubmesh) {
52       CORBA::Object_var Obj = aMeshOrSubmesh->GetObject();
53       if ( !CORBA::is_nil( Obj ) ) {
54         SMESH::SMESH_Mesh_var aMesh =
55           SObjectToInterface<SMESH::SMESH_Mesh>( theMeshOrSubmesh );
56         if ( !aMesh->_is_nil() )
57           return aMesh->GetShapeToMesh();
58         SMESH::SMESH_subMesh_var aSubmesh =
59           SObjectToInterface<SMESH::SMESH_subMesh>( theMeshOrSubmesh );
60         if ( !aSubmesh->_is_nil() )
61           return aSubmesh->GetSubShape();
62       }
63     }
64     return GEOM::GEOM_Object::_nil();
65   }
66
67   GEOM::GEOM_Object_ptr GetGeom (_PTR(SObject) theSO)
68   {
69     if (!theSO)
70       return GEOM::GEOM_Object::_nil();
71
72     _PTR(Study) aStudy = SMESH::GetActiveStudyDocument();
73     if (!aStudy)
74       return GEOM::GEOM_Object::_nil();
75
76     _PTR(ChildIterator) anIter (aStudy->NewChildIterator(theSO));
77     for (; anIter->More(); anIter->Next()) {
78       _PTR(SObject) aSObject = anIter->Value();
79       _PTR(SObject) aRefSOClient;
80       GEOM::GEOM_Object_var aMeshShape;
81
82       if (aSObject->ReferencedObject(aRefSOClient)) {
83         SALOMEDS_SObject* aRefSO = _CAST(SObject,aRefSOClient);
84         aMeshShape = GEOM::GEOM_Object::_narrow(aRefSO->GetObject());
85       } else {
86         SALOMEDS_SObject* aSO = _CAST(SObject,aSObject);
87         aMeshShape = GEOM::GEOM_Object::_narrow(aSO->GetObject());
88       }
89
90       if (!aMeshShape->_is_nil())
91         return aMeshShape._retn();
92     }
93     return GEOM::GEOM_Object::_nil();
94   }
95 }