mw_venc_common.h

#ifndef MW_VENC_COMMON_H
#define MW_VENC_COMMON_H
 
// Copyright (c) 2011-2020 Magewell Electronics Co., Ltd. (Nanjing)
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the "Software"),
// to deal in the Software without restriction, including without limitation
// the rights to use, copy, modify, merge, publish, distribute, sublicense,
// and/or sell copies of the Software, and to permit persons to whom the
// Software is furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included
// in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
 
#include <stdint.h>
 
typedef enum mw_venc_frame_type
{
    MW_VENC_FRAME_TYPE_UNKNOWN,                            
    MW_VENC_FRAME_TYPE_IDR,                                
    MW_VENC_FRAME_TYPE_I,                                
    MW_VENC_FRAME_TYPE_P,                                
    MW_VENC_FRAME_TYPE_B,                                
    MW_VENC_FRAME_TYPE_COUNT                            
}mw_venc_frame_type_t;
 
typedef struct mw_venc_frame_info
{
    mw_venc_frame_type_t frame_type;
    int32_t delay;
    int64_t pts;
}mw_venc_frame_info_t;
 
typedef void(*MW_ENCODER_CALLBACK)(void * user_ptr, const uint8_t * p_frame, uint32_t frame_len, mw_venc_frame_info_t *p_frame_info);
 
typedef void(*MW_ENCODER_CALLBACK_NALU)(void * user_ptr, const uint8_t *p_nalu_buf[], uint32_t nalu_len[], uint32_t nalu_num, mw_venc_frame_info_t *p_frame_info);
 
typedef enum mw_venc_code_type
{
    MW_VENC_CODE_TYPE_UNKNOWN,                            
    MW_VENC_CODE_TYPE_AVC,                                
    MW_VENC_CODE_TYPE_HEVC,                                
    MW_VENC_CODE_TYPE_H264 = MW_VENC_CODE_TYPE_AVC,        
    MW_VENC_CODE_TYPE_H265 = MW_VENC_CODE_TYPE_HEVC,    
    MW_VENC_CODE_TYPE_COUNT                                
}mw_venc_code_type_t;
 
typedef enum mw_venc_targetusage
{
    MW_VENC_TARGETUSAGE_UNKNOWN,                        
    MW_VENC_TARGETUSAGE_BEST_QUALITY,                    
    MW_VENC_TARGETUSAGE_BALANCED,                        
    MW_VENC_TARGETUSAGE_BEST_SPEED,                        
    MW_VENC_TARGETUSAGE_COUNT                            
}mw_venc_targetusage_t;
 
typedef enum mw_venc_rate_control_mode
{
    MW_VENC_RATECONTROL_UNKNOWN,                        
    MW_VENC_RATECONTROL_CBR,                            
    MW_VENC_RATECONTROL_VBR,                            
    MW_VENC_RATECONTROL_CQP,                            
    MW_VENC_RATECONTROL_COUNT                            
}mw_venc_rate_control_mode_t;
 
typedef enum mw_venc_profile
{
    MW_VENC_PROFILE_UNKNOWN,                            
    MW_VENC_PROFILE_H264_BASELINE,                        
    MW_VENC_PROFILE_H264_MAIN,                            
    MW_VENC_PROFILE_H264_HIGH,                            
    MW_VENC_PROFILE_H265_MAIN,                            
    MW_VENC_PROFILE_COUNT                                
}mw_venc_profile_t;
 
typedef enum mw_venc_level
{
    MW_VENC_LEVEL_UNKNOWN,                                
    MW_VENC_LEVEL_2_1,                                    
    MW_VENC_LEVEL_3_1,                                    
    MW_VENC_LEVEL_4_1,                                    
    MW_VENC_LEVEL_5_1,                                    
    MW_VENC_LEVEL_5_2,                                    
    MW_VENC_LEVEL_6_1,                                    
    MW_VENC_LEVEL_6_2,                                    
    MW_VENC_LEVEL_COUNT                                    
}mw_venc_level_t;
 
