AdExGroup.h

Go to the documentation of this file.

/*
* Copyright 2014-2018 Ankur Sinha and 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 ADEXGROUP_H_
#define ADEXGROUP_H_

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


namespace auryn {

class AdExGroup : public NeuronGroup
{
private:
    AurynFloat e_rest, e_reset, e_rev_gaba, e_rev_ampa,e_thr, g_leak, c_mem, deltat;
    AurynFloat tau_ampa, tau_gaba, tau_mem;
    AurynFloat scale_ampa, scale_gaba, scale_mem, scale_w, scale_current;
    AurynFloat * t_w;
    AurynFloat a; 
        AurynFloat b; 
        AurynFloat tau_w; 
    unsigned short refractory_time;
    auryn_vector_ushort * ref;

    AurynStateVector * w;

    AurynStateVector * I_exc;
    AurynStateVector * I_inh;
    AurynStateVector * I_leak;
    AurynStateVector * temp;

    AurynFloat * t_g_ampa;
    AurynFloat * t_g_gaba;
    AurynFloat * t_mem;
    unsigned short * t_ref;

    void init();
    void calculate_scale_constants();
    inline void integrate_state();
    inline void check_thresholds();
    virtual std::string get_output_line(NeuronID i);
    virtual void load_input_line(NeuronID i, const char * buf);

        void virtual_serialize(boost::archive::binary_oarchive & ar, const unsigned int version );
        void virtual_serialize(boost::archive::binary_iarchive & ar, const unsigned int version );
public:
    AdExGroup(NeuronID size);
    virtual ~AdExGroup();

    void set_refractory_period(AurynDouble t);

    void set_delta_t(AurynFloat d);

    void set_g_leak(AurynFloat g);

    void set_c_mem(AurynFloat cm);

    void set_a(AurynFloat _a);

    void set_b(AurynFloat _b);

    void set_e_reset(AurynFloat ereset);

    void set_e_rest(AurynFloat erest);

    void set_e_thr(AurynFloat ethr);

    void set_tau_w(AurynFloat tauw);


    void set_tau_ampa(AurynFloat tau);

    AurynFloat get_tau_ampa();

    void set_tau_gaba(AurynFloat tau);

    AurynFloat get_tau_gaba();

    void clear();

    void evolve();
};

}

#endif /*ADEXGROUP_H_*/