WaveletPacketTree.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 WAVELETPACKETTREE_H
#define WAVELETPACKETTREE_H
 
#include "WaveletMath.h"
#include "WaveletPacketTreeTemplateBase.h"
 
namespace panwave {
 
class Wavelet;
 
class WaveletPacketTree : public WaveletPacketTreeTemplateBase<2> {
 public:
  WaveletPacketTree(size_t height, const Wavelet* wavelet,
                    DyadicMode dyadic_mode = DyadicMode::Odd,
                    PaddingMode padding_mode = PaddingMode::Zeroes);
  ~WaveletPacketTree() override = default;
 
  void Decompose() override;
  void Reconstruct(size_t level) override;
 
 protected:
  void IsolateLevel(size_t level);
 
  void DecomposeNode(size_t node);
  void ReconstructNode(size_t node);
 
 private:
  DyadicMode dyadic_mode_;
  PaddingMode padding_mode_;
};
 
}  // namespace panwave
 
#endif  // WAVELETPACKETTREE_H