/*************************************************************************** * Copyright (c) 2013 Luke Parry * * 2022 WandererFan * * * * 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 #endif #include #include #include #include #include #include #include #include #include #include #include // generated from DrawViewDimensionPy.xml #include "DrawViewDimension.h" #include "DimensionFormatter.h" #include "DrawUtil.h" #include "DrawViewPart.h" #include "Geometry.h" #include "GeometryMatcher.h" #include "Preferences.h" using namespace TechDraw; using namespace Part; using DU = DrawUtil; //=========================================================================== // DrawViewDimension //=========================================================================== PROPERTY_SOURCE(TechDraw::DrawViewDimension, TechDraw::DrawView) const char* DrawViewDimension::TypeEnums[] = {"Distance", "DistanceX", "DistanceY", "DistanceZ", "Radius", "Diameter", "Angle", "Angle3Pt", nullptr}; const char* DrawViewDimension::MeasureTypeEnums[] = {"True", "Projected", nullptr}; // constraint to set the step size to 0.1 static const App::PropertyQuantityConstraint::Constraints ToleranceConstraint = {-DBL_MAX, DBL_MAX, 0.1}; // constraint to force positive values static const App::PropertyQuantityConstraint::Constraints PositiveConstraint = {0.0, DBL_MAX, 0.1}; DrawViewDimension::DrawViewDimension() { //create the formatter since it will be needed to set default property values m_formatter = new DimensionFormatter(this); ADD_PROPERTY_TYPE(References2D, (nullptr, nullptr), "", (App::Prop_None), "Projected Geometry References"); References2D.setScope(App::LinkScope::Global); ADD_PROPERTY_TYPE(References3D, (nullptr, nullptr), "", (App::Prop_None), "3D Geometry References"); References3D.setScope(App::LinkScope::Global); ADD_PROPERTY_TYPE(FormatSpec, (getDefaultFormatSpec()), "Format", App::Prop_Output, "Dimension format"); ADD_PROPERTY_TYPE(FormatSpecOverTolerance, (getDefaultFormatSpec(true)), "Format", App::Prop_Output, "Dimension overtolerance format"); ADD_PROPERTY_TYPE(FormatSpecUnderTolerance, (getDefaultFormatSpec(true)), "Format", App::Prop_Output, "Dimension undertolerance format"); ADD_PROPERTY_TYPE(Arbitrary, (false), "Format", App::Prop_Output, "Value overridden by user"); ADD_PROPERTY_TYPE(ArbitraryTolerances, (false), "Format", App::Prop_Output, "Tolerance values overridden by user"); Type.setEnums(TypeEnums);//dimension type: length, radius etc ADD_PROPERTY(Type, ((long)0)); MeasureType.setEnums(MeasureTypeEnums); ADD_PROPERTY(MeasureType, ((long)1));//Projected (or True) measurement ADD_PROPERTY_TYPE(TheoreticalExact, (false), "", App::Prop_Output, "If theoretical exact (basic) dimension"); ADD_PROPERTY_TYPE(EqualTolerance, (true), "", App::Prop_Output, "If over- and undertolerance are equal"); ADD_PROPERTY_TYPE(OverTolerance, (0.0), "", App::Prop_Output, "Overtolerance value\nIf 'Equal Tolerance' is true this is also\nthe negated " "value for 'Under Tolerance'"); OverTolerance.setUnit(Base::Unit::Length); OverTolerance.setConstraints(&ToleranceConstraint); ADD_PROPERTY_TYPE(UnderTolerance, (0.0), "", App::Prop_Output, "Undertolerance value\nIf 'Equal Tolerance' is true it will be replaced\nby " "negative value of 'Over Tolerance'"); UnderTolerance.setUnit(Base::Unit::Length); UnderTolerance.setConstraints(&ToleranceConstraint); ADD_PROPERTY_TYPE(Inverted, (false), "", App::Prop_Output, "The dimensional value is displayed inverted"); ADD_PROPERTY_TYPE(AngleOverride, (false), "Override", App::Prop_Output, "User specified angles"); ADD_PROPERTY_TYPE(LineAngle, (0.0), "Override", App::Prop_Output, "Dimension line angle"); ADD_PROPERTY_TYPE(ExtensionAngle, (0.0), "Override", App::Prop_Output, "Extension line angle"); ADD_PROPERTY_TYPE(SavedGeometry, () ,"References",(App::PropertyType)(App::Prop_None),"Reference Geometry"); SavedGeometry.setOrderRelevant(true); // hide the DrawView properties that don't apply to Dimensions ScaleType.setStatus(App::Property::ReadOnly, true); ScaleType.setStatus(App::Property::Hidden, true); Scale.setStatus(App::Property::ReadOnly, true); Scale.setStatus(App::Property::Hidden, true); Rotation.setStatus(App::Property::ReadOnly, true); Rotation.setStatus(App::Property::Hidden, true); Caption.setStatus(App::Property::Hidden, true); LockPosition.setStatus(App::Property::Hidden, true); // by default EqualTolerance is true, thus make UnderTolerance read-only UnderTolerance.setStatus(App::Property::ReadOnly, true); FormatSpecUnderTolerance.setStatus(App::Property::ReadOnly, true); measurement = new Measure::Measurement(); //TODO: should have better initial datumLabel position than (0, 0) in the DVP?? something closer to the object being measured? //initialize the descriptive geometry. //TODO: should this be more like DVP with a "geometry object"? resetLinear(); resetAngular(); resetArc(); m_hasGeometry = false; m_matcher = new GeometryMatcher(this); m_referencesCorrect = true; } DrawViewDimension::~DrawViewDimension() { delete measurement; measurement = nullptr; delete m_formatter; delete m_matcher; } void DrawViewDimension::resetLinear() { m_linearPoints.first(Base::Vector3d(0, 0, 0)); m_linearPoints.second(Base::Vector3d(0, 0, 0)); } void DrawViewDimension::resetAngular() { m_anglePoints.first(Base::Vector3d(0, 0, 0)); m_anglePoints.second(Base::Vector3d(0, 0, 0)); m_anglePoints.vertex(Base::Vector3d(0, 0, 0)); } void DrawViewDimension::resetArc() { m_arcPoints.isArc = false; m_arcPoints.center = Base::Vector3d(0, 0, 0); m_arcPoints.onCurve.first(Base::Vector3d(0, 0, 0)); m_arcPoints.onCurve.second(Base::Vector3d(0, 0, 0)); m_arcPoints.arcEnds.first(Base::Vector3d(0, 0, 0)); m_arcPoints.arcEnds.second(Base::Vector3d(0, 0, 0)); m_arcPoints.midArc = Base::Vector3d(0, 0, 0); m_arcPoints.arcCW = false; } void DrawViewDimension::onChanged(const App::Property* prop) { if (prop == &References3D) { //have to rebuild the Measurement object clear3DMeasurements();//Measurement object if (!(References3D.getValues()).empty()) { setAll3DMeasurement(); } } if (isRestoring()) { DrawView::onChanged(prop); return; } if (prop == &References2D) { updateSavedGeometry(); } else if (prop == &References3D) { // remove the old measurement object clear3DMeasurements(); if (!(References3D.getValues()).empty()) { // rebuild the Measurement object setAll3DMeasurement(); } else if (MeasureType.isValue("True")) {//empty 3dRefs, but True MeasureType.touch(); //run MeasureType logic for this case } updateSavedGeometry(); return; } else if (prop == &Type) {//why?? FormatSpec.setValue(getDefaultFormatSpec().c_str()); DimensionType type = static_cast(Type.getValue()); if (type == DimensionType::Angle || type == DimensionType::Angle3Pt) { OverTolerance.setUnit(Base::Unit::Angle); UnderTolerance.setUnit(Base::Unit::Angle); } else { OverTolerance.setUnit(Base::Unit::Length); UnderTolerance.setUnit(Base::Unit::Length); } return; } else if (prop == &TheoreticalExact) { // if TheoreticalExact disable tolerances and set them to zero if (TheoreticalExact.getValue()) { OverTolerance.setValue(0.0); UnderTolerance.setValue(0.0); OverTolerance.setReadOnly(true); UnderTolerance.setReadOnly(true); FormatSpecOverTolerance.setReadOnly(true); FormatSpecUnderTolerance.setReadOnly(true); ArbitraryTolerances.setValue(false); ArbitraryTolerances.setReadOnly(true); } else { OverTolerance.setReadOnly(false); FormatSpecOverTolerance.setReadOnly(false); ArbitraryTolerances.setReadOnly(false); if (!EqualTolerance.getValue()) { UnderTolerance.setReadOnly(false); FormatSpecUnderTolerance.setReadOnly(false); } } return; } else if (prop == &EqualTolerance) { // if EqualTolerance set negated overtolerance for untertolerance // then also the OverTolerance must be positive if (EqualTolerance.getValue()) { // if OverTolerance is negative or zero, first set it to zero if (OverTolerance.getValue() < 0) { OverTolerance.setValue(0.0); } OverTolerance.setConstraints(&PositiveConstraint); UnderTolerance.setValue(-1.0 * OverTolerance.getValue()); UnderTolerance.setUnit(OverTolerance.getUnit()); UnderTolerance.setReadOnly(true); FormatSpecUnderTolerance.setValue(FormatSpecOverTolerance.getValue()); FormatSpecUnderTolerance.setReadOnly(true); } else { OverTolerance.setConstraints(&ToleranceConstraint); if (!TheoreticalExact.getValue()) { UnderTolerance.setReadOnly(false); FormatSpecUnderTolerance.setReadOnly(false); } } return; } else if (prop == &OverTolerance) { // if EqualTolerance set negated overtolerance for untertolerance if (EqualTolerance.getValue()) { UnderTolerance.setValue(-1.0 * OverTolerance.getValue()); UnderTolerance.setUnit(OverTolerance.getUnit()); } return; } else if (prop == &FormatSpecOverTolerance) { if (!ArbitraryTolerances.getValue()) { FormatSpecUnderTolerance.setValue(FormatSpecOverTolerance.getValue()); } return; } else if (prop == &FormatSpecUnderTolerance) { if (!ArbitraryTolerances.getValue()) { FormatSpecOverTolerance.setValue(FormatSpecUnderTolerance.getValue()); } return; } DrawView::onChanged(prop); } void DrawViewDimension::Restore(Base::XMLReader& reader) // Old drawings did not have the equal tolerance options. // We cannot just introduce it as being set to true because that would e.g. destroy tolerances like +1-2 // Therefore set it to false for existing documents { EqualTolerance.setValue(false); DrawView::Restore(reader); } void DrawViewDimension::onDocumentRestored() { if (has3DReferences()) { setAll3DMeasurement(); } DimensionType type = static_cast(Type.getValue()); if (type == DimensionType::Angle || type == DimensionType::Angle3Pt) { OverTolerance.setUnit(Base::Unit::Angle); UnderTolerance.setUnit(Base::Unit::Angle); } } void DrawViewDimension::handleChangedPropertyType(Base::XMLReader& reader, const char* TypeName, App::Property* prop) { if (prop == &OverTolerance && strcmp(TypeName, "App::PropertyFloat") == 0) { App::PropertyFloat v; v.Restore(reader); OverTolerance.setValue(v.getValue()); } else if (prop == &UnderTolerance && strcmp(TypeName, "App::PropertyFloat") == 0) { App::PropertyFloat v; v.Restore(reader); UnderTolerance.setValue(v.getValue()); } else { TechDraw::DrawView::handleChangedPropertyType(reader, TypeName, prop); } // Over/Undertolerance were further changed from App::PropertyQuantity to App::PropertyQuantityConstraint if (prop == &OverTolerance && strcmp(TypeName, "App::PropertyQuantity") == 0) { App::PropertyQuantity OverToleranceProperty; // restore the PropertyQuantity to be able to set its value OverToleranceProperty.Restore(reader); OverTolerance.setValue(OverToleranceProperty.getValue()); } else if (prop == &UnderTolerance && strcmp(TypeName, "App::PropertyQuantity") == 0) { App::PropertyQuantity UnderToleranceProperty; // restore the PropertyQuantity to be able to set its value UnderToleranceProperty.Restore(reader); UnderTolerance.setValue(UnderToleranceProperty.getValue()); } } short DrawViewDimension::mustExecute() const { if (!isRestoring()) { if (References2D.isTouched() || References3D.isTouched() || Type.isTouched()) { return true; } } return DrawView::mustExecute(); } App::DocumentObjectExecReturn* DrawViewDimension::execute() { // Base::Console().Message("DVD::execute() - %s\n", getNameInDocument()); if (!okToProceed()) { return App::DocumentObject::StdReturn; } resetLinear(); resetAngular(); resetArc(); m_referencesCorrect = compareSavedGeometry(); if (!m_referencesCorrect) { m_referencesCorrect = fixExactMatch(); if (!m_referencesCorrect) { handleNoExactMatch(); } } if (References3D.getValues().empty() && !checkReferences2D()) { Base::Console().Warning("%s has invalid 2D References\n", getNameInDocument()); return new App::DocumentObjectExecReturn("Dimension object has invalid 2d references"); } //we have either or both valid References3D and References2D ReferenceVector references = getEffectiveReferences(); if (Type.isValue("Distance") || Type.isValue("DistanceX") || Type.isValue("DistanceY")) { if (getRefType() == oneEdge) { m_linearPoints = getPointsOneEdge(references); } else if (getRefType() == twoEdge) { m_linearPoints = getPointsTwoEdges(references); } else if (getRefType() == twoVertex) { m_linearPoints = getPointsTwoVerts(references); } else if (getRefType() == vertexEdge) { m_linearPoints = getPointsEdgeVert(references); } m_hasGeometry = true; } else if (Type.isValue("Radius")) { m_arcPoints = getArcParameters(references); m_hasGeometry = true; } else if (Type.isValue("Diameter")) { m_arcPoints = getArcParameters(references); m_hasGeometry = true; } else if (Type.isValue("Angle")) { if (getRefType() != twoEdge) { throw Base::RuntimeError("Angle dimension has non-edge references"); } m_anglePoints = getAnglePointsTwoEdges(references); m_hasGeometry = true; } else if (Type.isValue("Angle3Pt")) { if (getRefType() != threeVertex) { throw Base::RuntimeError("3 point angle dimension has non-vertex references"); } m_anglePoints = getAnglePointsThreeVerts(references); m_hasGeometry = true; } overrideKeepUpdated(false); return DrawView::execute(); } // true if we have enough information to execute, false otherwise bool DrawViewDimension::okToProceed() { if (!keepUpdated()) { return false; } DrawViewPart* dvp = getViewPart(); if (!dvp) return false; if (!has2DReferences() && !has3DReferences()) { //no references, can't do anything return App::DocumentObject::StdReturn; } if (!getViewPart()->hasGeometry()) { //can't do anything until Source has geometry return false; } return true; } bool DrawViewDimension::isMultiValueSchema() const { return m_formatter->isMultiValueSchema(); } std::string DrawViewDimension::formatValue(qreal value, QString qFormatSpec, int partial, bool isDim) { return m_formatter->formatValue(value, qFormatSpec, partial, isDim); } bool DrawViewDimension::haveTolerance() { //if a numeric tolerance is specified AND //tolerances are NOT arbitrary if ((!DrawUtil::fpCompare(OverTolerance.getValue(), 0.0) || !DrawUtil::fpCompare(UnderTolerance.getValue(), 0.0)) && !ArbitraryTolerances.getValue()) { return true; } return false; } std::string DrawViewDimension::getFormattedToleranceValue(int partial) { return m_formatter->getFormattedToleranceValue(partial); } ////get over and under tolerances std::pair DrawViewDimension::getFormattedToleranceValues(int partial) { return m_formatter->getFormattedToleranceValues(partial); } ////partial = 2 unit only std::string DrawViewDimension::getFormattedDimensionValue(int partial) { return m_formatter->getFormattedDimensionValue(partial); } QStringList DrawViewDimension::getPrefixSuffixSpec(QString fSpec) { return m_formatter->getPrefixSuffixSpec(fSpec); } //!NOTE: this returns the Dimension value in internal units (ie mm)!!!! double DrawViewDimension::getDimValue() { // Base::Console().Message("DVD::getDimValue()\n"); double result = 0.0; if (!has2DReferences() && !has3DReferences()) { //nothing to measure return result; } if (!getViewPart()) { return result; } if (!getViewPart()->hasGeometry()) { //happens when loading saved document return result; } if (MeasureType.isValue("True")) { // True Values if (!measurement->has3DReferences()) { Base::Console().Warning("%s - True dimension has no 3D References\n", getNameInDocument()); return result; } if (Type.isValue("Distance") || Type.isValue("DistanceX") || Type.isValue("DistanceY")) { result = measurement->length(); } else if (Type.isValue("Radius")) { result = measurement->radius(); } else if (Type.isValue("Diameter")) { result = 2.0 * measurement->radius(); } else if (Type.isValue("Angle") || Type.isValue("Angle3Pt")) { result = measurement->angle(); } else {//tarfu throw Base::ValueError("getDimValue() - Unknown Dimension Type (3)"); } } else { // Projected Values if (!checkReferences2D()) { Base::Console().Warning("DVD::getDimValue - %s - 2D references are corrupt (5)\n", getNameInDocument()); return result; } if (Type.isValue("Distance") || Type.isValue("DistanceX") || Type.isValue("DistanceY")) { pointPair pts = getLinearPoints(); Base::Vector3d dimVec = pts.first() - pts.second(); if (Type.isValue("Distance")) { result = dimVec.Length() / getViewPart()->getScale(); } else if (Type.isValue("DistanceX")) { result = fabs(dimVec.x) / getViewPart()->getScale(); } else { result = fabs(dimVec.y) / getViewPart()->getScale(); } } else if (Type.isValue("Radius")) { arcPoints pts = m_arcPoints; result = pts.radius / getViewPart()->getScale();//Projected BaseGeom is scaled for drawing } else if (Type.isValue("Diameter")) { arcPoints pts = m_arcPoints; result = (pts.radius * 2.0) / getViewPart()->getScale();//Projected BaseGeom is scaled for drawing } else if (Type.isValue("Angle") || Type.isValue("Angle3Pt")) {//same as case "Angle"? anglePoints pts = m_anglePoints; Base::Vector3d vertex = pts.vertex(); Base::Vector3d leg0 = pts.first() - vertex; Base::Vector3d leg1 = pts.second() - vertex; double legAngle = leg0.GetAngle(leg1) * 180.0 / M_PI; result = legAngle; } } result = fabs(result); if (Inverted.getValue()) { if (Type.isValue("Angle") || Type.isValue("Angle3Pt")) { result = 360 - result; } else { result = -result; } } return result; } pointPair DrawViewDimension::getPointsOneEdge(ReferenceVector references) { // Base::Console().Message("DVD::getPointsOneEdge()\n"); App::DocumentObject* refObject = references.front().getObject(); int iSubelement = DrawUtil::getIndexFromName(references.front().getSubName()); if (refObject->isDerivedFrom(TechDraw::DrawViewPart::getClassTypeId()) && !references.at(0).getSubName().empty()) { //TODO: Notify if not straight line Edge? //this is a 2d object (a DVP + subelements) TechDraw::BaseGeomPtr geom = getViewPart()->getGeomByIndex(iSubelement); if (!geom) { std::stringstream ssMessage; ssMessage << getNameInDocument() << " can not find geometry for 2d reference (1)"; throw Base::RuntimeError(ssMessage.str()); } if (geom->getGeomType() != TechDraw::GeomType::GENERIC) { std::stringstream ssMessage; ssMessage << getNameInDocument() << " 2d reference is a " << geom->geomTypeName(); throw Base::RuntimeError(ssMessage.str()); } TechDraw::GenericPtr generic = std::static_pointer_cast(geom); return {generic->points[0], generic->points[1]}; } //this is a 3d object //get the endpoints of the edge in the DVP's coordinates Base::Vector3d edgeEnd0, edgeEnd1; TopoDS_Shape geometry = references.front().getGeometry(); if (geometry.IsNull() || geometry.ShapeType() != TopAbs_EDGE) { throw Base::RuntimeError("Geometry for dimension reference is null."); } const TopoDS_Edge& edge = TopoDS::Edge(geometry); gp_Pnt gEnd0 = BRep_Tool::Pnt(TopExp::FirstVertex(edge)); gp_Pnt gEnd1 = BRep_Tool::Pnt(TopExp::LastVertex(edge)); pointPair pts(DrawUtil::toVector3d(gEnd0), DrawUtil::toVector3d(gEnd1)); pts.move(getViewPart()->getCurrentCentroid()); pts.project(getViewPart()); return pts; } pointPair DrawViewDimension::getPointsTwoEdges(ReferenceVector references) { // Base::Console().Message("DVD::getPointsTwoEdges() - %s\n", getNameInDocument()); App::DocumentObject* refObject = references.front().getObject(); int iSubelement0 = DrawUtil::getIndexFromName(references.at(0).getSubName()); int iSubelement1 = DrawUtil::getIndexFromName(references.at(1).getSubName()); if (refObject->isDerivedFrom(TechDraw::DrawViewPart::getClassTypeId()) && !references.at(0).getSubName().empty()) { //this is a 2d object (a DVP + subelements) TechDraw::BaseGeomPtr geom0 = getViewPart()->getGeomByIndex(iSubelement0); TechDraw::BaseGeomPtr geom1 = getViewPart()->getGeomByIndex(iSubelement1); if (!geom0 || !geom1) { std::stringstream ssMessage; ssMessage << getNameInDocument() << " can not find geometry for 2d reference (2)"; throw Base::RuntimeError(ssMessage.str()); } return closestPoints(geom0->getOCCEdge(), geom1->getOCCEdge()); } //this is a 3d object TopoDS_Shape geometry0 = references.at(0).getGeometry(); TopoDS_Shape geometry1 = references.at(1).getGeometry(); if (geometry0.IsNull() || geometry1.IsNull() || geometry0.ShapeType() != TopAbs_EDGE || geometry1.ShapeType() != TopAbs_EDGE) { throw Base::RuntimeError("Geometry for dimension reference is null."); } pointPair pts = closestPoints(geometry0, geometry1); pts.move(getViewPart()->getCurrentCentroid()); pts.project(getViewPart()); return pts; } pointPair DrawViewDimension::getPointsTwoVerts(ReferenceVector references) { // Base::Console().Message("DVD::getPointsTwoVerts() - %s\n", getNameInDocument()); App::DocumentObject* refObject = references.front().getObject(); int iSubelement0 = DrawUtil::getIndexFromName(references.at(0).getSubName()); int iSubelement1 = DrawUtil::getIndexFromName(references.at(1).getSubName()); if (refObject->isDerivedFrom(TechDraw::DrawViewPart::getClassTypeId()) && !references.at(0).getSubName().empty()) { //this is a 2d object (a DVP + subelements) TechDraw::VertexPtr v0 = getViewPart()->getProjVertexByIndex(iSubelement0); TechDraw::VertexPtr v1 = getViewPart()->getProjVertexByIndex(iSubelement1); if (!v0 || !v1) { std::stringstream ssMessage; ssMessage << getNameInDocument() << " can not find geometry for 2d reference (3)"; throw Base::RuntimeError(ssMessage.str()); } return {v0->point(), v1->point()}; } //this is a 3d object TopoDS_Shape geometry0 = references.at(0).getGeometry(); TopoDS_Shape geometry1 = references.at(1).getGeometry(); if (geometry0.IsNull() || geometry1.IsNull() || geometry0.ShapeType() != TopAbs_VERTEX || geometry1.ShapeType() != TopAbs_VERTEX) { throw Base::RuntimeError("Geometry for dimension reference is null."); } const TopoDS_Vertex& vertex0 = TopoDS::Vertex(geometry0); const TopoDS_Vertex& vertex1 = TopoDS::Vertex(geometry1); gp_Pnt gPoint0 = BRep_Tool::Pnt(vertex0); gp_Pnt gPoint1 = BRep_Tool::Pnt(vertex1); pointPair pts(DrawUtil::toVector3d(gPoint0), DrawUtil::toVector3d(gPoint1)); pts.move(getViewPart()->getCurrentCentroid()); pts.project(getViewPart()); return pts; } pointPair DrawViewDimension::getPointsEdgeVert(ReferenceVector references) { // Base::Console().Message("DVD::getPointsEdgeVert() - %s\n", getNameInDocument()); App::DocumentObject* refObject = references.front().getObject(); int iSubelement0 = DrawUtil::getIndexFromName(references.at(0).getSubName()); int iSubelement1 = DrawUtil::getIndexFromName(references.at(1).getSubName()); if (refObject->isDerivedFrom(TechDraw::DrawViewPart::getClassTypeId()) && !references.at(0).getSubName().empty()) { //this is a 2d object (a DVP + subelements) TechDraw::BaseGeomPtr edge; TechDraw::VertexPtr vertex; if (DrawUtil::getGeomTypeFromName(references.at(0).getSubName()) == "Edge") { edge = getViewPart()->getGeomByIndex(iSubelement0); vertex = getViewPart()->getProjVertexByIndex(iSubelement1); } else { edge = getViewPart()->getGeomByIndex(iSubelement1); vertex = getViewPart()->getProjVertexByIndex(iSubelement0); } if (!vertex || !edge) { throw Base::RuntimeError("Missing geometry for dimension (4)"); } return closestPoints(edge->getOCCEdge(), vertex->getOCCVertex()); } //this is a 3d object TopoDS_Shape geometry0 = references.at(0).getGeometry(); TopoDS_Shape geometry1 = references.at(1).getGeometry(); if (geometry0.IsNull() || geometry1.IsNull() || geometry0.ShapeType() != TopAbs_VERTEX || geometry1.ShapeType() != TopAbs_VERTEX) { throw Base::RuntimeError("Geometry for dimension reference is null."); } pointPair pts = closestPoints(geometry0, geometry1); pts.move(getViewPart()->getCurrentCentroid()); pts.project(getViewPart()); return pts; } arcPoints DrawViewDimension::getArcParameters(ReferenceVector references) { // Base::Console().Message("DVD::getArcParameters()\n"); App::DocumentObject* refObject = references.front().getObject(); int iSubelement = DrawUtil::getIndexFromName(references.front().getSubName()); if (refObject->isDerivedFrom(TechDraw::DrawViewPart::getClassTypeId()) && !references.at(0).getSubName().empty()) { //this is a 2d object (a DVP + subelements) TechDraw::BaseGeomPtr geom = getViewPart()->getGeomByIndex(iSubelement); if (!geom) { std::stringstream ssMessage; ssMessage << getNameInDocument() << " can not find geometry for 2d reference (4)"; throw Base::RuntimeError(ssMessage.str()); } return arcPointsFromBaseGeom(getViewPart()->getGeomByIndex(iSubelement)); } //this is a 3d reference TopoDS_Shape geometry = references.front().getGeometry(); if (geometry.IsNull() || geometry.ShapeType() != TopAbs_EDGE) { throw Base::RuntimeError("Geometry for dimension reference is null."); } const TopoDS_Edge& edge = TopoDS::Edge(geometry); arcPoints pts = arcPointsFromEdge(edge); pts.move(getViewPart()->getCurrentCentroid()); pts.project(getViewPart()); return pts; } arcPoints DrawViewDimension::arcPointsFromBaseGeom(TechDraw::BaseGeomPtr base) { TechDraw::CirclePtr circle; arcPoints pts; pts.center = Base::Vector3d(0.0, 0.0, 0.0); pts.radius = 0.0; if ((base && base->getGeomType() == TechDraw::GeomType::CIRCLE) || (base && base->getGeomType() == TechDraw::GeomType::ARCOFCIRCLE)) { circle = std::static_pointer_cast(base); pts.center = Base::Vector3d(circle->center.x, circle->center.y, 0.0); pts.radius = circle->radius; if (base->getGeomType() == TechDraw::GeomType::ARCOFCIRCLE) { TechDraw::AOCPtr aoc = std::static_pointer_cast(circle); pts.isArc = true; pts.onCurve.first(Base::Vector3d(aoc->midPnt.x, aoc->midPnt.y, 0.0)); pts.midArc = Base::Vector3d(aoc->midPnt.x, aoc->midPnt.y, 0.0); pts.arcEnds.first(Base::Vector3d(aoc->startPnt.x, aoc->startPnt.y, 0.0)); pts.arcEnds.second(Base::Vector3d(aoc->endPnt.x, aoc->endPnt.y, 0.0)); pts.arcCW = aoc->cw; } else { pts.isArc = false; pts.onCurve.first(pts.center + Base::Vector3d(1, 0, 0) * circle->radius);//arbitrary point on edge pts.onCurve.second( pts.center + Base::Vector3d(-1, 0, 0) * circle->radius);//arbitrary point on edge } } else if ((base && base->getGeomType() == TechDraw::GeomType::ELLIPSE) || (base && base->getGeomType() == TechDraw::GeomType::ARCOFELLIPSE)) { TechDraw::EllipsePtr ellipse = std::static_pointer_cast(base); if (ellipse->closed()) { double r1 = ellipse->minor; double r2 = ellipse->major; double rAvg = (r1 + r2) / 2.0; pts.center = Base::Vector3d(ellipse->center.x, ellipse->center.y, 0.0); pts.radius = rAvg; pts.isArc = false; pts.onCurve.first(pts.center + Base::Vector3d(1, 0, 0) * rAvg);//arbitrary point on edge pts.onCurve.second(pts.center + Base::Vector3d(-1, 0, 0) * rAvg);//arbitrary point on edge } else { TechDraw::AOEPtr aoe = std::static_pointer_cast(base); double r1 = aoe->minor; double r2 = aoe->major; double rAvg = (r1 + r2) / 2.0; pts.isArc = true; pts.center = Base::Vector3d(aoe->center.x, aoe->center.y, 0.0); pts.radius = rAvg; pts.arcEnds.first(Base::Vector3d(aoe->startPnt.x, aoe->startPnt.y, 0.0)); pts.arcEnds.second(Base::Vector3d(aoe->endPnt.x, aoe->endPnt.y, 0.0)); pts.midArc = Base::Vector3d(aoe->midPnt.x, aoe->midPnt.y, 0.0); pts.arcCW = aoe->cw; pts.onCurve.first(Base::Vector3d(aoe->midPnt.x, aoe->midPnt.y, 0.0));//for radius // pts.onCurve.first(pts.center + Base::Vector3d(1, 0,0) * rAvg); //for diameter pts.onCurve.second(pts.center + Base::Vector3d(-1, 0, 0) * rAvg);//arbitrary point on edge } } else if (base && base->getGeomType() == TechDraw::GeomType::BSPLINE) { TechDraw::BSplinePtr spline = std::static_pointer_cast(base); if (spline->isCircle()) { bool arc; double rad; Base::Vector3d center; //bool circ = GeometryUtils::getCircleParms(spline->getOCCEdge(), rad, center, arc); pts.center = Base::Vector3d(center.x, center.y, 0.0); pts.radius = rad; pts.arcEnds.first(Base::Vector3d(spline->startPnt.x, spline->startPnt.y, 0.0)); pts.arcEnds.second(Base::Vector3d(spline->endPnt.x, spline->endPnt.y, 0.0)); pts.midArc = Base::Vector3d(spline->midPnt.x, spline->midPnt.y, 0.0); pts.isArc = arc; pts.arcCW = spline->cw; if (arc) { pts.onCurve.first(Base::Vector3d(spline->midPnt.x, spline->midPnt.y, 0.0)); } else { pts.onCurve.first(pts.center + Base::Vector3d(1, 0, 0) * rad);//arbitrary point on edge pts.onCurve.second(pts.center + Base::Vector3d(-1, 0, 0) * rad);//arbitrary point on edge } } else { //fubar - can't have non-circular spline as target of Diameter dimension, but this is already //checked, so something has gone badly wrong. Base::Console().Error("%s: can not make a Circle from this BSpline edge\n", getNameInDocument()); throw Base::RuntimeError("Bad BSpline geometry for arc dimension"); } } else { std::stringstream ssMessage; ssMessage << getNameInDocument() << " 2d reference is a " << base->geomTypeName(); throw Base::RuntimeError(ssMessage.str()); } return pts; } arcPoints DrawViewDimension::arcPointsFromEdge(TopoDS_Edge occEdge) { arcPoints pts; pts.isArc = !BRep_Tool::IsClosed(occEdge); pts.arcCW = false; BRepAdaptor_Curve adapt(occEdge); double pFirst = adapt.FirstParameter(); double pLast = adapt.LastParameter(); double pMid = (pFirst + pLast) / 2.0; BRepLProp_CLProps props(adapt, pFirst, 0, Precision::Confusion()); pts.arcEnds.first(DrawUtil::toVector3d(props.Value())); props.SetParameter(pLast); pts.arcEnds.second(DrawUtil::toVector3d(props.Value())); props.SetParameter(pMid); pts.onCurve.first(DrawUtil::toVector3d(props.Value())); pts.onCurve.second(DrawUtil::toVector3d(props.Value())); pts.midArc = DrawUtil::toVector3d(props.Value()); if (adapt.GetType() == GeomAbs_Circle) { gp_Circ circle = adapt.Circle(); pts.center = DrawUtil::toVector3d(circle.Location()); pts.radius = circle.Radius(); if (pts.isArc) { //part of circle gp_Ax1 axis = circle.Axis(); gp_Vec startVec = DrawUtil::togp_Vec(pts.arcEnds.first() - pts.center); gp_Vec endVec = DrawUtil::togp_Vec(pts.arcEnds.second() - pts.center); double angle = startVec.AngleWithRef(endVec, axis.Direction().XYZ()); pts.arcCW = (angle < 0.0); } else { //full circle pts.onCurve.first(pts.center + Base::Vector3d(1, 0, 0) * pts.radius);//arbitrary point on edge pts.onCurve.second(pts.center + Base::Vector3d(-1, 0, 0) * pts.radius);//arbitrary point on edge } } else if (adapt.GetType() == GeomAbs_Ellipse) { gp_Elips ellipse = adapt.Ellipse(); pts.center = DrawUtil::toVector3d(ellipse.Location()); pts.radius = (ellipse.MajorRadius() + ellipse.MinorRadius()) / 2.0; if (pts.isArc) { //part of ellipse gp_Ax1 axis = ellipse.Axis(); gp_Vec startVec = DrawUtil::togp_Vec(pts.arcEnds.first() - pts.center); gp_Vec endVec = DrawUtil::togp_Vec(pts.arcEnds.second() - pts.center); double angle = startVec.AngleWithRef(endVec, axis.Direction().XYZ()); pts.arcCW = (angle < 0.0); } else { //full ellipse pts.onCurve.first(pts.center + Base::Vector3d(1, 0, 0) * pts.radius);//arbitrary point on edge pts.onCurve.second(pts.center + Base::Vector3d(-1, 0, 0) * pts.radius);//arbitrary point on edge } } else if (adapt.GetType() == GeomAbs_BSplineCurve) { if (GeometryUtils::isCircle(occEdge)) { bool isArc(false); TopoDS_Edge circleEdge = GeometryUtils::asCircle(occEdge, isArc); pts.isArc = isArc; BRepAdaptor_Curve adaptCircle(circleEdge); if (adaptCircle.GetType() != GeomAbs_Circle) { throw Base::RuntimeError("failed to get circle from bspline"); } gp_Circ circle = adapt.Circle(); //TODO: same code as above. reuse opportunity. pts.center = DrawUtil::toVector3d(circle.Location()); pts.radius = circle.Radius(); if (pts.isArc) { //part of circle gp_Ax1 axis = circle.Axis(); gp_Vec startVec = DrawUtil::togp_Vec(pts.arcEnds.first() - pts.center); gp_Vec endVec = DrawUtil::togp_Vec(pts.arcEnds.second() - pts.center); double angle = startVec.AngleWithRef(endVec, axis.Direction().XYZ()); pts.arcCW = (angle < 0.0); } else { //full circle pts.onCurve.first(pts.center + Base::Vector3d(1, 0, 0) * pts.radius);//arbitrary point on edge pts.onCurve.second( pts.center + Base::Vector3d(-1, 0, 0) * pts.radius);//arbitrary point on edge } } else { throw Base::RuntimeError("failed to make circle from bspline"); } } else { throw Base::RuntimeError("can not get arc points from this edge"); } return pts; } anglePoints DrawViewDimension::getAnglePointsTwoEdges(ReferenceVector references) { //Base::Console().Message("DVD::getAnglePointsTwoEdges() - %s\n", getNameInDocument()); App::DocumentObject* refObject = references.front().getObject(); int iSubelement0 = DrawUtil::getIndexFromName(references.at(0).getSubName()); int iSubelement1 = DrawUtil::getIndexFromName(references.at(1).getSubName()); if (refObject->isDerivedFrom(TechDraw::DrawViewPart::getClassTypeId()) && !references.at(0).getSubName().empty()) { //this is a 2d object (a DVP + subelements) TechDraw::BaseGeomPtr geom0 = getViewPart()->getGeomByIndex(iSubelement0); TechDraw::BaseGeomPtr geom1 = getViewPart()->getGeomByIndex(iSubelement1); if (!geom0 || !geom1) { std::stringstream ssMessage; ssMessage << getNameInDocument() << " can not find geometry for 2d reference (5)"; throw Base::RuntimeError(ssMessage.str()); } if (geom0->getGeomType() != TechDraw::GeomType::GENERIC) { std::stringstream ssMessage; ssMessage << getNameInDocument() << " first 2d reference is a " << geom0->geomTypeName(); throw Base::RuntimeError(ssMessage.str()); } if (geom1->getGeomType() != TechDraw::GeomType::GENERIC) { std::stringstream ssMessage; ssMessage << getNameInDocument() << " second 2d reference is a " << geom0->geomTypeName(); throw Base::RuntimeError(ssMessage.str()); } TechDraw::GenericPtr generic0 = std::static_pointer_cast(geom0); TechDraw::GenericPtr generic1 = std::static_pointer_cast(geom1); Base::Vector3d apex = generic0->apparentInter(generic1); Base::Vector3d farPoint0, farPoint1; //pick the end of generic0 farthest from the apex if ((generic0->getStartPoint() - apex).Length() > (generic0->getEndPoint() - apex).Length()) { farPoint0 = generic0->getStartPoint(); } else { farPoint0 = generic0->getEndPoint(); } //pick the end of generic1 farthest from the apex if ((generic1->getStartPoint() - apex).Length() > (generic1->getEndPoint() - apex).Length()) { farPoint1 = generic1->getStartPoint(); } else { farPoint1 = generic1->getEndPoint(); } Base::Vector3d leg0Dir = (generic0->getStartPoint() - generic0->getEndPoint()).Normalize(); Base::Vector3d leg1Dir = (generic1->getStartPoint() - generic1->getEndPoint()).Normalize(); if (DrawUtil::fpCompare(fabs(leg0Dir.Dot(leg1Dir)), 1.0)) { //legs of the angle are parallel. throw Base::RuntimeError("Can not make angle from parallel edges"); } Base::Vector3d extenPoint0 = farPoint0;//extension line points Base::Vector3d extenPoint1 = farPoint1; if (DrawUtil::fpCompare(fabs(leg0Dir.Dot(leg1Dir)), 0.0)) { //legs of angle are perpendicular farPoints will do } else { //legs of the angle are skew //project farthest points onto opposite edge Base::Vector3d projFar0OnLeg1 = farPoint0.Perpendicular(apex, leg1Dir); Base::Vector3d projFar1OnLeg0 = farPoint1.Perpendicular(apex, leg0Dir); if (DrawUtil::isBetween(projFar0OnLeg1, generic1->getStartPoint(), generic1->getEndPoint())) { extenPoint1 = projFar0OnLeg1; } else if (DrawUtil::isBetween(projFar1OnLeg0, generic0->getStartPoint(), generic0->getEndPoint())) { extenPoint0 = projFar1OnLeg0; } } anglePoints pts; pts.first(extenPoint0); pts.second(extenPoint1); pts.vertex(apex); return pts; } //this is a 3d object TopoDS_Shape geometry0 = references.at(0).getGeometry(); TopoDS_Shape geometry1 = references.at(1).getGeometry(); if (geometry0.IsNull() || geometry1.IsNull() || geometry0.ShapeType() != TopAbs_EDGE || geometry1.ShapeType() != TopAbs_EDGE) { throw Base::RuntimeError("Geometry for dimension reference is null."); } TopoDS_Edge edge0 = TopoDS::Edge(geometry0); BRepAdaptor_Curve adapt0(edge0); TopoDS_Edge edge1 = TopoDS::Edge(geometry1); BRepAdaptor_Curve adapt1(edge1); if (adapt0.GetType() != GeomAbs_Line || adapt1.GetType() != GeomAbs_Line) { throw Base::RuntimeError("Geometry for angle dimension must be lines."); } gp_Pnt gStart0 = BRep_Tool::Pnt(TopExp::FirstVertex(edge0)); gp_Pnt gEnd0 = BRep_Tool::Pnt(TopExp::LastVertex(edge0)); gp_Vec gDir0(gEnd0.XYZ() - gStart0.XYZ()); gp_Pnt gStart1 = BRep_Tool::Pnt(TopExp::FirstVertex(edge1)); gp_Pnt gEnd1 = BRep_Tool::Pnt(TopExp::LastVertex(edge1)); gp_Vec gDir1(gEnd1.XYZ() - gStart1.XYZ()); Base::Vector3d vApex; bool haveIntersection = DrawUtil::intersect2Lines3d( DrawUtil::toVector3d(gStart0), DrawUtil::toVector3d(gDir0), DrawUtil::toVector3d(gStart1), DrawUtil::toVector3d(gDir1), vApex); if (!haveIntersection) { throw Base::RuntimeError("Geometry for 3d angle dimension does not intersect"); } gp_Pnt gApex = DrawUtil::togp_Pnt(vApex); gp_Pnt gFar0 = gEnd0; if (gStart0.Distance(gApex) > gEnd0.Distance(gApex)) { gFar0 = gStart0; } gp_Pnt gFar1 = gEnd1; if (gStart1.Distance(gApex) > gEnd1.Distance(gApex)) { gFar1 = gStart1; } anglePoints pts(DrawUtil::toVector3d(gApex), DrawUtil::toVector3d(gFar0), DrawUtil::toVector3d(gFar1)); pts.move(getViewPart()->getCurrentCentroid()); pts.project(getViewPart()); return pts; } //TODO: this makes assumptions about the order of references (p - v - p). is this checked somewhere? anglePoints DrawViewDimension::getAnglePointsThreeVerts(ReferenceVector references) { // Base::Console().Message("DVD::getAnglePointsThreeVerts() - %s\n", getNameInDocument()); if (references.size() < 3) { throw Base::RuntimeError("Not enough references to make angle dimension"); } App::DocumentObject* refObject = references.front().getObject(); int iSubelement0 = DrawUtil::getIndexFromName(references.at(0).getSubName()); int iSubelement1 = DrawUtil::getIndexFromName(references.at(1).getSubName()); int iSubelement2 = DrawUtil::getIndexFromName(references.at(2).getSubName()); if (refObject->isDerivedFrom(TechDraw::DrawViewPart::getClassTypeId()) && !references.at(0).getSubName().empty()) { //this is a 2d object (a DVP + subelements) TechDraw::VertexPtr vert0 = getViewPart()->getProjVertexByIndex(iSubelement0); TechDraw::VertexPtr vert1 = getViewPart()->getProjVertexByIndex(iSubelement1); TechDraw::VertexPtr vert2 = getViewPart()->getProjVertexByIndex(iSubelement2); if (!vert0 || !vert1 || !vert2) { throw Base::RuntimeError("References for three point angle dimension are not vertices"); } anglePoints pts(vert1->point(), vert0->point(), vert2->point()); return pts; } //this is a 3d object TopoDS_Shape geometry0 = references.at(0).getGeometry(); TopoDS_Shape geometry1 = references.at(1).getGeometry(); TopoDS_Shape geometry2 = references.at(2).getGeometry(); if (geometry0.IsNull() || geometry1.IsNull() || geometry2.IsNull() || geometry0.ShapeType() != TopAbs_VERTEX || geometry1.ShapeType() != TopAbs_VERTEX || geometry2.ShapeType() != TopAbs_VERTEX) { throw Base::RuntimeError("Geometry for dimension reference is null."); } TopoDS_Vertex vertex0 = TopoDS::Vertex(geometry0); gp_Pnt point0 = BRep_Tool::Pnt(vertex0); TopoDS_Vertex vertex1 = TopoDS::Vertex(geometry1); gp_Pnt point1 = BRep_Tool::Pnt(vertex1); TopoDS_Vertex vertex2 = TopoDS::Vertex(geometry2); gp_Pnt point2 = BRep_Tool::Pnt(vertex2); anglePoints pts(DrawUtil::toVector3d(point1), DrawUtil::toVector3d(point0), DrawUtil::toVector3d(point2)); pts.move(getViewPart()->getCurrentCentroid()); pts.project(getViewPart()); return pts; } DrawViewPart* DrawViewDimension::getViewPart() const { if (References2D.getValues().empty()) { return nullptr; } return dynamic_cast(References2D.getValues().at(0)); } //return the references controlling this dimension. 3d references are used when available //otherwise 2d references are returned. no checking is performed. Result is pairs of (object, subName) ReferenceVector DrawViewDimension::getEffectiveReferences() const { // Base::Console().Message("DVD::getEffectiveReferences()\n"); const std::vector& objects3d = References3D.getValues(); const std::vector& subElements3d = References3D.getSubValues(); const std::vector& objects = References2D.getValues(); const std::vector& subElements = References2D.getSubValues(); ReferenceVector effectiveRefs; if (!objects3d.empty()) { //use 3d references by preference int refCount = objects3d.size(); for (int i = 0; i < refCount; i++) { ReferenceEntry ref(objects3d.at(i), std::string(subElements3d.at(i))); effectiveRefs.push_back(ref); } } else { //use 2d references if necessary int refCount = objects.size(); for (int i = 0; i < refCount; i++) { ReferenceEntry ref(objects.at(i), subElements.at(i)); effectiveRefs.push_back(ref); } } return effectiveRefs; } //return the 2d references as a ReferenceVector ReferenceVector DrawViewDimension::getReferences2d() const { const std::vector& objects = References2D.getValues(); const std::vector& subElements = References2D.getSubValues(); ReferenceVector refs2d; int refCount = objects.size(); for (int i = 0; i < refCount; i++) { ReferenceEntry ref(objects.at(i), subElements.at(i)); refs2d.push_back(ref); } return refs2d; } //return the 3d references as a ReferenceVector ReferenceVector DrawViewDimension::getReferences3d() const { const std::vector& objects3d = References3D.getValues(); const std::vector& subElements3d = References3D.getSubValues(); ReferenceVector refs3d; int refCount = objects3d.size(); for (int i = 0; i < refCount; i++) { ReferenceEntry ref(objects3d.at(i), subElements3d.at(i)); refs3d.push_back(ref); } return refs3d; } void DrawViewDimension::replaceReferenceSubElement2d(int iRef, std::string& newSubelement) { // Base::Console().Message("DVD::replaceReferenceSubElement2d(%d, %s)\n", iRef, newSubelement.c_str()); ReferenceVector refs = getReferences2d(); refs.at(iRef).setSubName(newSubelement); setReferences2d(refs); } void DrawViewDimension::replaceReferenceSubElement3d(int iRef, std::string& newSubelement) { ReferenceVector refs = getReferences3d(); refs.at(iRef).setSubName(newSubelement); setReferences3d(refs); } //what configuration of references do we have - Vertex-Vertex, Edge-Vertex, Edge, ... int DrawViewDimension::getRefType() const { if (isExtentDim()) { return RefType::extent; } ReferenceVector refs = getEffectiveReferences(); std::vector subNames; //std::vector subNames = getEffectiveSubNames(); //??? for (auto& ref : refs) { if (ref.getSubName().empty()) { //skip this one continue; } subNames.push_back(ref.getSubName()); } if (subNames.empty()) { //something went wrong, there were no subNames. Base::Console().Message("DVD::getRefType - %s - there are no subNames.\n", getNameInDocument()); return 0; } return getRefTypeSubElements(subNames); } //TODO: Gui/DimensionValidators.cpp has almost the same code //decide what the reference configuration is by examining the names of the sub elements int DrawViewDimension::getRefTypeSubElements(const std::vector& subElements) { int refType = invalidRef; int refEdges = 0, refVertices = 0; for (const auto& se : subElements) { if (DrawUtil::getGeomTypeFromName(se) == "Vertex") { refVertices++; } if (DrawUtil::getGeomTypeFromName(se) == "Edge") { refEdges++; } } if (refEdges == 0 && refVertices == 2) { refType = twoVertex; } if (refEdges == 0 && refVertices == 3) { refType = threeVertex; } if (refEdges == 1 && refVertices == 0) { refType = oneEdge; } if (refEdges == 1 && refVertices == 1) { refType = vertexEdge; } if (refEdges == 2 && refVertices == 0) { refType = twoEdge; } return refType; } //! validate 2D references - only checks if the target exists bool DrawViewDimension::checkReferences2D() const { // Base::Console().Message("DVD::checkReferences2d() - %s\n", getNameInDocument()); const std::vector& objects = References2D.getValues(); if (objects.empty()) { return false; } const std::vector& subElements = References2D.getSubValues(); if (subElements.empty()) { //must have at least 1 null string entry to balance DVP return false; } if (subElements.front().empty() && !References3D.getValues().empty()) { //this is (probably) a dim with 3d refs return true; } for (auto& s : subElements) { if (s.empty()) { return false; } int idx = DrawUtil::getIndexFromName(s); if (DrawUtil::getGeomTypeFromName(s) == "Edge") { TechDraw::BaseGeomPtr geom = getViewPart()->getGeomByIndex(idx); if (!geom) { return false; } } else if (DrawUtil::getGeomTypeFromName(s) == "Vertex") { TechDraw::VertexPtr v = getViewPart()->getProjVertexByIndex(idx); if (!v) { return false; } } } return true; } void DrawViewDimension::updateSavedGeometry() { // Base::Console().Message("DVD::updateSavedGeometry() - %s - savedGeometry: %d\n", // getNameInDocument(), SavedGeometry.getValues().size()); ReferenceVector references = getEffectiveReferences(); if (references.empty()) { // no references to save return; } std::vector newGeometry; const std::vector oldGeometry = SavedGeometry.getValues(); //need to clean up old geometry objects here? size_t iOldGeom(0); for (auto& entry : references) { if (entry.getSubName().empty()) { // view only reference has no geometry. continue; } if (entry.isValid()) { newGeometry.push_back(entry.asTopoShape()); } else { // use old geometry entry? null shape? have to put something in the vector // so SavedGeometry and references stay in sync. if (iOldGeom < oldGeometry.size()) { newGeometry.push_back(oldGeometry.at(iOldGeom)); } else { newGeometry.push_back(Part::TopoShape()); } } iOldGeom++; } if (!newGeometry.empty()) { SavedGeometry.setValues(newGeometry); } } // routines related to detecting that references no longer point to the same geometry as // when they were created. // returns true if the saved geometry is the same as the current reference geometry // returns false if the saved geometry is different from the the current reference geometry bool DrawViewDimension::compareSavedGeometry() { // Base::Console().Message("DVD::compareSavedGeometry() - isRestoring: %d\n", isRestoring()); const std::vector savedGeometry = SavedGeometry.getValues(); if (savedGeometry.empty()) { // no saved geometry, so we have nothing to compare, so we don't know if there has been a change // this should return false, since something != nothing // Base::Console().Warning("%s has no saved reference geometry!\n", getNameInDocument()); return false; } ReferenceVector references = getEffectiveReferences(); std::vector referenceGeometry; for (auto& entry : references) { referenceGeometry.push_back(entry.asTopoShape()); } if (savedGeometry.size() != referenceGeometry.size()) { // Base::Console().Message("DVD::compareSavedGeometry - geometry sizes have changed\n"); return false; } int geometryCount = savedGeometry.size(); int iGeom = 0; for ( ; iGeom < geometryCount; iGeom++) { if (savedGeometry.at(iGeom).getTypeId() != referenceGeometry.at(iGeom).getTypeId()) { // Base::Console().Message("DVD::compareSavedGeometry - saved geometry (%d) has different type\n", iGeom); return false; } } //saved and reference geometry have same count and types for (iGeom = 0; iGeom < geometryCount; iGeom++) { Part::TopoShape temp = savedGeometry.at(iGeom); if (!m_matcher->compareGeometry(temp, referenceGeometry.at(iGeom)) ) { // Base::Console().Message("DVD::compareSavedGeometry - saved geometry (%d) does not match current geometry\n", iGeom); return false; } } //free the reference geometry? return true; } // deal with the situation where references do not point to the same geometry as // when they were created. bool DrawViewDimension::fixExactMatch() { // Base::Console().Message("DVD::fixExactMatch() - reference geometry has changed\n"); if (!Preferences::autoCorrectDimRefs()) { return false; } ReferenceVector references = getEffectiveReferences(); if (references.empty()) { // could not get refs, something is wrong! return false; } if (SavedGeometry.getValues().empty()) { // there is no saved geometry, so we can't repair anything. return false; } std::vector< std::pair > refsToFix2d; std::vector< std::pair > refsToFix3d; bool success(true); size_t referenceCount = references.size(); size_t iRef = 0; for ( ; iRef < referenceCount; iRef++) { std::string newReference(""); TopoDS_Shape geomShape = references.at(iRef).getGeometry(); if (geomShape.IsNull()) { // Base::Console().Message("DVD::fixExactMatch - no geometry found for reference: %d\n", iRef); return false; } if (references.at(iRef).is3d()) { if (geomShape.ShapeType() == TopAbs_VERTEX) { newReference = recoverChangedVertex3d(iRef); } else { newReference = recoverChangedEdge3d(iRef); } if (!newReference.empty()) { std::pair toFix(iRef, newReference); refsToFix3d.push_back(toFix); } else { Base::Console().Warning("%s - no exact match for changed 3d reference: %d\n", getNameInDocument(), iRef); success = false; } } else { if (geomShape.ShapeType() == TopAbs_VERTEX) { newReference = recoverChangedVertex2d(iRef); } else { newReference = recoverChangedEdge2d(iRef); } if (!newReference.empty()) { std::pair toFix(iRef, newReference); refsToFix2d.push_back(toFix); } else { Base::Console().Warning("%s - no exact match for changed 2d reference: %d\n", getNameInDocument(), iRef); success = false; } } } for (auto& fix : refsToFix2d) { replaceReferenceSubElement2d(fix.first, fix.second); } for (auto& fix : refsToFix3d) { replaceReferenceSubElement3d(fix.first, fix.second); } return success; } // deal with situation where the current geometry does not match the saved geometry, // but we did not find an exact match in the geometry pile void DrawViewDimension::handleNoExactMatch() { // Base::Console().Message("DVD::handleNoExactMatch()\n"); // Base::Console().Message("%s - trying to match changed geometry - stage 2\n", getNameInDocument()); // this is where we insert the clever logic to determine that the changed geometry // actually still represents the "front top left" edge. // after figuring out the new reference, save the geometry // updateSavedGeometry(); m_referencesCorrect = true; } //find an edge in the view that matches the reference entry's type and characteristics std::string DrawViewDimension::recoverChangedEdge2d(int iReference) { // Base::Console().Message("DVD::recoverChangedEdge2d(ref: %d)\n", iReference); double scale = getViewPart()->getScale(); Part::TopoShape savedGeometryItem = SavedGeometry.getValues().at(iReference); std::vector gEdges = getViewPart()->getEdgeGeometry(); int iEdge = 0; for (auto& edge : gEdges) { Part::TopoShape temp = edge->asTopoShape(scale); if (savedGeometryItem.getTypeId() != temp.getTypeId()) { // if the typeIds don't match, we can not compare the geometry // Base::Console().Message("DVD::recoverChangedEdge2d - types do not match\n"); iEdge++; continue; } bool isSame = m_matcher->compareGeometry(savedGeometryItem, temp); // Base::Console().Message("DVD::recoverChangedEdge2d - iEdge: %d isSame: %d\n", iEdge, isSame); if (isSame) { return std::string("Edge") + std::to_string(iEdge); } iEdge++; } return std::string(""); } std::string DrawViewDimension::recoverChangedVertex2d(int iReference) { // Base::Console().Message("DVD::recoverChangedVertex2d(%d)\n", iReference); double scale = getViewPart()->getScale(); std::vector savedAll = SavedGeometry.getValues(); if (savedAll.empty() || iReference >= savedAll.size()) { return std::string(); } Part::TopoShape savedGeometryItem = SavedGeometry.getValues().at(iReference); std::vector gVertexAll = getViewPart()->getVertexGeometry(); int iVertex = 0; for (auto& vert : gVertexAll) { Part::TopoShape temp = vert->asTopoShape(scale); bool isSame = m_matcher->compareGeometry(savedGeometryItem, temp); if (isSame) { return std::string("Vertex") + std::to_string(iVertex); } iVertex++; } return std::string(""); } std::string DrawViewDimension::recoverChangedEdge3d(int iReference) { // Base::Console().Message("DVD::recoverChangedEdge3d(%d)\n", iReference); Part::TopoShape savedGeometryItem = SavedGeometry.getValues().at(iReference); ReferenceVector references = getEffectiveReferences(); App::DocumentObject* searchObject = references.at(iReference).getObject(); Part::TopoShape shape = Part::Feature::getTopoShape(searchObject); App::GeoFeature* geoFeat = dynamic_cast(searchObject); //does a feature in a body get the body's globalPlacement?? if (geoFeat) { shape.setPlacement(geoFeat->globalPlacement()); } //TODO: these TopoShapes will have to be released when we are finished with them std::vector edgesAll = getEdges(shape); int iEdge = 1; //note that edge numbering starts at 1! for (auto& edge : edgesAll) { bool isSame = m_matcher->compareGeometry(savedGeometryItem, edge); if (isSame) { return std::string("Edge") + std::to_string(iEdge); } iEdge++; } return std::string(""); } // based on Part::TopoShapePyImp::getShapes. Produces a vector of unique edges within the shape std::vector DrawViewDimension::getEdges(const TopoShape& inShape) { std::vector ret; TopTools_IndexedMapOfShape M; TopExp_Explorer Ex(inShape.getShape(), TopAbs_EDGE); while (Ex.More()) { M.Add(Ex.Current()); Ex.Next(); } for (Standard_Integer k = 1; k <= M.Extent(); k++) { const TopoDS_Shape& shape = M(k); ret.push_back(TopoShape(shape)); } return ret; } // as recoverChangedVertex2d, but 3d references do not need to be unscaled std::string DrawViewDimension::recoverChangedVertex3d(int iReference) { // Base::Console().Message("DVD::recoverChangedVertex3d(%d)\n", iReference); Part::TopoShape savedGeometryItem = SavedGeometry.getValues().at(iReference); ReferenceVector references = getEffectiveReferences(); App::DocumentObject* searchObject = references.at(iReference).getObject(); Part::TopoShape shape = Part::Feature::getTopoShape(searchObject); App::GeoFeature* geoFeat = dynamic_cast(searchObject); if (geoFeat) { shape.setPlacement(geoFeat->globalPlacement()); } //TODO: these TopoShapes will have to be released when we are finished with them std::vector vertsAll = getVertexes(shape); int iVert = 1; //note that vertex numbering starts at 1! for (auto& vert : vertsAll) { bool isSame = m_matcher->compareGeometry(savedGeometryItem, vert); if (isSame) { return std::string("Vertex") + std::to_string(iVert); } iVert++; } return std::string(""); } // based on Part::TopoShapePyImp::getShapes std::vector DrawViewDimension::getVertexes(const TopoShape& inShape) { std::vector ret; TopTools_IndexedMapOfShape M; TopExp_Explorer Ex(inShape.getShape(), TopAbs_VERTEX); while (Ex.More()) { M.Add(Ex.Current()); Ex.Next(); } for (Standard_Integer k = 1; k <= M.Extent(); k++) { const TopoDS_Shape& shape = M(k); ret.push_back(TopoShape(shape)); } return ret; } pointPair DrawViewDimension::closestPoints(TopoDS_Shape s1, TopoDS_Shape s2) const { pointPair result; BRepExtrema_DistShapeShape extss(s1, s2); if (!extss.IsDone()) { throw Base::RuntimeError("DVD::closestPoints - BRepExtrema_DistShapeShape failed"); } int count = extss.NbSolution(); if (count != 0) { gp_Pnt p = extss.PointOnShape1(1); result.first(Base::Vector3d(p.X(), p.Y(), p.Z())); p = extss.PointOnShape2(1); result.second(Base::Vector3d(p.X(), p.Y(), p.Z())); }//TODO: else { explode } return result; } //set the reference property from a reference vector void DrawViewDimension::setReferences2d(ReferenceVector refs) { // Base::Console().Message("DVD::setReferences2d(%d)\n", refs.size()); std::vector objects; std::vector subNames; if (objects.size() != subNames.size()) { throw Base::IndexError("DVD::setReferences2d - objects and subNames do not match."); } for (size_t iRef = 0; iRef < refs.size(); iRef++) { objects.push_back(refs.at(iRef).getObject()); subNames.push_back(refs.at(iRef).getSubName()); } References2D.setValues(objects, subNames); } //set the reference property from a reference vector void DrawViewDimension::setReferences3d(ReferenceVector refs) { if (refs.empty() && !References3D.getValues().empty()) { //clear the property of any old links References3D.setValue(nullptr, nullptr); return; } std::vector objects; std::vector subNames; if (objects.size() != subNames.size()) { throw Base::IndexError("DVD::setReferences3d - objects and subNames do not match."); } for (size_t iRef = 0; iRef < refs.size(); iRef++) { objects.push_back(refs.at(iRef).getObject()); subNames.push_back(refs.at(iRef).getSubName()); } References3D.setValues(objects, subNames); } //!add Dimension 3D references to measurement void DrawViewDimension::setAll3DMeasurement() { // Base::Console().Message("DVD::setAll3dMeasurement()\n"); measurement->clear(); const std::vector& Objs = References3D.getValues(); const std::vector& Subs = References3D.getSubValues(); int end = Objs.size(); int i = 0; for (; i < end; i++) { static_cast(measurement->addReference3D(Objs.at(i), Subs.at(i))); } } //delete all previous measurements void DrawViewDimension::clear3DMeasurements() { //set sublinklist to empty? measurement->clear(); } void DrawViewDimension::dumpRefs2D(const char* text) const { Base::Console().Message("DUMP - %s\n", text); const std::vector& objects = References2D.getValues(); const std::vector& subElements = References2D.getSubValues(); std::vector::const_iterator objIt = objects.begin(); std::vector::const_iterator subIt = subElements.begin(); int i = 0; for (; objIt != objects.end(); objIt++, subIt++, i++) { Base::Console().Message("DUMP - ref: %d object: %s subElement: %s\n", i, (*objIt)->getNameInDocument(), (*subIt).c_str()); } } double DrawViewDimension::dist2Segs(Base::Vector3d s1, Base::Vector3d e1, Base::Vector3d s2, Base::Vector3d e2) const { gp_Pnt start(s1.x, s1.y, 0.0); gp_Pnt end(e1.x, e1.y, 0.0); TopoDS_Vertex v1 = BRepBuilderAPI_MakeVertex(start); TopoDS_Vertex v2 = BRepBuilderAPI_MakeVertex(end); BRepBuilderAPI_MakeEdge makeEdge1(v1, v2); TopoDS_Edge edge1 = makeEdge1.Edge(); start = gp_Pnt(s2.x, s2.y, 0.0); end = gp_Pnt(e2.x, e2.y, 0.0); v1 = BRepBuilderAPI_MakeVertex(start); v2 = BRepBuilderAPI_MakeVertex(end); BRepBuilderAPI_MakeEdge makeEdge2(v1, v2); TopoDS_Edge edge2 = makeEdge2.Edge(); BRepExtrema_DistShapeShape extss(edge1, edge2); if (!extss.IsDone()) { throw Base::RuntimeError("DVD::dist2Segs - BRepExtrema_DistShapeShape failed"); } int count = extss.NbSolution(); double minDist = 0.0; if (count != 0) { minDist = extss.Value(); }//TODO: else { explode } return minDist; } bool DrawViewDimension::leaderIntersectsArc(Base::Vector3d s, Base::Vector3d pointOnCircle) { bool result = false; const std::vector& subElements = References2D.getSubValues(); int idx = DrawUtil::getIndexFromName(subElements[0]); TechDraw::BaseGeomPtr base = getViewPart()->getGeomByIndex(idx); if (base && base->getGeomType() == TechDraw::GeomType::ARCOFCIRCLE) { TechDraw::AOCPtr aoc = std::static_pointer_cast(base); if (aoc->intersectsArc(s, pointOnCircle)) { result = true; } } else if (base && base->getGeomType() == TechDraw::GeomType::BSPLINE) { TechDraw::BSplinePtr spline = std::static_pointer_cast(base); if (spline->isCircle()) { if (spline->intersectsArc(s, pointOnCircle)) { result = true; } } } return result; } void DrawViewDimension::saveArrowPositions(const Base::Vector2d positions[]) { if (!positions) { m_arrowPositions.first(Base::Vector3d(0.0, 0.0, 0.0)); m_arrowPositions.second(Base::Vector3d(0.0, 0.0, 0.0)); } else { double scale = getViewPart()->getScale(); m_arrowPositions.first(Base::Vector3d(positions[0].x, positions[0].y, 0.0) / scale); m_arrowPositions.second(Base::Vector3d(positions[1].x, positions[1].y, 0.0) / scale); } } //return position within parent view of dimension arrow heads/dimline endpoints //note positions are in apparent coord (inverted y). pointPair DrawViewDimension::getArrowPositions() { return m_arrowPositions; } bool DrawViewDimension::has2DReferences() const { // Base::Console().Message("DVD::has2DReferences() - %s\n",getNameInDocument()); const std::vector& objects = References2D.getValues(); const std::vector& subNames = References2D.getSubValues(); if (objects.empty()) { //we don't even have a DVP return false; } if (subNames.front().empty()) { //this is ok, as we must have a null string entry to balance DVP in first object position return true; } //we have a reference to a DVP and at least 1 subName entry, so we have 2d references return true; } //there is no special structure to 3d references, so anything > 0 is good bool DrawViewDimension::has3DReferences() const { return (References3D.getSize() > 0); } //has arbitrary or nonzero tolerance bool DrawViewDimension::hasOverUnderTolerance() const { if (ArbitraryTolerances.getValue() || !DrawUtil::fpCompare(OverTolerance.getValue(), 0.0) || !DrawUtil::fpCompare(UnderTolerance.getValue(), 0.0)) { return true; } return false; } bool DrawViewDimension::showUnits() const { return Preferences::getPreferenceGroup("Dimensions")->GetBool("ShowUnits", false); } bool DrawViewDimension::useDecimals() const { return Preferences::useGlobalDecimals(); } std::string DrawViewDimension::getPrefixForDimType() const { if (Type.isValue("Radius")) { return "R"; } else if (Type.isValue("Diameter")) { return std::string(Preferences::getPreferenceGroup("Dimensions")->GetASCII("DiameterSymbol", "\xe2\x8c\x80"));// Diameter symbol } return ""; } std::string DrawViewDimension::getDefaultFormatSpec(bool isToleranceFormat) const { return m_formatter->getDefaultFormatSpec(isToleranceFormat); } bool DrawViewDimension::isExtentDim() const { std::string name(getNameInDocument()); if (name.substr(0, 9) == "DimExtent") { return true; } return false; } PyObject* DrawViewDimension::getPyObject() { if (PythonObject.is(Py::_None())) { // ref counter is set to 1 PythonObject = Py::Object(new DrawViewDimensionPy(this), true); } return Py::new_reference_to(PythonObject); }