// SPDX-License-Identifier: LGPL-2.1-or-later /*************************************************************************** * Copyright (c) 2013 Werner Mayer * * * * This file is part of the FreeCAD CAx development system. * * * * This library is free software; you can redistribute it and/or * * modify it under the terms of the GNU Library General Public * * License as published by the Free Software Foundation; either * * version 2 of the License, or (at your option) any later version. * * * * This library 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 Library General Public License for more details. * * * * You should have received a copy of the GNU Library General Public * * License along with this library; see the file COPYING.LIB. If not, * * write to the Free Software Foundation, Inc., 59 Temple Place, * * Suite 330, Boston, MA 02111-1307, USA * * * ***************************************************************************/ #if defined(__MINGW32__) # define WNT // avoid conflict with GUID #endif #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include // for Precision::Confusion() #include #include #include #include #include #include #include #include "ImportOCAF.h" #include "Tools.h" #ifdef HAVE_TBB # include # include # include #endif using namespace Import; #define OCAF_KEEP_PLACEMENT ImportOCAF::ImportOCAF(Handle(TDocStd_Document) h, App::Document* d, const std::string& name) : pDoc(h) , doc(d) , default_name(name) { aShapeTool = XCAFDoc_DocumentTool::ShapeTool(pDoc->Main()); aColorTool = XCAFDoc_DocumentTool::ColorTool(pDoc->Main()); } ImportOCAF::~ImportOCAF() = default; void ImportOCAF::tryPlacementFromLoc(App::GeoFeature* part, const TopLoc_Location& part_loc) { gp_Trsf trf; Base::Matrix4D mtrx; if (part_loc.IsIdentity()) { trf = part_loc.Transformation(); } else { trf = TopLoc_Location(part_loc.FirstDatum()).Transformation(); } Part::TopoShape::convertToMatrix(trf, mtrx); tryPlacementFromMatrix(part, mtrx); } void ImportOCAF::tryPlacementFromMatrix(App::GeoFeature* part, const Base::Matrix4D& mat) { try { Base::Placement pl; pl.fromMatrix(mat); part->Placement.setValue(pl); } catch (const Base::ValueError& e) { e.reportException(); } } void ImportOCAF::loadShapes() { std::vector lValue; myRefShapes.clear(); loadShapes(pDoc->Main(), TopLoc_Location(), default_name, "", false, lValue); lValue.clear(); } void ImportOCAF::setMerge(bool merge) { this->merge = merge; } void ImportOCAF::loadShapes( const TDF_Label& label, const TopLoc_Location& loc, const std::string& defaultname, const std::string& assembly, bool isRef, std::vector& lValue ) { int hash = 0; #ifdef HAVE_TBB using namespace tbb; task_group g; #endif TopoDS_Shape aShape; std::vector localValue; if (aShapeTool->GetShape(label, aShape)) { hash = Part::ShapeMapHasher {}(aShape); } Handle(TDataStd_Name) name; std::string part_name = defaultname; if (label.FindAttribute(TDataStd_Name::GetID(), name)) { TCollection_ExtendedString extstr = name->Get(); char* str = new char[extstr.LengthOfCString() + 1]; extstr.ToUTF8CString(str); part_name = str; delete[] str; if (part_name.empty()) { part_name = defaultname; } else { bool ws = true; for (char it : part_name) { if (it != ' ') { ws = false; break; } } if (ws) { part_name = defaultname; } } } TopLoc_Location part_loc = loc; Handle(XCAFDoc_Location) hLoc; if (label.FindAttribute(XCAFDoc_Location::GetID(), hLoc)) { if (isRef) { part_loc = part_loc * hLoc->Get(); } else { part_loc = hLoc->Get(); } } #ifdef FC_DEBUG Base::Console().log( "H:%d, N:%s, T:%d, A:%d, S:%d, C:%d, SS:%d, F:%d, R:%d, C:%d, SS:%d\n", hash, part_name.c_str(), aShapeTool->IsTopLevel(label), aShapeTool->IsAssembly(label), aShapeTool->IsShape(label), aShapeTool->IsCompound(label), aShapeTool->IsSimpleShape(label), aShapeTool->IsFree(label), aShapeTool->IsReference(label), aShapeTool->IsComponent(label), aShapeTool->IsSubShape(label) ); #endif #if defined(OCAF_KEEP_PLACEMENT) std::string asm_name = part_name; (void)assembly; #else std::string asm_name = assembly; if (aShapeTool->IsAssembly(label)) { asm_name = part_name; } #endif TDF_Label ref; if (aShapeTool->IsReference(label) && aShapeTool->GetReferredShape(label, ref)) { loadShapes(ref, part_loc, part_name, asm_name, true, lValue); } if (isRef || myRefShapes.find(hash) == myRefShapes.end()) { TopoDS_Shape aShape; if (isRef && aShapeTool->GetShape(label, aShape)) { myRefShapes.insert(Part::ShapeMapHasher {}(aShape)); } if (aShapeTool->IsSimpleShape(label) && (isRef || aShapeTool->IsFree(label))) { if (!asm_name.empty()) { part_name = asm_name; } // TODO: The merge parameter (last one from createShape) should become an // Importer/Exporter option within the FreeCAD preference menu Currently it is merging // STEP Compound Shape into a single Shape Part::Feature which is an OpenCascade // computed Compound if (isRef) { createShape(label, loc, part_name, lValue, this->merge); } else { createShape(label, part_loc, part_name, localValue, this->merge); } } else { if (aShapeTool->IsSimpleShape(label)) { // We are not creating a list of Part::Feature in that case but just // a single Part::Feature which has as a Shape a Compound of the Subshapes contained // within the global shape // This is standard behavior of many STEP reader and avoid to register a crazy // amount of Shape within the Tree as STEP file do mostly contain large assemblies return; } // This is probably an Assembly let's try to create a Compound with the name for (TDF_ChildIterator it(label); it.More(); it.Next()) { if (isRef) { loadShapes(it.Value(), part_loc, part_name, asm_name, false, localValue); } else { loadShapes(it.Value(), part_loc, part_name, asm_name, isRef, localValue); } } if (!localValue.empty()) { if (aShapeTool->IsAssembly(label)) { App::Part* pcPart = nullptr; pcPart = doc->addObject(asm_name.c_str()); pcPart->Label.setValue(asm_name); pcPart->addObjects(localValue); // STEP reader is now a hierarchical reader. Node and leaf must have // there local placement updated and relative to the STEP file content // standard FreeCAD placement was absolute we are now moving to relative tryPlacementFromLoc(pcPart, part_loc); lValue.push_back(pcPart); } } } } } void ImportOCAF::createShape( const TDF_Label& label, const TopLoc_Location& loc, const std::string& name, std::vector& lValue, bool mergeShape ) { const TopoDS_Shape& aShape = aShapeTool->GetShape(label); #ifdef HAVE_TBB using namespace tbb; task_group g; #endif if (!aShape.IsNull() && aShape.ShapeType() == TopAbs_COMPOUND) { TopExp_Explorer xp; int ctSolids = 0, ctShells = 0, ctVertices = 0, ctEdges = 0; std::vector localValue; App::Part* pcPart = nullptr; if (mergeShape) { // We should do that only if there is more than a single shape inside // Computing Compounds takes time // We must keep track of the Color. If there is more than 1 Color into // a STEP Compound then the Merge can't be done and we cancel the operation BRep_Builder builder; TopoDS_Compound comp; builder.MakeCompound(comp); for (xp.Init(aShape, TopAbs_SOLID); xp.More(); xp.Next(), ctSolids++) { const TopoDS_Shape& sh = xp.Current(); if (!sh.IsNull()) { builder.Add(comp, sh); } } for (xp.Init(aShape, TopAbs_SHELL, TopAbs_SOLID); xp.More(); xp.Next(), ctShells++) { const TopoDS_Shape& sh = xp.Current(); if (!sh.IsNull()) { builder.Add(comp, sh); } } for (xp.Init(aShape, TopAbs_EDGE); xp.More(); xp.Next(), ctEdges++) { const TopoDS_Shape& sh = xp.Current(); if (!sh.IsNull()) { builder.Add(comp, sh); } } for (xp.Init(aShape, TopAbs_VERTEX); xp.More(); xp.Next(), ctVertices++) { const TopoDS_Shape& sh = xp.Current(); if (!sh.IsNull()) { builder.Add(comp, sh); } } // Ok we got a Compound which is computed // Just need to add it to a Part::Feature and push it to lValue if (!comp.IsNull() && (ctSolids || ctShells || ctEdges || ctVertices)) { Part::Feature* part = doc->addObject(); // Let's allocate the relative placement of the Compound from the STEP file tryPlacementFromLoc(part, loc); if (!loc.IsIdentity()) { part->Shape.setValue(comp.Moved(loc)); } else { part->Shape.setValue(comp); } part->Label.setValue(name); lValue.push_back(part); loadColors(part, aShape); } } else { for (xp.Init(aShape, TopAbs_SOLID); xp.More(); xp.Next(), ctSolids++) { createShape(xp.Current(), loc, name, localValue); } for (xp.Init(aShape, TopAbs_SHELL, TopAbs_SOLID); xp.More(); xp.Next(), ctShells++) { createShape(xp.Current(), loc, name, localValue); } } if (!localValue.empty() && !mergeShape) { pcPart = doc->addObject(name.c_str()); pcPart->Label.setValue(name); // localValue contain the objects that must added to the local Part // We must add the PartOrigin and the Part itself pcPart->addObjects(localValue); lValue.push_back(pcPart); } if (ctSolids > 0 || ctShells > 0) { return; } } else if (!aShape.IsNull()) { createShape(aShape, loc, name, lValue); } } void ImportOCAF::createShape( const TopoDS_Shape& aShape, const TopLoc_Location& loc, const std::string& name, std::vector& lvalue ) { Part::Feature* part = doc->addObject(); if (!loc.IsIdentity()) { part->Shape.setValue(aShape.Moved(loc)); } else { part->Shape.setValue(aShape); } part->Label.setValue(name); lvalue.push_back(part); loadColors(part, aShape); } void ImportOCAF::loadColors(Part::Feature* part, const TopoDS_Shape& aShape) { Quantity_ColorRGBA aColor; Base::Color color(0.8f, 0.8f, 0.8f); if (aColorTool->GetColor(aShape, XCAFDoc_ColorGen, aColor) || aColorTool->GetColor(aShape, XCAFDoc_ColorSurf, aColor) || aColorTool->GetColor(aShape, XCAFDoc_ColorCurv, aColor)) { color = Tools::convertColor(aColor); std::vector colors; colors.push_back(color); applyColors(part, colors); } TopTools_IndexedMapOfShape faces; TopExp_Explorer xp(aShape, TopAbs_FACE); while (xp.More()) { faces.Add(xp.Current()); xp.Next(); } bool found_face_color = false; std::vector faceColors; faceColors.resize(faces.Extent(), color); xp.Init(aShape, TopAbs_FACE); while (xp.More()) { if (aColorTool->GetColor(xp.Current(), XCAFDoc_ColorGen, aColor) || aColorTool->GetColor(xp.Current(), XCAFDoc_ColorSurf, aColor) || aColorTool->GetColor(xp.Current(), XCAFDoc_ColorCurv, aColor)) { int index = faces.FindIndex(xp.Current()); color = Tools::convertColor(aColor); faceColors[index - 1] = color; found_face_color = true; } xp.Next(); } if (found_face_color) { applyColors(part, faceColors); } } // ---------------------------------------------------------------------------- ImportOCAFCmd::ImportOCAFCmd(Handle(TDocStd_Document) h, App::Document* d, const std::string& name) : ImportOCAF(h, d, name) {} void ImportOCAFCmd::applyColors(Part::Feature* part, const std::vector& colors) { partColors[part] = colors; } // ---------------------------------------------------------------------------- ImportXCAF::ImportXCAF(Handle(TDocStd_Document) h, App::Document* d, const std::string& name) : hdoc(h) , doc(d) , default_name(name) { aShapeTool = XCAFDoc_DocumentTool::ShapeTool(hdoc->Main()); hColors = XCAFDoc_DocumentTool::ColorTool(hdoc->Main()); } ImportXCAF::~ImportXCAF() = default; void ImportXCAF::loadShapes() { // collect sequence of labels to display TDF_LabelSequence shapeLabels, colorLabels; aShapeTool->GetFreeShapes(shapeLabels); hColors->GetColors(colorLabels); // set presentations and show for (Standard_Integer i = 1; i <= shapeLabels.Length(); i++) { // get the shapes and attributes const TDF_Label& label = shapeLabels.Value(i); loadShapes(label); } std::map::iterator it; // go through solids for (it = mySolids.begin(); it != mySolids.end(); ++it) { createShape(it->second, true, true); } // go through shells for (it = myShells.begin(); it != myShells.end(); ++it) { createShape(it->second, true, true); } // go through compounds for (it = myCompds.begin(); it != myCompds.end(); ++it) { createShape(it->second, true, true); } // do the rest if (!myShapes.empty()) { BRep_Builder builder; TopoDS_Compound comp; builder.MakeCompound(comp); for (it = myShapes.begin(); it != myShapes.end(); ++it) { builder.Add(comp, it->second); } createShape(comp, true, false); } } void ImportXCAF::createShape(const TopoDS_Shape& shape, bool perface, bool setname) const { Part::Feature* part; part = doc->addObject(default_name.c_str()); part->Label.setValue(default_name); part->Shape.setValue(shape); std::map::const_iterator jt; jt = myColorMap.find(Part::ShapeMapHasher {}(shape)); Base::Color partColor(0.8f, 0.8f, 0.8f); // set label name if defined if (setname && !myNameMap.empty()) { std::map::const_iterator jt; jt = myNameMap.find(Part::ShapeMapHasher {}(shape)); if (jt != myNameMap.end()) { part->Label.setValue(jt->second); } } // check for colors per face if (perface && !myColorMap.empty()) { TopTools_IndexedMapOfShape faces; TopExp_Explorer xp(shape, TopAbs_FACE); while (xp.More()) { faces.Add(xp.Current()); xp.Next(); } std::vector faceColors; faceColors.resize(faces.Extent(), partColor); xp.Init(shape, TopAbs_FACE); while (xp.More()) { jt = myColorMap.find(Part::ShapeMapHasher {}(xp.Current())); if (jt != myColorMap.end()) { int index = faces.FindIndex(xp.Current()); faceColors[index - 1] = Tools::convertColor(jt->second); } xp.Next(); } } } void ImportXCAF::loadShapes(const TDF_Label& label) { TopoDS_Shape aShape; if (aShapeTool->GetShape(label, aShape)) { if (aShapeTool->IsTopLevel(label)) { int ctSolids = 0, ctShells = 0, ctComps = 0; // add the shapes TopExp_Explorer xp; for (xp.Init(aShape, TopAbs_SOLID); xp.More(); xp.Next(), ctSolids++) { this->mySolids[Part::ShapeMapHasher {}(xp.Current())] = (xp.Current()); } for (xp.Init(aShape, TopAbs_SHELL, TopAbs_SOLID); xp.More(); xp.Next(), ctShells++) { this->myShells[Part::ShapeMapHasher {}(xp.Current())] = (xp.Current()); } // if no solids and no shells were found then go for compounds if (ctSolids == 0 && ctShells == 0) { for (xp.Init(aShape, TopAbs_COMPOUND); xp.More(); xp.Next(), ctComps++) { this->myCompds[Part::ShapeMapHasher {}(xp.Current())] = (xp.Current()); } } if (ctComps == 0) { for (xp.Init(aShape, TopAbs_FACE, TopAbs_SHELL); xp.More(); xp.Next()) { this->myShapes[Part::ShapeMapHasher {}(xp.Current())] = (xp.Current()); } for (xp.Init(aShape, TopAbs_WIRE, TopAbs_FACE); xp.More(); xp.Next()) { this->myShapes[Part::ShapeMapHasher {}(xp.Current())] = (xp.Current()); } for (xp.Init(aShape, TopAbs_EDGE, TopAbs_WIRE); xp.More(); xp.Next()) { this->myShapes[Part::ShapeMapHasher {}(xp.Current())] = (xp.Current()); } for (xp.Init(aShape, TopAbs_VERTEX, TopAbs_EDGE); xp.More(); xp.Next()) { this->myShapes[Part::ShapeMapHasher {}(xp.Current())] = (xp.Current()); } } } // getting color Quantity_ColorRGBA col; if (hColors->GetColor(label, XCAFDoc_ColorGen, col) || hColors->GetColor(label, XCAFDoc_ColorSurf, col) || hColors->GetColor(label, XCAFDoc_ColorCurv, col)) { // add defined color myColorMap[Part::ShapeMapHasher {}(aShape)] = col; } else { // http://www.opencascade.org/org/forum/thread_17107/ TopoDS_Iterator it; for (it.Initialize(aShape); it.More(); it.Next()) { if (hColors->GetColor(it.Value(), XCAFDoc_ColorGen, col) || hColors->GetColor(it.Value(), XCAFDoc_ColorSurf, col) || hColors->GetColor(it.Value(), XCAFDoc_ColorCurv, col)) { // add defined color myColorMap[Part::ShapeMapHasher {}(it.Value())] = col; } } } // getting names Handle(TDataStd_Name) name; if (label.FindAttribute(TDataStd_Name::GetID(), name)) { TCollection_ExtendedString extstr = name->Get(); char* str = new char[extstr.LengthOfCString() + 1]; extstr.ToUTF8CString(str); std::string labelName(str); if (!labelName.empty()) { myNameMap[Part::ShapeMapHasher {}(aShape)] = labelName; } delete[] str; } if (label.HasChild()) { TDF_ChildIterator it; for (it.Initialize(label); it.More(); it.Next()) { loadShapes(it.Value()); } } } }