Salome HOME
Merge from V5_1_main branch 24/11/2010
[modules/geom.git] / src / GEOMClient / GEOM_Client.cxx
index ac8d7acf8db9d0a9bab575f7682832599d10a10b..0e07e69d3bbd9a8ae93aeff9206695e316728869 100644 (file)
-using namespace std;
-//  File      : GEOM_Client.cxx
-//  Created   : 
-//  Author    : Yves FRICAUD/Lucien PIGNOLONI
-//  Project   : SALOME
-//  Module    : GEOM
-//  Copyright : Open CASCADE
+//  Copyright (C) 2007-2010  CEA/DEN, EDF R&D, OPEN CASCADE
+//
+//  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
+//  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
+//
+//  This library is free software; you can redistribute it and/or
+//  modify it under the terms of the GNU Lesser General Public
+//  License as published by the Free Software Foundation; either
+//  version 2.1 of the License.
+//
+//  This library is distributed in the hope that it will be useful,
+//  but WITHOUT ANY WARRANTY; without even the implied warranty of
+//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+//  Lesser General Public License for more details.
+//
+//  You should have received a copy of the GNU Lesser General Public
+//  License along with this library; if not, write to the Free Software
+//  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//
+//  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+//
+
+//  GEOM GEOMClient : tool to transfer BREP files from GEOM server to GEOM client
+//  File   : GEOM_Client.cxx
+//  Author : Yves FRICAUD/Lucien PIGNOLONI
+//  Module : GEOM
 //  $Header$
+//
+#include <Standard_Stream.hxx>
+
+#include <Standard_Stream.hxx>
+
+#include <sstream>
 
 #include "GEOM_Client.hxx"
 #include <SALOMEconfig.h>
+#include "Basics_Utils.hxx"
 #include "utilities.h"
 
-#include CORBA_SERVER_HEADER(GEOM_Gen)
-
 #include <BRep_Builder.hxx>
 #include <BRepTools.hxx>
 #include <TopoDS_Shape.hxx>
 #include <TopoDS_Compound.hxx>
 #include <TCollection_AsciiString.hxx>
 #include <TopExp_Explorer.hxx>
+#include <TopExp.hxx>
 #include <TopAbs.hxx>
-#include <TColStd_MapOfInteger.hxx>
-#include <TopoDS_Iterator.hxx>
-#include <TopTools_MapOfShape.hxx>
-#include <TopTools_ListIteratorOfListOfShape.hxx>
-#include <TopTools_ListOfShape.hxx>
+#include <TopTools_IndexedMapOfShape.hxx>
+
+#ifdef WNT
+#include <process.h>
+#else
+#include <unistd.h>
+#endif
+
+#include CORBA_SERVER_HEADER(SALOMEDS)
+#include CORBA_SERVER_HEADER(GEOM_Gen)
 
+#define HST_CLIENT_LEN 256
 
 
 //=======================================================================
 // function : Load()
-// purpose  : 
+// purpose  :
 //=======================================================================
-static TopoDS_Shape Load( GEOM::GEOM_Gen_ptr geom, GEOM::GEOM_Shape_ptr aShape )
+TopoDS_Shape GEOM_Client::Load( GEOM::GEOM_Gen_ptr geom, GEOM::GEOM_Object_ptr aShape )
 {
-  TopoDS_Shape S;
-  /* get sequence of bytes of resulting brep shape from GEOM server */
-  GEOM::GEOM_Shape::TMPFile_var SeqFile = aShape->GetShapeStream();
-  int sizebuf = SeqFile->length();
-  char* buf;
-  buf = (char*) &SeqFile[0];
-  istrstream streamBrep(buf,sizebuf);
-  BRep_Builder aBuilder;
-  BRepTools::Read(S, streamBrep, aBuilder);
-  return S;
+  std::string hst_client = Kernel_Utils::GetHostname();
+
+  Engines::Container_var ctn_server = geom->GetContainerRef();
+  long                   pid_server = ctn_server->getPID();
+
+  CORBA::String_var hostname = ctn_server->getHostName();
+  if ( pid_client == pid_server && !strcmp(hst_client.c_str(), hostname.in()) ) {
+    TopoDS_Shape* S = (TopoDS_Shape*)(aShape->getShape());
+    return(*S);
+  } else {
+    /* get sequence of bytes of resulting brep shape from GEOM server */
+    TopoDS_Shape S;
+    SALOMEDS::TMPFile_var SeqFile = aShape->GetShapeStream();
+    int sizebuf = SeqFile->length();
+    char* buf;
+    buf = (char*) &SeqFile[0];
+    std::istringstream streamBrep(buf);
+    BRep_Builder aBuilder;
+    BRepTools::Read(S, streamBrep, aBuilder);
+    return(S);
+  }
 }
 
