ivi_dsp.h 14.9 KB
Newer Older
1 2 3
/*
 * DSP functions for Indeo Video Interactive codecs (Indeo4 and Indeo5)
 *
Kostya Shishkov's avatar
Kostya Shishkov committed
4
 * Copyright (c) 2009-2011 Maxim Poliakovski
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
 *
 * 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
 */

/**
24
 * @file
25
 * DSP functions (inverse transforms, motion compensations, wavelet recomposition)
26 27 28 29 30 31
 * for Indeo Video Interactive codecs.
 */

#ifndef AVCODEC_IVI_DSP_H
#define AVCODEC_IVI_DSP_H

32
#include <stddef.h>
33 34 35
#include <stdint.h>

#include "ivi.h"
36 37 38 39

/**
 *  5/3 wavelet recomposition filter for Indeo5
 *
40 41 42
 *  @param[in]   plane        pointer to the descriptor of the plane being processed
 *  @param[out]  dst          pointer to the destination buffer
 *  @param[in]   dst_pitch    pitch of the destination buffer
43 44
 */
void ff_ivi_recompose53(const IVIPlaneDesc *plane, uint8_t *dst,
45
                        const ptrdiff_t dst_pitch);
46

Kostya Shishkov's avatar
Kostya Shishkov committed
47 48 49 50 51 52 53 54
/**
 *  Haar wavelet recomposition filter for Indeo 4
 *
 *  @param[in]  plane        pointer to the descriptor of the plane being processed
 *  @param[out] dst          pointer to the destination buffer
 *  @param[in]  dst_pitch    pitch of the destination buffer
 */
void ff_ivi_recompose_haar(const IVIPlaneDesc *plane, uint8_t *dst,
55
                           const ptrdiff_t dst_pitch);
Kostya Shishkov's avatar
Kostya Shishkov committed
56 57 58 59 60 61 62 63 64 65 66

/**
 *  two-dimensional inverse Haar 8x8 transform for Indeo 4
 *
 *  @param[in]  in        pointer to the vector of transform coefficients
 *  @param[out] out       pointer to the output buffer (frame)
 *  @param[in]  pitch     pitch to move to the next y line
 *  @param[in]  flags     pointer to the array of column flags:
 *                        != 0 - non_empty column, 0 - empty one
 *                        (this array must be filled by caller)
 */
67
void ff_ivi_inverse_haar_8x8(const int32_t *in, int16_t *out, ptrdiff_t pitch,
Kostya Shishkov's avatar
Kostya Shishkov committed
68
                             const uint8_t *flags);
69 70 71 72
void ff_ivi_inverse_haar_8x1(const int32_t *in, int16_t *out, uint32_t pitch,
                             const uint8_t *flags);
void ff_ivi_inverse_haar_1x8(const int32_t *in, int16_t *out, uint32_t pitch,
                             const uint8_t *flags);
Kostya Shishkov's avatar
Kostya Shishkov committed
73

74 75 76 77 78 79 80 81 82 83
/**
 *  one-dimensional inverse 8-point Haar transform on rows for Indeo 4
 *
 *  @param[in]  in        pointer to the vector of transform coefficients
 *  @param[out] out       pointer to the output buffer (frame)
 *  @param[in]  pitch     pitch to move to the next y line
 *  @param[in]  flags     pointer to the array of column flags:
 *                        != 0 - non_empty column, 0 - empty one
 *                        (this array must be filled by caller)
 */
84
void ff_ivi_row_haar8(const int32_t *in, int16_t *out, ptrdiff_t pitch,
85 86 87 88 89 90 91 92 93 94 95 96
                      const uint8_t *flags);

/**
 *  one-dimensional inverse 8-point Haar transform on columns for Indeo 4
 *
 *  @param[in]  in        pointer to the vector of transform coefficients
 *  @param[out] out       pointer to the output buffer (frame)
 *  @param[in]  pitch     pitch to move to the next y line
 *  @param[in]  flags     pointer to the array of column flags:
 *                        != 0 - non_empty column, 0 - empty one
 *                        (this array must be filled by caller)
 */
97
void ff_ivi_col_haar8(const int32_t *in, int16_t *out, ptrdiff_t pitch,
98 99 100 101 102 103 104 105 106 107 108 109
                      const uint8_t *flags);

