StationaryWaveletPacketTree.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 STATIONARYWAVELETPACKETTREE_H
#define STATIONARYWAVELETPACKETTREE_H
 
#include <vector>
 
#include "Tree.h"
#include "Wavelet.h"
#include "WaveletMath.h"
#include "WaveletPacketTreeTemplateBase.h"
 
namespace panwave {
 
class StationaryWaveletPacketTree : public WaveletPacketTreeTemplateBase<4> {
 public:
  StationaryWaveletPacketTree(size_t height, const Wavelet* wavelet,
                              PaddingMode padding_mode = PaddingMode::Zeroes);
  ~StationaryWaveletPacketTree() override = default;
 
  void Decompose() override;
  void Reconstruct(size_t level) override;
 
 protected:
  void DecomposeNode(size_t node);
  void ReconstructNode(size_t node);
 
  void ReconstructAccumulate(size_t leaf_node,
                             std::vector<double>* accumulated_signal);
 
 private:
  PaddingMode padding_mode_;
};
 
};  // namespace panwave
 
#endif  // STATIONARYWAVELETPACKETTREE_H