roar/mawj/mawj.py

335 lines
15 KiB
Python
Raw Normal View History

2018-07-05 13:16:25 +00:00
#!/usr/bin/env python3
#
# mawj.py -- MiRCART Animation via Waves Generator
# Copyright (c) 2018 Lucio Andrés Illanes Albornoz <lucio@lucioillanes.de>
# This project is licensed under the terms of the MIT license.
#
from OpenGL.GL import *
from PIL import Image
import cv2, math, numpy, os, sys, wx, wx.glcanvas
MiRCARTColours = [
# {{{ MiRCARTColours: mIRC colour number to RGBA map given none of ^[BFV_] (bold, italic, reverse, underline],
[1.00, 1.00, 1.00], # White
[0.00, 0.00, 0.00], # Black
[0.00, 0.00, 0.73], # Blue
[0.00, 0.73, 0.00], # Green
[1.00, 0.33, 0.33], # Light Red
[0.73, 0.00, 0.00], # Red
[0.73, 0.00, 0.73], # Purple
[0.73, 0.73, 0.00], # Yellow
[1.00, 1.00, 0.33], # Light Yellow
[0.33, 1.00, 0.33], # Light Green
[0.00, 0.73, 0.73], # Cyan
[0.33, 1.00, 1.00], # Light Cyan
[0.33, 0.33, 1.00], # Light Blue
[1.00, 0.33, 1.00], # Pink
[0.33, 0.33, 0.33], # Grey
[0.73, 0.73, 0.73], # Light Grey
]
# }}}
class MawjMiRCARTImporter(object):
"""XXX"""
# {{{ _CellState(): Cell state
class _CellState():
CS_NONE = 0x00
CS_BOLD = 0x01
CS_ITALIC = 0x02
CS_UNDERLINE = 0x04
# }}}
# {{{ _ParseState(): Parsing loop state
class _ParseState():
PS_CHAR = 1
PS_COLOUR_DIGIT0 = 2
PS_COLOUR_DIGIT1 = 3
# }}}
# {{{ _flipCellStateBit(self, cellState, bit): XXX
def _flipCellStateBit(self, cellState, bit):
if cellState & bit:
return cellState & ~bit
else:
return cellState | bit
# }}}
# {{{ _parseCharAsColourSpec(self, colourSpec, curColours): XXX
def _parseCharAsColourSpec(self, colourSpec, curColours):
if len(colourSpec) > 0:
colourSpec = colourSpec.split(",")
if len(colourSpec) == 2 \
and len(colourSpec[1]) > 0:
return [int(colourSpec[0] or curColours[0]), \
int(colourSpec[1])]
elif len(colourSpec) == 1 \
or len(colourSpec[1]) == 0:
return [int(colourSpec[0]), curColours[1]]
else:
return [15, 1]
# }}}
# {{{ fromTextFile(self, pathName): XXX
def fromTextFile(self, pathName):
self.inFile = open(pathName, "r")
self.inSize = self.outMap = None;
inCurColourSpec = ""; inCurRow = -1;
inLine = self.inFile.readline()
inSize = [0, 0]; outMap = []; inMaxCols = 0;
while inLine:
inCellState = self._CellState.CS_NONE
inParseState = self._ParseState.PS_CHAR
inCurCol = 0; inMaxCol = len(inLine);
inCurColourDigits = 0; inCurColours = [15, 1]; inCurColourSpec = "";
inCurRow += 1; outMap.append([]); inRowCols = 0; inSize[1] += 1;
while inCurCol < inMaxCol:
inChar = inLine[inCurCol]
if inChar in set("\r\n"): \
inCurCol += 1
elif inParseState == self._ParseState.PS_CHAR:
inCurCol += 1
if inChar == "":
inCellState = self._flipCellStateBit( \
inCellState, self._CellState.CS_BOLD)
elif inChar == "":
inParseState = self._ParseState.PS_COLOUR_DIGIT0
elif inChar == "":
inCellState = self._flipCellStateBit( \
inCellState, self._CellState.CS_ITALIC)
elif inChar == "":
inCellState |= self._CellState.CS_NONE
inCurColours = [15, 1]
elif inChar == "":
inCurColours = [inCurColours[1], inCurColours[0]]
elif inChar == "":
inCellState = self._flipCellStateBit( \
inCellState, self._CellState.CS_UNDERLINE)
else:
inRowCols += 1
outMap[inCurRow].append([*inCurColours, inCellState, inChar])
elif inParseState == self._ParseState.PS_COLOUR_DIGIT0 \
or inParseState == self._ParseState.PS_COLOUR_DIGIT1:
if inChar == "," \
and inParseState == self._ParseState.PS_COLOUR_DIGIT0:
if (inCurCol + 1) < inMaxCol \
and not inLine[inCurCol + 1] in set("0123456789"):
inCurColours = self._parseCharAsColourSpec( \
inCurColourSpec, inCurColours)
inCurColourDigits = 0; inCurColourSpec = "";
inParseState = self._ParseState.PS_CHAR
else:
inCurCol += 1
inCurColourDigits = 0; inCurColourSpec += inChar;
inParseState = self._ParseState.PS_COLOUR_DIGIT1
elif inChar in set("0123456789") \
and inCurColourDigits == 0:
inCurCol += 1
inCurColourDigits += 1; inCurColourSpec += inChar;
elif inChar in set("0123456789") \
and inCurColourDigits == 1 \
and inCurColourSpec[-1] == "0":
inCurCol += 1
inCurColourDigits += 1; inCurColourSpec += inChar;
elif inChar in set("012345") \
and inCurColourDigits == 1 \
and inCurColourSpec[-1] == "1":
inCurCol += 1
inCurColourDigits += 1; inCurColourSpec += inChar;
else:
inCurColours = self._parseCharAsColourSpec( \
inCurColourSpec, inCurColours)
inCurColourDigits = 0; inCurColourSpec = "";
inParseState = self._ParseState.PS_CHAR
inMaxCols = max(inMaxCols, inRowCols)
inLine = self.inFile.readline()
inSize[0] = inMaxCols; self.inSize = inSize; self.outMap = outMap;
self.inFile.close()
# }}}
# {{{ __init__(self, inFile): initialisation method
def __init__(self, inFile):
self.inFile = inFile; self.inSize = self.outMap = None;
self.fromTextFile(inFile)
# }}}
class MawjFrame(wx.Frame):
"""XXX"""
# {{{ __init__(self, parent, size, videoPath=None): initialisation method
def __init__(self, parent, size, videoPath=None):
super().__init__(parent, size=size)
self.panelSkin = wx.Panel(self, wx.ID_ANY)
self.panelGLCanvas = MawjGLCanvasPanel(self.panelSkin, size=[s-128 for s in size], videoPath=videoPath)
self.Bind(wx.EVT_MOUSEWHEEL, self.panelGLCanvas.onMouseWheel)
self.SetFocus(); self.Show(True);
# }}}
class MawjGLCanvasPanel(wx.glcanvas.GLCanvas, wx.Panel):
"""XXX"""
# {{{ drawAll(self, deltaX=0, deltaY=0, deltaZ=0, paintFlag=False): XXX
def drawAll(self, deltaX=0, deltaY=0, deltaZ=0, paintFlag=False):
self.SetCurrent(self.glContext)
if not self.hasGLInit:
self.onGLInit(); self.hasGLInit = True;
if deltaX != 0 or deltaY != 0 or deltaZ != 0 or paintFlag:
self.isDirty = True
if self.isDirty:
glClear(GL_COLOR_BUFFER_BIT|GL_DEPTH_BUFFER_BIT)
glBegin(GL_QUADS)
curPos = [0,0,0]; cubeSize = (0.1,0.2);
for numRow in range(len(self.MiRCART.outMap)):
for numCol in range(len(self.MiRCART.outMap[numRow])):
cubeColour = MiRCARTColours[self.MiRCART.outMap[numRow][numCol][1]]
self.drawIrcCube(colour=cubeColour, pos=curPos, size=cubeSize)
curPos[0] += cubeSize[0]
curPos[0] = 0; curPos[1] -= cubeSize[1];
glEnd()
if deltaX != 0 or deltaY != 0:
w, h = self.GetClientSize(); w = max(w, 1.0); h = max(h, 1.0);
glRotatef(deltaY * (180.0/h), 1.0, 0.0, 0.0)
glRotatef(deltaX * (180.0/w), 0.0, 1.0, 0.0)
if deltaZ != 0:
glTranslatef(0, 0, deltaZ)
self.isDirty = False; self.SwapBuffers(); self.snap();
# }}}
# {{{ drawIrcCube(self, colour=(1.0, 0.0, 0.0), pos=(0,0,0), size=(0.1,0.2)): XXX
def drawIrcCube(self, colour=(1.0, 0.0, 0.0), pos=(0,0,0), size=(0.1,0.2)):
# Top Right, Top Left, Bottom Left, Bottom Right
glColor3f(*colour)
glNormal3f(0.0, 0.0, 1.0)
glVertex3f(pos[0], pos[1], pos[2])
glVertex3f(pos[0]-size[0], pos[1], pos[2])
glVertex3f(pos[0]-size[0], pos[1]-size[1], pos[2])
glVertex3f(pos[0], pos[1]-size[1], pos[2])
# }}}
# {{{ onGLInit(self): XXX
def onGLInit(self):
# <https://www.opengl.org/discussion_boards/showthread.php/125843-default-camera?p=954801&viewfull=1#post954801>
glViewport(0, 0, *self.curSize)
glMatrixMode(GL_PROJECTION)
glLoadIdentity(); glFrustum(-1, 1, -1, 1, 1, 100);
glMatrixMode(GL_MODELVIEW)
glEnable(GL_DEPTH_TEST)
glClearColor(0, 0, 0, 1); glClearDepth(1);
glTranslatef(-5.0, 3.0, -5)
# <https://www.khronos.org/opengl/wiki/How_lighting_works#Good_Settings.>
glLight(GL_LIGHT0, GL_AMBIENT, (0, 0, 0, 1))
glLight(GL_LIGHT0, GL_DIFFUSE, (1, 1, 1, 1))
glLight(GL_LIGHT0, GL_POSITION, (1, 1, 0+2, 0))
glLight(GL_LIGHT0, GL_SPECULAR, (1, 1, 1, 1))
glLightModelfv(GL_LIGHT_MODEL_AMBIENT, (0.2, 0.2, 0.2, 1))
glEnable(GL_LIGHTING); glEnable(GL_LIGHT0);
glEnable(GL_COLOR_MATERIAL)
glColorMaterial(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE)
if self.videoPath != None:
width, height = self.GetClientSize()
fourcc = cv2.VideoWriter_fourcc("X", "V", "I", "D")
self.videoWriter = cv2.VideoWriter(self.videoPath, fourcc, 25, (width, height), True)
# }}}
# {{{ onMouseDown(self, event): XXX
def onMouseDown(self, event):
self.CaptureMouse(); self.mouseXlast, self.mouseYlast = event.GetPosition();
# }}}
# {{{ onMouseMotion(self, event): XXX
def onMouseMotion(self, event):
if event.Dragging():
if event.LeftIsDown():
mouseX, mouseY = event.GetPosition()
self.drawAll(deltaX=mouseX-self.mouseXlast, deltaY=mouseY-self.mouseYlast)
self.mouseXlast, self.mouseYlast = mouseX, mouseY
elif event.RightIsDown():
mouseX, mouseY = event.GetPosition()
self.swirlAll(2, 30.0)
self.drawAll(paintFlag=True)
self.mouseXlast, self.mouseYlast = mouseX, mouseY
# }}}
# {{{ onMouseWheel(self, event): XXX
def onMouseWheel(self, event):
wheelRotation = event.GetWheelRotation()
if wheelRotation < 0:
self.drawAll(deltaZ=-1)
else:
self.drawAll(deltaZ=1)
# }}}
# {{{ onMouseUp(self, event): XXX
def onMouseUp(self, event):
self.ReleaseMouse(); self.mouseXlast = self.mouseYlast = 0;
# }}}
# {{{ onPaint(self, event): XXX
def onPaint(self, event):
eventDc = wx.PaintDC(self)
eventUpdates = wx.RegionIterator(self.GetUpdateRegion())
paintFlag = True if eventUpdates.HaveRects() else False
self.drawAll(paintFlag=paintFlag)
# }}}
# {{{ snap(self): XXX
def snap(self):
width, height = self.GetClientSize()
if sys.byteorder == "little":
screenshot = glReadPixels(0, 0, width, height, GL_BGR, GL_UNSIGNED_BYTE)
else:
screenshot = glReadPixels(0, 0, width, height, GL_RGB, GL_UNSIGNED_BYTE)
screenshot = numpy.flipud(numpy.frombuffer(screenshot, numpy.uint8).reshape((height, width, 3)))
self.videoWriter.write(screenshot)
# }}}
# {{{ swirlAll(self, radiusDiv=4, step=30.0): XXX
def swirlAll(self, radiusDiv=4, step=30.0):
# <http://geekofficedog.blogspot.de/2013/04/hello-swirl-swirl-effect-tutorial-in.html>
width = len(self.MiRCART.outMap[0]); height = len(self.MiRCART.outMap);
centerX = math.floor(width / 2); centerY = math.floor(height / 2);
size = width if width < height else height
radius = math.floor(size / radiusDiv)
newMap = self.MiRCART.outMap.copy()
for y in range(-radius, radius):
for x in range(-radius, radius):
# Transform the pixel cartesian coordinates (x, y) to polar coordinates (r, alpha)
# Remember that the angle alpha is in radians, transform it to degrees
r = math.sqrt(pow(x, 2) + pow(y, 2)); alpha = math.atan2(y, x);
degrees = (alpha * 180.0) / math.pi
# Shift the angle by a constant delta
# Note the '-' sign was changed by '+' the inverted function
degrees -= step
# Transform back from polar coordinates to cartesian
alpha = (degrees * math.pi) / 180.0
newY = math.floor(r * math.sin(alpha)); newX = math.floor(r * math.cos(alpha));
try:
# Calculate the pixel array position
# Get the new pixel location
srcPosY = y + centerY; srcPosX = x + centerX;
destPosY = newY + centerY; destPosX = newX + centerX;
newMap[destPosY][destPosX] = self.MiRCART.outMap[srcPosY][srcPosX]
except IndexError:
continue
self.MiRCART.outMap = newMap
# }}}
# {{{ __init__(self, parent, size, defaultPos=(24,24), videoPath=None): initialisation method
def __init__(self, parent, size, defaultPos=(24,24), videoPath=None):
super().__init__(parent, pos=defaultPos, size=size)
self.curPos = list(defaultPos); self.curSize = list(size);
self.glContext = wx.glcanvas.GLContext(self);
self.hasGLInit = False; self.isDirty = True;
self.mouseXlast = self.mouseYlast = 0;
self.Bind(wx.EVT_LEFT_DOWN, self.onMouseDown)
self.Bind(wx.EVT_LEFT_UP, self.onMouseUp)
self.Bind(wx.EVT_MOTION, self.onMouseMotion)
self.Bind(wx.EVT_PAINT, self.onPaint)
self.MiRCART = MawjMiRCARTImporter("puke.txt"); self.videoPath = videoPath;
# }}}
#
# Entry point
def main(*argv):
wxApp = wx.App(False)
appFrame = MawjFrame(None, size=(1152,864), videoPath=os.path.join("F:\\", "mawj.avi"))
wxApp.MainLoop()
if __name__ == "__main__":
main(*sys.argv)
# vim:expandtab foldmethod=marker sw=4 ts=4 tw=120