mirror of
https://github.com/DJ2LS/FreeDATA
synced 2024-05-14 08:04:33 +00:00
71 lines
3.1 KiB
Python
71 lines
3.1 KiB
Python
|
import numpy as np
|
||
|
|
|
||
|
|
"""
|
||
|
|
morse code generator
|
||
|
|
MorseCodePlayer().text_to_signal("DJ2LS-1")
|
||
|
|
|
||
|
|
|
||
|
|
"""
|
||
|
|
class MorseCodePlayer:
|
||
|
|
def __init__(self, wpm=150, f=1500, fs=48000):
|
||
|
|
self.wpm = wpm
|
||
|
|
self.f0 = f
|
||
|
|
self.fs = fs
|
||
|
|
self.dot_duration = 1.2/(self.wpm)
|
||
|
|
self.dash_duration = 3*self.dot_duration
|
||
|
|
self.pause_duration = self.dot_duration
|
||
|
|
self.word_pause_duration = 7*self.dot_duration
|
||
|
|
self.morse_alphabet = {
|
||
|
|
'A': '.-', 'B': '-...', 'C': '-.-.', 'D': '-..', 'E': '.', 'F': '..-.', 'G': '--.', 'H': '....',
|
||
|
|
'I': '..', 'J': '.---', 'K': '-.-', 'L': '.-..', 'M': '--', 'N': '-.', 'O': '---', 'P': '.--.',
|
||
|
|
'Q': '--.-', 'R': '.-.', 'S': '...', 'T': '-', 'U': '..-', 'V': '...-', 'W': '.--', 'X': '-..-',
|
||
|
|
'Y': '-.--', 'Z': '--..', '0': '-----', '1': '.----', '2': '..---', '3': '...--', '4': '....-',
|
||
|
|
'5': '.....', '6': '-....', '7': '--...', '8': '---..', '9': '----.', '.': '.-.-.-', ',': '--..--',
|
||
|
|
'?': '..--..', "'": '.----.', '!': '-.-.--', '/': '-..-.', '(': '-.--.', ')': '-.--.-', '&': '.-...',
|
||
|
|
':': '---...', ';': '-.-.-.', '=': '-...-', '+': '.-.-.', '-': '-....-', '_': '..--.-', '"': '.-..-.',
|
||
|
|
'$': '...-..-', '@': '.--.-.'
|
||
|
|
}
|
||
|
|
|
||
|
|
def text_to_morse(self, text):
|
||
|
|
morse = ''
|
||
|
|
for char in text:
|
||
|
|
if char.upper() in self.morse_alphabet:
|
||
|
|
morse += self.morse_alphabet[char.upper()] + ' '
|
||
|
|
elif char == ' ':
|
||
|
|
morse += ' '
|
||
|
|
return morse
|
||
|
|
|
||
|
|
def morse_to_signal(self, morse):
|
||
|
|
signal = np.array([], dtype=np.int16)
|
||
|
|
for char in morse:
|
||
|
|
if char == '.':
|
||
|
|
duration = int(self.dot_duration * self.fs)
|
||
|
|
s = np.sin(2 * np.pi * self.f0 * np.arange(duration) / self.fs)
|
||
|
|
signal = np.concatenate((signal, s * 32767))
|
||
|
|
pause_duration = int(self.pause_duration * self.fs)
|
||
|
|
signal = np.concatenate((signal, np.zeros(pause_duration, dtype=np.int16)))
|
||
|
|
elif char == '-':
|
||
|
|
duration = int(self.dash_duration * self.fs)
|
||
|
|
s = np.sin(2 * np.pi * self.f0 * np.arange(duration) / self.fs)
|
||
|
|
signal = np.concatenate((signal, s * 32767))
|
||
|
|
pause_duration = int(self.pause_duration * self.fs)
|
||
|
|
signal = np.concatenate((signal, np.zeros(pause_duration, dtype=np.int16)))
|
||
|
|
elif char == ' ':
|
||
|
|
pause_duration = int(self.word_pause_duration * self.fs)
|
||
|
|
signal = np.concatenate((signal, np.zeros(pause_duration, dtype=np.int16)))
|
||
|
|
pause_duration = int(self.pause_duration * self.fs)
|
||
|
|
signal = np.concatenate((signal, np.zeros(pause_duration, dtype=np.int16)))
|
||
|
|
|
||
|
|
pause_duration = int(self.word_pause_duration * self.fs)
|
||
|
|
#signal = np.concatenate((signal, np.zeros(pause_duration, dtype=np.int16)))
|
||
|
|
|
||
|
|
# Convert the signal to mono (single-channel)
|
||
|
|
#signal = signal.reshape(-1, 1)
|
||
|
|
|
||
|
|
return signal
|
||
|
|
|
||
|
|
def text_to_signal(self, text):
|
||
|
|
morse = self.text_to_morse(text)
|
||
|
|
return self.morse_to_signal(morse)
|
||
|
|
|