/*************************************************************************** * Copyright (c) 2013 Luke Parry * * * * 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" #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 #include #include #include #endif // #ifndef _PreComp_ #include #include #include #include #include #include #include #include "DrawUtil.h" #include "GeometryObject.h" #include "DrawViewPart.h" #include "DrawViewDetail.h" using namespace TechDrawGeometry; using namespace TechDraw; using namespace std; struct EdgePoints { gp_Pnt v1, v2; TopoDS_Edge edge; }; GeometryObject::GeometryObject(const string& parent, TechDraw::DrawView* parentObj) : m_parentName(parent), m_parent(parentObj), m_isoCount(0), m_isPersp(false), m_focus(100.0), m_usePolygonHLR(false) { } GeometryObject::~GeometryObject() { clear(); } const std::vector GeometryObject::getVisibleFaceEdges(const bool smooth, const bool seam) const { std::vector result; bool smoothOK = smooth; bool seamOK = seam; for (auto& e:edgeGeom) { if (e->visible) { switch (e->classOfEdge) { case ecHARD: case ecOUTLINE: result.push_back(e); break; case ecSMOOTH: if (smoothOK) { result.push_back(e); } break; case ecSEAM: if (seamOK) { result.push_back(e); } break; default: ; } } } return result; } void GeometryObject::clear() { for(std::vector::iterator it = edgeGeom.begin(); it != edgeGeom.end(); ++it) { delete *it; *it = 0; } for(std::vector::iterator it = faceGeom.begin(); it != faceGeom.end(); ++it) { delete *it; *it = 0; } for(std::vector::iterator it = vertexGeom.begin(); it != vertexGeom.end(); ++it) { delete *it; *it = 0; } vertexGeom.clear(); faceGeom.clear(); edgeGeom.clear(); } //!set up a hidden line remover and project a shape with it void GeometryObject::projectShape(const TopoDS_Shape& input, const gp_Ax2 viewAxis) { // Base::Console().Message("GO::projectShape()\n"); // Clear previous Geometry clear(); auto start = chrono::high_resolution_clock::now(); Handle(HLRBRep_Algo) brep_hlr = NULL; try { brep_hlr = new HLRBRep_Algo(); brep_hlr->Add(input, m_isoCount); if (m_isPersp) { double fLength = std::max(Precision::Confusion(),m_focus); // HLRAlgo_Projector projector( projAxis, fLength ); HLRAlgo_Projector projector( viewAxis, fLength ); brep_hlr->Projector(projector); } else { // HLRAlgo_Projector projector( projAxis ); HLRAlgo_Projector projector( viewAxis ); brep_hlr->Projector(projector); } brep_hlr->Update(); brep_hlr->Hide(); } catch (Standard_Failure e) { Base::Console().Error("GO::projectShape - OCC error - %s - while projecting shape\n", e.GetMessageString()); } catch (...) { throw Base::RuntimeError("GeometryObject::projectShape - unknown error occurred while projecting shape"); // Standard_Failure::Raise("GeometryObject::projectShape - error occurred while projecting shape"); } auto end = chrono::high_resolution_clock::now(); auto diff = end - start; double diffOut = chrono::duration (diff).count(); Base::Console().Log("TIMING - %s GO spent: %.3f millisecs in HLRBRep_Algo & co\n",m_parentName.c_str(),diffOut); start = chrono::high_resolution_clock::now(); try { HLRBRep_HLRToShape hlrToShape(brep_hlr); visHard = hlrToShape.VCompound(); visSmooth = hlrToShape.Rg1LineVCompound(); visSeam = hlrToShape.RgNLineVCompound(); visOutline = hlrToShape.OutLineVCompound(); visIso = hlrToShape.IsoLineVCompound(); hidHard = hlrToShape.HCompound(); hidSmooth = hlrToShape.Rg1LineHCompound(); hidSeam = hlrToShape.RgNLineHCompound(); hidOutline = hlrToShape.OutLineHCompound(); hidIso = hlrToShape.IsoLineHCompound(); //need these 3d curves to prevent "zero edges" later BRepLib::BuildCurves3d(visHard); BRepLib::BuildCurves3d(visSmooth); BRepLib::BuildCurves3d(visSeam); BRepLib::BuildCurves3d(visOutline); BRepLib::BuildCurves3d(visIso); BRepLib::BuildCurves3d(hidHard); BRepLib::BuildCurves3d(hidSmooth); BRepLib::BuildCurves3d(hidSeam); BRepLib::BuildCurves3d(hidOutline); BRepLib::BuildCurves3d(hidIso); } catch (Standard_Failure e) { Base::Console().Error("GO::projectShape - OCC error - %s - while extracting edges\n", e.GetMessageString()); } catch (...) { throw Base::RuntimeError("GeometryObject::projectShape - error occurred while extracting edges"); } end = chrono::high_resolution_clock::now(); diff = end - start; diffOut = chrono::duration (diff).count(); Base::Console().Log("TIMING - %s GO spent: %.3f millisecs in hlrToShape and BuildCurves\n",m_parentName.c_str(),diffOut); } //!set up a hidden line remover and project a shape with it void GeometryObject::projectShapeWithPolygonAlgo(const TopoDS_Shape& input, const gp_Ax2 viewAxis) { // Clear previous Geometry clear(); //work around for Mantis issue #3332 //if 3332 gets fixed in OCC, this will produce shifted views and will need //to be reverted. TopoDS_Shape inCopy; if (!m_isPersp) { gp_Pnt gCenter = findCentroid(input, viewAxis); Base::Vector3d motion(-gCenter.X(),-gCenter.Y(),-gCenter.Z()); inCopy = moveShape(input,motion); } else { BRepBuilderAPI_Copy BuilderCopy(input); inCopy = BuilderCopy.Shape(); } auto start = chrono::high_resolution_clock::now(); Handle(HLRBRep_PolyAlgo) brep_hlrPoly = NULL; try { TopExp_Explorer faces(inCopy, TopAbs_FACE); for (int i = 1; faces.More(); faces.Next(), i++) { const TopoDS_Face& f = TopoDS::Face(faces.Current()); if (!f.IsNull()) { BRepMesh_IncrementalMesh(f, 0.10); //Poly Algo requires a mesh! } } brep_hlrPoly = new HLRBRep_PolyAlgo(); brep_hlrPoly->Load(inCopy); if (m_isPersp) { double fLength = std::max(Precision::Confusion(), m_focus); HLRAlgo_Projector projector(viewAxis, fLength); brep_hlrPoly->Projector(projector); } else { // non perspective HLRAlgo_Projector projector(viewAxis); brep_hlrPoly->Projector(projector); } brep_hlrPoly->Update(); } catch (Standard_Failure e) { Base::Console().Error("GO::projectShapeWithPolygonAlgo - OCC error - %s - while projecting shape\n", e.GetMessageString()); } catch (...) { throw Base::RuntimeError("GeometryObject::projectShapeWithPolygonAlgo - error occurred while projecting shape"); // Standard_Failure::Raise("GeometryObject::projectShapeWithPolygonAlgo - error occurred while projecting shape"); } try { HLRBRep_PolyHLRToShape polyhlrToShape; polyhlrToShape.Update(brep_hlrPoly); visHard = polyhlrToShape.VCompound(); visSmooth = polyhlrToShape.Rg1LineVCompound(); visSeam = polyhlrToShape.RgNLineVCompound(); visOutline = polyhlrToShape.OutLineVCompound(); hidHard = polyhlrToShape.HCompound(); hidSmooth = polyhlrToShape.Rg1LineHCompound(); hidSeam = polyhlrToShape.RgNLineHCompound(); hidOutline = polyhlrToShape.OutLineHCompound(); //need these 3d curves to prevent "zero edges" later BRepLib::BuildCurves3d(visHard); BRepLib::BuildCurves3d(visSmooth); BRepLib::BuildCurves3d(visSeam); BRepLib::BuildCurves3d(visOutline); BRepLib::BuildCurves3d(hidHard); BRepLib::BuildCurves3d(hidSmooth); BRepLib::BuildCurves3d(hidSeam); BRepLib::BuildCurves3d(hidOutline); } catch (Standard_Failure e) { Base::Console().Error("GO::projectShapeWithPolygonAlgo - OCC error - %s - while extracting edges\n", e.GetMessageString()); } catch (...) { throw Base::RuntimeError("GeometryObject::projectShapeWithPolygonAlgo - error occurred while extracting edges"); // Standard_Failure::Raise("GeometryObject::projectShapeWithPolygonAlgo - error occurred while extracting edges"); } auto end = chrono::high_resolution_clock::now(); auto diff = end - start; double diffOut = chrono::duration (diff).count(); Base::Console().Log("TIMING - %s GO spent: %.3f millisecs in HLRBRep_PolyAlgo & co\n", m_parentName.c_str(), diffOut); } //!add edges meeting filter criteria for category, visibility void GeometryObject::extractGeometry(edgeClass category, bool visible) { TopoDS_Shape filtEdges; if (visible) { switch (category) { case ecHARD: filtEdges = visHard; break; case ecOUTLINE: filtEdges = visOutline; break; case ecSMOOTH: filtEdges = visSmooth; break; case ecSEAM: filtEdges = visSeam; break; case ecUVISO: filtEdges = visIso; break; default: Base::Console().Warning("GeometryObject::ExtractGeometry - unsupported visible edgeClass: %d\n",category); return; } } else { switch (category) { case ecHARD: filtEdges = hidHard; break; case ecOUTLINE: filtEdges = hidOutline; break; case ecSMOOTH: filtEdges = hidSmooth; break; case ecSEAM: filtEdges = hidSeam; break; case ecUVISO: filtEdges = hidIso; break; default: Base::Console().Warning("GeometryObject::ExtractGeometry - unsupported hidden edgeClass: %d\n",category); return; } } addGeomFromCompound(filtEdges, category, visible); } //! update edgeGeom and vertexGeom from Compound of edges void GeometryObject::addGeomFromCompound(TopoDS_Shape edgeCompound, edgeClass category, bool visible) { // Base::Console().Message("GO::addGeomFromCompound()\n"); if(edgeCompound.IsNull()) { Base::Console().Log("TechDraw::GeometryObject::addGeomFromCompound edgeCompound is NULL\n"); return; // There is no OpenCascade Geometry to be calculated } BaseGeom* base; TopExp_Explorer edges(edgeCompound, TopAbs_EDGE); int i = 1; for ( ; edges.More(); edges.Next(),i++) { const TopoDS_Edge& edge = TopoDS::Edge(edges.Current()); if (edge.IsNull()) { //Base::Console().Log("INFO - GO::addGeomFromCompound - edge: %d is NULL\n",i); continue; } if (DrawUtil::isZeroEdge(edge)) { Base::Console().Log("INFO - GO::addGeomFromCompound - edge: %d is zeroEdge\n",i); continue; } base = BaseGeom::baseFactory(edge); if (base == nullptr) { Base::Console().Log("Error - GO::addGeomFromCompound - baseFactory failed for edge: %d\n",i); throw Base::ValueError("GeometryObject::addGeomFromCompound - baseFactory failed"); } base->classOfEdge = category; base->visible = visible; edgeGeom.push_back(base); //add vertices of new edge if not already in list if (visible) { BaseGeom* lastAdded = edgeGeom.back(); bool v1Add = true, v2Add = true; bool c1Add = true; TechDrawGeometry::Vertex* v1 = new TechDrawGeometry::Vertex(lastAdded->getStartPoint()); TechDrawGeometry::Vertex* v2 = new TechDrawGeometry::Vertex(lastAdded->getEndPoint()); TechDrawGeometry::Circle* circle = dynamic_cast(lastAdded); TechDrawGeometry::Vertex* c1 = nullptr; if (circle) { c1 = new TechDrawGeometry::Vertex(circle->center); c1->isCenter = true; c1->visible = true; } std::vector::iterator itVertex = vertexGeom.begin(); for (; itVertex != vertexGeom.end(); itVertex++) { if ((*itVertex)->isEqual(v1,Precision::Confusion())) { v1Add = false; } if ((*itVertex)->isEqual(v2,Precision::Confusion())) { v2Add = false; } if (circle ) { if ((*itVertex)->isEqual(c1,Precision::Confusion())) { c1Add = false; } } } if (v1Add) { vertexGeom.push_back(v1); v1->visible = true; } else { delete v1; } if (v2Add) { vertexGeom.push_back(v2); v2->visible = true; } else { delete v2; } if (circle) { if (c1Add) { vertexGeom.push_back(c1); c1->visible = true; } else { delete c1; } } } } //end TopExp } int GeometryObject::addRandomVertex(Base::Vector3d pos) { // Base::Console().Message("GO::addRandomVertex(%s)\n",DrawUtil::formatVector(pos).c_str()); TechDrawGeometry::Vertex* v = new TechDrawGeometry::Vertex(pos.x, - pos.y); vertexGeom.push_back(v); int idx = vertexGeom.size() - 1; return idx; } //void GeometryObject::removeRandomVertex(TechDraw::CosmeticVertex* cv) //{ // Base::Console().Message("GO::removeRandomVertex(cv)\n"); //} //void GeometryObject::removeRandomVertex(int idx) //{ // Base::Console().Message("GO::removeRandomVertex(%d)\n", idx); //} int GeometryObject::addRandomEdge(TechDrawGeometry::BaseGeom* base) { // Base::Console().Message("GO::addRandomEdge() - cosmetic: %d\n", base->cosmetic); base->cosmetic = true; edgeGeom.push_back(base); int idx = edgeGeom.size() - 1; return idx; } //void GeometryObject::removeRandomEdge(TechDraw::CosmeticEdge* ce) //{ // Base::Console().Message("GO::removeRandomEdge(ce)\n"); //} //void GeometryObject::removeRandomEdge(int idx) //{ // Base::Console().Message("GO::removeRandomEdge(%d)\n",idx); //} //! empty Face geometry void GeometryObject::clearFaceGeom() { faceGeom.clear(); } //! add a Face to Face Geometry void GeometryObject::addFaceGeom(Face* f) { faceGeom.push_back(f); } TechDraw::DrawViewDetail* GeometryObject::isParentDetail() { TechDraw::DrawViewDetail* result = nullptr; if (m_parent != nullptr) { TechDraw::DrawViewDetail* detail = dynamic_cast(m_parent); if (detail != nullptr) { result = detail; } } return result; } bool GeometryObject::isWithinArc(double theta, double first, double last, bool cw) const { if (fabs(last - first) >= 2 * M_PI) { return true; } // Put params within [0, 2*pi) - not totally sure this is necessary theta = fmod(theta, 2 * M_PI); if (theta < 0) { theta += 2 * M_PI; } first = fmod(first, 2 * M_PI); if (first < 0) { first += 2 * M_PI; } last = fmod(last, 2 * M_PI); if (last < 0) { last += 2 * M_PI; } if (cw) { if (first > last) { return theta <= first && theta >= last; } else { return theta <= first || theta >= last; } } else { if (first > last) { return theta >= first || theta <= last; } else { return theta >= first && theta <= last; } } } Base::BoundBox3d GeometryObject::calcBoundingBox() const { // Base::Console().Message("GO::calcBoundingBox() - edges: %d\n", edgeGeom.size()); Bnd_Box testBox; testBox.SetGap(0.0); if (!edgeGeom.empty()) { for (std::vector::const_iterator it( edgeGeom.begin() ); it != edgeGeom.end(); ++it) { BRepBndLib::Add((*it)->occEdge, testBox); } } double xMin = 0,xMax = 0,yMin = 0,yMax = 0, zMin = 0, zMax = 0; if (testBox.IsVoid()) { Base::Console().Log("INFO - GO::calcBoundingBox - testBox is void\n"); } else { testBox.Get(xMin,yMin,zMin,xMax,yMax,zMax); } Base::BoundBox3d bbox(xMin,yMin,zMin,xMax,yMax,zMax); return bbox; } void GeometryObject::pruneVertexGeom(Base::Vector3d center, double radius) { const std::vector& oldVerts = getVertexGeometry(); std::vector newVerts; for (auto& v: oldVerts) { Base::Vector3d v3 = v->getAs3D(); double length = (v3 - center).Length(); if (length < Precision::Confusion()) { continue; } else if (length < radius) { newVerts.push_back(v); } } vertexGeom = newVerts; } //! does this GeometryObject already have this vertex bool GeometryObject::findVertex(Base::Vector2d v) { bool found = false; std::vector::iterator it = vertexGeom.begin(); for (; it != vertexGeom.end(); it++) { double dist = (v - (*it)->pnt).Length(); if (dist < Precision::Confusion()) { found = true; break; } } return found; } /// utility non-class member functions //! gets a coordinate system that matches view system used in 3D with +Z up (or +Y up if necessary) //! used for individual views, but not secondary views in projection groups //! flip determines Y mirror or not. // getViewAxis 1 gp_Ax2 TechDrawGeometry::getViewAxis(const Base::Vector3d origin, const Base::Vector3d& direction, const bool flip) { (void) flip; gp_Ax2 viewAxis; gp_Pnt inputCenter(origin.x,origin.y,origin.z); Base::Vector3d stdZ(0.0,0.0,1.0); Base::Vector3d stdOrg(0.0,0.0,0.0); Base::Vector3d flipDirection(direction.x,-direction.y,direction.z); if (!flip) { flipDirection = Base::Vector3d(direction.x,direction.y,direction.z); } Base::Vector3d cross = flipDirection; // //special case if (TechDraw::DrawUtil::checkParallel(flipDirection, stdZ)) { cross = Base::Vector3d(1.0,0.0,0.0); } else { cross.Normalize(); cross = cross.Cross(stdZ); } if (cross.IsEqual(stdOrg,FLT_EPSILON)) { viewAxis = gp_Ax2(inputCenter, gp_Dir(flipDirection.x, flipDirection.y, flipDirection.z)); return viewAxis; } viewAxis = gp_Ax2(inputCenter, gp_Dir(flipDirection.x, flipDirection.y, flipDirection.z), gp_Dir(cross.x, cross.y, cross.z)); return viewAxis; } //! gets a coordinate system specified by Z and X directions //getViewAxis 2 gp_Ax2 TechDrawGeometry::getViewAxis(const Base::Vector3d origin, const Base::Vector3d& direction, const Base::Vector3d& xAxis, const bool flip) { (void) flip; gp_Pnt inputCenter(origin.x,origin.y,origin.z); Base::Vector3d flipDirection(direction.x,-direction.y,direction.z); if (!flip) { flipDirection = Base::Vector3d(direction.x,direction.y,direction.z); } gp_Ax2 viewAxis; viewAxis = gp_Ax2(inputCenter, gp_Dir(flipDirection.x, flipDirection.y, flipDirection.z), gp_Dir(xAxis.x, xAxis.y, xAxis.z)); return viewAxis; } //! Returns the centroid of shape, as viewed according to direction gp_Pnt TechDrawGeometry::findCentroid(const TopoDS_Shape &shape, const Base::Vector3d &direction) { Base::Vector3d origin(0.0,0.0,0.0); gp_Ax2 viewAxis = getViewAxis(origin,direction); return findCentroid(shape,viewAxis); } //! Returns the centroid of shape, as viewed according to direction gp_Pnt TechDrawGeometry::findCentroid(const TopoDS_Shape &shape, const gp_Ax2 viewAxis) { // Base::Vector3d origin(0.0,0.0,0.0); // gp_Ax2 viewAxis = getViewAxis(origin,direction); gp_Trsf tempTransform; tempTransform.SetTransformation(viewAxis); BRepBuilderAPI_Transform builder(shape, tempTransform); Bnd_Box tBounds; BRepBndLib::Add(builder.Shape(), tBounds); tBounds.SetGap(0.0); Standard_Real xMin, yMin, zMin, xMax, yMax, zMax; tBounds.Get(xMin, yMin, zMin, xMax, yMax, zMax); Standard_Real x = (xMin + xMax) / 2.0, y = (yMin + yMax) / 2.0, z = (zMin + zMax) / 2.0; // Get centroid back into object space tempTransform.Inverted().Transforms(x, y, z); return gp_Pnt(x, y, z); } Base::Vector3d TechDrawGeometry::findCentroidVec(const TopoDS_Shape &shape, const Base::Vector3d &direction) { gp_Pnt p = TechDrawGeometry::findCentroid(shape,direction); Base::Vector3d result(p.X(),p.Y(),p.Z()); return result; } //!scales & mirrors a shape about a center TopoDS_Shape TechDrawGeometry::mirrorShape(const TopoDS_Shape &input, const gp_Pnt& inputCenter, double scale) { TopoDS_Shape transShape; if (input.IsNull()) { return transShape; } try { // Make tempTransform scale the object around it's centre point and // mirror about the Y axis gp_Trsf tempTransform; //BRepBuilderAPI_Transform will loop forever if asked to use 0.0 as scale if (!(scale > 0.0)) { tempTransform.SetScale(inputCenter, 1.0); } else { tempTransform.SetScale(inputCenter, scale); } gp_Trsf mirrorTransform; mirrorTransform.SetMirror( gp_Ax2(inputCenter, gp_Dir(0, -1, 0)) ); tempTransform.Multiply(mirrorTransform); // Apply that transform to the shape. This should preserve the centre. BRepBuilderAPI_Transform mkTrf(input, tempTransform); transShape = mkTrf.Shape(); } catch (...) { Base::Console().Log("GeometryObject::mirrorShape - mirror/scale failed.\n"); return transShape; } return transShape; } //!rotates a shape about a viewAxis TopoDS_Shape TechDrawGeometry::rotateShape(const TopoDS_Shape &input, gp_Ax2& viewAxis, double rotAngle) { TopoDS_Shape transShape; if (input.IsNull()) { return transShape; } gp_Ax1 rotAxis = viewAxis.Axis(); double rotation = rotAngle * M_PI/180.0; try { gp_Trsf tempTransform; tempTransform.SetRotation(rotAxis,rotation); BRepBuilderAPI_Transform mkTrf(input, tempTransform); transShape = mkTrf.Shape(); } catch (...) { Base::Console().Log("GeometryObject::rotateShape - rotate failed.\n"); return transShape; } return transShape; } //!scales a shape about a origin TopoDS_Shape TechDrawGeometry::scaleShape(const TopoDS_Shape &input, double scale) { TopoDS_Shape transShape; try { gp_Trsf scaleTransform; scaleTransform.SetScale(gp_Pnt(0,0,0), scale); BRepBuilderAPI_Transform mkTrf(input, scaleTransform); transShape = mkTrf.Shape(); } catch (...) { Base::Console().Log("GeometryObject::scaleShape - scale failed.\n"); return transShape; } return transShape; } //!moves a shape TopoDS_Shape TechDrawGeometry::moveShape(const TopoDS_Shape &input, const Base::Vector3d& motion) { TopoDS_Shape transShape; try { gp_Trsf xlate; xlate.SetTranslation(gp_Vec(motion.x,motion.y,motion.z)); BRepBuilderAPI_Transform mkTrf(input, xlate); transShape = mkTrf.Shape(); } catch (...) { Base::Console().Log("GeometryObject::moveShape - move failed.\n"); return transShape; } return transShape; }