ntv2demohevccommon.h

Go to the documentation of this file.

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

#include "ntv2rp188.h"
#include "ajabase/common/videotypes.h"
#include "ajabase/common/circularbuffer.h"
#include "ajabase/system/debug.h"
#include "ntv2democommon.h"
#include "ntv2m31enums.h"
#include "ntv2m31.h"
#include <string>

typedef struct
{
    uint32_t *      pVideoBuffer;           
    uint32_t        videoBufferSize;        
    uint32_t        videoDataSize;          
    uint32_t        videoDataSize2;         
    uint32_t *      pInfoBuffer;            
    uint32_t        infoBufferSize;         
    uint32_t        infoDataSize;           
    uint32_t        infoDataSize2;          
    uint32_t *      pAudioBuffer;           
    uint32_t        audioBufferSize;        
    uint32_t        audioDataSize;          
    uint32_t        timeCodeDBB;            
    uint32_t        timeCodeLow;            
    uint32_t        timeCodeHigh;           
    int64_t         frameTime;              
    bool            lastFrame;              
} AVHevcDataBuffer;


typedef struct
{
    ULWord          framesProcessed;
    ULWord          framesDropped;
    ULWord          bufferLevel;
} AVHevcStatus;


class CNTV2DemoHevcCommon
{
    //  Public Instance Methods
    public:
                                    CNTV2DemoHevcCommon ();
                                    ~CNTV2DemoHevcCommon ();
    
        AJA_PixelFormat             GetAJAPixelFormat(NTV2FrameBufferFormat pixelFormat);
        AJA_FrameRate               GetAJAFrameRate(NTV2FrameRate frameRate);

        AJAStatus                   CreateHevcFile(const std::string & inFileName, uint32_t maxFrames);
        void                        CloseHevcFile(void);
        void                        WriteHevcData(void* pBuffer, uint32_t bufferSize);
    
        AJAStatus                   CreateEncFile(const std::string & inFileName, uint32_t maxFrames);
        void                        CloseEncFile(void);
        void                        WriteEncData(void* pBuffer, uint32_t bufferSize);
    
        AJAStatus                   CreateAiffFile(const std::string & inFileName, uint32_t numChannels, uint32_t maxFrames, uint32_t bufferSize);
        void                        CloseAiffFile(void);
        void                        WriteAiffHeader(void);
        void                        WriteAiffData(void* pBuffer, uint32_t numChannels, uint32_t numSamples);

        AJAStatus                   CreateRawFile(const std::string & inFileName, uint32_t maxFrames);
        void                        CloseRawFile(void);
        void                        WriteRawData(void* pBuffer, uint32_t bufferSize);
    
        AJAStatus                   OpenYuv420File(const std::string & inFileName, const uint32_t width, const uint32_t height);
        void                        CloseYuv420File(void);
        AJAStatus                   ReadYuv420Frame(void* pBuffer, uint32_t numFrame);
        AJAStatus                   ConvertYuv420FrameToNV12(void* pSrcBuffer, void* pDstBuffer, uint32_t bufferSize);
        uint64_t                    YuvFileSize() { return mYuvFileSize; }
        uint32_t                    YuvNumFrames() { return mYuvNumTotalFrames; }
        uint32_t                    YuvWidth() { return mYuvFrameWidth; }
        uint32_t                    YuvHeight() { return mYuvFrameHeight; }

        uint32_t                    AlignDataBuffer(void* pBuffer, uint32_t bufferSize, uint32_t dataSize, uint32_t alignBytes, uint8_t fill);
        AJAStatus                   DetermineInputFormat(NTV2VideoFormat sdiFormat, bool quad, NTV2VideoFormat& videoFormat);

        AJAStatus                   SetupHEVC (CNTV2m31 * pM31, M31VideoPreset preset, M31Channel encodeChannel, bool multiStream, bool withInfo);

    //  Private Member Data
    private:
        FILE *                      mHevcFd;                                                
        FILE *                      mEncFd;                                                 
        FILE *                      mAiffFd;                                                
        FILE *                      mYuvFd;                                                 
        FILE *                      mRawFd;                                                 
    
        uint32_t                    mHevcFileFrameCount;                                    
        uint32_t                    mMaxHevcFrames;                                         

        uint32_t                    mEncFileFrameCount;                                     
        uint32_t                    mMaxEncFrames;                                          

        uint32_t                    mAiffFileFrameCount;                                    
        uint32_t                    mMaxAiffFrames;                                         
    
        uint32_t                    mAiffTotalSize;                                         
        uint32_t                    mAiffNumSamples;                                        
        uint32_t                    mAiffNumChannels;                                       
        uint8_t*                    mAiffWriteBuffer;                                       
    
        uint64_t                    mYuvFileSize;
        uint32_t                    mYuvFrameWidth;
        uint32_t                    mYuvFrameHeight;
        uint32_t                    mYuvNumTotalFrames;
        uint32_t                    mYuvFrameSize;

        uint32_t                    mRawFileFrameCount;                                    
        uint32_t                    mMaxRawFrames;                                         

};  //  CNTV2DemoHevcCommon


#endif  //  _NTV2DEMOHEVCCOMMON_H