git.sesse.net Git - ffmpeg/blob - libavcodec/j2k_dwt.h

   1 /*
   2  * Discrete wavelet transform
   3  * Copyright (c) 2007 Kamil Nowosad
   4  *
   5  * This file is part of FFmpeg.
   6  *
   7  * FFmpeg is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * FFmpeg is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with FFmpeg; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20  */
  21
  22 #ifndef AVCODEC_DWT_H
  23 #define AVCODEC_DWT_H
  24
  25 /**
  26  * Discrete wavelet transform
  27  * @file
  28  * @author Kamil Nowosad
  29  */
  30
  31 #include "avcodec.h"
  32
  33 #define FF_DWT_MAX_DECLVLS 32 ///< max number of decomposition levels
  34
  35 enum DWTType{
  36     FF_DWT97,
  37     FF_DWT53
  38 };
  39
  40 typedef struct {
  41     ///line lengths {horizontal, vertical} in consecutive decomposition levels
  42     uint16_t linelen[FF_DWT_MAX_DECLVLS][2];
  43     uint8_t  mod[FF_DWT_MAX_DECLVLS][2]; ///< coordinates (x0, y0) of decomp. levels mod 2
  44     uint8_t  ndeclevels;                 ///< number of decomposition levels
  45     uint8_t  type;                       ///< 0 for 9/7; 1 for 5/3
  46     void     *linebuf;                   ///< buffer used by transform (int or float)
  47 } DWTContext;
  48
  49 /**
  50  * initialize DWT
  51  * @param s DWT context
  52  * @param border coordinates of transformed region {{x0, x1}, {y0, y1}}
  53  * @param decomp_levels number of decomposition levels
  54  * @param type 0 for DWT 9/7; 1 for DWT 5/3
  55  */
  56 int ff_j2k_dwt_init(DWTContext *s, uint16_t border[2][2], int decomp_levels, int type);
  57
  58 int ff_j2k_dwt_encode(DWTContext *s, int *t);
  59 int ff_j2k_dwt_decode(DWTContext *s, int *t);
  60
  61 void ff_j2k_dwt_destroy(DWTContext *s);
  62
  63 #endif /* AVCODEC_DWT_H */