bids_edf_reader.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef BIDS_EDF_READER_H
#define BIDS_EDF_READER_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "bids_abstract_format_reader.h"
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QDateTime>
#include <QVector>
#include <QFile>
 
//=============================================================================================================
// DEFINE NAMESPACE BIDSLIB
//=============================================================================================================
 
namespace BIDSLIB
{
 
//=============================================================================================================
struct BIDSSHARED_EXPORT EDFChannelInfo
{
    int channelNumber{-1};
    QString label;
    QString transducerType;
    QString physicalDimension;
    QString prefiltering;
    float physicalMin{0.0f};
    float physicalMax{0.0f};
    long digitalMin{0};
    long digitalMax{0};
    long samplesPerRecord{0};
    long sampleCount{0};
    float frequency{0.0f};
    bool isMeasurement{false};
 
    FIFFLIB::FiffChInfo toFiffChInfo() const;

    float toSi() const;
};
 
//=============================================================================================================
class BIDSSHARED_EXPORT EDFReader : public AbstractFormatReader
{
public:
    //=========================================================================================================
    EDFReader();
 
    ~EDFReader() override;
 
    // AbstractFormatReader interface
    bool open(const QString& sFilePath) override;
    FIFFLIB::FiffInfo getInfo() const override;
    Eigen::MatrixXf readRawSegment(int iStartSampleIdx, int iEndSampleIdx) const override;
    long getSampleCount() const override;
    float getFrequency() const override;
    int getChannelCount() const override;
    FIFFLIB::FiffRawData toFiffRawData() const override;
    QString formatName() const override;
    bool supportsExtension(const QString& sExtension) const override;
 
    //=========================================================================================================
    QVector<EDFChannelInfo> getAllChannelInfos() const;
 
    //=========================================================================================================
    QVector<EDFChannelInfo> getMeasurementChannelInfos() const;
 
private:
    // EDF header field byte lengths
    enum EDFHeaderFieldLengths
    {
        EDF_VERSION = 8,
        LOCAL_PATIENT_INFO = 80,
        LOCAL_RECORD_INFO = 80,
        STARTDATE = 8,
        STARTTIME = 8,
        NUM_BYTES_HEADER = 8,
        HEADER_RESERVED = 44,
        NUM_DATA_RECORDS = 8,
        DURATION_DATA_RECS = 8,
        NUM_SIGNALS = 4,
        SIG_LABEL = 16,
        SIG_TRANSDUCER = 80,
        SIG_PHYS_DIM = 8,
        SIG_PHYS_MIN = 8,
        SIG_PHYS_MAX = 8,
        SIG_DIG_MIN = 8,
        SIG_DIG_MAX = 8,
        SIG_PREFILTERING = 80,
        SIG_NUM_SAMPLES = 8,
        SIG_RESERVED = 32,
    };
 
    void parseHeader(QIODevice* pDev);
 
    QString m_sFilePath;
 
    // Header data
    QString m_sVersionNo;
    QString m_sPatientId;
    QString m_sRecordingId;
    QDateTime m_startDateTime;
    int m_iNumBytesInHeader{0};
    int m_iNumDataRecords{0};
    float m_fDataRecordsDuration{0.0f};
    int m_iNumChannels{0};
    int m_iNumBytesPerDataRecord{0};
 
    QVector<EDFChannelInfo> m_vAllChannels;
    QVector<EDFChannelInfo> m_vMeasChannels;
 
    mutable QFile m_file;
    bool m_bIsOpen{false};
    int m_iBytesPerSample{2}; // 3 for BDF
};
 
} // namespace BIDSLIB
 
#endif // BIDS_EDF_READER_H