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

   1 /*
   2  * DSP functions for Indeo Video Interactive codecs (Indeo4 and Indeo5)
   3  *
   4  * Copyright (c) 2009-2011 Maxim Poliakovski
   5  *
   6  * This file is part of FFmpeg.
   7  *
   8  * FFmpeg is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2.1 of the License, or (at your option) any later version.
  12  *
  13  * FFmpeg is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with FFmpeg; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21  */
  22
  23 /**
  24  * @file
  25  * DSP functions (inverse transforms, motion compensations, wavelet recompostion)
  26  * for Indeo Video Interactive codecs.
  27  */
  28
  29 #ifndef AVCODEC_IVI_DSP_H
  30 #define AVCODEC_IVI_DSP_H
  31
  32 #include "avcodec.h"
  33 #include "ivi_common.h"
  34
  35 /**
  36  *  5/3 wavelet recomposition filter for Indeo5
  37  *
  38  *  @param[in]   plane        pointer to the descriptor of the plane being processed
  39  *  @param[out]  dst          pointer to the destination buffer
  40  *  @param[in]   dst_pitch    pitch of the destination buffer
  41  */
  42 void ff_ivi_recompose53(const IVIPlaneDesc *plane, uint8_t *dst,
  43                         const int dst_pitch);
  44
  45 /**
  46  *  Haar wavelet recomposition filter for Indeo 4
  47  *
  48  *  @param[in]  plane        pointer to the descriptor of the plane being processed
  49  *  @param[out] dst          pointer to the destination buffer
  50  *  @param[in]  dst_pitch    pitch of the destination buffer
  51  */
  52 void ff_ivi_recompose_haar(const IVIPlaneDesc *plane, uint8_t *dst,
  53                            const int dst_pitch);
  54
  55 /**
  56  *  two-dimensional inverse Haar 8x8 transform for Indeo 4
  57  *
  58  *  @param[in]  in        pointer to the vector of transform coefficients
  59  *  @param[out] out       pointer to the output buffer (frame)
  60  *  @param[in]  pitch     pitch to move to the next y line
  61  *  @param[in]  flags     pointer to the array of column flags:
  62  *                        != 0 - non_empty column, 0 - empty one
  63  *                        (this array must be filled by caller)
  64  */
  65 void ff_ivi_inverse_haar_8x8(const int32_t *in, int16_t *out, uint32_t pitch,
  66                              const uint8_t *flags);
  67 void ff_ivi_inverse_haar_8x1(const int32_t *in, int16_t *out, uint32_t pitch,
  68                              const uint8_t *flags);
  69 void ff_ivi_inverse_haar_1x8(const int32_t *in, int16_t *out, uint32_t pitch,
  70                              const uint8_t *flags);
  71
  72 /**
  73  *  one-dimensional inverse 8-point Haar transform on rows for Indeo 4
  74  *
  75  *  @param[in]  in        pointer to the vector of transform coefficients
  76  *  @param[out] out       pointer to the output buffer (frame)
  77  *  @param[in]  pitch     pitch to move to the next y line
  78  *  @param[in]  flags     pointer to the array of column flags:
  79  *                        != 0 - non_empty column, 0 - empty one
  80  *                        (this array must be filled by caller)
  81  */
  82 void ff_ivi_row_haar8(const int32_t *in, int16_t *out, uint32_t pitch,
  83                       const uint8_t *flags);
  84
  85 /**
  86  *  one-dimensional inverse 8-point Haar transform on columns for Indeo 4
  87  *
  88  *  @param[in]  in        pointer to the vector of transform coefficients
  89  *  @param[out] out       pointer to the output buffer (frame)
  90  *  @param[in]  pitch     pitch to move to the next y line
  91  *  @param[in]  flags     pointer to the array of column flags:
  92  *                        != 0 - non_empty column, 0 - empty one
  93  *                        (this array must be filled by caller)
  94  */
  95 void ff_ivi_col_haar8(const int32_t *in, int16_t *out, uint32_t pitch,
  96                       const uint8_t *flags);
  97
  98 /**
  99  *  two-dimensional inverse Haar 4x4 transform for Indeo 4
 100  *
 101  *  @param[in]  in        pointer to the vector of transform coefficients
 102  *  @param[out] out       pointer to the output buffer (frame)
 103  *  @param[in]  pitch     pitch to move to the next y line
 104  *  @param[in]  flags     pointer to the array of column flags:
 105  *                        != 0 - non_empty column, 0 - empty one
 106  *                        (this array must be filled by caller)
 107  */
 108 void ff_ivi_inverse_haar_4x4(const int32_t *in, int16_t *out, uint32_t pitch,
 109                              const uint8_t *flags);
 110
 111 /**
 112  *  one-dimensional inverse 4-point Haar transform on rows for Indeo 4
 113  *
 114  *  @param[in]  in        pointer to the vector of transform coefficients
 115  *  @param[out] out       pointer to the output buffer (frame)
 116  *  @param[in]  pitch     pitch to move to the next y line
 117  *  @param[in]  flags     pointer to the array of column flags:
 118  *                        != 0 - non_empty column, 0 - empty one
 119  *                        (this array must be filled by caller)
 120  */
 121 void ff_ivi_row_haar4(const int32_t *in, int16_t *out, uint32_t pitch,
 122                       const uint8_t *flags);
 123
 124 /**
 125  *  one-dimensional inverse 4-point Haar transform on columns for Indeo 4
 126  *
 127  *  @param[in]  in        pointer to the vector of transform coefficients
 128  *  @param[out] out       pointer to the output buffer (frame)
 129  *  @param[in]  pitch     pitch to move to the next y line
 130  *  @param[in]  flags     pointer to the array of column flags:
 131  *                        != 0 - non_empty column, 0 - empty one
 132  *                        (this array must be filled by caller)
 133  */
 134 void ff_ivi_col_haar4(const int32_t *in, int16_t *out, uint32_t pitch,
 135                       const uint8_t *flags);
 136
 137 /**
 138  *  DC-only two-dimensional inverse Haar transform for Indeo 4.
 139  *  Performing the inverse transform in this case is equivalent to
 140  *  spreading DC_coeff >> 3 over the whole block.
 141  *
 142  *  @param[in]  in          pointer to the dc coefficient
 143  *  @param[out] out         pointer to the output buffer (frame)
 144  *  @param[in]  pitch       pitch to move to the next y line
 145  *  @param[in]  blk_size    transform block size
 146  */
 147 void ff_ivi_dc_haar_2d(const int32_t *in, int16_t *out, uint32_t pitch,
 148                        int blk_size);
 149
 150 /**
 151  *  two-dimensional inverse slant 8x8 transform
 152  *
 153  *  @param[in]    in      pointer to the vector of transform coefficients
 154  *  @param[out]   out     pointer to the output buffer (frame)
 155  *  @param[in]    pitch   pitch to move to the next y line
 156  *  @param[in]    flags   pointer to the array of column flags:
 157  *                        != 0 - non_empty column, 0 - empty one
 158  *                        (this array must be filled by caller)
 159  */
 160 void ff_ivi_inverse_slant_8x8(const int32_t *in, int16_t *out, uint32_t pitch,
 161                               const uint8_t *flags);
 162
 163 /**
 164  *  two-dimensional inverse slant 4x4 transform
 165  *
 166  *  @param[in]    in      pointer to the vector of transform coefficients
 167  *  @param[out]   out     pointer to the output buffer (frame)
 168  *  @param[in]    pitch   pitch to move to the next y line
 169  *  @param[in]    flags   pointer to the array of column flags:
 170  *                        != 0 - non_empty column, 0 - empty one
 171  *                        (this array must be filled by caller)
 172  */
 173 void ff_ivi_inverse_slant_4x4(const int32_t *in, int16_t *out, uint32_t pitch,
 174                               const uint8_t *flags);
 175
 176 /**
 177  *  DC-only two-dimensional inverse slant transform.
 178  *  Performing the inverse slant transform in this case is equivalent to
 179  *  spreading (DC_coeff + 1)/2 over the whole block.
 180  *  It works much faster than performing the slant transform on a vector of zeroes.
 181  *
 182  *  @param[in]    in          pointer to the dc coefficient
 183  *  @param[out]   out         pointer to the output buffer (frame)
 184  *  @param[in]    pitch       pitch to move to the next y line
 185  *  @param[in]    blk_size    transform block size
 186  */
 187 void ff_ivi_dc_slant_2d(const int32_t *in, int16_t *out, uint32_t pitch, int blk_size);
 188
 189 /**
 190  *  inverse 1D row slant transform
 191  *
 192  *  @param[in]    in      pointer to the vector of transform coefficients
 193  *  @param[out]   out     pointer to the output buffer (frame)
 194  *  @param[in]    pitch   pitch to move to the next y line
 195  *  @param[in]    flags   pointer to the array of column flags (unused here)
 196  */
 197 void ff_ivi_row_slant8(const int32_t *in, int16_t *out, uint32_t pitch,
 198                        const uint8_t *flags);
 199
 200 /**
 201  *  inverse 1D column slant transform
 202  *
 203  *  @param[in]    in      pointer to the vector of transform coefficients
 204  *  @param[out]   out     pointer to the output buffer (frame)
 205  *  @param[in]    pitch   pitch to move to the next y line
 206  *  @param[in]    flags   pointer to the array of column flags:
 207  *                        != 0 - non_empty column, 0 - empty one
 208  *                        (this array must be filled by caller)
 209  */
 210 void ff_ivi_col_slant8(const int32_t *in, int16_t *out, uint32_t pitch,
 211                        const uint8_t *flags);
 212
 213 /**
 214  *  inverse 1D row slant transform
 215  *
 216  *  @param[in]    in      pointer to the vector of transform coefficients
 217  *  @param[out]   out     pointer to the output buffer (frame)
 218  *  @param[in]    pitch   pitch to move to the next y line
 219  *  @param[in]    flags   pointer to the array of column flags (unused here)
 220  */
 221 void ff_ivi_row_slant4(const int32_t *in, int16_t *out, uint32_t pitch,
 222                        const uint8_t *flags);
 223
 224 /**
 225  *  inverse 1D column slant transform
 226  *
 227  *  @param[in]    in      pointer to the vector of transform coefficients
 228  *  @param[out]   out     pointer to the output buffer (frame)
 229  *  @param[in]    pitch   pitch to move to the next y line
 230  *  @param[in]    flags   pointer to the array of column flags:
 231  *                        != 0 - non_empty column, 0 - empty one
 232  *                        (this array must be filled by caller)
 233  */
 234 void ff_ivi_col_slant4(const int32_t *in, int16_t *out, uint32_t pitch,
 235                        const uint8_t *flags);
 236
 237 /**
 238  *  DC-only inverse row slant transform
 239  */
 240 void ff_ivi_dc_row_slant(const int32_t *in, int16_t *out, uint32_t pitch, int blk_size);
 241
 242 /**
 243  *  DC-only inverse column slant transform
 244  */
 245 void ff_ivi_dc_col_slant(const int32_t *in, int16_t *out, uint32_t pitch, int blk_size);
 246
 247 /**
 248  *  Copy the pixels into the frame buffer.
 249  */
 250 void ff_ivi_put_pixels_8x8(const int32_t *in, int16_t *out, uint32_t pitch, const uint8_t *flags);
 251
 252 /**
 253  *  Copy the DC coefficient into the first pixel of the block and
 254  *  zero all others.
 255  */
 256 void ff_ivi_put_dc_pixel_8x8(const int32_t *in, int16_t *out, uint32_t pitch, int blk_size);
 257
 258 /**
 259  *  8x8 block motion compensation with adding delta
 260  *
 261  *  @param[in,out]   buf      pointer to the block in the current frame buffer containing delta
 262  *  @param[in]       ref_buf  pointer to the corresponding block in the reference frame
 263  *  @param[in]       pitch    pitch for moving to the next y line
 264  *  @param[in]       mc_type  interpolation type
 265  */
 266 void ff_ivi_mc_8x8_delta(int16_t *buf, const int16_t *ref_buf, uint32_t pitch, int mc_type);
 267
 268 /**
 269  *  4x4 block motion compensation with adding delta
 270  *
 271  *  @param[in,out]   buf      pointer to the block in the current frame buffer containing delta
 272  *  @param[in]       ref_buf  pointer to the corresponding block in the reference frame
 273  *  @param[in]       pitch    pitch for moving to the next y line
 274  *  @param[in]       mc_type  interpolation type
 275  */
 276 void ff_ivi_mc_4x4_delta(int16_t *buf, const int16_t *ref_buf, uint32_t pitch, int mc_type);
 277
 278 /**
 279  *  motion compensation without adding delta
 280  *
 281  *  @param[in,out]  buf      pointer to the block in the current frame receiving the result
 282  *  @param[in]      ref_buf  pointer to the corresponding block in the reference frame
 283  *  @param[in]      pitch    pitch for moving to the next y line
 284  *  @param[in]      mc_type  interpolation type
 285  */
 286 void ff_ivi_mc_8x8_no_delta(int16_t *buf, const int16_t *ref_buf, uint32_t pitch, int mc_type);
 287
 288 /**
 289  *  4x4 block motion compensation without adding delta
 290  *
 291  *  @param[in,out]  buf      pointer to the block in the current frame receiving the result
 292  *  @param[in]      ref_buf  pointer to the corresponding block in the reference frame
 293  *  @param[in]      pitch    pitch for moving to the next y line
 294  *  @param[in]      mc_type  interpolation type
 295  */
 296 void ff_ivi_mc_4x4_no_delta(int16_t *buf, const int16_t *ref_buf, uint32_t pitch, int mc_type);
 297
 298 #endif /* AVCODEC_IVI_DSP_H */