Some minor changes in tests. Now the initialization plugin performs correctly in...

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

diff --git a/src/Model/Model_AttributeDocRef.h b/src/Model/Model_AttributeDocRef.h
index 2218557242fc5931548c5350ec8240f30f723465..04eacc4372978ddebb19e8e1f2f78c2fb55e2709 100644 (file)
--- a/src/Model/Model_AttributeDocRef.h
+++ b/src/Model/Model_AttributeDocRef.h
@@ -1,13 +1,15 @@
+// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
+
 // File:        Model_AttributeDocRef.h
 // Created:     2 Apr 2014
 // Author:      Mikhail PONIKAROV
 
-#ifndef Model_AttributeDocRef_HeaderFile
-#define Model_AttributeDocRef_HeaderFile
+#ifndef Model_AttributeDocRef_H_
+#define Model_AttributeDocRef_H_
 
 #include "Model.h"
 #include "ModelAPI_AttributeDocRef.h"
-#include <TDataStd_Comment.hxx>
+#include <TDataStd_Integer.hxx>
 #include <TDF_Label.hxx>
 
 /**\class Model_AttributeDocRef
@@ -15,21 +17,25 @@
  * \brief Attribute that contains reference to another document.
  */
 
-class MODEL_EXPORT Model_AttributeDocRef : public ModelAPI_AttributeDocRef
+class Model_AttributeDocRef : public ModelAPI_AttributeDocRef
 {
-  Handle_TDataStd_Comment myComment; ///< reference to document is identified as string-id
-public:
+  Handle(TDataStd_Integer) myID; ///< persistent and unique identifier of the document in the application
+ public:
+
   /// Defines the document referenced from this attribute
-  virtual void setValue(std::shared_ptr<ModelAPI_Document> theDoc);
+  MODEL_EXPORT virtual void setValue(std::shared_ptr<ModelAPI_Document> theDoc);
 
   /// Returns document referenced from this attribute
-  virtual std::shared_ptr<ModelAPI_Document> value();
+  MODEL_EXPORT virtual std::shared_ptr<ModelAPI_Document> value();
+
+  /// Returns the persisten ID of the document
+  MODEL_EXPORT virtual int docId();
 
-protected:
+ protected:
   /// Initializes attibutes
   Model_AttributeDocRef(TDF_Label& theLabel);
 
-  friend class Model_Object;
+  friend class Model_Data;
 };
 
 #endif