/**
 *  two-dimensional inverse Haar 4x4 transform for Indeo 4
 *
 *  @param[in]  in        pointer to the vector of transform coefficients
 *  @param[out] out       pointer to the output buffer (frame)
 *  @param[in]  pitch     pitch to move to the next y line
 *  @param[in]  flags     pointer to the array of column flags:
 *                        != 0 - non_empty column, 0 - empty one
 *                        (this array must be filled by caller)
 */
110
void ff_ivi_inverse_haar_4x4(const int32_t *in, int16_t *out, ptrdiff_t pitch,
111 112 113 114 115 116 117 118 119 120 121 122
                             const uint8_t *flags);

/**
 *  one-dimensional inverse 4-point Haar transform on rows for Indeo 4
 *
 *  @param[in]  in        pointer to the vector of transform coefficients
 *  @param[out] out       pointer to the output buffer (frame)
 *  @param[in]  pitch     pitch to move to the next y line
 *  @param[in]  flags     pointer to the array of column flags:
 *                        != 0 - non_empty column, 0 - empty one
 *                        (this array must be filled by caller)
 */
123
void ff_ivi_row_haar4(const int32_t *in, int16_t *out, ptrdiff_t pitch,
124 125 126 127 128 129 130 131 132 133 134 135
                      const uint8_t *flags);

/**
 *  one-dimensional inverse 4-point Haar transform on columns for Indeo 4
 *
 *  @param[in]  in        pointer to the vector of transform coefficients
 *  @param[out] out       pointer to the output buffer (frame)
 *  @param[in]  pitch     pitch to move to the next y line
 *  @param[in]  flags     pointer to the array of column flags:
 *                        != 0 - non_empty column, 0 - empty one
 *                        (this array must be filled by caller)
 */
136
void ff_ivi_col_haar4(const int32_t *in, int16_t *out, ptrdiff_t pitch,
137 138
                      const uint8_t *flags);

Kostya Shishkov's avatar
Kostya Shishkov committed
139 140 141 142 143 144 145 146 147 148
/**
 *  DC-only two-dimensional inverse Haar transform for Indeo 4.
 *  Performing the inverse transform in this case is equivalent to
 *  spreading DC_coeff >> 3 over the whole block.
 *
 *  @param[in]  in          pointer to the dc coefficient
 *  @param[out] out         pointer to the output buffer (frame)
 *  @param[in]  pitch       pitch to move to the next y line
 *  @param[in]  blk_size    transform block size
 */
149
void ff_ivi_dc_haar_2d(const int32_t *in, int16_t *out, ptrdiff_t pitch,
Kostya Shishkov's avatar
Kostya Shishkov committed
150 151
                       int blk_size);

152 153 154
/**
 *  two-dimensional inverse slant 8x8 transform
 *
155 156 157 158
 *  @param[in]    in      pointer to the vector of transform coefficients
 *  @param[out]   out     pointer to the output buffer (frame)
 *  @param[in]    pitch   pitch to move to the next y line
 *  @param[in]    flags   pointer to the array of column flags:
159 160 161
 *                        != 0 - non_empty column, 0 - empty one
 *                        (this array must be filled by caller)
 */
162
void ff_ivi_inverse_slant_8x8(const int32_t *in, int16_t *out, ptrdiff_t pitch,
163 164 165 166 167
                              const uint8_t *flags);

/**
 *  two-dimensional inverse slant 4x4 transform
 *
168 169 170 171
 *  @param[in]    in      pointer to the vector of transform coefficients
 *  @param[out]   out     pointer to the output buffer (frame)
 *  @param[in]    pitch   pitch to move to the next y line
 *  @param[in]    flags   pointer to the array of column flags:
172 173 174
 *                        != 0 - non_empty column, 0 - empty one
 *                        (this array must be filled by caller)
 */
175
void ff_ivi_inverse_slant_4x4(const int32_t *in, int16_t *out, ptrdiff_t pitch,
176 177 178 179 180 181 182 183
                              const uint8_t *flags);

/**
 *  DC-only two-dimensional inverse slant transform.
 *  Performing the inverse slant transform in this case is equivalent to
 *  spreading (DC_coeff + 1)/2 over the whole block.
 *  It works much faster than performing the slant transform on a vector of zeroes.
 *
184 185 186 187
 *  @param[in]    in          pointer to the dc coefficient
 *  @param[out]   out         pointer to the output buffer (frame)
 *  @param[in]    pitch       pitch to move to the next y line
 *  @param[in]    blk_size    transform block size
188
 */
