def print_axis_and_parabola(original,
threshold,
stepRho=1,
stepTheta=np.pi / 180):
"""
Prints axis and parabola in the original image
"""
# original image
image = cv2.cvtColor(original.copy(), cv2.COLOR_GRAY2RGB)
# axis
axis = Axis(original, threshold)
original_axis = axis.get_axis()
rotate_angle, axis_90degree = axis.get_axis_in_90degree()
axis.draw_axis(image)
axis.draw_axis_origin(image)
# parabola
parabola = Parabola(original, rotate_angle, original_axis,
axis.axis_origin)
a, b, c = parabola.get_parabola()
parabola.draw_parabola(image)
return image
