Removed noisegate (again)
This commit is contained in:
Logan Cusano
130
NoiseGate-v2.py
130
NoiseGate-v2.py
@@ -1,130 +0,0 @@
|
||||
import pyaudio
|
||||
import struct
|
||||
import numpy
|
||||
import time
|
||||
from threading import Thread
|
||||
|
||||
sound_buffer = []
|
||||
|
||||
|
||||
class AudioSteam(Thread):
|
||||
def __init__(self, _channels: int = 1, _sample_rate: int = 48000, _frames_per_buffer: int = 2048,
|
||||
_input_device_index: int = None, _output_device_index: int = None, _input: bool = True,
|
||||
_output: bool = True):
|
||||
super(AudioSteam, self).__init__()
|
||||
self.paInstance_kwargs = {
|
||||
'format': pyaudio.paFloat32,
|
||||
'channels': _channels,
|
||||
'rate': _sample_rate,
|
||||
'input': _input,
|
||||
'output': _output,
|
||||
'frames_per_buffer': _frames_per_buffer,
|
||||
'stream_callback': callback
|
||||
}
|
||||
|
||||
if _input_device_index:
|
||||
if _input:
|
||||
self.paInstance_kwargs['input_device_index'] = _input_device_index
|
||||
else:
|
||||
print(f"[AudioStream.__init__]:\tInput was not enabled."
|
||||
f" Reinitialize with '_input=True'")
|
||||
|
||||
if _output_device_index:
|
||||
if _output:
|
||||
self.paInstance_kwargs['output_device_index'] = _output_device_index
|
||||
else:
|
||||
print(f"[AudioStream.__init__]:\tOutput was not enabled."
|
||||
f" Reinitialize with '_output=True'")
|
||||
|
||||
# Init PyAudio instance
|
||||
self.paInstance = pyaudio.PyAudio()
|
||||
|
||||
# Define and initialize stream object if we have been passed a device ID (pyaudio.open)
|
||||
self.stream = None
|
||||
if _output_device_index and _input_device_index:
|
||||
self.init_stream()
|
||||
|
||||
# temp section
|
||||
|
||||
def init_stream(self, _new_output_device_index: int = None, _new_input_device_index: int = None):
|
||||
# Check what device was asked to be changed (or set)
|
||||
if _new_input_device_index:
|
||||
if self.paInstance_kwargs['input']:
|
||||
self.paInstance_kwargs['input_device_index'] = _new_input_device_index
|
||||
else:
|
||||
print(f"[AudioStream.init_stream]:\tInput was not enabled when initialized."
|
||||
f" Reinitialize with '_input=True'")
|
||||
|
||||
if _new_output_device_index:
|
||||
if self.paInstance_kwargs['output']:
|
||||
self.paInstance_kwargs['output_device_index'] = _new_output_device_index
|
||||
else:
|
||||
print(f"[AudioStream.init_stream]:\tOutput was not enabled when initialized."
|
||||
f" Reinitialize with '_output=True'")
|
||||
|
||||
self.close_if_open()
|
||||
|
||||
# Reopen the stream
|
||||
self.stream = self.paInstance.open(**self.paInstance_kwargs)
|
||||
|
||||
def close_if_open(self):
|
||||
# Stop the stream if it is started
|
||||
if self.stream:
|
||||
if self.stream.is_active():
|
||||
self.stream.stop_stream()
|
||||
self.stream.close()
|
||||
print(f"[ReopenStream.close_if_open]:\t Stream was open; It was closed.")
|
||||
|
||||
def list_devices(self, _show_input_devices: bool = True, _show_output_devices: bool = True):
|
||||
info = self.paInstance.get_host_api_info_by_index(0)
|
||||
numdevices = info.get('deviceCount')
|
||||
|
||||
for i in range(0, numdevices):
|
||||
if (self.paInstance.get_device_info_by_host_api_device_index(0, i).get('maxInputChannels')) > 0:
|
||||
if _show_input_devices:
|
||||
print("Input Device id ", i, " - ",
|
||||
self.paInstance.get_device_info_by_host_api_device_index(0, i).get('name'))
|
||||
|
||||
if (self.paInstance.get_device_info_by_host_api_device_index(0, i).get('maxOutputChannels')) > 0:
|
||||
if _show_output_devices:
|
||||
print("Output Device id ", i, " - ",
|
||||
self.paInstance.get_device_info_by_host_api_device_index(0, i).get('name'))
|
||||
|
||||
|
||||
class NoiseGate(AudioSteam):
|
||||
def __init__(self, **kwargs):
|
||||
super().__init__(**kwargs)
|
||||
|
||||
def run(self) -> None:
|
||||
# Start the audio stream
|
||||
self.stream.start_stream()
|
||||
|
||||
global sound_buffer
|
||||
# While the stream is running, display the stream buffer in floats?
|
||||
while self.stream.is_active():
|
||||
if len(sound_buffer) > 0:
|
||||
for buffer in sound_buffer:
|
||||
volume_normalization = numpy.linalg.norm(numpy.fromstring(buffer)) * 10
|
||||
print(str(float(volume_normalization)))
|
||||
|
||||
self.stream.stop_stream()
|
||||
self.stream.close()
|
||||
|
||||
self.paInstance.terminate()
|
||||
|
||||
|
||||
def callback(in_data, frame_count, time_info, status):
|
||||
global sound_buffer
|
||||
|
||||
sound_buffer.append(in_data)
|
||||
|
||||
return in_data, pyaudio.paContinue
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
input_index = int(input("Input:\t"))
|
||||
output_index = int(input("Output:\t"))
|
||||
|
||||
ng = NoiseGate(_input_device_index=input_index, _output_device_index=output_index)
|
||||
|
||||
ng.start()
|
||||
67
NoiseGate.py
67
NoiseGate.py
@@ -1,67 +0,0 @@
|
||||
import threading
|
||||
import time
|
||||
|
||||
import numpy as np
|
||||
import discord
|
||||
import sounddevice as sd
|
||||
from threading import Thread, Event
|
||||
|
||||
noise_gate_trigger = 0 # Set this value for the trigger on the noise-gate
|
||||
voice_connection = None
|
||||
audio_stream = None
|
||||
last_callback_value = 0
|
||||
|
||||
|
||||
class NoiseGate(Thread):
|
||||
def __init__(self, trigger_value: int = 700):
|
||||
global noise_gate_trigger
|
||||
super(NoiseGate, self).__init__()
|
||||
self.stream = None
|
||||
self.stop_NG = threading.Event()
|
||||
self.NG_Started = threading.Event()
|
||||
noise_gate_trigger = trigger_value
|
||||
|
||||
def init_stream(self, num, _voice_connection, _audio_stream):
|
||||
global voice_connection, audio_stream
|
||||
self.stream = sd.InputStream(device=num, callback=stream_callback)
|
||||
voice_connection = _voice_connection
|
||||
audio_stream = _audio_stream
|
||||
|
||||
def run(self) -> None:
|
||||
self.NG_Started.set()
|
||||
self.stream.start()
|
||||
while not self.stop_NG.is_set():
|
||||
#print("Main loop start")
|
||||
if last_callback_value >= noise_gate_trigger:
|
||||
trigger_noise_gate(True)
|
||||
while last_callback_value >= noise_gate_trigger:
|
||||
time.sleep(6)
|
||||
trigger_noise_gate(False)
|
||||
#print("Main loop end")
|
||||
|
||||
self.stream.stop()
|
||||
self.stream.close()
|
||||
self.NG_Started.clear()
|
||||
voice_connection.stop()
|
||||
print('Thread #%s stopped' % self.ident)
|
||||
|
||||
|
||||
def stream_callback(indata, *args):
|
||||
global last_callback_value
|
||||
volume_normalization = np.linalg.norm(indata) * 10
|
||||
last_callback_value = int(volume_normalization)
|
||||
#print(f"Callback Value: {last_callback_value}")
|
||||
|
||||
|
||||
def trigger_noise_gate(triggered: bool):
|
||||
if triggered:
|
||||
if not voice_connection.is_playing():
|
||||
voice_connection.play(discord.PCMAudio(audio_stream))
|
||||
# print("|" * int(volume_normalization / 4))
|
||||
print("Noise Gate was Triggered")
|
||||
#time.sleep(10)
|
||||
else:
|
||||
if voice_connection.is_playing():
|
||||
print("Noise Gate stopped")
|
||||
voice_connection.pause()
|
||||
# try disconnecting and reconnecting
|
||||
Reference in New Issue
Block a user