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
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
|
#! /usr/bin/env python
import sys, os, gzip
from socket import *
from select import select
import cStringIO, struct, zlib
import time
sys.path.insert(0, os.pardir)
from common.msgstruct import *
from common import hostchooser
import modes
from modes import KeyPressed, KeyReleased
#import psyco; psyco.full()
SOURCEDIR = os.pardir
def loadpixmap(dpy, data, colorkey=None):
f = cStringIO.StringIO(data)
sig = f.readline().strip()
assert sig == "P6"
while 1:
line = f.readline().strip()
if not line.startswith('#'):
break
wh = line.split()
w, h = map(int, wh)
sig = f.readline().strip()
assert sig == "255"
data = f.read()
f.close()
if colorkey is None:
colorkey = -1
elif colorkey < 0:
r, g, b = struct.unpack("BBB", self.data[:3])
colorkey = b | (g<<8) | (r<<16)
return dpy.pixmap(w, h, data, colorkey)
class Icon:
def __init__(self, bitmap, (x, y, w, h), alpha):
self.rect = x, y, w, h
self.size = w, h
self.bitmap = bitmap
self.alpha = alpha
class Playback:
gameident = 'Playback'
def __init__(self, filename, mode=('x', 'off', {})):
f = gzip.open(filename, 'rb')
self.screenmode = mode
self.width = None
self.deffiles = {}
self.defbitmaps = {}
self.deficons = {}
self.icons = {}
self.frames = []
inbuf = ''
while 1:
values, inbuf = decodemessage(inbuf)
if not values:
# incomplete message
data = f.read(8192)
if not data:
break
inbuf += data
else:
#print values[0],
fn = Playback.MESSAGES.get(values[0], self.msg_unknown)
fn(self, *values[1:])
print '%d frames in file.' % len(self.frames)
f.close()
assert self.width, "no playfield definition found in file"
self.dpy = modes.open_dpy(self.screenmode,
self.width, self.height, self.gameident)
self.dpy.clear() # backcolor is ignored
self.sprites = []
self.buildicons()
self.go(0)
def buildicons(self):
bitmaps = {}
for bmpcode, (data, colorkey) in self.defbitmaps.items():
if isinstance(data, str):
data = zlib.decompress(data)
else:
data = self.deffiles[data]
bitmaps[bmpcode] = loadpixmap(self.dpy, data, colorkey)
for icocode, (bmpcode, rect, alpha) in self.deficons.items():
self.icons[icocode] = Icon(bitmaps[bmpcode], rect, alpha)
def go(self, n):
self.n = n
self.update_sprites(self.frames[n])
self.dpy.flip()
def save(self, filename=None):
"shm only!"
w, h, data, reserved = self.dpy.getppm((0, 0, self.width, self.height))
f = open(filename or ('frame%d.ppm' % self.n), 'wb')
print >> f, 'P6'
print >> f, w, h
print >> f, 255
for i in range(0, len(data), 4):
f.write(data[i+2]+data[i+1]+data[i])
f.close()
def update_sprites(self, udpdata):
sprites = self.sprites
unpack = struct.unpack
base = 0
for j in range(len(sprites)):
if sprites[j][0] != udpdata[base:base+6]:
removes = sprites[j:]
del sprites[j:]
removes.reverse()
eraser = self.dpy.putppm
for reserved, eraseargs in removes:
eraser(*eraseargs)
break
base += 6
try:
overlayer = self.dpy.overlayppm
except AttributeError:
getter = self.dpy.getppm
setter = self.dpy.putppm
#print "%d sprites redrawn" % (len(udpdata)/6-j)
for j in range(base, len(udpdata)-5, 6):
info = udpdata[j:j+6]
x, y, icocode = unpack("!hhh", info[:6])
try:
ico = self.icons[icocode]
sprites.append((info, (x, y, getter((x, y) + ico.size))))
setter(x, y, ico.bitmap, ico.rect)
except KeyError:
#print "bad ico code", icocode
pass # ignore sprites with bad ico (probably not defined yet)
else:
for j in range(base, len(udpdata)-5, 6):
info = udpdata[j:j+6]
x, y, icocode = unpack("!hhh", info[:6])
try:
ico = self.icons[icocode]
overlay = overlayer(x, y, ico.bitmap, ico.rect, ico.alpha)
sprites.append((info, overlay))
except KeyError:
#print "bad ico code", icocode
pass # ignore sprites with bad ico (probably not defined yet)
def msg_unknown(self, *rest):
pass
def msg_patch_file(self, fileid, position, data, lendata=None, *rest):
try:
s = self.deffiles[fileid]
except KeyError:
s = ''
if len(s) < position:
s += '\x00' * (position-len(s))
s = s[:position] + data + s[position+len(s):]
self.deffiles[fileid] = s
def msg_zpatch_file(self, fileid, position, data, *rest):
data1 = zlib.decompress(data)
self.msg_patch_file(fileid, position, data1, len(data), *rest)
def msg_md5_file(self, fileid, filename, position, length, checksum, *rest):
fn = os.path.join(SOURCEDIR, filename)
f = open(fn, 'rb')
f.seek(position)
data = f.read(length)
f.close()
assert len(data) == length
self.msg_patch_file(fileid, position, data)
def msg_def_playfield(self, width, height, *rest):
self.width, self.height = width, height
def msg_def_icon(self, bmpcode, icocode, x, y, w, h, alpha=255, *rest):
self.deficons[icocode] = bmpcode, (x, y, w, h), alpha
def msg_def_bitmap(self, bmpcode, data, colorkey=None, *rest):
self.defbitmaps[bmpcode] = data, colorkey
def msg_recorded(self, data):
self.frames.append(data)
MESSAGES = {
MSG_PATCH_FILE : msg_patch_file,
MSG_ZPATCH_FILE : msg_zpatch_file,
MSG_MD5_FILE : msg_md5_file,
MSG_DEF_PLAYFIELD: msg_def_playfield,
MSG_DEF_ICON : msg_def_icon,
MSG_DEF_BITMAP : msg_def_bitmap,
MSG_RECORDED : msg_recorded,
}
if __name__ == '__main__' and len(sys.argv) > 1:
p = Playback(sys.argv[1])
|
