src/Model/Model_AttributeRefList.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 // File:        ModelAPI_AttributeRefList.cxx
   4 // Created:     8 May 2014
   5 // Author:      Mikhail PONIKAROV
   6
   7 #include "Model_AttributeRefList.h"
   8 #include "Model_Application.h"
   9 #include "Model_Data.h"
  10 #include <ModelAPI_Feature.h>
  11 #include <TDF_ListIteratorOfLabelList.hxx>
  12
  13 using namespace std;
  14
  15 void Model_AttributeRefList::append(ObjectPtr theObject)
  16 {
  17   std::shared_ptr<Model_Data> aData = std::dynamic_pointer_cast<Model_Data>(theObject->data());
  18   myRef->Append(aData->label().Father());  // store label of the object
  19   // do it before the transaction finish to make just created/removed objects know dependencies
  20   // and reference from composite feature is removed automatically
  21   ADD_BACK_REF(theObject);
  22
  23   owner()->data()->sendAttributeUpdated(this);
  24 }
  25
  26 void Model_AttributeRefList::remove(ObjectPtr theObject)
  27 {
  28   std::shared_ptr<Model_Data> aData;
  29   if (theObject.get() != NULL) {
  30     aData = std::dynamic_pointer_cast<Model_Data>(theObject->data());
  31     myRef->Remove(aData->label().Father());
  32     REMOVE_BACK_REF(theObject);
  33   }
  34   else { // in case of empty object remove, the first empty object is removed from the list
  35     std::shared_ptr<Model_Document> aDoc = std::dynamic_pointer_cast<Model_Document>(
  36         owner()->document());
  37     if (aDoc) {
  38       const TDF_LabelList& aList = myRef->List();
  39       for (TDF_ListIteratorOfLabelList aLIter(aList); aLIter.More(); aLIter.Next()) {
  40         ObjectPtr anObj = aDoc->object(aLIter.Value());
  41         if (anObj.get() == NULL) {
  42           myRef->Remove(aLIter.Value());
  43           REMOVE_BACK_REF(theObject);
  44           break;
  45         }
  46       }
  47     }
  48   }
  49   owner()->data()->sendAttributeUpdated(this);
  50 }
  51
  52 int Model_AttributeRefList::size() const
  53 {
  54   return myRef->Extent();
  55 }
  56
  57 list<ObjectPtr> Model_AttributeRefList::list()
  58 {
  59   std::list<ObjectPtr> aResult;
  60   std::shared_ptr<Model_Document> aDoc = std::dynamic_pointer_cast<Model_Document>(
  61       owner()->document());
  62   if (aDoc) {
  63     const TDF_LabelList& aList = myRef->List();
  64     for (TDF_ListIteratorOfLabelList aLIter(aList); aLIter.More(); aLIter.Next()) {
  65       ObjectPtr anObj = aDoc->object(aLIter.Value());
  66       aResult.push_back(anObj);
  67     }
  68   }
  69   return aResult;
  70 }
  71
  72 ObjectPtr Model_AttributeRefList::object(const int theIndex) const
  73 {
  74   std::shared_ptr<Model_Document> aDoc = std::dynamic_pointer_cast<Model_Document>(
  75       owner()->document());
  76   if (aDoc) {
  77     const TDF_LabelList& aList = myRef->List();
  78     int anIndex = 0;
  79     for (TDF_ListIteratorOfLabelList aLIter(aList); aLIter.More(); aLIter.Next(), anIndex++) {
  80       if (anIndex == theIndex)
  81         return aDoc->object(aLIter.Value());
  82     }
  83   }
  84   return ObjectPtr();
  85 }
  86
  87 Model_AttributeRefList::Model_AttributeRefList(TDF_Label& theLabel)
  88 {
  89   myIsInitialized = theLabel.FindAttribute(TDataStd_ReferenceList::GetID(), myRef) == Standard_True;
  90   if (!myIsInitialized) {
  91     myRef = TDataStd_ReferenceList::Set(theLabel);
  92   }
  93 }