def play(frames, delay=0, magnify=1, background=None, blit_position=None, show_frame_number=True):
w = window.Window(visible=False, resizable=True)
arr = numpy.zeros([100,100], dtype=numpy.uint8)
aii = ArrayInterfaceImage(arr)
img = aii.texture
if background is None:
checks = image.create(32, 32, image.CheckerImagePattern())
background = image.TileableTexture.create_for_image(checks)
background_tiled = True
else:
if background.dtype != np.uint8:
'converting background to uint8'
background = np.array(background, dtype=np.uint8)*255
background_aii = ArrayInterfaceImage(background)
background = background_aii.texture
background_tiled = False
if background_tiled:
w.width = img.width
w.height = img.height
w.set_visible()
else:
w.width = background.width
w.height = background.height
w.set_visible()
LIBGL_ALWAYS_SOFTWARE = 1
glEnable(GL_BLEND)
glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
for i, f in enumerate(frames):
arr = f
if arr is not None:
if arr.dtype != np.uint8:
'converting'
arr = np.array(arr, dtype=np.uint8)*255
if magnify != 1:
newshape = [arr.shape[0]*magnify, arr.shape[1]*magnify]
arr = nim.rebin(arr, newshape)
try:
aii.view_new_array(arr)
except: # size changed
print 'size changed!'
#w.width = arr.shape[1]
#w.height = arr.shape[0]
aii = ArrayInterfaceImage(arr)
img = aii.texture
w.dispatch_events()
w.clear()
if background_tiled:
background.blit_tiled(0, 0, 0, 100, 100) #w.width, w.height)
else:
background.blit(0,0,0)
# add some overlays:
if show_frame_number:
s = 'frame: ' + str(i)
label = pyglet.text.Label(s,
font_name='courier',
font_size=36,
x=10, y=10)
label.draw()
if blit_position is None:
img.blit(0, 0, 0)
else:
img.blit(blit_position[i][1], blit_position[i][0], 0)
#frame_number_text.draw()
if 0:
r = arr.shape[0]/2.
body_axis = npmovie.kalmanobj.long_axis[f]
wing_axis = npmovie.kalmanobj.wingaxisR[f]
wing_center = npmovie.kalmanobj.wingcenterR[f]
#print wing_center, wing_axis
if mode == 'wingimgR':
print f, uframe.wingimgR.sum()
pyglet.graphics.draw(2, pyglet.gl.GL_LINES,('v2i', (int(r), int(r), int(body_axis[1]*10*magnify)+int(r), int(body_axis[0]*10*magnify)+int(r))))
try:
pyglet.graphics.draw(2, pyglet.gl.GL_LINES,('v2i', (int(r), int(r), int(wing_center[1]*magnify), int(wing_center[0]*magnify))))
pyglet.graphics.draw(1, pyglet.gl.GL_POINTS,('v2i', (int(wing_center[1]*magnify), int(wing_center[0]*magnify))))
pass
except:
pass
w.flip()
time.sleep(delay) # slow down the playback
w.close()
def play(frames,
delay=0,
magnify=1,
background=None,
blit_position=None,
show_frame_number=True):
w = window.Window(visible=False, resizable=True)
arr = numpy.zeros([100, 100], dtype=numpy.uint8)
aii = ArrayInterfaceImage(arr)
img = aii.texture
if background is None:
checks = image.create(32, 32, image.CheckerImagePattern())
background = image.TileableTexture.create_for_image(checks)
background_tiled = True
else:
if background.dtype != np.uint8:
'converting background to uint8'
background = np.array(background, dtype=np.uint8) * 255
background_aii = ArrayInterfaceImage(background)
background = background_aii.texture
background_tiled = False
if background_tiled:
w.width = img.width
w.height = img.height
w.set_visible()
else:
w.width = background.width
w.height = background.height
w.set_visible()
LIBGL_ALWAYS_SOFTWARE = 1
glEnable(GL_BLEND)
glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
for i, f in enumerate(frames):
arr = f
if arr is not None:
if arr.dtype != np.uint8:
'converting'
arr = np.array(arr, dtype=np.uint8) * 255
if magnify != 1:
newshape = [arr.shape[0] * magnify, arr.shape[1] * magnify]
arr = nim.rebin(arr, newshape)
try:
aii.view_new_array(arr)
except: # size changed
print 'size changed!'
#w.width = arr.shape[1]
#w.height = arr.shape[0]
aii = ArrayInterfaceImage(arr)
img = aii.texture
w.dispatch_events()
w.clear()
if background_tiled:
background.blit_tiled(0, 0, 0, 100, 100) #w.width, w.height)
else:
background.blit(0, 0, 0)
# add some overlays:
if show_frame_number:
s = 'frame: ' + str(i)
label = pyglet.text.Label(s,
font_name='courier',
font_size=36,
x=10,
y=10)
label.draw()
if blit_position is None:
img.blit(0, 0, 0)
else:
img.blit(blit_position[i][1], blit_position[i][0], 0)
#frame_number_text.draw()
if 0:
r = arr.shape[0] / 2.
body_axis = npmovie.kalmanobj.long_axis[f]
wing_axis = npmovie.kalmanobj.wingaxisR[f]
wing_center = npmovie.kalmanobj.wingcenterR[f]
#print wing_center, wing_axis
if mode == 'wingimgR':
print f, uframe.wingimgR.sum()
pyglet.graphics.draw(
2, pyglet.gl.GL_LINES,
('v2i',
(int(r), int(r), int(body_axis[1] * 10 * magnify) +
int(r), int(body_axis[0] * 10 * magnify) + int(r))))
try:
pyglet.graphics.draw(
2, pyglet.gl.GL_LINES,
('v2i', (int(r), int(r), int(wing_center[1] * magnify),
int(wing_center[0] * magnify))))
pyglet.graphics.draw(1, pyglet.gl.GL_POINTS,
('v2i',
(int(wing_center[1] * magnify),
int(wing_center[0] * magnify))))
pass
except:
pass
w.flip()
time.sleep(delay) # slow down the playback
w.close()
def play_npmovie(npmovie, delay=0, magnify=1, mode="uimg", frames=None, framerange=None, print_frames=False):
w = window.Window(visible=False, resizable=True)
arr = numpy.zeros([60, 60], dtype=numpy.uint8)
aii = ArrayInterfaceImage(arr)
img = aii.texture
w.width = img.width
w.height = img.height
w.set_visible()
checks = image.create(32, 32, image.CheckerImagePattern())
background = image.TileableTexture.create_for_image(checks)
glEnable(GL_BLEND)
glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
if frames is None:
if framerange is not None:
start = framerange[0]
stop = framerange[1]
if stop == -1:
stop = len(npmovie.uframes)
frames = np.arange(start, stop)
else:
frames = np.arange(1, len(npmovie.uframes))
for f in frames:
uframe = npmovie.uframes[f]
if uframe.uimg is not None:
try:
if mode == "uimg":
arr = uframe.uimg
if mode == "legs":
arr = uframe.legs
if mode == "body":
arr = uframe.body
if mode == "wings":
arr = uframe.wings
if mode == "absdiff":
arr = uframe.absdiff
if mode == "wingimg2":
arr = uframe.wingimg2
if mode == "wingimgR":
arr = uframe.wingimgR
if mode == "wingimgL":
arr = uframe.wingimgL
if mode == "wingimg":
arr = uframe.wingimg
if mode == "flysegs":
arr = uframe.flysegs
if mode == "diffthresh":
arr = uframe.diffthresh
if mode == "full":
arr = npmovie.background
arr[uframe.indices[0] : uframe.indices[1], uframe.indices[2] : uframe.indices[3]] = uframe.uimg
except:
arr = None
if arr is not None:
if arr.dtype != np.uint8:
"converting"
arr = np.array(arr, dtype=np.uint8) * 255
if magnify != 1:
newshape = [arr.shape[0] * magnify, arr.shape[1] * magnify]
arr = nim.rebin(arr, newshape)
try:
aii.view_new_array(arr)
except: # size changed
w.width = arr.shape[1]
w.height = arr.shape[0]
aii = ArrayInterfaceImage(arr)
img = aii.texture
w.dispatch_events()
background.blit_tiled(0, 0, 0, w.width, w.height)
img.blit(0, 0, 0)
# add some overlays:
r = arr.shape[0] / 2.0
body_axis = npmovie.kalmanobj.long_axis[f]
if 0:
wing_axis = npmovie.kalmanobj.wingaxisR[f]
wing_center = npmovie.kalmanobj.wingcenterR[f]
# print wing_center, wing_axis
if mode == "wingimgR":
print f, uframe.wingimgR.sum()
pyglet.graphics.draw(
2,
pyglet.gl.GL_LINES,
(
"v2i",
(
int(r),
int(r),
int(body_axis[1] * 10 * magnify) + int(r),
int(body_axis[0] * 10 * magnify) + int(r),
),
),
)
try:
pyglet.graphics.draw(
2,
pyglet.gl.GL_LINES,
("v2i", (int(r), int(r), int(wing_center[1] * magnify), int(wing_center[0] * magnify))),
)
pyglet.graphics.draw(
1,
pyglet.gl.GL_POINTS,
("v2i", (int(wing_center[1] * magnify), int(wing_center[0] * magnify))),
)
pass
except:
pass
w.flip()
time.sleep(delay) # slow down the playback
w.close()
