src/StdMeshers_I/StdMeshers_ObjRefUlils.cxx

   1 // Copyright (C) 2007-2013  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.
  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 SMESH : implementaion of SMESH idl descriptions
  24 // File      : StdMeshers_ObjRefUlils.cxx
  25 // Created   : Wed Oct 18 15:38:22 2006
  26 // Author    : Edward AGAPOV (eap)
  27 //
  28 #include "StdMeshers_ObjRefUlils.hxx"
  29
  30 #include <SALOMEDS_wrap.hxx>
  31 #include <TopoDS_Shape.hxx>
  32
  33 using namespace std;
  34
  35 //=======================================================================
  36 //function : GeomObjectToEntry
  37 //purpose  : Return study entry of GEOM Object
  38 //=======================================================================
  39
  40 std::string StdMeshers_ObjRefUlils::GeomObjectToEntry(GEOM::GEOM_Object_ptr&  theGeomObject)
  41 {
  42   if ( CORBA::is_nil( theGeomObject ))
  43     return "NULL_OBJECT";
  44
  45   CORBA::String_var entry = theGeomObject->GetStudyEntry();
  46   return entry.in();
  47 }
  48
  49 //=======================================================================
  50 //function : EntryOrShapeToGeomObject
  51 //purpose  :  Return GEOM Object by its sytudy entry or TopoDS_Shape
  52 //=======================================================================
  53
  54 GEOM::GEOM_Object_ptr
  55 StdMeshers_ObjRefUlils::EntryOrShapeToGeomObject (const std::string&  theEntry,
  56                                                   const TopoDS_Shape& theShape)
  57 {
  58   GEOM::GEOM_Object_var geom = GEOM::GEOM_Object::_nil();
  59
  60   // try by entry
  61   if (SMESH_Gen_i* gen = SMESH_Gen_i::GetSMESHGen()) {
  62     SALOMEDS::Study_var study = gen->GetCurrentStudy();
  63     if ( ! theEntry.empty() && ! study->_is_nil() ) {
  64       SALOMEDS::SObject_wrap sobj = study->FindObjectID( theEntry.c_str() );
  65       CORBA::Object_var       obj = gen->SObjectToObject( sobj );
  66       geom = GEOM::GEOM_Object::_narrow( obj );
  67     }
  68   }
  69   // try by TopoDS_Shape
  70   if ( geom->_is_nil() )
  71     geom = ShapeToGeomObject( theShape );
  72
  73   return geom._retn();
  74 }
  75
  76 //================================================================================
  77   /*!
  78    * \brief Store the shape in the stream
  79     * \param theShape - shape to store
  80     * \param stream - the stream
  81    */
  82 //================================================================================
  83
  84 void StdMeshers_ObjRefUlils::SaveToStream( const TopoDS_Shape& theShape, ostream & stream)
  85 {
  86   bool ok = false;
  87   if ( !theShape.IsNull() ) {
  88     if (SMESH_Gen_i* gen = SMESH_Gen_i::GetSMESHGen()) {
  89       GEOM::GEOM_Object_var geom = gen->ShapeToGeomObject( theShape );
  90       if ( ! geom->_is_nil() ) {
  91         SALOMEDS::SObject_wrap sobj = gen->ObjectToSObject( gen->GetCurrentStudy(), geom );
  92         if ( !sobj->_is_nil() ) {
  93           CORBA::String_var entry = sobj->GetID();
  94           stream << " " << entry.in();
  95           ok = true;
  96         }
  97       }
  98     }
  99   }
 100   if ( ! ok )
 101     stream << " NULL_SHAPE ";
 102 }
 103
 104 //================================================================================
 105   /*!
 106    * \brief Retrieve a shape from the stream
 107     * \param stream - the stream
 108     * \retval TopoDS_Shape - resulting shape
 109    */
 110 //================================================================================
 111
 112 TopoDS_Shape StdMeshers_ObjRefUlils::LoadFromStream( istream & stream)
 113 {
 114   if (SMESH_Gen_i* gen = SMESH_Gen_i::GetSMESHGen()) {
 115     SALOMEDS::Study_var study = gen->GetCurrentStudy();
 116     if ( ! study->_is_nil() ) {
 117       string str;
 118       if (stream >> str) {
 119         SALOMEDS::SObject_wrap sobj = study->FindObjectID( str.c_str() );
 120         CORBA::Object_var       obj = gen->SObjectToObject( sobj );
 121         GEOM::GEOM_Object_var  geom = GEOM::GEOM_Object::_narrow( obj );
 122         return gen->GeomObjectToShape( geom.in() );
 123       }
 124     }
 125   }
 126   return TopoDS_Shape();
 127 }
 128
 129 //================================================================================
 130   /*!
 131    * \brief Store the CORBA object in the stream
 132     * \param obj - object to store
 133     * \param stream - the stream
 134    */
 135 //================================================================================
 136
 137 void StdMeshers_ObjRefUlils::SaveToStream( CORBA::Object_ptr obj,
 138                                            std::ostream & stream)
 139 {
 140   bool ok = false;
 141   if ( !CORBA::is_nil( obj ) ) {
 142     if (SMESH_Gen_i* gen = SMESH_Gen_i::GetSMESHGen()) {
 143       stream << " " << gen->GetObjectId( obj );
 144       ok = true;
 145     }
 146   }
 147   if ( ! ok )
 148     stream << " NULL_OBJECT ";
 149 }
 150
 151 //=======================================================================
 152 //function : SaveToStream
 153 //purpose  : Store the study entry of object in the stream
 154 //=======================================================================
 155
 156 void StdMeshers_ObjRefUlils::SaveToStream( const std::string& studyEntry,
 157                                            std::ostream &     stream)
 158 {
 159   if ( studyEntry.find_first_not_of( ' ' ) == std::string::npos )
 160     stream << " NULL_OBJECT ";
 161   else
 162     stream << " " << studyEntry;
 163 }