fiff_sparse_matrix.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef FIFFSPARSEMATRIX_H
#define FIFFSPARSEMATRIX_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "fiff_global.h"
#include "fiff_types.h"
#include "fiff_tag.h"
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
#include <Eigen/SparseCore>
 
//=============================================================================================================
// STL INCLUDES
//=============================================================================================================
 
#include <memory>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QSharedPointer>
#include <QDebug>
 
//=============================================================================================================
// DEFINE NAMESPACE FIFFLIB
//=============================================================================================================
 
namespace FIFFLIB
{
 
//=============================================================================================================
class FIFFSHARED_EXPORT FiffSparseMatrix
{
public:
    using SPtr = QSharedPointer<FiffSparseMatrix>;            
    using ConstSPtr = QSharedPointer<const FiffSparseMatrix>; 
    using UPtr = std::unique_ptr<FiffSparseMatrix>;           
    using ConstUPtr = std::unique_ptr<const FiffSparseMatrix>;  
 
    //=========================================================================================================
    FiffSparseMatrix();
 
    //=========================================================================================================
    FiffSparseMatrix(const FiffSparseMatrix& mat) = default;
 
    //=========================================================================================================
    FiffSparseMatrix(FiffSparseMatrix&& mat) = default;
 
    //=========================================================================================================
    FiffSparseMatrix& operator=(const FiffSparseMatrix&) = default;
 
    //=========================================================================================================
    FiffSparseMatrix& operator=(FiffSparseMatrix&&) = default;
 
    //=========================================================================================================
    ~FiffSparseMatrix() = default;
 
    //============================= fiff_sparse.c =============================

    static std::vector<int> fiff_get_matrix_sparse_dims(const FIFFLIB::FiffTag::UPtr& tag);

    static FiffSparseMatrix::UPtr fiff_get_float_sparse_matrix(const FIFFLIB::FiffTag::UPtr& tag);
 
    //============================= mne_sparse_matop.c =============================

    static FiffSparseMatrix::UPtr create_sparse_rcs(int nrow,
                                                    int ncol,
                                                    int *nnz,
                                                    int **colindex,
                                                    float **vals);

    FiffSparseMatrix::UPtr mne_add_upper_triangle_rcs();

    FiffSparseMatrix::UPtr pickLowerTriangleRcs() const;

    inline bool is_empty() const;

    Eigen::SparseMatrix<double> toEigenSparse() const;

    static FiffSparseMatrix fromEigenSparse(const Eigen::SparseMatrix<double>& mat);
 
public:
    FIFFLIB::fiff_int_t   coding;    
    FIFFLIB::fiff_int_t   m;         
    FIFFLIB::fiff_int_t   n;         
    FIFFLIB::fiff_int_t   nz;        
    Eigen::VectorXf       data;      
    Eigen::VectorXi       inds;      
    Eigen::VectorXi       ptrs;      
 
};
 
//=============================================================================================================
// INLINE DEFINITIONS
//=============================================================================================================
 
inline bool FiffSparseMatrix::is_empty() const
{
    return nz <= 0 || data.size() == 0;
}
 
} // NAMESPACE FIFFLIB
 
#endif // FIFFSPARSEMATRIX_H