Search code examples
pythongeometrypython-imaging-librarydrawingline

Drawing equidistant segments in python with opencv


Given an image, I draw a line inside it in the following way:

def slope(x1,y1,x2,y2):

    ###finding slope
    if x2!=x1:
        return((y2-y1)/(x2-x1))
    else:
        return 'NA'

def draw_line(image,lineThickness, colors, points):
    x1,y1, x2, y2=points
    m=slope(x1,y1,x2,y2)
    h,w=image.shape[:2]
    if m!='NA':
        px=0
        py=-(x1-0)*m+y1
        ##ending point
        qx=w
        qy=-(x2-w)*m+y2
    else:
    ### if slope is zero, draw a line with x=x1 and y=0 and y=height
        px,py=x1,0
        qx,qy=x1,h

    cv2.line(image, (int(px), int(py)), (int(qx), int(qy)), colors, lineThickness)
    return (px, py), (qx, qy)

This code (coming from SA) ensure that given a line (defined by two points), it covers the whole image.

Now, I would like to draw multiple parallel lines over a single image, with a parameters that specify how much the lines need to be distant from each other

My current attempt is this:

#Utility to find if two segments intersect
def ccw(A,B,C):
    Ax, Ay = A
    Bx, By = B
    Cx, Cy = C
    return (Cy-Ay) * (Bx-Ax) > (By-Ay) * (Cx-Ax)

#Utility to find if two segments intersect
def intersect(A,B,C,D):
    return ccw(A,C,D) != ccw(B,C,D) and ccw(A,B,C) != ccw(A,B,D)

#Utility to find if a segment intersect a square
def intersect_square(A,B, s_size):
    C = (1,1)
    D = (1, s_size-1)
    E = (s_size-1, +1)
    F = (s_size-1, s_size-1)
    return (ccw(A,C,D) != ccw(B,C,D) and ccw(A,B,C) != ccw(A,B,D)) or \
            (ccw(A,E,F) != ccw(B,E,F) and ccw(A,B,E) != ccw(A,B,F)) or \
            (ccw(A,C,E) != ccw(B,C,E) and ccw(A,B,C) != ccw(A,B,E)) or  \
            (ccw(A,D,F) != ccw(B,D,F) and ccw(A,B,D) != ccw(A,B,F))

#Utility to draw a white canvas         
def draw_white_image(x_size,y_size):
    img = np.zeros([x_size,y_size,3],dtype=np.uint8)
    img.fill(255) # or img[:] = 255
    return img

def draw_parallel_lines(img, thickness, colors, points, space):

    #Draw the first line
    draw_line(img, thickness, colors, points)
    x1, y1, x2, y2 = points
    flag = True
    while(flag):
        y1 += space
        y2 += space
        new_points = draw_line(img, thickness, colors, (x1,y1,x2,y2))
        flag = intersect_square(new_points[0], new_points[1], h)

This code move the y coordinates of my initial points, until the new lines I generate are outside the square. Sadly, this code gives this result:

enter image description here

The lines are not equidistant. I spent a few hours on this and I'm kinda sad now. Please help


Solution

  • If line segment is defined by points (x1,y1) and (x2,y2), it has direction vector:

    (dx, dy) = (x2-x1, y2-y1)
    

    Normalized (unit length) vector:

    len = sqrt((x2-x1)^2 + (y2-y1)^2)
    (udx, udy) = (dx / len, dy / len)
    

    Perpendicular vector:

    (px, py) = (-udy, udx)  
    

    Base point for parallel line at distance dist:

    (x1', y1') = (x1 + px * dist, y1 + py * dist)
    or
    (x1', y1') = (x1 - udy * dist, y1 + udx * dist) 
    

    Another end point:

    (x2', y2') = (x1' + dx, y1' + dy)
    

    To get parallel line in another direction, negate signs of px,py