1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
|
#!/usr/bin/env python3
"""This module contains functions for visualizing QLC+ functions as images.
For audio support,
"""
import subprocess as subp
import tempfile
import wave
import os
import warnings
from shutil import which
from PIL import Image, ImageDraw, ImageFont
from .workspace import Show, Function, SHOW
BLUE = (0,0,255)
RED = (255,0,0)
WHITE = (255,255,255)
GRAY = (127,127,127)
GREEN = (0, 255, 0)
BLACK = (0,0,0)
YELLOW = (255,255,0)
BAMBER = (255,127,0)
CBLUE = (127,127,255)
NOGEL = (255,255,127)
PURPLE = (255,0,255)
def get_wave(fname):
"""Load an audio file into a wave."""
with tempfile.NamedTemporaryFile(delete=False,suffix=".wav") as f:
tname = f.name
subp.call(["ffmpeg", "-i", fname, "-acodec", "pcm_s16le", "-ar", "44100", "-y", tname], stdout=subp.DEVNULL, stderr=subp.DEVNULL)
return tname,wave.open(tname,mode="rb")
def chunks(l, n):
"""Split the iterable l into chunks of at most n elements."""
for i in range(0, len(l), n):
yield l[i:i+n]
def s16letoi(lsb, msb):
"""Convert S16LE (2's complement) to an integer."""
val = (msb << 8) + lsb
if msb&0b1000000 > 0: ## Negative
return (-1)*((1 << 16) - val)
return val
def render_image(f:Function):
"""Render a QLC+ function.
This function delegates appropriately, based on the type of function.
"""
if f.type == SHOW:
return render_image_show(f)
else:
raise ValueError("Don't know how to render %s" % f.type)
def render_image_show(s:Show):
"""Render a show to a PIL image."""
vals, acues = s.render_all()
vals = list(vals)
values = {}
for t,cv in vals:
for c, v in cv:
if c not in values:
values[c] = 0
cheight = 270
offset = 200
numheight = 100
width = int(vals[-1][0]/10 + 1)+offset
height = (len(values)+1)*cheight+numheight
ah = lambda y: height-1-y
font = ImageFont.truetype(font="arial.ttf", size=int(offset*0.8))
nfont = ImageFont.truetype(font="arial.ttf", size=int(numheight*0.8))
colormap = {7:RED, 8:RED, 9:RED, 19: BAMBER, 12: CBLUE, 24: NOGEL, 4: BLUE, 5: BLUE, 6: BLUE,
25: GREEN, 26: GREEN, 27: GREEN, 13: YELLOW, 14: YELLOW, 15: YELLOW, 16: YELLOW,
18: PURPLE, 22: PURPLE, 21: PURPLE}
order = [7,8,9,25,26,27,4,5,6,12,24,19,13,14,15,16,18,22,21]
im = Image.new("RGB", (width, height), color=WHITE)
draw = ImageDraw.Draw(im)
channels = [i for i in order if i in values]+["A"]
for c in range(len(values)+1):
draw.line(((0, ah(c*cheight)), (width-1,ah(c*cheight))), fill=GRAY, width=1)
draw.text((0,ah(c*cheight + 200)), text=str(channels[c]), font=font, fill=GRAY)
channels = channels[:-1]
draw.line(((offset-1,0),(offset-1,height-1)), fill=GRAY, width=1)
draw.line(((0,numheight-1), (width-1, numheight-1)), fill=GRAY, width=3)
atime = []
if which("ffmpeg") is None:
warnings.warn("Cannot find ffmpeg, audio is disabled", RuntimeWarning)
else:
for a in acues:
stime = a[0]
skipperiods = stime//10 + (stime%10 > 0)
## Add 0 entries as necessary to the audio list
atime += [0]*max(0,skipperiods-len(atime))
tname, wave = get_wave(a[2])
nchannels, sampwidth, framerate, nframes, *_ = wave.getparams()
if sampwidth != 2:
raise ValueError("Only 16-bit wave is supported")
skip = framerate//100
if stime%10 > 0:
wave.readframes((stime%10)*(skip//10))
for n,t in enumerate(range(0,nframes, skip), skipperiods):
asum = 0
count = 0
for i in range(skip//10):
for c in chunks(wave.readframes(10)[:nchannels*sampwidth],sampwidth):
asum += abs(s16letoi(*c))
count += 1
if skip%10 > 0:
wave.readframes(skip%10)
if not count:
break
if len(atime) > n:
atime[n] = min(atime[n]+asum/count, 32767)
else:
atime.append(asum/count)
wave.close()
os.remove(tname)
break
t = 0
while vals:
#old = {c: v for c,v in values.items()}
while vals and vals[0][0] < t:
_, cv = vals.pop(0)
values.update(cv)
curx = t//10 + offset
if t % 500 == 0:
draw.line(((curx+(t%2000==0),0), (curx+(t%2000==0), height-1)), fill=GRAY, width = 1 + 3*(t % 2000 == 0))
if t % 2000 == 0:
draw.text((curx+1, 5), text=str(t//1000), font=nfont, fill=GRAY)
for n,c in enumerate(channels):
v = values[c]
cury = ah(n*cheight+1+v)
fill = colormap[c]
draw.point((curx, cury), fill=fill)
draw.line(((curx,ah(n*cheight+1)),(curx, cury)), fill=fill, width=1)
if atime:
aval = abs(int(atime.pop(0)/256))
draw.line([(curx, ah(len(values)*cheight+128-aval)),(curx, ah(len(values)*cheight+128+aval))], fill=BLACK, width=1)
t += 10
return im
if __name__ == "__main__":
import argparse
from .workspace import Workspace
parser = argparse.ArgumentParser(description="Render a show to an image.")
parser.add_argument("workspace", type=str, help="Workspace file")
parser.add_argument("show_id", type=int, help="Show ID")
parser.add_argument("output", type=str, help="Output file")
args = parser.parse_args()
print("Loading workspace...")
w = Workspace.load(args.workspace)
s = w.functions[args.show_id]
print("Rendering show \"%s\"..." % s.name)
render_image(s).save(args.output)
print("Done!")
|
