def __init__(self, width, height, background=[255,255,255]):
self.width = width
self.height = height
self.triangles = []
self.batch = Batch()
self.bg_colour = background
has_fbo = gl.gl_info.have_extension('GL_EXT_framebuffer_object')
#setup a framebuffer
self.fb = gl.GLuint()
gl.glGenFramebuffersEXT(1, ctypes.byref(self.fb))
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, self.fb)
#allocate a texture for the fb to render to
tex = image.Texture.create_for_size(gl.GL_TEXTURE_2D, width, height, gl.GL_RGBA)
gl.glBindTexture(gl.GL_TEXTURE_2D, tex.id)
gl.glFramebufferTexture2DEXT(gl.GL_FRAMEBUFFER_EXT, gl.GL_COLOR_ATTACHMENT0_EXT, gl.GL_TEXTURE_2D, tex.id, 0)
status = gl.glCheckFramebufferStatusEXT(gl.GL_FRAMEBUFFER_EXT)
assert status == gl.GL_FRAMEBUFFER_COMPLETE_EXT
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, 0)
self.bg = self.batch.add( 6,
gl.GL_TRIANGLES,None,
("v2i/static", (0,0,0,height,width,height,width,height,width,0,0,0)),
("c3B/static",background*6)
)
def __init__(self, width, height, background=[255, 255, 255]):
self.width = width
self.height = height
self.triangles = []
self.batch = Batch()
self.bg_colour = background
has_fbo = gl.gl_info.have_extension('GL_EXT_framebuffer_object')
#setup a framebuffer
self.fb = gl.GLuint()
gl.glGenFramebuffersEXT(1, ctypes.byref(self.fb))
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, self.fb)
#allocate a texture for the fb to render to
tex = image.Texture.create_for_size(gl.GL_TEXTURE_2D, width, height,
gl.GL_RGBA)
gl.glBindTexture(gl.GL_TEXTURE_2D, tex.id)
gl.glFramebufferTexture2DEXT(gl.GL_FRAMEBUFFER_EXT,
gl.GL_COLOR_ATTACHMENT0_EXT,
gl.GL_TEXTURE_2D, tex.id, 0)
status = gl.glCheckFramebufferStatusEXT(gl.GL_FRAMEBUFFER_EXT)
assert status == gl.GL_FRAMEBUFFER_COMPLETE_EXT
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, 0)
self.bg = self.batch.add(
6, gl.GL_TRIANGLES, None,
("v2i/static",
(0, 0, 0, height, width, height, width, height, width, 0, 0, 0)),
("c3B/static", background * 6))
def __init__(self, texture=None):
from pyglet import gl
fbo_id = gl.GLuint(0)
gl.glGenFramebuffersEXT(1, gl.byref(fbo_id))
self.fbo_id = fbo_id
self._texture = None
if texture:
self.bind()
self.texture = texture
self.unbind()
def __init__(self, texture=None):
from pyglet import gl
fbo_id = gl.GLuint(0)
gl.glGenFramebuffersEXT(1, gl.byref(fbo_id))
self.fbo_id = fbo_id
self._texture = None
if texture:
self.bind()
self.texture = texture
self.unbind()
def __init__(self, width, height):
if not gl.gl_info.have_extension('GL_EXT_framebuffer_object'):
raise Exception('framebuffer object extension not available')
texture_id = buffer_texture(width, height)
self.texture = Texture(width, height, gl.GL_TEXTURE_2D, texture_id)
framebuffer_id = gl.GLuint()
gl.glGenFramebuffersEXT(1, byref(framebuffer_id))
self.id = framebuffer_id.value
with self:
gl.glFramebufferTexture2DEXT(gl.GL_FRAMEBUFFER_EXT,
gl.GL_COLOR_ATTACHMENT0_EXT + 0,
gl.GL_TEXTURE_2D, texture_id, 0)
def __init__(self, width, height):
if not gl.gl_info.have_extension('GL_EXT_framebuffer_object'):
raise Exception('framebuffer object extension not available')
texture_id = buffer_texture(width, height)
self.texture = Texture(width, height, gl.GL_TEXTURE_2D, texture_id)
framebuffer_id = gl.GLuint()
gl.glGenFramebuffersEXT(1, byref(framebuffer_id))
self.id = framebuffer_id.value
with self:
gl.glFramebufferTexture2DEXT(gl.GL_FRAMEBUFFER_EXT,
gl.GL_COLOR_ATTACHMENT0_EXT + 0,
gl.GL_TEXTURE_2D,
texture_id,
0)
def __init__(self, width = 512, height = 512, screen_size=None ):
self.render_target_size = width, height
# create the framebuffer
self._buffer = (c_uint * 1)() ; gl.glGenFramebuffersEXT(1,self._buffer)
self._buffer = self._buffer[0]
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, self._buffer)
self.rgb_texture = self._create_texture(texture_size=self.render_target_size)
gl.glFramebufferTexture2DEXT(gl.GL_FRAMEBUFFER_EXT, gl.GL_COLOR_ATTACHMENT0_EXT, gl.GL_TEXTURE_2D, self.rgb_texture, 0)
status = gl.glCheckFramebufferStatusEXT(gl.GL_FRAMEBUFFER_EXT)
if status != gl.GL_FRAMEBUFFER_COMPLETE_EXT:
print("ERROR on FRAMEBUFFER")
return
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, 0)
def __init__(self):
"""Framebuffer Object (FBO) for off-screen rendering.
A framebuffer object contains one or more
framebuffer-attachable images. These can be either
renderbuffers or texture images.
For now the framebuffer object handles only textures.
"""
## Create a framebuffer object
framebuffer = gl.GLuint()
gl.glGenFramebuffersEXT(1, ctypes.byref(framebuffer))
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, framebuffer)
self.id = framebuffer
self.MAX_COLOR_ATTACHMENTS = MAX_COLOR_ATTACHMENTS
self._textures = []
def renderToTexture(w, h, function):
import ctypes
from pyglet import gl
from pyglet import image
from pyglet.gl import gl_info
global has_fbo
if has_fbo is None:
has_fbo = gl_info.have_extension('GL_EXT_framebuffer_object')
# enforce dimensions are ints
w, h = int(w), int(h)
# set up viewport
gl.glPushAttrib(gl.GL_VIEWPORT_BIT | gl.GL_TRANSFORM_BIT)
gl.glViewport(0, 0, w, h)
gl.glMatrixMode(gl.GL_PROJECTION)
gl.glPushMatrix()
gl.glLoadIdentity()
gl.glOrtho(0, w, 0, h, -1, 1)
gl.glMatrixMode(gl.GL_MODELVIEW)
if has_fbo:
# render directly to texture
# create our frame buffer
fbo = gl.GLuint()
gl.glGenFramebuffersEXT(1, ctypes.byref(fbo))
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, fbo)
# allocate a texture and add to the frame buffer
tex = image.Texture.create_for_size(gl.GL_TEXTURE_2D, w, h, gl.GL_RGBA)
gl.glBindTexture(gl.GL_TEXTURE_2D, tex.id)
gl.glFramebufferTexture2DEXT(gl.GL_FRAMEBUFFER_EXT,
gl.GL_COLOR_ATTACHMENT0_EXT,
gl.GL_TEXTURE_2D, tex.id, 0)
status = gl.glCheckFramebufferStatusEXT(gl.GL_FRAMEBUFFER_EXT)
assert status == gl.GL_FRAMEBUFFER_COMPLETE_EXT
# now render
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, fbo)
function()
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, 0)
# clean up
gl.glDeleteFramebuffersEXT(1, ctypes.byref(fbo))
else:
# render and copy to texture
# render
function()
# grab the buffer and copy contents to the texture
buffer = image.get_buffer_manager().get_color_buffer()
tex = image.Texture.create_for_size(gl.GL_TEXTURE_2D, w, h, gl.GL_RGBA)
tex.blit_into(buffer.get_region(0, 0, w, h), 0, 0, 0)
gl.glMatrixMode(gl.GL_PROJECTION)
gl.glPopMatrix()
gl.glPopAttrib()
# return the region (the whole texture will most likely be larger)
return tex.get_region(0, 0, w, h)
def __init__(self):
framebuffer = gl.GLuint(0)
gl.glGenFramebuffersEXT(1, ct.byref(framebuffer))
self.buff_id = framebuffer
def __init__(self):
"""Create a new framebuffer object"""
id = gl.GLuint(0)
gl.glGenFramebuffersEXT(1, ct.byref(id))
self._id = id.value
def __init__(self):
"""Create a new framebuffer object"""
id = gl.GLuint(0)
gl.glGenFramebuffersEXT(1, ct.byref(id))
self._id = id.value
reaction_shader.unbind()
# Color shader
# ------------
vertex_shader = open('./color.vert').read()
fragment_shader = open('./color.frag').read()
color_shader = Shader(vertex_shader, fragment_shader)
color_shader.bind()
color_shader.uniformi('texture', 0)
color_shader.unbind()
# Framebuffer
# -----------
framebuffer = gl.GLuint(0)
gl.glGenFramebuffersEXT(1, ctypes.byref(framebuffer))
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, framebuffer)
gl.glFramebufferTexture2DEXT(gl.GL_FRAMEBUFFER_EXT, gl.GL_COLOR_ATTACHMENT0_EXT,
gl.GL_TEXTURE_2D, texture_uv.id, 0);
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, 0)
@window.event
def on_mouse_drag(x, y, dx, dy, button, modifiers):
sprite.x = int((x/float(window.width)) * width)-sprite.width//2
sprite.y = int((y/float(window.height)) * height)-sprite.height//2
# Compute
gl.glViewport(0, 0, width, height)
gl.glMatrixMode(gl.GL_PROJECTION)
def renderToTexture(w, h, function):
import ctypes
from pyglet import gl
from pyglet import image
from pyglet.gl import gl_info
global has_fbo
if has_fbo is None:
has_fbo = gl_info.have_extension("GL_EXT_framebuffer_object")
# enforce dimensions are ints
w, h = int(w), int(h)
# set up viewport
gl.glPushAttrib(gl.GL_VIEWPORT_BIT | gl.GL_TRANSFORM_BIT)
gl.glViewport(0, 0, w, h)
gl.glMatrixMode(gl.GL_PROJECTION)
gl.glPushMatrix()
gl.glLoadIdentity()
gl.glOrtho(0, w, 0, h, -1, 1)
gl.glMatrixMode(gl.GL_MODELVIEW)
if has_fbo:
# render directly to texture
# create our frame buffer
fbo = gl.GLuint()
gl.glGenFramebuffersEXT(1, ctypes.byref(fbo))
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, fbo)
# allocate a texture and add to the frame buffer
tex = image.Texture.create_for_size(gl.GL_TEXTURE_2D, w, h, gl.GL_RGBA)
gl.glBindTexture(gl.GL_TEXTURE_2D, tex.id)
gl.glFramebufferTexture2DEXT(gl.GL_FRAMEBUFFER_EXT, gl.GL_COLOR_ATTACHMENT0_EXT, gl.GL_TEXTURE_2D, tex.id, 0)
status = gl.glCheckFramebufferStatusEXT(gl.GL_FRAMEBUFFER_EXT)
assert status == gl.GL_FRAMEBUFFER_COMPLETE_EXT
# now render
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, fbo)
function()
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, 0)
# clean up
gl.glDeleteFramebuffersEXT(1, ctypes.byref(fbo))
else:
# render and copy to texture
# render
function()
# grab the buffer and copy contents to the texture
buffer = image.get_buffer_manager().get_color_buffer()
tex = image.Texture.create_for_size(gl.GL_TEXTURE_2D, w, h, gl.GL_RGBA)
tex.blit_into(buffer.get_region(0, 0, w, h), 0, 0, 0)
gl.glMatrixMode(gl.GL_PROJECTION)
gl.glPopMatrix()
gl.glPopAttrib()
# return the region (the whole texture will most likely be larger)
return tex.get_region(0, 0, w, h)
def __init__(self, width, height):
"""Creates a FBO"""
self.initialized = False
assert self.supported()
self.width = width
self.height = height
self.framebuffer_id = ctypes.c_uint(0)
self.depthbuffer_id = ctypes.c_uint(0)
self.texture_id = ctypes.c_uint(0)
# Frame buffer
gl.glGenFramebuffersEXT(
1, # number of buffers created
ctypes.byref(self.framebuffer_id), # dest. id
)
self.initialized = True
with self._bound_context(gl.GL_FRAMEBUFFER_EXT):
# Depth buffer
gl.glGenRenderbuffersEXT(
1, # no. of buffers created
ctypes.byref(self.depthbuffer_id), # dest. id
)
gl.glBindRenderbufferEXT(
gl.GL_RENDERBUFFER_EXT, # target
self.depthbuffer_id, # id
)
gl.glRenderbufferStorageEXT(
gl.GL_RENDERBUFFER_EXT, # target
gl.GL_DEPTH_COMPONENT, # internal format
self.width, self.height, # size
)
gl.glFramebufferRenderbufferEXT(
gl.GL_FRAMEBUFFER_EXT, # target
gl.GL_DEPTH_ATTACHMENT_EXT, # attachment point
gl.GL_RENDERBUFFER_EXT, # renderbuffer target
self.depthbuffer_id, # renderbuffer id
)
# Target Texture
gl.glGenTextures(
1, # no. of textures
ctypes.byref(self.texture_id), # dest. id
)
gl.glBindTexture(
gl.GL_TEXTURE_2D, # target
self.texture_id, # texture id
)
# Black magic (props to pyprocessing!)
# (nearest works, as well as linear)
gl.glTexParameteri(
gl.GL_TEXTURE_2D, # target
gl.GL_TEXTURE_MAG_FILTER, # property name
gl.GL_LINEAR, # value
)
gl.glTexParameteri(
gl.GL_TEXTURE_2D, # target
gl.GL_TEXTURE_MIN_FILTER, # property name
gl.GL_LINEAR, # value
)
# Attach texture to FBO
gl.glTexImage2D(
gl.GL_TEXTURE_2D, # target
0, # mipmap level (0=default)
gl.GL_RGBA8, # internal format
self.width, self.height, # size
0, # border
gl.GL_RGBA, # format
gl.GL_UNSIGNED_BYTE, # type
None, # data
)
gl.glFramebufferTexture2DEXT(
gl.GL_FRAMEBUFFER_EXT, # target
gl.GL_COLOR_ATTACHMENT0_EXT, # attachment point
gl.GL_TEXTURE_2D, # texture target
self.texture_id, # tex id
0, # mipmap level
)
# sanity check
status = gl.glCheckFramebufferStatusEXT(gl.GL_FRAMEBUFFER_EXT)
assert status == gl.GL_FRAMEBUFFER_COMPLETE_EXT
def __init__(self):
self.buff_id = gl.GLuint(0)
gl.glGenFramebuffersEXT(1, ct.byref(self.buff_id))
def __init__(self):
framebuffer = gl.GLuint(0)
gl.glGenFramebuffersEXT(1, ct.byref(framebuffer))
self.buff_id = framebuffer
def __init__(self):
self.buff_id = gl.GLuint(0)
gl.glGenFramebuffersEXT(1, ct.byref(self.buff_id))
def main(image_file, num_polygons=250, resume=False):
global image_pixels
global keeps
global newdrawing, olddrawing
global blitted
pic = image.load(image_file)
width = pic.width
height = pic.height
size = width*height
#setup the framebuffer
# create our frame buffer
fbo = gl.GLuint()
gl.glGenFramebuffersEXT(1, ctypes.byref(fbo))
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, fbo)
# allocate a texture and add to the frame buffer
tex = image.Texture.create_for_size(gl.GL_TEXTURE_2D, width, height, gl.GL_RGBA)
gl.glBindTexture(gl.GL_TEXTURE_2D, tex.id)
gl.glFramebufferTexture2DEXT(gl.GL_FRAMEBUFFER_EXT,
gl.GL_COLOR_ATTACHMENT0_EXT, gl.GL_TEXTURE_2D, tex.id, 0)
status = gl.glCheckFramebufferStatusEXT(gl.GL_FRAMEBUFFER_EXT)
assert status == gl.GL_FRAMEBUFFER_COMPLETE_EXT
newdrawing = Drawing(width,height,fbo)
try:
os.path.isfile(resume)
except TypeError:
svg_file = image_file + '.svg'
if resume == True:
newdrawing.svg_import(svg_file)
else:
newdrawing.generate(num_polygons)
else:
newdrawing.svg_import(resume)
svg_file = resume
w = window.Window(width*2,height,"cows", vsync = False)
w.set_visible(True)
gl.glEnable(gl.GL_BLEND)
gl.glBlendFunc(gl.GL_SRC_ALPHA, gl.GL_ONE_MINUS_SRC_ALPHA)
fps = pyglet.clock.Clock()
#use this for pixel dumps
a = (gl.GLubyte * (4*size))(0)
@w.event
def on_close():
gl.glDeleteFramebuffersEXT(1, ctypes.byref(olddrawing.fb))
olddrawing.svg_export(image_file, svg_file)
@w.event
def on_draw():
#w.clear()
global parent
global parentdiff
global olddrawing,newdrawing
global blitted
global image_pixels
global keeps
global i
if not blitted:
"""
At the start we've not seen the target before,
so draw it and store the pixel data.
"""
gl.glBindFramebufferEXT(gl.GL_FRAMEBUFFER_EXT, 0)
gl.glClear(gl.GL_COLOR_BUFFER_BIT)
pic.blit(0,0)
blitted = 1
image_pixels = (gl.GLubyte * (4*size))(0)
gl.glReadPixels(0,
0,
newdrawing.width,
newdrawing.height,
gl.GL_RGBA,
gl.GL_UNSIGNED_BYTE,
image_pixels
)
image_pixels = np.frombuffer(image_pixels, dtype=np.uint8).astype(np.int32)
# Draw the new child
newdrawing.draw()
# Read the pixel data for the child and find out if its any good
gl.glReadPixels(0,
0,
newdrawing.width,
newdrawing.height,
gl.GL_RGBA, gl.GL_UNSIGNED_BYTE, a)
diff = compute_diff(a)
if parent == None or diff < parentdiff:
# The new drawing is better.
# Redraw the parent as this child.
# Set this child's diff as the new one to beat.
parent = image.ImageData(newdrawing.width,newdrawing.height,"RGBA",a)
parentdiff = diff
draw_parent(parent, newdrawing.width)
else:
# The new drawing sucks. Replace it
newdrawing = olddrawing
i += 1
if (i % 20 == 0):
# Use the window title to let the user know how we're doing
w.set_caption(str(fps.get_fps())+" "+str(parentdiff) + " " + str(log(parentdiff,10))+ " " + str(i))
fps.tick()
# Isolate and mutate the child on each clock tick
pyglet.clock.schedule(update)
# Set it running
pyglet.app.run()
