MjN(@dZddlZddlZddlZddlmZddlZdejd<iZ dZ da ej Z hdZd Zed Zed Zdadadadad Zd ZdZddZde fdZdZe fdZe fdZe fdZddZddZ ddZ!dD] Z"ee"dS)aANotification sounds for Wan2GP video generation application. Pure Python audio notification system with multiple backend support. Set WAN2GP_DISABLE_AUDIO=1 on headless servers to skip pygame/sounddevice initialization. Set WAN2GP_FORCE_AUDIO=1 to bypass the automatic device detection when audio hardware is available. N)Path1PYGAME_HIDE_SUPPORT_PROMPTiDF>ronyestruectj|d}|t vS)N)osenvirongetstriplower _TRUE_VALUES)namevalues GC:\Users\johnn\OneDrive\Documents\AI\shared\utils\notification_sound.py _env_flagrs6 JNN4 $ $E ;;==   L 00WAN2GP_FORCE_AUDIOWAN2GP_DISABLE_AUDIOctd}|sdS |d}n#ttf$rd}YnwxYw|sdSd|vrdStd }|sd S d |D}n#t$rYd SwxYwd |D}|sdSdS)Nz/proc/asound/cards)Fz//proc/asound/cards is missing (no ALSA devices)ignore)errorsr )Fz8ALSA reports no soundcards (/proc/asound/cards is empty)z no soundcards)Fz/ALSA reports no soundcards (/proc/asound/cards)z/dev/snd)Fz/dev/snd is not availablecg|] }|j S)r).0entrys r z*_linux_audio_available..6sCCCu CCCr)Fz!no permission to inspect /dev/sndc<g|]}|d|S))pcmcontrolChwCmidiC) startswith)rrs rrz*_linux_audio_available..9s;  ??> ? ? r)Fz2no ALSA pcm/control devices exposed under /dev/sndTN) rexists read_textOSErrorUnicodeDecodeErrorrriterdirPermissionError) cards_path cards_contentsnd_path device_namespcm_likes r_linux_audio_availabler2%sM*++J     HGG",,H,== ' (     QPP---////GGJH ??  211:CC0@0@0B0BCCC :::999: H KJJ :s!>AA+C CCctrdStrdStjdrt SdS)Nr&)Fz!disabled via WAN2GP_DISABLE_AUDIOlinux)_FORCE_AUDIO_BACKENDS_DISABLE_AUDIO_BACKENDSsysplatformr%r2rrr_detect_audio_supportr9CsFz:99 |w''(%''' :rc`tr tsdStt\aatS)NF)_audio_backends_failedr5_audio_support_cacher9_audio_support_reasonrrr_should_try_audio_backendsr>Ms5&;u#6K6M6M33 rcd}|rd|d}|tkrt||atjdtjdS)Nz1Audio notification fallback: using terminal beep.zAudio notification fallback: z; using terminal beep.)_last_beep_noticeprintr7stdoutwriteflush)reasonmessages r_terminal_beeprHVsmAG QP&PPP### g#JTJr2c(|dkrtjgStdtd|}|dkr |dz dz}n8|dkrd|dz dz dzz}n#|dkrd |dz dz dzz}nd |dz dz d zz}|}d \}}}d }tjd|t ||zd}tj|dztjz|zdztj|dztjz|zdzztj|dztjz|zdzz}tjtj|} | dkr|| z d z}d} | |}dd} | |}t||dzkr>t d|z} tj |} |d| dz| | d<|| z}||zdz}tjtj|} | dkr|| z dz}|S)z2Generate pleasant C major chord notification soundrdg9@?rIg?Kg?g?g?)gGzZp@gGzt@gx@皙?Fg?g333333?皙?c t|}td|z}td|z}td|z}tj|}|dkr.tjtjdd|d|d|<|dkr"|||z}}tjdd||||<|dkr8||z }dtjdtjdd|zz||d<||zS) NrQg?rMr333333?)lenintnponespowerlinspaceexp) wave_datalength attack_time decay_time release_timeenvelope start_idxend_idxs rapply_adsr_envelopez8_generate_notification_beep..apply_adsr_envelopesY#,'' v&& 3<(( 76?? ??%'Xbk!Q .L.La%P%PH\k\ " >>!,kJ.FwI*,+az*J*JHYw& ' !  -I#'"&bk!Q 6U6U1U*V*V#VHYZZ 8##rc*tdtt|dz}|dzdkr|dz }tj||z }tj||dzd}tj||d}||dz| dzS) NrTgMbP?rPrrSedge)modesame)maxrXrWrYrZpadconvolve)signal cutoff_ratio window_sizekernelpaddedfiltereds rsimple_lowpass_filterz:_generate_notification_beep..simple_lowpass_filters!SVu!45566 ?a   1 K%% 3 q 0v>>>;vvF;;; q(K<1+<<==rgQ?Ng{Gz?rU)rO) rYarrayrkminr\rXsinpiabsrW zeros_like)volume sample_rate volume_mappedfreq_cfreq_efreq_gdurationtwave max_amplituderfrt delay_samplesreverbs r_generate_notification_beepras| {{x|| CV$$ % %F||$#- 2 t3t;; 2" 4<<  t3t;; F4FFFH Ax[8%;!>>> !  & &D 4yy;!###D;.// t$$!%o ~o!6!=}~~f} &=3 DF26$<<((Mtm#d* Krc\|tvrt|t|<t|S)z!Return cached waveform for volume)_cached_waveformsr)r|s r_get_cached_waveformrs, &&&$?$G$G&! V $$rc ddl}t5ts9|j|ddd|jda|j}| |ddkr ddddS|d ztj }t|j d kr| }tj t|dftj }||dddf<||ddd f<|j|}|j|t%t||z d zd z}|j| ddddS#1swxYwYdS#t*$rYdSt,$r}t/d|Yd}~dSd}~wwxYw)zPlay audio with pygame backendrNirPi) frequencysizechannelsbufferTFrS)dtypeirIzPygame error: )pygame _mixer_lock_mixer_initializedmixerpre_initinitget_initastyperYint16rWshapeflattenzerossndarray make_soundstopplayrXtimewait ImportError ExceptionrB) audio_datar}r mixer_info audio_int16 stereo_datasound duration_mses rplay_audio_with_pygamersh"   % * %% #PQZ]%^^^ !!###%)"..00J!Z]a%7%7        &-55bh??K;$%%)))1133 (C $4$4a#8IIIK +K1  +K1 O..{;;E L      JJLLLc*oo ;dBCCbHK K  [ ) ) )3                  6 uu  "q""###uuuuusN F3A)F&6 F3DF& F3&F**F3-F*.F33 G& G& G!!G&c ddl}||||dS#t$rYdSt$r}t d|Yd}~dSd}~wwxYw)z$Play audio using sounddevice backendrNTFzSounddevice error: ) sounddevicerrrrrB)rr}sdrs rplay_audio_with_sounddevicers       K(((  t uu  'A''(((uuuuus.2 A% A%A  A%ctjdkrdS ddl}ddl}ddl}ddl}|}tj |d| j d} | |d5}| d|d |||d zt"j} || dddn #1swxYwY|||j tj|rtj|nH#YnDxYw# tj|rtj|ww#YwxYwxYwd S#t2$rYdSt4$r} t7d | Yd} ~ dSd} ~ wwxYw) z0Play audio using winsound backend (Windows only)win32FrN notification_z.wavwrSrPrTzWinsound error: )r7r8winsoundrtempfileuuid gettempdirr pathjoinuuid4hexopen setnchannels setsampwidth setframeraterrYr writeframestobytes PlaySound SND_FILENAMEr'unlinkrrrB) rr}rrrrtemp_dir temp_filenamewav_filerrs rplay_audio_with_winsoundrsb |wu----------------&&(( X/Utzz||?O/U/U/UVV  =#.. <(%%a(((%%a(((%%k222)E199"(CC $$[%8%8%:%:;;;  < < < < < < < < < < < < < < <   }h.C D D D 7>>-00-Im,,,   7>>-00-Im,,,,- t uu  $$$%%%uuuuusAG4F B D FD##F&D#'F3E:9G:E><GF?3F86F?8F<:F??G G7 G7G22G7c@|dkrdSt|}t|dkrdStstpd}t |dSt t tg}|D]!} ||rdS#t$rYwxYwda t ddS)z-Play notification sound with specified volumerNzaudio backends unavailableTzall audio backends failed) rrWr>r=rHrrrrr;)r|rrFaudio_backendsbackends rplay_notification_soundrs {{%f--J :! % ' '&F*Fv,.IKcdN! wz""      H "./////s. A== B  B cbfd}tj|ddS)z5Play notification sound asynchronously (non-blocking)c| tdS#t$r}td|Yd}~dSd}~wwxYw)Nz"Error playing notification sound: )rrrB)rr|s r play_soundz+play_notification_async..play_sound2sd < #F + + + + + < < < :q:: ; ; ; ; ; ; ; ; ; rHrrrrrrrrvolrrrrs +. '( in ))) 111 " "677#)$:;;<       (*|VVVVp%%%4@%%%%N9E    6B    D00004====$$$$ Cr