2010-06-25 10 views
6

Sto disegnando un po 'di linee con ImageDraw.line() della libreria di immagini Python, ma mi sembrano orribili dal momento che non riesco a trovare un modo per antialearle. Come posso le righe anti-alias in PIL? Se PIL non può farlo, c'è un'altra libreria di manipolazione di immagini Python che può?Disegnare una linea anti-alias con la libreria di immagini di Python

+0

risposta possibile: http://stackoverflow.com/questions/1828345/any-way-to-make-nice-antialiased- round-corners-for-images-in-python – unutbu

+0

Bene, suppongo che sia una soluzione, se una lenta. – snostorm

risposta

3

aggdraw fornisce un disegno migliore di PIL.

+0

Sembra carino, ma non è confezionato per Ubuntu. Ho bisogno di questo. – snostorm

+0

Le linee appaiono ancora agitate in aggdraw usando il metodo di base (punti, penna). –

+0

È ancora disponibile per Python3? – DrMickeyLauer

11

Questa è una funzione molto rapidamente compromessa per tracciare una linea anti-alias con PIL che ho scritto dopo aver cercato su Google lo stesso problema, vedendo questo post e non riuscendo ad installare aggdraw e ad essere in una scadenza ristretta. È un'implementazione dell'algoritmo di Xiaolin Wu. Spero che aiuti chiunque a cercare la stessa cosa !!

:)

"""Library to draw an antialiased line.""" 
# http://stackoverflow.com/questions/3122049/drawing-an-anti-aliased-line-with-thepython-imaging-library 
# https://en.wikipedia.org/wiki/Xiaolin_Wu%27s_line_algorithm 
import math 


def plot(draw, img, x, y, c, col, steep, dash_interval): 
    """Draws an antiliased pixel on a line.""" 
    if steep: 
     x, y = y, x 
    if x < img.size[0] and y < img.size[1] and x >= 0 and y >= 0: 
     c = c * (float(col[3])/255.0) 
     p = img.getpixel((x, y)) 
     x = int(x) 
     y = int(y) 
     if dash_interval: 
      d = dash_interval - 1 
      if (x/dash_interval) % d == 0 and (y/dash_interval) % d == 0: 
       return 
     draw.point((x, y), fill=(
      int((p[0] * (1 - c)) + col[0] * c), 
      int((p[1] * (1 - c)) + col[1] * c), 
      int((p[2] * (1 - c)) + col[2] * c), 255)) 


def iround(x): 
    """Rounds x to the nearest integer.""" 
    return ipart(x + 0.5) 


def ipart(x): 
    """Floors x.""" 
    return math.floor(x) 


def fpart(x): 
    """Returns the fractional part of x.""" 
    return x - math.floor(x) 


def rfpart(x): 
    """Returns the 1 minus the fractional part of x.""" 
    return 1 - fpart(x) 


def draw_line_antialiased(draw, img, x1, y1, x2, y2, col, dash_interval=None): 
    """Draw an antialised line in the PIL ImageDraw. 

    Implements the Xialon Wu antialiasing algorithm. 

    col - color 
    """ 
    dx = x2 - x1 
    if not dx: 
     draw.line((x1, y1, x2, y2), fill=col, width=1) 
     return 

    dy = y2 - y1 
    steep = abs(dx) < abs(dy) 
    if steep: 
     x1, y1 = y1, x1 
     x2, y2 = y2, x2 
     dx, dy = dy, dx 
    if x2 < x1: 
     x1, x2 = x2, x1 
     y1, y2 = y2, y1 
    gradient = float(dy)/float(dx) 

    # handle first endpoint 
    xend = round(x1) 
    yend = y1 + gradient * (xend - x1) 
    xgap = rfpart(x1 + 0.5) 
    xpxl1 = xend # this will be used in the main loop 
    ypxl1 = ipart(yend) 
    plot(draw, img, xpxl1, ypxl1, rfpart(yend) * xgap, col, steep, 
     dash_interval) 
    plot(draw, img, xpxl1, ypxl1 + 1, fpart(yend) * xgap, col, steep, 
     dash_interval) 
    intery = yend + gradient # first y-intersection for the main loop 

    # handle second endpoint 
    xend = round(x2) 
    yend = y2 + gradient * (xend - x2) 
    xgap = fpart(x2 + 0.5) 
    xpxl2 = xend # this will be used in the main loop 
    ypxl2 = ipart(yend) 
    plot(draw, img, xpxl2, ypxl2, rfpart(yend) * xgap, col, steep, 
     dash_interval) 
    plot(draw, img, xpxl2, ypxl2 + 1, fpart(yend) * xgap, col, steep, 
     dash_interval) 

    # main loop 
    for x in range(int(xpxl1 + 1), int(xpxl2)): 
     plot(draw, img, x, ipart(intery), rfpart(intery), col, steep, 
      dash_interval) 
     plot(draw, img, x, ipart(intery) + 1, fpart(intery), col, steep, 
      dash_interval) 
     intery = intery + gradient 
+0

Che cos'è col? Che cosa è ripido? Una piccola spiegazione sarebbe molto apprezzata. –

+0

@RamenRecon, ero curioso di sapere anche col ripido. Basato sulla mia lettura del codice, steep è un booleano per se xdelta è minore di ydelta e quindi se la linea è ripida che probabilmente influisce sull'algoritmo, ma è usata solo internamente dalla funzione drawLine e la sua funzione helper di trama quindi non è nulla l'utente deve preoccuparsi. D'altra parte, l'utente deve fornire il colore Col e sembra essere un colore RGBA a 4-tuple (rosso, gr, blu, alfa) che viene usato per decidere il valore di trasparenza alfa di ogni punto (cioè l'effetto antialiasing). Curioso di provarlo e vedere quanto è veloce ... –

+1

Btw @toastie yours è l'unica implementazione Python manuale per l'antialiasing che si vede ovunque. Sono sorpreso che la tua risposta non abbia più upvotes, sono sicuro che abbia aiutato un sacco di persone. +1 –

2

ho avuto un problema simile, le mie battute avevano spigoli dove si cambia direzione. Ho trovato un indizio su come le linee vengono disegnate in IOS e si avvicinò con questo codice. Mette i cappucci arrotondati alle estremità delle linee e pulisce davvero le cose. Non esattamente anti-aliasing, ma sono totalmente nuovo per PIL e ha avuto un momento così difficile trovare una risposta che ho pensato di condividere. Ha bisogno di qualche ritocco e non v'è probabilmente un modo migliore, ma fa quello che mi serve :)


    from PIL import Image 
    import ImageDraw 

    class Point: 
     def __init__(self, x, y): 
      self.x = x 
      self.y = y 

    class DrawLines: 
     def draw(self, points, color, imageName): 
      img = Image.new("RGBA", [1440,1080], (255,255,255,0)) 
      draw = ImageDraw.Draw(img) 
      linePoints = [] 
      for point in points: 
       draw.ellipse((point.x-7, point.y-7, point.x+7, point.y+7), fill=color) 
       linePoints.append(point.x) 
       linePoints.append(point.y) 
      draw.line(linePoints, fill=color, width=14) 
      img.save(imageName) 

    p1 = Point(100,200) 
    p2 = Point(190,250) 
    points = [p1,p2] 

    red = (255,0,0) 
    drawLines = DrawLines() 
    drawLines.draw(points, red, "C:\\test.png") 
Problemi correlati