/;ji jddlZddlmZmZddlmZmZmZGdde Z GddeZ dS)N)ABCabstractmethod)OptionalDictAnyceZdZdZdS)WatermarkingExceptionz8Exception raised for errors in the watermarking process.N)__name__ __module__ __qualname____doc__K/root/voice-cloning/.venv/lib/python3.11/site-packages/perth/watermarker.pyr r sBBDrr c eZdZdZe d dejdeejdedejfdZ e dd ejded eedejfd Z dejdede fd Z dS)WatermarkerBasez Base class for all audio watermarking algorithms. This abstract class defines the interface that all watermarking implementations must follow, providing methods for watermark application and extraction. NDwav watermark sample_ratereturnc t)a Apply a watermark to an audio signal. Args: wav: Input audio signal as numpy array watermark: Optional watermark data to embed. If None, a default watermark may be generated. sample_rate: Sample rate of the audio signal in Hz **kwargs: Additional algorithm-specific parameters Returns: Watermarked audio signal as numpy array Raises: WatermarkingException: If watermarking fails NotImplementedError)selfrrrkwargss rapply_watermarkzWatermarkerBase.apply_watermark$"###rwatermarked_wavwatermark_lengthc t)a Extract a watermark from a watermarked audio signal. Args: watermarked_wav: Watermarked audio signal as numpy array sample_rate: Sample rate of the audio signal in Hz watermark_length: Optional expected length of the watermark **kwargs: Additional algorithm-specific parameters Returns: Extracted watermark data as numpy array Raises: WatermarkingException: If watermark extraction fails r)rrrr rs r get_watermarkzWatermarkerBase.get_watermark'rrcdS)a6 Verify if the audio is compatible with this watermarking method. Args: wav: Input audio signal as numpy array sample_rate: Sample rate of the audio signal in Hz Returns: True if the audio is compatible, False otherwise Tr)rrrs rverify_compatibilityz$WatermarkerBase.verify_compatibility;s tr)Nr)rN) r r r r rnpndarrayrintrr"boolr$rrrrr sQU+0$$2:$(2::N$%($?Az$$$^$&LQ8<$$RZ$c$(0 $KM:$$$^$&         rr) numpyr%abcrrtypingrrr Exceptionr rrrrr-s########&&&&&&&&&&     I   ;;;;;c;;;;;r