-
 //=======================================================================
 // function : Create()
-// purpose  : 
+// purpose  : Create in client not in a container
 //=======================================================================
 GEOM_Client::GEOM_Client()
 {
+  pid_client =
+#ifdef WNT
+    (long)_getpid();
+#else
+    (long)getpid();
+#endif
 }
 
+//=======================================================================
+// function : Create()
+// purpose  :
+//=======================================================================
+GEOM_Client::GEOM_Client(Engines::Container_ptr client)
+{
+  pid_client = client->getPID();
+}
 
 //=======================================================================
 // function : Find()
-// purpose  : 
+// purpose  :
 //=======================================================================
 Standard_Integer GEOM_Client::Find( const TCollection_AsciiString& IOR, TopoDS_Shape& S )
 {
@@ -71,6 +126,20 @@ Standard_Integer GEOM_Client::Find( const TCollection_AsciiString& IOR, TopoDS_S
   return 0;
 }
 
+//=======================================================================
+// function : Find()
+// purpose  :
+//=======================================================================
+Standard_Integer GEOM_Client::Find( const TopoDS_Shape& S, TCollection_AsciiString& IOR )
+{
+  for ( Standard_Integer i = 1; i<= myShapes.Length(); i++ ) {
+    if (myShapes.Value(i) == S) {
+      IOR = myIORs.Value(i);
+      return i;
+    }
+  }
+  return 0;
+}
 
 //=======================================================================
 // function : Bind()
@@ -86,21 +155,20 @@ void GEOM_Client::Bind( const TCollection_AsciiString& IOR, const TopoDS_Shape&
 // function : RemoveShapeFromBuffer()
 // purpose  : Remove shape from Client Buffer
 //=======================================================================
-void GEOM_Client::RemoveShapeFromBuffer( const TCollection_AsciiString& shapeIOR )
+void GEOM_Client::RemoveShapeFromBuffer( const TCollection_AsciiString& IOR)
 {
   if( myIORs.IsEmpty() )
-    return ;
-  
-  TopoDS_Shape S ;
-  Standard_Integer anIndex = Find( shapeIOR, S ) ;
+    return;
+
+  TopoDS_Shape S;
+  Standard_Integer anIndex = Find( IOR, S );
   if( anIndex != 0 ) {
-    myIORs.Remove(anIndex) ;
-    myShapes.Remove(anIndex) ;
+    myIORs.Remove(anIndex);
+    myShapes.Remove(anIndex);
   }
-  return ;
+  return;
 }
 
-
 //=======================================================================
 // function : ClearClientBuffer()
 // purpose  : purge buffer
@@ -108,10 +176,10 @@ void GEOM_Client::RemoveShapeFromBuffer( const TCollection_AsciiString& shapeIOR
 void GEOM_Client::ClearClientBuffer()
 {
   if( myIORs.IsEmpty() )
-    return ;
-  myIORs.Clear() ;
-  myShapes.Clear() ;
-  return ;
+    return;
+  myIORs.Clear();
+  myShapes.Clear();
+  return;
 }
 
 //=======================================================================
@@ -120,114 +188,54 @@ void GEOM_Client::ClearClientBuffer()
 //=======================================================================
 unsigned int GEOM_Client::BufferLength()
 {
-  return myIORs.Length() ;
+  return myIORs.Length();
 }
 
-
 //=======================================================================
 // function : GetShape()
-// purpose  : 
-//=======================================================================
-
-TopoDS_Shape GEOM_Client::GetShape( GEOM::GEOM_Gen_ptr geom, GEOM::GEOM_Shape_ptr aShape ) 
-{ 
-
-  TopoDS_Shape            S; 
-  TCollection_AsciiString IOR(aShape->Name()); 
-  Standard_Integer        anIndex = Find(IOR, S); 
-  
-  BRep_Builder B; 
-
-  if (anIndex !=0 ) { 
-    return S ; 
-   } 
-
-  /******* in case of a MAIN GEOM::SHAPE ********/ 
-  if (aShape->IsMainShape()) { 
-    S = Load(geom, aShape); 
-    Bind(IOR,S); 
-    return S; 
-  } 
-
-  /******* in case of SUB GEOM::SHAPE ***********/ 
-  // Load and Explore the Main Shape 
-  TopoDS_Shape MainShape = GetShape (geom, geom->GetIORFromString(aShape->MainName())); 
-  GEOM::GEOM_Shape::ListOfSubShapeID_var list = aShape->Index(); 
-
-  Standard_Integer j = 1; 
-  TopExp_Explorer exp; 
-  TopAbs_ShapeEnum ShapeType = TopAbs_ShapeEnum(aShape->ShapeType()); 
-  
-  /* Case of only one subshape */ 
-  if (list->length() == 1) 
-  { 
-    if (ShapeType == TopAbs_COMPOUND) 
-      { 
-       TopoDS_Iterator it; 
-       TopTools_ListOfShape CL; 
-       CL.Append( MainShape ); 
-       TopTools_ListIteratorOfListOfShape itC; 
-       for (itC.Initialize( CL ); itC.More(); itC.Next()) 
-         { 
-           for (it.Initialize( itC.Value() );  it.More(); it.Next()) 
-             { 
-               if ( it.Value().ShapeType() == TopAbs_COMPOUND) 
-                 {
-                   if (j == list[0]) 
-                     { 
-                       S = it.Value(); 
-                       Bind(IOR, S); 
-                       return S; 
-                     } 
-                   j++; 
-                   CL.Append( it.Value() ); 
-                 }
-             } 
-         } 
-      } 
-    else 
-      { 
-       TopTools_MapOfShape M; 
-       for (exp.Init(MainShape, ShapeType); exp.More(); exp.Next()) { 
-         if ( M.Add(exp.Current()) ) 
-           { 
-             if (j == list[0]) 
-               { 
-                 S = exp.Current(); 
-                 Bind(IOR, S); 
-                 return S; 
-               } 
-             j++; 
-           } 
-       } 
-      } 
-  } 
-  
-  /* Case of a compound containing two or more sub shapes (not a main shape compound !)   */ 
-
-  /* Warning : the compound when representing sub shapes must be explored in a sub type   */ 
-  /* that is NOT ShapeType=aShape->ShapeType()= TopAbs_COMPOUND !                         */ 
-  /* We have to retrieve the exact sub type of shapes contained in the compound first !   */ 
-  TopoDS_Iterator it ; 
-  TopAbs_ShapeEnum exactSubType ; 
-  S = Load( geom, aShape ); 
-  it.Initialize( S, true, true ) ; 
-  it.More(); 
-  exactSubType = it.Value().ShapeType() ; 
-
-  TColStd_MapOfInteger MapIndex; 
-  Standard_Integer nbSS = list->length(); 
-  TopoDS_Compound Comp; 
-  B.MakeCompound(Comp); 
-  
-  for (Standard_Integer i=1; i<=nbSS; i++) 
-    MapIndex.Add(list[i-1]); 
-  
-  for (exp.Init(MainShape, exactSubType), j=1; exp.More() ; exp.Next(), j++) { 
-    if ( MapIndex.Contains(j) ) { 
-      B.Add( Comp, exp.Current() ); 
-    } 
-  } 
-  Bind(IOR, Comp); 
-  return Comp; 
-} 
+// purpose  :
+//=======================================================================
+TopoDS_Shape GEOM_Client::GetShape( GEOM::GEOM_Gen_ptr geom, GEOM::GEOM_Object_ptr aShape )
+{
+  TopoDS_Shape S;
+  CORBA::String_var anIOR = geom->GetStringFromIOR(aShape);
+  TCollection_AsciiString IOR = (char*)anIOR.in();
+  Standard_Integer anIndex = Find(IOR, S);
+
+  if (anIndex != 0) return S;
+
+  /******* in case of a MAIN GEOM::SHAPE ********/
+  if (aShape->IsMainShape()) {
+    S = Load(geom, aShape);
+    Bind(IOR, S);
+    return S;
+  }
+
+  /******* in case of SUB GEOM::SHAPE ***********/
+  // Load and Explore the Main Shape
+  TopoDS_Shape aMainShape = GetShape (geom, aShape->GetMainShape());
+  GEOM::ListOfLong_var list = aShape->GetSubShapeIndices();
+
+  TopTools_IndexedMapOfShape anIndices;
+  TopExp::MapShapes(aMainShape, anIndices);
+
+  /* Case of only one subshape */
+  if (list->length() == 1 && list[0] > 0) {
+    S = anIndices.FindKey(list[0]);
+  }
+  else {
+    BRep_Builder B;
+    TopoDS_Compound aCompound;
+    B.MakeCompound(aCompound);
+    for (int i = 0; i < list->length(); i++) {
+      if (0 < list[i] && list[i] <= anIndices.Extent()) {
+        TopoDS_Shape aSubShape = anIndices.FindKey(list[i]);
+        B.Add(aCompound, aSubShape);
+      }
+    }
+
+    S = aCompound;
+  }
+  Bind(IOR, S);
+  return S;
+}