Reader.h

//
// This file is part of libGDF.
//
// libGDF is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as
// published by the Free Software Foundation, either version 3 of
// the License, or (at your option) any later version.
//
// libGDF is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with libGDF.  If not, see <http://www.gnu.org/licenses/>.
//
// Copyright 2010 Martin Billinger

#ifndef __READER_H_INCLUDED__
#define __READER_H_INCLUDED__

#include "GDF/Record.h"
#include "GDF/EventHeader.h"
#include "GDF/GDFHeaderAccess.h"
#include "GDF/Types.h"
#include "GDF/tools.h"
#include <vector>
#include <string>
#include <fstream>

namespace gdf
{

    class Reader
    {
    public:
        Reader( );

        virtual ~Reader( );

        void open( const std::string filename );

        void close( );

        void enableCache( bool b );

        virtual void initCache( );

        virtual void resetCache( );

        size_t findRecord( uint16 channel_idx, size_t sample_idx );


        void getSignals( std::vector< std::vector<double> > &buffer, double start_time = 0, double end_time = -1, std::vector<uint16> signal_indices = std::vector<uint16>() );


        void getSignal( uint16 channel_idx, double *buffer, size_t start = 0, size_t end = 0 );


        double getSample( uint16 channel_idx, size_t sample_idx );

        Record *getRecordPtr( size_t index );

        void readRecord( size_t index, Record *rec );

        void precacheRecords( size_t start, size_t end );

        EventHeader *getEventHeader( );

        const GDFHeaderAccess &getHeaderAccess_readonly( ) const { return m_header; }

        const MainHeader &getMainHeader_readonly( ) const { return m_header.getMainHeader_readonly( ); }

        const SignalHeader &getSignalHeader_readonly( size_t idx ) const { return m_header.getSignalHeader_readonly(idx); }

    protected:
        void readEvents( );

        std::string m_filename;
        GDFHeaderAccess m_header;
        EventHeader *m_events;
        std::vector< Record* > m_record_cache;
        Record* m_record_nocache;
        std::list<size_t> m_cache_entries;
        std::ifstream m_file;
        bool m_cache_enabled;

        size_t m_record_length; 
        size_t m_record_offset; 
        size_t m_event_offset;  
    };
}

#endif // __READER_H_INCLUDED__