1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
|
/* ------------------------------------------------------------------
* 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_ */
|
