ntv2encodehevcfile.h

Go to the documentation of this file.

/* 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