mirror of https://github.com/lalbornoz/roar.git
69 lines
2.9 KiB
Python
69 lines
2.9 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# ENNTool -- mIRC art animation tool (for EFnet #MiRCART) (WIP)
|
|
# Copyright (c) 2018 Lucio Andrés Illanes Albornoz <lucio@lucioillanes.de>
|
|
# This project is licensed under the terms of the MIT license.
|
|
#
|
|
# References:
|
|
# Tue, 03 Jul 2018 20:35:52 +0200 [1] <https://stackoverflow.com/questions/23930671/how-to-create-n-dim-numpy-array-from-a-pointer>
|
|
# Wed, 04 Jul 2018 10:02:22 +0200 [2] <http://www.songho.ca/opengl/gl_pbo.html>
|
|
#
|
|
|
|
from OpenGL.GL import *
|
|
import ctypes, cv2, numpy
|
|
|
|
class ENNToolGLVideoWriter(object):
|
|
"""XXX"""
|
|
|
|
# {{{ _copyFrames(self): XXX
|
|
def _copyFrames(self):
|
|
for numPbo in range(self.pboCount):
|
|
glBindBuffer(GL_PIXEL_PACK_BUFFER, self.pboList[numPbo])
|
|
frameBufferPtr = glMapBuffer(GL_PIXEL_PACK_BUFFER, GL_READ_ONLY)
|
|
frameBufferPtr = ctypes.cast(frameBufferPtr, ctypes.POINTER(ctypes.c_ubyte))
|
|
frameBuffer = numpy.ctypeslib.as_array(frameBufferPtr, shape=(self.videoSize[1], self.videoSize[0], 3))
|
|
frameBuffer = numpy.flipud(frameBuffer)
|
|
self.videoWriter.write(frameBuffer)
|
|
glUnmapBuffer(GL_PIXEL_PACK_BUFFER)
|
|
# }}}
|
|
# {{{ _initCv2(self, videoFps, videoPath, videoSize): XXX
|
|
def _initCv2(self, videoFps, videoPath, videoSize):
|
|
fourcc = cv2.VideoWriter_fourcc(*"avc1")
|
|
videoWriter = cv2.VideoWriter(videoPath, fourcc, videoFps, tuple(videoSize), True)
|
|
return videoWriter
|
|
# }}}
|
|
# {{{ _initPbos(self, pboCount, videoSize): XXX
|
|
def _initPbos(self, pboCount, videoSize):
|
|
pboBufs, pboCur, pboList = [None] * pboCount, 0, [None] * pboCount
|
|
for numPbo in range(pboCount):
|
|
pboList[numPbo] = glGenBuffers(1)
|
|
glBindBuffer(GL_PIXEL_PACK_BUFFER, pboList[numPbo])
|
|
glBufferData(GL_PIXEL_PACK_BUFFER,
|
|
videoSize[0] * videoSize[1] * 3, None, GL_STREAM_READ)
|
|
glBindBuffer(GL_PIXEL_PACK_BUFFER, 0)
|
|
return pboBufs, pboCount, pboCur, pboList
|
|
# }}}
|
|
# {{{ saveFrame(self): XXX
|
|
def saveFrame(self):
|
|
glBindBuffer(GL_PIXEL_PACK_BUFFER, self.pboList[self.pboCur])
|
|
glReadPixels(0, 0, self.videoSize[0], self.videoSize[1], GL_BGR, GL_UNSIGNED_BYTE, ctypes.c_void_p(0))
|
|
self.pboCur += 1
|
|
if self.pboCur >= self.pboCount:
|
|
self._copyFrames(); self.pboCur = 0;
|
|
glBindBuffer(GL_PIXEL_PACK_BUFFER, 0)
|
|
# }}}
|
|
# {{{ saveVideo(self): XXX
|
|
def saveVideo(self):
|
|
return
|
|
# }}}
|
|
# {{{ __init__(self, videoPath, videoSize, videoFps=25): XXX
|
|
def __init__(self, videoPath, videoSize, videoFps=25):
|
|
videoWriter = self._initCv2(videoFps, videoPath, videoSize)
|
|
self.pboBufs, self.pboCount, \
|
|
self.pboCur, self.pboList = self._initPbos(videoFps, videoSize)
|
|
self.videoFps, self.videoPath, \
|
|
self.videoSize, self.videoWriter = videoFps, videoPath, videoSize, videoWriter
|
|
# }}}
|
|
|
|
# vim:expandtab foldmethod=marker sw=4 ts=4 tw=120
|