New file |
| | |
| | | import pyaudio |
| | | import wave |
| | | from datetime import datetime |
| | | |
| | | # Generate a file name from current date and time |
| | | current_date_time = datetime.now().strftime("%Y-%m-%d--%H-%M-%S") |
| | | file_name = 'recording_' + current_date_time |
| | | |
| | | # Initialize PyAudio |
| | | p = pyaudio.PyAudio() |
| | | |
| | | # No prompts needed for the startup version of this script |
| | | ''' |
| | | # List available input devices |
| | | print("Available audio input devices:") |
| | | for i in range(p.get_device_count()): |
| | | device_info = p.get_device_info_by_index(i) |
| | | if device_info['maxInputChannels'] > 0: |
| | | print(f"Device ID {i}: {device_info['name']} (Max channels: {device_info['maxInputChannels']})") |
| | | |
| | | # Prompt user to select a device |
| | | device_index = int(input("Please select the device ID to use for recording: ")) |
| | | n_channels = int(input("Please enter the number of channels to use for recording: ")) |
| | | duration_sec = int(input("Please enter the desired duration (in seconds) of the recording: ")) |
| | | #file_name = str(input("Please enter a name for the output file: ")) |
| | | ''' |
| | | |
| | | # Parameters |
| | | CHANNELS = 19 # Number of channels |
| | | SAMPLE_RATE = 48000 |
| | | CHUNK = 32 # Number of frames per buffer |
| | | FORMAT = pyaudio.paInt16 # 16-bit audio |
| | | RECORD_SECONDS = 10 # Duration to record (in seconds) |
| | | OUTPUT_FILENAME = "/home/rpi1/Desktop/" + file_name + ".wav" |
| | | MONITORING = True # To write to a text file as long as the device is recording |
| | | |
| | | # Get the chosen device's info to validate it can support the entered number of channels and 48kHz |
| | | device_info = p.get_device_info_by_index(2) |
| | | max_channels = device_info['maxInputChannels'] |
| | | if max_channels < CHANNELS: |
| | | print(f"Error: The selected device only supports {max_channels} channels, but {CHANNELS} channels are required.") |
| | | p.terminate() |
| | | exit(1) |
| | | |
| | | # Open stream |
| | | stream = p.open(format=FORMAT, |
| | | channels=CHANNELS, |
| | | rate=SAMPLE_RATE, |
| | | input=True, |
| | | input_device_index=2, |
| | | frames_per_buffer=CHUNK) |
| | | |
| | | print(f"Recording {RECORD_SECONDS} seconds of audio from device ID {2}...") |
| | | |
| | | frames = [] |
| | | |
| | | print (f"Check status {stream.is_active()}") |
| | | |
| | | if MONITORING: |
| | | open("/home/rpi1/Desktop/current_recording_" + current_date_time + ".txt","w+") |
| | | |
| | | for i in range(int(SAMPLE_RATE / CHUNK * RECORD_SECONDS)): |
| | | |
| | | # print (i, f"RECORDING: Check status {stream.is_active()}") # this line is for testing |
| | | |
| | | try: |
| | | # Record audio |
| | | # data = stream.read(CHUNK) |
| | | data = stream.read(CHUNK, exception_on_overflow = False) # this line is for testing |
| | | frames.append(data) |
| | | |
| | | if MONITORING: |
| | | if i%int(10*SAMPLE_RATE/CHUNK) == 0: # writing current time into "current_recording.txt" every 10 seconds |
| | | current_second = int(i*CHUNK/SAMPLE_RATE) |
| | | with open('current_recording_' + current_date_time + '.txt', 'a') as f: |
| | | f.write(f"{current_second}\n") |
| | | |
| | | except: |
| | | print ("Error") |
| | | |
| | | print("Finished recording.") |
| | | |
| | | |
| | | print (f"Check status {stream.is_active()}") |
| | | # Stop and close the stream |
| | | stream.stop_stream() |
| | | stream.close() |
| | | |
| | | # Terminate PyAudio |
| | | p.terminate() |
| | | |
| | | # Save the recorded audio to a file |
| | | wf = wave.open(OUTPUT_FILENAME, 'wb') |
| | | wf.setnchannels(CHANNELS) |
| | | wf.setsampwidth(p.get_sample_size(FORMAT)) |
| | | wf.setframerate(SAMPLE_RATE) |
| | | wf.writeframes(b''.join(frames)) |
| | | wf.close() |
| | | |
| | | print(f"Audio saved to {OUTPUT_FILENAME}.") |
| | | |