PIL Image.fromstring из буфера PyOpengl имеет неправильный размер - PullRequest
Новая
       24

PIL Image.fromstring из буфера PyOpengl имеет неправильный размер

4 голосов
/ 18 августа 2011

Я использую PyOpenGL для рисования 2D-изображения.Затем я хочу использовать библиотеку изображений Python (PIL) для сохранения этого изображения на диск.Я использую GLUT для отображения изображения, которое отлично работает.Но когда я использую PIL для хранения изображения, он извлекает неправильное отсечение.Он имеет неправильный размер.

Вот минимальный пример, который воспроизводит эффект, и я также прикрепляю вывод, чтобы сделать его более четким без выполнения какого-либо кода. 

from OpenGL.GL import *
from OpenGL.GLUT import *
from PIL import Image

width, height = 640, 480

def DrawStuff():
    poly1 = [(0,0), (640,0), (0,480)]
    color = (0.5, 0.4, 0.3, 0.8)
    glClear(GL_COLOR_BUFFER_BIT)
    glPushMatrix()
    glLineWidth(5.0)

    glColor4f(*color)
    glBegin(GL_POLYGON)
    glVertex2f(poly1[0][0], poly1[0][1])
    glVertex2f(poly1[1][0], poly1[1][1])
    glVertex2f(poly1[2][0], poly1[2][1])    
    glVertex2f(poly1[0][0], poly1[0][1])
    glEnd() # GL_POLYGON

    glPopMatrix()
    glPixelStorei(GL_PACK_ALIGNMENT, 1)
    data = glReadPixels(0, 0, width, height, GL_RGBA, GL_UNSIGNED_BYTE)
    image = Image.fromstring("RGBA", (width, height), data)
    image.show()
    image.save('out.png', 'PNG')
    glutSwapBuffers()


# glut initialization
glutInit(sys.argv)
glutInitDisplayMode(GLUT_DOUBLE | GLUT_RGBA)
glutCreateWindow("Draw Polygons")
glutInitWindowSize(width, height)

# set the function to draw
glutDisplayFunc(DrawStuff)

# enable the alpha blending
glEnable(GL_BLEND)
glBlendFunc (GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);

# prepare for 2D drawing
glMatrixMode(GL_PROJECTION)
glLoadIdentity()
glOrtho(0, width, height, 0, 0, 1)
glDisable(GL_DEPTH_TEST)
glMatrixMode(GL_MODELVIEW)

# start the mainloop
glutMainLoop ()

this is how it looks int the GLUT window and how it is supposed to look like thisкак это выглядит в окне GLUT и как оно должно выглядеть

and this is how the saved image looks like, а вот так выглядит сохраненное изображение

Ответы [ 2 ]

1 голос
/ 19 августа 2011

Мне удалось решить мою собственную проблему.

Сначала я попробовал следующее решение, которое также могло бы помочь людям с сопутствующими проблемами: solution1

Но затем, путем обширных проб и ошибок, я обнаружил, что решение намногопроще.

Мне просто пришлось поменять две строки: 

glutCreateWindow("Draw Polygons")
glutInitWindowSize(width, height)

на 

glutInitWindowSize(width, height)
glutCreateWindow("Draw Polygons")

Видимо, размер должен быть установлен перед окном

0 голосов
/ 18 августа 2011

Следует учитывать, что в OpenGL система координат начинается в другом месте, чем в PIL.Посмотрите на это .

...