o 2j1 @s4ddlmZddlZddlmmZddlmZddl m Z m Z ddl m Z GdddeZGdd d eZGd d d eZGd d d eZGdddeZejejdededejdejfddZejejdedejdejfddZejdeejdejdejfddZejejGdddZdS) )IntEnumN) err_check) VideoFormat get_pix_fmt)alloc_video_framec@seZdZUeZded<eZded<eZ ded<e Z ded<e Z d ed <eZd ed <eZd ed<eZded<eZded<eZded<eZded<dS) Interpolationz Fast bilinear FAST_BILINEARBilinearBILINEARz2-tap cubic B-splineBICUBIC ExperimentalXzNearest neighbor / pointPOINTzArea averagingAREAzBicubic luma / Bilinear chromaBICUBLINzGaussian approximationGAUSSzUnwindowed SincSINCz3-tap sinc/sincLANCZOSzUnwindowed natural cubic splineSPLINEN)__name__ __module__ __qualname__SWS_FAST_BILINEARr__annotations__ SWS_BILINEARr SWS_BICUBICr SWS_Xr SWS_POINTrSWS_AREAr SWS_BICUBLINr SWS_GAUSSrSWS_SINCr SWS_LANCZOSr SWS_SPLINErr$r$O/var/www/html/whisper/venv/lib/python3.10/site-packages/av/video/reformatter.pyr s          rc@sDeZdZeZeZeZe Z e Z e ZeZeZeZeZe Ze Ze ZeZdS) ColorspaceN)rrr SWS_CS_ITU709ITU709 SWS_CS_FCCFCC SWS_CS_ITU601ITU601 SWS_CS_ITU624ITU624SWS_CS_SMPTE170M SMPTE170MSWS_CS_SMPTE240M SMPTE240MSWS_CS_DEFAULTDEFAULTitu709fccitu601itu624 smpte170m smpte240mdefaultr$r$r$r%r&sr&c@sFeZdZUejZded<ejZded<ej Z ded<ej Z ded<d S) ColorRange Unspecified UNSPECIFIEDz%MPEG (limited) YUV range, 219*2^(n-8)MPEGzJPEG (full) YUV range, 2^n-1JPEGzNot part of ABINBN) rrrlibAVCOL_RANGE_UNSPECIFIEDr>rAVCOL_RANGE_MPEGr?AVCOL_RANGE_JPEGr@AVCOL_RANGE_NBrAr$r$r$r%r<*s r<c@seZdZUdZejZded<ejZ ded<ej Z ded<ej Z ded <ejZd ed <ejZd ed <ejZded<ejZded<ejZded<ejZded<ejZded<ejZded<ejZded<ej Z!ded<ej"Z#ded<ej$Z%d ed!<ej&Z'd"ed#<d$S)%ColorTrczuTransfer characteristic (gamma curve) of a video frame. Maps to FFmpeg's ``AVColorTransferCharacteristic``. zBT.709BT709r=r>zGamma 2.2 (BT.470M)GAMMA22zGamma 2.8 (BT.470BG)GAMMA28z SMPTE 170Mr0 SMPTE 240Mr2LinearLINEARzLogarithmic (100:1 range)LOGz"Logarithmic (100*sqrt(10):1 range)LOG_SQRTzIEC 61966-2-4 (sRGB) IEC61966_2_4zBT.1361 extended colour gamut BT1361_ECGzIEC 61966-2-1 (sYCC) IEC61966_2_1zBT.2020 10-bit BT2020_10zBT.2020 12-bit BT2020_12zSMPTE 2084 (PQ, HDR10) SMPTE2084z SMPTE 428-1SMPTE428zARIB STD-B67 (HLG) ARIB_STD_B67N)(rrr__doc__rBAVCOL_TRC_BT709rHrAVCOL_TRC_UNSPECIFIEDr>AVCOL_TRC_GAMMA22rIAVCOL_TRC_GAMMA28rJAVCOL_TRC_SMPTE170Mr0AVCOL_TRC_SMPTE240Mr2AVCOL_TRC_LINEARrM AVCOL_TRC_LOGrNAVCOL_TRC_LOG_SQRTrOAVCOL_TRC_IEC61966_2_4rPAVCOL_TRC_BT1361_ECGrQAVCOL_TRC_IEC61966_2_1rRAVCOL_TRC_BT2020_10rSAVCOL_TRC_BT2020_12rTAVCOL_TRC_SMPTE2084rUAVCOL_TRC_SMPTE428rVAVCOL_TRC_ARIB_STD_B67rWr$r$r$r%rG1s& rGc@seZdZUdZejZded<ejZ ded<ej Z ded<ej Z ded <ejZd ed <ejZd ed <ejZded<ejZded<ejZded<ejZded<ejZded<ejZded<dS)ColorPrimarieszRColor primaries of a video frame. Maps to FFmpeg's ``AVColorPrimaries``. zBT.709 / sRGB / sYCCrHr=r>zBT.470MBT470MzBT.470BG / BT.601-6 625BT470BGzSMPTE 170M / BT.601-6 525r0rKr2zGeneric film (Illuminant C)FILMzBT.2020 / BT.2100BT2020zSMPTE 428-1 / XYZrVzSMPTE 431-2 (DCI-P3)SMPTE431zSMPTE 432-1 (Display P3)SMPTE432zEBU 3213-E / JEDEC P22EBU3213N)rrrrXrBAVCOL_PRI_BT709rHrAVCOL_PRI_UNSPECIFIEDr>AVCOL_PRI_BT470MrkAVCOL_PRI_BT470BGrlAVCOL_PRI_SMPTE170Mr0AVCOL_PRI_SMPTE240Mr2AVCOL_PRI_FILMrmAVCOL_PRI_BT2020rnAVCOL_PRI_SMPTE428rVAVCOL_PRI_SMPTE431roAVCOL_PRI_SMPTE432rpAVCOL_PRI_EBU3213rqr$r$r$r%rjJs rjvalue enum_classr;returncCsT|dur|St||r|jSt|tr|St|tr||jStd|d|j)NzCannot convert z to ) isinstancer~intstr ValueErrorr)r~rr;r$r$r%_resolve_enum_value^s    rformatcCs,|dur|St|trtt|jSt|SN)rrcythoncastpix_fmtr)rr;r$r$r%_resolve_formatos  rframe colorspace color_rangecCsn|tjkr ttj||_|tkrtj|_dS|t kr!tj |_dS|t kr+tj |_dS|t kr5tj|_dSdS)zCSet AVFrame colorspace/range from SWS_CS_* and AVColorRange values.N)rBrCrr AVColorRangerr'AVCOL_SPC_BT709rr) AVCOL_SPC_FCCr+AVCOL_SPC_SMPTE170Mr1AVCOL_SPC_SMPTE240M)rrrr$r$r%_set_frame_colorspaceys     rc@seZdZdZddZ           ddefddZejdedej d ej d e j d ej d ej d ej dej dej dej dej dej dej dej fddZ dS)VideoReformatterzAn object for reformatting size and pixel format of :class:`.VideoFrame`. It is most efficient to have a reformatter object for each set of parameters you will use as calling :meth:`reformat` will reconfigure the internal object. cCs<tjtt|jWddS1swYdSr)rnogilsws_free_contextaddressptr)selfr$r$r% __dealloc__s"zVideoReformatter.__dealloc__Nrc Cst||jj} t|t|jj}t|t|jj}t|tt}t|t d}t| t d}t| t d}t| t d}| dur;| nd}|durC|n|jj }|durM|n|jj }| du}| du}||||| ||||||||||S)awCreate a new :class:`VideoFrame` with the given width/height/format/colorspace. Returns the same frame untouched if nothing needs to be done to it. :param int width: New width, or ``None`` for the same width. :param int height: New height, or ``None`` for the same height. :param format: New format, or ``None`` for the same format. :type format: :class:`.VideoFormat` or ``str`` :param src_colorspace: Current colorspace, or ``None`` for the frame colorspace. :type src_colorspace: :class:`Colorspace` or ``str`` :param dst_colorspace: Desired colorspace, or ``None`` for the frame colorspace. :type dst_colorspace: :class:`Colorspace` or ``str`` :param interpolation: The interpolation method to use, or ``None`` for ``BILINEAR``. :type interpolation: :class:`Interpolation` or ``str`` :param src_color_range: Current color range, or ``None`` for the ``UNSPECIFIED``. :type src_color_range: :class:`ColorRange` or ``str`` :param dst_color_range: Desired color range, or ``None`` for the ``UNSPECIFIED``. :type dst_color_range: :class:`ColorRange` or ``str`` :param dst_color_trc: Desired transfer characteristic to tag on the output frame, or ``None`` to preserve the source frame's value. This sets frame metadata only; it does not perform a pixel-level transfer function conversion. :type dst_color_trc: :class:`ColorTrc` or ``int`` :param dst_color_primaries: Desired color primaries to tag on the output frame, or ``None`` to preserve the source frame's value. :type dst_color_primaries: :class:`ColorPrimaries` or ``int`` :param int threads: How many threads to use for scaling, or ``0`` for automatic selection based on the number of available CPUs. Defaults to ``0`` (auto). rN)rrrrr&rrrrr<rGrjwidthheight _reformat)rrrrrsrc_colorspacedst_colorspace interpolationsrc_color_rangedst_color_range dst_color_trcdst_color_primariesthreads c_dst_formatc_src_colorspacec_dst_colorspacec_interpolationc_src_color_rangec_dst_color_rangec_dst_color_trcc_dst_color_primaries c_threadsc_widthc_heightset_dst_color_trcset_dst_color_primariesr$r$r%reformatsJ,     zVideoReformatter.reformatrr dst_formatrrrrrrrrrrcCs|jjdkr tdttj|jj}||kr1||jjkr1||jjkr1||kr1|| kr1| s1| s1|S|jj rVt }t t |j|jd|j |_ ||}ttj|jj}|jtjkrjt|_|jtjkrjtd||j_ttj||j_t }|||||||jj}|jj}t|j||t|j|| tjt|j|j|j}Wdn1swY||j_||j_t || rttj| |j_| rttj| |j_|S)NrzFrame does not have format set.zCould not allocate SwsContext) rrrrrrB AVPixelFormatrr hw_frames_ctxrrav_hwframe_transfer_datapts_init_user_attributesNULLsws_alloc_context MemoryErrorruintflags_copy_internal_attributes_initrrrrsws_scale_frameAVColorTransferCharacteristic color_trcAVColorPrimariescolor_primaries)rrrrrrrrrrrrrrr src_formatframe_sw new_frameframe_src_colorspaceframe_src_color_rangeretr$r$r%rs`      zVideoReformatter._reformat) NNNNNNNNNNN)rrrrXr VideoFramerrcfuncrrBrbintrr$r$r$r%rs^ U     r) enumrrcython.cimports.libavcimportslibavrBcython.cimports.av.errorrcython.cimports.av.video.formatrrcython.cimports.av.video.framerrr&r<rGrjrinlineobjectrrrrpointerAVFramerfinalcclassrr$r$r$r%sH