tools/DFBrowserPane/DFBrowserPane_TDataStdTreeNodeModel.cxx

   1 // Created on: 2017-06-16
   2 // Created by: Natalia ERMOLAEVA
   3 // Copyright (c) 2017 OPEN CASCADE SAS
   4 //
   5 // This file is part of Open CASCADE Technology software library.
   6 //
   7 // This library is free software; you can redistribute it and/or modify it under
   8 // the terms of the GNU Lesser General Public License version 2.1 as published
   9 // by the Free Software Foundation, with special exception defined in the file
  10 // OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
  11 // distribution for complete text of the license and disclaimer of any warranty.
  12 //
  13 // Alternatively, this file may be used under the terms of Open CASCADE
  14 // commercial license or contractual agreement.
  15
  16 #include <inspector/DFBrowserPane_TDataStdTreeNodeModel.hxx>
  17 #include <inspector/DFBrowserPane_TDataStdTreeNodeItem.hxx>
  18
  19 #include <TDataStd_TreeNode.hxx>
  20
  21 #include <QAbstractItemModel>
  22
  23 // =======================================================================
  24 // function : Constructor
  25 // purpose :
  26 // =======================================================================
  27 DFBrowserPane_TDataStdTreeNodeModel::DFBrowserPane_TDataStdTreeNodeModel (QObject* theParent)
  28 : TreeModel_ModelBase (theParent)
  29 {
  30   m_pRootItem = DFBrowserPane_TDataStdTreeNodeItem::CreateItem (TreeModel_ItemBasePtr(), 0, 0);
  31 }
  32
  33 // =======================================================================
  34 // function : SetAttribute
  35 // purpose :
  36 // =======================================================================
  37 void DFBrowserPane_TDataStdTreeNodeModel::SetAttribute (const Handle(TDF_Attribute)& theAttribute)
  38 {
  39   DFBrowserPane_TDataStdTreeNodeItemPtr aRootItem = itemDynamicCast<DFBrowserPane_TDataStdTreeNodeItem>(m_pRootItem);
  40   Reset();
  41   aRootItem->SetAttribute (theAttribute);
  42   EmitLayoutChanged();
  43 }
  44
  45 // =======================================================================
  46 // function : FindIndex
  47 // purpose :
  48 // =======================================================================
  49 QModelIndex DFBrowserPane_TDataStdTreeNodeModel::FindIndex (const Handle(TDF_Attribute)& theAttribute,
  50                                                             const QModelIndex theParentIndex)
  51 {
  52   QModelIndex aParentIndex = theParentIndex;
  53
  54   if (!aParentIndex.isValid())
  55     aParentIndex = index (0, 0);
  56
  57   DFBrowserPane_TDataStdTreeNodeItemPtr aParentItem = itemDynamicCast<DFBrowserPane_TDataStdTreeNodeItem>
  58     (TreeModel_ModelBase::GetItemByIndex (aParentIndex));
  59
  60   if (aParentItem->GetAttribute() == theAttribute)
  61     return aParentIndex;
  62
  63   for (int aChildId = 0, aCount = aParentItem->rowCount(); aChildId < aCount; aChildId++)
  64   {
  65     QModelIndex anIndex = index (aChildId, 0, aParentIndex);
  66     TreeModel_ItemBasePtr anItemBase = TreeModel_ModelBase::GetItemByIndex (anIndex);
  67     DFBrowserPane_TDataStdTreeNodeItemPtr anItem = itemDynamicCast<DFBrowserPane_TDataStdTreeNodeItem>(anItemBase);
  68
  69     if (anItem->GetAttribute() == theAttribute)
  70       return anIndex;
  71
  72     QModelIndex aSubIndex = FindIndex (theAttribute, anIndex);
  73     if (aSubIndex.isValid())
  74       return aSubIndex;
  75   }
  76   return QModelIndex();
  77 }