| 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 | } |