refs #30 - Sketch base GUI: create, draw lines

[modules/shaper.git] / src / ModelAPI / ModelAPI_Feature.h

diff --git a/src/ModelAPI/ModelAPI_Feature.h b/src/ModelAPI/ModelAPI_Feature.h
index 8c1a82e75e2a3a22380065e42dab13463cba0662..eb4ba54f992dc60664127fa71ba529a540dea8c8 100644 (file)
--- a/src/ModelAPI/ModelAPI_Feature.h
+++ b/src/ModelAPI/ModelAPI_Feature.h
@@ -22,6 +22,7 @@ class ModelAPI_Document;
 class ModelAPI_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,6 +37,9 @@ 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 boost::shared_ptr<ModelAPI_Data> data() {return myData;}
 
@@ -44,6 +48,10 @@ public:
   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(boost::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