j&ddlZddlmZddlZddlmcmZddlm Z ddl m Z ddl m Z ddlmZGddeZGd d eZejeeZejeejZejd ejejejd efd ZejejGddZejdZ ejejGddZ!dS)N)IntEnum)Codec) Dictionary) err_check)get_video_formatceZdZejZejZejZ ej Z ej Z ejZejZejZejZejZejZejZejZdZdZdS) HWDeviceType N) __name__ __module__ __qualname__libAV_HWDEVICE_TYPE_NONEnoneAV_HWDEVICE_TYPE_VDPAUvdpauAV_HWDEVICE_TYPE_CUDAcudaAV_HWDEVICE_TYPE_VAAPIvaapiAV_HWDEVICE_TYPE_DXVA2dxva2AV_HWDEVICE_TYPE_QSVqsvAV_HWDEVICE_TYPE_VIDEOTOOLBOX videotoolboxAV_HWDEVICE_TYPE_D3D11VAd3d11vaAV_HWDEVICE_TYPE_DRMdrmAV_HWDEVICE_TYPE_OPENCLopenclAV_HWDEVICE_TYPE_MEDIACODEC mediacodecAV_HWDEVICE_TYPE_VULKANvulkanAV_HWDEVICE_TYPE_D3D12VAd3d12vaamfohcodecu/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib64/python3.11/site-packages/av/codec/hwaccel.pyr r s  $D  &E  $D  &E  &E "C4L*G "C  (F0J  (F*G CGGGr-r cJeZdZdZejZejZej Z ej Z dS)HWConfigMethodrN) r r rrr'AV_CODEC_HW_CONFIG_METHOD_HW_DEVICE_CTX hw_device_ctx'AV_CODEC_HW_CONFIG_METHOD_HW_FRAMES_CTX hw_frame_ctx"AV_CODEC_HW_CONFIG_METHOD_INTERNALinternal AV_CODEC_HW_CONFIG_METHOD_AD_HOCad_hocr,r-r.r0r0s8 D 3>L5H  1FFFr-r0ptrreturnc ttjtj|S#t$rYnwxYwt t }|||ttjtj|<|SN) _singletonscythoncast Py_ssize_tKeyErrorHWConfig_cinit_sentinel_init)r9configs r. wrap_hwconfigrF-s~ 6;v'8#>>??      00F LL7=K F-s334 Ms ), 99ceZdZdZejdejejej dej fdZ dZ e dZe dZe dZe d Zd S) rBc6|turtddS)NzCannot instantiate CodecContext)rC RuntimeError)selfsentinels r.__init__zHWConfig.__init__<s$ ? * *@AA A + *r-r9r:c||_dSr<)r9)rJr9s r.rDzHWConfig._init@sr-c d|jjdtj|jd|jr |jjndd|jdtj tj |j dd S)Nz) __class__r rav_hwdevice_get_type_name device_typeformatname is_supportedr>r?r@r9rJs r.__repr__zHWConfig.__repr__Fs c4>* c c89IJJ c c*.+?dk&&4 c c!- c c6<[ARTXT\5]5] b c c c r-c4t|jjSr<)r r9rSrWs r.rSzHWConfig.device_typeNsDH0111r-c8t|jjddS)Nr)rr9pix_fmtrWs r.rTzHWConfig.formatRs 0!Q777r-c4t|jjSr<)r0r9methodsrWs r.r]zHWConfig.methodsVsdh.///r-cNt|jjtjzSr<)boolr9r]rr1rWs r.rVzHWConfig.is_supportedZsDH$s'RRSSSr-N)r r rrLr>cfuncpointerconstrAVCodecHWConfigvoidrDrXpropertyrSrTr]rVr,r-r.rBrB9sBBB \>&,s/B"CD \    22X288X800X0TTXTTTr-rBcg}tj} tj|}|tjkrn5|tjt |Z|S)zkReturn the names of the hardware device types FFmpeg was built with, e.g. ``["cuda", "videotoolbox"]``.)rrav_hwdevice_iterate_typesappendrRr )resultrOs r.hwdevices_availablerj_shF5AF  )! , , ) ) )  c3LOODDEEE F Mr-cReZdZdZ d dZd dedefdZd deded efd Z d Z dS)HWAccelaHWAccel(device_type, device=None, allow_software_fallback=True, options=None, flags=None, is_hw_owned=False) Settings for hardware-accelerated decoding and encoding. Pass an instance to :func:`av.open` to decode on the device, or to :meth:`OutputContainer.add_stream ` to encode on it. :param device_type: The kind of device, e.g. ``"cuda"``, ``"vaapi"``, ``"videotoolbox"``. See :func:`hwdevices_available` for the types supported by the loaded FFmpeg. :type device_type: str or HWDeviceType :param str device: An optional device identifier, e.g. a DRI render node path (``"/dev/dri/renderD128"``) for VAAPI or a device index (``"1"``) for CUDA. Uses the default device if ``None``. :param bool allow_software_fallback: Whether decoding falls back to a software decoder when the hardware decoder cannot handle the stream. :param dict options: Options passed to ``av_hwdevice_ctx_create``. :param int flags: Flags passed to ``av_hwdevice_ctx_create``. :param bool is_hw_owned: If True, decoded frames stay on the device (for consumption via DLPack, for example) instead of being downloaded to system memory. NTFct|tr||_nht|tr't t j||_n,t|tr||_ntd||_d|_ |jtj kr|rt ||_ |dn||_ ||_ |sint||_|jtj kr"|jr|jdd|sdn||_t"j|_d|_dS)NzUnknown type for device_typer primary_ctx1) isinstancer _device_typestrintrav_hwdevice_find_type_by_name ValueError is_hw_owned device_idr_deviceallow_software_fallbackdictoptions setdefaultflagsr>NULLr9rE)rJrSdeviceryr{r}rvs r.rLzHWAccel.__init__s3 k< 0 0 = +D    S ) ) = #C$Ek$R$R S SD    S ) ) = +D  ;<< <&   1 1 1f 1 [[DN%~ttf; '>$!(;rrd7mm   1 1 1d6F 1 L # #M3 7 7 7#.QQ ; r-codec for_encodingc tj}|r|tjz}|jD]+}|jj|zs|jr|j|jkr*ntd|||_ tj }|j r|j }|}t|j}t!tjtj|j|jj||j|jdS)Nz!No supported hardware config for )rr1r3hardware_configsr9r]rqrSNotImplementedErrorrEr>r~rxencoderr{rav_hwdevice_ctx_createaddressr})rJrrsupported_methodsrEc_device device_bytes c_optionss r._initialize_hw_contextzHWAccel._initialize_hw_contexts),(S  M !L L , S SFJ&)::   V%74;L%L%L E%&Q%&Q&QRR R "(+ < $<..00L#H *4< 8 8   &tx(( &         r-r:c|jrtdt|j|j|j|j|j}||||S)z>Create a new hardware accelerator context with the given codecz$Hardware context already initialized)rSrryr{rv)r) r9rIrlrqrxryr{rvr)rJrrrets r.createzHWAccel.createsl 8 GEFF F)<$($@L(     ""5|"DDD r-cn|jr-tjtj|jdSdSr<)r9rav_buffer_unrefr>rrWs r. __dealloc__zHWAccel.__dealloc__s9 8 :  tx 8 8 9 9 9 9 9 : :r-)NTNNF)F) r r r__doc__rLrr_rrlrrr,r-r.rlrlms4 $B! ! E! ! ! ! ! F  E  '    :::::r-rl)"weakrefenumrr>cython.cimports.libavcimportslibavrcython.cimports.av.codec.codecrcython.cimports.av.dictionaryrcython.cimports.av.errorrcython.cimports.av.video.formatrr r0declareobjectrCWeakValueDictionaryr=r`rarbrcrBrFfinalcclassccallrjrlr,r-r.rs #########000000444444......<<<<<<7&22222W222!&.22fnV%@W%@%B%BCC v~fl33F&GHX!T!T!T!T!T!T!T!TH   m:m:m:m:m:m:m:m:m:m:r-