Salome HOME
bos #29332 [CEA 26899] GroupOnGeom returns None
[modules/geom.git] / src / GEOMClient / GEOM_Client.cxx
index b5ee8f58e84bbedff4fc74dbba4e00caf7c6ee89..871cd861a93274e97ebe3191cd97bef9ce16bba1 100644 (file)
-//  GEOM GEOMClient : tool to transfer BREP files from GEOM server to GEOM client
+// Copyright (C) 2007-2021  CEA/DEN, EDF R&D, OPEN CASCADE
 //
-//  Copyright (C) 2003  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.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org 
+// 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, or (at your option) any later version.
 //
+// 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
 //
 //  File   : GEOM_Client.cxx
 //  Author : Yves FRICAUD/Lucien PIGNOLONI
 //  Module : GEOM
-//  $Header$
 
-using namespace std;
+#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 WIN32
+#include <process.h>
+#else
 #include <unistd.h>
+#endif
 
-#define HST_CLIENT_LEN 256
+#include CORBA_SERVER_HEADER(SALOMEDS)
+#include CORBA_SERVER_HEADER(GEOM_Gen)
 
+#define HST_CLIENT_LEN 256
 
 //=======================================================================
 // function : Load()
-// purpose  : 
+// purpose  :
 //=======================================================================
-TopoDS_Shape GEOM_Client::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 )
 {
-    char hst_client[HST_CLIENT_LEN];
-    gethostname(hst_client, HST_CLIENT_LEN);
+  std::string hst_client = Kernel_Utils::GetHostname();
 
-    Engines::Container_var ctn_server = geom->GetContainerRef();
-    long                   pid_server = ctn_server->getPID();
+  Engines::Container_var ctn_server = geom->GetContainerRef();
+  long                   pid_server = ctn_server->getPID();
 
-    if ( (pid_client==pid_server) && (strcmp(hst_client, ctn_server->getHostName())==0) ) {
-        TopoDS_Shape* S = (TopoDS_Shape*)(aShape->getShape());
-        return(*S);
-    } else {
-        /* get sequence of bytes of resulting brep shape from GEOM server */
-        TopoDS_Shape S;
-        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);
-    };
+  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  : Create in client not in a container
 //=======================================================================
 GEOM_Client::GEOM_Client()
 {
-  pid_client = (-1);
+  pid_client =
+#ifdef WIN32
+    (long)_getpid();
+#else
+    (long)getpid();
+#endif
 }
 
 //=======================================================================
 // function : Create()
-// purpose  : 
+// purpose  : Copy constructor
+//=======================================================================
+GEOM_Client::GEOM_Client(const GEOM_Client& client)
+{
+  _mySubShapes = client._mySubShapes;
+  myShapesMap = client.myShapesMap;
+  pid_client = client.pid_client;
+}
+
+//=======================================================================
+// function : Create()
+// purpose  :
 //=======================================================================
 GEOM_Client::GEOM_Client(Engines::Container_ptr client)
 {
   pid_client = client->getPID();
 }
 
+//=======================================================================
+// function : get_client()
+// purpose  : Static method to have the only one instance of GEOM_Client
+//=======================================================================
+#ifdef SINGLE_CLIENT
+GEOM_Client& GEOM_Client::get_client()
+{
+  static GEOM_Client a;
+  return a;
+}
+#else
+GEOM_Client GEOM_Client::get_client()
+{
+  return GEOM_Client();
+}
+#endif
+
 //=======================================================================
 // function : Find()
-// purpose  : 
+// purpose  :
 //=======================================================================
-Standard_Integer GEOM_Client::Find( const TCollection_AsciiString& IOR, TopoDS_Shape& S )
+Standard_Boolean GEOM_Client::Find (const TCollection_AsciiString& IOR, TopoDS_Shape& S)
 {
-  for ( Standard_Integer i = 1; i<= myIORs.Length(); i++ ) {
-    if (myIORs.Value(i).IsEqual(IOR)) {
-      S = myShapes.Value(i);
-      return i;
-    }
+  std::map< TCollection_AsciiString , TopoDS_Shape >::iterator i2s = myShapesMap.find( IOR );
+  if ( i2s != myShapesMap.end() ) {
+    S = i2s->second;
+    return Standard_True;
   }
-  return 0;
+  return Standard_False;
 }
 
+//=======================================================================
+// function : Find()
+// purpose  :
+//=======================================================================
+Standard_Boolean GEOM_Client::Find (const TopoDS_Shape& S, TCollection_AsciiString& IOR)
+{
+  std::map< TCollection_AsciiString, TopoDS_Shape >::const_iterator it;
+  for (it = myShapesMap.begin(); it != myShapesMap.end(); ++it) {
+    if ((*it).second == S) {
+      IOR = (*it).first;
+      return Standard_True;
+    }
+  }
+  return Standard_False;
+}
 
 //=======================================================================
 // function : Bind()
 // purpose  :
 //=======================================================================
-void GEOM_Client::Bind( const TCollection_AsciiString& IOR, const TopoDS_Shape& S )
+void GEOM_Client::Bind( const TCollection_AsciiString& IOR, const TopoDS_Shape& S, int Tick )
 {
-  myIORs.Append(IOR);
-  myShapes.Append(S);
+  myShapesMap[IOR] = S;
+  myTicksMap[IOR] = Tick;
 }
 
 //=======================================================================
 // 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 ) ;
-  if( anIndex != 0 ) {
-    myIORs.Remove(anIndex) ;
-    myShapes.Remove(anIndex) ;
-  }
-  return ;
+  if ( myShapesMap.erase( IOR ))
+    _mySubShapes.erase( IOR );
 }
 
