For the issue #1856 : prepare the persistence mechanism for nodes states

diff --git a/src/Model/Model_Document.cpp b/src/Model/Model_Document.cpp
index c8b567cf72be377dc583fac64e9aebebe231647e..79febfc4e347d82f36b6e9efe05f492e68d9728a 100755 (executable)
--- a/src/Model/Model_Document.cpp
+++ b/src/Model/Model_Document.cpp
@@ -67,9 +67,10 @@ static const int TAG_GENERAL = 1;  // general properties tag
 
 // general sub-labels
 /// where the reference to the current feature label is located (or no attribute if null feature)
-static const int TAG_CURRENT_FEATURE = 1;
+static const int TAG_CURRENT_FEATURE = 1; ///< reference to the current feature
 static const int TAG_CURRENT_TRANSACTION = 2; ///< integer, index of the transaction
 static const int TAG_SELECTION_FEATURE = 3; ///< integer, tag of the selection feature label
+static const int TAG_NODES_STATE = 4; ///< array, tag of the Object Browser nodes states
 
 Model_Document::Model_Document(const int theID, const std::string theKind)
     : myID(theID), myKind(theKind), myIsActive(false),
@@ -1538,3 +1539,29 @@ bool Model_Document::isLater(FeaturePtr theLater, FeaturePtr theCurrent) const
 {
   return myObjs->isLater(theLater, theCurrent);
 }
+
+void Model_Document::storeNodesState(const std::list<bool>& theStates)
+{
+  TDF_Label aLab = generalLabel().FindChild(TAG_NODES_STATE);
+  aLab.ForgetAllAttributes();
+  if (!theStates.empty()) {
+    Handle(TDataStd_BooleanArray) anArray =
+      TDataStd_BooleanArray::Set(aLab, 0, int(theStates.size()) - 1);
+    std::list<bool>::const_iterator aState = theStates.begin();
+    for(int anIndex = 0; aState != theStates.end(); aState++, anIndex++) {
+      anArray->SetValue(anIndex, *aState);
+    }
+  }
+}
+
+void Model_Document::restoreNodesState(std::list<bool>& theStates) const
+{
+  TDF_Label aLab = generalLabel().FindChild(TAG_NODES_STATE);
+  Handle(TDataStd_BooleanArray) anArray;
+  if (aLab.FindAttribute(TDataStd_BooleanArray::GetID(), anArray)) {
+    int anUpper = anArray->Upper();
+    for(int anIndex = 0; anIndex <= anUpper; anIndex++) {
+      theStates.push_back(anArray->Value(anIndex) == Standard_True);
+    }
+  }
+}

diff --git a/src/Model/Model_Document.h b/src/Model/Model_Document.h
index 80daeb237f27a8ccc4d089b077d5a7c779b3f56e..1356f5c09fa18d4f125ddf89f1c26bbfbf11476e 100644 (file)
--- a/src/Model/Model_Document.h
+++ b/src/Model/Model_Document.h
@@ -280,6 +280,14 @@ class Model_Document : public ModelAPI_Document
   //! for calculation of selection externally from the document
   std::shared_ptr<ModelAPI_AttributeSelectionList> selectionInPartFeature();
 
+  /// Stores in the document boolean flags: states of the nodes in the object browser.
+  /// Normally is called outside of the transaction, just before "save".
+  void storeNodesState(const std::list<bool>& theStates);
+
+  /// Returns the stored nodes states. Normally it is calles just after "open".
+  /// Appends the values to theStates list.
+  void restoreNodesState(std::list<bool>& theStates) const;
+
   friend class Model_Application;
   friend class Model_Session;
   friend class Model_Update;

diff --git a/src/ModelAPI/ModelAPI_Document.h b/src/ModelAPI/ModelAPI_Document.h
index 90b87621b71869b567d5e2c7640124583fa07ecf..6b0d2d1ea1b244bb31962006ac700f4bc7759d1a 100644 (file)
--- a/src/ModelAPI/ModelAPI_Document.h
+++ b/src/ModelAPI/ModelAPI_Document.h
@@ -176,6 +176,14 @@ public:
   //! Internally makes document know that feature was removed or added in history after creation
   MODELAPI_EXPORT virtual void updateHistory(const std::string theGroup) = 0;
 
+  /// Stores in the document boolean flags: states of the nodes in the object browser.
+  /// Normally is called outside of the transaction, just before "save".
+  MODELAPI_EXPORT void storeNodesState(const std::list<bool>& theStates);
+
+  /// Returns the stored nodes states. Normally it is calles just after "open".
+  /// Appends the values to theStates list.
+  MODELAPI_EXPORT void restoreNodesState(std::list<bool>& theStates) const;
+
 protected:
   //! Only for SWIG wrapping it is here
   MODELAPI_EXPORT ModelAPI_Document();