morse funcs

morse.py

#! /usr/bin/env python

keylist = " 5H4S!V3I?F@U$^2E L R - A P W J .6B=D/X7N C K Y T Z G Q M8: O9)0"

def decode(code):
    o = i = len(keylist) >> 1
    for c in code:
        o >>= 1
        i += o if c == '-' else -o
    return keylist[i]

print decode("****")
print decode("**")
print decode("")

# Take this a step further and recognize that each bit in the index to a given
# glyph can be treated as a branch decision in the tree:
def encode(text):
    output = ""
    for c in text.upper():
        if c == ' ':
            continue
        i = keylist.find(c)
        code = ""
        for b in range(5)[::-1]:
            if decode(code) == c:
                break
            code += "-" if i>>1 & (1<<b) else "*"
        output += " " + code
    return output.strip()

print encode("hi tgunz sos plz")