def drawTangentConnectionTemplate(corner1, corner2, connection, bm,
getRadiusesForIntersections,
getConnectionEndPoints):
c1 = Vector((corner1.x, corner1.y, defaultZ))
c2 = Vector((corner2.x, corner2.y, defaultZ))
r1, r2 = getRadiusesForIntersections(connection.radius, corner1.radius,
corner2.radius)
geomCalc = GeometryCalculator()
intersections = geomCalc.getCircleIntersections(c1, r1, c2, r2)
if intersections == None:
assignCornerEndPoint(corner1, None)
assignCornerStartPoint(corner2, None)
return
center = None
if len(intersections) == 1:
center = intersections[0]
elif len(intersections) == 2:
if not connection.flipCenter:
center = intersections[1]
else:
center = intersections[0]
connectionStartPoint, connectionEndPoint = getConnectionEndPoints(
geomCalc, center, c1, corner1.radius, c2, corner2.radius,
connection.radius)
assignCornerEndPoint(corner1, connectionStartPoint)
assignCornerStartPoint(corner2, connectionEndPoint)
angleDeg, angleRad = geomCalc.getPositiveAngleBetween3Points(
connectionStartPoint, center, connectionEndPoint)
if (angleDeg == None) or (angleRad == None):
return
if connection.flipAngle:
angleRad = -(2 * pi - angleRad)
spinAxis = Vector((0, 0, 1))
v0 = bm.verts.new(connectionEndPoint)
result = bmesh.ops.spin(bm, geom = [v0], cent = center, axis = spinAxis, \
angle = angleRad, steps = connection.sides, use_duplicate = False)
def drawTangentConnectionTemplate(corner1, corner2, connection, bm, getRadiusesForIntersections, getConnectionEndPoints):
c1 = Vector((corner1.x, corner1.y, defaultZ))
c2 = Vector((corner2.x, corner2.y, defaultZ))
r1, r2 = getRadiusesForIntersections(connection.radius, corner1.radius, corner2.radius)
geomCalc = GeometryCalculator()
intersections = geomCalc.getCircleIntersections(c1, r1, c2, r2)
if intersections == None:
assignCornerEndPoint(corner1, None)
assignCornerStartPoint(corner2, None)
return
center = None
if len(intersections) == 1:
center = intersections[0]
elif len(intersections) == 2:
if not connection.flipCenter:
center = intersections[1]
else:
center = intersections[0]
connectionStartPoint, connectionEndPoint = getConnectionEndPoints(geomCalc, center, c1, corner1.radius, c2, corner2.radius, connection.radius)
assignCornerEndPoint(corner1, connectionStartPoint)
assignCornerStartPoint(corner2, connectionEndPoint)
angleDeg, angleRad = geomCalc.getPositiveAngleBetween3Points(connectionStartPoint, center, connectionEndPoint)
if (angleDeg == None) or (angleRad == None):
return
if connection.flipAngle:
angleRad = -(2 * pi - angleRad)
spinAxis = Vector((0, 0, 1))
v0 = bm.verts.new(connectionEndPoint)
result = bmesh.ops.spin(bm, geom = [v0], cent = center, axis = spinAxis, \
angle = angleRad, steps = connection.sides, use_duplicate = False)
class TestGCCircleIntersections(unittest.TestCase):
def setUp(self):
self.geomCalc = GeometryCalculator()
def tearDown(self):
pass
# ## 2 intersections
def testTwoIntersectionsSmallOverlap(self):
c2 = Vector((0, 0, 0))
r2 = 1.0
c1 = Vector((4, 0, 0))
r1 = 3.5
intersection2 = Vector((0.5937, 0.805, 0.0))
intersection1 = Vector((0.5937, -0.805, 0.0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(2, len(output))
self.assertAlmostEqual(intersection1[0], output[0][0], places=3)
self.assertAlmostEqual(intersection1[1], output[0][1], places=3)
self.assertAlmostEqual(intersection1[2], output[0][2], places=3)
self.assertAlmostEqual(intersection2[0], output[1][0], places=3)
self.assertAlmostEqual(intersection2[1], output[1][1], places=3)
self.assertAlmostEqual(intersection2[2], output[1][2], places=3)
def testTwoIntersectionsSmallOverlapMinusDiagonal(self):
c2 = Vector((0, 0, 0))
r2 = 3.5
c1 = Vector((-4, -2, 0))
r1 = 1.0
intersection1 = Vector((-3.2177, -1.3770, 0.0))
intersection2 = Vector((-3.0322, -1.7479, 0.0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(2, len(output))
self.assertAlmostEqual(intersection1[0], output[0][0], places=3)
self.assertAlmostEqual(intersection1[1], output[0][1], places=3)
self.assertAlmostEqual(intersection1[2], output[0][2], places=3)
self.assertAlmostEqual(intersection2[0], output[1][0], places=3)
self.assertAlmostEqual(intersection2[1], output[1][1], places=3)
self.assertAlmostEqual(intersection2[2], output[1][2], places=3)
def testTwoIntersectionsLargeOverlapDiagonal(self):
c2 = Vector((0, 0, 0))
r2 = 3.5
c1 = Vector((2, 2, 0))
r1 = 1.5
intersection2 = Vector((1.2192, 3.2807, 0.0))
intersection1 = Vector((3.2807, 1.2192, 0.0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(2, len(output))
self.assertAlmostEqual(intersection1[0], output[0][0], places=3)
self.assertAlmostEqual(intersection1[1], output[0][1], places=3)
self.assertAlmostEqual(intersection1[2], output[0][2], places=3)
self.assertAlmostEqual(intersection2[0], output[1][0], places=3)
self.assertAlmostEqual(intersection2[1], output[1][1], places=3)
self.assertAlmostEqual(intersection2[2], output[1][2], places=3)
def testForTwoIntersectionsLargeOverlapPlusX(self):
c1 = Vector((0, 0, 0))
r1 = 6.0
c2 = Vector((4, 0, 0))
r2 = 2.5
intersection1 = Vector((5.71875, 1.8155, 0.0))
intersection2 = Vector((5.71875, -1.8155, 0.0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(2, len(output))
self.assertAlmostEqual(intersection1[0], output[0][0], places=3)
self.assertAlmostEqual(intersection1[1], output[0][1], places=3)
self.assertAlmostEqual(intersection1[2], output[0][2], places=3)
self.assertAlmostEqual(intersection2[0], output[1][0], places=3)
self.assertAlmostEqual(intersection2[1], output[1][1], places=3)
self.assertAlmostEqual(intersection2[2], output[1][2], places=3)
def testTwoIntersectionsLargeOverlapMinusX(self):
self.geomCalc.selectedPlane = self.geomCalc.YZ
c1 = Vector((0, 0, 0))
r1 = 6.0
# c2 = Vector((-4, 0, 0))
c2 = Vector((0, -4, 0))
r2 = 2.5
# intersection1 = Vector((-5.71875, -1.8155, 0.0))
# intersection2 = Vector((-5.71875, 1.8155, 0.0))
intersection1 = Vector((0.0, -5.71875, -1.8155))
intersection2 = Vector((0.0, -5.71875, 1.8155))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(2, len(output))
self.assertAlmostEqual(intersection1[0], output[0][0], places=3)
self.assertAlmostEqual(intersection1[1], output[0][1], places=3)
self.assertAlmostEqual(intersection1[2], output[0][2], places=3)
self.assertAlmostEqual(intersection2[0], output[1][0], places=3)
self.assertAlmostEqual(intersection2[1], output[1][1], places=3)
self.assertAlmostEqual(intersection2[2], output[1][2], places=3)
def testIntersectionsLargeOverlapMinusY(self):
c1 = Vector((0, 0, 0))
r1 = 6.0
c2 = Vector((0, -4, 0))
r2 = 2.5
intersection1 = Vector((1.8155, -5.71875, 0.0))
intersection2 = Vector((-1.8155, -5.71875, 0.0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(2, len(output))
self.assertAlmostEqual(intersection1[0], output[0][0], places=3)
self.assertAlmostEqual(intersection1[1], output[0][1], places=3)
self.assertAlmostEqual(intersection1[2], output[0][2], places=3)
self.assertAlmostEqual(intersection2[0], output[1][0], places=3)
self.assertAlmostEqual(intersection2[1], output[1][1], places=3)
self.assertAlmostEqual(intersection2[2], output[1][2], places=3)
### Tangency tests ###
def testTangentCirclesWithEqualRadiusReturnOnePoint2(self):
c1 = Vector((0, 0, 0))
r1 = 2.0
c2 = Vector((4, 0, 0))
r2 = 2.0
intersection = Vector((2, 0, 0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(1, len(output))
self.assertEqual(intersection, output[0])
def testTangentCirclesWithEqualRadiusReturnOnePoint3(self):
c1 = Vector((0, 0, 0))
r1 = 3.0
c2 = Vector((6, 0, 0))
r2 = 3.0
intersection = Vector((3, 0, 0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(1, len(output))
self.assertEqual(intersection, output[0])
def testTangentCirclesAlongXReturnOnePoint(self):
c1 = Vector((0, 0, 0))
r1 = 2.0
c2 = Vector((6, 0, 0))
r2 = 4.0
intersection = Vector((2, 0, 0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(1, len(output))
self.assertEqual(intersection, output[0])
def testInnerTangentCirclesAlongXReturnOnePoint(self):
c1 = Vector((3, 0, 0))
r1 = 1.0
c2 = Vector((6, 0, 0))
r2 = 4.0
intersection = Vector((2, 0, 0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(1, len(output))
self.assertEqual(intersection, output[0])
def testTangentCirclesAlongYReturnOnePoint(self):
c1 = Vector((0, 0, 0))
r1 = 2.0
c2 = Vector((0, 6, 0))
r2 = 4.0
intersection = Vector((0, 2, 0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(1, len(output))
self.assertEqual(intersection, output[0])
###################################################
# ## no intersection
def testNoIntersectionsWhenCenterDistanceLargerThenRadiusesSum(self):
c1 = Vector((0, 0, 0))
r1 = 2.0
c2 = Vector((10, 0, 0))
r2 = 4.0
intersection = None
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(intersection, output)
def testNoIntersectionsWhenCircleInsideTheOtherCircle(self):
c1 = Vector((0, 0, 0))
r1 = 20.0
c2 = Vector((10, 0, 0))
r2 = 4.0
intersection = None
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(intersection, output)
##############################
def testGetPerpendicularVectorXY(self):
self.geomCalc.selectedPlane = self.geomCalc.XY
v = Vector((1, 2, 0))
vp = Vector((-2, 1, 0))
calcVec = self.geomCalc.getPerpendicularVector(v)
self.assertEqual(calcVec, vp)
def testGetPerpendicularVectorXZ(self):
self.geomCalc.selectedPlane = self.geomCalc.XZ
v = Vector((1, 0, 5))
vp = Vector((-5, 0, 1))
calcVec = self.geomCalc.getPerpendicularVector(v)
self.assertEqual(calcVec, vp)
def testGetPerpendicularVectorYZ(self):
self.geomCalc.selectedPlane = self.geomCalc.YZ
v = Vector((0, 2, 3))
vp = Vector((0, -3, 2))
calcVec = self.geomCalc.getPerpendicularVector(v)
self.assertEqual(calcVec, vp)
class TestGCCircleIntersections(unittest.TestCase):
def setUp(self):
self.geomCalc = GeometryCalculator()
def tearDown(self):
pass
# ## 2 intersections
def testTwoIntersectionsSmallOverlap(self):
c2 = Vector((0, 0, 0))
r2 = 1.0
c1 = Vector((4, 0, 0))
r1 = 3.5
intersection2 = Vector((0.5937, 0.805, 0.0))
intersection1 = Vector((0.5937, -0.805, 0.0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(2, len(output))
self.assertAlmostEqual(intersection1[0], output[0][0], places = 3)
self.assertAlmostEqual(intersection1[1], output[0][1], places = 3)
self.assertAlmostEqual(intersection1[2], output[0][2], places = 3)
self.assertAlmostEqual(intersection2[0], output[1][0], places = 3)
self.assertAlmostEqual(intersection2[1], output[1][1], places = 3)
self.assertAlmostEqual(intersection2[2], output[1][2], places = 3)
def testTwoIntersectionsSmallOverlapMinusDiagonal(self):
c2 = Vector((0, 0, 0))
r2 = 3.5
c1 = Vector((-4, -2, 0))
r1 = 1.0
intersection1 = Vector((-3.2177, -1.3770, 0.0))
intersection2 = Vector((-3.0322, -1.7479, 0.0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(2, len(output))
self.assertAlmostEqual(intersection1[0], output[0][0], places = 3)
self.assertAlmostEqual(intersection1[1], output[0][1], places = 3)
self.assertAlmostEqual(intersection1[2], output[0][2], places = 3)
self.assertAlmostEqual(intersection2[0], output[1][0], places = 3)
self.assertAlmostEqual(intersection2[1], output[1][1], places = 3)
self.assertAlmostEqual(intersection2[2], output[1][2], places = 3)
def testTwoIntersectionsLargeOverlapDiagonal(self):
c2 = Vector((0, 0, 0))
r2 = 3.5
c1 = Vector((2, 2, 0))
r1 = 1.5
intersection2 = Vector((1.2192, 3.2807, 0.0))
intersection1 = Vector((3.2807, 1.2192, 0.0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(2, len(output))
self.assertAlmostEqual(intersection1[0], output[0][0], places = 3)
self.assertAlmostEqual(intersection1[1], output[0][1], places = 3)
self.assertAlmostEqual(intersection1[2], output[0][2], places = 3)
self.assertAlmostEqual(intersection2[0], output[1][0], places = 3)
self.assertAlmostEqual(intersection2[1], output[1][1], places = 3)
self.assertAlmostEqual(intersection2[2], output[1][2], places = 3)
def testForTwoIntersectionsLargeOverlapPlusX(self):
c1 = Vector((0, 0, 0))
r1 = 6.0
c2 = Vector((4, 0, 0))
r2 = 2.5
intersection1 = Vector((5.71875, 1.8155, 0.0))
intersection2 = Vector((5.71875, -1.8155, 0.0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(2, len(output))
self.assertAlmostEqual(intersection1[0], output[0][0], places = 3)
self.assertAlmostEqual(intersection1[1], output[0][1], places = 3)
self.assertAlmostEqual(intersection1[2], output[0][2], places = 3)
self.assertAlmostEqual(intersection2[0], output[1][0], places = 3)
self.assertAlmostEqual(intersection2[1], output[1][1], places = 3)
self.assertAlmostEqual(intersection2[2], output[1][2], places = 3)
def testTwoIntersectionsLargeOverlapMinusX(self):
self.geomCalc.selectedPlane = self.geomCalc.YZ
c1 = Vector((0, 0, 0))
r1 = 6.0
# c2 = Vector((-4, 0, 0))
c2 = Vector((0, -4, 0))
r2 = 2.5
# intersection1 = Vector((-5.71875, -1.8155, 0.0))
# intersection2 = Vector((-5.71875, 1.8155, 0.0))
intersection1 = Vector((0.0, -5.71875, -1.8155))
intersection2 = Vector((0.0, -5.71875, 1.8155))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(2, len(output))
self.assertAlmostEqual(intersection1[0], output[0][0], places = 3)
self.assertAlmostEqual(intersection1[1], output[0][1], places = 3)
self.assertAlmostEqual(intersection1[2], output[0][2], places = 3)
self.assertAlmostEqual(intersection2[0], output[1][0], places = 3)
self.assertAlmostEqual(intersection2[1], output[1][1], places = 3)
self.assertAlmostEqual(intersection2[2], output[1][2], places = 3)
def testIntersectionsLargeOverlapMinusY(self):
c1 = Vector((0, 0, 0))
r1 = 6.0
c2 = Vector((0, -4, 0))
r2 = 2.5
intersection1 = Vector((1.8155, -5.71875, 0.0))
intersection2 = Vector((-1.8155, -5.71875, 0.0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(2, len(output))
self.assertAlmostEqual(intersection1[0], output[0][0], places = 3)
self.assertAlmostEqual(intersection1[1], output[0][1], places = 3)
self.assertAlmostEqual(intersection1[2], output[0][2], places = 3)
self.assertAlmostEqual(intersection2[0], output[1][0], places = 3)
self.assertAlmostEqual(intersection2[1], output[1][1], places = 3)
self.assertAlmostEqual(intersection2[2], output[1][2], places = 3)
### Tangency tests ###
def testTangentCirclesWithEqualRadiusReturnOnePoint2(self):
c1 = Vector((0, 0, 0))
r1 = 2.0
c2 = Vector((4, 0, 0))
r2 = 2.0
intersection = Vector((2, 0, 0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(1, len(output))
self.assertEqual(intersection, output[0])
def testTangentCirclesWithEqualRadiusReturnOnePoint3(self):
c1 = Vector((0, 0, 0))
r1 = 3.0
c2 = Vector((6, 0, 0))
r2 = 3.0
intersection = Vector((3, 0, 0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(1, len(output))
self.assertEqual(intersection, output[0])
def testTangentCirclesAlongXReturnOnePoint(self):
c1 = Vector((0, 0, 0))
r1 = 2.0
c2 = Vector((6, 0, 0))
r2 = 4.0
intersection = Vector((2, 0, 0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(1, len(output))
self.assertEqual(intersection, output[0])
def testInnerTangentCirclesAlongXReturnOnePoint(self):
c1 = Vector((3, 0, 0))
r1 = 1.0
c2 = Vector((6, 0, 0))
r2 = 4.0
intersection = Vector((2, 0, 0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(1, len(output))
self.assertEqual(intersection, output[0])
def testTangentCirclesAlongYReturnOnePoint(self):
c1 = Vector((0, 0, 0))
r1 = 2.0
c2 = Vector((0, 6, 0))
r2 = 4.0
intersection = Vector((0, 2, 0))
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(1, len(output))
self.assertEqual(intersection, output[0])
###################################################
# ## no intersection
def testNoIntersectionsWhenCenterDistanceLargerThenRadiusesSum(self):
c1 = Vector((0, 0, 0))
r1 = 2.0
c2 = Vector((10, 0, 0))
r2 = 4.0
intersection = None
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(intersection, output)
def testNoIntersectionsWhenCircleInsideTheOtherCircle(self):
c1 = Vector((0, 0, 0))
r1 = 20.0
c2 = Vector((10, 0, 0))
r2 = 4.0
intersection = None
output = self.geomCalc.getCircleIntersections(c1, r1, c2, r2)
self.assertEqual(intersection, output)
##############################
def testGetPerpendicularVectorXY(self):
self.geomCalc.selectedPlane = self.geomCalc.XY
v = Vector((1, 2, 0))
vp = Vector((-2, 1, 0))
calcVec = self.geomCalc.getPerpendicularVector(v)
self.assertEqual(calcVec, vp)
def testGetPerpendicularVectorXZ(self):
self.geomCalc.selectedPlane = self.geomCalc.XZ
v = Vector((1, 0, 5))
vp = Vector((-5, 0, 1))
calcVec = self.geomCalc.getPerpendicularVector(v)
self.assertEqual(calcVec, vp)
def testGetPerpendicularVectorYZ(self):
self.geomCalc.selectedPlane = self.geomCalc.YZ
v = Vector((0, 2, 3))
vp = Vector((0, -3, 2))
calcVec = self.geomCalc.getPerpendicularVector(v)
self.assertEqual(calcVec, vp)
