-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
140 lines (123 loc) · 3.56 KB
/
main.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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
import argparse
import queue
import sys
# from matplotlib.animation import FuncAnimation
# import matplotlib.pyplot as plt
import numpy as np
import sounddevice as sd
import pigpio
print("imports done")
def int_or_str(text):
try:
return int(text)
except ValueError:
return text
parser = argparse.ArgumentParser(add_help=False)
parser.add_argument(
"-l",
"--list-devices",
action="store_true",
help="show list of audio devices and exit",
)
args, remaining = parser.parse_known_args()
if args.list_devices:
print(sd.query_devices())
parser.exit(0)
parser = argparse.ArgumentParser(
description=__doc__,
formatter_class=argparse.RawDescriptionHelpFormatter,
parents=[parser],
)
parser.add_argument(
"channels",
type=int,
default=[1],
nargs="*",
metavar="CHANNEL",
help="input channels to plot (default: the first)",
)
parser.add_argument(
"-d", "--device", type=int_or_str, help="input device (numeric ID or substring)"
)
parser.add_argument(
"-w",
"--window",
type=float,
default=200,
metavar="DURATION",
help="visible time slot (default: %(default)s ms)",
)
parser.add_argument(
"-i",
"--interval",
type=float,
default=30,
help="minimum time between plot updates (default: %(default)s ms)",
)
parser.add_argument("-b", "--blocksize", type=int, help="block size (in samples)")
parser.add_argument(
"-r", "--samplerate", type=float, help="sampling rate of audio device"
)
parser.add_argument(
"-n",
"--downsample",
type=int,
default=10,
metavar="N",
help="display every Nth sample (default: %(default)s)",
)
parser.add_argument(
"-g", "--graph", action="store_true", help="plot audio on a live graph"
)
args = parser.parse_args(remaining)
if any(c < 1 for c in args.channels):
parser.error("argument CHANNEL: must be >= 1")
mapping = [c - 1 for c in args.channels] # Channel numbers start with 1
q = queue.Queue()
previous_brightnesses = queue.Queue()
rpi = pigpio.pi()
pwm_pin = 12
rpi.set_PWM_range(pwm_pin, 100)
print("init done")
rpi = pigpio.pi()
pwm_pin = 12
rpi.set_PWM_range(pwm_pin, 100)
previous_brightness_q = queue.Queue()
previous_brightness_q.put(int(0.5) * 100)
def update_led_brightness(data):
sound_amplitude = max(max(data), abs(min(data)))
MIN_BRIGHTNESS = 0.3
MAX_BRIGHTNESS = 1
previous_brightness = previous_brightness_q.get()
led_brightness = int(min(sound_amplitude + MIN_BRIGHTNESS, MAX_BRIGHTNESS) * 100)
previous_brightness_q.put(led_brightness)
if abs(previous_brightness - led_brightness) > 2:
for brightness in np.linspace(previous_brightness, led_brightness, 20):
rpi.set_PWM_dutycycle(pwm_pin, int(brightness))
print(int(brightness))
def audio_callback(indata, frames, time, status):
"""This is called (from a separate thread) for each audio block."""
print("in callback")
if status:
print(status, file=sys.stderr)
# Fancy indexing with mapping creates a (necessary!) copy:
data = indata[:: args.downsample, mapping]
q.put(data)
update_led_brightness(data)
try:
if args.samplerate is None:
device_info = sd.query_devices(args.device, "input")
args.samplerate = device_info["default_samplerate"]
stream = sd.InputStream(
device=args.device,
channels=max(args.channels),
samplerate=args.samplerate,
callback=audio_callback,
)
print("stream open")
with stream:
input()
except Exception as e:
parser.exit(type(e).__name__ + ": " + str(e))
finally:
rpi.stop()