typedef enum mw_venc_fourcc
{
    MW_VENC_FOURCC_UNKNOWN,                                
    MW_VENC_FOURCC_NV12,                                
    MW_VENC_FOURCC_NV21,                                
    MW_VENC_FOURCC_YV12,                                
    MW_VENC_FOURCC_I420,                                
    MW_VENC_FOURCC_YUY2,                                
    MW_VENC_FOURCC_P010,                                
    MW_VENC_FOURCC_BGRA,                                
    MW_VENC_FOURCC_RGBA,                                
    MW_VENC_FOURCC_ARGB,                                
    MW_VENC_FOURCC_ABGR,                                
    MW_VENC_FOURCC_COUNT                                
}mw_venc_fourcc_t;
 
typedef struct mw_venc_rate_control
{
    mw_venc_rate_control_mode_t mode;                    
    union{
        struct {
            uint32_t target_bitrate;                    
            uint32_t max_bitrate;                        
        };
        struct {
            uint8_t qpi;                                
            uint8_t qpb;                                
            uint8_t qpp;                                
            uint8_t reserved;                            
        };
    };
}mw_venc_rate_control_t;
 
typedef struct mw_venc_fps{
    int32_t num;                                    
    int32_t den;                                    
}mw_venc_fps_t;
 
typedef struct mw_venc_extdata {
    uint8_t *p_extdata;                                    
    uint32_t extdata_len;                                
    uint32_t len[3];                                    
}mw_venc_extdata_t;
 
typedef enum mw_venc_property
{
    MW_VENC_PROPERTY_UNKNOWN,                            
    MW_VENC_PROPERTY_RATE_CONTROL,                        
    MW_VENC_PROPERTY_FPS,                                
    MW_VENC_PROPERTY_GOP_SIZE,                            
    MW_VENC_PROPERTY_SLICE_NUM,                            
    MW_VENC_PROPERTY_GOP_REF_SIZE,                        
    MW_VENC_PROPERTY_EXTDATA,                           
    MW_VENC_PROPERTY_FORCE_IDR,                         
    MW_VENC_PROPERTY_COUNT                                
}mw_venc_property_t;
 
typedef enum mw_venc_status
{
    MW_VENC_STATUS_SUCCESS,                                
    MW_VENC_STATUS_FAIL,                                
    MW_VENC_STATUS_UNSUPPORT,                            
    MW_VENC_STATUS_BUSY,                                
    MW_VENC_STATUS_INVALID_PARAM,                        
    MW_VENC_STATUS_UNKNOWN_ERROR,                        
    MW_VENC_STATUS_COUNT                                
}mw_venc_status_t;
 
 
//
 
typedef enum mw_venc_color_primaries {
    MW_VENC_COLOR_PRI_RESERVED0 = 0,
    MW_VENC_COLOR_PRI_BT709 = 1,  
    MW_VENC_COLOR_PRI_UNSPECIFIED = 2,
    MW_VENC_COLOR_PRI_RESERVED = 3,
    MW_VENC_COLOR_PRI_BT470M = 4,  
    
    MW_VENC_COLOR_PRI_BT470BG = 5,  
    MW_VENC_COLOR_PRI_SMPTE170M = 6,  
    MW_VENC_COLOR_PRI_SMPTE240M = 7,  
    MW_VENC_COLOR_PRI_FILM = 8,  
    MW_VENC_COLOR_PRI_BT2020 = 9,  
    MW_VENC_COLOR_PRI_SMPTE428 = 10, 
    MW_VENC_COLOR_PRI_SMPTEST428_1 = MW_VENC_COLOR_PRI_SMPTE428,
    MW_VENC_COLOR_PRI_SMPTE431 = 11, 
    MW_VENC_COLOR_PRI_SMPTE432 = 12, 
    MW_VENC_COLOR_PRI_JEDEC_P22 = 22, 
    MW_VENC_COLOR_PRI_COUNT                
}mw_venc_color_primaries_t;
 
