nvmedia_iep_output_extradata.h

Go to the documentation of this file.

/*
 * Copyright (c) 2020, NVIDIA CORPORATION.  All rights reserved.  All
 * information contained herein is proprietary and confidential to NVIDIA
 * Corporation.  Any use, reproduction, or disclosure without the written
 * permission of NVIDIA Corporation is prohibited.
 */
 
#ifndef NVMEDIA_IEP_OUTPUT_EXTRA_DATA_H
#define NVMEDIA_IEP_OUTPUT_EXTRA_DATA_H
 
#include "nvmedia_iep_common_extradata.h"
 
#ifdef __cplusplus
extern "C" {
#endif
 
#define MV_BUFFER_HEADER 0xFFFEFDFCU
 
typedef enum {
    NVMEDIA_ENCODE_H26X_FRAME_TYPE_P=0,
    NVMEDIA_ENCODE_H26X_FRAME_TYPE_B,
    NVMEDIA_ENCODE_H26X_FRAME_TYPE_I,
    NVMEDIA_ENCODE_H26X_FRAME_TYPE_IDR,
    NVMEDIA_ENCODE_H26X_FRAME_TYPE_END
} NvMediaEncodeH26xFrameType;
 
typedef struct {
    uint32_t ulFrameId;
    NvMediaBool bIdrFrame;
    NvMediaBool bLTRefFrame;
    uint32_t  ulPictureOrderCnt;
    uint32_t  ulFrameNum;
    uint32_t  ulLTRFrameIdx;
} NvMediaEncodeFrameFullProp;
 
typedef struct {
    uint32_t  ulCycleCount;
    uint32_t  ulPresetTime;
    uint32_t  ulFlushTime;
    uint32_t  ulEncodeTime;
    uint32_t  ulFetchTime;
} NvMediaFrameStats;
 
typedef union {
    struct {
        NvMediaEncodeH26xFrameType eFrameType;
        NvMediaBool bRefPic;
        NvMediaBool bIntraRefresh;
        uint32_t uIntraMBCount;
        uint32_t uInterMBCount;
    } h264Extradata;
    struct {
        NvMediaEncodeH26xFrameType eFrameType;
        NvMediaBool bRefPic;
        NvMediaBool bIntraRefresh;
        uint32_t uIntraCU32x32Count;
        uint32_t uInterCU32x32Count;
        uint32_t uIntraCU16x16Count;
        uint32_t uInterCU16x16Count;
        uint32_t uIntraCU8x8Count;
        uint32_t uInterCU8x8Count;
    } h265Extradata;
} NvMediaEncodeCodecExData;
 
typedef struct
{
    uint32_t MagicNum;
    uint32_t buffersize;
    uint16_t blocksize;
    uint16_t width_in_blocksize;
    uint16_t height_in_blocksize;
    uint16_t reserved;
} NvMediaEncodeMVBufferHeader;
 
typedef struct {
    int32_t mv_x;
    int32_t mv_y;
} NvMediaEncodeMVData;
 
typedef struct {
    uint32_t ulExtraDataSize;
    NvMediaBool bkeyFrame;
    NvMediaBool bEndOfFrame;
    uint32_t ulHdrSize;
    int16_t  AvgQP;
    NvMediaBool bIsGoldenOrAlternateFrame;
    NvMediaBool bValidReconCRC;
    uint32_t ulReconCRC_Y;
    uint32_t ulReconCRC_U;
    uint32_t ulReconCRC_V;
    uint32_t ulFrameMinQP;
    uint32_t ulFrameMaxQP;
    NvMediaBool bRPSFeedback;
    uint32_t  ulCurrentRefFrameId;
    uint32_t  ulActiveRefFrames;
    NvMediaEncodeFrameFullProp RPSList[NVMEDIA_ENCODE_MAX_RPS_SIZE];
    NvMediaBool bMVbufferdump;
    uint32_t MVBufferDumpSize;
    uint32_t MVBufferDumpStartOffset;
    NvMediaFrameStats FrameStats;
    uint32_t ulHrdBitrate;
    uint32_t ulVbvBufSize;
    NvMediaVideoCodec codec;
    NvMediaEncodeCodecExData codecExData;
} NvMediaEncodeOutputExtradata;
 
#ifdef __cplusplus
};     /* extern "C" */
#endif
 
#endif /* NVMEDIA_IEP_OUTPUT_EXTRA_DATA_H */