mvar_model.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef MVARMODEL_H
#define MVARMODEL_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../conn_global.h"
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QVector>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
//=============================================================================================================
// DEFINE NAMESPACE CONNLIB
//=============================================================================================================
 
namespace CONNLIB {
 
//=============================================================================================================
// CONNLIB FORWARD DECLARATIONS
//=============================================================================================================
 
//=============================================================================================================
class CONNSHARED_EXPORT MvarModel
{
 
public:
    //=========================================================================================================
    MvarModel() = default;
 
    //=========================================================================================================
    void fit(const Eigen::MatrixXd& data, int p = 0);
 
    //=========================================================================================================
    QVector<Eigen::MatrixXd> coefficients() const;
 
    //=========================================================================================================
    Eigen::MatrixXd noiseCov() const;
 
    //=========================================================================================================
    int order() const;
 
    //=========================================================================================================
    QVector<Eigen::MatrixXcd> transferFunction(const Eigen::VectorXd& freqs) const;
 
    //=========================================================================================================
    QVector<Eigen::MatrixXcd> spectralMatrix(const Eigen::VectorXd& freqs) const;
 
private:
    //=========================================================================================================
    void fitLevinsonDurbin(const Eigen::MatrixXd& data, int p);
 
    //=========================================================================================================
    int selectOrderBIC(const Eigen::MatrixXd& data, int maxOrder = 20) const;
 
    QVector<Eigen::MatrixXd>    m_coeffs;       
    Eigen::MatrixXd             m_noiseCov;     
    int                         m_order = 0;    
    int                         m_nChannels = 0;
};
 
//=============================================================================================================
// INLINE DEFINITIONS
//=============================================================================================================
} // namespace CONNLIB
 
#endif // MVARMODEL_H