189
void ff_ivi_dc_slant_2d(const int32_t *in, int16_t *out, ptrdiff_t pitch, int blk_size);
190 191 192 193

/**
 *  inverse 1D row slant transform
 *
194 195 196 197
 *  @param[in]    in      pointer to the vector of transform coefficients
 *  @param[out]   out     pointer to the output buffer (frame)
 *  @param[in]    pitch   pitch to move to the next y line
 *  @param[in]    flags   pointer to the array of column flags (unused here)
198
 */
199
void ff_ivi_row_slant8(const int32_t *in, int16_t *out, ptrdiff_t pitch,
200 201 202 203 204
                       const uint8_t *flags);

/**
 *  inverse 1D column slant transform
 *
205 206 207 208
 *  @param[in]    in      pointer to the vector of transform coefficients
 *  @param[out]   out     pointer to the output buffer (frame)
 *  @param[in]    pitch   pitch to move to the next y line
 *  @param[in]    flags   pointer to the array of column flags:
209 210 211
 *                        != 0 - non_empty column, 0 - empty one
 *                        (this array must be filled by caller)
 */
212
void ff_ivi_col_slant8(const int32_t *in, int16_t *out, ptrdiff_t pitch,
213 214
                       const uint8_t *flags);

215 216 217 218 219 220 221 222
/**
 *  inverse 1D row slant transform
 *
 *  @param[in]    in      pointer to the vector of transform coefficients
 *  @param[out]   out     pointer to the output buffer (frame)
 *  @param[in]    pitch   pitch to move to the next y line
 *  @param[in]    flags   pointer to the array of column flags (unused here)
 */
223
void ff_ivi_row_slant4(const int32_t *in, int16_t *out, ptrdiff_t pitch,
224 225 226 227 228 229 230 231 232 233 234 235
                       const uint8_t *flags);

/**
 *  inverse 1D column slant transform
 *
 *  @param[in]    in      pointer to the vector of transform coefficients
 *  @param[out]   out     pointer to the output buffer (frame)
 *  @param[in]    pitch   pitch to move to the next y line
 *  @param[in]    flags   pointer to the array of column flags:
 *                        != 0 - non_empty column, 0 - empty one
 *                        (this array must be filled by caller)
 */
236
void ff_ivi_col_slant4(const int32_t *in, int16_t *out, ptrdiff_t pitch,
237 238
                       const uint8_t *flags);

239 240 241
/**
 *  DC-only inverse row slant transform
 */
242
void ff_ivi_dc_row_slant(const int32_t *in, int16_t *out, ptrdiff_t pitch, int blk_size);
243 244 245 246

/**
 *  DC-only inverse column slant transform
 */
247
void ff_ivi_dc_col_slant(const int32_t *in, int16_t *out, ptrdiff_t pitch, int blk_size);
248 249

/**
250
 *  Copy the pixels into the frame buffer.
251
 */
252
void ff_ivi_put_pixels_8x8(const int32_t *in, int16_t *out, ptrdiff_t pitch, const uint8_t *flags);
253 254

/**
255 256
 *  Copy the DC coefficient into the first pixel of the block and
 *  zero all others.
257
 */
258
void ff_ivi_put_dc_pixel_8x8(const int32_t *in, int16_t *out, ptrdiff_t pitch, int blk_size);
259 260 261 262

/**
 *  8x8 block motion compensation with adding delta
 *
263 264 265 266
 *  @param[in,out]   buf      pointer to the block in the current frame buffer containing delta
 *  @param[in]       ref_buf  pointer to the corresponding block in the reference frame
 *  @param[in]       pitch    pitch for moving to the next y line
 *  @param[in]       mc_type  interpolation type
267
 */
268
void ff_ivi_mc_8x8_delta(int16_t *buf, const int16_t *ref_buf, ptrdiff_t pitch, int mc_type);
269 270 271 272

/**
 *  4x4 block motion compensation with adding delta
 *
273 274 275 276
 *  @param[in,out]   buf      pointer to the block in the current frame buffer containing delta
 *  @param[in]       ref_buf  pointer to the corresponding block in the reference frame
 *  @param[in]       pitch    pitch for moving to the next y line
 *  @param[in]       mc_type  interpolation type
277
 */
