2011-12-18 26 views
9

Tengo una aplicación donde me gustaría poder generar imágenes PNG a partir de datos en Python. He hecho algunas búsquedas y encontré "PIL" que parecía bastante anticuado. ¿Hay alguna otra biblioteca que sería mejor para esto?Creando un archivo PNG en Python

Gracias,

+1

¿Cómo está PIL desactualizado? –

+1

utiliza un buen PIL anterior (si aún no está desarrollando más allá de py3.2) – joaquin

Respuesta

6

Trate de usar el módulo png para empezar.

+0

Gracias, eso es genial – adam

29

Los archivos PNG simples se pueden generar bastante fácilmente a partir del código Python puro; todo lo que necesita es el módulo zlib estándar y algunos bytes de codificación para escribir los fragmentos. Aquí hay un ejemplo completo que el lector casual puede usar como un iniciador para su propio generador png:

#! /usr/bin/python 
""" Converts a list of list into gray-scale PNG image. """ 
__copyright__ = "Copyright (C) 2014 Guido Draheim" 
__licence__ = "Public Domain" 

import zlib 
import struct 

def makeGrayPNG(data, height = None, width = None): 
    def I1(value): 
     return struct.pack("!B", value & (2**8-1)) 
    def I4(value): 
     return struct.pack("!I", value & (2**32-1)) 
    # compute width&height from data if not explicit 
    if height is None: 
     height = len(data) # rows 
    if width is None: 
     width = 0 
     for row in data: 
      if width < len(row): 
       width = len(row) 
    # generate these chunks depending on image type 
    makeIHDR = True 
    makeIDAT = True 
    makeIEND = True 
    png = b"\x89" + "PNG\r\n\x1A\n".encode('ascii') 
    if makeIHDR: 
     colortype = 0 # true gray image (no palette) 
     bitdepth = 8 # with one byte per pixel (0..255) 
     compression = 0 # zlib (no choice here) 
     filtertype = 0 # adaptive (each scanline seperately) 
     interlaced = 0 # no 
     IHDR = I4(width) + I4(height) + I1(bitdepth) 
     IHDR += I1(colortype) + I1(compression) 
     IHDR += I1(filtertype) + I1(interlaced) 
     block = "IHDR".encode('ascii') + IHDR 
     png += I4(len(IHDR)) + block + I4(zlib.crc32(block)) 
    if makeIDAT: 
     raw = b"" 
     for y in xrange(height): 
      raw += b"\0" # no filter for this scanline 
      for x in xrange(width): 
       c = b"\0" # default black pixel 
       if y < len(data) and x < len(data[y]): 
        c = I1(data[y][x]) 
       raw += c 
     compressor = zlib.compressobj() 
     compressed = compressor.compress(raw) 
     compressed += compressor.flush() #!! 
     block = "IDAT".encode('ascii') + compressed 
     png += I4(len(compressed)) + block + I4(zlib.crc32(block)) 
    if makeIEND: 
     block = "IEND".encode('ascii') 
     png += I4(0) + block + I4(zlib.crc32(block)) 
    return png 

def _example(): 
    with open("cross3x3.png","wb") as f: 
     f.write(makeGrayPNG([[0,255,0],[255,255,255],[0,255,0]])) 
+3

Un poco más de profundidad que la mayoría de nosotros, probablemente incluyendo OP, estamos buscando, pero +1 para compartir tanta información. Este es un gran material de referencia si alguna vez quiero profundizar con pngs. – mightypile

+0

Este ejemplo es impresionante. ¿Puede cambiar cualquier cambio para mostrar cómo agregar una paleta (para algunos colores básicos de rojo/verde/etc. y tener un canal alfa)? – 576i

+0

Quería dar una respuesta muy breve. Obviamente, hay bastantes opciones para construir y codificar una imagen. Afortunadamente, el estándar PNG es bastante claro sobre qué poner en cada fragmento (https://www.w3.org/TR/PNG). Para evitar el relleno de bits uno simplemente buscaría las opciones para codificar siempre un valor en exactamente un byte. ///// Para un RGBA que puede ser muy simple: un fragmento PLTE no tiene opciones de codificación => siempre es exactamente 8 bits por canal de color, ni más ni menos. Entonces, cada entrada tiene 4 bytes (R, G, B, A). La profundidad de bits de IHDR se refiere a IDAT: con 8 bits uno puede escribir solo bytes de índice nuevamente. –