network.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef NETWORK_H
#define NETWORK_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../connectivity_global.h"
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QSharedPointer>
#include <QList>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
//=============================================================================================================
// DEFINE NAMESPACE CONNECTIVITYLIB
//=============================================================================================================
 
namespace CONNECTIVITYLIB
{
 
//=============================================================================================================
// CONNECTIVITYLIB FORWARD DECLARATIONS
//=============================================================================================================
 
class NetworkEdge;
class NetworkNode;

struct VisualizationInfo
{
    QString sMethod = "Map";                                    
    QString sColormap = "Viridis";                              
    Eigen::Vector4i colNodes = Eigen::Vector4i(255, 0, 0, 255); 
    Eigen::Vector4i colEdges = Eigen::Vector4i(255, 0, 0, 255); 
};
 
//=============================================================================================================
 
class CONNECTIVITYSHARED_EXPORT Network
{
public:
    typedef QSharedPointer<Network> SPtr;            
    typedef QSharedPointer<const Network> ConstSPtr; 
 
    //=========================================================================================================
    explicit Network(const QString& sConnectivityMethod = "Unknown",
                     double dThreshold = 0.0);
 
    //=========================================================================================================
    Eigen::MatrixXd getFullConnectivityMatrix(bool bGetMirroredVersion = true) const;
 
    //=========================================================================================================
    Eigen::MatrixXd getThresholdedConnectivityMatrix(bool bGetMirroredVersion = true) const;
 
    //=========================================================================================================
    const QList<QSharedPointer<NetworkEdge>>& getFullEdges() const;
 
    //=========================================================================================================
    const QList<QSharedPointer<NetworkEdge>>& getThresholdedEdges() const;
 
    //=========================================================================================================
    const QList<QSharedPointer<NetworkNode>>& getNodes() const;
 
    //=========================================================================================================
    QSharedPointer<NetworkEdge> getEdgeAt(int i);
 
    //=========================================================================================================
    QSharedPointer<NetworkNode> getNodeAt(int i);
 
    //=========================================================================================================
    qint16 getFullDistribution() const;
 
    //=========================================================================================================
    qint16 getThresholdedDistribution() const;
 
    //=========================================================================================================
    void setConnectivityMethod(const QString& sConnectivityMethod);
 
    //=========================================================================================================
    QString getConnectivityMethod() const;
 
    //=========================================================================================================
    QPair<double, double> getMinMaxFullWeights() const;
 
    //=========================================================================================================
    QPair<double, double> getMinMaxThresholdedWeights() const;
 
    //=========================================================================================================
    QPair<int, int> getMinMaxFullDegrees() const;
 
    //=========================================================================================================
    QPair<int, int> getMinMaxThresholdedDegrees() const;
 
    //=========================================================================================================
    QPair<int, int> getMinMaxFullIndegrees() const;
 
    //=========================================================================================================
    QPair<int, int> getMinMaxThresholdedIndegrees() const;
 
    //=========================================================================================================
    QPair<int, int> getMinMaxFullOutdegrees() const;
 
    //=========================================================================================================
    QPair<int, int> getMinMaxThresholdedOutdegrees() const;
 
    //=========================================================================================================
    void setThreshold(double dThreshold = 0.0);
 
    //=========================================================================================================
    double getThreshold();
 
    //=========================================================================================================
    void setFrequencyRange(float fLowerFreq, float fUpperFreq);
 
    //=========================================================================================================
    const QPair<float, float>& getFrequencyRange() const;
 
    //=========================================================================================================
    void append(QSharedPointer<NetworkEdge> newEdge);
 
    //=========================================================================================================
    void append(QSharedPointer<NetworkNode> newNode);
 
    //=========================================================================================================
    bool isEmpty() const;
 
    //=========================================================================================================
    void normalize();
 
    //=========================================================================================================
    VisualizationInfo getVisualizationInfo() const;
 
    //=========================================================================================================
    void setVisualizationInfo(const VisualizationInfo& visualizationInfo);
 
    //=========================================================================================================
    float getSamplingFrequency() const;
 
    //=========================================================================================================
    void setSamplingFrequency(float fSFreq);
 
    //=========================================================================================================
    int getUsedFreqBins() const;
 
    //=========================================================================================================
    void setUsedFreqBins(int iNumberFreqBins);
 
    //=========================================================================================================
    void setFFTSize(int iFFTSize);
 
    //=========================================================================================================
    int getFFTSize();
 
protected:
    QList<QSharedPointer<NetworkEdge>> m_lFullEdges;        
    QList<QSharedPointer<NetworkEdge>> m_lThresholdedEdges; 
 
    QList<QSharedPointer<NetworkNode>> m_lNodes; 
 
    Eigen::MatrixXd m_matDistMatrix; 
 
    QString m_sConnectivityMethod; 
 
    QPair<double, double> m_minMaxFullWeights;        
    QPair<double, double> m_minMaxThresholdedWeights; 
    QPair<float, float> m_minMaxFrequency;            
 
    double m_dThreshold;   
    float m_fSFreq;        
    int m_iNumberFreqBins; 
    int m_iFFTSize;        
 
    VisualizationInfo m_visualizationInfo; 
};
 
//=============================================================================================================
// INLINE DEFINITIONS
//=============================================================================================================
} // namespace CONNECTIVITYLIB
 
#ifndef metatype_networks
#define metatype_networks
Q_DECLARE_METATYPE(CONNECTIVITYLIB::Network);
#endif
 
#ifndef metatype_networkslist
#define metatype_networkslist
Q_DECLARE_METATYPE(QList<CONNECTIVITYLIB::Network>);
#endif
 
#ifndef metatype_networkssptr
#define metatype_networkssptr
Q_DECLARE_METATYPE(CONNECTIVITYLIB::Network::SPtr);
#endif
 
#ifndef metatype_networkssptrlist
#define metatype_networkssptrlist
Q_DECLARE_METATYPE(QList<CONNECTIVITYLIB::Network::SPtr>);
#endif
 
#endif // NETWORK_H