Salome HOME
Migration to VC9 and boost::shared_ptr with connection to SALOME
[modules/shaper.git] / src / Model / Model_Document.h
index d96aa15dd039747a06d2fb7113adab3947353a7d..f98788c5ec8c3f802a5bde4228a8350ad7712c68 100644 (file)
@@ -7,20 +7,19 @@
 
 #include <Model.h>
 #include <ModelAPI_Document.h>
+
 #include <TDocStd_Document.hxx>
 #include <map>
+#include <set>
 
 class Handle_Model_Document;
 
 /**\class Model_Document
  * \ingroup DataModel
- * \brief Document for internal data structure of any object storage. Corresponds to the SALOME study.
- * Document contains all data of te SALOME Study specific to this module
- * that must be written into the HDF file.
+ * \brief Document for internal data structure of any object storage.
+ * Document contains all data that must be stored/retrived in the file.
  * Also it provides acces to this data: open/save, transactions management etc.
- * to provide access to all stored data.
  */
-
 class Model_Document: public ModelAPI_Document
 {
 public:
@@ -29,59 +28,65 @@ public:
   //! \param theFileName full name of the file to load
   //! \param theStudyID identifier of the SALOME study to associate with loaded file
   //! \returns true if file was loaded successfully
-  MODEL_EXPORT bool load(const char* theFileName);
+  MODEL_EXPORT virtual bool load(const char* theFileName);
 
   //! Saves the OCAF document to the file.
   //! \param theFileName full name of the file to store
   //! \returns true if file was stored successfully
-  MODEL_EXPORT bool save(const char* theFileName);
+  MODEL_EXPORT virtual bool save(const char* theFileName);
 
   //! Removes document data
-  MODEL_EXPORT void close();
+  MODEL_EXPORT virtual void close();
 
   //! Starts a new operation (opens a tansaction)
-  MODEL_EXPORT void startOperation();
+  MODEL_EXPORT virtual void startOperation();
   //! Finishes the previously started operation (closes the transaction)
-  MODEL_EXPORT void finishOperation();
+  //! Returns true if transaction in this document is not empty and really was performed
+  MODEL_EXPORT virtual void finishOperation();
   //! Aborts the operation 
-  MODEL_EXPORT void abortOperation();
+  MODEL_EXPORT virtual void abortOperation();
   //! Returns true if operation has been started, but not yet finished or aborted
-  MODEL_EXPORT bool isOperation();
+  MODEL_EXPORT virtual bool isOperation();
   //! Returns true if document was modified (since creation/opening)
-  MODEL_EXPORT bool isModified();
+  MODEL_EXPORT virtual bool isModified();
 
   //! Returns True if there are available Undos
-  MODEL_EXPORT bool canUndo();
+  MODEL_EXPORT virtual bool canUndo();
   //! Undoes last operation
-  MODEL_EXPORT void undo();
+  MODEL_EXPORT virtual void undo();
   //! Returns True if there are available Redos
-  MODEL_EXPORT bool canRedo();
+  MODEL_EXPORT virtual bool canRedo();
   //! Redoes last operation
-  MODEL_EXPORT void redo();
+  MODEL_EXPORT virtual void redo();
 
-  //! Adds to the document the new feature of the given group id
-  //! \param theFeature a feature object that will be connected to the document in this method
-  //! \param theGroupID identifier of the groups of objects (must be greater than zero)
-  MODEL_EXPORT virtual void addFeature(std::shared_ptr<ModelAPI_Feature> theFeature,
-    const std::string theGroupID);
+  //! Adds to the document the new feature of the given feature id
+  //! \param creates feature and puts it in the document
+  MODEL_EXPORT virtual boost::shared_ptr<ModelAPI_Feature> addFeature(std::string theID);
 
   //! Returns the existing feature by the label
   //! \param theLabel base label of the feature
-  MODEL_EXPORT virtual std::shared_ptr<ModelAPI_Feature> feature(TDF_Label& theLabel);
+  MODEL_EXPORT virtual boost::shared_ptr<ModelAPI_Feature> feature(TDF_Label& theLabel);
 
   //! Adds a new sub-document by the identifier, or returns existing one if it is already exist
-  MODEL_EXPORT virtual std::shared_ptr<ModelAPI_Document> subDocument(std::string theDocID);
+  MODEL_EXPORT virtual boost::shared_ptr<ModelAPI_Document> subDocument(std::string theDocID);
 
   //! Creates an iterator of the features by the specific groups
-  MODEL_EXPORT virtual std::shared_ptr<ModelAPI_Iterator> featuresIterator(
+  MODEL_EXPORT virtual boost::shared_ptr<ModelAPI_Iterator> featuresIterator(
     const std::string theGroup);
 
   MODEL_EXPORT virtual const std::string& id() const {return myID;}
 
   //! Returns the feature in the group by the index (started from zero)
-  MODEL_EXPORT virtual std::shared_ptr<ModelAPI_Feature> 
+  MODEL_EXPORT virtual boost::shared_ptr<ModelAPI_Feature> 
     feature(const std::string& theGroupID, const int theIndex);
 
+  ///! Returns the vector of groups already added to the document
+  MODEL_EXPORT virtual const std::vector<std::string>& getGroups() const;
+
+  //! Returns the index of feature in the group (zero based)
+  //! \retruns -1 if not found
+  MODEL_EXPORT virtual int featureIndex(boost::shared_ptr<ModelAPI_Feature> theFeature);
+
 protected:
 
   //! Returns (creates if needed) the group label
@@ -89,8 +94,13 @@ protected:
 
   //! Initializes feature with a unique name in this group (unique name is generated as 
   //! feature type + "_" + index
-  void setUniqueName(
-    std::shared_ptr<ModelAPI_Feature> theFeature, const std::string theGroupID);
+  void setUniqueName(boost::shared_ptr<ModelAPI_Feature> theFeature);
+
+  //! Adds to the document the new feature
+  void addFeature(const boost::shared_ptr<ModelAPI_Feature> theFeature);
+
+  //! Synchronizes myGroups, myGroupsNames, myFeatures and mySubs list with the updated document
+  void synchronizeFeatures();
 
   //! Creates new document with binary file format
   Model_Document(const std::string theID);
@@ -100,8 +110,16 @@ protected:
 private:
   std::string myID; ///< identifier of the document in the application
   Handle_TDocStd_Document myDoc; ///< OCAF document
-  int myTransactionsAfterSave; ///< number of transactions after the last "save" call, used for "IsModified" method
-  std::map<std::string, TDF_Label> myGroups; ///< root labels of the features groups identified by names
+  /// number of transactions after the last "save" call, used for "IsModified" method
+  int myTransactionsAfterSave;
+  /// root labels of the features groups identified by names
+  std::map<std::string, TDF_Label> myGroups;
+  std::vector<std::string> myGroupsNames; ///< names of added groups to the document
+  /// Features managed by this document: by group name
+  std::map<std::string, std::vector<boost::shared_ptr<ModelAPI_Feature> > > myFeatures;
+  std::set<std::string> mySubs; ///< set of identifiers of sub-documents of this document
+  /// transaction indexes (related to myTransactionsAfterSave) which were empty in this doc
+  std::map<int, bool> myIsEmptyTr;
 };
 
 #endif