src/GEOMClient/GEOM_Client.cxx

   1 //  Copyright (C) 2007-2010  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 //  GEOM GEOMClient : tool to transfer BREP files from GEOM server to GEOM client
  24 //  File   : GEOM_Client.cxx
  25 //  Author : Yves FRICAUD/Lucien PIGNOLONI
  26 //  Module : GEOM
  27 //  $Header$
  28 //
  29 #include <Standard_Stream.hxx>
  30
  31 #include <Standard_Stream.hxx>
  32
  33 #include <strstream>
  34
  35 #include "GEOM_Client.hxx"
  36 #include <SALOMEconfig.h>
  37 #include "Basics_Utils.hxx"
  38 #include "utilities.h"
  39
  40 #include <BRep_Builder.hxx>
  41 #include <BRepTools.hxx>
  42 #include <TopoDS_Shape.hxx>
  43 #include <TopoDS_Compound.hxx>
  44 #include <TCollection_AsciiString.hxx>
  45 #include <TopExp_Explorer.hxx>
  46 #include <TopExp.hxx>
  47 #include <TopAbs.hxx>
  48 #include <TopTools_IndexedMapOfShape.hxx>
  49
  50 #ifdef WNT
  51 #include <process.h>
  52 #else
  53 #include <unistd.h>
  54 #endif
  55
  56 #include CORBA_SERVER_HEADER(SALOMEDS)
  57 #include CORBA_SERVER_HEADER(GEOM_Gen)
  58
  59 #define HST_CLIENT_LEN 256
  60
  61
  62 //=======================================================================
  63 // function : Load()
  64 // purpose  :
  65 //=======================================================================
  66 TopoDS_Shape GEOM_Client::Load( GEOM::GEOM_Gen_ptr geom, GEOM::GEOM_Object_ptr aShape )
  67 {
  68   std::string hst_client = Kernel_Utils::GetHostname();
  69
  70   Engines::Container_var ctn_server = geom->GetContainerRef();
  71   long                   pid_server = ctn_server->getPID();
  72
  73   CORBA::String_var hostname = ctn_server->getHostName();
  74   if ( pid_client == pid_server && !strcmp(hst_client.c_str(), hostname.in()) ) {
  75     TopoDS_Shape* S = (TopoDS_Shape*)(aShape->getShape());
  76     return(*S);
  77   } else {
  78     /* get sequence of bytes of resulting brep shape from GEOM server */
  79     TopoDS_Shape S;
  80     SALOMEDS::TMPFile_var SeqFile = aShape->GetShapeStream();
  81     int sizebuf = SeqFile->length();
  82     char* buf;
  83     buf = (char*) &SeqFile[0];
  84     std::istrstream streamBrep(buf,sizebuf);
  85     BRep_Builder aBuilder;
  86     BRepTools::Read(S, streamBrep, aBuilder);
  87     return(S);
  88   }
  89 }
  90
  91 //=======================================================================
  92 // function : Create()
  93 // purpose  : Create in client not in a container
  94 //=======================================================================
  95 GEOM_Client::GEOM_Client()
  96 {
  97   pid_client =
  98 #ifdef WNT
  99     (long)_getpid();
 100 #else
 101     (long)getpid();
 102 #endif
 103 }
 104
 105 //=======================================================================
 106 // function : Create()
 107 // purpose  :
 108 //=======================================================================
 109 GEOM_Client::GEOM_Client(Engines::Container_ptr client)
 110 {
 111   pid_client = client->getPID();
 112 }
 113
 114 //=======================================================================
 115 // function : Find()
 116 // purpose  :
 117 //=======================================================================
 118 Standard_Integer GEOM_Client::Find( const TCollection_AsciiString& IOR, TopoDS_Shape& S )
 119 {
 120   for ( Standard_Integer i = 1; i<= myIORs.Length(); i++ ) {
 121     if (myIORs.Value(i).IsEqual(IOR)) {
 122       S = myShapes.Value(i);
 123       return i;
 124     }
 125   }
 126   return 0;
 127 }
 128
 129 //=======================================================================
 130 // function : Find()
 131 // purpose  :
 132 //=======================================================================
 133 Standard_Integer GEOM_Client::Find( const TopoDS_Shape& S, TCollection_AsciiString& IOR )
 134 {
 135   for ( Standard_Integer i = 1; i<= myShapes.Length(); i++ ) {
 136     if (myShapes.Value(i) == S) {
 137       IOR = myIORs.Value(i);
 138       return i;
 139     }
 140   }
 141   return 0;
 142 }
 143
 144 //=======================================================================
 145 // function : Bind()
 146 // purpose  :
 147 //=======================================================================
 148 void GEOM_Client::Bind( const TCollection_AsciiString& IOR, const TopoDS_Shape& S )
 149 {
 150   myIORs.Append(IOR);
 151   myShapes.Append(S);
 152 }
 153
 154 //=======================================================================
 155 // function : RemoveShapeFromBuffer()
 156 // purpose  : Remove shape from Client Buffer
 157 //=======================================================================
 158 void GEOM_Client::RemoveShapeFromBuffer( const TCollection_AsciiString& IOR)
 159 {
 160   if( myIORs.IsEmpty() )
 161     return;
 162
 163   TopoDS_Shape S;
 164   Standard_Integer anIndex = Find( IOR, S );
 165   if( anIndex != 0 ) {
 166     myIORs.Remove(anIndex);
 167     myShapes.Remove(anIndex);
 168   }
 169   return;
 170 }
 171
 172 //=======================================================================
 173 // function : ClearClientBuffer()
 174 // purpose  : purge buffer
 175 //=======================================================================
 176 void GEOM_Client::ClearClientBuffer()
 177 {
 178   if( myIORs.IsEmpty() )
 179     return;
 180   myIORs.Clear();
 181   myShapes.Clear();
 182   return;
 183 }
 184
 185 //=======================================================================
 186 // function : BufferLength()
 187 // purpose  :
 188 //=======================================================================
 189 unsigned int GEOM_Client::BufferLength()
 190 {
 191   return myIORs.Length();
 192 }
 193
 194 //=======================================================================
 195 // function : GetShape()
 196 // purpose  :
 197 //=======================================================================
 198 TopoDS_Shape GEOM_Client::GetShape( GEOM::GEOM_Gen_ptr geom, GEOM::GEOM_Object_ptr aShape )
 199 {
 200   TopoDS_Shape S;
 201   CORBA::String_var anIOR = geom->GetStringFromIOR(aShape);
 202   TCollection_AsciiString IOR = (char*)anIOR.in();
 203   Standard_Integer anIndex = Find(IOR, S);
 204
 205   if (anIndex != 0) return S;
 206
 207   /******* in case of a MAIN GEOM::SHAPE ********/
 208   if (aShape->IsMainShape()) {
 209     S = Load(geom, aShape);
 210     Bind(IOR, S);
 211     return S;
 212   }
 213
 214   /******* in case of SUB GEOM::SHAPE ***********/
 215   // Load and Explore the Main Shape
 216   TopoDS_Shape aMainShape = GetShape (geom, aShape->GetMainShape());
 217   GEOM::ListOfLong_var list = aShape->GetSubShapeIndices();
 218
 219   TopTools_IndexedMapOfShape anIndices;
 220   TopExp::MapShapes(aMainShape, anIndices);
 221
 222   /* Case of only one subshape */
 223   if (list->length() == 1 && list[0] > 0) {
 224     S = anIndices.FindKey(list[0]);
 225   }
 226   else {
 227     BRep_Builder B;
 228     TopoDS_Compound aCompound;
 229     B.MakeCompound(aCompound);
 230     for (int i = 0; i < list->length(); i++) {
 231       if (0 < list[i] && list[i] <= anIndices.Extent()) {
 232         TopoDS_Shape aSubShape = anIndices.FindKey(list[i]);
 233         B.Add(aCompound, aSubShape);
 234       }
 235     }
 236
 237     S = aCompound;
 238   }
 239   Bind(IOR, S);
 240   return S;
 241 }