typedef enum mw_venc_color_transfer_characteristic {
    MW_VENC_COLOR_TRC_RESERVED0 = 0,
    MW_VENC_COLOR_TRC_BT709 = 1,  
    MW_VENC_COLOR_TRC_UNSPECIFIED = 2,
    MW_VENC_COLOR_TRC_RESERVED = 3,
    MW_VENC_COLOR_TRC_GAMMA22 = 4,  
    MW_VENC_COLOR_TRC_GAMMA28 = 5,  
    MW_VENC_COLOR_TRC_SMPTE170M = 6,  
    MW_VENC_COLOR_TRC_SMPTE240M = 7,
    MW_VENC_COLOR_TRC_LINEAR = 8,  
    MW_VENC_COLOR_TRC_LOG = 9,  
    MW_VENC_COLOR_TRC_LOG_SQRT = 10, 
    MW_VENC_COLOR_TRC_IEC61966_2_4 = 11, 
    MW_VENC_COLOR_TRC_BT1361_ECG = 12, 
    MW_VENC_COLOR_TRC_IEC61966_2_1 = 13, 
    MW_VENC_COLOR_TRC_BT2020_10 = 14, 
    MW_VENC_COLOR_TRC_BT2020_12 = 15, 
    MW_VENC_COLOR_TRC_SMPTE2084 = 16, 
    MW_VENC_COLOR_TRC_SMPTEST2084 = MW_VENC_COLOR_TRC_SMPTE2084,
    MW_VENC_COLOR_TRC_SMPTE428 = 17, 
    MW_VENC_COLOR_TRC_SMPTEST428_1 = MW_VENC_COLOR_TRC_SMPTE428,
    MW_VENC_COLOR_TRC_ARIB_STD_B67 = 18, 
    MW_VENC_COLOR_TRC_COUNT                 
}mw_venc_color_transfer_characteristic_t;
 
typedef enum mw_venc_color_space {
    MW_VENC_COLOR_SPACE_RGB = 0,  
    MW_VENC_COLOR_SPACE_BT709 = 1,  
    MW_VENC_COLOR_SPACE_UNSPECIFIED = 2,
    MW_VENC_COLOR_SPACE_RESERVED = 3,
    MW_VENC_COLOR_SPACE_FCC = 4,  
    MW_VENC_COLOR_SPACE_BT470BG = 5,  
    MW_VENC_COLOR_SPACE_SMPTE170M = 6,  
    MW_VENC_COLOR_SPACE_SMPTE240M = 7,  
    MW_VENC_COLOR_SPACE_YCGCO = 8,  
    MW_VENC_COLOR_SPACE_YCOCG = MW_VENC_COLOR_SPACE_YCGCO,
    MW_VENC_COLOR_SPACE_BT2020_NCL = 9,  
    MW_VENC_COLOR_SPACE_BT2020_CL = 10, 
    MW_VENC_COLOR_SPACE_SMPTE2085 = 11, 
    MW_VENC_COLOR_SPACE_CHROMA_DERIVED_NCL = 12, 
    MW_VENC_COLOR_SPACE_CHROMA_DERIVED_CL = 13, 
    MW_VENC_COLOR_SPACE_ICTCP = 14, 
    MW_VENC_COLOR_SPACE_COUNT                
}mw_venc_color_space_t;
 
 
typedef struct mw_venc_param {
    mw_venc_code_type_t code_type;                        
    mw_venc_fourcc_t fourcc;                            
    mw_venc_targetusage_t targetusage;                    
    mw_venc_rate_control_t rate_control;                
    int32_t width;                                        
    int32_t height;                                        
    mw_venc_fps_t fps;                                    
    int32_t slice_num;                                    
    int32_t gop_pic_size;                                
    int32_t gop_ref_size;                                
    mw_venc_profile_t profile;                            
    mw_venc_level_t level;                                
    int32_t intel_async_depth;
    int32_t yuv_is_full_range;
    mw_venc_color_primaries_t color_primaries;
    mw_venc_color_transfer_characteristic_t color_trc;
    mw_venc_color_space_t color_space;
}mw_venc_param_t;
 
#endif