0024023: Revamp the OCCT Handle -- downcast (automatic)
[occt.git] / src / TDF / TDF_ClosureTool.cxx
1 // Created by: DAUTRY Philippe
2 // Copyright (c) 1998-1999 Matra Datavision
3 // Copyright (c) 1999-2014 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 //              -------------------
17
18 // Version:     0.0
19 //Version       Date            Purpose
20 //              0.0     Sep  8 1998     Creation
21
22
23 #include <TDF_ClosureTool.ixx>
24
25 //#include <TDF_Reference.hxx>
26 #include <TDF_Attribute.hxx>
27 #include <TDF_AttributeIterator.hxx>
28 #include <TDF_AttributeMap.hxx>
29 #include <TDF_ChildIterator.hxx>
30 #include <TDF_Label.hxx>
31 #include <TDF_LabelMap.hxx>
32 #include <TDF_ListIteratorOfLabelList.hxx>
33 #include <TDF_MapIteratorOfAttributeMap.hxx>
34 #include <TDF_MapIteratorOfLabelMap.hxx>
35
36 #include <Standard_TypeMismatch.hxx>
37
38 #define DeclareAndSpeedCast(V,T,Vdown) Handle(T) Vdown = Handle(T)::DownCast (V)
39 #define DeclareConstAndSpeedCast(V,T,Vdown) const Handle(T)& Vdown = (Handle(T)&) V
40 #define SpeedCast(V,T,Vdown) Vdown = Handle(T)::DownCast (V)
41
42
43 //=======================================================================
44 //function : Closure
45 //purpose  : Builds the transitive closure whithout attribute filter.
46 //=======================================================================
47
48 void TDF_ClosureTool::Closure
49 (const Handle(TDF_DataSet)& aDataSet) 
50 {
51   TDF_IDFilter Filter(Standard_False); // "Keep all"
52   TDF_ClosureMode Mode; // All modes are set to true.
53   TDF_ClosureTool::Closure(aDataSet, Filter, Mode);
54 }
55
56
57 //=======================================================================
58 //function : Closure
59 //purpose  : Builds the transitive closure with an attribute filter.
60 //=======================================================================
61
62 void TDF_ClosureTool::Closure
63 (const Handle(TDF_DataSet)& aDataSet,
64  const TDF_IDFilter& aFilter,
65  const TDF_ClosureMode& aMode) 
66 {
67   TDF_LabelMap&     labMap = aDataSet->Labels();
68   TDF_AttributeMap& attMap = aDataSet->Attributes();
69   TDF_LabelList&    rootLst  = aDataSet->Roots();
70
71   // Memorizes the roots for future uses.
72   rootLst.Clear();
73   TDF_MapIteratorOfLabelMap labMItr(labMap);
74   for (; labMItr.More(); labMItr.Next()) rootLst.Append(labMItr.Key());
75
76   // Iterates on roots.
77   TDF_ListIteratorOfLabelList labLItr(rootLst);
78   for (; labLItr.More(); labLItr.Next()) {
79     const TDF_Label& lab = labLItr.Value();
80     if (lab.HasAttribute())
81       TDF_ClosureTool::LabelAttributes(lab,labMap,attMap,aFilter,aMode);
82     TDF_ClosureTool::Closure(lab,labMap,attMap,aFilter,aMode); 
83   }
84 }
85
86
87 //=======================================================================
88 //function : Closure
89 //purpose  : Internal closure method.
90 //=======================================================================
91
92 void TDF_ClosureTool::Closure
93 (const TDF_Label& aLabel,
94  TDF_LabelMap& aLabMap,
95  TDF_AttributeMap& anAttMap,
96  const TDF_IDFilter& aFilter,
97  const TDF_ClosureMode& aMode) 
98 {
99   TDF_Label upLab;
100   for (TDF_ChildIterator childItr(aLabel,Standard_True);
101        childItr.More();childItr.Next()){
102     const TDF_Label& locLab = childItr.Value();
103     // On ne peut faire cette optimisation car il faudrait d'abord
104     // qu'aucun label donne comme Root ne soit fils d'un autre label root!
105     if (locLab.HasAttribute()) { // && aLabMap.Add(locLab)) {
106       aLabMap.Add(locLab);
107       upLab = locLab.Father();
108       while (aLabMap.Add(upLab)) upLab = upLab.Father();
109       TDF_ClosureTool::LabelAttributes(locLab,aLabMap,anAttMap,aFilter,aMode);
110     }
111   }
112 }
113
114
115
116
117 //=======================================================================
118 //function : LabelAttributes
119 //purpose  : Internal method: adds the attributes to <aDataSet>.
120 //=======================================================================
121
122 void TDF_ClosureTool::LabelAttributes
123 (const TDF_Label& aLabel,
124  TDF_LabelMap& aLabMap,
125  TDF_AttributeMap& anAttMap,
126  const TDF_IDFilter& aFilter,
127  const TDF_ClosureMode& aMode) 
128 {
129   Handle(TDF_DataSet) tmpDataSet;
130   Standard_Boolean BindLabel;
131   TDF_MapIteratorOfAttributeMap attMItr;
132   TDF_MapIteratorOfLabelMap labMItr;
133
134   // Attributes directly attached to the label.
135   for (TDF_AttributeIterator attItr(aLabel); attItr.More(); attItr.Next()) {
136     const Handle(TDF_Attribute) locAtt1 = attItr.Value();
137     if (aFilter.IsKept(locAtt1)) {
138       if (anAttMap.Add(locAtt1)) {
139         // locAtt1 not yet in the map.
140
141         // Labels & Attributes referenced by the attribute.
142         tmpDataSet = new TDF_DataSet();
143         if (aMode.References()) {
144           // 1 - The referenced attributes
145           // 1.1 - A referenced attribute has a label : adds the label;
146           // 1.2 - A referenced attribute has no label : adds the attribute;
147           // 2 - Adds the referenced labels.
148
149           locAtt1->References(tmpDataSet);
150
151           // 1 - The referenced attributes
152           const TDF_AttributeMap& tmpAttMap = tmpDataSet->Attributes();
153           for (attMItr.Initialize(tmpAttMap);
154                attMItr.More(); attMItr.Next()) {
155             const Handle(TDF_Attribute)& locAtt2 = attMItr.Key();
156             BindLabel = Standard_False;
157             if (!locAtt2.IsNull()) {
158               const TDF_Label& locLab2 = locAtt2->Label();
159               BindLabel = !locLab2.IsNull();
160               if (BindLabel) {
161                 // 1.1 - A referenced attribute has a label.
162                 if (aLabMap.Add(locLab2))
163                   TDF_ClosureTool::Closure(locLab2,
164                                            aLabMap,anAttMap,aFilter,aMode);
165               }
166               else {
167                 // 1.2 - A referenced attribute has no label.
168                 // We suppose locAtt2 has no referenced attribute itself.
169                 anAttMap.Add(locAtt2);
170               }
171             }
172           }
173
174           // 2 - Adds the referenced labels.
175           const TDF_LabelMap& tmpLabMap = tmpDataSet->Labels();
176           for (labMItr.Initialize(tmpLabMap);
177                labMItr.More(); labMItr.Next()) {
178             const TDF_Label& locLab1 = labMItr.Key();
179             if (aLabMap.Add(locLab1))
180               TDF_ClosureTool::Closure(locLab1,
181                                        aLabMap,anAttMap,aFilter,aMode);
182           }
183         }
184       }
185     }
186   }
187 }