278
void ff_ivi_mc_4x4_delta(int16_t *buf, const int16_t *ref_buf, ptrdiff_t pitch, int mc_type);
279 280 281 282

/**
 *  motion compensation without adding delta
 *
283 284 285 286
 *  @param[in,out]  buf      pointer to the block in the current frame receiving the result
 *  @param[in]      ref_buf  pointer to the corresponding block in the reference frame
 *  @param[in]      pitch    pitch for moving to the next y line
 *  @param[in]      mc_type  interpolation type
287
 */
288
void ff_ivi_mc_8x8_no_delta(int16_t *buf, const int16_t *ref_buf, ptrdiff_t pitch, int mc_type);
289 290 291 292

/**
 *  4x4 block motion compensation without adding delta
 *
293 294 295 296
 *  @param[in,out]  buf      pointer to the block in the current frame receiving the result
 *  @param[in]      ref_buf  pointer to the corresponding block in the reference frame
 *  @param[in]      pitch    pitch for moving to the next y line
 *  @param[in]      mc_type  interpolation type
297
 */
298
void ff_ivi_mc_4x4_no_delta(int16_t *buf, const int16_t *ref_buf, ptrdiff_t pitch, int mc_type);
299

300 301 302 303 304 305 306 307 308 309
/**
 *  8x8 block motion compensation with adding delta
 *
 *  @param[in,out]  buf      pointer to the block in the current frame buffer containing delta
 *  @param[in]      ref_buf  pointer to the corresponding block in the backward reference frame
 *  @param[in]      ref_buf2 pointer to the corresponding block in the forward reference frame
 *  @param[in]      pitch    pitch for moving to the next y line
 *  @param[in]      mc_type  interpolation type for backward reference
 *  @param[in]      mc_type2 interpolation type for forward reference
 */
310
void ff_ivi_mc_avg_8x8_delta(int16_t *buf, const int16_t *ref_buf, const int16_t *ref_buf2, ptrdiff_t pitch, int mc_type, int mc_type2);
311 312 313 314 315 316 317 318 319 320 321

/**
 *  4x4 block motion compensation with adding delta
 *
 *  @param[in,out]  buf      pointer to the block in the current frame buffer containing delta
 *  @param[in]      ref_buf  pointer to the corresponding block in the backward reference frame
 *  @param[in]      ref_buf2 pointer to the corresponding block in the forward reference frame
 *  @param[in]      pitch    pitch for moving to the next y line
 *  @param[in]      mc_type  interpolation type for backward reference
 *  @param[in]      mc_type2 interpolation type for forward reference
 */
322
void ff_ivi_mc_avg_4x4_delta(int16_t *buf, const int16_t *ref_buf, const int16_t *ref_buf2, ptrdiff_t pitch, int mc_type, int mc_type2);
323 324 325 326 327 328 329 330 331 332 333

/**
 *  motion compensation without adding delta for B-frames
 *
 *  @param[in,out]  buf      pointer to the block in the current frame receiving the result
 *  @param[in]      ref_buf  pointer to the corresponding block in the backward reference frame
 *  @param[in]      ref_buf2 pointer to the corresponding block in the forward reference frame
 *  @param[in]      pitch    pitch for moving to the next y line
 *  @param[in]      mc_type  interpolation type for backward reference
 *  @param[in]      mc_type2 interpolation type for forward reference
 */
334
void ff_ivi_mc_avg_8x8_no_delta(int16_t *buf, const int16_t *ref_buf, const int16_t *ref_buf2, ptrdiff_t pitch, int mc_type, int mc_type2);
335 336 337 338 339 340 341 342 343 344 345

/**
 *  4x4 block motion compensation without adding delta for B-frames
 *
 *  @param[in,out]  buf      pointer to the block in the current frame receiving the result
 *  @param[in]      ref_buf  pointer to the corresponding block in the backward reference frame
 *  @param[in]      ref_buf2 pointer to the corresponding block in the forward reference frame
 *  @param[in]      pitch    pitch for moving to the next y line
 *  @param[in]      mc_type  interpolation type for backward reference
 *  @param[in]      mc_type2 interpolation type for forward reference
 */
346
void ff_ivi_mc_avg_4x4_no_delta(int16_t *buf, const int16_t *ref_buf, const int16_t *ref_buf2, ptrdiff_t pitch, int mc_type, int mc_type2);
347

348
#endif /* AVCODEC_IVI_DSP_H */