blob: 8293576b1a006ebb3966a60f9500b36d675d8aa5 [file] [log] [blame]
/* ------------------------------------------------------------------
* Copyright (C) 1998-2009 PacketVideo
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied.
* See the License for the specific language governing permissions
* and limitations under the License.
* -------------------------------------------------------------------
*/
#ifndef _MP4ENC_LIB_H_
#define _MP4ENC_LIB_H_
#include "mp4def.h" // typedef
#include "mp4lib_int.h" // main video structure
#ifdef __cplusplus
extern "C"
{
#endif
/* defined in vop.c */
PV_STATUS EncodeVop(VideoEncData *video);
PV_STATUS EncodeSlice(VideoEncData *video);
PV_STATUS EncodeVideoPacketHeader(VideoEncData *video, int MB_number,
int quant_scale, Int insert);
#ifdef ALLOW_VOP_NOT_CODED
PV_STATUS EncodeVopNotCoded(VideoEncData *video, UChar *bstream, Int *size, ULong modTime);
#endif
/* defined in combined_decode.c */
PV_STATUS EncodeFrameCombinedMode(VideoEncData *video);
PV_STATUS EncodeSliceCombinedMode(VideoEncData *video);
/* defined in datapart_decode.c */
PV_STATUS EncodeFrameDataPartMode(VideoEncData *video);
PV_STATUS EncodeSliceDataPartMode(VideoEncData *video);
/* defined in fastcodeMB.c */
//void m4v_memset(void *adr_dst, uint8 value, uint32 size);
PV_STATUS CodeMB_H263(VideoEncData *video, approxDCT *function, Int offsetQP, Int ncoefblck[]);
#ifndef NO_MPEG_QUANT
PV_STATUS CodeMB_MPEG(VideoEncData *video, approxDCT *function, Int offsetQP, Int ncoefblck[]);
#endif
Int getBlockSAV(Short block[]);
Int Sad8x8(UChar *rec, UChar *prev, Int lx);
Int getBlockSum(UChar *rec, Int lx);
/* defined in dct.c */
void blockIdct(Short *block);
void blockIdct_SSE(Short *input);
void BlockDCTEnc(Short *blockData, Short *blockCoeff);
/*---- FastQuant.c -----*/
Int cal_dc_scalerENC(Int QP, Int type) ;
Int BlockQuantDequantH263Inter(Short *rcoeff, Short *qcoeff, struct QPstruct *QuantParam,
UChar bitmapcol[ ], UChar *bitmaprow, UInt *bitmapzz,
Int dctMode, Int comp, Int dummy, UChar shortHeader);
Int BlockQuantDequantH263Intra(Short *rcoeff, Short *qcoeff, struct QPstruct *QuantParam,
UChar bitmapcol[ ], UChar *bitmaprow, UInt *bitmapzz,
Int dctMode, Int comp, Int dc_scaler, UChar shortHeader);
Int BlockQuantDequantH263DCInter(Short *rcoeff, Short *qcoeff, struct QPstruct *QuantParam,
UChar *bitmaprow, UInt *bitmapzz, Int dummy, UChar shortHeader);
Int BlockQuantDequantH263DCIntra(Short *rcoeff, Short *qcoeff, struct QPstruct *QuantParam,
UChar *bitmaprow, UInt *bitmapzz, Int dc_scaler, UChar shortHeader);
#ifndef NO_MPEG_QUANT
Int BlockQuantDequantMPEGInter(Short *rcoeff, Short *qcoeff, Int QP, Int *qmat,
UChar bitmapcol[ ], UChar *bitmaprow, UInt *bitmapzz,
Int DctMode, Int comp, Int dc_scaler);
Int BlockQuantDequantMPEGIntra(Short *rcoeff, Short *qcoeff, Int QP, Int *qmat,
UChar bitmapcol[ ], UChar *bitmaprow, UInt *bitmapzz,
Int DctMode, Int comp, Int dc_scaler);
Int BlockQuantDequantMPEGDCInter(Short *rcoeff, Short *qcoeff, Int QP, Int *qmat,
UChar bitmapcol[ ], UChar *bitmaprow, UInt *bitmapzz, Int dummy);
Int BlockQuantDequantMPEGDCIntra(Short *rcoeff, Short *qcoeff, Int QP, Int *qmat,
UChar bitmapcol[ ], UChar *bitmaprow, UInt *bitmapzz, Int dc_scaler);
#endif
/*---- FastIDCT.c -----*/
void BlockIDCTMotionComp(Short *block, UChar *bitmapcol, UChar bitmaprow,
Int dctMode, UChar *rec, UChar *prev, Int lx_intra_zeroMV);
/* defined in motion_comp.c */
void getMotionCompensatedMB(VideoEncData *video, Int ind_x, Int ind_y, Int offset);
void EncPrediction_INTER(Int xpred, Int ypred, UChar *c_prev, UChar *c_rec,
Int width, Int round1);
void EncPrediction_INTER4V(Int xpred, Int ypred, MOT *mot, UChar *c_prev, UChar *c_rec,
Int width, Int round1);
void EncPrediction_Chrom(Int xpred, Int ypred, UChar *cu_prev, UChar *cv_prev, UChar *cu_rec,
UChar *cv_rec, Int pitch_uv, Int width_uv, Int height_uv, Int round1);
void get_MB(UChar *c_prev, UChar *c_prev_u , UChar *c_prev_v,
Short mb[6][64], Int width, Int width_uv);
void PutSkippedBlock(UChar *rec, UChar *prev, Int lx);
/* defined in motion_est.c */
void MotionEstimation(VideoEncData *video);
#ifdef HTFM
void InitHTFM(VideoEncData *video, HTFM_Stat *htfm_stat, double *newvar, Int *collect);
void UpdateHTFM(VideoEncData *video, double *newvar, double *exp_lamda, HTFM_Stat *htfm_stat);
#endif
/* defined in ME_utils.c */
void ChooseMode_C(UChar *Mode, UChar *cur, Int lx, Int min_SAD);
void ChooseMode_MMX(UChar *Mode, UChar *cur, Int lx, Int min_SAD);
void GetHalfPelMBRegion_C(UChar *cand, UChar *hmem, Int lx);
void GetHalfPelMBRegion_SSE(UChar *cand, UChar *hmem, Int lx);
void GetHalfPelBlkRegion(UChar *cand, UChar *hmem, Int lx);
void PaddingEdge(Vop *padVop);
void ComputeMBSum_C(UChar *cur, Int lx, MOT *mot_mb);
void ComputeMBSum_MMX(UChar *cur, Int lx, MOT *mot_mb);
void ComputeMBSum_SSE(UChar *cur, Int lx, MOT *mot_mb);
void GetHalfPelMBRegionPadding(UChar *ncand, UChar *hmem, Int lx, Int *reptl);
void GetHalfPelBlkRegionPadding(UChar *ncand, UChar *hmem, Int lx, Int *reptl);
/* defined in findhalfpel.c */
void FindHalfPelMB(VideoEncData *video, UChar *cur, MOT *mot, UChar *ncand,
Int xpos, Int ypos, Int *xhmin, Int *yhmin, Int hp_guess);
Int FindHalfPelBlk(VideoEncData *video, UChar *cur, MOT *mot, Int sad16, UChar *ncand8[],
UChar *mode, Int xpos, Int ypos, Int *xhmin, Int *yhmin, UChar *hp_mem);
/* defined in sad.c */
Int SAD_MB_HalfPel_Cxhyh(UChar *ref, UChar *blk, Int dmin_lx, void *extra_info);
Int SAD_MB_HalfPel_Cyh(UChar *ref, UChar *blk, Int dmin_lx, void *extra_info);
Int SAD_MB_HalfPel_Cxh(UChar *ref, UChar *blk, Int dmin_lx, void *extra_info);
Int SAD_MB_HalfPel_MMX(UChar *ref, UChar *blk, Int dmin_lx, void *extra_info);
Int SAD_MB_HalfPel_SSE(UChar *ref, UChar *blk, Int dmin_lx, void *extra_info);
Int SAD_Blk_HalfPel_C(UChar *ref, UChar *blk, Int dmin, Int lx, Int rx, Int xh, Int yh, void *extra_info);
Int SAD_Blk_HalfPel_MMX(UChar *ref, UChar *blk, Int dmin, Int lx, void *extra_info);
Int SAD_Blk_HalfPel_SSE(UChar *ref, UChar *blk, Int dmin, Int lx, void *extra_info);
Int SAD_Macroblock_C(UChar *ref, UChar *blk, Int dmin_lx, void *extra_info);
Int SAD_Macroblock_MMX(UChar *ref, UChar *blk, Int dmin_lx, void *extra_info);
Int SAD_Macroblock_SSE(UChar *ref, UChar *blk, Int dmin_lx, void *extra_info);
Int SAD_Block_C(UChar *ref, UChar *blk, Int dmin, Int lx, void *extra_info);
Int SAD_Block_MMX(UChar *ref, UChar *blk, Int dmin, Int lx, void *extra_info);
Int SAD_Block_SSE(UChar *ref, UChar *blk, Int dmin, Int lx, void *extra_info);
#ifdef HTFM /* Hypothesis Testing Fast Matching */
Int SAD_MB_HP_HTFM_Collectxhyh(UChar *ref, UChar *blk, Int dmin_x, void *extra_info);
Int SAD_MB_HP_HTFM_Collectyh(UChar *ref, UChar *blk, Int dmin_x, void *extra_info);
Int SAD_MB_HP_HTFM_Collectxh(UChar *ref, UChar *blk, Int dmin_x, void *extra_info);
Int SAD_MB_HP_HTFMxhyh(UChar *ref, UChar *blk, Int dmin_lx, void *extra_info);
Int SAD_MB_HP_HTFMyh(UChar *ref, UChar *blk, Int dmin_lx, void *extra_info);
Int SAD_MB_HP_HTFMxh(UChar *ref, UChar *blk, Int dmin_lx, void *extra_info);
Int SAD_MB_HTFM_Collect(UChar *ref, UChar *blk, Int dmin_lx, void *extra_info);
Int SAD_MB_HTFM(UChar *ref, UChar *blk, Int dmin_lx, void *extra_info);
#endif
/* on-the-fly padding */
Int SAD_Blk_PADDING(UChar *ref, UChar *cur, Int dmin, Int lx, void *extra_info);
Int SAD_MB_PADDING(UChar *ref, UChar *cur, Int dmin, Int lx, void *extra_info);
#ifdef HTFM
Int SAD_MB_PADDING_HTFM_Collect(UChar *ref, UChar *blk, Int dmin, Int lx, void *extra_info);
Int SAD_MB_PADDING_HTFM(UChar *ref, UChar *blk, Int dmin, Int lx, void *extra_info);
#endif
/* defined in rate_control.c */
/* These are APIs to rate control exposed to core encoder module. */
PV_STATUS RC_Initialize(void *video);
PV_STATUS RC_VopQPSetting(VideoEncData *video, rateControl *rc[]);
PV_STATUS RC_VopUpdateStat(VideoEncData *video, rateControl *rc);
PV_STATUS RC_MBQPSetting(VideoEncData *video, rateControl *rc, Int start_packet_header);
PV_STATUS RC_MBUpdateStat(VideoEncData *video, rateControl *rc, Int Bi, Int Hi);
PV_STATUS RC_Cleanup(rateControl *rc[], Int numLayers);
Int RC_GetSkipNextFrame(VideoEncData *video, Int currLayer);
Int RC_GetRemainingVops(VideoEncData *video, Int currLayer);
void RC_ResetSkipNextFrame(VideoEncData *video, Int currLayer);
PV_STATUS RC_UpdateBuffer(VideoEncData *video, Int currLayer, Int num_skip);
PV_STATUS RC_UpdateBXRCParams(void *input);
/* defined in vlc_encode.c */
void MBVlcEncodeDataPar_I_VOP(VideoEncData *video, Int ncoefblck[], void *blkCodePtr);
void MBVlcEncodeDataPar_P_VOP(VideoEncData *video, Int ncoefblck[], void *blkCodePtr);
void MBVlcEncodeCombined_I_VOP(VideoEncData *video, Int ncoefblck[], void *blkCodePtr);
void MBVlcEncodeCombined_P_VOP(VideoEncData *video, Int ncoefblck[], void *blkCodePtr);
void BlockCodeCoeff_ShortHeader(RunLevelBlock *RLB, BitstreamEncVideo *bs, Int j_start, Int j_stop, UChar Mode);
void BlockCodeCoeff_RVLC(RunLevelBlock *RLB, BitstreamEncVideo *bs, Int j_start, Int j_stop, UChar Mode);
void BlockCodeCoeff_Normal(RunLevelBlock *RLB, BitstreamEncVideo *bs, Int j_start, Int j_stop, UChar Mode);
#ifdef __cplusplus
}
#endif
#endif /* _MP4ENC_LIB_H_ */