Merge branch 'master' of newgeom:newgeom

[modules/shaper.git] / src / Model / Model_AttributeReference.h

diff --git a/src/Model/Model_AttributeReference.h b/src/Model/Model_AttributeReference.h
index e17855e48419ee63c2ae242fc0a15ad376c6c9ab..d9993f99a6049a57a0beb97ee7a3ca7b5e84c08a 100644 (file)
--- a/src/Model/Model_AttributeReference.h
+++ b/src/Model/Model_AttributeReference.h
@@ -2,11 +2,12 @@
 // Created:     8 May 2014
 // Author:      Mikhail PONIKAROV
 
-#ifndef Model_AttributeReference_HeaderFile
-#define Model_AttributeReference_HeaderFile
+#ifndef Model_AttributeReference_H_
+#define Model_AttributeReference_H_
 
 #include "Model.h"
 #include "ModelAPI_AttributeReference.h"
+#include <ModelAPI_Feature.h>
 #include <TDF_Label.hxx>
 #include <TDF_Reference.hxx>
 
@@ -17,13 +18,17 @@
 
 class Model_AttributeReference : public ModelAPI_AttributeReference
 {
-  Handle_TDF_Reference myRef; ///< references to the feature label
-public:
-  /// Defines the feature referenced from this attribute
-  MODEL_EXPORT virtual void setValue(boost::shared_ptr<ModelAPI_Feature> theFeature);
+  Handle_TDF_Reference myRef;  ///< references to the feature label
+ public:
+  /// Defines the object referenced from this attribute
+  MODEL_EXPORT virtual void setValue(ObjectPtr theObject);
 
-  /// Returns feature referenced from this attribute
-  MODEL_EXPORT virtual boost::shared_ptr<ModelAPI_Feature> value();
+  /// Returns object referenced from this attribute
+  MODEL_EXPORT virtual ObjectPtr value();
+
+  MODEL_EXPORT ~Model_AttributeReference();
+
+  MODEL_EXPORT virtual void setObject(const boost::shared_ptr<ModelAPI_Object>& theObject);
 
 protected:
   /// Objects are created for features automatically