i( xddlZddlZddlmZmZddlmZddlmZddl m Z m Z ddl m Z ddlmZdd lmZdd lmZeZe j*Zej.j1d d Zej.j5d d Zej.j5d dZej.j5d dZej.j1d dZej.j5d dZdee fdZ!de fdZ"de de fdZ#dde deee e fde de$fdZ%de deee e fde fdZ&y)N)ListDict)Console)config_manager) binary_pathsget_ffmpeg_path)need_to_force_to_ts)check_duration_v_a)fix_subtitle_extension)capture_ffmpeg_real_timeM3U8_CONVERSIONuse_gpu param_video param_audio param_finalsubtitle_dispositionsubtitle_disposition_language ffmpeg_cmdctr|jty|jt|jty)z Add encoding parameters to the ffmpeg command. Parameters: ffmpeg_cmd (List[str]): List of the FFmpeg command to modify N) PARAM_FINALextend PARAM_VIDEO PARAM_AUDIO)rs ]/home/nidran/Documenti/script/streaming-community/StreamingCommunity/core/processors/merge.pyadd_encoding_paramsr#s1+&+&+&returnc tdk(r5tjdgddd}|jj }ntdk(r7 tjgdddd}|jj }n@td k(r6tjd d gddd}|jj }nyd |vry d|vryd|vsd|vryy#tj t f$rZ tjgdddd}|jj }n #tj t f$rYYywxYwYwxYw#tj t f$rYywxYw)z Detects the GPU device type available on the system. Returns: str: The type of GPU device detected ('cuda', 'vaapi', 'qsv', or 'none'). linuxlspciT)capture_outputtextcheckwindows)wmicpathwin32_videocontrollergetname) powershellz-CommandzHGet-WmiObject win32_videocontroller | Select-Object -ExpandProperty Namenonedarwinsystem_profilerSPDisplaysDataTypenvidiacudaintelvaapiamdati)os_type subprocessrunstdoutlowerCalledProcessErrorFileNotFoundError)resultoutputs rdetect_gpu_device_typer?1s! g ^^WIdUYZF]]((*F  ! "#(`qu}AIMN,,. ^^%68L$M^bimuyzF]]((*F v    f_-113DE ""'^^-Sdhos{@F#]]002F"557HI"!"  ".  ) )+< =ssAE 5C?AE E E E E.5D$#E$E=E>E EEE EE E&%E& video_pathout_pathc`tg}tr5t}tj d||j d|gt |r|j ddg|j d|gt||j |dgt|d}t ||fS)z Mux video file using FFmpeg. Parameters: - video_path (str): The path to the video file. - out_path (str): The path to save the output file. z7[yellow]FFMPEG [cyan]Detected GPU for video join: [red]-hwaccelz-fmpegts-i-yz[yellow]FFMPEG [cyan]Join video) rUSE_GPUr?consoleprintrr rr )r@rArgpu_type_hwaccel result_jsons r join_videorL\s"#$J13 OP`Oabc:'789:&4*+tZ() #x&'+:7XYK G [  r audio_trackslimit_duration_diffcd}|D]}|jd}|jdd}t||\}} } } tjd|d|d| dd | dd | dd | |kDsetjd |d | dd|dd}t g} t r| j dtg| j d|gt|D]'\} }| j d|jdg)| j ddgtdt|dzD]} | j d| dgt|D]Y\} }|jdd}| j d| d|g| j d| d|jddg[t| |r| j gd| j |dgt| d}t|||fS)a Joins audio tracks with a video file using FFmpeg. Parameters: - video_path (str): The path to the video file. - audio_tracks (list[dict[str, str]]): A list of dictionaries containing information about audio tracks. Each dictionary should contain the 'path' and 'name' keys. - out_path (str): The path to save the output file. - limit_duration_diff (float): Maximum duration difference in seconds. Fr'r*unknownz$[yellow] - [cyan]Audio lang [red], [cyan]Path: [red]z, [cyan]Video duration: [red]z.2fzs, [cyan]Audio duration: [red]zs, [cyan]Diff: [red]sz)[yellow] WARN [cyan]Audio lang: [red]'z*' [cyan]has a duration difference of [red]z(s [cyan]which exceeds the limit of [red]zs.TrCrE-map0:vr z:az-metadata:s:a:z language=ztitle=)z -shortestz-strict experimentalrFz[yellow]FFMPEG [cyan]Join audio) r)r rHrIrrGrr? enumeraterangelenrr )r@rMrArN use_shortest audio_track audio_path audio_lang_diffvideo_durationaudio_durationri lang_coderKs r join_audiosrcsL#   __V,  __VY7 2DZQ[2\/4 )* "#T:6J!!(+A.446JV# v #/tXf-.. /00J#>2X X||FHLLY,OP  ? ~M`aiamamntvbAaBC Dv#'"~.. )005z7H7H7VWW Xvvvvv~NOS():_SE2C89:N 3a 7'~6 MC$LLR8>>@M5 "jjl002  K/=3K3KK3XMM$WX`XdXdeoXpWq"rs&)O  *   &   0ABIN O8T""J+:7[\K G [  r))'rmr7typingrr rich.consolerStreamingCommunity.utilsrStreamingCommunity.setuprrhelper.ex_videor helper.ex_audior helper.ex_subr capturer rH_detect_systemr6configget_boolrGget_listrrrrprqstrrr?rLfloatrcr~rrrsp !4B1/1- ) %, % % '    ( ():I F##,,-> N ##,,-> N ##,,-> N %,,556GI_` . 5 5 > >?PRq r 'DI '((V!!3!!#!!HB/CB/tDcN/CB/sB/inB/JM!sM!Dc3h4HM!TWM!r