Additional fix for the issue #1876 : now compsolid result is also splitted when selec...

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

diff --git a/src/Model/Model_AttributeReference.h b/src/Model/Model_AttributeReference.h
index e8f9c9997f45169bf31a51572817c839fc189180..ea9b721f527b64c4d4ef78d357a1f009e0fd4a24 100644 (file)
--- a/src/Model/Model_AttributeReference.h
+++ b/src/Model/Model_AttributeReference.h
@@ -1,3 +1,5 @@
+// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
+
 // File:        Model_AttributeReference.h
 // Created:     8 May 2014
 // Author:      Mikhail PONIKAROV
@@ -18,6 +20,7 @@
 
 class Model_AttributeReference : public ModelAPI_AttributeReference
 {
+  TDF_Label myLab; ///< the main label of this attribute
   Handle_TDF_Reference myRef;  ///< references to the feature label
  public:
   /// Defines the object referenced from this attribute
@@ -28,11 +31,20 @@ class Model_AttributeReference : public ModelAPI_AttributeReference
 
   MODEL_EXPORT ~Model_AttributeReference();
 
+  MODEL_EXPORT virtual void setObject(const std::shared_ptr<ModelAPI_Object>& theObject);
+
+  /// Returns true if attribute was  initialized by some value
+  MODEL_EXPORT virtual bool isInitialized();
+
+
 protected:
   /// Objects are created for features automatically
   MODEL_EXPORT Model_AttributeReference(TDF_Label& theLabel);
+  /// Reinitializes the internal state of the attribute (may be needed on undo/redo, abort, etc)
+  virtual void reinit();
 
   friend class Model_Data;
+  friend class Model_AttributeSelection;
 };
 
 #endif