geometryinfo.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef GEOMETRYINFO_H
#define GEOMETRYINFO_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_global.h"
 
#include <fiff/fiff_evoked.h>
 
//=============================================================================================================
// STL INCLUDES
//=============================================================================================================
 
#include <atomic>
#include <limits>
#include <functional>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QSharedPointer>
#include <QVector>
#include <vector>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
#include <Eigen/Sparse>
 
//=============================================================================================================
// DEFINE NAMESPACE
//=============================================================================================================
 
namespace DISP3DLIB
{
 
#define FLOAT_INFINITY std::numeric_limits<float>::infinity()
 
//=============================================================================================================
 
class DISP3DSHARED_EXPORT GeometryInfo
{
public:
    typedef QSharedPointer<GeometryInfo> SPtr;
    typedef QSharedPointer<const GeometryInfo> ConstSPtr;
 
    GeometryInfo() = delete;
 
    //=========================================================================================================
    static QSharedPointer<Eigen::MatrixXd> scdc(const Eigen::MatrixX3f& matVertices,
                                                const std::vector<Eigen::VectorXi>& vecNeighborVertices,
                                                Eigen::VectorXi& vecVertSubset,
                                                double dCancelDist = FLOAT_INFINITY);
 
    //=========================================================================================================
    static QSharedPointer<Eigen::SparseMatrix<float>> scdcInterpolationMat(
        const Eigen::MatrixX3f& matVertices,
        const std::vector<Eigen::VectorXi>& vecNeighborVertices,
        const Eigen::VectorXi& vecVertSubset,
        double (*interpolationFunction)(double),
        double dCancelDist,
        std::function<void(int, int)> progressCallback = nullptr,
        const std::atomic<bool>* cancelledFlag = nullptr);
 
    //=========================================================================================================
    static Eigen::VectorXi projectSensors(const Eigen::MatrixX3f& matVertices,
                                          const Eigen::MatrixX3f& matSensorPositions);
 
    //=========================================================================================================
    static Eigen::VectorXi filterBadChannels(QSharedPointer<Eigen::MatrixXd> matDistanceTable,
                                             const FIFFLIB::FiffInfo& fiffInfo,
                                             qint32 iSensorType);
 
protected:
    static inline double squared(double dBase);
 
    static Eigen::VectorXi nearestNeighbor(const Eigen::MatrixX3f& matVertices,
                                           const Eigen::MatrixX3f& matSensorPositions,
                                           qint32 iBegin,
                                           qint32 iEnd);
 
    static void iterativeDijkstra(QSharedPointer<Eigen::MatrixXd> matOutputDistMatrix,
                                  const Eigen::MatrixX3f& matVertices,
                                  const std::vector<Eigen::VectorXi>& vecNeighborVertices,
                                  const Eigen::VectorXi& vecVertSubset,
                                  qint32 iBegin,
                                  qint32 iEnd,
                                  double dCancelDistance);
};
 
//=============================================================================================================
// INLINE DEFINITIONS
//=============================================================================================================
 
double GeometryInfo::squared(double dBase)
{
    return dBase * dBase;
}
 
} // namespace DISP3DLIB
 
#endif // GEOMETRYINFO_H