1 |
/************************************************************************** |
/***************************************************************************** |
2 |
* |
* |
3 |
* Modifications: |
* XVID MPEG-4 VIDEO CODEC |
4 |
|
* - Motion estimation fuctions header file - |
5 |
* |
* |
6 |
* 02.12.2001 motion estimation/compensation split |
* Copyright(C) 2002 Michael Militzer <isibaar@xvid.org> |
|
* 16.11.2001 const/uint32_t changes to MBMotionEstComp() |
|
|
* 26.08.2001 added inter4v_mode parameter to MBMotionEstComp() |
|
7 |
* |
* |
8 |
* Michael Militzer <isibaar@videocoding.de> |
* This file is part of XviD, a free MPEG-4 video encoder/decoder |
9 |
* |
* |
10 |
**************************************************************************/ |
* XviD is free software; you can redistribute it and/or modify it |
11 |
|
* under the terms of the GNU General Public License as published by |
12 |
|
* the Free Software Foundation; either version 2 of the License, or |
13 |
|
* (at your option) any later version. |
14 |
|
* |
15 |
|
* This program is distributed in the hope that it will be useful, |
16 |
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
17 |
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
18 |
|
* GNU General Public License for more details. |
19 |
|
* |
20 |
|
* You should have received a copy of the GNU General Public License |
21 |
|
* along with this program; if not, write to the Free Software |
22 |
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
23 |
|
* |
24 |
|
* Under section 8 of the GNU General Public License, the copyright |
25 |
|
* holders of XVID explicitly forbid distribution in the following |
26 |
|
* countries: |
27 |
|
* |
28 |
|
* - Japan |
29 |
|
* - United States of America |
30 |
|
* |
31 |
|
* Linking XviD statically or dynamically with other modules is making a |
32 |
|
* combined work based on XviD. Thus, the terms and conditions of the |
33 |
|
* GNU General Public License cover the whole combination. |
34 |
|
* |
35 |
|
* As a special exception, the copyright holders of XviD give you |
36 |
|
* permission to link XviD with independent modules that communicate with |
37 |
|
* XviD solely through the VFW1.1 and DShow interfaces, regardless of the |
38 |
|
* license terms of these independent modules, and to copy and distribute |
39 |
|
* the resulting combined work under terms of your choice, provided that |
40 |
|
* every copy of the combined work is accompanied by a complete copy of |
41 |
|
* the source code of XviD (the version of XviD used to produce the |
42 |
|
* combined work), being distributed under the terms of the GNU General |
43 |
|
* Public License plus this exception. An independent module is a module |
44 |
|
* which is not derived from or based on XviD. |
45 |
|
* |
46 |
|
* Note that people who make modified versions of XviD are not obligated |
47 |
|
* to grant this special exception for their modified versions; it is |
48 |
|
* their choice whether to do so. The GNU General Public License gives |
49 |
|
* permission to release a modified version without this exception; this |
50 |
|
* exception also makes it possible to release a modified version which |
51 |
|
* carries forward this exception. |
52 |
|
* |
53 |
|
* $Id: mbfunctions.h,v 1.15 2002-11-17 00:51:10 edgomez Exp $ |
54 |
|
* |
55 |
|
****************************************************************************/ |
56 |
|
|
57 |
#ifndef _ENCORE_BLOCK_H |
#ifndef _MBFUNCTIONS_H |
58 |
#define _ENCORE_BLOCK_H |
#define _MBFUNCTIONS_H |
59 |
|
|
60 |
#include "../encoder.h" |
#include "../encoder.h" |
61 |
#include "../bitstream/bitstream.h" |
#include "../bitstream/bitstream.h" |
62 |
|
|
63 |
|
/***************************************************************************** |
64 |
|
* Prototypes |
65 |
|
****************************************************************************/ |
66 |
|
|
67 |
|
|
68 |
/** MotionEstimation **/ |
/* MotionEstimation */ |
69 |
|
|
70 |
bool MotionEstimation( |
bool MotionEstimation(MBParam * const pParam, |
71 |
MACROBLOCK * const pMBs, |
FRAMEINFO * const current, |
72 |
MBParam * const pParam, |
FRAMEINFO * const reference, |
|
const IMAGE * const pRef, |
|
73 |
const IMAGE * const pRefH, |
const IMAGE * const pRefH, |
74 |
const IMAGE * const pRefV, |
const IMAGE * const pRefV, |
75 |
const IMAGE * const pRefHV, |
const IMAGE * const pRefHV, |
|
IMAGE * const pCurrent, |
|
76 |
const uint32_t iLimit); |
const uint32_t iLimit); |
77 |
|
|
78 |
|
|
79 |
/** MBMotionCompensation **/ |
bool SMP_MotionEstimation(MBParam * const pParam, |
80 |
void MBMotionCompensation( |
FRAMEINFO * const current, |
81 |
MACROBLOCK * const pMB, |
FRAMEINFO * const reference, |
82 |
|
const IMAGE * const pRefH, |
83 |
|
const IMAGE * const pRefV, |
84 |
|
const IMAGE * const pRefHV, |
85 |
|
const uint32_t iLimit); |
86 |
|
|
87 |
|
|
88 |
|
|
89 |
|
/* MBMotionCompensation */ |
90 |
|
|
91 |
|
void MBMotionCompensation(MACROBLOCK * const pMB, |
92 |
const uint32_t j, |
const uint32_t j, |
93 |
const uint32_t i, |
const uint32_t i, |
94 |
const IMAGE * const pRef, |
const IMAGE * const pRef, |
96 |
const IMAGE * const pRefV, |
const IMAGE * const pRefV, |
97 |
const IMAGE * const pRefHV, |
const IMAGE * const pRefHV, |
98 |
IMAGE * const pCurrent, |
IMAGE * const pCurrent, |
99 |
int16_t dct_codes[][64], |
int16_t dct_codes[6 * 64], |
100 |
const uint32_t width, |
const uint32_t width, |
101 |
const uint32_t height, |
const uint32_t height, |
102 |
const uint32_t edged_width, |
const uint32_t edged_width, |
103 |
const uint32_t rounding); |
const uint32_t rounding); |
104 |
|
|
105 |
|
|
106 |
/** MBTransQuant.c **/ |
/* MBTransQuant.c */ |
|
|
|
107 |
|
|
108 |
void MBTransQuantIntra(const MBParam *pParam, |
void MBTransQuantIntra(const MBParam *pParam, |
109 |
|
FRAMEINFO * frame, |
110 |
|
MACROBLOCK * pMB, |
111 |
const uint32_t x_pos, /* <-- The x position of the MB to be searched */ |
const uint32_t x_pos, /* <-- The x position of the MB to be searched */ |
112 |
const uint32_t y_pos, /* <-- The y position of the MB to be searched */ |
const uint32_t y_pos, /* <-- The y position of the MB to be searched */ |
113 |
int16_t data[][64], /* <-> the data of the MB to be coded */ |
int16_t data[6 * 64], /* <-> the data of the MB to be coded */ |
114 |
int16_t qcoeff[][64], /* <-> the quantized DCT coefficients */ |
int16_t qcoeff[6 * 64]); /* <-> the quantized DCT coefficients */ |
115 |
IMAGE * const pCurrent /* <-> the reconstructed image ( function will update one |
|
|
MB in it with data from data[] ) */ |
|
|
); |
|
116 |
|
|
117 |
|
|
118 |
uint8_t MBTransQuantInter(const MBParam *pParam, /* <-- the parameter for DCT transformation |
void MBTransQuantIntra2(const MBParam * pParam, |
119 |
and Quantization */ |
FRAMEINFO * frame, |
120 |
|
MACROBLOCK * pMB, |
121 |
const uint32_t x_pos, /* <-- The x position of the MB to be searched */ |
const uint32_t x_pos, /* <-- The x position of the MB to be searched */ |
122 |
const uint32_t y_pos, /* <-- The y position of the MB to be searched */ |
const uint32_t y_pos, /* <-- The y position of the MB to be searched */ |
123 |
int16_t data[][64], /* <-> the data of the MB to be coded */ |
int16_t data[6 * 64], /* <-> the data of the MB to be coded */ |
124 |
int16_t qcoeff[][64], /* <-> the quantized DCT coefficients */ |
int16_t qcoeff[6 * 64]); /* <-> the quantized DCT coefficients */ |
125 |
IMAGE * const pCurrent /* <-> the reconstructed image ( function will |
|
126 |
update one MB in it with data from data[] ) */ |
|
127 |
); |
|
128 |
|
uint8_t MBTransQuantInter(const MBParam * pParam, |
129 |
|
FRAMEINFO * frame, |
130 |
|
MACROBLOCK * pMB, |
131 |
|
const uint32_t x_pos, |
132 |
|
const uint32_t y_pos, |
133 |
|
int16_t data[6 * 64], |
134 |
|
int16_t qcoeff[6 * 64]); |
135 |
|
|
136 |
|
|
137 |
|
uint8_t MBTransQuantInter2(const MBParam * pParam, |
138 |
|
FRAMEINFO * frame, |
139 |
|
MACROBLOCK * pMB, |
140 |
|
const uint32_t x_pos, |
141 |
|
const uint32_t y_pos, |
142 |
|
int16_t data[6 * 64], |
143 |
|
int16_t qcoeff[6 * 64]); |
144 |
|
|
145 |
|
uint8_t MBTransQuantInterBVOP(const MBParam * pParam, |
146 |
|
FRAMEINFO * frame, |
147 |
|
MACROBLOCK * pMB, |
148 |
|
int16_t data[6 * 64], |
149 |
|
int16_t qcoeff[6 * 64]); |
150 |
|
|
151 |
|
void MBTrans(const MBParam * pParam, |
152 |
|
FRAMEINFO * frame, |
153 |
|
MACROBLOCK * pMB, |
154 |
|
const uint32_t x_pos, |
155 |
|
const uint32_t y_pos, |
156 |
|
int16_t data[6 * 64]); |
157 |
|
|
158 |
|
void MBfDCT(const MBParam * pParam, |
159 |
|
FRAMEINFO * frame, |
160 |
|
MACROBLOCK * pMB, |
161 |
|
int16_t data[6 * 64]); |
162 |
|
|
163 |
|
uint8_t MBQuantInter(const MBParam * pParam, |
164 |
|
const int iQuant, |
165 |
|
int16_t data[6 * 64], |
166 |
|
int16_t qcoeff[6 * 64]); |
167 |
|
|
168 |
|
void MBQuantDeQuantIntra(const MBParam * pParam, |
169 |
|
FRAMEINFO * frame, |
170 |
|
MACROBLOCK *pMB, |
171 |
|
int16_t qcoeff[6 * 64], |
172 |
|
int16_t data[6*64]); |
173 |
|
|
174 |
|
void MBQuantIntra(const MBParam * pParam, |
175 |
|
FRAMEINFO * frame, |
176 |
|
MACROBLOCK *pMB, |
177 |
|
int16_t data[6*64], |
178 |
|
int16_t qcoeff[6 * 64]); |
179 |
|
|
180 |
|
void MBDeQuantIntra(const MBParam * pParam, |
181 |
|
const int iQuant, |
182 |
|
int16_t qcoeff[6 * 64], |
183 |
|
int16_t data[6*64]); |
184 |
|
|
185 |
|
void MBDeQuantInter(const MBParam * pParam, |
186 |
|
const int iQuant, |
187 |
|
int16_t data[6 * 64], |
188 |
|
int16_t qcoeff[6 * 64], |
189 |
|
const uint8_t cbp); |
190 |
|
|
191 |
|
|
192 |
|
void MBiDCT(int16_t data[6 * 64], |
193 |
|
const uint8_t cbp); |
194 |
|
|
195 |
|
|
196 |
|
void MBTransAdd(const MBParam * pParam, |
197 |
|
FRAMEINFO * frame, |
198 |
|
MACROBLOCK * pMB, |
199 |
|
const uint32_t x_pos, |
200 |
|
const uint32_t y_pos, |
201 |
|
int16_t data[6 * 64], |
202 |
|
const uint8_t cbp); |
203 |
|
|
204 |
|
|
205 |
|
|
206 |
|
/* interlacing */ |
207 |
|
|
208 |
|
uint32_t MBDecideFieldDCT(int16_t data[6 * 64]); /* <- decide whether to use field-based DCT for interlacing */ |
209 |
|
|
210 |
|
void MBFrameToField(int16_t data[6 * 64]); /* de-interlace vertical Y blocks */ |
211 |
|
|
212 |
|
|
213 |
|
/* MBCoding.c */ |
214 |
|
|
215 |
|
void MBSkip(Bitstream * bs); /* just the bitstream. Since MB is skipped, no info is needed */ |
216 |
|
|
|
/** MBCoding.c **/ |
|
217 |
|
|
218 |
void MBCoding(const MBParam *pParam, /* <-- the parameter for coding of the bitstream */ |
void MBCoding(const FRAMEINFO * frame, /* <-- the parameter for coding of the bitstream */ |
219 |
MACROBLOCK *pMB, /* <-- Info of the MB to be coded */ |
MACROBLOCK *pMB, /* <-- Info of the MB to be coded */ |
220 |
int16_t qcoeff[][64], /* <-- the quantized DCT coefficients */ |
int16_t qcoeff[6 * 64], /* <-- the quantized DCT coefficients */ |
221 |
Bitstream * bs, /* <-> the bitstream */ |
Bitstream * bs, /* <-> the bitstream */ |
222 |
Statistics * pStat /* <-> statistical data collected for current frame */ |
Statistics * pStat); /* <-> statistical data collected for current frame */ |
|
); |
|
223 |
|
|
224 |
#endif |
#endif |