使用pygame / pyglet从数组重建图像时颜色不正确

时间:2013-03-09 14:05:28

标签: python pygame pyglet

我编写了这个程序,它将使用numpy和Image(PIL)库将图像作为一堆矩阵读取,并使用pyglet(和opengl)重建图像。

使用pyglet的代码如下:

import Image
import numpy
import window
import sys
import pyglet
import random
a=numpy.asarray(Image.open(sys.argv[1]))
h,w= a.shape[0],a.shape[1]
s=a[0]
print s.shape

#######################################
def display():
    x_a=0;y_a=h
    for page in a:
        for array in page: 
            j=array[2]
            k=array[1]
            l=array[0]
            pyglet.gl.glColor3f(l,j,k)
            pyglet.gl.glVertex2i(x_a,y_a)
            x_a+=1
        y_a-=1  
        x_a=0
######################################33
def on_draw(self):
    global w,h

    self.clear
    pyglet.gl.glClear(pyglet.gl.GL_COLOR_BUFFER_BIT)
    pyglet.gl.glBegin(pyglet.gl.GL_POINTS)
    display()
    pyglet.gl.glEnd()
    pyglet.image.get_buffer_manager().get_color_buffer().save('screenshot.png')
window.win.on_draw=on_draw

#######################################

u=window.win(w,h)
pyglet.app.run()

修改相同的代码以使用pygame库(并且没有任何opengl用法)

import pygame
import numpy
import Image
import sys
from pygame import gfxdraw

color=(255,255,255)

a=numpy.asarray(Image.open(sys.argv[1]))
h,w=a.shape[0],a.shape[1]

pygame.init()
screen = pygame.display.set_mode((w,h))

def uu():
    y_a=0
    for page in a:
        x_a=0
        for array in page:
            co=(array[0],array[1],array[2])
            pygame.gfxdraw.pixel(screen,x_a,y_a,co)
            x_a+=1
        y_a+=1

uu()
done = False

while not done:
        for event in pygame.event.get():
                if event.type == pygame.QUIT:
                        done = True

        pygame.display.flip()

pyglet与pygame的结果:

pyglet vs pygame

所以我的问题是......为什么会出现问题?我使用opengl逐像素地绘制图片的方式是否有问题,或者现在还有其他东西超出了我的理解范围?

1 个答案:

答案 0 :(得分:1)

Pygame.Color期望整数在0-255范围内,而pyglet.gl.glColor3f期望浮点数在0.0-1.0范围内。像这样的转换可以解决您的问题:

j=array[0] / 255.0
k=array[1] / 255.0
l=array[2] / 255.0
pyglet.gl.glColor3f(j,k,l)