/home/breidi/Dropbox/signalserver/include/hardware/eeg_simulator.h

Go to the documentation of this file.

/*
    This file is part of the TOBI SignalServer.

    Commercial Usage
    Licensees holding valid Graz University of Technology Commercial
    licenses may use this file in accordance with the Graz University
    of Technology Commercial License Agreement provided with the
    Software or, alternatively, in accordance with the terms contained in
    a written agreement between you and Graz University of Technology.

    --------------------------------------------------

    GNU General Public License Usage
    Alternatively, this file may be used under the terms of the GNU
    General Public License version 3.0 as published by the Free Software
    Foundation and appearing in the file gpl.txt included in the
    packaging of this file.  Please review the following information to
    ensure the GNU General Public License version 3.0 requirements will be
    met: http://www.gnu.org/copyleft/gpl.html.

    In case of GNU General Public License Usage ,the TOBI SignalServer
    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 General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with the TOBI SignalServer. If not, see <http://www.gnu.org/licenses/>.

    Copyright 2010 Graz University of Technology
    Contact: SignalServer@tobi-project.org
*/

#ifndef EEG_SIMULATOR_H
#define EEG_SIMULATOR_H

#include "hardware/hw_thread.h"
#include "hardware/artificial_signal_source.h"
#include "hardware/hw_thread_builder.h"
#include "hardware/eeg_sim_msg_parser.h"

#include <boost/random/mersenne_twister.hpp>
#include <boost/random/normal_distribution.hpp>
#include <boost/random/variate_generator.hpp>

#include <map>

namespace tobiss
{

//-----------------------------------------------------------------------------

class EEGSimulator : public ArtificialSignalSource
{
  public:

//  static const unsigned int MESSAGE_VERSION = 1;

    EEGSimulator(boost::asio::io_service& io,
                 ticpp::Iterator<ticpp::Element> hw);
    virtual ~EEGSimulator();

//------------------------------------------

  private:
//    /**
//    * @struct EEGConfig
//    * @brief Simple struct holding parameters to configure the EEG signal.
//    */
//    struct EEGConfig
//    {
//      public:
//        double scaling_;
//        double offset_;
//    };

//    /**
//    * @struct SineConfig
//    * @brief Simple struct holding parameters to configure the sine waves.
//    */
//    struct SineConfig
//    {
//      public:
//        double freq_;
//        double amplitude_;
//        double phase_;
//    };

//    enum MessageType
//    {
//      Invalid,
//      GetConfig,
//      Config
//    };

    virtual void generateSignal();

    void setHardware(ticpp::Iterator<ticpp::Element>const &hw);
    virtual void setDeviceSettings(ticpp::Iterator<ticpp::Element>const &father);
    virtual void setChannelSettings(ticpp::Iterator<ticpp::Element>const &father);

    void setPort(ticpp::Iterator<ticpp::Element>const &elem);

    void setDeviceEEGConfig(ticpp::Iterator<ticpp::Element>const &elem);
    void setDeviceSineConfig(ticpp::Iterator<ticpp::Element>const &elem);

    void setChannelEEGConfig(ticpp::Iterator<ticpp::Element>const &father);
    void setChannelSineConfig(ticpp::Iterator<ticpp::Element>const &father);

    void checkEEGConfigAttributes(ticpp::Iterator<ticpp::Element>const &elem);
    void checkSineConfigAttributes(ticpp::Iterator<ticpp::Element>const &elem);

    EEGSimMsgParser::EEGConfig getEEGConfig(ticpp::Iterator<ticpp::Element>const &elem);
    EEGSimMsgParser::SineConfig getSineConfig(ticpp::Iterator<ticpp::Element>const &elem);

    void handleAsyncRead(const boost::system::error_code& ec, std::size_t bytes_transferred );

    void acceptHandler(const boost::system::error_code& error);

//    void checkMessage();
//    MessageType getMessageType();
//    void parseConfigMessage(std::map<boost::uint16_t, EEGConfig>& eeg,
//                      std::multimap<boost::uint16_t, SineConfig>& sine);

    void updateEEGConfig(std::map<boost::uint16_t, EEGSimMsgParser::EEGConfig>& eeg);
    void updateSineConfig(std::multimap<boost::uint16_t, EEGSimMsgParser::SineConfig>& sine);

//    std::string buildConfigString();
    void setSineConfigInMultimap(boost::uint16_t ch, EEGSimMsgParser::SineConfig config);

//------------------------------------------

  private:
    static const HWThreadBuilderTemplateRegistrator<EEGSimulator> factory_registrator_;
    boost::mt19937 twister_;
    boost::normal_distribution<> eeg_dist_;
    boost::variate_generator<boost::mt19937&, boost::normal_distribution<> > eeg_gen_;

    boost::asio::ip::tcp::acceptor acceptor_;
    boost::asio::ip::tcp::socket socket_;
    boost::uint32_t port_;
    std::string     peer_ip_;
    bool connected_;


    boost::asio::streambuf message_buffer_;
    std::string str_buffer_;

//    std::map<std::string, MessageType>            msg_types_map_;

    std::map<boost::uint16_t, EEGSimMsgParser::EEGConfig>          eeg_config_;       
    std::multimap<boost::uint16_t, EEGSimMsgParser::SineConfig>    sine_configs_;     

    EEGSimMsgParser                               parser_;

    //--------------------------------------
    // Constants:

    static const std::string xml_eeg_sim_port_;
    static const std::string xml_eeg_config_;
    static const std::string xml_sine_config_;
    static const std::string xml_scaling_;
    static const std::string xml_offset_;
    static const std::string xml_frequ_;
    static const std::string xml_amplitude_;
    static const std::string xml_phase_;

};

}  // tobiss




#endif // EEG_SIMULATOR_H