ntv2encodehevc.h

Go to the documentation of this file.

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