ntv2encodehevcvif.h

Go to the documentation of this file.

/* SPDX-License-Identifier: MIT */
#ifndef _NTV2ENCODEHEVCVIF_H
#define _NTV2ENCODEHEVCVIF_H

#include "ntv2enums.h"
#include "ntv2devicefeatures.h"
#include "ntv2devicescanner.h"
#include "ntv2demohevccommon.h"
//#include "ajacircularbuffer.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 NTV2EncodeHEVCVif
{
    //  Public Instance Methods
    public:
        NTV2EncodeHEVCVif ( const std::string           inDeviceSpecifier   = "0",
                            const M31VideoPreset        inM31Preset         = M31_FILE_1280X720_420_8_5994p,
                            const NTV2FrameBufferFormat inPixelFormat       = NTV2_FBF_10BIT_YCBCR_420PL2,
                            const uint32_t              inAudioChannels     = 0,
                            const bool                  inInfoData          = false,
                            const uint32_t              inMaxFrames         = 0xffffffff);

        virtual                 ~NTV2EncodeHEVCVif ();

        virtual AJAStatus       Init (void);

        virtual AJAStatus       Run (void);

        virtual void            Quit (void);

        virtual void            GetStatus (AVHevcStatus * outInputStatus);

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

        virtual void            StartAVFileThread (void);

        virtual void            VideoInputWorker (void);

        virtual void            CodecHevcWorker (void);

        virtual void            AVFileWorker (void);

        virtual void            TransferPictureInfo(CNTV2m31 *  pM31);

        //  Protected Class Methods
    protected:
        static void             VideoInputThreadStatic (AJAThread * pThread, void * pContext);

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

        static void             AVFileThreadStatic (AJAThread * pThread, void * pContext);
        
    //  Private Member Data
    private:
        AJAThread                   mACInputThread;         
        AJAThread                   mCodecHevcThread;       
        AJAThread                   mAVFileThread;          
        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             mInputFormat;           
        NTV2VideoFormat             mVideoFormat;           
        NTV2FrameBufferFormat       mCapturePixelFormat;    
        NTV2FrameBufferFormat       mOverlayPixelFormat;    
        NTV2FrameBufferFormat       mCodecPixelFormat;      
        NTV2FrameRate               mFrameRate;             
        NTV2AudioRate               mAudioRate;             
        bool                        mWithInfo;              
        NTV2AudioSystem             mAudioSystem;           
        NTV2EveryFrameTaskMode      mSavedTaskMode;         
        uint32_t                    mNumAudioChannels;      
        uint32_t                    mFileAudioChannels;     
        uint32_t                    mMaxFrames;             

        bool                        mLastFrame;             
        bool                        mLastFrameInput;        
        bool                        mLastFrameHevc;         
        bool                        mLastFrameVideo;        
        bool                        mGlobalQuit;            
        uint32_t                    mQueueSize;             
        uint32_t                    mVideoBufferSize;       
        uint32_t                    mPicInfoBufferSize;     
        uint32_t                    mEncInfoBufferSize;     
        uint32_t                    mAudioBufferSize;       
        uint32_t                    mOverlayBufferSize;     

        AVHevcDataBuffer                        mACInputBuffer [VIDEO_RING_SIZE];           
        AJACircularBuffer <AVHevcDataBuffer *>  mACInputCircularBuffer;                     

        AVHevcDataBuffer                        mVideoHevcBuffer [VIDEO_RING_SIZE];         
        AJACircularBuffer <AVHevcDataBuffer *>  mVideoHevcCircularBuffer;                   

        AVHevcDataBuffer *          mFrameData;                                             
        uint32_t *                  mSilentBuffer;                                          
        uint32_t                    mVideoInputFrameCount;                                  
        uint32_t                    mCodecHevcFrameCount;                                   
        uint32_t                    mAVFileFrameCount;                                      
        uint32_t                    mRawFrameCount;                                         
        uint32_t                    mInfoFrameCount;                                        

        uint32_t *                  mOverlayBuffer [2];                                     
        uint32_t                    mOverlayFrame [2];                                      
        uint32_t                    mOverlayIndex;                                          

        AJATimeBase                 mTimeBase;                                              

};  //  NTV2EncodeHEVCVIF

#endif  //  _NTV2ENCODEHEVCVIF_H