[cvs] / xvidcore / src / quant / adapt_quant.c Repository:
ViewVC logotype

Annotation of /xvidcore/src/quant/adapt_quant.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.6.2.1 - (view) (download)

1 : Isibaar 1.1 #include "../portab.h"
2 :     #include "adapt_quant.h"
3 :    
4 : edgomez 1.3 #include <stdlib.h> /* free, malloc */
5 : edgomez 1.2
6 : Isibaar 1.1 #define MAX(a,b) (((a) > (b)) ? (a) : (b))
7 :     #define RDIFF(a,b) ((int)(a+0.5)-(int)(b+0.5))
8 :    
9 : edgomez 1.3 int
10 :     normalize_quantizer_field(float *in,
11 :     int *out,
12 :     int num,
13 :     int min_quant,
14 :     int max_quant)
15 : Isibaar 1.1 {
16 :     int i;
17 :     int finished;
18 : edgomez 1.3
19 :     do {
20 : Isibaar 1.1 finished = 1;
21 : edgomez 1.3 for (i = 1; i < num; i++) {
22 :     if (RDIFF(in[i], in[i - 1]) > 2) {
23 : Isibaar 1.1 in[i] -= (float) 0.5;
24 :     finished = 0;
25 : edgomez 1.3 } else if (RDIFF(in[i], in[i - 1]) < -2) {
26 :     in[i - 1] -= (float) 0.5;
27 : Isibaar 1.1 finished = 0;
28 :     }
29 : edgomez 1.3
30 :     if (in[i] > max_quant) {
31 : edgomez 1.2 in[i] = (float) max_quant;
32 :     finished = 0;
33 :     }
34 : edgomez 1.3 if (in[i] < min_quant) {
35 : edgomez 1.2 in[i] = (float) min_quant;
36 :     finished = 0;
37 :     }
38 : edgomez 1.3 if (in[i - 1] > max_quant) {
39 :     in[i - 1] = (float) max_quant;
40 : edgomez 1.2 finished = 0;
41 :     }
42 : edgomez 1.3 if (in[i - 1] < min_quant) {
43 :     in[i - 1] = (float) min_quant;
44 : edgomez 1.2 finished = 0;
45 :     }
46 : Isibaar 1.1 }
47 : edgomez 1.3 } while (!finished);
48 :    
49 : Isibaar 1.1 out[0] = 0;
50 :     for (i = 1; i < num; i++)
51 : edgomez 1.3 out[i] = RDIFF(in[i], in[i - 1]);
52 :    
53 : Isibaar 1.1 return (int) (in[0] + 0.5);
54 :     }
55 :    
56 : edgomez 1.3 int
57 :     adaptive_quantization(unsigned char *buf,
58 :     int stride,
59 :     int *intquant,
60 :     int framequant,
61 :     int min_quant,
62 :     int max_quant,
63 :     int mb_width,
64 :     int mb_height) // no qstride because normalization
65 : Isibaar 1.1 {
66 : edgomez 1.3 int i, j, k, l;
67 :    
68 : edgomez 1.6 float *quant;
69 : Isibaar 1.1 unsigned char *ptr;
70 :     float *val;
71 : Isibaar 1.5 float global = 0.;
72 :     uint32_t mid_range = 0;
73 :    
74 :     const float DarkAmpl = 14 / 2;
75 :     const float BrightAmpl = 10 / 2;
76 :     const float DarkThres = 70;
77 :     const float BrightThres = 200;
78 :    
79 :     const float GlobalDarkThres = 60;
80 :     const float GlobalBrightThres = 170;
81 :    
82 :     const float MidRangeThres = 20;
83 :     const float UpperLimit = 200;
84 :     const float LowerLimit = 25;
85 : Isibaar 1.1
86 : edgomez 1.3
87 : edgomez 1.6 if (!(quant = (float *) malloc(mb_width * mb_height * sizeof(float))))
88 :     return(-1);
89 : Isibaar 1.1
90 : edgomez 1.6 if(!(val = (float *) malloc(mb_width * mb_height * sizeof(float))))
91 :     return(-1);
92 : Isibaar 1.1
93 : edgomez 1.3 for (k = 0; k < mb_height; k++) {
94 :     for (l = 0; l < mb_width; l++) // do this for all macroblocks individually
95 : Isibaar 1.1 {
96 : edgomez 1.3 quant[k * mb_width + l] = (float) framequant;
97 :    
98 : Isibaar 1.1 // calculate luminance-masking
99 : edgomez 1.3 ptr = &buf[16 * k * stride + 16 * l]; // address of MB
100 : Isibaar 1.1
101 : edgomez 1.3 val[k * mb_width + l] = 0.;
102 :    
103 :     for (i = 0; i < 16; i++)
104 :     for (j = 0; j < 16; j++)
105 :     val[k * mb_width + l] += ptr[i * stride + j];
106 :     val[k * mb_width + l] /= 256.;
107 : Isibaar 1.5 global +=val[k * mb_width + l];
108 :    
109 :     if ((val[k * mb_width + l] > LowerLimit) &&
110 :     (val[k * mb_width + l] < UpperLimit))
111 :     mid_range++;
112 : Isibaar 1.1 }
113 :     }
114 :    
115 : Isibaar 1.5 global /=mb_width * mb_height;
116 : Isibaar 1.1
117 : Isibaar 1.5 if (((global <GlobalBrightThres) &&(global >GlobalDarkThres))
118 :     || (mid_range < MidRangeThres)) {
119 :     for (k = 0; k < mb_height; k++) {
120 :     for (l = 0; l < mb_width; l++) // do this for all macroblocks individually
121 :     {
122 :     if (val[k * mb_width + l] < DarkThres)
123 :     quant[k * mb_width + l] +=
124 :     DarkAmpl * (DarkThres -
125 :     val[k * mb_width + l]) / DarkThres;
126 :     else if (val[k * mb_width + l] > BrightThres)
127 :     quant[k * mb_width + l] +=
128 :     BrightAmpl * (val[k * mb_width + l] -
129 :     BrightThres) / (255 - BrightThres);
130 :     }
131 : Isibaar 1.1 }
132 :     }
133 : suxen_drol 1.6.2.1
134 :     i = normalize_quantizer_field(quant, intquant,
135 :     mb_width * mb_height,
136 :     min_quant, max_quant);
137 :    
138 : Isibaar 1.1 free(val);
139 : edgomez 1.6 free(quant);
140 : suxen_drol 1.6.2.1
141 :     return(i);
142 :    
143 : edgomez 1.2 }

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