Mac/Demo/sound/morselib.py

   1 """Translate text strings to Morse code"""
   2
   3 FRAMERATE = 22050
   4 SAMPWIDTH = 2
   5
   6 BASEFREQ = 441
   7 OCTAVE = 2
   8
   9 DOT = 30
  10 DAH = 80
  11
  12 morsetab = {
  13         'a': '.-',
  14         'b': '-...',
  15         'c': '-.-.',
  16         'd': '-..',
  17         'e': '.',
  18         'f': '..-.',
  19         'g': '--.',
  20         'h': '....',
  21         'i': '..',
  22         'j': '.---',
  23         'k': '-.-',
  24         'l': '.-..',
  25         'm': '--',
  26         'n': '-.',
  27         'o': '---',
  28         'p': '.--.',
  29         'q': '--.-',
  30         'r': '.-.',
  31         's': '...',
  32         't': '-',
  33         'u': '..-',
  34         'v': '...-',
  35         'w': '.--',
  36         'x': '-..-',
  37         'y': '-.--',
  38         'z': '--..',
  39         '0': '-----',
  40         '1': '.----',
  41         '2': '..---',
  42         '3': '...--',
  43         '4': '....-',
  44         '5': '.....',
  45         '6': '-....',
  46         '7': '--...',
  47         '8': '---..',
  48         '9': '----.',
  49         ',': '--..--',
  50         '.': '.-.-.-',
  51         '?': '..--..',
  52         ';': '-.-.-.',
  53         ':': '---...',
  54         "'": '.----.',
  55         '-': '-....-',
  56         '/': '-..-.',
  57         '(': '-.--.-',
  58         ')': '-.--.-', # XXX same as code for '(' ???
  59         '_': '..--.-',
  60         ' ': ' '
  61 }
  62
  63 def morsecode(s):
  64         from string import lower
  65         m = ''
  66         for c in s:
  67                 c = lower(c)
  68                 if morsetab.has_key(c):
  69                         c = morsetab[c] + ' '
  70                 else:
  71                         c = '? '
  72                 m = m + c
  73         return m
  74
  75
  76 class BaseMorse:
  77         "base class for morse transmissions"
  78
  79         def __init__(self):
  80                 "constructor"
  81                 self.dots = DOT
  82                 self.dahs = DAH
  83
  84         def noise(self, duration):
  85                 "beep for given duration"
  86                 pass
  87
  88         def pause(self, duration):
  89                 "pause for given duration"
  90                 pass
  91
  92         def dot(self):
  93                 "short beep"
  94                 self.noise(self.dots)
  95
  96         def dah(self):
  97                 "long beep"
  98                 self.noise(self.dahs)
  99
 100         def pdot(self):
 101                 "pause as long as a dot"
 102                 self.pause(self.dots)
 103
 104         def pdah(self):
 105                 "pause as long as a dah"
 106                 self.pause(self.dahs)
 107
 108         def sendmorse(self, s):
 109                 for c in s:
 110                         if c == '.': self.dot()
 111                         elif c == '-': self.dah()
 112                         else: self.pdah()
 113                         self.pdot()
 114
 115         def sendascii(self, s):
 116                 self.sendmorse(morsecode(s))
 117
 118         def send(self, s):
 119                 self.sendascii(s)
 120
 121
 122 import Audio_mac
 123 class MyAudio(Audio_mac.Play_Audio_mac):
 124         def _callback(self, *args):
 125                 if hasattr(self, 'usercallback'): self.usercallback()
 126                 apply(Audio_mac.Play_Audio_mac._callback, (self,) + args)
 127
 128
 129 class MacMorse(BaseMorse):
 130         "Mac specific class to play Morse code"
 131
 132         def __init__(self):
 133                 BaseMorse.__init__(self)
 134                 self.dev = MyAudio()
 135                 self.dev.setoutrate(FRAMERATE)
 136                 self.dev.setsampwidth(SAMPWIDTH)
 137                 self.dev.setnchannels(1)
 138                 self.dev.usercallback = self.usercallback
 139                 sinewave = ''
 140                 n = int(FRAMERATE / BASEFREQ)
 141                 octave = OCTAVE
 142                 from math import sin, pi
 143                 for i in range(n):
 144                         val = int(sin(2 * pi * i * octave / n) * 0x7fff)
 145                         sample = chr((val >> 8) & 255) + chr(val & 255)
 146                         sinewave = sinewave + sample[:SAMPWIDTH]
 147                 self.sinewave = sinewave
 148                 self.silence = '\0' * (n*SAMPWIDTH)
 149                 self.morsequeue = ''
 150
 151         def __del__(self):
 152                 self.close()
 153
 154         def close(self):
 155                 self.dev = None
 156
 157         def pause(self, duration):
 158                 self.dev.writeframes(self.silence * duration)
 159
 160         def noise(self, duration):
 161                 self.dev.writeframes(self.sinewave * duration)
 162
 163         def sendmorse(self, s):
 164                 self.morsequeue = self.morsequeue + s
 165                 self.dev.usercallback()
 166                 self.dev.usercallback()
 167                 self.dev.usercallback()
 168
 169         def usercallback(self):
 170                 if self.morsequeue:
 171                         c, self.morsequeue = self.morsequeue[0], self.morsequeue[1:]
 172                         if c == '.': self.dot()
 173                         elif c == '-': self.dah()
 174                         else: self.pdah()
 175                         self.pdot()
 176
 177
 178 def test():
 179         m = MacMorse()
 180         while 1:
 181                 try:
 182                         line = raw_input('Morse line: ')
 183                 except (EOFError, KeyboardInterrupt):
 184                         break
 185                 m.send(line)
 186                 while m.morsequeue: pass
 187
 188 test()