dc/db3/ntv2encodehevc_8h_source.html

/* SPDX-License-Identifier: MIT */

#ifndef _NTV2ENCODEHEVC_H

#define _NTV2ENCODEHEVC_H


#include "ntv2enums.h"

#include "ntv2devicefeatures.h"

#include "ntv2devicescanner.h"

#include "ntv2demohevccommon.h"


#include "ntv2m31enums.h"


#include "ajabase/common/videotypes.h"

#include "ajabase/common/circularbuffer.h"

#include "ajabase/common/timebase.h"

#include "ajabase/common/timecode.h"

#include "ajabase/common/timecodeburn.h"

#include "ajabase/system/thread.h"


#include "ntv2m31.h"


#define VIDEO_RING_SIZE         60

#define AUDIO_RING_SIZE         (3*VIDEO_RING_SIZE)


class NTV2EncodeHEVC

{

    //  Public Instance Methods

    public:

        NTV2EncodeHEVC (const std::string           inDeviceSpecifier   = "0",

                        const NTV2Channel           inChannel           = NTV2_CHANNEL1,

                        const M31VideoPreset        inM31Preset         = M31_FILE_1280X720_420_8_5994p,

                        const NTV2FrameBufferFormat inPixelFormat       = NTV2_FBF_10BIT_YCBCR_420PL2,

                        const bool                  inQuadMode          = false,

                        const uint32_t              inAudioChannels     = 0,

                        const bool                  inTimeCodeBurn      = false,

                        const bool                  inInfoMode          = false,

                        const bool                  inTsiMode           = false,

                        const uint32_t              inMaxFrames         = 0xffffffff);


        virtual                 ~NTV2EncodeHEVC ();


        virtual AJAStatus       Init (void);


        virtual AJAStatus       Run (void);


        virtual void            Quit (void);


        virtual void            GetStatus (AVHevcStatus & outStatus);


        virtual M31VideoPreset  GetCodecPreset (void);


    //  Protected Instance Methods

    protected:

        virtual AJAStatus       SetupVideo (void);


        virtual AJAStatus       SetupAudio (void);


        virtual void            RouteInputSignal (void);


        virtual void            SetupHostBuffers (void);


        virtual void            SetupAutoCirculate (void);


        virtual void            StartVideoInputThread (void);


        virtual void            StartVideoProcessThread (void);


        virtual void            StartCodecRawThread (void);


        virtual void            StartCodecHevcThread (void);


        virtual void            StartVideoFileThread (void);


        virtual void            StartAudioFileThread (void);


        virtual void            VideoInputWorker (void);


        virtual void            VideoProcessWorker (void);


        virtual void            CodecRawWorker (void);


        virtual void            CodecHevcWorker (void);


        virtual void            VideoFileWorker (void);


        virtual void            AudioFileWorker (void);


        virtual AJAStatus       ProcessVideoFrame (AVHevcDataBuffer * pSrcFrame, AVHevcDataBuffer * pDstFrame, uint32_t frameNumber);


        //  Protected Class Methods

    protected:

        static void             VideoInputThreadStatic (AJAThread * pThread, void * pContext);


        static void             VideoProcessThreadStatic (AJAThread * pThread, void * pContext);


        static void             CodecRawThreadStatic (AJAThread * pThread, void * pContext);


        static void             CodecHevcThreadStatic (AJAThread * pThread, void * pContext);


        static void             VideoFileThreadStatic (AJAThread * pThread, void * pContext);


        static void             AudioFileThreadStatic (AJAThread * pThread, void * pContext);


    //  Private Member Data

    private:

        AJAThread                   mACInputThread;

        AJAThread                   mVideoProcessThread;

        AJAThread                   mCodecRawThread;

        AJAThread                   mCodecHevcThread;

        AJAThread                   mVideoFileThread;

        AJAThread                   mAudioFileThread;

        CNTV2m31 *                  mM31;

        CNTV2DemoHevcCommon *       mHevcCommon;


        CNTV2Card                   mDevice;

        NTV2DeviceID                mDeviceID;

        const std::string           mDeviceSpecifier;

        bool                        mWithAudio;

        NTV2Channel                 mInputChannel;

        NTV2Channel                 mOutputChannel;

        M31Channel                  mEncodeChannel;

        M31VideoPreset              mPreset;

        NTV2InputSource             mInputSource;

        NTV2VideoFormat             mVideoFormat;

        NTV2FrameBufferFormat       mPixelFormat;

        bool                        mQuad;

        bool                        mTsi;

        bool                        m60p;

        bool                        mVif;

        bool                        mInterlaced;

        bool                        mMultiStream;

        bool                        mWithInfo;

        bool                        mWithAnc;

        NTV2AudioSystem             mAudioSystem;

        NTV2EveryFrameTaskMode      mSavedTaskMode;

        uint32_t                    mNumAudioChannels;

        uint32_t                    mFileAudioChannels;

        uint32_t                    mMaxFrames;


        bool                        mLastFrame;

        bool                        mLastFrameInput;

        bool                        mLastFrameRaw;

        bool                        mLastFrameHevc;

        bool                        mLastFrameVideo;

        bool                        mLastFrameAudio;

        bool                        mGlobalQuit;

        uint32_t                    mQueueSize;

        uint32_t                    mVideoBufferSize;

        uint32_t                    mPicInfoBufferSize;

        uint32_t                    mEncInfoBufferSize;

        uint32_t                    mAudioBufferSize;


        AVHevcDataBuffer                        mACInputBuffer [VIDEO_RING_SIZE];

        AJACircularBuffer <AVHevcDataBuffer *>  mACInputCircularBuffer;


        AVHevcDataBuffer                        mVideoRawBuffer [VIDEO_RING_SIZE];

        AJACircularBuffer <AVHevcDataBuffer *>  mVideoRawCircularBuffer;


        AVHevcDataBuffer                        mVideoHevcBuffer [VIDEO_RING_SIZE];

        AJACircularBuffer <AVHevcDataBuffer *>  mVideoHevcCircularBuffer;


        AVHevcDataBuffer                        mAudioInputBuffer [AUDIO_RING_SIZE];

        AJACircularBuffer <AVHevcDataBuffer *>  mAudioInputCircularBuffer;


        uint32_t                    mVideoInputFrameCount;

        uint32_t                    mVideoProcessFrameCount;

        uint32_t                    mCodecRawFrameCount;

        uint32_t                    mCodecHevcFrameCount;

        uint32_t                    mVideoFileFrameCount;

        uint32_t                    mAudioFileFrameCount;


        AJATimeBase                 mTimeBase;

        AJATimeCode                 mTimeCode;

        AJATimeCodeBurn             mTimeCodeBurn;


};  //  NTV2EncodeHEVC


#endif  //  _NTV2ENCODEHEVC_H