/*
 * // Copyright (c) Radzivon Bartoshyk 12/2025. All rights reserved.
 * //
 * // Redistribution and use in source and binary forms, with or without modification,
 * // are permitted provided that the following conditions are met:
 * //
 * // 1.  Redistributions of source code must retain the above copyright notice, this
 * // list of conditions and the following disclaimer.
 * //
 * // 2.  Redistributions in binary form must reproduce the above copyright notice,
 * // this list of conditions and the following disclaimer in the documentation
 * // and/or other materials provided with the distribution.
 * //
 * // 3.  Neither the name of the copyright holder nor the names of its
 * // contributors may be used to endorse or promote products derived from
 * // this software without specific prior written permission.
 * //
 * // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * // DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
 * // FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * // DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * // SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 * // CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * // OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
use crate::border_mode::{BorderInterpolation, BorderMode};
use crate::err::{OscletError, try_vec};
use crate::mla::fmla;
use crate::util::{
    dwt_length, idwt_length, low_pass_to_high_from_arr, sixteen_taps_size_for_input,
};
use crate::{DwtForwardExecutor, DwtInverseExecutor, DwtSize, IncompleteDwtExecutor};
use std::arch::aarch64::*;

pub(crate) struct NeonWavelet16TapsF32 {
    border_mode: BorderMode,
    low_pass: [f32; 16],
    high_pass: [f32; 16],
}

impl NeonWavelet16TapsF32 {
    pub(crate) fn new(border_mode: BorderMode, wavelet: &[f32; 16]) -> Self {
        Self {
            border_mode,
            low_pass: *wavelet,
            high_pass: low_pass_to_high_from_arr(wavelet),
        }
    }
}

impl DwtForwardExecutor<f32> for NeonWavelet16TapsF32 {
    fn execute_forward(
        &self,
        input: &[f32],
        approx: &mut [f32],
        details: &mut [f32],
    ) -> Result<(), OscletError> {
        let mut scratch = try_vec![f32::default(); self.required_scratch_size(input.len())];
        self.execute_forward_with_scratch(input, approx, details, &mut scratch)
    }

    fn execute_forward_with_scratch(
        &self,
        input: &[f32],
        approx: &mut [f32],
        details: &mut [f32],
        scratch: &mut [f32],
    ) -> Result<(), OscletError> {
        let half = dwt_length(input.len(), 16);

        if input.len() < 16 {
            return Err(OscletError::MinFilterSize(input.len(), 16));
        }

        if approx.len() != half {
            return Err(OscletError::ApproxDetailsSize(approx.len()));
        }
        if details.len() != half {
            return Err(OscletError::ApproxDetailsSize(details.len()));
        }

        let required_size = self.required_scratch_size(input.len());
        if scratch.len() < required_size {
            return Err(OscletError::ScratchSize(required_size, scratch.len()));
        }

        unsafe {
            let h0 = vld1q_f32(self.low_pass.as_ptr());
            let h4 = vld1q_f32(self.low_pass[4..].as_ptr());
            let h8 = vld1q_f32(self.low_pass[8..].as_ptr());
            let h12 = vld1q_f32(self.low_pass[12..].as_ptr());
            let g0 = vld1q_f32(self.high_pass.as_ptr());
            let g4 = vld1q_f32(self.high_pass[4..].as_ptr());
            let g8 = vld1q_f32(self.high_pass[8..].as_ptr());
            let g12 = vld1q_f32(self.high_pass[12..].as_ptr());

            let interpolation = BorderInterpolation::new(self.border_mode, 0, input.len() as isize);

            let (front_approx, approx) = approx.split_at_mut(7);
            let (front_detail, details) = details.split_at_mut(7);

            for (i, (approx, detail)) in front_approx
                .iter_mut()
                .zip(front_detail.iter_mut())
                .enumerate()
            {
                let base = 2 * i as isize - 14;

                let x0 = interpolation.interpolate(input, base);
                let x1 = interpolation.interpolate(input, base + 1);
                let x2 = interpolation.interpolate(input, base + 2);
                let x3 = interpolation.interpolate(input, base + 3);

                let val0 = vld1q_f32([x0, x1, x2, x3].as_ptr());

                let x4 = interpolation.interpolate(input, base + 4);
                let x5 = interpolation.interpolate(input, base + 5);
                let x6 = interpolation.interpolate(input, base + 6);
                let x7 = interpolation.interpolate(input, base + 7);

                let val1 = vld1q_f32([x4, x5, x6, x7].as_ptr());

                let x8 = interpolation.interpolate(input, base + 8);
                let x9 = interpolation.interpolate(input, base + 9);
                let x10 = interpolation.interpolate(input, base + 10);
                let x11 = interpolation.interpolate(input, base + 11);

                let val2 = vld1q_f32([x8, x9, x10, x11].as_ptr());

                let x12 = interpolation.interpolate(input, base + 12);
                let x13 = interpolation.interpolate(input, base + 13);
                let x14 = *input.get_unchecked((base + 14) as usize);
                let x15 = *input.get_unchecked((base + 15) as usize);

                let val3 = vld1q_f32([x12, x13, x14, x15].as_ptr());

                let a = vfmaq_f32(
                    vfmaq_f32(vfmaq_f32(vmulq_f32(val0, h0), val1, h4), val2, h8),
                    val3,
                    h12,
                );
                let d = vfmaq_f32(
                    vfmaq_f32(vfmaq_f32(vmulq_f32(val0, g0), val1, g4), val2, g8),
                    val3,
                    g12,
                );

                let q0 = vpadd_f32(
                    vadd_f32(vget_low_f32(a), vget_high_f32(a)),
                    vadd_f32(vget_low_f32(d), vget_high_f32(d)),
                );

                vst1_lane_f32::<0>(approx, q0);
                vst1_lane_f32::<1>(detail, q0);
            }

            let (approx, approx_rem) =
                approx.split_at_mut(sixteen_taps_size_for_input(input.len(), approx.len()));
            let (details, details_rem) =
                details.split_at_mut(sixteen_taps_size_for_input(input.len(), details.len()));

            let base_start = approx.len();

            let mut processed = 0usize;

            for (i, (approx, detail)) in approx
                .as_chunks_mut::<2>()
                .0
                .iter_mut()
                .zip(details.as_chunks_mut::<2>().0.iter_mut())
                .enumerate()
            {
                let base = 2 * 2 * i;
                let input = input.get_unchecked(base..);

                let xw0 = vld1q_f32(input.as_ptr());
                let xw1 = vld1q_f32(input.get_unchecked(4..).as_ptr());
                let xw2 = vld1q_f32(input.get_unchecked(8..).as_ptr());
                let xw3 = vld1q_f32(input.get_unchecked(12..).as_ptr());
                let xw4 = vld1_f32(input.get_unchecked(16..).as_ptr());

                let xw1_0 = vcombine_f32(vget_high_f32(xw0), vget_low_f32(xw1));
                let xw1_1 = vcombine_f32(vget_high_f32(xw1), vget_low_f32(xw2));
                let xw1_2 = vcombine_f32(vget_high_f32(xw2), vget_low_f32(xw3));
                let xw1_3 = vcombine_f32(vget_high_f32(xw3), xw4);

                let a0 = vfmaq_f32(
                    vfmaq_f32(vfmaq_f32(vmulq_f32(xw0, h0), xw1, h4), xw2, h8),
                    xw3,
                    h12,
                );
                let d0 = vfmaq_f32(
                    vfmaq_f32(vfmaq_f32(vmulq_f32(xw0, g0), xw1, g4), xw2, g8),
                    xw3,
                    g12,
                );

                let a1 = vfmaq_f32(
                    vfmaq_f32(vfmaq_f32(vmulq_f32(xw1_0, h0), xw1_1, h4), xw1_2, h8),
                    xw1_3,
                    h12,
                );
                let d1 = vfmaq_f32(
                    vfmaq_f32(vfmaq_f32(vmulq_f32(xw1_0, g0), xw1_1, g4), xw1_2, g8),
                    xw1_3,
                    g12,
                );

                let xa = vpaddq_f32(a0, a1);
                let xd = vpaddq_f32(d0, d1);

                let va = vpadd_f32(vget_low_f32(xa), vget_high_f32(xa));
                let vd = vpadd_f32(vget_low_f32(xd), vget_high_f32(xd));

                vst1_f32(approx.as_mut_ptr(), va);
                vst1_f32(detail.as_mut_ptr(), vd);

                processed += 2;
            }

            let approx = approx.chunks_exact_mut(2).into_remainder();
            let details = details.chunks_exact_mut(2).into_remainder();

            for (i, (approx, detail)) in approx.iter_mut().zip(details.iter_mut()).enumerate() {
                let base = 2 * (i + processed);
                let input = input.get_unchecked(base..);

                let xw0 = vld1q_f32(input.as_ptr());
                let xw1 = vld1q_f32(input.get_unchecked(4..).as_ptr());
                let xw2 = vld1q_f32(input.get_unchecked(8..).as_ptr());
                let xw3 = vld1q_f32(input.get_unchecked(12..).as_ptr());

                let a = vfmaq_f32(
                    vfmaq_f32(vfmaq_f32(vmulq_f32(xw0, h0), xw1, h4), xw2, h8),
                    xw3,
                    h12,
                );
                let d = vfmaq_f32(
                    vfmaq_f32(vfmaq_f32(vmulq_f32(xw0, g0), xw1, g4), xw2, g8),
                    xw3,
                    g12,
                );

                let q0 = vpadd_f32(
                    vadd_f32(vget_low_f32(a), vget_high_f32(a)),
                    vadd_f32(vget_low_f32(d), vget_high_f32(d)),
                );

                vst1_lane_f32::<0>(approx, q0);
                vst1_lane_f32::<1>(detail, q0);
            }

            for (i, (approx, detail)) in approx_rem
                .iter_mut()
                .zip(details_rem.iter_mut())
                .enumerate()
            {
                let base = 2 * (i + base_start);

                let x0 = *input.get_unchecked(base);
                let x1 = interpolation.interpolate(input, base as isize + 1);
                let x2 = interpolation.interpolate(input, base as isize + 2);
                let x3 = interpolation.interpolate(input, base as isize + 3);

                let val0 = vld1q_f32([x0, x1, x2, x3].as_ptr());

                let x4 = interpolation.interpolate(input, base as isize + 4);
                let x5 = interpolation.interpolate(input, base as isize + 5);
                let x6 = interpolation.interpolate(input, base as isize + 6);
                let x7 = interpolation.interpolate(input, base as isize + 7);

                let val1 = vld1q_f32([x4, x5, x6, x7].as_ptr());

                let x8 = interpolation.interpolate(input, base as isize + 8);
                let x9 = interpolation.interpolate(input, base as isize + 9);
                let x10 = interpolation.interpolate(input, base as isize + 10);
                let x11 = interpolation.interpolate(input, base as isize + 11);

                let val2 = vld1q_f32([x8, x9, x10, x11].as_ptr());

                let x12 = interpolation.interpolate(input, base as isize + 12);
                let x13 = interpolation.interpolate(input, base as isize + 13);
                let x14 = interpolation.interpolate(input, base as isize + 14);
                let x15 = interpolation.interpolate(input, base as isize + 15);

                let val3 = vld1q_f32([x12, x13, x14, x15].as_ptr());

                let a = vfmaq_f32(
                    vfmaq_f32(vfmaq_f32(vmulq_f32(val0, h0), val1, h4), val2, h8),
                    val3,
                    h12,
                );
                let d = vfmaq_f32(
                    vfmaq_f32(vfmaq_f32(vmulq_f32(val0, g0), val1, g4), val2, g8),
                    val3,
                    g12,
                );

                let q0 = vpadd_f32(
                    vadd_f32(vget_low_f32(a), vget_high_f32(a)),
                    vadd_f32(vget_low_f32(d), vget_high_f32(d)),
                );

                vst1_lane_f32::<0>(approx, q0);
                vst1_lane_f32::<1>(detail, q0);
            }
        }
        Ok(())
    }

    fn required_scratch_size(&self, _: usize) -> usize {
        0
    }

    fn dwt_size(&self, input_length: usize) -> DwtSize {
        DwtSize::new(dwt_length(input_length, self.filter_length()))
    }
}

impl DwtInverseExecutor<f32> for NeonWavelet16TapsF32 {
    fn execute_inverse(
        &self,
        approx: &[f32],
        details: &[f32],
        output: &mut [f32],
    ) -> Result<(), OscletError> {
        if approx.len() != details.len() {
            return Err(OscletError::ApproxDetailsNotMatches(
                approx.len(),
                details.len(),
            ));
        }

        let rec_len = idwt_length(approx.len(), 16);

        if output.len() != rec_len {
            return Err(OscletError::OutputSizeIsNotValid(output.len(), rec_len));
        }

        const FILTER_OFFSET: usize = 14;
        const FILTER_LENGTH: usize = 16;

        unsafe {
            let safe_start = FILTER_OFFSET;
            // 2*x - off + len >= output.len()
            // x >= (output.len() + off - len)/2
            let mut safe_end = ((output.len() + FILTER_OFFSET).saturating_sub(FILTER_LENGTH)) / 2;

            if safe_start < safe_end {
                for i in 0..safe_start {
                    let (h, g) = (*approx.get_unchecked(i), *details.get_unchecked(i));
                    let k = 2 * i as isize - FILTER_OFFSET as isize;
                    for j in 0..FILTER_LENGTH {
                        let k = k + j as isize;
                        if k >= 0 && k < rec_len as isize {
                            *output.get_unchecked_mut(k as usize) = fmla(
                                self.low_pass[j],
                                h,
                                fmla(self.high_pass[j], g, *output.get_unchecked(k as usize)),
                            );
                        }
                    }
                }

                let h0 = vld1q_f32(self.low_pass.as_ptr());
                let h4 = vld1q_f32(self.low_pass[4..].as_ptr());
                let h8 = vld1q_f32(self.low_pass[8..].as_ptr());
                let h12 = vld1q_f32(self.low_pass[12..].as_ptr());
                let g0 = vld1q_f32(self.high_pass.as_ptr());
                let g4 = vld1q_f32(self.high_pass[4..].as_ptr());
                let g8 = vld1q_f32(self.high_pass[8..].as_ptr());
                let g12 = vld1q_f32(self.high_pass[12..].as_ptr());

                let mut ui = safe_start;

                while ui + 4 <= safe_end {
                    let (h, g) = (
                        vld1q_f32(approx.get_unchecked(ui)),
                        vld1q_f32(details.get_unchecked(ui)),
                    );

                    let k = 2 * ui as isize - FILTER_OFFSET as isize;
                    let part = output.get_unchecked_mut(k as usize..);

                    macro_rules! step_by {
                        ($part: expr, $h: expr, $g: expr, $w: expr) => {
                            let xw0 = vld1q_f32($part.get_unchecked(2 * $w..).as_ptr());
                            let xw1 = vld1q_f32($part.get_unchecked(4 + 2 * $w..).as_ptr());
                            let xw2 = vld1q_f32($part.get_unchecked(8 + 2 * $w..).as_ptr());
                            let xw3 = vld1q_f32($part.get_unchecked(12 + 2 * $w..).as_ptr());

                            let q0 =
                                vfmaq_laneq_f32::<$w>(vfmaq_laneq_f32::<$w>(xw0, h0, $h), g0, $g);
                            let q2 =
                                vfmaq_laneq_f32::<$w>(vfmaq_laneq_f32::<$w>(xw1, h4, $h), g4, $g);
                            let q8 =
                                vfmaq_laneq_f32::<$w>(vfmaq_laneq_f32::<$w>(xw2, h8, $h), g8, $g);
                            let q12 =
                                vfmaq_laneq_f32::<$w>(vfmaq_laneq_f32::<$w>(xw3, h12, $h), g12, $g);

                            vst1q_f32($part.get_unchecked_mut(2 * $w..).as_mut_ptr(), q0);
                            vst1q_f32($part.get_unchecked_mut(4 + 2 * $w..).as_mut_ptr(), q2);
                            vst1q_f32($part.get_unchecked_mut(8 + 2 * $w..).as_mut_ptr(), q8);
                            vst1q_f32($part.get_unchecked_mut(12 + 2 * $w..).as_mut_ptr(), q12);
                        };
                    }

                    step_by!(part, h, g, 0);
                    step_by!(part, h, g, 1);
                    step_by!(part, h, g, 2);
                    step_by!(part, h, g, 3);

                    ui += 4;
                }

                for i in ui..safe_end {
                    let (h, g) = (*approx.get_unchecked(i), *details.get_unchecked(i));
                    let k = 2 * i as isize - FILTER_OFFSET as isize;
                    let part = output.get_unchecked_mut(k as usize..);
                    let xw0 = vld1q_f32(part.as_ptr());
                    let xw1 = vld1q_f32(part.get_unchecked(4..).as_ptr());
                    let xw2 = vld1q_f32(part.get_unchecked(8..).as_ptr());
                    let xw3 = vld1q_f32(part.get_unchecked(12..).as_ptr());

                    let q0 = vfmaq_n_f32(vfmaq_n_f32(xw0, h0, h), g0, g);
                    let q4 = vfmaq_n_f32(vfmaq_n_f32(xw1, h4, h), g4, g);
                    let q8 = vfmaq_n_f32(vfmaq_n_f32(xw2, h8, h), g8, g);
                    let q12 = vfmaq_n_f32(vfmaq_n_f32(xw3, h12, h), g12, g);

                    vst1q_f32(part.as_mut_ptr(), q0);
                    vst1q_f32(part.get_unchecked_mut(4..).as_mut_ptr(), q4);
                    vst1q_f32(part.get_unchecked_mut(8..).as_mut_ptr(), q8);
                    vst1q_f32(part.get_unchecked_mut(12..).as_mut_ptr(), q12);
                }
            } else {
                safe_end = 0usize;
            }

            for i in safe_end..approx.len() {
                let (h, g) = (*approx.get_unchecked(i), *details.get_unchecked(i));
                let k = 2 * i as isize - FILTER_OFFSET as isize;
                for j in 0..FILTER_LENGTH {
                    let k = k + j as isize;
                    if k >= 0 && k < rec_len as isize {
                        *output.get_unchecked_mut(k as usize) = fmla(
                            self.low_pass[j],
                            h,
                            fmla(self.high_pass[j], g, *output.get_unchecked(k as usize)),
                        );
                    }
                }
            }
        }
        Ok(())
    }

    fn idwt_size(&self, input_length: DwtSize) -> usize {
        idwt_length(input_length.approx_length, self.filter_length())
    }
}

impl IncompleteDwtExecutor<f32> for NeonWavelet16TapsF32 {
    fn filter_length(&self) -> usize {
        16
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::{DaubechiesFamily, WaveletFilterProvider};

    #[test]
    fn test_db8_odd() {
        let input = vec![
            1.0, 2.0, 3.0, 4.0, 2.0, 1.0, 0.0, 1.0, 2.4, 6.5, 2.4, 6.4, 5.2, 0.6, 0.5, 1.3, 2.5,
            5.1, 0.5, 0.6, 0.5,
        ];

        let db8 = NeonWavelet16TapsF32::new(
            BorderMode::Wrap,
            DaubechiesFamily::Db8
                .get_wavelet()
                .as_ref()
                .try_into()
                .unwrap(),
        );
        let out_length = dwt_length(input.len(), 16);
        let mut approx = vec![0.0; out_length];
        let mut details = vec![0.0; out_length];
        db8.execute_forward(&input, &mut approx, &mut details)
            .unwrap();

        const REFERENCE_APPROX: [f32; 18] = [
            5.0284267712823505,
            8.541490130983899,
            1.9934790471164194,
            2.333844690825454,
            4.506488742584248,
            0.9811126491855129,
            1.0171300022100558,
            4.59743726177117,
            3.6575696466567176,
            -0.1642348305647749,
            3.974315145990313,
            7.189070703036446,
            6.2502770929916815,
            0.1407324473575215,
            4.960922595291658,
            2.3705953140504503,
            0.502497757811914,
            2.77235543211178,
        ];
        const REFERENCE_DETAILS: [f32; 18] = [
            -1.0840275905588241,
            0.45372515952711434,
            -0.7922888215418205,
            -0.1891602885685358,
            -0.4587880839845288,
            -2.392006887002547,
            -1.9956755891867481,
            2.4874710348921214,
            -1.2482576077146206,
            -1.371344951587655,
            -0.8335157086113016,
            0.32967560099755683,
            0.1852242463136476,
            0.7904451976777772,
            -0.06646607760682827,
            1.050248702399062,
            3.62357438049428,
            -1.3109650760645661,
        ];

        approx.iter().enumerate().for_each(|(i, x)| {
            assert!(
                (REFERENCE_APPROX[i] - x).abs() < 1e-3,
                "approx difference expected to be < 1e-3, but values were ref {}, derived {}",
                REFERENCE_APPROX[i],
                x
            );
        });
        details.iter().enumerate().for_each(|(i, x)| {
            assert!(
                (REFERENCE_DETAILS[i] - x).abs() < 1e-3,
                "details difference expected to be < 1e-3, but values were ref {}, derived {}",
                REFERENCE_DETAILS[i],
                x
            );
        });

        let mut reconstructed = vec![0.0; idwt_length(approx.len(), 16)];
        db8.execute_inverse(&approx, &details, &mut reconstructed)
            .unwrap();
        reconstructed.iter().take(input.len()).enumerate().for_each(|(i, x)| {
            assert!(
                (input[i] - x).abs() < 1e-3,
                "reconstructed difference expected to be < 1e-3, but values were ref {}, derived {}",
                input[i],
                x
            );
        });
    }

    #[test]
    fn test_db8_even() {
        let input = vec![
            1.0, 2.0, 3.0, 4.0, 2.0, 1.0, 0.0, 1.0, 2.4, 6.5, 2.4, 6.4, 5.2, 0.6, 0.5, 1.3, 6.5,
            1.23,
        ];

        let db8 = NeonWavelet16TapsF32::new(
            BorderMode::Wrap,
            DaubechiesFamily::Db8
                .get_wavelet()
                .as_ref()
                .try_into()
                .unwrap(),
        );
        let out_length = dwt_length(input.len(), 16);
        let mut approx = vec![0.0; out_length];
        let mut details = vec![0.0; out_length];
        db8.execute_forward(&input, &mut approx, &mut details)
            .unwrap();

        const REFERENCE_APPROX: [f32; 16] = [
            -0.1820269500881618,
            3.9947444538407444,
            7.29633393733095,
            5.742896296102855,
            1.1839137971766258,
            5.394627310820256,
            1.5665797328118078,
            4.59743726177117,
            3.6607260794370826,
            -0.1820269500881618,
            3.9947444538407444,
            7.29633393733095,
            5.742896296102855,
            1.1839137971766258,
            5.394627310820256,
            1.5665797328118078,
        ];
        const REFERENCE_DETAILS: [f32; 16] = [
            3.6650183488164876,
            -2.065110756232226,
            0.2719729833884861,
            -0.5029963198341018,
            -0.4108687969580707,
            -2.390762763838438,
            -1.9968617802797932,
            2.4874710348921214,
            0.21381806542339166,
            3.6650183488164876,
            -2.065110756232226,
            0.2719729833884861,
            -0.5029963198341018,
            -0.4108687969580707,
            -2.390762763838438,
            -1.9968617802797932,
        ];

        approx.iter().enumerate().for_each(|(i, x)| {
            assert!(
                (REFERENCE_APPROX[i] - x).abs() < 1e-3,
                "approx difference expected to be < 1e-3, but values were ref {}, derived {}",
                REFERENCE_APPROX[i],
                x
            );
        });
        details.iter().enumerate().for_each(|(i, x)| {
            assert!(
                (REFERENCE_DETAILS[i] - x).abs() < 1e-3,
                "details difference expected to be < 1e-3, but values were ref {}, derived {}",
                REFERENCE_DETAILS[i],
                x
            );
        });

        let mut reconstructed = vec![0.0; idwt_length(approx.len(), 16)];
        db8.execute_inverse(&approx, &details, &mut reconstructed)
            .unwrap();
        reconstructed.iter().take(input.len()).enumerate().for_each(|(i, x)| {
            assert!(
                (input[i] - x).abs() < 1e-3,
                "reconstructed difference expected to be < 1e-3, but values were ref {}, derived {}",
                input[i],
                x
            );
        });
    }

    #[test]
    fn test_db8_even_big() {
        let data_length = 86;
        let mut input = vec![0.; data_length];
        for i in 0..data_length {
            input[i] = i as f32 / data_length as f32;
        }
        let db8 = NeonWavelet16TapsF32::new(
            BorderMode::Wrap,
            DaubechiesFamily::Db8
                .get_wavelet()
                .as_ref()
                .try_into()
                .unwrap(),
        );
        let out_length = dwt_length(input.len(), 16);
        let mut approx = vec![0.0; out_length];
        let mut details = vec![0.0; out_length];
        db8.execute_forward(&input, &mut approx, &mut details)
            .unwrap();

        let mut reconstructed = vec![0.0; idwt_length(approx.len(), 16)];
        db8.execute_inverse(&approx, &details, &mut reconstructed)
            .unwrap();
        reconstructed.iter().take(input.len()).enumerate().for_each(|(i, x)| {
            assert!(
                (input[i] - x).abs() < 1e-3,
                "reconstructed difference expected to be < 1e-3, but values were ref {}, derived {}",
                input[i],
                x
            );
        });
    }
}
