1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
/* Copyright (c) 2006-2012 Filip Wasilewski <http://en.ig.ma/>
* Copyright (c) 2012-2016 The PyWavelets Developers
* <https://github.com/PyWavelets/pywt>
* See COPYING for license details.
*/
/* Common constants, typedefs and functions */
#pragma once
#include <stdlib.h>
#include <memory.h>
#ifdef HAVE_C99_COMPLEX
/* For templating, we need typedefs without spaces for complex types. */
typedef float _Complex float_complex;
typedef double _Complex double_complex;
#endif
/* ##### Typedefs ##### */
#ifdef PY_EXTENSION
/* another declaration is in .c file generated by Pyrex */
#ifndef PY_SSIZE_T_CLEAN
#define PY_SSIZE_T_CLEAN
#include "Python.h"
#include "structmember.h"
#ifndef PY_LONG_LONG
#define PY_LONG_LONG LONG_LONG
#endif
#else
#include "Python.h"
#endif
/* on Solaris/SmartOS system, index_t is used in sys/types.h, so use pytw_index_t */
typedef Py_ssize_t pywt_index_t;
#else
typedef int pywt_index_t;
#endif
/* standard c memory management */
#define wtmalloc(size) malloc(size)
#define wtfree(ptr) free(ptr)
#define wtcalloc(len, size) calloc(len, size)
#ifdef _MSC_VER
#include <intrin.h>
#endif
typedef struct {
size_t * shape;
pywt_index_t * strides;
size_t ndim;
} ArrayInfo;
typedef enum {
COEF_APPROX = 0,
COEF_DETAIL = 1,
} Coefficient;
typedef enum {
DWT_TRANSFORM = 0,
SWT_TRANSFORM = 1,
} DiscreteTransformType;
/* Signal extension modes */
typedef enum {
MODE_INVALID = -1,
MODE_ZEROPAD = 0, /* default, signal extended with zeros */
MODE_SYMMETRIC, /* signal extended symmetrically (mirror)
* also known as half-sample symmetric
* For extensions greater than signal length,
* mirror back and forth:
* 2 3 3 2 1 | 1 2 3 | 3 2 1 1 2
*/
MODE_CONSTANT_EDGE, /* signal extended with the border value */
MODE_SMOOTH, /* linear extrapolation (first derivative) */
MODE_PERIODIC, /* signal is treated as being periodic */
MODE_PERIODIZATION, /* signal is treated as being periodic, minimal output length */
MODE_REFLECT, /* signal extended symmetrically (reflect)
* also known as whole-sample symmetric
* For extensions greater than signal length,
* reflect back and forth without repeating edge values:
* 1 2 3 2 | 1 2 3 | 2 1 2 3
*/
MODE_ANTISYMMETRIC, /* antisymmetric version of "MODE_SYMMETRIC"
* also known as half-sample antisymmetric
* 2 3 -3 -2 -1 | 1 2 3 | -3 -2 -1 1 2
*/
MODE_ANTIREFLECT, /* antisymmetric version of "MODE_REFLECT"
* also known as whole-sample antisymmetric
* 0 -1 -2 -1 0 | 1 2 3 | 4 5 6 5 4
*/
MODE_MAX,
} MODE;
/* ##### Calculating buffer lengths for various operations ##### */
/*
* Length of DWT coeffs for specified input data length, filter length and
* signal extension mode.
*/
size_t dwt_buffer_length(size_t input_len, size_t filter_len, MODE mode);
/*
* Length of reconstructed signal for specified input coeffs length and filter
* length. It is used for direct reconstruction from coefficients (normal
* convolution of upsampled coeffs with filter).
*/
size_t reconstruction_buffer_length(size_t coeffs_len, size_t filter_len);
/*
* Length of IDWT reconstructed signal for specified input coeffs length, filter
* length and extension mode.
*/
size_t idwt_buffer_length(size_t coeffs_len, size_t filter_len, MODE mode);
/* Length of SWT coefficients for specified input signal length (== input_len) */
size_t swt_buffer_length(size_t input_len);
/* Maximum useful level of DWT decomposition. */
unsigned char dwt_max_level(size_t input_len, size_t filter_len);
/* Maximum useful level of SWT decomposition. */
unsigned char swt_max_level(size_t input_len);
|
