jddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZejejGd d eZdS) N)libav)Packet)Stream)avrational_to_fraction) VideoFrame)int32_t)memcpyc2eZdZdZdZejddedzfdZejdde dzfdZ ej dZ ej d Z d Zdd Zed ZedZedZedZedZdS) VideoStreamc|jd|jdt|ddSd|jd|jd|jr |jjndd|jjd|jjdt|dd S)Nz at 0xx> z, z at 0x) codec_contextindexidnameformatwidthheightselfs t/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib64/python3.11/site-packages/av/video/stream.py__repr__zVideoStream.__repr__s   %VtzVVDVVVV V >  > >TY > >#';8t{D > >;?;M;S > >!( > >024 = > > > c|dvr(tdt|jd|d|j(tdt|jd|dt |j|S)N) frameraterate'z' object has no attribute ')AttributeErrortype__name__rgetattr)rrs r __getattr__zVideoStream.__getattr__s ( ( ( KDJJ'KKDKKK    % KDJJ'KKDKKK t)4000rNframec|tj|j|}|D]}||_|jj|j_ |S)z Encode an :class:`.VideoFrame` and return a list of :class:`.Packet`. :rtype: list[Packet] .. seealso:: This is mostly a passthrough to :meth:`.CodecContext.encode`. ) _assert_has_codec_contextlibAVERROR_ENCODER_NOT_FOUNDrencode_streamptrr stream_index)rr%packetspackets rr*zVideoStream.encode"s] &&s'DEEE$++E22 5 5F!FN&*hnFJ # #rr/c^||j|S)z Decode a :class:`.Packet` and return a list of :class:`.VideoFrame`. :rtype: list[VideoFrame] .. seealso:: This is a passthrough to :meth:`.CodecContext.decode`. )r'rdecode)rr/s rr1zVideoStream.decode3s- &&(((!((000rctj||j|jjj|j_|j|jr|dSdSdSN)r_finalize_for_outputrr,ravg_frame_rate_has_display_matrix_apply_display_matrixrs rr4z VideoStream._finalize_for_output?sh#D)))   )&*&8&<&FDH #   )d.F )  & & ( ( ( ( ( * ) ) )rczdtjtz}tjtj|jjjtj|jjj tj |d}|tj krtdt|j|j|dS)N rz+could not allocate display matrix side data)cythonsizeofrr(av_packet_side_data_newaddressr,codecparcoded_side_datanb_coded_side_dataAV_PKT_DATA_DISPLAYMATRIXNULL MemoryErrorr data_display_matrix)rnsds rr7z!VideoStream._apply_display_matrixIsFM'222363N N48,< = = N48,? @ @  ) 4 4    KLL Lrw,a00000rc| d|_dSd|D}t|dkrtdtdD]}|||j|<d|_dS)aBSet the display matrix written to the container as coded side data. ``matrix`` is a sequence of 9 integers in FFmpeg's ``AV_PKT_DATA_DISPLAYMATRIX`` layout, or ``None`` to clear. Must be called before the first frame is encoded. See :meth:`set_display_rotation` for a higher-level helper. NFc,g|]}t|S)int).0vs r z2VideoStream.set_display_matrix..cs'''1A'''rr9z+display matrix must have exactly 9 elementsT)r6len ValueErrorrangerE)rmatrixvalsis rset_display_matrixzVideoStream.set_display_matrixXs~ >',D $ F''''' t99>>JKK Kq . .A&*1gD  # ##'   rFctj|jt| tj|jt |t |d|_dS)ahSet the container display matrix from a rotation and optional flips. ``degrees`` is a counter-clockwise rotation (matching the value read back from :attr:`VideoFrame.rotation`); ``hflip`` / ``vflip`` mirror after it. Together these express all eight EXIF orientations. Must be called before the first frame is encoded. TN)r(av_display_rotation_setrEfloatav_display_matrix_flipboolr6)rdegreeshflipvflips rset_display_rotationz VideoStream.set_display_rotationksT #D$85>>/JJJ "4#7ed5kkRRR#'   rcXttj|jjS)z The average frame rate of this video stream. This is calculated when the file is opened by looking at the first few frames and averaging their rate. :type: fractions.Fraction | None )rr:r=r,r5rs r average_ratezVideoStream.average_rateys!&fnTX5L&M&MNNNrcXttj|jjS)a The base frame rate of this stream. This is calculated as the lowest framerate at which the timestamps of frames can be represented accurately. See :ffmpeg:`AVStream.r_frame_rate` for more. :type: fractions.Fraction | None )rr:r=r, r_frame_raters r base_ratezVideoStream.base_rates!&fnTX5J&K&KLLLrctjtj|jtj}t tj|S)zThe guessed frame rate of this stream. This is a wrapper around :ffmpeg:`av_guess_frame_rate`, and uses multiple heuristics to decide what is "the" frame rate. :type: fractions.Fraction | None )r(av_guess_frame_rater:rBr,rr=)rvals r guessed_ratezVideoStream.guessed_rates="5 K6;  &fnS&9&9:::rctj|jj|jtj}t t j|S)aThe guessed sample aspect ratio (SAR) of this stream. This is a wrapper around :ffmpeg:`av_guess_sample_aspect_ratio`, and uses multiple heuristics to decide what is "the" sample aspect ratio. :type: fractions.Fraction | None )r(av_guess_sample_aspect_ratio containerr,r:rBrr=)rsars rsample_aspect_ratiozVideoStream.sample_aspect_ratios@"> N &+  &fnS&9&9:::rchtjtj}tjtj|jtj|j|jj |j jz|jj |j jzdttj|S)zThe guessed display aspect ratio (DAR) of this stream. This is calculated from :meth:`.VideoStream.guessed_sample_aspect_ratio`. :type: fractions.Fraction | None i) r:declarer( AVRational av_reducer=numdenrrrlrr)rdars rdisplay_aspect_ratioz VideoStream.display_aspect_ratiosnS^,, N37 # # N37 # # K  8 < < K !9!= =     &fnS&9&9:::rr3)FF)r" __module__ __qualname__rr$r:ccallrr*rr1cfuncr4r7rUr^propertyr`rcrgrlrtrJrrr r s    1 1 1 \J-\  \ 1 1Vd] 1 1 1\ 1 \))\) \ 1 1\ 1(((& ( ( ( ( O OX O M MX M ; ;X ; ; ;X ;;;X;;;rr )r:cython.cimportsrr(cython.cimports.av.packetrcython.cimports.av.streamrcython.cimports.av.utilsrcython.cimports.av.video.framercython.cimports.libc.stdintrcython.cimports.libc.stringr finalcclassr rJrrrs ((((((,,,,,,,,,,,,;;;;;;555555//////......r;r;r;r;r;&r;r;r;r;r;r