File: deringing.cpp

package info (click to toggle)
libjpeg 0.0~git20210129.91985dc-1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 3,664 kB
  • sloc: cpp: 36,104; makefile: 613; ansic: 275; sh: 54; python: 38; perl: 11
file content (230 lines) | stat: -rw-r--r-- 7,600 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
/*************************************************************************

    This project implements a complete(!) JPEG (Recommendation ITU-T
    T.81 | ISO/IEC 10918-1) codec, plus a library that can be used to
    encode and decode JPEG streams. 
    It also implements ISO/IEC 18477 aka JPEG XT which is an extension
    towards intermediate, high-dynamic-range lossy and lossless coding
    of JPEG. In specific, it supports ISO/IEC 18477-3/-6/-7/-8 encoding.

    Note that only Profiles C and D of ISO/IEC 18477-7 are supported
    here. Check the JPEG XT reference software for a full implementation
    of ISO/IEC 18477-7.

    Copyright (C) 2012-2018 Thomas Richter, University of Stuttgart and
    Accusoft. (C) 2019-2020 Thomas Richter, Fraunhofer IIS.

    This program is available under two licenses, GPLv3 and the ITU
    Software licence Annex A Option 2, RAND conditions.

    For the full text of the GPU license option, see README.license.gpl.
    For the full text of the ITU license option, see README.license.itu.
    
    You may freely select between these two options.

    For the GPL option, please note the following:

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.

*************************************************************************/
/*
** This class implements a simple deblocking filter to avoid
** DCT artifacts (Gibbs Phenomenon)
**
** $Id: deringing.cpp,v 1.2 2016/10/28 13:58:54 thor Exp $
**
*/

/// Includes
#include "tools/environment.hpp"
#include "marker/frame.hpp"
#include "dct/dct.hpp"
#include "dct/deringing.hpp"
#include "std/string.hpp"
///

/// DeRinger::DeRinger
// Create a new deblocker for a given frame and component.
DeRinger::DeRinger(class Frame *frame,class DCT *dct)
  : JKeeper(frame->EnvironOf()), m_pDCT(dct)
{ 
  int preshift = dct->PreshiftOf();
  //
  // Min and max are the minimum and maximum sample values that
  // would be reconstructed to the extreme values.
  m_lMin   = (1L << preshift) - 1;
  m_lMax   = ((1L << frame->HiddenPrecisionOf()) - 1) << preshift;
  m_lDelta = (1L << preshift);
}
///

/// DeRinger::~DeRinger
DeRinger::~DeRinger(void)
{
}
///

#if ACCUSOFT_CODE
/// DeRinger::Smooth
// Run a simple Gaussian smoothing filter on the second argument, place
// the result in the first argument, or copy from the original if mask is
// not set.
void DeRinger::Smooth(LONG target[64],const LONG src[64],const LONG mask[64])
{
  int x,y;

  for(y = 0;y < 8;y++) {
    for(x = 0;x < 8;x++) {
      if (mask[x + (y << 3)]) {
        LONG center = src[x                   + (y << 3)];
        LONG left   = src[((x > 0)?(x - 1):x) + (y << 3)];
        LONG right  = src[((x < 7)?(x + 1):x) + (y << 3)];
        LONG top    = src[x                   + (((y > 0)?(y - 1):(y)) << 3)];
        LONG bottom = src[x                   + (((y < 7)?(y + 1):(y)) << 3)];
        LONG round  = (center << 2) + left + right + top + bottom;
        //
        // Insert the average now.
        round = ((round | 1) + 3) >> 3;
        //
        // Make sure we do not smooth the pixels out in the wrong direction,
        // we must at least preserve minimum and maximum.
        if (center <= m_lMin && round > m_lMin) {
          round = m_lMin;
        } else if (center >= m_lMax && round < m_lMax) {
          round = m_lMax;
        } 
        target[x + (y << 3)] = round;
      } else {
        // Pixels except the minimum and maximum are not affected.
        target[x + (y << 3)] = src[x + (y << 3)];
      }
    }
  }
}
///
#endif  

