o Vù¡i™ ã@sPddlZddlmZmZddlmZmZmZGdd„de ƒZ Gdd„deƒZ dS)éN)ÚABCÚabstractmethod)ÚOptionalÚDictÚAnyc@seZdZdZdS)ÚWatermarkingExceptionz8Exception raised for errors in the watermarking process.N)Ú__name__Ú __module__Ú __qualname__Ú__doc__©r r úE/home/ubuntu/.local/lib/python3.10/site-packages/perth/watermarker.pyrsrc @s‚eZdZdZe  ddejdeejdedejfdd „ƒZ e  dd ejded eedejfd d „ƒZ dejdede fdd„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. NéD¬ÚwavÚ watermarkÚ sample_rateÚreturncKó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)ÚselfrrrÚkwargsr r r Úapply_watermarkózWatermarkerBase.apply_watermarkÚwatermarked_wavÚwatermark_lengthcKr)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)rrrrrr r r Ú get_watermark'rzWatermarkerBase.get_watermarkcCsdS)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 )rrrr r r Úverify_compatibility;s z$WatermarkerBase.verify_compatibility)Nr)rN) rr r r rÚnpÚndarrayrÚintrrÚboolrr r r r r s&ÿÿ ÿÿÿ ÿr) ÚnumpyrÚabcrrÚtypingrrrÚ Exceptionrrr r r r Ús