Salome HOME
To show notebook variables in the Object Browser.
[modules/geom.git] / src / GEOM_I / GEOM_Gen_i.cc
index 2ae6d0a596aa1a8ea1b4b33d1c79fea3033bd522..b36a652f7981dab7152f75e29e12ebb7b078a979 100644 (file)
@@ -273,6 +273,35 @@ SALOMEDS::SObject_ptr GEOM_Gen_i::PublishInStudy(SALOMEDS::Study_ptr theStudy,
   SALOMEDS::AttributeName_var aNameAttrib = SALOMEDS::AttributeName::_narrow(anAttr);
   aNameAttrib->SetValue(aShapeName.ToCString());
 
+  //Set NoteBook variables used in the object creation
+  TCollection_AsciiString aParams(aShape->GetParameters());
+  if(!aParams.IsEmpty()) {
+    TCollection_AsciiString aVars;
+    int nbVars = 0;
+    int n = 1;
+    TCollection_AsciiString aParam = aParams.Token(":",1);
+    while( aParam.Length() != 0 ) {
+      aParam = aParams.Token(":",n);
+      if(theStudy->IsVariable(aParam.ToCString())){
+       aVars+=aParam;
+       nbVars++;
+      }
+      if(aParam.Length() == 0)
+       break;
+      aVars+=":";
+      n++;
+    }
+    aVars.Remove(aVars.Length(),1);
+    if(nbVars > 0 ) {
+      anAttr = aStudyBuilder->FindOrCreateAttribute(aResultSO, "AttributeString");
+      SALOMEDS::AttributeString_var aStringAttrib = SALOMEDS::AttributeString::_narrow(anAttr);
+      aStringAttrib->SetValue(aVars.ToCString());
+    }
+  }
+  
+  //Set a name of the GEOM object
+  aShape->SetName(theName);
+
   return aResultSO._retn();
 }
 
