Zombie MythPyWii Update
Friday, September 12th, 2008I can’t sleep, so I thought it would be wise to give you an update on my previous post about MythPyWii, despite feeling a bit like the living dead must feel. (And having to frequently remove Artie from the keyboard, bless her.)
Unfortunately I have not been able to get a video of MythPyWii yet because – minor technical hitch – I can’t find a VGA cable to connect my PC to my TV! (Wanted to show off the TV at the same time, it’s a good excuse!) I have, however, set up completely diskless booting on the old PC – and it’s working great! Took me a little bit of hacking (2 hours) to get it working well with my setup mind, not bad for my first ever diskless box! Hopefully it gives me a chance to outline every required piece of software on mythbuntu too. 
I’ve also been hacking away at MythPyWii, it now has the following improvements:
- Button repeats are sensible (if you hold “up”, it will simulate pressing up, pause 0.5 seconds, and then repeat every 0.15 seconds)
- Manually repeating a button works better (previously we ignored any button press 0.5s after the previous one. Now, instead, we reset the delay every time a different key is pressed/released on the Wiimote, so you can tap up as fast as you want (and as fast as mythfrontend can handle))
- Slowmo/doubletime – I’ve added time stretching. To activate, hold the wiimote flat, hold B and A together, and then twist, as with fast forwarding. When you let go, myth will be left in slowmo, if this is not what you want, adjust timestretch to 1x using the same method! Enjoy!
- Fractionally tidier code. This is a mess already (as you would expect for a first attempt at a programming language) – if anyone out there rocks at Python, any tips would be appreciated!
Here’s the latest code:
#!/usr/bin/env python
"""
Copyright (c) 2008, Benjie Gillam
All rights reserved.
Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
* Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
* Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.
* Neither the name of MythPyWii nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
"""
# By Benjie Gillam http://www.benjiegillam.com/mythpywii/
import cwiid, time, StringIO, sys, asyncore, socket
from math import log, floor, atan, sqrt, cos, exp
# Note to self - list of good documentation:
# cwiid: http://flx.proyectoanonimo.com/proyectos/cwiid/
# myth telnet: http://www.mythtv.org/wiki/index.php/Telnet_socket
def do_scale(input, max, divisor=None):
if divisor is None: divisor = max
if (input > 1): input = 1
if (input < -1): input = -1
input = int(input * divisor)
if input>max: input = max
elif input < -max: input = -max
return input
class MythSocket(asyncore.dispatcher):
firstData = True
data = ""
prompt="\n# "
owner = None
buffer = ""
callbacks = []
oktosend = True
def __init__(self, owner):
self.owner = owner
asyncore.dispatcher.__init__(self)
self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
self.connect(("localhost", 6546))
def handle_connect(self):
print "Connected"
def handle_close(self):
print "Closed"
self.close()
def handle_read(self):
self.data = self.data + self.recv(8192)
while len(self.data)>0:
a = self.data.find(self.prompt)
if a>-1:
self.oktosend = True
result = self.data[:a]
self.data = self.data[a+len(self.prompt):]
if not self.firstData:
print "<<<", result
cb = self.callbacks.pop(0)
if cb:
cb(result)
else:
print "Logged in to MythFrontend"
self.firstData = False
else:
break;
def writable(self):
return (self.oktosend) and (len(self.buffer) > 0) and (self.buffer.find("\n") > 0)
def handle_write(self):
a = self.buffer.find("\n")
sent = self.send(self.buffer[:a+1])
print ">>>", self.buffer[:sent-1]
self.buffer = self.buffer[sent:]
self.oktosend = False
def cmd(self, data, cb = None):
self.buffer += data + "\n"
self.callbacks.append(cb)
def raw(self, data):
cmds = data.split("\n")
for cmd in cmds:
if len(cmd.strip())>0:
self.cmd(cmd)
def ok(self):
return len(self.callbacks) == len(self.buffer) == 0
class WiiMyth:
wii_calibration = False
wm = None
ms = None
wii_calibration = None
#Initialize variables
reportvals = {"accel":cwiid.RPT_ACC, "button":cwiid.RPT_BTN, "ext":cwiid.RPT_EXT, "status":cwiid.RPT_STATUS}
report={"accel":True, "button":True}
state = {"acc":[0, 0, 1]}
lasttime = 0.0
laststate = {}
responsiveness = 0.15
firstPress = True
firstPressDelay = 0.5
maxButtons = 0
#wii_rel = lambda v, axis: float(v - self.wii_calibration[0][axis]) / (
# self.wii_calibration[1][axis] - self.wii_calibration[0][axis])
def wii_rel(self, v, axis):
return float(v - self.wii_calibration[0][axis]) / (
self.wii_calibration[1][axis] - self.wii_calibration[0][axis])
def wmconnect(self):
print "Please press 1&2 on the wiimote..."
try:
self.wm = cwiid.Wiimote()
except:
self.wm = None
if self.ms is not None:
self.ms.close()
self.ms = None
return None
self.ms = MythSocket(self)
print "Connected..."
self.wm.rumble=1
time.sleep(.2)
self.wm.rumble=0
# Wiimote calibration data (cache this)
self.wii_calibration = self.wm.get_acc_cal(cwiid.EXT_NONE)
return self.wm
def wmcb(self, messages):
state = self.state
for message in messages:
if message[0] == cwiid.MESG_BTN:
state["buttons"] = message[1]
#elif message[0] == cwiid.MESG_STATUS:
# print "\nStatus: ", message[1]
elif message[0] == cwiid.MESG_ERROR:
if message[1] == cwiid.ERROR_DISCONNECT:
self.wm = None
if self.ms is not None:
self.ms.close()
self.ms = None
continue
else:
print "ERROR: ", message[1]
elif message[0] == cwiid.MESG_ACC:
state["acc"] = message[1]
else:
print "Unknown message!", message
laststate = self.laststate
if ('buttons' in laststate) and (laststate['buttons'] <> state['buttons']):
if state['buttons'] == 0:
self.maxButtons = 0
elif state['buttons'] < self.maxButtons:
continue
else:
self.maxButtons = state['buttons']
self.lasttime = 0
self.firstPress = True
if laststate['buttons'] == cwiid.BTN_B and not state['buttons'] == cwiid.BTN_B:
del state['BTN_B']
self.ms.cmd('play speed normal')
if (laststate['buttons'] & cwiid.BTN_A and laststate['buttons'] & cwiid.BTN_B) and not (state['buttons'] & cwiid.BTN_A and state['buttons'] & cwiid.BTN_B):
del state['BTN_AB']
#self.ms.cmd('play speed normal')
if self.ms.ok() and (self.wm is not None) and (state["buttons"] > 0) and (time.time() > self.lasttime+self.responsiveness):
self.lasttime = time.time()
wasFirstPress = False
if self.firstPress:
wasFirstPress = True
self.lasttime = self.lasttime + self.firstPressDelay
self.firstPress = False
# Stuff that doesn't need roll/etc calculations
if state["buttons"] == cwiid.BTN_HOME:
self.ms.cmd('key escape')
if state["buttons"] == cwiid.BTN_A:
self.ms.cmd('key enter')
if state["buttons"] == cwiid.BTN_MINUS:
self.ms.cmd('key d')
if state["buttons"] == cwiid.BTN_UP:
self.ms.cmd('key up')
if state["buttons"] == cwiid.BTN_DOWN:
self.ms.cmd('key down')
if state["buttons"] == cwiid.BTN_LEFT:
self.ms.cmd('key left')
if state["buttons"] == cwiid.BTN_RIGHT:
self.ms.cmd('key right')
if state["buttons"] == cwiid.BTN_PLUS:
self.ms.cmd('key p')
if state["buttons"] == cwiid.BTN_1:
self.ms.cmd('key i')
if state["buttons"] == cwiid.BTN_2:
self.ms.cmd('key m')
# Do we need to calculate roll, etc?
# Currently only BTN_B needs this.
calcAcc = state["buttons"] & cwiid.BTN_B
if calcAcc:
# Calculate the roll/etc.
X = self.wii_rel(state["acc"][cwiid.X], cwiid.X)
Y = self.wii_rel(state["acc"][cwiid.Y], cwiid.Y)
Z = self.wii_rel(state["acc"][cwiid.Z], cwiid.Z)
if (Z==0): Z=0.00000001 # Hackishly prevents divide by zeros
roll = atan(X/Z)
if (Z <= 0.0):
if (X>0): roll += 3.14159
else: roll -= 3.14159
pitch = atan(Y/Z*cos(roll))
#print "X: %f, Y: %f, Z: %f; R: %f, P: %f; B: %d \r" % (X, Y, Z, roll, pitch, state["buttons"]),
sys.stdout.flush()
if state["buttons"] & cwiid.BTN_B and state["buttons"] & cwiid.BTN_LEFT:
self.ms.cmd('play seek beginning')
if state["buttons"] & cwiid.BTN_B and state["buttons"] & cwiid.BTN_A:
speed=do_scale(roll/3.14159, 20, 25)
if (speed<-10): speed = -10
state['BTN_AB'] = speed
cmd = ""
# on first press, press a,
# after then use the diff to press left/right
if not 'BTN_AB' in laststate:
# # query location
# Playback Recorded 00:04:20 of 00:25:31 1x 30210 2008-09-10T09:18:00 6523 /video/30210_20080910091800.mpg 25
cmd += "play speed normal\nkey a\n"#"play speed normal\n"
else:
speed = speed - laststate['BTN_AB']
if speed > 0:
cmd += abs(speed)*"key right\n"
elif speed < 0:
cmd += abs(speed)*"key left\n"
if speed <> 0:
self.wm.rumble=1
time.sleep(.05)
self.wm.rumble=0
if cmd is not None and cmd:
self.ms.raw(cmd)
if state["buttons"] == cwiid.BTN_B:
speed=do_scale(roll/3.14159, 8, 13)
state['BTN_B'] = speed
if not 'BTN_B' in laststate:
# # query location
# Playback Recorded 00:04:20 of 00:25:31 1x 30210 2008-09-10T09:18:00 6523 /video/30210_20080910091800.mpg 25
cmd = ""#"play speed normal\n"
if speed > 0:
cmd += "key .\n"
elif speed < 0:
cmd += "key ,\n"
if speed <> 0:
cmd += "key "+str(abs(speed)-1)+"\n"
#print cmd
elif laststate['BTN_B']<>speed:
self.wm.rumble=1
time.sleep(.05)
self.wm.rumble=0
if speed == 0:
cmd = "play speed normal"
elif ((laststate['BTN_B'] > 0) and (speed > 0)) or ((laststate['BTN_B'] < 0) and (speed < 0)):
cmd = "key "+str(abs(speed)-1)+"\n"
elif speed>0:
cmd = "key .\nkey "+str(abs(speed)-1)+"\n"
else:
cmd = "key ,\nkey "+str(abs(speed)-1)+"\n"
else:
cmd = None
if cmd is not None:
self.ms.raw(cmd)
self.laststate = state.copy() #NOTE TO SELF: REMEMBER .copy() !!!
def mythLocation(self, data):
#Playback Recorded 00:00:49 of 00:25:31 1x 30210 2008-09-10T09:18:00 1243 /video/30210_20080910091800.mpg 25
#PlaybackBox
temp = data.split(" ")
output = {}
output['mode'] = temp[0]
if output['mode'] == "Playback":
output['position'] = temp[2]
output['max'] = temp[4]
return output
def main(self):
while True:
if self.wm is None:
#Connect wiimote
self.wmconnect()
if self.wm:
#Tell Wiimote to display rock sign
self.wm.led = cwiid.LED1_ON | cwiid.LED4_ON
self.wm.rpt_mode = sum(self.reportvals[a] for a in self.report if self.report[a])
self.wm.enable(cwiid.FLAG_MESG_IFC | cwiid.FLAG_REPEAT_BTN)
self.wm.mesg_callback = self.wmcb
asyncore.loop(timeout=0, count=1)
time.sleep(0.05)
print "Exited Safely"
# Instantiate our class, and start.
inst = WiiMyth()
inst.main()
Download this code: /code/myth_py_wii.r14.py
Hopefully tomorrow I can have it up and running, get a video up, and decent install instructions. Hopefully.
![Reblog this post [with Zemanta]](http://img.zemanta.com/reblog_c.png?x-id=a8dcc3d0-8dce-4fdc-a906-fe3028caae05)
![Reblog this post [with Zemanta]](http://img.zemanta.com/reblog_c.png?x-id=3c6b13d5-6170-4311-9dbd-1569799d4bb7)
![Reblog this post [with Zemanta]](http://img.zemanta.com/reblog_c.png?x-id=6438cd77-346d-40d7-85aa-5d71dcb0573f)
