wi dZddlZddlZddlZddlZddlmZ ddl m Z m Z m Z mZddlmZddlmZd%dZdZd&d Zd&d Zd%d Zd Zd'dededefdZdefdZdedejfdZdejdefdZ dede!fdZ"d(dede ede ejeffdZ#dejdededdfdZ$ d(d ejd!ejded"e eddf d#Z%d ejd!ejde eeffd$Z&dS))z: Utility functions for audio processing and watermarking. N)TupleOptionalDictAny)sqrt)modeTc  |jd}g}td||D]l}||||z}|rF|jd|kr5tj|tj||jdz }||m|S)Nr)shaperangenpappendzeros)data window_lengthpad n_samplesframesidxchunks [/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/perth/utils.py_signal_to_framesrs 2I FQ =11S},,-  R5;r?]22IeRX}u{2/N%P%PQQE e Mc*tj|SN)r hstack)rs r_frames_to_signalrs 9V  rc|jjdks Jd|dz }d|z d|zdz }}||z||tjS)Nfz)This function takes floating point arrays)dtypekindclipastyper int16)wav bit_depthunsigned_bit_depth range_min range_maxs r audio_to_rawr-sr 9>S "M "Q"455q=O7PST7TyI )O ! !)Y 7 7 > >rx H HHrc|dz }d|z d|zdz }}d\}}|||z ||z z||z z zddtjS)Nr!r")gg?r )r%r&r float32)r(r)r*r+r,abs rraw_pcm16_tofloatr2$sx"Q"455qV#@>^!4!4v!=>>?  1I''HRXiY61IJJKKM %i00 WfW rcTt|djdS)Nr!)axisr )rsqueeze)watermark_vectors rr8r88s'  q ) ) ) . 6 6r : ::r@nkreturnc|dkrdSdtdz|z }dtdz |z }t||z||zz tdz S)Nrr!)rint)rCrDalphabetas rmodified_binets_fibonnacirK<s_  Avvaa a[A E QK1 D ! *d1gg5 6 66rr:ctj|f}tj||kddS)Nr!r)r randomwheremean)r:r<s rgenerate_dummy_watermarkrPHs<   &++I 8I  0 00!Q 7 77rr<c>tjd|DS)Nc,g|]}t|S)rH.0chars r z*watermark_str_to_numpy..Ns5554SYY555r)r r5r<s rwatermark_str_to_numpyrYMs! 8559555 6 66rc@dd|DS)Nc34K|]}t|VdSr)strrTs r z)watermark_numpy_to_str..Rs(333t99333333r)joinrXs rwatermark_numpy_to_strr`Qs# 7733333 3 33rc4td|DS)Ncg|]}|dvS))10rSrTs rrWz-validate_string_watermark..Vs===4J&===r)anyrXs rvalidate_string_watermarkrfUs ==9=== > >>r audio_pathsrc tj||\}}||fS#t$r}td|d|d}~wwxYw)z Load an audio file using librosa. Args: audio_path: Path to the audio file sr: Target sample rate. If None, the native sample rate is used. Returns: Tuple of (audio_data, sample_rate) )rhzCould not load audio file z: N)librosaload ExceptionIOError)rgrhaudio sample_ratees r load_audiorqYsnF$\*<<<{k!! FFFD:DDDDEEEFs A>A audio_data file_pathroctjtj|}tj|dt j|||dS)z Save audio data to a file. Args: audio_data: Audio data as a numpy array file_path: Output file path sample_rate: Sample rate for the audio file T)exist_okN)ospathdirnameabspathmakedirssfwrite)rrrsro directorys r save_audior~ksT : :;;IK D))))HY K00000roriginal watermarked output_pathc6tjddd\}}tjt ||z }|d||dd|d||dd |dd |dd |dd |d ||z }|d|||dd |dd |ddtj tj tj |tj}tj|dd||d|dd|dd |dd||djd|ddtj|r)tj|tjdStjdS)a$ Plot a comparison between original and watermarked audio. Args: original: Original audio data watermarked: Watermarked audio data sample_rate: Sample rate of the audio output_path: Path to save the plot. If None, plot is shown interactively. r!) )figsizergffffff?Original)rIlabel WatermarkedzWaveform ComparisonzTime (s) Amplitudez#Difference (Watermarked - Original) Difference)reftimelogr")x_axisy_axisrhaxzSpectrogram of DifferencezFrequency (Hz)z %+2.0f dB)rformatN)pltsubplotsr aranger4plot set_title set_xlabel set_ylabellegendrjamplitude_to_dbabsstftmaxdisplayspecshowcolorbar collections tight_layoutsavefigcloseshow) rrrorfigaxsrdiffDs rplot_audio_comparisonrysR|Aq(333HC 9S]] # #k 1DFKKhcK<<<FKKkMKBBBF*+++Fj!!!Fk"""FMMOOO  !DFKKdF:;;;Fj!!!Fl###  w|D!!""   A OQve PSTUPVWWWF0111Fj!!!F&'''LLQ#A&3q6+LFFF K      rcbt|t|krtdtj||z dz}tj|dz}|}|dkrdtj||z znt d}t tjtj|tjtj|}|dkr,dtj|tj|z znt d}|||dS)ak Calculate audio quality metrics between original and watermarked audio. Args: original: Original audio data watermarked: Watermarked audio data Returns: Dictionary of quality metrics: - snr: Signal-to-Noise Ratio (dB) - mse: Mean Squared Error - psnr: Peak Signal-to-Noise Ratio (dB) z8Original and watermarked audio must have the same lengthr"rrinf)snrmsepsnr) r4 ValueErrorr rOlog10floatrrr)rrr signal_power noise_powerr max_valuers rcalculate_audio_metricsrs 8}}K((((STTT '8k)a/ 0 0C78q=))LK7BQ"rx {233 3 3ERWLLCBF26(++,,bfRVK5H5H.I.IJJI69Agg2RWS\\122 2 25<rs ------------IIIIhhhh    ;;; 7 7 7 7 7 7 7 78S8888 7c7bj77774bj4S4444??????FF3FHSMFU2:s?=SFFFF$ 12: 1# 1C 1D 1 1 1 1IM,,BJ,RZ,&),8@ ,QU,,,,^!bj!rz!dSVX]S]N^!!!!!!r