1 |
/***************************************************************************** |
/***************************************************************************** |
2 |
* |
* |
3 |
* XVID MPEG-4 VIDEO CODEC |
* XVID MPEG-4 VIDEO CODEC |
4 |
* - Motion estimation fuctions header file - |
* - MB related header - |
5 |
* |
* |
6 |
* Copyright(C) 2002 Michael Militzer <isibaar@xvid.org> |
* Copyright(C) 2001 Michael Militzer <isibaar@videocoding.de> |
7 |
* |
* |
8 |
* This file is part of XviD, a free MPEG-4 video encoder/decoder |
* This program is free software ; you can redistribute it and/or modify |
9 |
* |
* it under the terms of the GNU General Public License as published by |
|
* XviD is free software; you can redistribute it and/or modify it |
|
|
* under the terms of the GNU General Public License as published by |
|
10 |
* the Free Software Foundation; either version 2 of the License, or |
* the Free Software Foundation; either version 2 of the License, or |
11 |
* (at your option) any later version. |
* (at your option) any later version. |
12 |
* |
* |
19 |
* along with this program; if not, write to the Free Software |
* along with this program; if not, write to the Free Software |
20 |
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
21 |
* |
* |
22 |
* Under section 8 of the GNU General Public License, the copyright |
* $Id: mbfunctions.h,v 1.17.2.5 2003-06-28 15:48:08 chl Exp $ |
|
* holders of XVID explicitly forbid distribution in the following |
|
|
* countries: |
|
|
* |
|
|
* - Japan |
|
|
* - United States of America |
|
|
* |
|
|
* Linking XviD statically or dynamically with other modules is making a |
|
|
* combined work based on XviD. Thus, the terms and conditions of the |
|
|
* GNU General Public License cover the whole combination. |
|
|
* |
|
|
* As a special exception, the copyright holders of XviD give you |
|
|
* permission to link XviD with independent modules that communicate with |
|
|
* XviD solely through the VFW1.1 and DShow interfaces, regardless of the |
|
|
* license terms of these independent modules, and to copy and distribute |
|
|
* the resulting combined work under terms of your choice, provided that |
|
|
* every copy of the combined work is accompanied by a complete copy of |
|
|
* the source code of XviD (the version of XviD used to produce the |
|
|
* combined work), being distributed under the terms of the GNU General |
|
|
* Public License plus this exception. An independent module is a module |
|
|
* which is not derived from or based on XviD. |
|
|
* |
|
|
* Note that people who make modified versions of XviD are not obligated |
|
|
* to grant this special exception for their modified versions; it is |
|
|
* their choice whether to do so. The GNU General Public License gives |
|
|
* permission to release a modified version without this exception; this |
|
|
* exception also makes it possible to release a modified version which |
|
|
* carries forward this exception. |
|
|
* |
|
|
* $Id: mbfunctions.h,v 1.16 2002-11-28 07:27:37 suxen_drol Exp $ |
|
23 |
* |
* |
24 |
****************************************************************************/ |
****************************************************************************/ |
25 |
|
|
26 |
#ifndef _MBFUNCTIONS_H |
#ifndef _ENCORE_BLOCK_H |
27 |
#define _MBFUNCTIONS_H |
#define _ENCORE_BLOCK_H |
28 |
|
|
29 |
#include "../encoder.h" |
#include "../encoder.h" |
30 |
#include "../bitstream/bitstream.h" |
#include "../bitstream/bitstream.h" |
31 |
|
|
32 |
/***************************************************************************** |
/** MotionEstimation **/ |
|
* Prototypes |
|
|
****************************************************************************/ |
|
|
|
|
|
|
|
|
/* MotionEstimation */ |
|
33 |
|
|
34 |
bool MotionEstimation(MBParam * const pParam, |
bool MotionEstimation(MBParam * const pParam, |
35 |
FRAMEINFO * const current, |
FRAMEINFO * const current, |
37 |
const IMAGE * const pRefH, |
const IMAGE * const pRefH, |
38 |
const IMAGE * const pRefV, |
const IMAGE * const pRefV, |
39 |
const IMAGE * const pRefHV, |
const IMAGE * const pRefHV, |
40 |
|
const IMAGE * const pGMC, |
41 |
const uint32_t iLimit); |
const uint32_t iLimit); |
42 |
|
|
43 |
|
|
44 |
|
|
45 |
|
void |
46 |
|
GMEanalysis(const MBParam * const pParam, |
47 |
|
const FRAMEINFO * const current, |
48 |
|
const FRAMEINFO * const reference, |
49 |
|
const IMAGE * const pRefH, |
50 |
|
const IMAGE * const pRefV, |
51 |
|
const IMAGE * const pRefHV); |
52 |
|
|
53 |
|
WARPPOINTS |
54 |
|
GlobalMotionEst(MACROBLOCK * const pMBs, |
55 |
|
const MBParam * const pParam, |
56 |
|
const FRAMEINFO * const current, |
57 |
|
const FRAMEINFO * const reference, |
58 |
|
const IMAGE * const pRefH, |
59 |
|
const IMAGE * const pRefV, |
60 |
|
const IMAGE * const pRefHV); |
61 |
|
|
62 |
/* MBMotionCompensation */ |
int |
63 |
|
GlobalMotionEstRefine( |
64 |
void MBMotionCompensation(MACROBLOCK * const pMB, |
WARPPOINTS *const startwp, |
65 |
const uint32_t j, |
MACROBLOCK * const pMBs, |
66 |
const uint32_t i, |
const MBParam * const pParam, |
67 |
|
const FRAMEINFO * const current, |
68 |
|
const FRAMEINFO * const reference, |
69 |
|
const IMAGE * const pCurr, |
70 |
const IMAGE * const pRef, |
const IMAGE * const pRef, |
71 |
const IMAGE * const pRefH, |
const IMAGE * const pRefH, |
72 |
const IMAGE * const pRefV, |
const IMAGE * const pRefV, |
73 |
const IMAGE * const pRefHV, |
const IMAGE * const pRefHV); |
|
IMAGE * const pCurrent, |
|
|
int16_t dct_codes[6 * 64], |
|
|
const uint32_t width, |
|
|
const uint32_t height, |
|
|
const uint32_t edged_width, |
|
|
const uint32_t rounding); |
|
74 |
|
|
75 |
|
int |
76 |
|
globalSAD(const WARPPOINTS *const wp, |
77 |
|
const MBParam * const pParam, |
78 |
|
const MACROBLOCK * const pMBs, |
79 |
|
const FRAMEINFO * const current, |
80 |
|
const IMAGE * const pRef, |
81 |
|
const IMAGE * const pCurr, |
82 |
|
uint8_t *const GMCblock); |
83 |
|
|
|
/* MBTransQuant.c */ |
|
84 |
|
|
|
void MBTransQuantIntra(const MBParam * pParam, |
|
|
FRAMEINFO * frame, |
|
|
MACROBLOCK * pMB, |
|
|
const uint32_t x_pos, /* <-- The x position of the MB to be searched */ |
|
|
const uint32_t y_pos, /* <-- The y position of the MB to be searched */ |
|
|
int16_t data[6 * 64], /* <-> the data of the MB to be coded */ |
|
|
int16_t qcoeff[6 * 64]); /* <-> the quantized DCT coefficients */ |
|
85 |
|
|
86 |
|
|
87 |
|
/** MBMotionCompensation **/ |
88 |
|
|
89 |
void MBTransQuantIntra2(const MBParam * pParam, |
void |
90 |
FRAMEINFO * frame, |
MBMotionCompensation(MACROBLOCK * const mb, |
91 |
MACROBLOCK * pMB, |
const uint32_t i, |
92 |
|
const uint32_t j, |
93 |
|
const IMAGE * const ref, |
94 |
|
const IMAGE * const refh, |
95 |
|
const IMAGE * const refv, |
96 |
|
const IMAGE * const refhv, |
97 |
|
const IMAGE * const refGMC, |
98 |
|
IMAGE * const cur, |
99 |
|
int16_t * dct_codes, |
100 |
|
const uint32_t width, |
101 |
|
const uint32_t height, |
102 |
|
const uint32_t edged_width, |
103 |
|
const int32_t quarterpel, |
104 |
|
const int reduced_resolution, |
105 |
|
const int32_t rounding); |
106 |
|
|
107 |
|
/** MBTransQuant.c **/ |
108 |
|
|
109 |
|
|
110 |
|
void MBTransQuantIntra(const MBParam * const pParam, |
111 |
|
const FRAMEINFO * const frame, |
112 |
|
MACROBLOCK * const pMB, |
113 |
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 */ |
114 |
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 */ |
115 |
int16_t data[6 * 64], /* <-> the data of the MB to be coded */ |
int16_t data[6 * 64], /* <-> the data of the MB to be coded */ |
116 |
int16_t qcoeff[6 * 64]); /* <-> the quantized DCT coefficients */ |
int16_t qcoeff[6 * 64]); /* <-> the quantized DCT coefficients */ |
117 |
|
|
118 |
|
uint8_t MBTransQuantInter(const MBParam * const pParam, |
119 |
|
const FRAMEINFO * const frame, |
120 |
uint8_t MBTransQuantInter(const MBParam * pParam, |
MACROBLOCK * const pMB, |
|
FRAMEINFO * frame, |
|
|
MACROBLOCK * pMB, |
|
|
const uint32_t x_pos, |
|
|
const uint32_t y_pos, |
|
|
int16_t data[6 * 64], |
|
|
int16_t qcoeff[6 * 64]); |
|
|
|
|
|
|
|
|
uint8_t MBTransQuantInter2(const MBParam * pParam, |
|
|
FRAMEINFO * frame, |
|
|
MACROBLOCK * pMB, |
|
121 |
const uint32_t x_pos, |
const uint32_t x_pos, |
122 |
const uint32_t y_pos, |
const uint32_t y_pos, |
123 |
int16_t data[6 * 64], |
int16_t data[6 * 64], |
126 |
uint8_t MBTransQuantInterBVOP(const MBParam * pParam, |
uint8_t MBTransQuantInterBVOP(const MBParam * pParam, |
127 |
FRAMEINFO * frame, |
FRAMEINFO * frame, |
128 |
MACROBLOCK * pMB, |
MACROBLOCK * pMB, |
|
int16_t data[6 * 64], |
|
|
int16_t qcoeff[6 * 64]); |
|
|
|
|
|
void MBTrans(const MBParam * pParam, |
|
|
FRAMEINFO * frame, |
|
|
MACROBLOCK * pMB, |
|
129 |
const uint32_t x_pos, |
const uint32_t x_pos, |
130 |
const uint32_t y_pos, |
const uint32_t y_pos, |
|
int16_t data[6 * 64]); |
|
|
|
|
|
void MBfDCT(const MBParam * pParam, |
|
|
FRAMEINFO * frame, |
|
|
MACROBLOCK * pMB, |
|
|
int16_t data[6 * 64]); |
|
|
|
|
|
uint8_t MBQuantInter(const MBParam * pParam, |
|
|
const int iQuant, |
|
|
int16_t data[6 * 64], |
|
|
int16_t qcoeff[6 * 64]); |
|
|
|
|
|
void MBQuantDeQuantIntra(const MBParam * pParam, |
|
|
FRAMEINFO * frame, |
|
|
MACROBLOCK *pMB, |
|
|
int16_t qcoeff[6 * 64], |
|
|
int16_t data[6*64]); |
|
|
|
|
|
void MBQuantIntra(const MBParam * pParam, |
|
|
FRAMEINFO * frame, |
|
|
MACROBLOCK *pMB, |
|
131 |
int16_t data[6*64], |
int16_t data[6*64], |
132 |
int16_t qcoeff[6 * 64]); |
int16_t qcoeff[6 * 64]); |
133 |
|
|
|
void MBDeQuantIntra(const MBParam * pParam, |
|
|
const int iQuant, |
|
|
int16_t qcoeff[6 * 64], |
|
|
int16_t data[6*64]); |
|
|
|
|
|
void MBDeQuantInter(const MBParam * pParam, |
|
|
const int iQuant, |
|
|
int16_t data[6 * 64], |
|
|
int16_t qcoeff[6 * 64], |
|
|
const uint8_t cbp); |
|
134 |
|
|
135 |
|
typedef uint32_t (MBFIELDTEST) (int16_t data[6 * 64]); /* function pointer for field test */ |
136 |
|
typedef MBFIELDTEST *MBFIELDTEST_PTR; |
137 |
|
|
138 |
void MBiDCT(int16_t data[6 * 64], |
/* global field test pointer for xvid.c */ |
139 |
const uint8_t cbp); |
extern MBFIELDTEST_PTR MBFieldTest; |
140 |
|
|
141 |
|
/* field test implementations */ |
142 |
void MBTransAdd(const MBParam * pParam, |
MBFIELDTEST MBFieldTest_c; |
143 |
FRAMEINFO * frame, |
MBFIELDTEST MBFieldTest_mmx; |
|
MACROBLOCK * pMB, |
|
|
const uint32_t x_pos, |
|
|
const uint32_t y_pos, |
|
|
int16_t data[6 * 64], |
|
|
const uint8_t cbp); |
|
|
|
|
|
|
|
|
|
|
|
/* interlacing */ |
|
|
|
|
|
uint32_t MBDecideFieldDCT(int16_t data[6 * 64]); /* <- decide whether to use field-based DCT for interlacing */ |
|
144 |
|
|
145 |
void MBFrameToField(int16_t data[6 * 64]); /* de-interlace vertical Y blocks */ |
void MBFrameToField(int16_t data[6 * 64]); /* de-interlace vertical Y blocks */ |
146 |
|
|
147 |
|
|
148 |
/* MBCoding.c */ |
/** MBCoding.c **/ |
|
|
|
|
void MBSkip(Bitstream * bs); /* just the bitstream. Since MB is skipped, no info is needed */ |
|
|
|
|
149 |
|
|
150 |
void MBCoding(const FRAMEINFO * frame, /* <-- the parameter for coding of the bitstream */ |
void MBCoding(const FRAMEINFO * const frame, /* <-- the parameter for coding of the bitstream */ |
151 |
MACROBLOCK * pMB, /* <-- Info of the MB to be coded */ |
MACROBLOCK * pMB, /* <-- Info of the MB to be coded */ |
152 |
int16_t qcoeff[6 * 64], /* <-- the quantized DCT coefficients */ |
int16_t qcoeff[6 * 64], /* <-- the quantized DCT coefficients */ |
153 |
Bitstream * bs, /* <-> the bitstream */ |
Bitstream * bs, /* <-> the bitstream */ |