0030268: Inspectors - improvements in VInspector plugin
[occt.git] / tools / ShapeView / ShapeView_TreeModel.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/ShapeView_TreeModel.hxx>
17 #include <inspector/ShapeView_ItemRoot.hxx>
18 #include <inspector/ShapeView_TreeModel.hxx>
19 #include <inspector/ShapeView_ItemRoot.hxx>
20 #include <inspector/ShapeView_ItemShape.hxx>
21
22 // =======================================================================
23 // function : Constructor
24 // purpose :
25 // =======================================================================
26 ShapeView_TreeModel::ShapeView_TreeModel (QObject* theParent)
27 : TreeModel_ModelBase (theParent)
28 {
29 }
30
31 // =======================================================================
32 // function : createRootItem
33 // purpose :
34 // =======================================================================
35 TreeModel_ItemBasePtr ShapeView_TreeModel::createRootItem (const int theColumnId)
36 {
37   return ShapeView_ItemRoot::CreateItem (TreeModel_ItemBasePtr(), 0, theColumnId);
38 }
39
40 // =======================================================================
41 // function : AddShape
42 // purpose :
43 // =======================================================================
44 void ShapeView_TreeModel::AddShape(const TopoDS_Shape& theShape)
45 {
46   for (int aColId = 0, aNbColumns = columnCount(); aColId < aNbColumns; aColId++)
47   {
48     ShapeView_ItemRootPtr aRootItem = itemDynamicCast<ShapeView_ItemRoot>(RootItem (aColId));
49     aRootItem->AddShape(theShape);
50   }
51
52   Reset();
53   EmitLayoutChanged();
54 }
55
56 // =======================================================================
57 // function : RemoveAllShapes
58 // purpose :
59 // =======================================================================
60 void ShapeView_TreeModel::RemoveAllShapes()
61 {
62   for (int aColId = 0, aNbColumns = columnCount(); aColId < aNbColumns; aColId++)
63   {
64     ShapeView_ItemRootPtr aRootItem = itemDynamicCast<ShapeView_ItemRoot>(RootItem (aColId));
65     aRootItem->RemoveAllShapes();
66   }
67   Reset();
68   EmitLayoutChanged();
69 }
70
71 // =======================================================================
72 // function : FindIndex
73 // purpose :
74 // =======================================================================
75 QModelIndex ShapeView_TreeModel::FindIndex (const TopoDS_Shape& theShape) const
76 {
77   QModelIndex aParentIndex = index (0, 0);
78   TreeModel_ItemBasePtr aParentItem = TreeModel_ModelBase::GetItemByIndex (aParentIndex); // application item
79   for (int aChildId = 0, aCount = aParentItem->rowCount(); aChildId < aCount; aChildId++)
80   {
81     QModelIndex anIndex = index (aChildId, 0, aParentIndex);
82     ShapeView_ItemShapePtr anItem = itemDynamicCast<ShapeView_ItemShape> (TreeModel_ModelBase::GetItemByIndex (anIndex));
83     if (anItem && anItem->GetItemShape() == theShape)
84       return anIndex;
85   }
86   return QModelIndex();
87 }