Salome HOME
refs #30 - Sketch base GUI: create, draw lines
[modules/shaper.git] / src / ModelAPI / ModelAPI_Feature.h
index 7cb9256a08874c3c8e609625020ba2c5bcec78b3..eb4ba54f992dc60664127fa71ba529a540dea8c8 100644 (file)
@@ -9,7 +9,7 @@
 #include "ModelAPI_PluginManager.h"
 
 #include <string>
-#include <memory>
+#include <boost/shared_ptr.hpp>
 
 class ModelAPI_Data;
 class ModelAPI_Document;
@@ -21,7 +21,8 @@ class ModelAPI_Document;
  */
 class ModelAPI_Feature
 {
-  std::shared_ptr<ModelAPI_Data> myData; ///< manager of the data model of a feature
+  boost::shared_ptr<ModelAPI_Data> myData; ///< manager of the data model of a feature
+  boost::shared_ptr<ModelAPI_Document> myDoc; ///< document this feature belongs to
 
 public:
   /// Returns the kind of a feature (like "Point")
@@ -36,14 +37,21 @@ public:
   /// Computes or recomputes the result
   MODELAPI_EXPORT virtual void execute() = 0;
 
+  /// Returns true if this feature must be displayed in the history (top level of Part tree)
+  MODELAPI_EXPORT virtual bool isInHistory() {return true;}
+
   /// Returns the data manager of this feature
-  MODELAPI_EXPORT virtual std::shared_ptr<ModelAPI_Data> data() {return myData;}
+  MODELAPI_EXPORT virtual boost::shared_ptr<ModelAPI_Data> data() {return myData;}
 
   /// Must return document where the new feature must be added to
   /// By default it is current document
-  MODELAPI_EXPORT virtual std::shared_ptr<ModelAPI_Document> documentToAdd()
+  MODELAPI_EXPORT virtual boost::shared_ptr<ModelAPI_Document> documentToAdd()
   {return ModelAPI_PluginManager::get()->currentDocument();}
 
+  /// Returns document this feature belongs to
+  MODELAPI_EXPORT virtual boost::shared_ptr<ModelAPI_Document> document()
+  {return myDoc;}
+
   /// To virtually destroy the fields of successors
   virtual ~ModelAPI_Feature() {}
 
@@ -54,8 +62,16 @@ protected:
   {}
 
   /// Sets the data manager of an object (document does)
-  MODELAPI_EXPORT void setData(std::shared_ptr<ModelAPI_Data> theData) {myData = theData;}
+  MODELAPI_EXPORT virtual void setData(boost::shared_ptr<ModelAPI_Data> theData) 
+  {myData = theData;}
+  /// Sets the data manager of an object (document does)
+  MODELAPI_EXPORT void setDoc(boost::shared_ptr<ModelAPI_Document> theDoc) {myDoc = theDoc;}
+
   friend class Model_Document;
 };
 
+//! Pointer on feature object
+typedef boost::shared_ptr<ModelAPI_Feature> FeaturePtr;
+
+
 #endif