-
Notifications
You must be signed in to change notification settings - Fork 0
/
solution.py
113 lines (101 loc) · 2.25 KB
/
solution.py
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
import wave
morse_code = {
"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 get_morse_char(on, duration):
if on:
if duration > 1000:
return "-"
else:
return "."
else:
if duration > 2000:
return " / "
elif duration > 1000:
return " "
return ""
def count_consecutive(values):
result = []
count = 0
value = values[0]
for v in values:
if v == value:
count += 1
else:
if count > 1:
result.append((value, count))
count = 1
value = v
result.append((value, count))
return result
def get_morse_string(filename):
with wave.open(filename, "rb") as f:
frames = f.readframes(f.getnframes())
print("frames count", frames.count(128))
most_common = max(set(reversed(frames)), key=frames.count)
print("MOST", most_common)
values = [most_common != frame for frame in frames]
out = ""
for on, dur in count_consecutive(values):
out += get_morse_char(on, dur)
return out
def morse_to_text(morse):
return "".join([k for k, v in morse_code.items() if v == morse])
def decode(filename):
morse = get_morse_string(filename).split(" ")
return''.join(morse_to_text(m) for m in morse)
print(decode("morse.wav"))