-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathHalfBand16Avx512.h
156 lines (90 loc) · 4.33 KB
/
HalfBand16Avx512.h
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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
/*****************************************************************************
HalfBand16Avx512.h
Author: Laurent de Soras, 2021
Half-band filter, low-pass or high-pass, using AVX512 instruction set.
Works on vectors of 16 float.
This object must be aligned on a 64-byte boundary!
Template parameters:
- NC: number of coefficients, > 0
--- Legal stuff ---
This program is free software. It comes without any warranty, to
the extent permitted by applicable law. You can redistribute it
and/or modify it under the terms of the Do What The Fuck You Want
To Public License, Version 2, as published by Sam Hocevar. See
http://www.wtfpl.net/ for more details.
*Tab=3***********************************************************************/
#pragma once
#if ! defined (hiir_HalfBand16Avx512_HEADER_INCLUDED)
#define hiir_HalfBand16Avx512_HEADER_INCLUDED
/*\\\ INCLUDE FILES \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\*/
#include "hiir/def.h"
#include "hiir/StageDataAvx512.h"
#include <immintrin.h>
#include <array>
namespace hiir
{
template <int NC>
class HalfBand16Avx512
{
static_assert ((NC > 0), "Number of coefficient must be positive.");
/*\\\ PUBLIC \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\*/
public:
typedef float DataType;
static constexpr int _nbr_chn = 16;
static constexpr int NBR_COEFS = NC;
static constexpr double _delay = 0;
HalfBand16Avx512 () noexcept;
HalfBand16Avx512 (const HalfBand16Avx512 <NC> &other) = default;
HalfBand16Avx512 (HalfBand16Avx512 <NC> &&other) = default;
~HalfBand16Avx512 () = default;
HalfBand16Avx512 <NC> &
operator = (const HalfBand16Avx512 <NC> &other) = default;
HalfBand16Avx512 <NC> &
operator = (HalfBand16Avx512 <NC> &&other) = default;
void set_coefs (const double coef_arr []) noexcept;
hiir_FORCEINLINE __m512
process_sample (__m512 input) noexcept;
void process_block (float out_ptr [], const float in_ptr [], long nbr_spl) noexcept;
hiir_FORCEINLINE __m512
process_sample_hpf (__m512 input) noexcept;
void process_block_hpf (float out_ptr [], const float in_ptr [], long nbr_spl) noexcept;
hiir_FORCEINLINE void
process_sample_split (__m512 &low, __m512 &high, __m512 input) noexcept;
void process_block_split (float out_l_ptr [], float out_h_ptr [], const float in_ptr [], long nbr_spl) noexcept;
void clear_buffers () noexcept;
/*\\\ PROTECTED \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\*/
protected:
/*\\\ PRIVATE \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\*/
private:
static constexpr int _nbr_phases = 2;
typedef std::array <StageDataAvx512, NBR_COEFS + 2> Filter; // Stages 0 and 1 contain only input memories
typedef std::array <Filter, _nbr_phases> FilterBiPhase;
struct EvenOdd
{
alignas (64) __m512 _e;
alignas (64) __m512 _o;
};
hiir_FORCEINLINE EvenOdd
process_2_paths (__m512 input) noexcept;
template <typename FL, typename FH>
hiir_FORCEINLINE void
process_block_2_paths (float out_l_ptr [], float out_h_ptr [], const float in_ptr [], long nbr_spl, FL fnc_l, FH fnc_h) noexcept;
hiir_FORCEINLINE static void
store_low (float *ptr, __m512 tmp_0, __m512 tmp_1) noexcept;
hiir_FORCEINLINE static void
store_high (float *ptr, __m512 tmp_0, __m512 tmp_1) noexcept;
hiir_FORCEINLINE static void
bypass (float *, __m512, __m512) noexcept {}
FilterBiPhase _bifilter;
alignas (64) float
_prev [_nbr_chn];
int _phase; // 0 or 1
/*\\\ FORBIDDEN MEMBER FUNCTIONS \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\*/
private:
bool operator == (const HalfBand16Avx512 &other) const = delete;
bool operator != (const HalfBand16Avx512 &other) const = delete;
}; // class HalfBand16Avx512
} // namespace hiir
#include "hiir/HalfBand16Avx512.hpp"
#endif // hiir_HalfBand16Avx512_HEADER_INCLUDED
/*\\\ EOF \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\*/