Download src/Mod/Import/Gui/OCAFBrowser.cpp from AbdulElahGwaith/FreeCAD: direct link, hf CLI and curl.
- Browser
- Download file 12.4 kB
-
https://huggingface.co/AbdulElahGwaith/FreeCAD/resolve/main/src/Mod/Import/Gui/OCAFBrowser.cpp
- Command line
-
hf download hf://AbdulElahGwaith/FreeCAD/src/Mod/Import/Gui/OCAFBrowser.cpp
-
curl -L -o OCAFBrowser.cpp https://huggingface.co/AbdulElahGwaith/FreeCAD/resolve/main/src/Mod/Import/Gui/OCAFBrowser.cpp
12.4 kB
| // SPDX-License-Identifier: LGPL-2.1-or-later | |
| /*************************************************************************** | |
| * Copyright (c) 2023 Werner Mayer <wmayer[at]users.sourceforge.net> * | |
| * * | |
| * This file is part of FreeCAD. * | |
| * * | |
| * FreeCAD is free software: you can redistribute it and/or modify it * | |
| * under the terms of the GNU Lesser General Public License as * | |
| * published by the Free Software Foundation, either version 2.1 of the * | |
| * License, or (at your option) any later version. * | |
| * * | |
| * FreeCAD is distributed in the hope that it will be useful, but * | |
| * WITHOUT ANY WARRANTY; without even the implied warranty of * | |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * | |
| * Lesser General Public License for more details. * | |
| * * | |
| * You should have received a copy of the GNU Lesser General Public * | |
| * License along with FreeCAD. If not, see * | |
| * <https://www.gnu.org/licenses/>. * | |
| * * | |
| **************************************************************************/ | |
| using namespace ImportGui; | |
| OCAFBrowser::OCAFBrowser(const Handle(TDocStd_Document) & hDoc) | |
| : pDoc(hDoc) | |
| { | |
| myGroupIcon = QApplication::style()->standardIcon(QStyle::SP_DirIcon); | |
| TDataStd::IDList(myList); | |
| myList.Append(TDataStd_TreeNode::GetDefaultTreeID()); | |
| myList.Append(TDataStd_Integer::GetID()); | |
| myList.Append(TDocStd_Owner::GetID()); | |
| myList.Append(TNaming_NamedShape::GetID()); | |
| myList.Append(TNaming_UsedShapes::GetID()); | |
| myList.Append(XCAFDoc_Color::GetID()); | |
| myList.Append(XCAFDoc_ColorTool::GetID()); | |
| myList.Append(XCAFDoc_LayerTool::GetID()); | |
| myList.Append(XCAFDoc_ShapeTool::GetID()); | |
| myList.Append(XCAFDoc_ShapeMapTool::GetID()); | |
| myList.Append(XCAFDoc_Location::GetID()); | |
| } | |
| std::string OCAFBrowser::toString(const TCollection_ExtendedString& extstr) | |
| { | |
| // NOLINTBEGIN | |
| char* str = new char[extstr.LengthOfCString() + 1]; | |
| extstr.ToUTF8CString(str); | |
| std::string text(str); | |
| delete[] str; | |
| return text; | |
| // NOLINTEND | |
| } | |
| void OCAFBrowser::load(QTreeWidget* theTree) | |
| { | |
| theTree->clear(); | |
| QTreeWidgetItem* root = new QTreeWidgetItem(); | |
| root->setText(0, QLatin1String("0")); | |
| root->setIcon(0, myGroupIcon); | |
| theTree->addTopLevelItem(root); | |
| load(pDoc->GetData()->Root(), root, QStringLiteral("0")); | |
| } | |
| void OCAFBrowser::load(const TDF_Label& label, QTreeWidgetItem* item, const QString& str) | |
| { | |
| label.Dump(std::cout); | |
| Handle(TDataStd_Name) name; | |
| if (label.FindAttribute(TDataStd_Name::GetID(), name)) { | |
| QString text | |
| = QStringLiteral("%1 %2").arg(str, QString::fromUtf8(toString(name->Get()).c_str())); | |
| item->setText(0, text); | |
| } | |
| TDF_IDList localList; | |
| TDF_AttributeIterator itr(label); | |
| for (; itr.More(); itr.Next()) { | |
| localList.Append(itr.Value()->ID()); | |
| } | |
| // Hint: In OCC 7.7 use XCAFDoc::AttributeInfo | |
| for (TDF_ListIteratorOfIDList it(localList); it.More(); it.Next()) { | |
| Handle(TDF_Attribute) attr; | |
| if (label.FindAttribute(it.Value(), attr)) { | |
| QTreeWidgetItem* child = new QTreeWidgetItem(); | |
| item->addChild(child); | |
| if (it.Value() == TDataStd_Name::GetID()) { | |
| QString text; | |
| QTextStream str(&text); | |
| str << attr->DynamicType()->Name(); | |
| str << " = " << toString(Handle(TDataStd_Name)::DownCast(attr)->Get()).c_str(); | |
| child->setText(0, text); | |
| } | |
| else if (it.Value() == TNaming_UsedShapes::GetID()) { | |
| QString text; | |
| QTextStream str(&text); | |
| str << attr->DynamicType()->Name(); | |
| str << " = " << Handle(TNaming_UsedShapes)::DownCast(attr)->Map().Size(); | |
| child->setText(0, text); | |
| } | |
| else if (it.Value() == TDF_TagSource::GetID()) { | |
| QString text; | |
| QTextStream str(&text); | |
| str << attr->DynamicType()->Name(); | |
| str << " = " << Handle(TDF_TagSource)::DownCast(attr)->Get(); | |
| child->setText(0, text); | |
| } | |
| else if (it.Value() == TDataStd_Integer::GetID()) { | |
| QString text; | |
| QTextStream str(&text); | |
| str << attr->DynamicType()->Name(); | |
| str << " = " << Handle(TDataStd_Integer)::DownCast(attr)->Get(); | |
| child->setText(0, text); | |
| } | |
| else if (it.Value() == TDataStd_Real::GetID()) { | |
| QString text; | |
| QTextStream str(&text); | |
| str << attr->DynamicType()->Name(); | |
| str << " = " << Handle(TDataStd_Real)::DownCast(attr)->Get(); | |
| child->setText(0, text); | |
| } | |
| else if (it.Value() == XCAFDoc_Color::GetID()) { | |
| Quantity_ColorRGBA rgba = Handle(XCAFDoc_Color)::DownCast(attr)->GetColorRGBA(); | |
| Quantity_Color rgb = rgba.GetRGB(); | |
| QString tuple = QStringLiteral("(%1, %2, %3, %4)") | |
| .arg(rgb.Red()) | |
| .arg(rgb.Green()) | |
| .arg(rgb.Blue()) | |
| .arg(rgba.Alpha()); | |
| QString text; | |
| QTextStream str(&text); | |
| str << attr->DynamicType()->Name(); | |
| str << " = " << Quantity_Color::StringName(rgb.Name()) << " " << tuple; | |
| child->setText(0, text); | |
| } | |
| else if (it.Value() == XCAFDoc_ShapeMapTool::GetID()) { | |
| QString text; | |
| QTextStream str(&text); | |
| str << attr->DynamicType()->Name(); | |
| str << " = " << Handle(XCAFDoc_ShapeMapTool)::DownCast(attr)->GetMap().Size(); | |
| child->setText(0, text); | |
| } | |
| else if (it.Value() == TNaming_NamedShape::GetID()) { | |
| child->setText(0, toText(Handle(TNaming_NamedShape)::DownCast(attr))); | |
| } | |
| else if (!Handle(TDataStd_TreeNode)::DownCast(attr).IsNull()) { | |
| Handle(TDataStd_TreeNode) aTreeNode = Handle(TDataStd_TreeNode)::DownCast(attr); | |
| child->setText(0, toText(aTreeNode)); | |
| } | |
| else { | |
| child->setText(0, QLatin1String(attr->DynamicType()->Name())); | |
| } | |
| } | |
| } | |
| int index = 1; | |
| for (TDF_ChildIterator it(label); it.More(); it.Next(), index++) { | |
| QString text = QStringLiteral("%1:%2").arg(str).arg(index); | |
| QTreeWidgetItem* child = new QTreeWidgetItem(); | |
| child->setText(0, text); | |
| child->setIcon(0, myGroupIcon); | |
| item->addChild(child); | |
| load(it.Value(), child, text); | |
| } | |
| } | |
| QString OCAFBrowser::toText(const Handle(TNaming_NamedShape) & namedShape) | |
| { | |
| TopoDS_Shape shape = namedShape->Get(); | |
| QString text; | |
| QTextStream str(&text); | |
| str << namedShape->DynamicType()->Name() << " = "; | |
| if (!shape.IsNull()) { | |
| switch (shape.ShapeType()) { | |
| case TopAbs_COMPOUND: | |
| str << "COMPOUND PRIMITIVE"; | |
| break; | |
| case TopAbs_COMPSOLID: | |
| str << "COMPSOLID PRIMITIVE"; | |
| break; | |
| case TopAbs_SOLID: | |
| str << "SOLID PRIMITIVE"; | |
| break; | |
| case TopAbs_SHELL: | |
| str << "SHELL PRIMITIVE"; | |
| break; | |
| case TopAbs_FACE: | |
| str << "FACE PRIMITIVE"; | |
| break; | |
| case TopAbs_WIRE: | |
| str << "WIRE PRIMITIVE"; | |
| break; | |
| case TopAbs_EDGE: | |
| str << "EDGE PRIMITIVE"; | |
| break; | |
| case TopAbs_VERTEX: | |
| str << "VERTEX PRIMITIVE"; | |
| break; | |
| case TopAbs_SHAPE: | |
| str << "SHAPE PRIMITIVE"; | |
| break; | |
| } | |
| } | |
| return text; | |
| } | |
| QString OCAFBrowser::toText(const Handle(TDataStd_TreeNode) & treeNode) | |
| { | |
| Standard_CString type = ""; | |
| if (treeNode->ID() == XCAFDoc::ShapeRefGUID()) { | |
| type = "Shape Instance Link"; | |
| } | |
| else if (treeNode->ID() == XCAFDoc::ColorRefGUID(XCAFDoc_ColorGen)) { | |
| type = "Generic Color Link"; | |
| } | |
| else if (treeNode->ID() == XCAFDoc::ColorRefGUID(XCAFDoc_ColorSurf)) { | |
| type = "Surface Color Link"; | |
| } | |
| else if (treeNode->ID() == XCAFDoc::ColorRefGUID(XCAFDoc_ColorCurv)) { | |
| type = "Curve Color Link"; | |
| } | |
| else if (treeNode->ID() == XCAFDoc::DimTolRefGUID()) { | |
| type = "DGT Link"; | |
| } | |
| else if (treeNode->ID() == XCAFDoc::DatumRefGUID()) { | |
| type = "Datum Link"; | |
| } | |
| else if (treeNode->ID() == XCAFDoc::MaterialRefGUID()) { | |
| type = "Material Link"; | |
| } | |
| QString prefix = QString::fromLatin1(type); | |
| QString display; | |
| if (treeNode->HasFather()) { | |
| TDF_Label aLabel = treeNode->Father()->Label(); | |
| TCollection_AsciiString entry; | |
| TDF_Tool::Entry(aLabel, entry); | |
| QString suffix = QString::fromLatin1(entry.ToCString()); | |
| display = QStringLiteral("%1 ==> %2").arg(prefix, suffix); | |
| } | |
| else { | |
| Handle(TDataStd_TreeNode) aFirstChild = treeNode->First(); | |
| QStringList aRefs; | |
| while (!aFirstChild.IsNull()) { | |
| TDF_Label aLabel = aFirstChild->Label(); | |
| TCollection_AsciiString entry; | |
| TDF_Tool::Entry(aLabel, entry); | |
| QString suffix = QString::fromLatin1(entry.ToCString()); | |
| aRefs.append(suffix); | |
| aFirstChild = aFirstChild->Next(); | |
| } | |
| display = QStringLiteral("%1 <== %2").arg(prefix, aRefs.join(QChar::fromLatin1(','))); | |
| } | |
| QString text; | |
| QTextStream str(&text); | |
| str << treeNode->DynamicType()->Name(); | |
| if (!display.isEmpty()) { | |
| str << " = " << display; | |
| } | |
| return text; | |
| } | |
| void OCAFBrowser::showDialog(const QString& title, const Handle(TDocStd_Document) & hDoc) | |
| { | |
| // NOLINTBEGIN | |
| QDialog* dlg = new QDialog(Gui::getMainWindow()); | |
| QTreeWidget* tree = new QTreeWidget(); | |
| tree->setHeaderLabel(QStringLiteral("OCAF Browser")); | |
| QVBoxLayout* layout = new QVBoxLayout; | |
| layout->addWidget(tree); | |
| dlg->setLayout(layout); | |
| QDialogButtonBox* btn = new QDialogButtonBox(dlg); | |
| btn->setStandardButtons(QDialogButtonBox::Close); | |
| QObject::connect(btn, &QDialogButtonBox::rejected, dlg, &QDialog::reject); | |
| QHBoxLayout* boxlayout = new QHBoxLayout; | |
| boxlayout->addWidget(btn); | |
| layout->addLayout(boxlayout); | |
| dlg->setWindowTitle(title); | |
| dlg->setAttribute(Qt::WA_DeleteOnClose); | |
| dlg->show(); | |
| OCAFBrowser browse(hDoc); | |
| browse.load(dlg->findChild<QTreeWidget*>()); | |
| tree->expandAll(); | |
| dlg->resize(400, 600); | |
| // NOLINTEND | |
| } | |