WaveletMath.h

//-------------------------------------------------------------------------------------------------------
// Copyright (C) Taylor Woll and panwave contributors. All rights reserved.
// Licensed under the MIT license. See LICENSE.txt file in the project root for
// full license information.
//-------------------------------------------------------------------------------------------------------
 
#ifndef WAVELETMATH_H
#define WAVELETMATH_H
 
#include <cstdint>
#include <vector>
 
namespace panwave {
class Wavelet;
}  // namespace panwave
 
namespace panwave {
 
enum class DyadicMode : uint8_t { Even = 0, Odd };
 
enum class PaddingMode : uint8_t { Zeroes = 0, Symmetric };
 
class WaveletMath {
 public:
  static void Decompose(const std::vector<double>* data, const Wavelet* wavelet,
                        std::vector<double>* approx_coeffs,
                        std::vector<double>* details_coeffs,
                        DyadicMode dyadic_mode, PaddingMode padding_mode);
 
  static void Reconstruct(const std::vector<double>* coeffs,
                          const std::vector<double>* reconstruction_coeffs,
                          std::vector<double>* data, const Wavelet* wavelet,
                          size_t data_size, DyadicMode dyadic_mode,
                          PaddingMode padding_mode);
 
  static void DyadicUpsample(const std::vector<double>* data,
                             std::vector<double>* data_upsampled,
                             DyadicMode dyadic_mode);
 
  static void DyadicDownsample(const std::vector<double>* data,
                               std::vector<double>* data_downsampled,
                               DyadicMode dyadic_mode);
 
  static void Convolve(const std::vector<double>* data,
                       const std::vector<double>* coeffs,
                       std::vector<double>* result);
 
  static void Pad(const std::vector<double>* data,
                  std::vector<double>* extended_data, size_t pad_left,
                  size_t pad_right, PaddingMode padding_mode);
};
 
}  // namespace panwave
 
#endif  // WAVELETMATH_H