X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FModel%2FModel_AttributeReference.h;h=991f4e725544ae90283cad4ecd618c58bbeba063;hb=f367061802012d6909c2619ef6b6c4cba86fc503;hp=6406d0bfbfbf0fd8b828ab4fc0dca034c39423ec;hpb=f581964034f8df715c46a07c5ecb762492d6b4cf;p=modules%2Fshaper.git

diff --git a/src/Model/Model_AttributeReference.h b/src/Model/Model_AttributeReference.h
index 6406d0bfb..991f4e725 100644
--- a/src/Model/Model_AttributeReference.h
+++ b/src/Model/Model_AttributeReference.h
@@ -1,9 +1,11 @@
+// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
+
 // File:        Model_AttributeReference.h
 // 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"
@@ -18,19 +20,24 @@
 
 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(FeaturePtr 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 object referenced from this attribute
+  MODEL_EXPORT virtual ObjectPtr value();
+
+  MODEL_EXPORT ~Model_AttributeReference();
 
-  /// Returns feature referenced from this attribute
-  MODEL_EXPORT virtual FeaturePtr value();
+  MODEL_EXPORT virtual void setObject(const std::shared_ptr<ModelAPI_Object>& theObject);
 
 protected:
   /// Objects are created for features automatically
   MODEL_EXPORT Model_AttributeReference(TDF_Label& theLabel);
 
   friend class Model_Data;
+  friend class Model_AttributeSelection;
 };
 
 #endif