mirror of https://github.com/FFmpeg/FFmpeg.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
245 lines
7.3 KiB
245 lines
7.3 KiB
/* |
|
* Copyright (C) 2004-2010 Michael Niedermayer <michaelni@gmx.at> |
|
* |
|
* 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 |
|
*/ |
|
|
|
#ifndef AVCODEC_DWT_H |
|
#define AVCODEC_DWT_H |
|
|
|
#include <stdint.h> |
|
|
|
typedef int DWTELEM; |
|
typedef short IDWTELEM; |
|
|
|
#define MAX_DWT_SUPPORT 8 |
|
#define MAX_DECOMPOSITIONS 8 |
|
|
|
typedef struct { |
|
IDWTELEM *b[MAX_DWT_SUPPORT]; |
|
|
|
IDWTELEM *b0; |
|
IDWTELEM *b1; |
|
IDWTELEM *b2; |
|
IDWTELEM *b3; |
|
int y; |
|
} DWTCompose; |
|
|
|
/** Used to minimize the amount of memory used in order to optimize cache performance. **/ |
|
typedef struct slice_buffer_s { |
|
IDWTELEM * * line; ///< For use by idwt and predict_slices. |
|
IDWTELEM * * data_stack; ///< Used for internal purposes. |
|
int data_stack_top; |
|
int line_count; |
|
int line_width; |
|
int data_count; |
|
IDWTELEM * base_buffer; ///< Buffer that this structure is caching. |
|
} slice_buffer; |
|
|
|
struct DWTContext; |
|
|
|
// Possible prototypes for vertical_compose functions |
|
typedef void (*vertical_compose_2tap)(IDWTELEM *b0, IDWTELEM *b1, int width); |
|
typedef void (*vertical_compose_3tap)(IDWTELEM *b0, IDWTELEM *b1, IDWTELEM *b2, int width); |
|
typedef void (*vertical_compose_5tap)(IDWTELEM *b0, IDWTELEM *b1, IDWTELEM *b2, IDWTELEM *b3, IDWTELEM *b4, int width); |
|
typedef void (*vertical_compose_9tap)(IDWTELEM *dst, IDWTELEM *b[8], int width); |
|
|
|
typedef struct DWTContext { |
|
IDWTELEM *buffer; |
|
IDWTELEM *temp; |
|
int width; |
|
int height; |
|
int stride; |
|
int decomposition_count; |
|
int support; |
|
|
|
void (*spatial_compose)(struct DWTContext *cs, int level, int width, int height, int stride); |
|
void (*vertical_compose_l0)(); |
|
void (*vertical_compose_h0)(); |
|
void (*vertical_compose_l1)(); |
|
void (*vertical_compose_h1)(); |
|
void (*vertical_compose)(); ///< one set of lowpass and highpass combined |
|
void (*horizontal_compose)(IDWTELEM *b, IDWTELEM *tmp, int width); |
|
|
|
void (*vertical_compose97i)(IDWTELEM *b0, IDWTELEM *b1, IDWTELEM *b2, IDWTELEM *b3, IDWTELEM *b4, IDWTELEM *b5, int width); |
|
void (*horizontal_compose97i)(IDWTELEM *b, int width); |
|
void (*inner_add_yblock)(const uint8_t *obmc, const int obmc_stride, uint8_t * * block, int b_w, int b_h, int src_x, int src_y, int src_stride, slice_buffer * sb, int add, uint8_t * dst8); |
|
|
|
DWTCompose cs[MAX_DECOMPOSITIONS]; |
|
} DWTContext; |
|
|
|
enum dwt_type { |
|
DWT_SNOW_DAUB9_7, |
|
DWT_SNOW_LEGALL5_3, |
|
DWT_DIRAC_DD9_7, |
|
DWT_DIRAC_LEGALL5_3, |
|
DWT_DIRAC_DD13_7, |
|
DWT_DIRAC_HAAR0, |
|
DWT_DIRAC_HAAR1, |
|
DWT_DIRAC_FIDELITY, |
|
DWT_DIRAC_DAUB9_7, |
|
DWT_NUM_TYPES |
|
}; |
|
|
|
// -1 if an error occurred, e.g. the dwt_type isn't recognized |
|
int ff_spatial_idwt_init2(DWTContext *d, IDWTELEM *buffer, int width, int height, |
|
int stride, enum dwt_type type, int decomposition_count, |
|
IDWTELEM *temp); |
|
|
|
int ff_spatial_idwt2(IDWTELEM *buffer, int width, int height, int stride, |
|
enum dwt_type type, int decomposition_count, IDWTELEM *temp); |
|
|
|
void ff_spatial_idwt_slice2(DWTContext *d, int y); |
|
|
|
// shared stuff for simd optimiztions |
|
#define COMPOSE_53iL0(b0, b1, b2)\ |
|
(b1 - ((b0 + b2 + 2) >> 2)) |
|
|
|
#define COMPOSE_DIRAC53iH0(b0, b1, b2)\ |
|
(b1 + ((b0 + b2 + 1) >> 1)) |
|
|
|
#define COMPOSE_DD97iH0(b0, b1, b2, b3, b4)\ |
|
(b2 + ((-b0 + 9*b1 + 9*b3 - b4 + 8) >> 4)) |
|
|
|
#define COMPOSE_DD137iL0(b0, b1, b2, b3, b4)\ |
|
(b2 - ((-b0 + 9*b1 + 9*b3 - b4 + 16) >> 5)) |
|
|
|
#define COMPOSE_HAARiL0(b0, b1)\ |
|
(b0 - ((b1 + 1) >> 1)) |
|
|
|
#define COMPOSE_HAARiH0(b0, b1)\ |
|
(b0 + b1) |
|
|
|
#define COMPOSE_FIDELITYiL0(b0, b1, b2, b3, b4, b5, b6, b7, b8)\ |
|
(b4 - ((-8*(b0+b8) + 21*(b1+b7) - 46*(b2+b6) + 161*(b3+b5) + 128) >> 8)) |
|
|
|
#define COMPOSE_FIDELITYiH0(b0, b1, b2, b3, b4, b5, b6, b7, b8)\ |
|
(b4 + ((-2*(b0+b8) + 10*(b1+b7) - 25*(b2+b6) + 81*(b3+b5) + 128) >> 8)) |
|
|
|
#define COMPOSE_DAUB97iL1(b0, b1, b2)\ |
|
(b1 - ((1817*(b0 + b2) + 2048) >> 12)) |
|
|
|
#define COMPOSE_DAUB97iH1(b0, b1, b2)\ |
|
(b1 - (( 113*(b0 + b2) + 64) >> 7)) |
|
|
|
#define COMPOSE_DAUB97iL0(b0, b1, b2)\ |
|
(b1 + (( 217*(b0 + b2) + 2048) >> 12)) |
|
|
|
#define COMPOSE_DAUB97iH0(b0, b1, b2)\ |
|
(b1 + ((6497*(b0 + b2) + 2048) >> 12)) |
|
|
|
|
|
|
|
#define DWT_97 0 |
|
#define DWT_53 1 |
|
|
|
#define liftS lift |
|
#if 1 |
|
#define W_AM 3 |
|
#define W_AO 0 |
|
#define W_AS 1 |
|
|
|
#undef liftS |
|
#define W_BM 1 |
|
#define W_BO 8 |
|
#define W_BS 4 |
|
|
|
#define W_CM 1 |
|
#define W_CO 0 |
|
#define W_CS 0 |
|
|
|
#define W_DM 3 |
|
#define W_DO 4 |
|
#define W_DS 3 |
|
#elif 0 |
|
#define W_AM 55 |
|
#define W_AO 16 |
|
#define W_AS 5 |
|
|
|
#define W_BM 3 |
|
#define W_BO 32 |
|
#define W_BS 6 |
|
|
|
#define W_CM 127 |
|
#define W_CO 64 |
|
#define W_CS 7 |
|
|
|
#define W_DM 7 |
|
#define W_DO 8 |
|
#define W_DS 4 |
|
#elif 0 |
|
#define W_AM 97 |
|
#define W_AO 32 |
|
#define W_AS 6 |
|
|
|
#define W_BM 63 |
|
#define W_BO 512 |
|
#define W_BS 10 |
|
|
|
#define W_CM 13 |
|
#define W_CO 8 |
|
#define W_CS 4 |
|
|
|
#define W_DM 15 |
|
#define W_DO 16 |
|
#define W_DS 5 |
|
|
|
#else |
|
|
|
#define W_AM 203 |
|
#define W_AO 64 |
|
#define W_AS 7 |
|
|
|
#define W_BM 217 |
|
#define W_BO 2048 |
|
#define W_BS 12 |
|
|
|
#define W_CM 113 |
|
#define W_CO 64 |
|
#define W_CS 7 |
|
|
|
#define W_DM 227 |
|
#define W_DO 128 |
|
#define W_DS 9 |
|
#endif |
|
|
|
#define slice_buffer_get_line(slice_buf, line_num) ((slice_buf)->line[line_num] ? (slice_buf)->line[line_num] : ff_slice_buffer_load_line((slice_buf), (line_num))) |
|
//#define slice_buffer_get_line(slice_buf, line_num) (ff_slice_buffer_load_line((slice_buf), (line_num))) |
|
|
|
void ff_slice_buffer_init(slice_buffer * buf, int line_count, int max_allocated_lines, int line_width, IDWTELEM * base_buffer); |
|
void ff_slice_buffer_release(slice_buffer * buf, int line); |
|
void ff_slice_buffer_flush(slice_buffer * buf); |
|
void ff_slice_buffer_destroy(slice_buffer * buf); |
|
IDWTELEM * ff_slice_buffer_load_line(slice_buffer * buf, int line); |
|
|
|
void ff_snow_vertical_compose97i(IDWTELEM *b0, IDWTELEM *b1, IDWTELEM *b2, IDWTELEM *b3, IDWTELEM *b4, IDWTELEM *b5, int width); |
|
void ff_snow_horizontal_compose97i(IDWTELEM *b, int width); |
|
void ff_snow_inner_add_yblock(const uint8_t *obmc, const int obmc_stride, uint8_t * * block, int b_w, int b_h, int src_x, int src_y, int src_stride, slice_buffer * sb, int add, uint8_t * dst8); |
|
|
|
int ff_w53_32_c(void *v, uint8_t * pix1, uint8_t * pix2, int line_size, int h); |
|
int ff_w97_32_c(void *v, uint8_t * pix1, uint8_t * pix2, int line_size, int h); |
|
|
|
void ff_spatial_dwt(int *buffer, int width, int height, int stride, int type, int decomposition_count); |
|
|
|
void ff_spatial_idwt_buffered_init(DWTCompose *cs, slice_buffer * sb, int width, int height, int stride_line, int type, int decomposition_count); |
|
void ff_spatial_idwt_buffered_slice(DWTContext *dsp, DWTCompose *cs, slice_buffer * slice_buf, int width, int height, int stride_line, int type, int decomposition_count, int y); |
|
void ff_spatial_idwt(IDWTELEM *buffer, int width, int height, int stride, int type, int decomposition_count); |
|
|
|
void ff_dwt_init(DWTContext *c); |
|
void ff_dwt_init_x86(DWTContext *c); |
|
|
|
#endif /* AVCODEC_DWT_H */
|
|
|