CQTProcessor.hpp

Go to the documentation of this file.

//
//  CQTProcessor.hpp
//  CQTDSP
//
//  Created by Juan Sierra on 3/17/25.
//

 
#pragma once
 
#include <Eigen/Core>
 
#include "CQT.hpp"
#include "DoubleBuffer.h"
#include "Slicer.hpp"
#include "Splicer.hpp"
 
namespace jsa::cicuetea {

class CqtDenseProcessor
{
  public:
    CqtDenseProcessor(double sampleRate, Eigen::Index numSamples, double fraction,
                      double minFrequency, double maxFrequency, double refFrequency);

    virtual ~CqtDenseProcessor() = default;

    double processSample(double sample);

    virtual void processBlock(Eigen::ArrayXXcd& block) = 0;

    const Eigen::ArrayXd& getWindow() const { return win; }

    const NsgfCqtDense& getCqt() const { return cqt; }

    Eigen::Index getLatency() const { return slicer.getBlockSize(); }

    bool isValid() const { return cqt.isValid(); }
 
  protected:
    NsgfCqtDense cqt; 
 
  private:
    Eigen::ArrayXd   xi;      
    Eigen::ArrayXd   win;     
    Eigen::ArrayXXcd Xcq;     
    Slicer           slicer;  
    Splicer          splicer; 
};
 
//==========================================================================

class CqtSparseProcessor
{
  public:
    CqtSparseProcessor(double sampleRate, Eigen::Index numSamples, double fraction,
                       double minFrequency, double maxFrequency, double refFrequency);

    virtual ~CqtSparseProcessor() = default;

    double processSample(double sample);

    virtual void processBlock(NsgfCqtSparse::Coefs& block) = 0;

    const Eigen::ArrayXd& getWindow() const { return win; }

    const NsgfCqtSparse& getCqt() const { return cqt; }

    Eigen::Index getLatency() const { return slicer.getBlockSize(); }

    bool isValid() const { return cqt.isValid(); }
 
  protected:
    NsgfCqtSparse cqt; 
 
  private:
    Eigen::ArrayXd       xi;      
    Eigen::ArrayXd       win;     
    NsgfCqtSparse::Coefs Xcq;     
    Slicer               slicer;  
    Splicer              splicer; 
};
 
//==========================================================================

class SlidingCqtDenseProcessor
{
  public:
    SlidingCqtDenseProcessor(double sampleRate, Eigen::Index numSamples, double fraction,
                             double minFrequency, double maxFrequency, double refFrequency);

    virtual ~SlidingCqtDenseProcessor() = default;

    double processSample(double sample);

    virtual void processBlock(Eigen::ArrayXXcd& block) = 0;

    const Eigen::ArrayXd& getWindow() const { return win; }

    const NsgfCqtDense& getCqt() const { return cqt; }

    Eigen::Index getLatency() const { return slicer.getBlockSize() + slicer.getHopSize(); }

    bool isValid() const { return cqt.isValid(); }
 
  protected:
    NsgfCqtDense cqt; 
 
  private:
    Eigen::ArrayXd                 xi;      
    DoubleBuffer<Eigen::ArrayXXcd> Xcq;     
    DoubleBuffer<Eigen::ArrayXXcd> Zcq;     
    Eigen::ArrayXXcd               Ycq;     
    Eigen::ArrayXd                 win;     
    Slicer                         slicer;  
    Splicer                        splicer; 
};
 
//==========================================================================

class SlidingCqtSparseProcessor
{
  public:
    SlidingCqtSparseProcessor(double sampleRate, Eigen::Index numSamples,
                              double fraction, double minFrequency,
                              double maxFrequency, double refFrequency);

    virtual ~SlidingCqtSparseProcessor() = default;

    double processSample(double sample);

    virtual void processBlock(NsgfCqtSparse::Coefs& block) = 0;

    const Eigen::ArrayXd& getWindow() const { return win; }

    const Eigen::ArrayXd& getCqtWindow(Eigen::Index k) const { return Win[k]; }

    const NsgfCqtSparse& getCqt() const { return cqt; }

    Eigen::Index getLatency() const { return slicer.getBlockSize() + slicer.getHopSize(); }

    bool isValid() const { return cqt.isValid(); }
 
  protected:
    NsgfCqtSparse cqt; 
 
  private:
    Eigen::ArrayXd                     xi;      
    DoubleBuffer<NsgfCqtSparse::Coefs> Xcq;     
    DoubleBuffer<NsgfCqtSparse::Coefs> Zcq;     
    NsgfCqtSparse::Coefs               Ycq;     
    Eigen::ArrayXd                     win;     
    NsgfCqtSparse::Frame               Win;     
    Slicer                             slicer;  
    Splicer                            splicer; 
};
 
} // namespace jsa::cicuetea