iQ$ddlZdZdZdZdZy)Nc~tjtjtj|ddS)z Calculates the root mean square. Based on https://github.com/iver56/audiomentations/blob/master/audiomentations/core/utils.py F)dimkeepdim)torchsqrtmeansquare)sampless t/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/torch_audiomentations/utils/dsp.py calculate_rmsr s) ::ejjg!6BN OOc|d|dz zz }|S)a Given the Root Mean Square (RMS) of a clean sound and a desired signal-to-noise ratio (SNR), calculate the desired RMS of a noise sound to be mixed in. Based on https://github.com/Sato-Kunihiko/audio-SNR/blob/8d2c933b6c0afe6f1203251f4877e7a1068a6130/create_mixed_audio_file.py#L20 :param clean_rms: Root Mean Square (RMS) - a value between 0.0 and 1.0 :param snr: Signal-to-Noise (SNR) Ratio in dB - typically somewhere between -20 and 60 :return: ) clean_rmssnr noise_rmss r calculate_desired_noise_rmsr sRC"H-.I rcd|dz zS)Nrrr)decibelss r #convert_decibels_to_amplitude_ratiors (R-  rc2dtj|zS)Nr)rlog10)amplitude_ratios r #convert_amplitude_ratio_to_decibelsrs  O, ,,r)rr rrrrrr rs P !-r