2010-04-29 16 views
0

J'essaie d'utiliser l'algorithme de ligne de Bresenham pour calculer le champ de vision sur une grille. Le code que j'utilise calcule les lignes sans problème, mais j'ai du mal à le faire pour toujours renvoyer la ligne allant du point de départ au point de terminaison. Que dois-je faire pour que toutes les lignes retournées de course (X0, Y0) à (x1, y1)Modification de l'algorithme de la ligne de Bresenham

def bresenham_line(self, x0, y0, x1, y1): 
steep = abs(y1 - y0) > abs(x1 - x0) 
if steep: 
    x0, y0 = y0, x0 
    x1, y1 = y1, x1 

if x0 > x1: 
    x0, x1 = x1, x0 
    y0, y1 = y1, y0 

if y0 < y1: 
    ystep = 1 
else: 
    ystep = -1 

deltax = x1 - x0 
deltay = abs(y1 - y0) 
error = -deltax/2 
y = y0 

line = []  
for x in range(x0, x1 + 1): 
    if steep: 
     line.append((y,x)) 
    else: 
     line.append((x,y)) 

    error = error + deltay 
    if error > 0: 
     y = y + ystep 
     error = error - deltax 
return line 
+0

J'utilise xrange ici, comme x0 et x1 pourraient être assez éloignés. – clahey

Répondre

2

Rappelez-vous si vous avez activé x0 et x1, puis inversez la liste si vous avez fait.

if x0 > x1: 
    x0, x1 = x1, x0 
    y0, y1 = y1, y0 

devient

switched = False 
if x0 > x1: 
    switched = True 
    x0, x1 = x1, x0 
    y0, y1 = y1, y0 

et à la fin, il suffit d'ajouter:

if switched: 
    line.reverse()