@@ -287,6 +316,17 @@ SALOMEDS::TMPFile* GEOM_Gen_i::Save(SALOMEDS::SComponent_ptr theComponent,
   SALOMEDS::TMPFile_var aStreamFile;
   // Get a temporary directory to store a file
   std::string aTmpDir = (isMultiFile)?theURL:SALOMEDS_Tool::GetTmpDir();
+
+  // OCCT BUG: cannot save a document (in current folder)
+  // if directory name is empty
+  if (aTmpDir.size() == 0) {
+#ifdef WNT
+    aTmpDir = ".\\";
+#else
+    aTmpDir = "./";
+#endif
+  }
+
   // Create a list to store names of created files
   SALOMEDS::ListOfFileNames_var aSeq = new SALOMEDS::ListOfFileNames;
   aSeq->length(1);
@@ -299,7 +339,7 @@ SALOMEDS::TMPFile* GEOM_Gen_i::Save(SALOMEDS::SComponent_ptr theComponent,
   // Build a full file name of temporary file
   TCollection_AsciiString aFullName = TCollection_AsciiString((char*)aTmpDir.c_str()) + aNameWithExt;
   // Save GEOM component in this file
-  _impl->Save(theComponent->GetStudy()->StudyId(), aFullName.ToCString());
+  _impl->Save(theComponent->GetStudy()->StudyId(),(char*) aFullName.ToCString());
   // Conver a file to the byte stream
   aStreamFile = SALOMEDS_Tool::PutFilesToStream(aTmpDir.c_str(), aSeq.in(), isMultiFile);
   // Remove the created file and tmp directory
@@ -338,8 +378,20 @@ CORBA::Boolean GEOM_Gen_i::Load(SALOMEDS::SComponent_ptr theComponent,
 
   // Get a temporary directory for a file
   std::string aTmpDir = isMultiFile?theURL:SALOMEDS_Tool::GetTmpDir();
+
+  // OCCT BUG: cannot load a document (from current folder)
+  // if directory name is empty
+  if (aTmpDir.size() == 0) {
+#ifdef WNT
+    aTmpDir = ".\\";
+#else
+    aTmpDir = "./";
+#endif
+  }
+
   // Conver the byte stream theStream to a file and place it in tmp directory
-  SALOMEDS::ListOfFileNames_var aSeq = SALOMEDS_Tool::PutStreamToFiles(theStream, aTmpDir.c_str(), isMultiFile);
+  SALOMEDS::ListOfFileNames_var aSeq =
+    SALOMEDS_Tool::PutStreamToFiles(theStream, aTmpDir.c_str(), isMultiFile);
 
   // Prepare a file name to open
   TCollection_AsciiString aNameWithExt("");
@@ -349,7 +401,7 @@ CORBA::Boolean GEOM_Gen_i::Load(SALOMEDS::SComponent_ptr theComponent,
   TCollection_AsciiString aFullName = (TCollection_AsciiString((char*)aTmpDir.c_str()) + aNameWithExt);
 
   // Open document
-  if (!_impl->Load(theComponent->GetStudy()->StudyId(), aFullName.ToCString())) return false;
+  if (!_impl->Load(theComponent->GetStudy()->StudyId(),(char*) aFullName.ToCString())) return false;
 
   // Remove the created file and tmp directory
   if (!isMultiFile) SALOMEDS_Tool::RemoveTemporaryFiles(aTmpDir.c_str(), aSeq.in(), true);
@@ -496,7 +548,10 @@ char* GEOM_Gen_i::ComponentDataType()
 // function : AddInStudy
 // purpose  :
 //============================================================================
-SALOMEDS::SObject_ptr GEOM_Gen_i::AddInStudy(SALOMEDS::Study_ptr theStudy, GEOM::GEOM_Object_ptr theObject, const char* theName, GEOM::GEOM_Object_ptr theFather)
+SALOMEDS::SObject_ptr GEOM_Gen_i::AddInStudy (SALOMEDS::Study_ptr theStudy,
+                                             GEOM::GEOM_Object_ptr theObject,
+                                             const char* theName,
+                                             GEOM::GEOM_Object_ptr theFather)
 {
   SALOMEDS::SObject_var aResultSO;
   if(theObject->_is_nil() || theStudy->_is_nil()) return aResultSO;
@@ -533,14 +588,401 @@ SALOMEDS::SObject_ptr GEOM_Gen_i::AddInStudy(SALOMEDS::Study_ptr theStudy, GEOM:
   return aResultSO._retn();
 }
 
+//============================================================================
+// function : RestoreSubShapesO
+// purpose  : Publish sub-shapes, standing for arguments and sub-shapes of arguments.
+//            To be used from python scripts out of geompy.addToStudy (non-default usage)
+//============================================================================
+CORBA::Boolean GEOM_Gen_i::RestoreSubShapesO (SALOMEDS::Study_ptr     theStudy,
+                                             GEOM::GEOM_Object_ptr   theObject,
+                                             const GEOM::ListOfGO&   theArgs,
+                                             GEOM::find_shape_method theFindMethod,
+                                             CORBA::Boolean          theInheritFirstArg)
+{
+  if (CORBA::is_nil(theStudy) || CORBA::is_nil(theObject))
+    return false;
+
+  // find SObject in the study
+  CORBA::String_var anIORo = _orb->object_to_string(theObject);
+  SALOMEDS::SObject_var aSO = theStudy->FindObjectIOR(anIORo.in());
+  if (CORBA::is_nil(aSO))
+    return false;
+
+  return RestoreSubShapes(theStudy, theObject, aSO, theArgs, theFindMethod, theInheritFirstArg);
+}
+
+//============================================================================
+// function : RestoreSubShapesSO
+// purpose  : Publish sub-shapes, standing for arguments and sub-shapes of arguments.
+//            To be used from GUI and from geompy.addToStudy
+//============================================================================
+CORBA::Boolean GEOM_Gen_i::RestoreSubShapesSO (SALOMEDS::Study_ptr     theStudy,
+                                              SALOMEDS::SObject_ptr   theSObject,
+                                              const GEOM::ListOfGO&   theArgs,
+                                              GEOM::find_shape_method theFindMethod,
+                                              CORBA::Boolean          theInheritFirstArg)
+{
+  if (CORBA::is_nil(theStudy) || CORBA::is_nil(theSObject))
+    return false;
+
+  SALOMEDS::GenericAttribute_var anAttr;
+  if (!theSObject->FindAttribute(anAttr, "AttributeIOR"))
+    return false;
+
+  SALOMEDS::AttributeIOR_var anAttrIOR = SALOMEDS::AttributeIOR::_narrow(anAttr);
+  CORBA::String_var anIORso = anAttrIOR->Value();
+
+  // get Object from SObject
+  GEOM::GEOM_Object_var anO = GEOM::GEOM_Object::_narrow(_orb->string_to_object(anIORso));
+  if (CORBA::is_nil(anO))
+    return false;
+
+  return RestoreSubShapes(theStudy, anO, theSObject, theArgs, theFindMethod, theInheritFirstArg);
+}
+
+//============================================================================
+// function : RestoreSubShapes
+// purpose  : Private method. Works only if both theObject and theSObject
+//            are defined, and does not check, if they correspond to each other.
+//============================================================================
+CORBA::Boolean GEOM_Gen_i::RestoreSubShapes (SALOMEDS::Study_ptr     theStudy,
+                                            GEOM::GEOM_Object_ptr   theObject,
+                                            SALOMEDS::SObject_ptr   theSObject,
+                                            const GEOM::ListOfGO&   theArgs,
+                                            GEOM::find_shape_method theFindMethod,
+                                            CORBA::Boolean          theInheritFirstArg)
+{
+  if (CORBA::is_nil(theStudy) || CORBA::is_nil(theObject) || CORBA::is_nil(theSObject))
+    return false;
+
+  // Arguments to be published
+  GEOM::ListOfGO_var aList;
+
+  // If theArgs list is empty, we try to publish all arguments,
+  // otherwise publish only passed args
+  Standard_Integer nbArgsActual = -1; // -1 means unknown
+  Standard_Integer aLength = theArgs.length();
+  if (aLength > 0) {
+    aList = new GEOM::ListOfGO;
+    aList->length(aLength);
+    for (int i = 0; i < aLength; i++) {
+      aList[i] = theArgs[i];
+    }
+  }
+  else {
+    // Get all arguments
+    aList = theObject->GetDependency();
+    aLength = aList->length();
+    nbArgsActual = aLength;
+  }
+
+  if (aLength < 1)
+    return false;
+
+  if (theInheritFirstArg || (nbArgsActual == 1)) {
+    // Do not publish argument's reflection,
+    // but only reconstruct its published sub-shapes
+
+    GEOM::GEOM_Object_var anArgO = aList[0];
+    CORBA::String_var anIOR = _orb->object_to_string(anArgO);
+    SALOMEDS::SObject_var anArgSO = theStudy->FindObjectIOR(anIOR.in());
+
+    GEOM::ListOfGO_var aParts =
+      RestoreSubShapesOneLevel(theStudy, anArgSO, theSObject, theObject, theFindMethod);
+
+    // set the color of the transformed shape to the color of initial shape
+    theObject->SetColor(aList[0]->GetColor());
+
+    return (aParts->length() > 0);
+  }
+
+  // Get interface, containing method, which we will use to reconstruct sub-shapes
+  GEOM::GEOM_IShapesOperations_var aShapesOp = GetIShapesOperations(theStudy->StudyId());
+  GEOM::GEOM_IGroupOperations_var  aGroupOp  = GetIGroupOperations(theStudy->StudyId());
+
+  // Reconstruct arguments and tree of sub-shapes of the arguments
+  CORBA::String_var anIOR;
+  SALOMEDS::StudyBuilder_var aStudyBuilder = theStudy->NewBuilder();
+  for (Standard_Integer i = 0; i < aLength; i++)
+  {
+    GEOM::GEOM_Object_var anArgO = aList[i];
+    if (!CORBA::is_nil(anArgO)) {
+      anIOR = _orb->object_to_string(anArgO);
+      SALOMEDS::SObject_var anArgSO = theStudy->FindObjectIOR(anIOR.in());
+      TCollection_AsciiString anArgName;
+      if (CORBA::is_nil(anArgSO)) {
+       anArgName = "arg_";
+       anArgName += TCollection_AsciiString(i);
+      }
+      else {
+       anArgName = anArgSO->GetName();
+      }
+
+      // Find a sub-shape of theObject in place of the argument
+      GEOM::GEOM_Object_var aSubO;
+      switch (theFindMethod) {
+      case GEOM::FSM_GetInPlace:
+       {
+         // Use GetInPlace
+         aSubO = aShapesOp->GetInPlace(theObject, anArgO);
+       }
+       break;
+      case GEOM::FSM_Transformed:
+       {
+         // transformation, cannot use GetInPlace, operate with indices
+         GEOM::ListOfLong_var anIDs = anArgO->GetSubShapeIndices();
+         if (anIDs->length() > 1) {
+           // group
+           aSubO = aGroupOp->CreateGroup(theObject, aGroupOp->GetType(anArgO));
+           if (!CORBA::is_nil(aSubO))
+             aGroupOp->UnionIDs(aSubO, anIDs);
+         }
+         else {
+           // single sub-shape
+           aSubO = aShapesOp->GetSubShape(theObject, anIDs[0]);
+         }
+       }
+       break;
+      case GEOM::FSM_GetSame:
+       {
+         // Use GetSame
+         aSubO = aShapesOp->GetSame(theObject, anArgO);
+       }
+       break;
+      case GEOM::FSM_GetShapesOnShape:
+       {
+         // Use GetShapesOnShape. Can work only on solids, so it has sense to search only solids
+         aSubO = aShapesOp->GetShapesOnShapeAsCompound(anArgO, theObject,
+                                                       (short)GEOM::SOLID, GEOM::ST_ONIN);
+       }
+       break;
+      case GEOM::FSM_GetInPlaceByHistory:
+       {
+         // Use GetInPlaceByHistory
+         aSubO = aShapesOp->GetInPlaceByHistory(theObject, anArgO);
+       }
+       break;
+      default:
+       {}
+      }
+
+      if (!CORBA::is_nil(aSubO)) {
+       // Publish the sub-shape
+       TCollection_AsciiString aSubName ("from_");
+       aSubName += anArgName;
+       SALOMEDS::SObject_var aSubSO = aStudyBuilder->NewObject(theSObject);
+       aSubSO = PublishInStudy(theStudy, aSubSO, aSubO, aSubName.ToCString());
+       // Restore color
+       aSubO->SetColor(anArgO->GetColor());
+
+       if (!CORBA::is_nil(anArgSO)) {
+         // Restore published sub-shapes of the argument
+         if (theFindMethod == GEOM::FSM_GetInPlaceByHistory)
+           // pass theObject, because only it has the history
+           RestoreSubShapesOneLevel(theStudy, anArgSO, aSubSO, theObject, theFindMethod);
+         else
+           RestoreSubShapesOneLevel(theStudy, anArgSO, aSubSO, aSubO, theFindMethod);
+       }
+      }
+      else { // GetInPlace failed, try to build from published parts
+       if (!CORBA::is_nil(anArgSO)) {
+         SALOMEDS::SObject_var aSubSO = aStudyBuilder->NewObject(theSObject);
+
+         // Restore published sub-shapes of the argument
+         GEOM::ListOfGO_var aParts =
+           RestoreSubShapesOneLevel(theStudy, anArgSO, aSubSO, theObject, theFindMethod);
+
+         if (aParts->length() > 0) {
+           // try to build an argument from a set of its sub-shapes,
+           // that published and will be reconstructed
+           if (aParts->length() > 1) {
+             aSubO = aShapesOp->MakeCompound(aParts);
+           }
+           else {
+             aSubO = aParts[0];
+           }
+           if (!CORBA::is_nil(aSubO)) {
+             // Publish the sub-shape
+             TCollection_AsciiString aSubName ("from_parts_of_");
+             aSubName += anArgName;
+             aSubSO = PublishInStudy(theStudy, aSubSO, aSubO, aSubName.ToCString());
+             // Restore color
+             aSubO->SetColor(anArgO->GetColor());
+           }
+         }
+         else {
+           // remove created aSubSO, because no parts have been found
+           aStudyBuilder->RemoveObject(aSubSO);
+         }
+       }
+      } // try to build from published parts
+    }
+  } // process arguments
+
+  return true;
+}
+
+//============================================================================
+// function : RestoreSubShapesOneLevel
+// purpose  : Private method
+//============================================================================
+GEOM::ListOfGO* GEOM_Gen_i::RestoreSubShapesOneLevel (SALOMEDS::Study_ptr     theStudy,
+                                                     SALOMEDS::SObject_ptr   theOldSO,
+                                                     SALOMEDS::SObject_ptr   theNewSO,
+                                                     GEOM::GEOM_Object_ptr   theNewO,
+                                                     GEOM::find_shape_method theFindMethod)
+{
+  int i = 0;
+  GEOM::ListOfGO_var aParts = new GEOM::ListOfGO;
+  if (CORBA::is_nil(theStudy) || CORBA::is_nil(theOldSO) ||
+      CORBA::is_nil(theNewO) || CORBA::is_nil(theNewSO))
+    return aParts._retn();
+
+  SALOMEDS::StudyBuilder_var aStudyBuilder = theStudy->NewBuilder();
+
+  // Get interface, containing method, which we will use to reconstruct sub-shapes
+  GEOM::GEOM_IShapesOperations_var aShapesOp = GetIShapesOperations(theStudy->StudyId());
+  GEOM::GEOM_IGroupOperations_var  aGroupOp  = GetIGroupOperations(theStudy->StudyId());
+
+  // Reconstruct published sub-shapes
+  SALOMEDS::ChildIterator_var it = theStudy->NewChildIterator(theOldSO);
+
+  int aLen = 0;
+  for (it->Init(); it->More(); it->Next()) {
+    aLen++;
+  }
+  aParts->length(aLen);
+
+  for (it->Init(); it->More(); it->Next()) {
+    SALOMEDS::SObject_var anOldSubSO = it->Value();
+
+    TCollection_AsciiString anArgName = anOldSubSO->GetName();
+
+    SALOMEDS::GenericAttribute_var anAttr;
+    if (anOldSubSO->FindAttribute(anAttr, "AttributeIOR")) {
+      SALOMEDS::AttributeIOR_var anAttrIOR = SALOMEDS::AttributeIOR::_narrow(anAttr);
+      GEOM::GEOM_Object_var anOldSubO =
+       GEOM::GEOM_Object::_narrow(_orb->string_to_object(anAttrIOR->Value()));
+      if (!CORBA::is_nil(anOldSubO)) {
+       // Find a sub-shape of theNewO in place of anOldSubO
+       GEOM::GEOM_Object_var aNewSubO;
+       switch (theFindMethod) {
+       case GEOM::FSM_GetInPlace:
+         {
+           // Use GetInPlace
+           aNewSubO = aShapesOp->GetInPlace(theNewO, anOldSubO);
+         }
+         break;
+       case GEOM::FSM_Transformed:
+         {
+           // transformation, cannot use GetInPlace, operate with indices
+           GEOM::ListOfLong_var anIDs = anOldSubO->GetSubShapeIndices();
+           if (anIDs->length() > 1) {
+             // group
+             aNewSubO = aGroupOp->CreateGroup(theNewO, aGroupOp->GetType(anOldSubO));
+             if (!CORBA::is_nil(aNewSubO))
+               aGroupOp->UnionIDs(aNewSubO, anIDs);
+           }
+           else {
+             // single sub-shape
+             aNewSubO = aShapesOp->GetSubShape(theNewO, anIDs[0]);
+           }
+         }
+         break;
+       case GEOM::FSM_GetSame:
+         {
+           // Use GetSame
+           aNewSubO = aShapesOp->GetSame(theNewO, anOldSubO);
+         }
+         break;
+       case GEOM::FSM_GetShapesOnShape:
+         {
+           // Use GetShapesOnShape. Can work only on solids, so it has sense to search only solids
+           aNewSubO = aShapesOp->GetShapesOnShapeAsCompound(anOldSubO, theNewO,
+                                                            (short)GEOM::SOLID, GEOM::ST_ONIN);
+         }
+         break;
+       case GEOM::FSM_GetInPlaceByHistory:
+         {
+           // Use GetInPlaceByHistory
+           aNewSubO = aShapesOp->GetInPlaceByHistory(theNewO, anOldSubO);
+         }
+         break;
+       default:
+         {}
+       }
+
+       if (!CORBA::is_nil(aNewSubO)) {
+         // add the part to the list
+         aParts[i] = aNewSubO;
+         i++;
+
+         // Publish the sub-shape
+         TCollection_AsciiString aSubName ("from_");
+         aSubName += anArgName;
+         SALOMEDS::SObject_var aNewSubSO = aStudyBuilder->NewObject(theNewSO);
+         aNewSubSO = PublishInStudy(theStudy, aNewSubSO, aNewSubO, aSubName.ToCString());
+         // Restore color
+         aNewSubO->SetColor(anOldSubO->GetColor());
+
+         // Restore published sub-shapes of the argument
+         if (theFindMethod == GEOM::FSM_GetInPlaceByHistory)
+           // pass the main shape as Object, because only it has the history
+           RestoreSubShapesOneLevel(theStudy, anOldSubSO, aNewSubSO, theNewO, theFindMethod);
+         else
+           RestoreSubShapesOneLevel(theStudy, anOldSubSO, aNewSubSO, aNewSubO, theFindMethod);
+       }
+       else { // GetInPlace failed, try to build from published parts
+         SALOMEDS::SObject_var aNewSubSO = aStudyBuilder->NewObject(theNewSO);
+
+         // Restore published sub-shapes of the argument
+         GEOM::ListOfGO_var aParts =
+           RestoreSubShapesOneLevel(theStudy, anOldSubSO, aNewSubSO, theNewO, theFindMethod);
+
+         if (aParts->length() > 0) {
+           // try to build an object from a set of its sub-shapes,
+           // that published and will be reconstructed
+           if (aParts->length() > 1) {
+             aNewSubO = aShapesOp->MakeCompound(aParts);
+           }
+           else {
+             aNewSubO = aParts[0];
+           }
+
+           if (!CORBA::is_nil(aNewSubO)) {
+             // add the part to the list
+             aParts[i] = aNewSubO;
+             i++;
+
+             // Publish the sub-shape
+             TCollection_AsciiString aSubName = "from_parts_of_";
+             aSubName += anArgName;
+             aNewSubSO = PublishInStudy(theStudy, aNewSubSO, aNewSubO, aSubName.ToCString());
+             // Restore color
+             aNewSubO->SetColor(anOldSubO->GetColor());
+           }
+         }
+         else {
+           // remove created aSubSO, because no parts have been found
+           aStudyBuilder->RemoveObject(aNewSubSO);
+         }
+       } // try to build from published parts
+      }
+    }
+  } // iterate on published sub-shapes
+
+  aParts->length(i);
+  return aParts._retn();
+}
+
 //============================================================================
 // function : register()
 // purpose  : register 'name' in 'name_service'
 //============================================================================
 void GEOM_Gen_i::register_name(char * name)
 {
-  GEOM::GEOM_Gen_ptr g = GEOM::GEOM_Gen::_narrow(_this());
-  name_service->Register(g, CORBA::string_dup(name));
+  GEOM::GEOM_Gen_var g = _this();
+  name_service->Register(g, name);
 }
 
 //============================================================================
@@ -573,6 +1015,7 @@ GEOM::GEOM_IBasicOperations_ptr GEOM_Gen_i::GetIBasicOperations(CORBA::Long theS
 
   GEOM::GEOM_Gen_ptr engine = _this();
 
+  //transfer reference on engine
   GEOM_IBasicOperations_i* aServant =
     new GEOM_IBasicOperations_i(_poa, engine, _impl->GetIBasicOperations(theStudyID));
 
@@ -831,7 +1274,8 @@ GEOM::GEOM_Object_ptr GEOM_Gen_i::AddSubShape (GEOM::GEOM_Object_ptr theMainShap
 //=============================================================================
 void GEOM_Gen_i::RemoveObject(GEOM::GEOM_Object_ptr theObject)
 {
-  Handle(GEOM_Object) anObject = _impl->GetObject(theObject->GetStudyID(), theObject->GetEntry());
+  CORBA::String_var anEntry = theObject->GetEntry();
+  Handle(GEOM_Object) anObject = _impl->GetObject(theObject->GetStudyID(), anEntry);
   if (anObject.IsNull()) return;
   _impl->RemoveObject(anObject);
   return;
@@ -902,7 +1346,8 @@ GEOM_I_EXPORT
                                                const char * interfaceName)
   {
    GEOM_Gen_i * myGEOM_Gen_i = new GEOM_Gen_i(orb, poa, contId, instanceName, interfaceName);
-   myGEOM_Gen_i->register_name("/myGEOM_Gen"); // NRI : 11/07/2002 : Add for Supervision example
+   // Don't understand the reason of this register ????
+//   myGEOM_Gen_i->register_name("/myGEOM_Gen"); // NRI : 11/07/2002 : Add for Supervision example
    return myGEOM_Gen_i->getId();
   }
 }