lib/jxl/dct-inl.h

38 using FV = typename FVImpl<SZ>::type;
49       auto in1 = Load(FV<SZ>(), ain1 + i * SZ);
50       auto in2 = Load(FV<SZ>(), ain2 + (N - i - 1) * SZ);
51       Store(in1 + in2, FV<SZ>(), aout + i * SZ);
58       auto in1 = Load(FV<SZ>(), ain1 + i * SZ);
59       auto in2 = Load(FV<SZ>(), ain2 + (N - i - 1) * SZ);
60       Store(in1 - in2, FV<SZ>(), aout + i * SZ);
64     auto sqrt2 = Set(FV<SZ>(), kSqrt2);
65     auto in1 = Load(FV<SZ>(), coeff);
66     auto in2 = Load(FV<SZ>(), coeff + SZ);
67     Store(MulAdd(in1, sqrt2, in2), FV<SZ>(), coeff);
69       auto in1 = Load(FV<SZ>(), coeff + i * SZ);
70       auto in2 = Load(FV<SZ>(), coeff + (i + 1) * SZ);
71       Store(in1 + in2, FV<SZ>(), coeff + i * SZ);
76       auto in1 = Load(FV<SZ>(), coeff + i * SZ);
77       auto in2 = Load(FV<SZ>(), coeff + (i - 1) * SZ);
78       Store(in1 + in2, FV<SZ>(), coeff + i * SZ);
80     auto sqrt2 = Set(FV<SZ>(), kSqrt2);
81     auto in1 = Load(FV<SZ>(), coeff);
82     Store(in1 * sqrt2, FV<SZ>(), coeff);
88       auto in1 = Load(FV<SZ>(), ain + i * SZ);
89       Store(in1, FV<SZ>(), aout + 2 * i * SZ);
92       auto in1 = Load(FV<SZ>(), ain + i * SZ);
93       Store(in1, FV<SZ>(), aout + (2 * (i - N / 2) + 1) * SZ);
100       auto in1 = LoadU(FV<SZ>(), ain + 2 * i * ain_stride);
101       Store(in1, FV<SZ>(), aout + i * SZ);
104       auto in1 = LoadU(FV<SZ>(), ain + (2 * (i - N / 2) + 1) * ain_stride);
105       Store(in1, FV<SZ>(), aout + i * SZ);
111       auto in1 = Load(FV<SZ>(), coeff + (N / 2 + i) * SZ);
112       auto mul = Set(FV<SZ>(), WcMultipliers<N>::kMultipliers[i]);
113       Store(in1 * mul, FV<SZ>(), coeff + (N / 2 + i) * SZ);
119       auto mul = Set(FV<SZ>(), WcMultipliers<N>::kMultipliers[i]);
120       auto in1 = Load(FV<SZ>(), coeff + i * SZ);
121       auto in2 = Load(FV<SZ>(), coeff + (N / 2 + i) * SZ);
124       StoreU(out1, FV<SZ>(), out + i * out_stride);
125       StoreU(out2, FV<SZ>(), out + (N - i - 1) * out_stride);
132       Store(in.LoadPart(FV<SZ>(), i, off), FV<SZ>(), coeff + i * SZ);
138     auto mul = Set(FV<SZ>(), 1.0f / N);
140       out.StorePart(FV<SZ>(), mul * Load(FV<SZ>(), coeff + i * SZ), i, off);
156     auto in1 = Load(FV<SZ>(), mem);
157     auto in2 = Load(FV<SZ>(), mem + SZ);
158     Store(in1 + in2, FV<SZ>(), mem);
159     Store(in1 - in2, FV<SZ>(), mem + SZ);
185     StoreU(LoadU(FV<SZ>(), from), FV<SZ>(), to);
195     auto in1 = LoadU(FV<SZ>(), from);
196     auto in2 = LoadU(FV<SZ>(), from + from_stride);
197     StoreU(in1 + in2, FV<SZ>(), to);
198     StoreU(in1 - in2, FV<SZ>(), to + to_stride);
221   constexpr size_t SZ = MaxLanes(FV<M_or_0>());
223   for (size_t i = 0; i < M; i += Lanes(FV<M_or_0>())) {
236   constexpr size_t SZ = MaxLanes(FV<M_or_0>());
237   for (size_t i = 0; i < M; i += Lanes(FV<M_or_0>())) {
252 struct DCT1D<N, M, typename std::enable_if<(M > MaxLanes(FV<0>()))>::type> {
268 struct IDCT1D<N, M, typename std::enable_if<(M > MaxLanes(FV<0>()))>::type> {