\jddlmZddlZddlmZddlmZddlm Z GddeZ ejejGdd eZ ejejGd d Zy) )IntEnumN)libav)SideData)uint8_tcleZdZejZej ZejZ ejZ y)VideoEncParamsTypeN) __name__ __module__ __qualname__libAV_VIDEO_ENC_PARAMS_NONENONEAV_VIDEO_ENC_PARAMS_VP9VP9AV_VIDEO_ENC_PARAMS_H264H264AV_VIDEO_ENC_PARAMS_MPEG2MPEG2H/root/.hermes/venv/lib/python3.12/site-packages/av/sidedata/encparams.pyrr s.  ' 'D % %C  ' 'D  ) )Errc~eZdZdZedZedZedZedZedZ edZ dZ d Z y ) VideoEncParamscVd|jd|jd|jdS)Nz') nb_blocks codec_typeqpselfs r__repr__zVideoEncParams.__repr__s488H VZVeVeUffklplslskttuvvrctjtjtj|j j jS)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 AVVideoEncParamsptrdatarrs rrzVideoEncParams.nb_blockss6{{ NN3// 0$((-- ) rctjtjtj|j j jS)zk Offset in bytes from the beginning of this structure at which the array of blocks starts. )r#r$r%r r&r'r( blocks_offsetrs rr*zVideoEncParams.blocks_offset"s6 {{ NN3// 0$((-- - rctjtjtj|j j jS)zX Size of each block in bytes. May not match sizeof(AVVideoBlockParams). )r#r$r%r r&r'r( block_sizers rr,zVideoEncParams.block_size+s6 {{ NN3// 0$((-- * rctjtjtj|j j j}ttjtj|S)zH Type of the parameters (the codec they are used with). ) r#r$r%r r&r'r(typerint)r ts rrzVideoEncParams.codec_type4sR '-kk NN3// 0$((--' $ "&++fjj!"<==rctjtjtj|j j jS)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. )r#r$r%r r&r'r(rrs rrzVideoEncParams.qp>s2{{6>>#*>*>?ORRRrc .tjtjtj|j j }tdDcgc]+}tdDcgc]}|j||c}-c}}Scc}wcc}}w)z Quantisation parameter offset from the base (per-frame) qp for a given plane (first index) and AC/DC coefficients (second index). ) r#r$r%r r&r'r(rangedelta_qp)r pijs rr6zVideoEncParams.delta_qpHsh 39++ NN3// 0$((--3 ?DAhG584aAq!4GG4GsB*B B BcX|dks||jk\r tdt||S)z? Get the encoding parameters for a given block rz$Expected idx in range [0, nb_blocks))r ValueErrorVideoBlockParams)r idxs r block_paramszVideoEncParams.block_paramsSs/ 7cT^^+CD Dc**rcPddl}|jjjdzdz}|jjjdzdz}||z}t j t j\}}}|j|k7r tdt jt jtj|jjj}|tj j"k7r(|tj j$k7r t'd|j)||f|j*} t-|D]J}t/||} | j0dz}| j2dz}|j4| j6z| ||f<L| 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'heightwidthr#declarer/r RuntimeErrorr$r%r r&r(r.AVVideoEncParamsTyperrr;emptyint32r5r<src_ysrc_xrr6) r npmb_hmb_wnb_mb block_idxxyr.mapblocks rqp_mapzVideoEncParams.qp_map]sd  JJNN11B62= JJNN0025"< 4K ..4 1a >>U "L *0 NN3// 0$((--* $  C,,FF F00III56 6hhd|288h4u 1I$T95E r!A r!A%..0C1I  1  rN) r r r r!propertyrr*r,rrr6r>rWrrrrrsw  >>SSHH+%rrceZdZdedej ddfdZdZedZ edZ ed Z ed Z ed Z y) r<video_enc_paramsr=returnNc4tjtjt|jj }|j ||jzz}tjtjtj||z|_y)N) r#r$r%rr'r(r*r,r AVVideoBlockParams)r rZr=baseoffsets r__init__zVideoBlockParams.__init__ss(. NN7 #%5%9%9%>%>)   * *S3C3N3N-N N ;;v~~c.D.DEtf}Urc d|jd|jd|jd|jd|jd S)Nz$rqsx (9/**pXppf3!3!3!r