src/SketchPlugin/Test/TestSketchArcCircle.py

   1 """
   2     TestSketchArcCircle.py
   3
   4     static const std::string MY_SKETCH_ARC_ID("SketchArc");
   5     static const std::string MY_CENTER_ID = "ArcCenter";
   6     static const std::string MY_START_ID = "ArcStartPoint";
   7     static const std::string MY_END_ID = "ArcEndPoint";
   8     data()->addAttribute(SketchPlugin_Arc::CENTER_ID(), GeomDataAPI_Point2D::typeId());
   9     data()->addAttribute(SketchPlugin_Arc::START_ID(),  GeomDataAPI_Point2D::typeId());
  10     data()->addAttribute(SketchPlugin_Arc::END_ID(),    GeomDataAPI_Point2D::typeId());
  11
  12     static const std::string MY_CIRCLE_ID("SketchCircle");
  13     static const std::string MY_CIRCLE_CENTER_ID("CircleCenter");
  14     static const std::string MY_CIRCLE_RADIUS_ID("CircleRadius");
  15     data()->addAttribute(SketchPlugin_Circle::CENTER_ID(), GeomDataAPI_Point2D::typeId());
  16     data()->addAttribute(SketchPlugin_Circle::RADIUS_ID(), ModelAPI_AttributeDouble::typeId());
  17 """
  18
  19 #=========================================================================
  20 # Initialization of the test
  21 #=========================================================================
  22 from GeomDataAPI import *
  23 from ModelAPI import *
  24 import math
  25
  26 __updated__ = "2014-10-28"
  27
  28
  29 #=========================================================================
  30 # Auxiliary functions
  31 #=========================================================================
  32
  33 def angle(theCenter, theFirst, theLast):
  34     """
  35     subroutine to calculate angle given by 3 points
  36     """
  37     aDirX1 = theFirst.x() - theCenter.x()
  38     aDirY1 = theFirst.y() - theCenter.y()
  39     aLen1 = math.hypot(aDirX1, aDirY1)
  40     aDirX2 = theLast.x() - theCenter.x()
  41     aDirY2 = theLast.y() - theCenter.y()
  42     aLen2 = math.hypot(aDirX2, aDirY2)
  43     aDot = aDirX1 * aDirX2 + aDirY1 * aDirY2
  44     anAngle = math.acos(aDot / aLen1 / aLen2)
  45     return round(anAngle * 180. / math.pi, 6)
  46
  47 def distancePointPoint(thePointA, thePointB):
  48     """
  49     subroutine to calculate distance between two points
  50     result of calculated distance is has 10**-5 precision
  51     """
  52     xdiff = math.pow((thePointA.x() - thePointB.x()), 2)
  53     ydiff = math.pow((thePointA.y() - thePointB.y()), 2)
  54     return round(math.sqrt(xdiff + ydiff), 5)
  55
  56 def dot(thePoint11, thePoint12, thePoint21, thePoint22):
  57     """
  58     subroutine to calculate dit product between lines given by their points
  59     """
  60     aDirX1 = thePoint12.x() - thePoint11.x()
  61     aDirY1 = thePoint12.y() - thePoint11.y()
  62     aLen1 = math.hypot(aDirX1, aDirY1)
  63     aDirX2 = thePoint22.x() - thePoint21.x()
  64     aDirY2 = thePoint22.y() - thePoint21.y()
  65     aLen2 = math.hypot(aDirX2, aDirY2)
  66     aDot = aDirX1 * aDirX2 + aDirY1 * aDirY2
  67     return aDot / aLen1 / aLen2
  68
  69
  70 aSession = ModelAPI_Session.get()
  71 aDocument = aSession.moduleDocument()
  72 #=========================================================================
  73 # Creation of a sketch
  74 #=========================================================================
  75 aSession.startOperation()
  76 #aSketchFeature = aDocument.addFeature("Sketch")
  77 aSketchCommonFeature = aDocument.addFeature("Sketch")
  78 aSketchFeature = featureToCompositeFeature(aSketchCommonFeature)
  79 origin = geomDataAPI_Point(aSketchFeature.attribute("Origin"))
  80 origin.setValue(0, 0, 0)
  81 dirx = geomDataAPI_Dir(aSketchFeature.attribute("DirX"))
  82 dirx.setValue(1, 0, 0)
  83 norm = geomDataAPI_Dir(aSketchFeature.attribute("Norm"))
  84 norm.setValue(0, 0, 1)
  85 aSession.finishOperation()
  86 #=========================================================================
  87 # Creation of an arc
  88 # 1. Test SketchPlugin_Arc attributes
  89 # 2.
  90 #=========================================================================
  91 aSession.startOperation()
  92 aSketchReflist = aSketchFeature.reflist("Features")
  93 assert (not aSketchReflist.isInitialized())
  94 assert (aSketchReflist.size() == 0)
  95 assert (len(aSketchReflist.list()) == 0)
  96 aSketchArc = aSketchFeature.addFeature("SketchArc")
  97 assert (aSketchArc.getKind() == "SketchArc")
  98 anArcCentr = geomDataAPI_Point2D(aSketchArc.attribute("ArcCenter"))
  99 assert (anArcCentr.x() == 0)
 100 assert (anArcCentr.y() == 0)
 101 assert (not anArcCentr.isInitialized())
 102 anArcCentr.setValue(10., 10.)
 103 anArcStartPoint = geomDataAPI_Point2D(
 104     aSketchArc.attribute("ArcStartPoint"))
 105 assert (anArcStartPoint.x() == 0)
 106 assert (anArcStartPoint.y() == 0)
 107 assert (not anArcStartPoint.isInitialized())
 108 anArcStartPoint.setValue(0., 50.)
 109 anArcEndPoint = geomDataAPI_Point2D(aSketchArc.attribute("ArcEndPoint"))
 110 assert (anArcEndPoint.x() == 0)
 111 assert (anArcEndPoint.y() == 0)
 112 assert (not anArcEndPoint.isInitialized())
 113 anArcEndPoint.setValue(50., 0.)
 114 aSession.finishOperation()
 115 # check that values have been changed
 116 aSketchReflist = aSketchFeature.reflist("Features")
 117 assert (aSketchReflist.size() == 1)
 118 assert (len(aSketchReflist.list()) == 1)
 119 assert (anArcCentr.x() == 10.0)
 120 assert (anArcCentr.y() == 10.0)
 121 assert (anArcStartPoint.x() == 0.0)
 122 assert (anArcStartPoint.y() == 50.0)
 123 assert (anArcEndPoint.x() == 50.0)
 124 assert (anArcEndPoint.y() == 0.0)
 125 #=========================================================================
 126 # Edit the arc:
 127 # 1. Move whole arc
 128 # 2. Change the start point
 129 # 3. Change the radius of arc
 130 # 4. Change arc's angle
 131 #=========================================================================
 132 aSession.startOperation()
 133 deltaX, deltaY = 5., 10.
 134 anArcCentr.setValue(anArcCentr.x() + deltaX, anArcCentr.y() + deltaY)
 135 aSession.finishOperation()
 136 assert (anArcCentr.x() == 15)
 137 assert (anArcCentr.y() == 20)
 138 assert (math.fabs(distancePointPoint(anArcCentr, anArcStartPoint) - distancePointPoint(anArcCentr, anArcEndPoint)) < 1.e-10)
 139 # Change the start point
 140 aSession.startOperation()
 141 anArcStartPoint.setValue(anArcStartPoint.x() + deltaX, anArcStartPoint.y())
 142 aSession.finishOperation()
 143 assert (math.fabs(distancePointPoint(anArcCentr, anArcStartPoint) - distancePointPoint(anArcCentr, anArcEndPoint)) < 1.e-10)
 144 # Change radius
 145 RADIUS = 50
 146 aSession.startOperation()
 147 anArcRadius = aSketchArc.real("ArcRadius")
 148 anArcRadius.setValue(RADIUS)
 149 aSession.finishOperation()
 150 assert (math.fabs(distancePointPoint(anArcCentr, anArcStartPoint) - RADIUS) < 1.e-10)
 151 assert (math.fabs(distancePointPoint(anArcCentr, anArcEndPoint) - RADIUS) < 1.e-10)
 152 # Change angle
 153 ANGLE = 120
 154 aSession.startOperation()
 155 anArcAngle = aSketchArc.real("ArcAngle")
 156 anArcAngle.setValue(ANGLE)
 157 aSession.finishOperation()
 158 assert (math.fabs(angle(anArcCentr, anArcStartPoint, anArcEndPoint) - ANGLE) < 1.e-7)
 159 #=========================================================================
 160 # Check results of the Arc
 161 #=========================================================================
 162 aResult = aSketchArc.firstResult()
 163 aResultConstruction = modelAPI_ResultConstruction(aResult)
 164 aShape = aResultConstruction.shape()
 165 assert (aShape is not None)
 166 assert (not aShape.isNull())
 167 #=========================================================================
 168 # Create a circle
 169 # 1. Test SketchPlugin_Circle.h attributes
 170 # 2. ModelAPI_AttributeDouble attribute
 171 #=========================================================================
 172 aSession.startOperation()
 173 aSketchReflist = aSketchFeature.reflist("Features")
 174 # Arc is already added
 175 assert (aSketchReflist.size() == 1)
 176 assert (len(aSketchReflist.list()) == 1)
 177 aSketchCircle = aSketchFeature.addFeature("SketchCircle")
 178 assert (aSketchCircle.getKind() == "SketchCircle")
 179 anCircleCentr = geomDataAPI_Point2D(aSketchCircle.attribute("CircleCenter"))
 180 assert (anCircleCentr.x() == 0)
 181 assert (anCircleCentr.y() == 0)
 182 assert (not anCircleCentr.isInitialized())
 183 aCircleRadius = aSketchCircle.real("CircleRadius")
 184 assert (type(aCircleRadius) == ModelAPI_AttributeDouble)
 185 # ModelAPI_AttributeDouble.typeId() is checked in ModelAPI_TestConstants
 186 assert (aCircleRadius.attributeType() == ModelAPI_AttributeDouble.typeId())
 187 assert (aCircleRadius.value() == 0)
 188 anCircleCentr.setValue(-25., -25)
 189 aCircleRadius.setValue(25.)
 190 assert (anCircleCentr.x() == -25)
 191 assert (anCircleCentr.y() == -25)
 192 assert (aCircleRadius.value() == 25)
 193 aSession.finishOperation()
 194 #=========================================================================
 195 # Edit the Cricle
 196 # 1. Check that changing the centr of a circle does not affects radius
 197 # 2. and vise versa; also check that int is acceptable as well as a real
 198 #=========================================================================
 199 aSession.startOperation()
 200 anCircleCentr.setValue(10, 60)
 201 aSession.finishOperation()
 202 assert (anCircleCentr.x() == 10)
 203 assert (anCircleCentr.y() == 60)
 204 assert (aCircleRadius.value() == 25)
 205 aSession.startOperation()
 206 aCircleRadius.setValue(int(20))
 207 aSession.finishOperation()
 208 assert (anCircleCentr.x() == 10)
 209 assert (anCircleCentr.y() == 60)
 210 assert (aCircleRadius.value() == 20)
 211 #=========================================================================
 212 # Create an arc, tangent to the line
 213 #=========================================================================
 214 aSession.startOperation()
 215 aSketchLine = aSketchFeature.addFeature("SketchLine")
 216 aLineStart = geomDataAPI_Point2D(aSketchLine.attribute("StartPoint"))
 217 aLineEnd = geomDataAPI_Point2D(aSketchLine.attribute("EndPoint"))
 218 aLineStart.setValue(0., 0.)
 219 aLineEnd.setValue(50., 0.)
 220 aSession.finishOperation()
 221 aSession.startOperation()
 222 aSketchArcTangent = aSketchFeature.addFeature("SketchArc")
 223 aSketchArcTangent.string("ArcType").setValue("Tangent")
 224 anArcEndPoint = geomDataAPI_Point2D(aSketchArcTangent.attribute("ArcEndPoint"))
 225 aTangent = aSketchArcTangent.refattr("ArcTangentPoint")
 226 aTangent.setAttr(aLineEnd)
 227 anArcEndPoint.setValue(51., 1.)
 228 aSession.finishOperation()
 229 aSession.startOperation()
 230 anArcEndPoint.setValue(100., 25.)
 231 aSession.finishOperation()
 232 anArcCenter = geomDataAPI_Point2D(aSketchArcTangent.attribute("ArcCenter"))
 233 aDot = dot(anArcCenter, aLineEnd, aLineStart, aLineEnd)
 234 assert math.fabs(aDot) <= 2.e-4, "Observed dot product: {0}".format(aDot)
 235 #=========================================================================
 236 # Create an arc, tangent to the previous arc
 237 #=========================================================================
 238 aSession.startOperation()
 239 aSketchArcTangent2 = aSketchFeature.addFeature("SketchArc")
 240 aSketchArcTangent2.string("ArcType").setValue("Tangent")
 241 anArcEndPoint2 = geomDataAPI_Point2D(aSketchArcTangent2.attribute("ArcEndPoint"))
 242 aTangent = aSketchArcTangent2.refattr("ArcTangentPoint")
 243 aTangent.setAttr(anArcEndPoint)
 244 anArcEndPoint2.setValue(anArcEndPoint.x() + 1, anArcEndPoint.y() + 1)
 245 aSession.finishOperation()
 246 aSession.startOperation()
 247 anArcEndPoint2.setValue(50., 50.)
 248 aSession.finishOperation()
 249 #=========================================================================
 250 # End of test
 251 #=========================================================================