o 2j@sddlmZddlZddlmZddlmZddlm Z GdddeZ ej ej Gdd d eZ ej ej Gd d d ZdS) )IntEnumN)libav)SideData)uint8_tc@s$eZdZejZejZejZ ej Z dS)VideoEncParamsTypeN) __name__ __module__ __qualname__libAV_VIDEO_ENC_PARAMS_NONENONEAV_VIDEO_ENC_PARAMS_VP9VP9AV_VIDEO_ENC_PARAMS_H264H264AV_VIDEO_ENC_PARAMS_MPEG2MPEG2rrP/var/www/html/whisper/venv/lib/python3.10/site-packages/av/sidedata/encparams.pyr s  rc@sleZdZddZeddZeddZeddZed d Zed d Z ed dZ ddZ ddZ dS)VideoEncParamscCsd|jd|jd|jdS)Nz') nb_blocks codec_typeqpselfrrr__repr__szVideoEncParams.__repr__cCttjtj|jjjS)z Number of blocks in the array May be 0, in which case no per-block information is present. In this case the values of blocks_offset / block_size are unspecified and should not be accessed. )cythoncastpointerr AVVideoEncParamsptrdatarrrrrrs zVideoEncParams.nb_blockscCr)zk Offset in bytes from the beginning of this structure at which the array of blocks starts. )rrr r r!r"r# blocks_offsetrrrrr$" zVideoEncParams.blocks_offsetcCr)zX Size of each block in bytes. May not match sizeof(AVVideoBlockParams). )rrr r r!r"r# block_sizerrrrr&+r%zVideoEncParams.block_sizecCs,ttjtj|jjj}tttj |S)zH Type of the parameters (the codec they are used with). ) rrr r r!r"r#typerint)rtrrrr4s zVideoEncParams.codec_typecCr)a Base quantisation parameter for the frame. The final quantiser for a given block in a given plane is obtained from this value, possibly combined with `delta_qp` and the per-block delta in a manner documented for each type. )rrr r r!r"r#rrrrrr>szVideoEncParams.qpcs.ttjtj|jjfddtdDS)z Quantisation parameter offset from the base (per-frame) qp for a given plane (first index) and AC/DC coefficients (second index). cs$g|]fddtdDqS)csg|] }j|qSr)delta_qp).0j)iprr Qsz6VideoEncParams.delta_qp...)range)r+r.)r-rr/Qs$z+VideoEncParams.delta_qp..)rrr r r!r"r#r1rrr2rr*HszVideoEncParams.delta_qpcCs$|dks ||jkr tdt||S)z? Get the encoding parameters for a given block rz$Expected idx in range [0, nb_blocks))r ValueErrorVideoBlockParams)ridxrrr block_paramsSs zVideoEncParams.block_paramsc Csddl}|jjjdd}|jjjdd}||}ttj\}}}|j|kr,t dt tj t j |jjj}|t jjkrI|t jjkrItd|j||f|jd} t|D]}t||} | jd}| jd}|j| j| ||f<qW| S)z Convenience method that creates a 2-D map with the quantization parameters per macroblock. Only for MPEG2 and H264 encoded videos. rNz:Expected frame size to match number of blocks in side datazExpected MPEG2 or H264)dtype)numpyframer"heightwidthrdeclarer(r RuntimeErrorrr r r!r#r'AVVideoEncParamsTyperrr4emptyint32r1r5src_ysrc_xrr*) rnpmb_hmb_wnb_mb block_idxxyr'mapblockrrrqp_map]s0       zVideoEncParams.qp_mapN) rrr rpropertyrr$r&rrr*r7rOrrrrrs        rc@sheZdZdedejddfddZddZed d Z ed d Z ed dZ eddZ eddZ dS)r5video_enc_paramsr6returnNcCsDttjt|jj}|j||j}ttjtj |||_dS)N) rrr rr"r#r$r&r AVVideoBlockParams)rrQr6baseoffsetrrr__init__s zVideoBlockParams.__init__c Cs.d|jd|jd|jd|jd|jd S)Nz$s    s