1 |
|
/***************************************************************************** |
2 |
|
* |
3 |
|
* XVID MPEG-4 VIDEO CODEC |
4 |
|
* - Adaptive quantization functions - |
5 |
|
* |
6 |
|
* Copyright(C) 2002 Peter Ross <pross@xvid.org> |
7 |
|
* 2002 Christoph Lampert <gruel@web.de> |
8 |
|
* |
9 |
|
* This file is part of XviD, a free MPEG-4 video encoder/decoder |
10 |
|
* |
11 |
|
* XviD is free software; you can redistribute it and/or modify it |
12 |
|
* under the terms of the GNU General Public License as published by |
13 |
|
* the Free Software Foundation; either version 2 of the License, or |
14 |
|
* (at your option) any later version. |
15 |
|
* |
16 |
|
* This program is distributed in the hope that it will be useful, |
17 |
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
18 |
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
19 |
|
* GNU General Public License for more details. |
20 |
|
* |
21 |
|
* You should have received a copy of the GNU General Public License |
22 |
|
* along with this program; if not, write to the Free Software |
23 |
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
24 |
|
* |
25 |
|
* Under section 8 of the GNU General Public License, the copyright |
26 |
|
* holders of XVID explicitly forbid distribution in the following |
27 |
|
* countries: |
28 |
|
* |
29 |
|
* - Japan |
30 |
|
* - United States of America |
31 |
|
* |
32 |
|
* Linking XviD statically or dynamically with other modules is making a |
33 |
|
* combined work based on XviD. Thus, the terms and conditions of the |
34 |
|
* GNU General Public License cover the whole combination. |
35 |
|
* |
36 |
|
* As a special exception, the copyright holders of XviD give you |
37 |
|
* permission to link XviD with independent modules that communicate with |
38 |
|
* XviD solely through the VFW1.1 and DShow interfaces, regardless of the |
39 |
|
* license terms of these independent modules, and to copy and distribute |
40 |
|
* the resulting combined work under terms of your choice, provided that |
41 |
|
* every copy of the combined work is accompanied by a complete copy of |
42 |
|
* the source code of XviD (the version of XviD used to produce the |
43 |
|
* combined work), being distributed under the terms of the GNU General |
44 |
|
* Public License plus this exception. An independent module is a module |
45 |
|
* which is not derived from or based on XviD. |
46 |
|
* |
47 |
|
* Note that people who make modified versions of XviD are not obligated |
48 |
|
* to grant this special exception for their modified versions; it is |
49 |
|
* their choice whether to do so. The GNU General Public License gives |
50 |
|
* permission to release a modified version without this exception; this |
51 |
|
* exception also makes it possible to release a modified version which |
52 |
|
* carries forward this exception. |
53 |
|
* |
54 |
|
* $Id: adapt_quant.c,v 1.13 2002-11-26 23:44:11 edgomez Exp $ |
55 |
|
* |
56 |
|
****************************************************************************/ |
57 |
|
|
58 |
#include "../portab.h" |
#include "../portab.h" |
59 |
#include "adapt_quant.h" |
#include "adapt_quant.h" |
60 |
|
|
63 |
#define MAX(a,b) (((a) > (b)) ? (a) : (b)) |
#define MAX(a,b) (((a) > (b)) ? (a) : (b)) |
64 |
#define RDIFF(a,b) ((int)(a+0.5)-(int)(b+0.5)) |
#define RDIFF(a,b) ((int)(a+0.5)-(int)(b+0.5)) |
65 |
|
|
66 |
|
/***************************************************************************** |
67 |
|
* Functions |
68 |
|
****************************************************************************/ |
69 |
|
|
70 |
int |
int |
71 |
normalize_quantizer_field(float *in, |
normalize_quantizer_field(float *in, |
72 |
int *out, |
int *out, |
122 |
int min_quant, |
int min_quant, |
123 |
int max_quant, |
int max_quant, |
124 |
int mb_width, |
int mb_width, |
125 |
int mb_height) // no qstride because normalization |
int mb_height) /* no qstride because normalization */ |
126 |
{ |
{ |
127 |
int i, j, k, l; |
int i, j, k, l; |
128 |
|
|
148 |
if (!(quant = (float *) malloc(mb_width * mb_height * sizeof(float)))) |
if (!(quant = (float *) malloc(mb_width * mb_height * sizeof(float)))) |
149 |
return(-1); |
return(-1); |
150 |
|
|
151 |
if(!(val = (float *) malloc(mb_width * mb_height * sizeof(float)))) |
if(!(val = (float *) malloc(mb_width * mb_height * sizeof(float)))) { |
152 |
|
free(quant); |
153 |
return(-1); |
return(-1); |
154 |
|
} |
155 |
|
|
156 |
for (k = 0; k < mb_height; k++) { |
for (k = 0; k < mb_height; k++) { |
157 |
for (l = 0; l < mb_width; l++) // do this for all macroblocks individually |
for (l = 0; l < mb_width; l++) /* do this for all macroblocks individually */ |
158 |
{ |
{ |
159 |
quant[k * mb_width + l] = (float) framequant; |
quant[k * mb_width + l] = (float) framequant; |
160 |
|
|
161 |
// calculate luminance-masking |
/* calculate luminance-masking */ |
162 |
ptr = &buf[16 * k * stride + 16 * l]; // address of MB |
ptr = &buf[16 * k * stride + 16 * l]; /* address of MB */ |
163 |
|
|
164 |
val[k * mb_width + l] = 0.; |
val[k * mb_width + l] = 0.; |
165 |
|
|
180 |
if (((global <GlobalBrightThres) &&(global >GlobalDarkThres)) |
if (((global <GlobalBrightThres) &&(global >GlobalDarkThres)) |
181 |
|| (mid_range < MidRangeThres)) { |
|| (mid_range < MidRangeThres)) { |
182 |
for (k = 0; k < mb_height; k++) { |
for (k = 0; k < mb_height; k++) { |
183 |
for (l = 0; l < mb_width; l++) // do this for all macroblocks individually |
for (l = 0; l < mb_width; l++) /* do this for all macroblocks individually */ |
184 |
{ |
{ |
185 |
if (val[k * mb_width + l] < DarkThres) |
if (val[k * mb_width + l] < DarkThres) |
186 |
quant[k * mb_width + l] += |
quant[k * mb_width + l] += |
193 |
} |
} |
194 |
} |
} |
195 |
} |
} |
196 |
|
|
197 |
|
i = normalize_quantizer_field(quant, intquant, |
198 |
|
mb_width * mb_height, |
199 |
|
min_quant, max_quant); |
200 |
|
|
201 |
free(val); |
free(val); |
202 |
free(quant); |
free(quant); |
203 |
return normalize_quantizer_field(quant, intquant, mb_width * mb_height, |
|
204 |
min_quant, max_quant); |
return(i); |
205 |
|
|
206 |
} |
} |