Fix for removing sketch and sub-components not correct history position was defined

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

diff --git a/src/Model/Model_AttributeDocRef.h b/src/Model/Model_AttributeDocRef.h
index 778ed316bd0eceda4ef02932193287ca0b35bfb6..3a4638a69c69588cbd0013cd73821c4f06e90eff 100644 (file)
--- a/src/Model/Model_AttributeDocRef.h
+++ b/src/Model/Model_AttributeDocRef.h
@@ -1,9 +1,11 @@
+// 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"
@@ -17,15 +19,16 @@
 
 class Model_AttributeDocRef : public ModelAPI_AttributeDocRef
 {
-  Handle_TDataStd_Comment myComment; ///< reference to document is identified as string-id
-public:
+  Handle_TDataStd_Comment myComment;  ///< reference to document is identified as string-id
+  std::shared_ptr<ModelAPI_Document> myDoc; ///< document referenced by this attribute (if already loaded)
+ public:
   /// Defines the document referenced from this attribute
   MODEL_EXPORT virtual void setValue(std::shared_ptr<ModelAPI_Document> theDoc);
 
   /// Returns document referenced from this attribute
   MODEL_EXPORT virtual std::shared_ptr<ModelAPI_Document> value();
 
-protected:
+ protected:
   /// Initializes attibutes
   Model_AttributeDocRef(TDF_Label& theLabel);