InputChannelGroup.h

Go to the documentation of this file.

/* 
* Copyright 2014-2018 Friedemann Zenke
*
* This file is part of Auryn, a simulation package for plastic
* spiking neural networks.
* 
* Auryn is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
* 
* Auryn 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 Auryn.  If not, see <http://www.gnu.org/licenses/>.
*
* If you are using Auryn or parts of it for your work please cite:
* Zenke, F. and Gerstner, W., 2014. Limits to high-speed simulations 
* of spiking neural networks using general-purpose computers. 
* Front Neuroinform 8, 76. doi: 10.3389/fninf.2014.00076
*/

#ifndef INPUTCHANNELGROUP_H_
#define INPUTCHANNELGROUP_H_

#include "auryn_definitions.h"
#include "AurynVector.h"
#include "System.h"
#include "SpikingGroup.h"

#include <boost/random/mersenne_twister.hpp>
#include <boost/random/uniform_int.hpp>
#include <boost/random/variate_generator.hpp>
#include <boost/random/exponential_distribution.hpp>

namespace auryn {

class InputChannelGroup : public SpikingGroup
{
private:
        AurynTime * clk;
        static boost::mt19937 shared_noise_gen; 
        static boost::mt19937 rank_noise_gen; 

        // boost::uniform_01<> * dist;
        boost::variate_generator<boost::mt19937&, boost::uniform_01<> > * shared_noise;
        boost::variate_generator<boost::mt19937&, boost::exponential_distribution<> > * rank_noise;

        void init(AurynDouble rate, NeuronID chsize );

protected:
        AurynDouble lambda;

        AurynTime tstop;

        NeuronID channelsize;
        NeuronID nb_channels;

        AurynDouble timescale;
        int offset;
        AurynTime delay;

        AurynDouble * o;

        NeuronID * x;
        
public:
        AurynDouble * amplitudes; 
        AurynDouble * means;

        InputChannelGroup(NeuronID n, 
                        AurynDouble rate=5., 
                        NeuronID chsize=100
                        );
        virtual ~InputChannelGroup();
        virtual void evolve();
        void set_rate(AurynDouble rate);
        void set_timescale(AurynDouble scale);
        void set_offset(int off);

        void set_means(AurynDouble m);
        void set_mean(NeuronID channel, AurynDouble m);

        void set_amplitudes(AurynDouble amp);
        void set_amplitude(NeuronID channel, AurynDouble amp);

        void set_stoptime(AurynDouble stoptime);
        AurynDouble get_rate();
        void seed();
};

}

#endif /*INPUTCHANNELGROUP_H_*/