0031875: Draw Harness, ViewerTest - command vaspects -mostContinuity lacks g1 and...
[occt.git] / src / XCAFDoc / XCAFDoc_MaterialTool.cxx
1 // Copyright (c) 1999-2014 OPEN CASCADE SAS
2 //
3 // This file is part of Open CASCADE Technology software library.
4 //
5 // This library is free software; you can redistribute it and/or modify it under
6 // the terms of the GNU Lesser General Public License version 2.1 as published
7 // by the Free Software Foundation, with special exception defined in the file
8 // OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
9 // distribution for complete text of the license and disclaimer of any warranty.
10 //
11 // Alternatively, this file may be used under the terms of Open CASCADE
12 // commercial license or contractual agreement.
13
14 #include <XCAFDoc_MaterialTool.hxx>
15
16 #include <Standard_GUID.hxx>
17 #include <Standard_Type.hxx>
18 #include <TCollection_HAsciiString.hxx>
19 #include <TDataStd_Name.hxx>
20 #include <TDataStd_TreeNode.hxx>
21 #include <TDF_Attribute.hxx>
22 #include <TDF_ChildIDIterator.hxx>
23 #include <TDF_Label.hxx>
24 #include <TDF_RelocationTable.hxx>
25 #include <XCAFDoc.hxx>
26 #include <XCAFDoc_DocumentTool.hxx>
27 #include <XCAFDoc_Material.hxx>
28 #include <XCAFDoc_ShapeTool.hxx>
29
30 IMPLEMENT_DERIVED_ATTRIBUTE_WITH_TYPE(XCAFDoc_MaterialTool,TDataStd_GenericEmpty,"xcaf","MaterialTool")
31
32 //=======================================================================
33 //function : XCAFDoc_MaterialTool
34 //purpose  : 
35 //=======================================================================
36 XCAFDoc_MaterialTool::XCAFDoc_MaterialTool()
37 {
38 }
39
40
41 //=======================================================================
42 //function : Set
43 //purpose  : 
44 //=======================================================================
45
46 Handle(XCAFDoc_MaterialTool) XCAFDoc_MaterialTool::Set(const TDF_Label& L) 
47 {
48   Handle(XCAFDoc_MaterialTool) A;
49   if (!L.FindAttribute (XCAFDoc_MaterialTool::GetID(), A)) {
50     A = new XCAFDoc_MaterialTool ();
51     L.AddAttribute(A);
52     A->myShapeTool = XCAFDoc_DocumentTool::ShapeTool(L);
53   }
54   return A;
55 }
56
57
58 //=======================================================================
59 //function : GetID
60 //purpose  : 
61 //=======================================================================
62
63 const Standard_GUID& XCAFDoc_MaterialTool::GetID() 
64 {
65   static Standard_GUID MatTblID ("efd212f9-6dfd-11d4-b9c8-0060b0ee281b");
66   return MatTblID; 
67 }
68
69
70 //=======================================================================
71 //function : BaseLabel
72 //purpose  : 
73 //=======================================================================
74
75 TDF_Label XCAFDoc_MaterialTool::BaseLabel() const
76 {
77   return Label();
78 }
79
80
81 //=======================================================================
82 //function : ShapeTool
83 //purpose  : 
84 //=======================================================================
85
86 const Handle(XCAFDoc_ShapeTool)& XCAFDoc_MaterialTool::ShapeTool() 
87 {
88   if(myShapeTool.IsNull())
89     myShapeTool = XCAFDoc_DocumentTool::ShapeTool(Label());
90   return myShapeTool;
91 }
92
93
94 //=======================================================================
95 //function : IsMaterial
96 //purpose  : 
97 //=======================================================================
98
99 Standard_Boolean XCAFDoc_MaterialTool::IsMaterial(const TDF_Label& lab) const
100 {
101   Handle(XCAFDoc_Material) MatAttr;
102   if(lab.FindAttribute(XCAFDoc_Material::GetID(),MatAttr)) {
103     return Standard_True;
104   }
105   return Standard_False;
106 }
107
108
109 //=======================================================================
110 //function : GetMaterialLabels
111 //purpose  : 
112 //=======================================================================
113
114 void XCAFDoc_MaterialTool::GetMaterialLabels(TDF_LabelSequence& Labels) const
115 {
116   Labels.Clear();
117   TDF_ChildIterator ChildIterator( Label() ); 
118   for (; ChildIterator.More(); ChildIterator.Next()) {
119     TDF_Label L = ChildIterator.Value();
120     if ( IsMaterial(L)) Labels.Append(L);
121   }
122 }
123
124
125 //=======================================================================
126 //function : AddMaterial
127 //purpose  : 
128 //=======================================================================
129
130 TDF_Label XCAFDoc_MaterialTool::AddMaterial(const Handle(TCollection_HAsciiString)& aName,
131                                             const Handle(TCollection_HAsciiString)& aDescription,
132                                             const Standard_Real aDensity,
133                                             const Handle(TCollection_HAsciiString)& aDensName,
134                                             const Handle(TCollection_HAsciiString)& aDensValType) const
135 {
136   TDF_Label MatL;
137   TDF_TagSource aTag;
138   MatL = aTag.NewChild ( Label() );
139   XCAFDoc_Material::Set(MatL,aName,aDescription,aDensity,aDensName,aDensValType);
140   TDataStd_Name::Set(MatL,TCollection_AsciiString(aName->ToCString()));
141   return MatL;
142 }
143
144
145 //=======================================================================
146 //function : SetMaterial
147 //purpose  : 
148 //=======================================================================
149
150 void XCAFDoc_MaterialTool::SetMaterial(const TDF_Label& L,const TDF_Label& MatL) const
151 {
152   // set reference
153   Handle(TDataStd_TreeNode) refNode, mainNode;
154   mainNode = TDataStd_TreeNode::Set ( MatL, XCAFDoc::MaterialRefGUID() );
155   refNode  = TDataStd_TreeNode::Set ( L,       XCAFDoc::MaterialRefGUID() );
156   refNode->Remove(); // abv: fix against bug in TreeNode::Append()
157   mainNode->Append(refNode);
158 }
159
160
161 //=======================================================================
162 //function : SetMaterial
163 //purpose  : 
164 //=======================================================================
165
166 void XCAFDoc_MaterialTool::SetMaterial(const TDF_Label& L,
167                                        const Handle(TCollection_HAsciiString)& aName,
168                                        const Handle(TCollection_HAsciiString)& aDescription,
169                                        const Standard_Real aDensity,
170                                        const Handle(TCollection_HAsciiString)& aDensName,
171                                        const Handle(TCollection_HAsciiString)& aDensValType) const
172 {
173   TDF_Label MatL = AddMaterial(aName,aDescription,aDensity,aDensName,aDensValType);
174   SetMaterial(L,MatL);
175 }
176
177
178 //=======================================================================
179 //function : GetMaterial
180 //purpose  : 
181 //=======================================================================
182
183 Standard_Boolean XCAFDoc_MaterialTool::GetMaterial(const TDF_Label& MatL,
184                                                    Handle(TCollection_HAsciiString)& aName,
185                                                    Handle(TCollection_HAsciiString)& aDescription,
186                                                    Standard_Real& aDensity,
187                                                    Handle(TCollection_HAsciiString)& aDensName,
188                                                    Handle(TCollection_HAsciiString)& aDensValType) const
189 {
190   Handle(XCAFDoc_Material) MatAttr;
191   if(!MatL.FindAttribute(XCAFDoc_Material::GetID(),MatAttr)) {
192     return Standard_False;
193   }
194   aName = MatAttr->GetName();
195   aDescription = MatAttr->GetDescription();
196   aDensity = MatAttr->GetDensity();
197   aDensName = MatAttr->GetDensName();
198   aDensValType = MatAttr->GetDensValType();
199   
200   return Standard_True;
201 }
202
203
204 //=======================================================================
205 //function : GetDensityForShape
206 //purpose  : 
207 //=======================================================================
208
209 Standard_Real XCAFDoc_MaterialTool::GetDensityForShape(const TDF_Label& ShapeL)
210 {
211   Standard_Real Dens=0.0;
212   Handle(TDataStd_TreeNode) Node;
213   if ( ! ShapeL.FindAttribute ( XCAFDoc::MaterialRefGUID(), Node) ||
214        ! Node->HasFather() ) return Dens;
215   TDF_Label MatL = Node->Father()->Label();
216   Handle(XCAFDoc_Material) MatAttr;
217   if(!MatL.FindAttribute(XCAFDoc_Material::GetID(),MatAttr)) {
218     return Dens;
219   }
220   // defaul dimension fo density - gram/sm^3
221   // we transfer "sm" into "mm"
222   Dens = MatAttr->GetDensity()*0.001; 
223   return Dens;
224 }
225
226
227 //=======================================================================
228 //function : ID
229 //purpose  : 
230 //=======================================================================
231
232 const Standard_GUID& XCAFDoc_MaterialTool::ID() const
233 {
234   return GetID();
235 }
236
237 //=======================================================================
238 //function : DumpJson
239 //purpose  : 
240 //=======================================================================
241 void XCAFDoc_MaterialTool::DumpJson (Standard_OStream& theOStream, Standard_Integer theDepth) const
242 {
243   OCCT_DUMP_TRANSIENT_CLASS_BEGIN (theOStream)
244
245   OCCT_DUMP_BASE_CLASS (theOStream, theDepth, TDF_Attribute)
246
247   OCCT_DUMP_FIELD_VALUES_DUMPED (theOStream, theDepth, myShapeTool.get())
248 }