src/StdMeshers_I/StdMeshers_ObjRefUlils.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 SMESH : implementation 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     if ( ! theEntry.empty() ) {
  63       SALOMEDS::SObject_wrap sobj = SMESH_Gen_i::GetSMESHGen()->getStudyServant()->FindObjectID( theEntry.c_str() );
  64       CORBA::Object_var       obj = gen->SObjectToObject( sobj );
  65       geom = GEOM::GEOM_Object::_narrow( obj );
  66     }
  67   }
  68   // try by TopoDS_Shape
  69   if ( geom->_is_nil() )
  70     geom = ShapeToGeomObject( theShape );
  71
  72   return geom._retn();
  73 }
  74
  75 //================================================================================
  76   /*!
  77    * \brief Store the shape in the stream
  78     * \param theShape - shape to store
  79     * \param stream - the stream
  80    */
  81 //================================================================================
  82
  83 void StdMeshers_ObjRefUlils::SaveToStream( const TopoDS_Shape& theShape, ostream & stream)
  84 {
  85   bool ok = false;
  86   if ( !theShape.IsNull() ) {
  87     if (SMESH_Gen_i* gen = SMESH_Gen_i::GetSMESHGen()) {
  88       GEOM::GEOM_Object_var geom = gen->ShapeToGeomObject( theShape );
  89       if ( ! geom->_is_nil() ) {
  90         SALOMEDS::SObject_wrap sobj = gen->ObjectToSObject( geom );
  91         if ( !sobj->_is_nil() ) {
  92           CORBA::String_var entry = sobj->GetID();
  93           stream << " " << entry.in();
  94           ok = true;
  95         }
  96       }
  97     }
  98   }
  99   if ( ! ok )
 100     stream << " NULL_SHAPE ";
 101 }
 102
 103 //================================================================================
 104   /*!
 105    * \brief Retrieve a shape from the stream
 106     * \param stream - the stream
 107     * \retval TopoDS_Shape - resulting shape
 108    */
 109 //================================================================================
 110
 111 TopoDS_Shape StdMeshers_ObjRefUlils::LoadFromStream( istream &    stream,
 112                                                      std::string* entry)
 113 {
 114   string str;
 115   if (stream >> str) {
 116     if ( entry )
 117       * entry = str;
 118     if (SMESH_Gen_i* gen = SMESH_Gen_i::GetSMESHGen()) {
 119       SALOMEDS::SObject_wrap sobj = SMESH_Gen_i::GetSMESHGen()->getStudyServant()->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   if ( entry )
 126     entry->clear();
 127   return TopoDS_Shape();
 128 }
 129
 130 //================================================================================
 131   /*!
 132    * \brief Store the CORBA object in the stream
 133     * \param obj - object to store
 134     * \param stream - the stream
 135    */
 136 //================================================================================
 137
 138 void StdMeshers_ObjRefUlils::SaveToStream( CORBA::Object_ptr obj,
 139                                            std::ostream & stream)
 140 {
 141   bool ok = false;
 142   if ( !CORBA::is_nil( obj ) ) {
 143     if (SMESH_Gen_i* gen = SMESH_Gen_i::GetSMESHGen()) {
 144       stream << " " << gen->GetObjectId( obj );
 145       ok = true;
 146     }
 147   }
 148   if ( ! ok )
 149     stream << " NULL_OBJECT ";
 150 }
 151
 152 //=======================================================================
 153 //function : SaveToStream
 154 //purpose  : Store the study entry of object in the stream
 155 //=======================================================================
 156
 157 void StdMeshers_ObjRefUlils::SaveToStream( const std::string& studyEntry,
 158                                            std::ostream &     stream)
 159 {
 160   if ( studyEntry.find_first_not_of( ' ' ) == std::string::npos )
 161     stream << " NULL_OBJECT ";
 162   else
 163     stream << " " << studyEntry;
 164 }
 165
 166 //=======================================================================
 167 //function : EntryToShape
 168 //purpose  : Return TopoDS_Shape by a study entry
 169 //=======================================================================
 170
 171 TopoDS_Shape StdMeshers_ObjRefUlils::EntryToShape(const std::string theEntry)
 172 {
 173   TopoDS_Shape shape;
 174
 175   if (SMESH_Gen_i* gen = SMESH_Gen_i::GetSMESHGen()) {
 176     SALOMEDS::SObject_wrap sobj = SMESH_Gen_i::GetSMESHGen()->getStudyServant()->FindObjectID( theEntry.c_str() );
 177     CORBA::Object_var       obj = gen->SObjectToObject( sobj );
 178     GEOM::GEOM_Object_var  geom = GEOM::GEOM_Object::_narrow( obj );
 179     shape = gen->GeomObjectToShape( geom.in() );
 180   }
 181   return shape;
 182 }