d7/d9c/ntv2encodehevcfile_8h_source.html

/* SPDX-License-Identifier: MIT */

#ifndef _NTV2ENCODEHEVCFILE_H

#define _NTV2ENCODEHEVCFILE_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 RAWFILEPATH             "ElephantsDream.yuv"

#define RAWFILEWIDTH            1920

#define RAWFILEHEIGHT           1080

#define M31PRESET               M31_FILE_1920X1080_420_8_24p


class NTV2EncodeHEVCFile

{

    //  Public Instance Methods

    public:

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

                            const NTV2Channel           inChannel           = NTV2_CHANNEL1,

                            const std::string           inFilePath          = "",

                            const uint32_t              inFrameWidth        = 0,

                            const uint32_t              inFrameHeight       = 0,

                            const M31VideoPreset        inM31Preset         = M31PRESET);


        virtual                 ~NTV2EncodeHEVCFile ();


        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 void            SetupHostBuffers (void);


        virtual void            StartVideoInputThread (void);


        virtual void            StartVideoProcessThread (void);


        virtual void            StartCodecRawThread (void);


        virtual void            StartCodecHevcThread (void);


        virtual void            StartVideoFileThread (void);


        virtual void            VideoInputWorker (void);


        virtual void            VideoProcessWorker (void);


        virtual void            CodecRawWorker (void);


        virtual void            CodecHevcWorker (void);


        virtual void            VideoFileWorker (void);


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


        //  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);


    //  Private Member Data

    private:

        AJAThread                   mFileInputThread;

        AJAThread                   mVideoProcessThread;

        AJAThread                   mCodecRawThread;

        AJAThread                   mCodecHevcThread;

        AJAThread                   mVideoFileThread;

        CNTV2m31 *                  mM31;

        CNTV2DemoHevcCommon *       mHevcCommon;


        CNTV2Card                   mDevice;

        NTV2DeviceID                mDeviceID;

        const std::string           mDeviceSpecifier;

        const std::string           mFileName;

        uint32_t                    mFrameWidth;

        uint32_t                    mFrameHeight;


        NTV2Channel                 mInputChannel;

        M31Channel                  mEncodeChannel;

        M31VideoPreset              mPreset;

        NTV2VideoFormat             mVideoFormat;

        NTV2FrameBufferFormat       mPixelFormat;

        bool                        mQuad;

        bool                        mInterlaced;

        bool                        mMultiStream;

        NTV2EveryFrameTaskMode      mSavedTaskMode;


        bool                        mLastFrame;

        bool                        mLastFrameInput;

        bool                        mLastFrameRaw;

        bool                        mLastFrameHevc;

        bool                        mLastFrameVideo;

        bool                        mGlobalQuit;

        uint32_t                    mQueueSize;

        uint32_t                    mVideoBufferSize;

        uint32_t                    mPicInfoBufferSize;

        uint32_t                    mEncInfoBufferSize;


        AVHevcDataBuffer                        mFileInputBuffer [VIDEO_RING_SIZE];

        AJACircularBuffer <AVHevcDataBuffer *>  mFileInputCircularBuffer;


        AVHevcDataBuffer                        mVideoRawBuffer [VIDEO_RING_SIZE];

        AJACircularBuffer <AVHevcDataBuffer *>  mVideoRawCircularBuffer;


        AVHevcDataBuffer                        mVideoHevcBuffer [VIDEO_RING_SIZE];

        AJACircularBuffer <AVHevcDataBuffer *>  mVideoHevcCircularBuffer;


        uint32_t                    mVideoInputFrameCount;

        uint32_t                    mVideoProcessFrameCount;

        uint32_t                    mCodecRawFrameCount;

        uint32_t                    mCodecHevcFrameCount;

        uint32_t                    mVideoFileFrameCount;


        AJATimeBase                 mTimeBase;

        AJATimeCode                 mTimeCode;

        AJATimeCodeBurn             mTimeCodeBurn;


};  //  NTV2EncodeHEVCFile


#endif  //  _NTV2ENCODEHEVCFILE_H