nvmedia_iep_output_extradata.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2021-2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
 *
 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
 * property and proprietary rights in and to this material, related
 * documentation and any modifications thereto. Any use, reproduction,
 * disclosure or distribution of this material and related documentation
 * without an express license agreement from NVIDIA CORPORATION or
 * its affiliates is strictly prohibited.
 */
 
#ifndef NVMEDIA_IEP_OUTPUT_EXTRA_DATA_H
#define NVMEDIA_IEP_OUTPUT_EXTRA_DATA_H
 
#ifdef __cplusplus
extern "C" {
#endif
 
#include <stdint.h>
#include <stdbool.h>
 
#define NVMEDIA_ENCODE_MAX_RPS_SIZE 17U
 
#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;
    bool bIdrFrame;
    bool 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;
        bool bRefPic;
        bool bIntraRefresh;
        uint32_t uIntraMBCount;
        uint32_t uInterMBCount;
    } h264Extradata;
    struct {
        NvMediaEncodeH26xFrameType eFrameType;
        bool bRefPic;
        bool 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;
    bool bkeyFrame;
    bool bEndOfFrame;
    uint32_t ulHdrSize;
    int16_t  AvgQP;
    bool bIsGoldenOrAlternateFrame;
    bool bValidReconCRC;
    uint32_t ulReconCRC_Y;
    uint32_t ulReconCRC_U;
    uint32_t ulReconCRC_V;
    uint32_t ulFrameMinQP;
    uint32_t ulFrameMaxQP;
    bool bRPSFeedback;
    uint32_t  ulCurrentRefFrameId;
    uint32_t  ulActiveRefFrames;
    NvMediaEncodeFrameFullProp RPSList[NVMEDIA_ENCODE_MAX_RPS_SIZE];
    bool 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 */