o wi"@sTddlmZddlmZddlZddlZddlmZddl m Z Gddde Z dS) )division) iteritemsN)util) IIRfilterc@s<eZdZdZd ddZddZedd Zejd d Zd S) Metera Meter object which defines how the meter operates Defaults to the algorithm defined in ITU-R BS.1770-4. Parameters ---------- rate : float Sampling rate in Hz. filter_class : str Class of weigthing filter used. - 'K-weighting' - 'Fenton/Lee 1' - 'Fenton/Lee 2' - 'Dash et al.' - 'DeMan' block_size : float Gating block size in seconds. K-weighting皙?cCs||_||_||_dSN)rate filter_class block_size)selfr r r rM/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/pyloudnorm/meter.py__init__s zMeter.__init__c s$|}t||j|j|jdkrt||jddf}|jd|jd}t |j D]\}}t D]}| |dd|f|dd|f<q4q,gd|j}dd}d|} ||j} t t| ||| d} td| } tj| fdt D]8} | D]3}t ||| |j}t ||| d|j}d||jtt|||| f| |f<qq~ttjd td fd d | D}Wdn1swYfd d t|Dttjd td fdd t DWdn 1s wYddttfdd t Ddfdd t|Dt!tjd td ttfdd t DWdn 1s]wYtjd dddttfdd t D}Wd|S1swY|S)a Measure the integrated gated loudness of a signal. Uses the weighting filters and block size defined by the meter the integrated loudness is measured based upon the gating algorithm defined in the ITU-R BS.1770-4 specification. Input data must have shape (samples, ch) or (samples,) for mono audio. Supports up to 5 channels and follows the channel ordering: [Left, Right, Center, Left surround, Right surround] Params ------- data : ndarray Input multichannel audio data. Returns ------- LUFS : float Integrated gated loudness of the input measured in dB LUFS. rrN)?rr(\?rgQg?r)shapeignore)categoryc s:g|]ddttfddtDqS)&1$@cs g|] }||fqSrr.0i)Gjzrr Ys 8Meter.integrated_loudness...)nplog10sumranger)r numChannelsr)rrrYs:z-Meter.integrated_loudness..csg|] \}}|kr|qSrrrrl_j)Gamma_arrr\c&g|]tfddDqS)cg|]}|fqSrrrrrrrrrar r!meanr%J_grrrra&rrcg|] }||qSrrrr z_avg_gatedrrrcr*cs$g|]\}}|kr|kr|qSrrr')r)Gamma_rrrrfs$cr+)cr,rrr-r.rrrjr/r r0r%r2r4rrjr5)dividecr6rrrr7rrrnr*)copyr valid_audior r ndimr!reshaperr_filtersr$ apply_filterintroundarangezerosr#squarewarningscatch_warnings simplefilterRuntimeWarning enumerater" nan_to_numarrayerrstate)rdata input_data numSamplesr filter_stagechT_goverlapstepT numBlocksj_rangerrluLUFSr)rr)r9r3r&rr8rintegrated_loudness!s\    $   4  0 &. zMeter.integrated_loudnesscCs|jSr ) _filter_class)rrrrr rszMeter.filter_classcCsi|_||_|jdkr*tddtdd|jd|jd<tddd |jd |jd <dS|jd kr_td dtdd|jd|jd<tddd |jd |jd <tddtdd|jd|jd<dS|jdkrtddtdd|jd|jd<tddd |jd |jd <dS|jdkrtddd|jd |jd <tddd|jd|jd<dS|jdkrtddd|jd|jd<tddd|jd|jd<dS|jd krdStd!|j)"Nrg@rgp@ high_shelfgg?gC@ high_passz Fenton/Lee 1g@g@`@g@@peakingz Fenton/Lee 2 high_selfz Dash et al.g?gb@gݡsgrˢ E?g@@DeMangCo"@giwh-?gCxG@high_shelf_DeMangح?g^JWC@high_pass_DeMancustomzInvalid filter class:)r?r]rr!sqrtr ValueError)rvaluerrrr vs* " "& "    N)rr ) __name__ __module__ __qualname____doc__rr\propertyr setterrrrrrs Q r) __future__r future.utilsrrFnumpyr!r iirfilterrobjectrrrrrs