WEEG2.cpp

//---------------------------------------------------------------------------
//
// Project: OpenWalnut ( http://www.openwalnut.org )
//
// Copyright 2009 OpenWalnut Community, BSV@Uni-Leipzig and CNCF@MPI-CBS
// For more information see http://www.openwalnut.org/copying
//
// This file is part of OpenWalnut.
//
// OpenWalnut 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.
//
// OpenWalnut 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 OpenWalnut. If not, see <http://www.gnu.org/licenses/>.
//
//---------------------------------------------------------------------------
 
#include <cstddef>
#include <memory>
#include <sstream>
#include <string>
#include <vector>
 
 
#include "../common/exceptions/WOutOfBounds.h"
#include "WEEG2.h"
#include "WEEG2Segment.h"
#include "WEEGChannelInfo.h"
#include "WEEGPositionsLibrary.h"
#include "exceptions/WDHException.h"
#include "io/WPagerEEG.h"
 
 
std::shared_ptr< WPrototyped > WEEG2::m_prototype = std::shared_ptr< WPrototyped >();
 
WEEG2::WEEG2( std::shared_ptr< WPagerEEG > pager, std::shared_ptr< WEEGPositionsLibrary > positionsLibrary )
{
    if( !pager )
    {
        throw WDHException( std::string( "Couldn't construct new EEG: pager invalid" ) );
    }
 
    if( !positionsLibrary )
    {
        throw WDHException( std::string( "Couldn't construct new EEG: positions library invalid" ) );
    }
 
    std::size_t nbSegments = pager->getNumberOfSegments();
    if( nbSegments <= 0 || WRecording::MAX_RECORDING_SEGMENTS < nbSegments )
    {
        throw WDHException( std::string( "Couldn't construct new EEG: invalid number of segments" ) );
    }
 
    std::size_t nbChannels = pager->getNumberOfChannels();
    if( nbChannels <= 0 || WRecording::MAX_RECORDING_CHANNELS < nbChannels )
    {
        throw WDHException( std::string( "Couldn't construct new EEG: invalid number of channels" ) );
    }
 
    m_samplingRate = pager->getSamplingRate();
    if( m_samplingRate <= 0.0 || WRecording::MAX_RECORDING_SAMPLING_FREQUENCY < m_samplingRate )
    {
        throw WDHException( std::string( "Couldn't construct new EEG: invalid sampling rate" ) );
    }
 
    setFilename( pager->getFilename() );
 
    m_segments.reserve( nbSegments );
    for( std::size_t segmentID = 0; segmentID < nbSegments; ++segmentID )
    {
        m_segments.push_back( std::shared_ptr< WEEG2Segment >( new WEEG2Segment( segmentID, pager ) ) );
    }
 
    m_channelInfos.reserve( nbChannels );
    for( std::size_t channelID = 0; channelID < nbChannels; ++channelID )
    {
        m_channelInfos.push_back( std::shared_ptr< WEEGChannelInfo >( new WEEGChannelInfo( channelID, pager, positionsLibrary ) ) );
    }
 
    m_infoProperties->addProperty( "Segments", "The number of segments", static_cast< WPVBaseTypes::PV_INT >( nbSegments ) );
    m_infoProperties->addProperty( "Channels", "The number of channels", static_cast< WPVBaseTypes::PV_INT >( nbChannels ) );
    m_infoProperties->addProperty( "Sampling rate", "Sampling rate of the recording", static_cast< WPVBaseTypes::PV_DOUBLE >( m_samplingRate ) );
}
 
WEEG2::WEEG2()
{
}
 
std::size_t WEEG2::getNumberOfSegments() const
{
    return m_segments.size();
}
 
std::size_t WEEG2::getNumberOfChannels() const
{
    return m_channelInfos.size();
}
 
double WEEG2::getSamplingRate() const
{
    return m_samplingRate;
}
 
std::shared_ptr< WEEG2Segment > WEEG2::getSegment( std::size_t segmentID ) const
{
    if( segmentID >= m_segments.size() )
    {
        std::ostringstream stream;
        stream << "The EEG has no segment number " << segmentID;
        throw WOutOfBounds( stream.str() );
    }
 
    return m_segments[segmentID];
}
 
std::shared_ptr< WEEGChannelInfo > WEEG2::getChannelInfo( std::size_t channelID ) const
{
    if( channelID >= m_channelInfos.size() )
    {
        std::ostringstream stream;
        stream << "The EEG has no channel number " << channelID;
        throw WOutOfBounds( stream.str() );
    }
 
    return m_channelInfos[channelID];
}
 
const std::string WEEG2::getName() const
{
    return "WEEG2";
}
 
const std::string WEEG2::getDescription() const
{
    return "Contains EEG data";
}
 
std::shared_ptr< WPrototyped > WEEG2::getPrototype()
{
    if( !m_prototype )
    {
        m_prototype = std::shared_ptr< WPrototyped >( new WEEG2() );
    }
 
    return m_prototype;
}