diff options
author | Michael Niedermayer <michaelni@gmx.at> | 2013-04-22 19:41:31 +0200 |
---|---|---|
committer | Michael Niedermayer <michaelni@gmx.at> | 2013-04-22 20:00:54 +0200 |
commit | e2e9bee2daef9d5d5169f5360737eec8b76d7e0c (patch) | |
tree | 22ba6e50d34a036832d4ef8f903df335d3c9d63a /libavcodec/jpeg2000dwt.c | |
parent | c1c2b0b3390482bba39534f2f125d97e8964d7d5 (diff) | |
parent | f13888afcc5e528b80e4ae5bc4f0d9dadc76f6aa (diff) | |
download | ffmpeg-e2e9bee2daef9d5d5169f5360737eec8b76d7e0c.tar.gz |
Merge remote-tracking branch 'qatar/master'
* qatar/master:
avcodec: Bump minor for JPEG 2000 decoder
JPEG 2000 decoder for DCinema
The mqc code is merged, the rest is added independent of
the existing jpeg2000 decoder and encoder.
Conflicts:
Changelog
doc/general.texi
libavcodec/Makefile
libavcodec/allcodecs.c
libavcodec/mqc.c
libavcodec/mqc.h
libavcodec/mqcdec.c
libavcodec/version.h
tests/fate/video.mak
Merged-by: Michael Niedermayer <michaelni@gmx.at>
Diffstat (limited to 'libavcodec/jpeg2000dwt.c')
-rw-r--r-- | libavcodec/jpeg2000dwt.c | 371 |
1 files changed, 371 insertions, 0 deletions
diff --git a/libavcodec/jpeg2000dwt.c b/libavcodec/jpeg2000dwt.c new file mode 100644 index 0000000000..080b0a045d --- /dev/null +++ b/libavcodec/jpeg2000dwt.c @@ -0,0 +1,371 @@ +/* + * Discrete wavelet transform + * Copyright (c) 2007 Kamil Nowosad + * Copyright (c) 2013 Nicolas Bertrand <nicoinattendu@gmail.com> + * + * This file is part of FFmpeg. + * + * FFmpeg is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * FFmpeg 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 + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +/** + * @file + * Discrete wavelet transform + */ + +#include "libavutil/common.h" +#include "libavutil/mem.h" +#include "jpeg2000dwt.h" +#include "internal.h" + +/* Defines for 9/7 DWT lifting parameters. + * Parameters are in float. */ +#define F_LFTG_ALPHA 1.586134342059924f +#define F_LFTG_BETA 0.052980118572961f +#define F_LFTG_GAMMA 0.882911075530934f +#define F_LFTG_DELTA 0.443506852043971f +#define F_LFTG_K 1.230174104914001f +#define F_LFTG_X 1.625732422f +/* FIXME: Why use 1.625732422 instead of 1/F_LFTG_K? + * Incorrect value in JPEG2000 norm. + * see (ISO/IEC 15444:1 (version 2002) F.3.8.2 */ + +/* Lifting parameters in integer format. + * Computed as param = (float param) * (1 << 16) */ +#define I_LFTG_ALPHA 103949 +#define I_LFTG_BETA 3472 +#define I_LFTG_GAMMA 57862 +#define I_LFTG_DELTA 29066 +#define I_LFTG_K 80621 +#define I_LFTG_X 106544 + + +static inline void extend53(int *p, int i0, int i1) +{ + p[i0 - 1] = p[i0 + 1]; + p[i1] = p[i1 - 2]; + p[i0 - 2] = p[i0 + 2]; + p[i1 + 1] = p[i1 - 3]; +} + +static inline void extend97_float(float *p, int i0, int i1) +{ + int i; + + for (i = 1; i <= 4; i++) { + p[i0 - i] = p[i0 + i]; + p[i1 + i - 1] = p[i1 - i - 1]; + } +} + +static inline void extend97_int(int32_t *p, int i0, int i1) +{ + int i; + + for (i = 1; i <= 4; i++) { + p[i0 - i] = p[i0 + i]; + p[i1 + i - 1] = p[i1 - i - 1]; + } +} + +static void sr_1d53(int *p, int i0, int i1) +{ + int i; + + if (i1 == i0 + 1) + return; + + extend53(p, i0, i1); + + for (i = i0 / 2; i < i1 / 2 + 1; i++) + p[2 * i] -= (p[2 * i - 1] + p[2 * i + 1] + 2) >> 2; + for (i = i0 / 2; i < i1 / 2; i++) + p[2 * i + 1] += (p[2 * i] + p[2 * i + 2]) >> 1; +} + +static void dwt_decode53(DWTContext *s, int *t) +{ + int lev; + int w = s->linelen[s->ndeclevels - 1][0]; + int32_t *line = s->i_linebuf; + line += 3; + + for (lev = 0; lev < s->ndeclevels; lev++) { + int lh = s->linelen[lev][0], + lv = s->linelen[lev][1], + mh = s->mod[lev][0], + mv = s->mod[lev][1], + lp; + int *l; + + // HOR_SD + l = line + mh; + for (lp = 0; lp < lv; lp++) { + int i, j = 0; + // copy with interleaving + for (i = mh; i < lh; i += 2, j++) + l[i] = t[w * lp + j]; + for (i = 1 - mh; i < lh; i += 2, j++) + l[i] = t[w * lp + j]; + + sr_1d53(line, mh, mh + lh); + + for (i = 0; i < lh; i++) + t[w * lp + i] = l[i]; + } + + // VER_SD + l = line + mv; + for (lp = 0; lp < lh; lp++) { + int i, j = 0; + // copy with interleaving + for (i = mv; i < lv; i += 2, j++) + l[i] = t[w * j + lp]; + for (i = 1 - mv; i < lv; i += 2, j++) + l[i] = t[w * j + lp]; + + sr_1d53(line, mv, mv + lv); + + for (i = 0; i < lv; i++) + t[w * i + lp] = l[i]; + } + } +} + +static void sr_1d97_float(float *p, int i0, int i1) +{ + int i; + + if (i1 == i0 + 1) + return; + + extend97_float(p, i0, i1); + + /*step 1*/ + for (i = i0 / 2 - 1; i < i1 / 2 + 2; i++) + p[2 * i] *= F_LFTG_K; + /* step 2*/ + for (i = i0 / 2 - 2; i < i1 / 2 + 2; i++) + p[2 * i + 1] *= F_LFTG_X; + /* step 3*/ + for (i = i0 / 2 - 1; i < i1 / 2 + 2; i++) + p[2 * i] -= F_LFTG_DELTA * (p[2 * i - 1] + p[2 * i + 1]); + /* step 4 */ + for (i = i0 / 2 - 1; i < i1 / 2 + 1; i++) + p[2 * i + 1] -= F_LFTG_GAMMA * (p[2 * i] + p[2 * i + 2]); + /*step 5*/ + for (i = i0 / 2; i < i1 / 2 + 1; i++) + p[2 * i] += F_LFTG_BETA * (p[2 * i - 1] + p[2 * i + 1]); + /* step 6 */ + for (i = i0 / 2; i < i1 / 2; i++) + p[2 * i + 1] += F_LFTG_ALPHA * (p[2 * i] + p[2 * i + 2]); +} + +static void dwt_decode97_float(DWTContext *s, float *t) +{ + int lev; + int w = s->linelen[s->ndeclevels - 1][0]; + float *line = s->f_linebuf; + float *data = t; + /* position at index O of line range [0-5,w+5] cf. extend function */ + line += 5; + + for (lev = 0; lev < s->ndeclevels; lev++) { + int lh = s->linelen[lev][0], + lv = s->linelen[lev][1], + mh = s->mod[lev][0], + mv = s->mod[lev][1], + lp; + float *l; + // HOR_SD + l = line + mh; + for (lp = 0; lp < lv; lp++) { + int i, j = 0; + // copy with interleaving + for (i = mh; i < lh; i += 2, j++) + l[i] = data[w * lp + j]; + for (i = 1 - mh; i < lh; i += 2, j++) + l[i] = data[w * lp + j]; + + sr_1d97_float(line, mh, mh + lh); + + for (i = 0; i < lh; i++) + data[w * lp + i] = l[i]; + } + + // VER_SD + l = line + mv; + for (lp = 0; lp < lh; lp++) { + int i, j = 0; + // copy with interleaving + for (i = mv; i < lv; i += 2, j++) + l[i] = data[w * j + lp]; + for (i = 1 - mv; i < lv; i += 2, j++) + l[i] = data[w * j + lp]; + + sr_1d97_float(line, mv, mv + lv); + + for (i = 0; i < lv; i++) + data[w * i + lp] = l[i]; + } + } +} + +static void sr_1d97_int(int32_t *p, int i0, int i1) +{ + int i; + + if (i1 == i0 + 1) + return; + + extend97_int(p, i0, i1); + + /*step 1*/ + for (i = i0 / 2 - 1; i < i1 / 2 + 2; i++) + p[2 * i] = ((p[2 * i] * I_LFTG_K) + (1 << 15)) >> 16; + /* step 2*/ + for (i = i0 / 2 - 2; i < i1 / 2 + 2; i++) + p[2 * i + 1] = ((p[2 * i + 1] * I_LFTG_X) + (1 << 15)) >> 16; + /* step 3*/ + for (i = i0 / 2 - 1; i < i1 / 2 + 2; i++) + p[2 * i] -= (I_LFTG_DELTA * (p[2 * i - 1] + p[2 * i + 1]) + (1 << 15)) >> 16; + /* step 4 */ + for (i = i0 / 2 - 1; i < i1 / 2 + 1; i++) + p[2 * i + 1] -= (I_LFTG_GAMMA * (p[2 * i] + p[2 * i + 2]) + (1 << 15)) >> 16; + /*step 5*/ + for (i = i0 / 2; i < i1 / 2 + 1; i++) + p[2 * i] += (I_LFTG_BETA * (p[2 * i - 1] + p[2 * i + 1]) + (1 << 15)) >> 16; + /* step 6 */ + for (i = i0 / 2; i < i1 / 2; i++) + p[2 * i + 1] += (I_LFTG_ALPHA * (p[2 * i] + p[2 * i + 2]) + (1 << 15)) >> 16; +} + +static void dwt_decode97_int(DWTContext *s, int32_t *t) +{ + int lev; + int w = s->linelen[s->ndeclevels - 1][0]; + int32_t *line = s->i_linebuf; + int32_t *data = t; + /* position at index O of line range [0-5,w+5] cf. extend function */ + line += 5; + + for (lev = 0; lev < s->ndeclevels; lev++) { + int lh = s->linelen[lev][0], + lv = s->linelen[lev][1], + mh = s->mod[lev][0], + mv = s->mod[lev][1], + lp; + int32_t *l; + // HOR_SD + l = line + mh; + for (lp = 0; lp < lv; lp++) { + int i, j = 0; + // copy with interleaving + for (i = mh; i < lh; i += 2, j++) + l[i] = data[w * lp + j]; + for (i = 1 - mh; i < lh; i += 2, j++) + l[i] = data[w * lp + j]; + + sr_1d97_int(line, mh, mh + lh); + + for (i = 0; i < lh; i++) + data[w * lp + i] = l[i]; + } + + // VER_SD + l = line + mv; + for (lp = 0; lp < lh; lp++) { + int i, j = 0; + // copy with interleaving + for (i = mv; i < lv; i += 2, j++) + l[i] = data[w * j + lp]; + for (i = 1 - mv; i < lv; i += 2, j++) + l[i] = data[w * j + lp]; + + sr_1d97_int(line, mv, mv + lv); + + for (i = 0; i < lv; i++) + data[w * i + lp] = l[i]; + } + } +} + +int ff_jpeg2000_dwt_init(DWTContext *s, uint16_t border[2][2], + int decomp_levels, int type) +{ + int i, j, lev = decomp_levels, maxlen, + b[2][2]; + + s->ndeclevels = decomp_levels; + s->type = type; + + for (i = 0; i < 2; i++) + for (j = 0; j < 2; j++) + b[i][j] = border[i][j]; + + maxlen = FFMAX(b[0][1] - b[0][0], + b[1][1] - b[1][0]); + while (--lev >= 0) + for (i = 0; i < 2; i++) { + s->linelen[lev][i] = b[i][1] - b[i][0]; + s->mod[lev][i] = b[i][0] & 1; + for (j = 0; j < 2; j++) + b[i][j] = (b[i][j] + 1) >> 1; + } + switch (type) { + case FF_DWT97: + s->f_linebuf = av_malloc((maxlen + 12) * sizeof(*s->f_linebuf)); + if (!s->f_linebuf) + return AVERROR(ENOMEM); + break; + case FF_DWT97_INT: + s->i_linebuf = av_malloc((maxlen + 12) * sizeof(*s->i_linebuf)); + if (!s->i_linebuf) + return AVERROR(ENOMEM); + break; + case FF_DWT53: + s->i_linebuf = av_malloc((maxlen + 6) * sizeof(*s->i_linebuf)); + if (!s->i_linebuf) + return AVERROR(ENOMEM); + break; + default: + return -1; + } + return 0; +} + +int ff_dwt_decode(DWTContext *s, void *t) +{ + switch (s->type) { + case FF_DWT97: + dwt_decode97_float(s, t); + break; + case FF_DWT97_INT: + dwt_decode97_int(s, t); + break; + case FF_DWT53: + dwt_decode53(s, t); + break; + default: + return -1; + } + return 0; +} + +void ff_dwt_destroy(DWTContext *s) +{ + av_freep(&s->f_linebuf); + av_freep(&s->i_linebuf); +} |