/// DeRinger::DeRing
// Deblock the given image block (non-DCT-transformed) by including overshooting
// in the extreme image parts, or undershooting in the dark image regions.
#if ACCUSOFT_CODE 
void DeRinger::DeRing(const LONG block[64],LONG dst[64],LONG dcshift)
#else
void DeRinger::DeRing(const LONG [64],LONG [64],LONG)
#endif  
{
#if ACCUSOFT_CODE 
  int i,c1 = 0,c2 = 0;
  LONG mask[64]; // Is set to one if we can modify this coefficient.
  LONG delta = m_pDCT->BucketSizes()[0];
  LONG sum   = 0;
  if (delta < m_lDelta)
    delta = m_lDelta;
  //
  // Check how many samples are there that could require processing by this
  // filter. These are samples that are either below min or above max.
  for(i = 0;i < 64;i++) {
    LONG v = block[i];
    if (v <= m_lMin) {
      c1++;
      mask[i] = -1;
    } else if (v >= m_lMax) {
      c2++;
      mask[i] = +1;
    } else {
      mask[i] = 0;
    }
    sum += block[i];
  }
  m_pDCT->TransformBlock(block,dst,dcshift);
  //
  // If there are no extreme samples, this makes all no sense at all.
  if ((c1 > 0 && c1 < 64) || (c2 > 0 && c2 < 64)) {
    LONG mod[64],tmp[64];
    // Compute maximum overshoot levels for minimum and maximum
    // coefficients. First, do not increase by more than two
    // DC quantization buckets, but by a minimum level that ensures
    // that most artifacts are gone.
    LONG overshoot    = (delta << 1);
    LONG minovershoot = (31 << m_pDCT->PreshiftOf());
    if (overshoot < minovershoot)
      overshoot = minovershoot;
    // The maximum overshoot moves the average grey level just
    // to the maximum or minimum level.
    LONG max = ((m_lMax << 6) - sum) >> 6;
    LONG min = (sum - (m_lMin << 6)) >> 6;
    if (max > overshoot)
      max = overshoot;
    if (min > overshoot)
      min = overshoot;
    // Compute from the overshoot the maximum and minimum amplitude
    max = m_lMax + max;
    min = m_lMin - min;
    //
    // First increase the amplitude by all overshooting coefficients.
    for(i = 0;i < 64;i++) {
      if (mask[i] > 0) {
        mod[i] = max;
      } else if (mask[i] < 0) {
        mod[i] = min;
      } else {
        mod[i] = block[i];
      }
    }
    //
    // Now run a smoothing filter over those parts we modified to
    // improve compressibility. This is a simple Gaussian filter
    // that should hopefully remove high contrast edges due to the
    // large amplitudes, and thus the high frequencies.
    Smooth(tmp,mod,mask);
    Smooth(mod,tmp,mask);
    Smooth(tmp,mod,mask);
    //
    // Forward transform the smoothed data.
    m_pDCT->TransformBlock(tmp,mod,dcshift);
    //
    // Now copy the data back into the final DCT.
    // Try to cover the corrections as good as possible, but
    // avoid changing the magnitude category to keep the
    // rate increase minimal.
    for(i = 0;i < 64;i++) {
      LONG data = dst[i];
      LONG mdat = mod[i];
      if (i == 0) {
        dst[i] = mdat;
      } else if (data) {
        LONG max = 1;
        data     = (data < 0)?(-data):(data);
        while(data >>= 1) {
          max = (max << 1) | 1;
        }
        if (mdat > max) {
          dst[i] = max;
        } else if (mdat < -max) {
          dst[i] = -max;
        } else {
          dst[i] = mdat;
        }
      } else {
        dst[i] = data;
      }
    }
  }
#else
  JPG_THROW(NOT_IMPLEMENTED,"DeRinger::DeRing",
            "the de-ringing filter is not available in this software release");
#endif  
}
///