/*************************************************************************** * 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 * * * ***************************************************************************/ #include "PreCompiled.h" #if defined(__MINGW32__) # define WNT // avoid conflict with GUID #endif #ifndef _PreComp_ # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include # include #if OCC_VERSION_HEX >= 0x060500 # include # else # include # endif #endif #include "ImportOCAF.h" #include #include #include #include #include #include #include #include #include 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() { } void ImportOCAF::loadShapes() { myRefShapes.clear(); loadShapes(pDoc->Main(), TopLoc_Location(), default_name, "", false); } void ImportOCAF::loadShapes(const TDF_Label& label, const TopLoc_Location& loc, const std::string& defaultname, const std::string& /*assembly*/, bool isRef) { int hash = 0; TopoDS_Shape aShape; if (aShapeTool->GetShape(label,aShape)) { hash = aShape.HashCode(HashUpper); } 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 (std::string::iterator it = part_name.begin(); it != part_name.end(); ++it) { 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().Message("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; #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); } if (isRef || myRefShapes.find(hash) == myRefShapes.end()) { TopoDS_Shape aShape; if (isRef && aShapeTool->GetShape(label, aShape)) myRefShapes.insert(aShape.HashCode(HashUpper)); if (aShapeTool->IsSimpleShape(label) && (isRef || aShapeTool->IsFree(label))) { if (!asm_name.empty()) part_name = asm_name; if (isRef) createShape(label, loc, part_name); else createShape(label, part_loc, part_name); } else { for (TDF_ChildIterator it(label); it.More(); it.Next()) { loadShapes(it.Value(), part_loc, part_name, asm_name, isRef); } } } } void ImportOCAF::createShape(const TDF_Label& label, const TopLoc_Location& loc, const std::string& name) { const TopoDS_Shape& aShape = aShapeTool->GetShape(label); std::vector lValue; if (!aShape.IsNull() && aShape.ShapeType() == TopAbs_COMPOUND) { TopExp_Explorer xp; int ctSolids = 0, ctShells = 0; Part::Compound *pcCompound = static_cast(doc->addObject ("Part::Compound",name.c_str() )); for (xp.Init(aShape, TopAbs_SOLID); xp.More(); xp.Next(), ctSolids++) createShape(xp.Current(), loc, name, lValue); for (xp.Init(aShape, TopAbs_SHELL, TopAbs_SOLID); xp.More(); xp.Next(), ctShells++) createShape(xp.Current(), loc, name, lValue); pcCompound->Links.setValues(lValue); if (ctSolids > 0 || ctShells > 0) return; } 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 = static_cast(doc->addObject("Part::Feature")); if (!loc.IsIdentity()) part->Shape.setValue(aShape.Moved(loc)); else part->Shape.setValue(aShape); part->Label.setValue(name); lvalue.push_back(part); Quantity_Color aColor; App::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.r = (float)aColor.Red(); color.g = (float)aColor.Green(); color.b = (float)aColor.Blue(); std::vector colors; colors.push_back(color); applyColors(part, colors); #if 0//TODO Gui::ViewProvider* vp = Gui::Application::Instance->getViewProvider(part); if (vp && vp->isDerivedFrom(PartGui::ViewProviderPart::getClassTypeId())) { color.r = aColor.Red(); color.g = aColor.Green(); color.b = aColor.Blue(); static_cast(vp)->ShapeColor.setValue(color); } #endif } 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.r = (float)aColor.Red(); color.g = (float)aColor.Green(); color.b = (float)aColor.Blue(); faceColors[index-1] = color; found_face_color = true; } xp.Next(); } if (found_face_color) { applyColors(part, faceColors); #if 0//TODO Gui::ViewProvider* vp = Gui::Application::Instance->getViewProvider(part); if (vp && vp->isDerivedFrom(PartGui::ViewProviderPartExt::getClassTypeId())) { static_cast(vp)->DiffuseColor.setValues(faceColors); } #endif } } // ---------------------------------------------------------------------------- ExportOCAF::ExportOCAF(Handle_TDocStd_Document h) : pDoc(h) { aShapeTool = XCAFDoc_DocumentTool::ShapeTool(pDoc->Main()); aColorTool = XCAFDoc_DocumentTool::ColorTool(pDoc->Main()); #if defined(OCAF_KEEP_PLACEMENT) rootLabel = aShapeTool->NewShape(); TDataStd_Name::Set(rootLabel, "ASSEMBLY"); #else rootLabel = TDF_TagSource::NewChild(pDoc->Main()); #endif } void ExportOCAF::saveShape(Part::Feature* part, const std::vector& colors) { const TopoDS_Shape& shape = part->Shape.getValue(); if (shape.IsNull()) return; #if defined(OCAF_KEEP_PLACEMENT) // http://www.opencascade.org/org/forum/thread_18813/?forum=3 TopLoc_Location aLoc = shape.Location(); TopoDS_Shape baseShape = shape.Located(TopLoc_Location()); #else TopoDS_Shape baseShape = shape; #endif // Add shape and name TDF_Label shapeLabel = aShapeTool->NewShape(); aShapeTool->SetShape(shapeLabel, baseShape); TDataStd_Name::Set(shapeLabel, TCollection_ExtendedString(part->Label.getValue(), 1)); #if defined(OCAF_KEEP_PLACEMENT) aShapeTool->AddComponent(rootLabel, shapeLabel, aLoc); #endif // Add color information Quantity_Color col; std::set face_index; TopTools_IndexedMapOfShape faces; TopExp_Explorer xp(baseShape,TopAbs_FACE); while (xp.More()) { face_index.insert(faces.Add(xp.Current())); xp.Next(); } // define color per face? if (colors.size() == face_index.size()) { xp.Init(baseShape,TopAbs_FACE); while (xp.More()) { int index = faces.FindIndex(xp.Current()); if (face_index.find(index) != face_index.end()) { face_index.erase(index); //TDF_Label faceLabel = aShapeTool->AddSubShape(shapeLabel, xp.Current()); TDF_Label faceLabel= TDF_TagSource::NewChild(shapeLabel); aShapeTool->SetShape(faceLabel, xp.Current()); const App::Color& color = colors[index-1]; Quantity_Parameter mat[3]; mat[0] = color.r; mat[1] = color.g; mat[2] = color.b; col.SetValues(mat[0],mat[1],mat[2],Quantity_TOC_RGB); aColorTool->SetColor(faceLabel, col, XCAFDoc_ColorSurf); } xp.Next(); } } else if (!colors.empty()) { App::Color color = colors.front(); Quantity_Parameter mat[3]; mat[0] = color.r; mat[1] = color.g; mat[2] = color.b; col.SetValues(mat[0],mat[1],mat[2],Quantity_TOC_RGB); aColorTool->SetColor(shapeLabel, col, XCAFDoc_ColorGen); } } // ---------------------------------------------------------------------------- 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() { } 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 = static_cast(doc->addObject("Part::Feature", default_name.c_str())); part->Shape.setValue(shape); std::map::const_iterator jt; jt = myColorMap.find(shape.HashCode(INT_MAX)); App::Color partColor(0.8f,0.8f,0.8f); #if 0//TODO Gui::ViewProvider* vp = Gui::Application::Instance->getViewProvider(part); if (vp && vp->isDerivedFrom(PartGui::ViewProviderPart::getClassTypeId())) { if (jt != myColorMap.end()) { App::Color color; color.r = jt->second.Red(); color.g = jt->second.Green(); color.b = jt->second.Blue(); static_cast(vp)->ShapeColor.setValue(color); } partColor = static_cast(vp)->ShapeColor.getValue(); } #endif // set label name if defined if (setname && !myNameMap.empty()) { std::map::const_iterator jt; jt = myNameMap.find(shape.HashCode(INT_MAX)); 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(); } bool found_face_color = false; std::vector faceColors; faceColors.resize(faces.Extent(), partColor); xp.Init(shape,TopAbs_FACE); while (xp.More()) { jt = myColorMap.find(xp.Current().HashCode(INT_MAX)); if (jt != myColorMap.end()) { int index = faces.FindIndex(xp.Current()); App::Color color; color.r = (float)jt->second.Red(); color.g = (float)jt->second.Green(); color.b = (float)jt->second.Blue(); faceColors[index-1] = color; found_face_color = true; } xp.Next(); } if (found_face_color) { #if 0//TODO Gui::ViewProvider* vp = Gui::Application::Instance->getViewProvider(part); if (vp && vp->isDerivedFrom(PartGui::ViewProviderPartExt::getClassTypeId())) { static_cast(vp)->DiffuseColor.setValues(faceColors); } #endif } } } void ImportXCAF::loadShapes(const TDF_Label& label) { TopoDS_Shape aShape; if (aShapeTool->GetShape(label,aShape)) { //if (aShapeTool->IsReference(label)) { // TDF_Label reflabel; // if (aShapeTool->GetReferredShape(label, reflabel)) { // loadShapes(reflabel); // } //} 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[xp.Current().HashCode(INT_MAX)] = (xp.Current()); for (xp.Init(aShape, TopAbs_SHELL, TopAbs_SOLID); xp.More(); xp.Next(), ctShells++) this->myShells[xp.Current().HashCode(INT_MAX)] = (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[xp.Current().HashCode(INT_MAX)] = (xp.Current()); } if (ctComps == 0) { for (xp.Init(aShape, TopAbs_FACE, TopAbs_SHELL); xp.More(); xp.Next()) this->myShapes[xp.Current().HashCode(INT_MAX)] = (xp.Current()); for (xp.Init(aShape, TopAbs_WIRE, TopAbs_FACE); xp.More(); xp.Next()) this->myShapes[xp.Current().HashCode(INT_MAX)] = (xp.Current()); for (xp.Init(aShape, TopAbs_EDGE, TopAbs_WIRE); xp.More(); xp.Next()) this->myShapes[xp.Current().HashCode(INT_MAX)] = (xp.Current()); for (xp.Init(aShape, TopAbs_VERTEX, TopAbs_EDGE); xp.More(); xp.Next()) this->myShapes[xp.Current().HashCode(INT_MAX)] = (xp.Current()); } } // getting color Quantity_Color col; if (hColors->GetColor(label, XCAFDoc_ColorGen, col) || hColors->GetColor(label, XCAFDoc_ColorSurf, col) || hColors->GetColor(label, XCAFDoc_ColorCurv, col)) { // add defined color myColorMap[aShape.HashCode(INT_MAX)] = 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[it.Value().HashCode(INT_MAX)] = 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 label(str); if (!label.empty()) myNameMap[aShape.HashCode(INT_MAX)] = label; delete [] str; } #if 0 // http://www.opencascade.org/org/forum/thread_15174/ if (aShapeTool->IsAssembly(label)) { TDF_LabelSequence shapeLabels; aShapeTool->GetComponents(label, shapeLabels); Standard_Integer nbShapes = shapeLabels.Length(); for (Standard_Integer i = 1; i <= nbShapes; i++) { loadShapes(shapeLabels.Value(i)); } } #endif if (label.HasChild()) { TDF_ChildIterator it; for (it.Initialize(label); it.More(); it.Next()) { loadShapes(it.Value()); } } } }