0029018: Documentation - Provide user guide for Qt browser
[occt.git] / 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 }