/* ------------------------------------------------------------------ * 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_ */