[svn] / branches / dev-api-4 / xvidcore / src / quant / adapt_quant.c Repository:
ViewVC logotype

Annotation of /branches/dev-api-4/xvidcore/src/quant/adapt_quant.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1054 - (view) (download)

1 : edgomez 1054 /*****************************************************************************
2 :     *
3 :     * XVID MPEG-4 VIDEO CODEC
4 :     * - Deprecated code -
5 :     *
6 :     * Copyright(C) 2002 Peter Ross <pross@xvid.org>
7 :     * 2002 Christoph Lampert <gruel@web.de>
8 :     *
9 :     * This program is free software ; you can redistribute it and/or modify
10 :     * it under the terms of the GNU General Public License as published by
11 :     * the Free Software Foundation ; either version 2 of the License, or
12 :     * (at your option) any later version.
13 :     *
14 :     * This program is distributed in the hope that it will be useful,
15 :     * but WITHOUT ANY WARRANTY ; without even the implied warranty of
16 :     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 :     * GNU General Public License for more details.
18 :     *
19 :     * You should have received a copy of the GNU General Public License
20 :     * along with this program ; if not, write to the Free Software
21 :     * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 :     *
23 :     * $Id: adapt_quant.c,v 1.14.2.2 2003-06-09 13:55:10 edgomez Exp $
24 :     *
25 :     ****************************************************************************/
26 :    
27 : Isibaar 3 #include "../portab.h"
28 : edgomez 851 #include "../global.h"
29 : Isibaar 3 #include "adapt_quant.h"
30 :    
31 : edgomez 195 #include <stdlib.h> /* free, malloc */
32 : edgomez 18
33 : Isibaar 3 #define RDIFF(a,b) ((int)(a+0.5)-(int)(b+0.5))
34 :    
35 : edgomez 195 int
36 :     normalize_quantizer_field(float *in,
37 :     int *out,
38 :     int num,
39 :     int min_quant,
40 :     int max_quant)
41 : Isibaar 3 {
42 :     int i;
43 :     int finished;
44 : edgomez 195
45 :     do {
46 : Isibaar 3 finished = 1;
47 : edgomez 195 for (i = 1; i < num; i++) {
48 :     if (RDIFF(in[i], in[i - 1]) > 2) {
49 : Isibaar 3 in[i] -= (float) 0.5;
50 :     finished = 0;
51 : edgomez 195 } else if (RDIFF(in[i], in[i - 1]) < -2) {
52 :     in[i - 1] -= (float) 0.5;
53 : Isibaar 3 finished = 0;
54 :     }
55 : edgomez 195
56 :     if (in[i] > max_quant) {
57 : edgomez 18 in[i] = (float) max_quant;
58 :     finished = 0;
59 :     }
60 : edgomez 195 if (in[i] < min_quant) {
61 : edgomez 18 in[i] = (float) min_quant;
62 :     finished = 0;
63 :     }
64 : edgomez 195 if (in[i - 1] > max_quant) {
65 :     in[i - 1] = (float) max_quant;
66 : edgomez 18 finished = 0;
67 :     }
68 : edgomez 195 if (in[i - 1] < min_quant) {
69 :     in[i - 1] = (float) min_quant;
70 : edgomez 18 finished = 0;
71 :     }
72 : Isibaar 3 }
73 : edgomez 195 } while (!finished);
74 :    
75 : Isibaar 3 out[0] = 0;
76 :     for (i = 1; i < num; i++)
77 : edgomez 195 out[i] = RDIFF(in[i], in[i - 1]);
78 :    
79 : Isibaar 3 return (int) (in[0] + 0.5);
80 :     }
81 :    
82 : edgomez 195 int
83 :     adaptive_quantization(unsigned char *buf,
84 :     int stride,
85 :     int *intquant,
86 :     int framequant,
87 :     int min_quant,
88 :     int max_quant,
89 :     int mb_width,
90 : edgomez 1053 int mb_height) /* no qstride because normalization */
91 : Isibaar 3 {
92 : edgomez 195 int i, j, k, l;
93 :    
94 : edgomez 391 float *quant;
95 : Isibaar 3 unsigned char *ptr;
96 :     float *val;
97 : Isibaar 357 float global = 0.;
98 :     uint32_t mid_range = 0;
99 : Isibaar 3
100 : Isibaar 357 const float DarkAmpl = 14 / 2;
101 :     const float BrightAmpl = 10 / 2;
102 :     const float DarkThres = 70;
103 :     const float BrightThres = 200;
104 : Isibaar 3
105 : Isibaar 357 const float GlobalDarkThres = 60;
106 :     const float GlobalBrightThres = 170;
107 : Isibaar 3
108 : Isibaar 357 const float MidRangeThres = 20;
109 :     const float UpperLimit = 200;
110 :     const float LowerLimit = 25;
111 : Isibaar 3
112 : Isibaar 357
113 : edgomez 391 if (!(quant = (float *) malloc(mb_width * mb_height * sizeof(float))))
114 :     return(-1);
115 : Isibaar 3
116 : edgomez 851 if(!(val = (float *) malloc(mb_width * mb_height * sizeof(float))))
117 : edgomez 391 return(-1);
118 : Isibaar 3
119 : edgomez 195 for (k = 0; k < mb_height; k++) {
120 : edgomez 1053 for (l = 0; l < mb_width; l++) /* do this for all macroblocks individually */
121 : Isibaar 3 {
122 : edgomez 195 quant[k * mb_width + l] = (float) framequant;
123 :    
124 : edgomez 1053 /* calculate luminance-masking */
125 :     ptr = &buf[16 * k * stride + 16 * l]; /* address of MB */
126 : Isibaar 3
127 : edgomez 195 val[k * mb_width + l] = 0.;
128 :    
129 :     for (i = 0; i < 16; i++)
130 :     for (j = 0; j < 16; j++)
131 :     val[k * mb_width + l] += ptr[i * stride + j];
132 :     val[k * mb_width + l] /= 256.;
133 : Isibaar 357 global +=val[k * mb_width + l];
134 :    
135 :     if ((val[k * mb_width + l] > LowerLimit) &&
136 :     (val[k * mb_width + l] < UpperLimit))
137 :     mid_range++;
138 : Isibaar 3 }
139 :     }
140 :    
141 : Isibaar 357 global /=mb_width * mb_height;
142 : Isibaar 3
143 : Isibaar 357 if (((global <GlobalBrightThres) &&(global >GlobalDarkThres))
144 :     || (mid_range < MidRangeThres)) {
145 :     for (k = 0; k < mb_height; k++) {
146 : edgomez 1053 for (l = 0; l < mb_width; l++) /* do this for all macroblocks individually */
147 : Isibaar 357 {
148 :     if (val[k * mb_width + l] < DarkThres)
149 :     quant[k * mb_width + l] +=
150 :     DarkAmpl * (DarkThres -
151 :     val[k * mb_width + l]) / DarkThres;
152 :     else if (val[k * mb_width + l] > BrightThres)
153 :     quant[k * mb_width + l] +=
154 :     BrightAmpl * (val[k * mb_width + l] -
155 :     BrightThres) / (255 - BrightThres);
156 :     }
157 : Isibaar 3 }
158 :     }
159 : edgomez 495
160 :     i = normalize_quantizer_field(quant, intquant,
161 :     mb_width * mb_height,
162 :     min_quant, max_quant);
163 :    
164 : Isibaar 3 free(val);
165 : edgomez 391 free(quant);
166 : edgomez 495
167 :     return(i);
168 :    
169 : edgomez 18 }

No admin address has been configured
ViewVC Help
Powered by ViewVC 1.0.4