-
 //=======================================================================
 // function : ClearClientBuffer()
 // purpose  : purge buffer
 //=======================================================================
 void GEOM_Client::ClearClientBuffer()
 {
-  if( myIORs.IsEmpty() )
-    return ;
-  myIORs.Clear() ;
-  myShapes.Clear() ;
-  return ;
+  _mySubShapes.clear();
+  myShapesMap.clear();
 }
 
 //=======================================================================
-// function : BufferLength()
+// function : GetShape()
 // purpose  :
 //=======================================================================
-unsigned int GEOM_Client::BufferLength()
+TopoDS_Shape GEOM_Client::GetShape( GEOM::GEOM_Gen_ptr geom, GEOM::GEOM_Object_ptr aShape )
 {
-  return myIORs.Length() ;
-}
+  CORBA::String_var     anIOR = geom->GetStringFromIOR(aShape);
+  TCollection_AsciiString IOR = anIOR.in();
 
+  // Special treatment for groups: group value (shape) is computed on
+  // first demand, and its tick is set to main shape's tick after that;
+  // tick of modified group is decremented.
+  // There is a not fixed problem: if anyone modifies a group and updates
+  // it before call to this method, the group will not be changed in the buffer
+  bool isModifiedSubShape = false;
+  int aShapeTick = aShape->GetTick();
+  if ( !aShape->IsMainShape() ) {
+    int anOldTick = aShapeTick;
+    aShape->GetShapeType(); // compute subshape and update its tick, if not yet
+    aShapeTick = aShape->GetTick();
+    isModifiedSubShape = (anOldTick != aShapeTick);
+  }
 
-//=======================================================================
-// function : GetShape()
-// purpose  : 
-//=======================================================================
+  std::map< TCollection_AsciiString , int >::iterator i2t = myTicksMap.find( IOR );
+  if ( i2t != myTicksMap.end() ) {
+    if (i2t->second != aShapeTick || isModifiedSubShape) {
+      // The shape was modified, clean the stored one
+      RemoveShapeFromBuffer(IOR);
+    }
+  }
 
-TopoDS_Shape GEOM_Client::GetShape( GEOM::GEOM_Gen_ptr geom, GEOM::GEOM_Shape_ptr aShape ) 
-{ 
+  TopoDS_Shape S;
+  if ( Find( IOR, S ))
+    return S;
 
-  TopoDS_Shape            S; 
-  TCollection_AsciiString IOR(aShape->Name()); 
-  Standard_Integer        anIndex = Find(IOR, S); 
-  
-  BRep_Builder B; 
+  /******* in case of a MAIN GEOM::SHAPE ********/
+  if ( aShape->IsMainShape() ) {
+    S = Load(geom, aShape);
+    Bind(IOR, S, aShapeTick);
+    return S;
+  }
 
-  if (anIndex !=0 ) { 
-    return S ; 
-   } 
+  /******* in case of SUB GEOM::SHAPE ***********/
+  // Load and Explore the Main Shape
+  GEOM::GEOM_Object_var mainGO = aShape->GetMainShape();
+  TopoDS_Shape      aMainShape = GetShape( geom, mainGO );
+  GEOM::ListOfLong_var    list = aShape->GetSubShapeIndices();
 
-  /******* in case of a MAIN GEOM::SHAPE ********/ 
-  if (aShape->IsMainShape()) { 
-    S = Load(geom, aShape); 
-    Bind(IOR,S); 
-    return S; 
-  } 
+  CORBA::String_var      aMainIOR = geom->GetStringFromIOR( mainGO );
+  TCollection_AsciiString mainIOR = aMainIOR.in();
 
-  /******* 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(); 
+  //find subshapes only one time
+  auto pos2isnew = _mySubShapes.insert( std::make_pair( mainIOR, std::vector<TopoDS_Shape>() ));
+  std::vector<TopoDS_Shape> & subShapes = pos2isnew.first->second;
+  if ( pos2isnew.second )
+  {
+    TopTools_IndexedMapOfShape anIndices;
+    TopExp::MapShapes( aMainShape, anIndices );
+    subShapes.insert( subShapes.end(), anIndices.cbegin(), anIndices.cend() );
+  }
 
-  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 !)   */ 
+  /* Case of only one subshape, excluding groups of one main shape */
+  if ( list->length() == 1 &&
+       !(aShape->GetType() == 37 && list[0] == 1) &&
+       list[0] > 0 ) {
+    S = subShapes[list[0]-1];
+  }
+  else {
+    BRep_Builder B;
+    TopoDS_Compound aCompound;
+    B.MakeCompound(aCompound);
+    CORBA::Long nbSub = subShapes.size();
+    for ( size_t i = 0; i < list->length(); i++ )
+      if ( 0 < list[i] && list[i] <= nbSub )
+        B.Add(aCompound, subShapes[list[i]-1] );
 
-  /* 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() ; 
+    S = aCompound;
+  }
 
-  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; 
-} 
+  Bind(IOR, S, aShapeTick);
+  return S;
+}