网上给的算法只能画0-90°的线条(而且还不能水平和垂直线条),我写的这个可以画0-360°。。。调bug调了好久,现在分享给大家,但是没有做线条越界图片边缘的截断(我累了),所以不要让线条超越图片大小,或者你改一下
def BresenhamLine(img,x1, y1, x2, y2, color):
y_neg = False
dx = x2 - x1
if (dx < 0):#向左走,就将起点终点调换
temp = x2
x2 = x1
x1 = temp
temp = y2
y2 = y1
y1 = temp
dx = x2 - x1 # 重新计算dx
dy = y2 - y1
if (dy < 0): # 向下走,就将计算出来的y加个负号
y_neg=True
dy=-dy
x = x1
y = y1#从起点开始
if(dx==0):#打竖画线,向左走换了(x1,y1),(x2,y2),向下走没有换,注意
y=min(y1,y2)
y_end=max(y1,y2)
while(y<=y_end):
img[x, y, color] = 200
y+=1
return img
elif(dy==0):#打横画线
while (x <= x2):
img[x, y, color] = 200
x+= 1
return img
else:
k = dy /(1.0*dx)#斜率
if (abs(k) <= 1):#画横斜线,|斜率|<=1
sign_k = True # 斜率标记
xend = x2
else: #| 斜率|>1
sign_k = False# 等会关于y=x对称,计算|斜率|<=1的值,画|斜率|>1的画
#先打横了来算,画的时候才对称
xend = x1+dy
temp=dy
dy=dx
dx=temp
k = dy / (1.0*dx)# 斜率
p = 2 * dy - dx#P0 = 2*dy - dx
v = 2 * dy
u = p - dx#Pk+1 = Pk + 2*dy - 2*dx,u 表示 2*dy - 2*dx
############开始画线###################3
b=y1-x1
while ( x <= xend):
if (sign_k == True):#画横斜线,|斜率|<=1
if(y_neg==True):#向下走,画横线,只需要关于y=y0上下翻转就好了
#SetPixel(hDC, x, y, color);
img[x, 2*y1-y, color] = 200#(x,-y)
else:#向上走,画横线,最正常了
img[x, y, color] = 200#(x,y)
else :#画竖斜线,|斜率|>1
#SetPixel(hDC, y, x, color);
if (y_neg == True):#画竖斜线,又向下走,要先关于y=kx+b对称,之后再关于y=y0上下翻转
# SetPixel(hDC, x, y, color);
img[y-b, 2*y1-(x+b), color] = 200#(y,-x)
else:#向上走,画竖斜线,关于y=kx+b对称
img[y-b, x+b, color] = 200#(y,x)
if (p>0):
y+=1
p = p + u
#cout << "1 ";
else:
p = p + v
#//cout << "0 ";
x+=1
return img