o Si@sjddlZddlmZddlmZmZmZmZmZddl m Z ddl m Z ddl mZeGdddZdS) N) dataclass)ListLiteralOptionalTupleUnion)CutSet)Codec) resolve_seedc@seZdZUdZeeed<dZee e e e ffed<dZ e ee ed<dZ eed<d Ze ed <d Zeeed fed <dZe ejed<dZeed<dddZdedefddZdS)Compressa Applies a lossy compression algorithm filter to each Cut in a CutSet. The audio is decompressed back to raw waveforms. The compression is applied with a probability of ``p``. The codec is randomly selected the list of provided codecs, with optional weights controlling the selection. If compression level is provided as an interval, then the actual value is sampled uniformly from the provided interval. :param codecs: A list of codecs (supported: "opus", "mp3", "vorbis", "gsm") :param compression_level: A single value or an interval. 0.0 = lowest compression (highest bitrate), 1.0 = highest compression (lowest bitrate). If an interval is provided, the value is sampled uniformly. :param codec_weights: Optional weights for each codec (default: equal weights). :param p: The probability of applying the low-pass filter (default: 0.5). :param randgen: An optional random number generator (default: a new instance). :param preserve_id: Whether to preserve the original cut ID (default: False). codecsg?compression_levelN codec_weightsFcompress_custom_fieldsg?p*)trng randomizedseedrng preserve_idreturncCsDt|jttt|jksJdt|jttfr= rz+Probability p must be between 0 and 1, got zLExpected codec_weights to be a list with the same length as codecs, got len(z ) != len()css|]}|dkVqdS)rN).0wrrZ/home/ubuntu/.local/lib/python3.10/site-packages/lhotse/dataset/cut_transforms/compress.py 8s z)Compress.__post_init__..z&All codec weights must be non-negativecSsg|]}dqS)g?r)r_rrr =sz*Compress.__post_init__..z-Either rng or seed must be provided, not both)sortedr listset isinstancer rrlenrrallrr ValueErrorrandomRandomr )selfmin_compressionmax_compressionrrr __post_init__&s6$   (   zCompress.__post_init__cutsc Csg}|D]V}|j|jkrUt|jttfr'|j\}}|j|||}n|j}|jj|j|j d^}}|j |||j d} |j sO| j d|d|d| _ || q||qt|S)N)weights)codecr rr z.2f)rr)rr%r rrchoicesr rcompressrridappendr) r+r/compressed_cutscutr,r-r r1r new_cutrrr__call__Ds*   zCompress.__call__)rN)__name__ __module__ __qualname____doc__rr __annotations__r rfloatrrrrboolrrintrrr)r*rr.rr9rrrrr s      r )r) dataclassesrtypingrrrrrlhotserlhotse.augmentation.compressr lhotse.dataset.dataloadingr r rrrrs