IF.h

Go to the documentation of this file.

 
#ifndef SPIKE_IF_H
#define SPIKE_IF_H
 
#include <cassert>
#include <cmath>
#include <ostream>
#include <random>
#include <vector>
 
#include "Spike/Neuron/Neuron.h"
#include "Spike/Signal/CosineSignal.h"
#include "Spike/Signal/Signal.h"
#include "Spike/SpikeTrain/SpikeTrain.h"
#include "Spike/TimeFrame/TimeFrame.h"
 
namespace Spike {
 
class IF : public Neuron {
protected:
  double mu;                             
  double D;                              
  std::random_device rd;                 
  std::mt19937 generator;                
  std::normal_distribution<double> dist; 
 
public:
  explicit IF(const std::string &input_file);
 
  IF(double mu, double D);
 
  [[nodiscard]] virtual double drift(double v) const = 0;
 
  [[nodiscard]] double diffusion() const;
 
  void get_spikes(SpikeTrain &spike_train) override;
 
  void get_spikes(Signal &signal, SpikeTrain &spike_train) override;
 
  void get_voltage_curve(const TimeFrame &time, std::vector<double> &v);
 
  void set_mu(double mu_new) { this->mu = mu_new; };
 
  void set_D(double D_new) { this->D = D_new; };
 
  [[nodiscard]] double get_D() const { return D; };
 
  virtual void print(std::ostream &out) const = 0;
 
  friend std::ostream &operator<<(std::ostream &out, const IF &neuron);
};
 
class PIF : public IF {
public:
  PIF(double mu, double D);
 
  explicit PIF(const std::string &input_file);
 
  [[nodiscard]] double drift(double v) const override;
 
  void print(std::ostream &out) const override {
    out << "PIF(mu: " << mu << ", D: " << D << ")";
  }
};
 
class LIF : public IF {
public:
  LIF(double mu, double D);
 
  explicit LIF(const std::string &input_file);
 
  [[nodiscard]] double drift(double v) const override;
 
  void print(std::ostream &out) const override {
    out << "LIF(mu: " << mu << ", D: " << D << ")";
  }
};
 
} // namespace Spike
 
#endif // SPIKE_IF_H