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

Diff of /trunk/xvidcore/src/quant/adapt_quant.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 3, Fri Mar 8 02:46:11 2002 UTC revision 677, Tue Nov 26 23:44:11 2002 UTC
# Line 1  Line 1 
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    
61    #include <stdlib.h>                             /* free, malloc */
62    
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  int normalize_quantizer_field(float *in, int *out, int num, int min_quant, int max_quant)  /*****************************************************************************
67     * Functions
68     ****************************************************************************/
69    
70    int
71    normalize_quantizer_field(float *in,
72                                                      int *out,
73                                                      int num,
74                                                      int min_quant,
75                                                      int max_quant)
76  {  {
77          int i;          int i;
78          int finished;          int finished;
79    
80          do          do {
         {  
81                  finished = 1;                  finished = 1;
82                  for(i = 1; i < num; i++)                  for (i = 1; i < num; i++) {
83                  {                          if (RDIFF(in[i], in[i - 1]) > 2) {
                         if(RDIFF(in[i], in[i-1]) > 2)  
             {  
84                                  in[i] -= (float) 0.5;                                  in[i] -= (float) 0.5;
85                                  finished = 0;                                  finished = 0;
86                          }                          } else if (RDIFF(in[i], in[i - 1]) < -2) {
                         else if(RDIFF(in[i], in[i-1]) < -2)  
                         {  
87                                  in[i-1] -= (float) 0.5;                                  in[i-1] -= (float) 0.5;
88                                  finished = 0;                                  finished = 0;
89                          }                          }
90    
91            if(in[i] > max_quant)                          if (in[i] > max_quant) {
                   {  
92                            in[i] = (float) max_quant;                            in[i] = (float) max_quant;
93                            finished = 0;                            finished = 0;
94                    }                    }
95            if(in[i] < min_quant)                          if (in[i] < min_quant) {
                   {  
96                            in[i] = (float) min_quant;                            in[i] = (float) min_quant;
97                            finished = 0;                            finished = 0;
98                    }                    }
99            if(in[i-1] > max_quant)                          if (in[i - 1] > max_quant) {
                   {  
100                            in[i-1] = (float) max_quant;                            in[i-1] = (float) max_quant;
101                            finished = 0;                            finished = 0;
102                    }                    }
103            if(in[i-1] < min_quant)                          if (in[i - 1] < min_quant) {
                   {  
104                            in[i-1] = (float) min_quant;                            in[i-1] = (float) min_quant;
105                            finished = 0;                            finished = 0;
106                    }                    }
# Line 55  Line 114 
114          return (int) (in[0] + 0.5);          return (int) (in[0] + 0.5);
115  }  }
116    
117  int adaptive_quantization(unsigned char* buf, int stride, int* intquant,  int
118          int framequant, int min_quant, int max_quant,  adaptive_quantization(unsigned char *buf,
119                  int mb_width, int mb_height)  // no qstride because normalization                                            int stride,
120                                              int *intquant,
121                                              int framequant,
122                                              int min_quant,
123                                              int max_quant,
124                                              int mb_width,
125                                              int mb_height)        /* no qstride because normalization */
126  {  {
127          int i,j,k,l;          int i,j,k,l;
128    
129          static float *quant;          float *quant;
130          unsigned char *ptr;          unsigned char *ptr;
131          float *val;          float *val;
132          float global = 0.;          float global = 0.;
# Line 80  Line 145 
145          const float LowerLimit = 25;          const float LowerLimit = 25;
146    
147    
         if(!quant)  
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          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);
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    
# Line 103  Line 169 
169                                  val[k*mb_width+l] /= 256.;                                  val[k*mb_width+l] /= 256.;
170                                  global += val[k*mb_width+l];                                  global += val[k*mb_width+l];
171    
172                                  if((val[k*mb_width+l] > LowerLimit) && (val[k*mb_width+l] < UpperLimit))                          if ((val[k * mb_width + l] > LowerLimit) &&
173                                    (val[k * mb_width + l] < UpperLimit))
174                                          mid_range++;                                          mid_range++;
175                  }                  }
176          }          }
177    
178          global /= mb_width*mb_height;          global /= mb_width*mb_height;
179    
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] += DarkAmpl*(DarkThres-val[k*mb_width+l])/DarkThres;                                          quant[k * mb_width + l] +=
187                                                    DarkAmpl * (DarkThres -
188                                                                            val[k * mb_width + l]) / DarkThres;
189                                  else if (val[k*mb_width+l]>BrightThres)                                  else if (val[k*mb_width+l]>BrightThres)
190                                          quant[k*mb_width+l] += BrightAmpl*(val[k*mb_width+l]-BrightThres)/(255-BrightThres);                                          quant[k * mb_width + l] +=
191                                                    BrightAmpl * (val[k * mb_width + l] -
192                                                                              BrightThres) / (255 - BrightThres);
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          return normalize_quantizer_field(quant, intquant, mb_width*mb_height, min_quant, max_quant);          free(quant);
203    
204            return(i);
205    
206  }  }
Legend:
Removed from v.3  
changed lines
  Added in v.677
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4