#pragma once
#include <knp/core/population.h>
#include <knp/core/projection.h>
#include <knp/core/messaging/messaging.h>
#include <knp/neuron-traits/all_traits.h>
#include <knp/synapse-traits/all_traits.h>
#include <functional>
#include <vector>

namespace knp_gpu_experiment
{
using Neuron = knp::neuron_traits::SynapticResourceSTDPAltAILIFNeuron;
using Synapse = knp::synapse_traits::SynapticResourceSTDPDeltaSynapse;
using Population = knp::core::Population<Neuron>;
using Projections = std::vector<std::reference_wrapper<knp::core::Projection<Synapse>>>;
using Reference = void (*)(Population&, Projections&, const knp::core::messaging::SpikeMessage&, knp::core::Step);
void train(Population&, Projections&, const knp::core::messaging::SpikeMessage&, knp::core::Step, Reference);
void flush();
void clear();
}