Python使用DDA算法和中點Bresenham算法畫直線

先上效果圖

代碼

#!/usr/bin/env python
# coding=utf-8
from pylab import *
from matplotlib.ticker import MultipleLocator
import matplotlib.patches as patches

'''
1. 輸入直線兩端點 x0,y0 xn,yn
2. 計算初始值delta_x, delta_y,k=delta_y/delta_x,d=0, x=x0,y=y0
3. 繪製點x,y
4. d更新爲d+k,若d>0.5,則x,y更新爲x+1,y+1,d=d-1;不然x,y更新爲x+1,y
5. 重複3,4直到直線畫完
'''


def init(ax, width):
    # 設置長寬
    ax.axis([0, width, 0, width])

# 設置主刻度標籤的位置,標籤文本的格式
    majorLocator = MultipleLocator(1)
    minorLocator = MultipleLocator(0.5)
    ax.xaxis.set_major_locator(majorLocator)
    ax.yaxis.set_major_locator(majorLocator)
    # ax.xaxis.set_minor_locator(minorLocator)
    # ax.yaxis.set_minor_locator(minorLocator)
    ax.grid(True)  # x座標軸的網格使用主刻度


def add_pixel(x, y, ax, c):
    x = round(x)

    y = round(y)
    if c == 1:
        ax.add_patch(patches.Rectangle((x - 0.5, y - 0.5), 1, 1, color='b'))
        ax.plot(x, y, 'r.')
    else:
        ax.add_patch(patches.Rectangle((x - 0.5, y - 0.5), 1, 1))
        ax.plot(x, y, 'y.')


if __name__ == '__main__':

    # 將一行的字符串分割並轉化爲數字
    x0, y0, x1, y1, width = map(int, input("輸入直線的兩點和畫布的邊長: ").split(' '))
    if x0>x1:
        x0,x1=x1,x0
        y0,y1=y1,y0
    ax = subplot(121, aspect='equal',
            title='modified Bresenham')  # 改進的bresenham
    ax.plot([x0, x1], [y0, y1], '-k')
    bx = subplot(122, aspect='equal', title='DDA')  # DDA
    bx.plot([x0, x1], [y0, y1], '-k')
    # 圖形初始化
    init(ax, width)
    init(bx, width)
    delta_x = x1 - x0
    delta_y = y1 - y0
    d = 0
    if delta_x == 0:
        k = 999999999
    else:
        k = delta_y / delta_x
    x = round(x0)
    y = round(y0)
    '''
    DDA算法
    '''
    if k > -1 and k < 1:
        # X 最大位移
        while True:
            if x > x1:
                break
            add_pixel(x, y, bx, 1)
            x = x+1
            y = y+k
    elif k >= 1:
        # Y 最大位移
        while True:
            if y > y1:
                break
            add_pixel(x, y, bx, 1)
            y = y+1
            x = x+1/k
    else:
        while True:
            if y < y1:
                break
            add_pixel(x, y, bx, 1)
            y = y-1
            x = x-1/k

    '''
    k的範圍
    1.  (0,1)    x爲最大位移，y正向增長
    2.  (1,+inf) y爲最大位移，x正向增長
    3.  (0,-1)   x爲最大位移，y負向增長
    4.  (-1,-inf)y爲最大位移，y減少。x正向增長
    '''
    x = x0
    y = y0
    if k > 1:
        while True:
            if y > y1:
                break
            add_pixel(x, y, ax, 0)
            y = y + 1
            d = d + 1 / k
            if d > 0.5:
                x = x + 1
                d = d - 1
    elif k > 0:
        while True:
            if x > x1:
                break
            add_pixel(x, y, ax, 0)
            x = x + 1
            d = d + k
            if d > 0.5:
                y = y + 1
                d = d - 1
    elif k > -1:
        while True:
            if x > x1:
                break
            add_pixel(x, y, ax, 0)
            x = x + 1
            d = d - k
            if d > 0.5:
                y = y - 1
                d = d - 1
    else:
        while True:
            if y < y1:
                break
            add_pixel(x, y, ax, 0)
            y = y - 1
            d = d - 1 / k
            if d > 0.5:
                x = x + 1
                d = d - 1

    show()