2011-12-18 17 views
9

Ho un'applicazione in cui mi piacerebbe essere in grado di generare immagini PNG da dati in Python. Ho fatto qualche ricerca e ho trovato "PIL" che sembrava abbastanza obsoleto. C'è qualche altra libreria che sarebbe meglio per questo?Creazione di un file PNG in Python

Grazie,

+1

Come è il PIL obsoleto? –

+1

si utilizza il buon vecchio PIL (se non si sta già sviluppando oltre py3.2) – joaquin

risposta

6

Provare a utilizzare il modulo png per cominciare.

+0

Grazie, è fantastico – adam

29

I semplici file PNG possono essere generati abbastanza facilmente dal codice Python puro: tutto ciò che serve è il modulo zlib standard e alcuni byte-codifica per scrivere i blocchi. Ecco un esempio completo che il lettore casuale può utilizzare come iniziatore per il proprio generatore di 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 po 'più di profondità della maggior parte di noi, probabilmente incluso OP, stanno cercando, ma +1 per condividere così tante informazioni. Questo è un ottimo materiale di riferimento se voglio approfondire le mie conoscenze. – mightypile

+0

Questo esempio è fantastico. Qualsiasi modifica che puoi modificare per mostrare come aggiungere una palette (per alcuni colori base rosso/verde/etc e avere un canale alfa)? – 576i

+0

Volevo fornire una risposta molto breve. Ovviamente ci sono parecchie opzioni per costruire e codificare un'immagine. Fortunatamente lo standard PNG è abbastanza chiaro su cosa inserire in ogni blocco (https://www.w3.org/TR/PNG). Per evitare di riempire bit, si dovrebbe semplicemente cercare le opzioni per codificare sempre un valore in esattamente un byte. ///// Per un RGBA che può essere molto semplice: un chunk PLTE non ha opzioni di codifica => è sempre esattamente 8bit per canale colore, non più e non meno. Quindi ogni voce è di 4 byte (R, G, B, A). Il bitdepth IHDR si riferisce all'IDAT - con 8 bit si possono scrivere nuovamente solo i byte indice. –