0029018: Documentation - Provide user guide for Qt browser
[occt.git] / tools / DFBrowserPane / DFBrowserPane_TDataStdTreeNode.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_TDataStdTreeNode.hxx>
17
18 #include <inspector/DFBrowserPane_AttributePaneModel.hxx>
19 #include <inspector/DFBrowserPane_TableView.hxx>
20 #include <inspector/DFBrowserPane_TDataStdTreeNodeItem.hxx>
21 #include <inspector/DFBrowserPane_TDataStdTreeNodeModel.hxx>
22 #include <inspector/DFBrowserPane_Tools.hxx>
23
24 #include <TDataStd_TreeNode.hxx>
25
26 #include <QHeaderView>
27 #include <QVariant>
28 #include <QTableView>
29 #include <QTreeView>
30 #include <QVBoxLayout>
31 #include <QWidget>
32 #include <QItemSelectionModel>
33
34 // =======================================================================
35 // function : Constructor
36 // purpose :
37 // =======================================================================
38 DFBrowserPane_TDataStdTreeNode::DFBrowserPane_TDataStdTreeNode()
39 : DFBrowserPane_AttributePane(), myTreeNodeView (0)
40 {
41   myModel = new DFBrowserPane_TDataStdTreeNodeModel (0);
42   mySelectionModels.clear(); // do not use selection model of parent pane
43   mySelectionModels.push_back (new QItemSelectionModel (myModel));
44 }
45
46 // =======================================================================
47 // function : CreateWidget
48 // purpose :
49 // =======================================================================
50 QWidget* DFBrowserPane_TDataStdTreeNode::CreateWidget (QWidget* theParent)
51 {
52   QWidget* aMainWidget = new QWidget (theParent);
53   aMainWidget->setVisible (false);
54
55   myTableView = new DFBrowserPane_TableView (aMainWidget, getTableColumnWidths());
56   myTableView->SetVisibleHorizontalHeader (false);
57   DFBrowserPane_TableView::SetFixedRowCount (1, myTableView->GetTableView());
58   myTableView->SetModel (myPaneModel);
59
60   QVBoxLayout* aLay = new QVBoxLayout (aMainWidget);
61   aLay->setContentsMargins (0, 0, 0, 0);
62   aLay->addWidget (myTableView);
63
64   myTreeNodeView = new QTreeView (theParent);
65   myTreeNodeView->header()->setVisible (false);
66   myTreeNodeView->setModel (myModel);
67   myTreeNodeView->setSelectionModel (mySelectionModels.front());
68   myTreeNodeView->setSelectionBehavior (QAbstractItemView::SelectRows);
69
70   aLay->addWidget (myTreeNodeView);
71   aLay->setStretch (1, 1);
72   return aMainWidget;
73 }
74
75 // =======================================================================
76 // function : Init
77 // purpose :
78 // =======================================================================
79 void DFBrowserPane_TDataStdTreeNode::Init (const Handle(TDF_Attribute)& theAttribute)
80 {
81   Handle(TDataStd_TreeNode) aTreeNode = Handle(TDataStd_TreeNode)::DownCast (theAttribute);
82
83   bool aDefaultGUID = aTreeNode->ID() != aTreeNode->GetDefaultTreeID();
84   if (myTableView)
85     myTableView->setVisible (!aDefaultGUID);
86
87   if (!aDefaultGUID)
88   {
89     QList<QVariant> aValues;
90     char aStr[256];
91     aTreeNode->ID().ToCString (aStr);
92     TCollection_AsciiString aString(aStr);
93     aValues << "GetDefaultTreeID" << DFBrowserPane_Tools::ToString (aString);
94     getPaneModel()->Init (aValues);
95     if (myTableView)
96       myTableView->GetTableView()->resizeColumnToContents (0);
97   }
98
99
100   DFBrowserPane_TDataStdTreeNodeModel* aModel = dynamic_cast<DFBrowserPane_TDataStdTreeNodeModel*> (myModel);
101   aModel->Reset();
102
103   if (!aTreeNode.IsNull())
104   {
105     Handle(TDataStd_TreeNode) aRootItem = aTreeNode->Root();
106     aModel->SetAttribute (aRootItem);
107
108     QModelIndex anIndex = aModel->FindIndex (theAttribute, QModelIndex());
109     if (myTreeNodeView && anIndex.isValid())
110     {
111       myTreeNodeView->setExpanded (anIndex.parent(), true);
112       myTreeNodeView->scrollTo (anIndex);
113
114       TreeModel_ItemBasePtr anAttributeItem = TreeModel_ModelBase::GetItemByIndex (anIndex);
115       DFBrowserPane_TDataStdTreeNodeItemPtr anAttributeNodeItem =
116               itemDynamicCast<DFBrowserPane_TDataStdTreeNodeItem>(anAttributeItem);
117       anAttributeNodeItem->setCurrentAttribute (true);
118     }
119   }
120   aModel->EmitLayoutChanged();
121 }
122
123 // =======================================================================
124 // function : GetShortAttributeInfo
125 // purpose :
126 // =======================================================================
127 void DFBrowserPane_TDataStdTreeNode::GetShortAttributeInfo (const Handle(TDF_Attribute)& theAttribute,
128                                                             QList<QVariant>& theValues)
129 {
130   Handle(TDataStd_TreeNode) aTNAttribute = Handle(TDataStd_TreeNode)::DownCast (theAttribute);
131   bool aDefaultGUID = aTNAttribute->ID() != aTNAttribute->GetDefaultTreeID();
132   QString aGUIDPrefix;
133   if (!aDefaultGUID) {
134     QList<QVariant> aValues;
135     char aStr[256];
136     aTNAttribute->ID().ToCString (aStr);
137     TCollection_AsciiString aString(aStr);
138     aGUIDPrefix = DFBrowserPane_Tools::ToString(aString);
139   }
140
141   if (aTNAttribute->HasFather())
142   {
143     TDF_Label aLabel = aTNAttribute->Father()->Label();
144     theValues.append (QString ("%1 ==> %2").arg(aGUIDPrefix).arg (DFBrowserPane_Tools::GetEntry (aLabel).ToCString()));
145   }
146   else 
147   {
148     Handle(TDataStd_TreeNode) aFirstChild = aTNAttribute->First();
149     QStringList aRefs;
150     while (! aFirstChild.IsNull() ) 
151     {
152       TDF_Label aLabel = aFirstChild->Label();
153       aRefs.append (DFBrowserPane_Tools::GetEntry (aLabel).ToCString());
154       aFirstChild = aFirstChild->Next();
155     }
156     theValues.append (QString ("%1 <== (%2)").arg(aGUIDPrefix).arg (aRefs.join (", ")));
157   }
158 }
159
160 // =======================================================================
161 // function : GetReferences
162 // purpose :
163 // =======================================================================
164 void DFBrowserPane_TDataStdTreeNode::GetReferences (const Handle(TDF_Attribute)& theAttribute,
165                                                     NCollection_List<TDF_Label>& theRefLabels,
166                                                     Handle(Standard_Transient)&)
167 {
168   Handle(TDataStd_TreeNode) anAttribute = Handle(TDataStd_TreeNode)::DownCast (theAttribute);
169   if (anAttribute.IsNull())
170     return;
171
172   QModelIndexList aSelectedIndices = myTreeNodeView->selectionModel()->selectedIndexes();
173   for (int aSelectedId = 0, aSize = aSelectedIndices.size(); aSelectedId < aSize; aSelectedId++)
174   {
175     QModelIndex anIndex = aSelectedIndices[aSelectedId];
176
177     TreeModel_ItemBasePtr anAttributeItem = TreeModel_ModelBase::GetItemByIndex (anIndex);
178     DFBrowserPane_TDataStdTreeNodeItemPtr anAttributeNodeItem =
179               itemDynamicCast<DFBrowserPane_TDataStdTreeNodeItem>(anAttributeItem);
180
181     Handle(TDF_Attribute) aNodeAttribute = anAttributeNodeItem->GetAttribute();
182     if (aNodeAttribute.IsNull())
183       continue;
184
185     theRefLabels.Append (aNodeAttribute->Label());
186   }
187 }