1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
{# USES_VARIABLES { N, _clock_t, count,
_source_start, _source_stop} #}
{# WRITES_TO_READ_ONLY_VARIABLES { N, count } #}
{% extends 'common_group.cpp' %}
{% block maincode %}
//// MAIN CODE ////////////
{# Get the name of the array that stores these events (e.g. the spikespace array) #}
{% set _eventspace = get_array_name(eventspace_variable) %}
int32_t _num_events = {{_eventspace}}[_num{{eventspace_variable.name}}-1];
if (_num_events > 0)
{
size_t _start_idx = _num_events;
size_t _end_idx = _num_events;
for(size_t _j=0; _j<_num_events; _j++)
{
const int _idx = {{_eventspace}}[_j];
if (_idx >= _source_start) {
_start_idx = _j;
break;
}
}
for(size_t _j=_num_events-1; _j>=_start_idx; _j--)
{
const int _idx = {{_eventspace}}[_j];
if (_idx < _source_stop) {
break;
}
_end_idx = _j;
}
_num_events = _end_idx - _start_idx;
if (_num_events > 0) {
const size_t _vectorisation_idx = 1;
{{scalar_code|autoindent}}
for(size_t _j=_start_idx; _j<_end_idx; _j++)
{
const size_t _idx = {{_eventspace}}[_j];
const size_t _vectorisation_idx = _idx;
{{vector_code|autoindent}}
{% for varname, var in record_variables | dictsort %}
{{get_array_name(var, access_data=False)}}.push_back(_to_record_{{varname}});
{% endfor %}
{{count}}[_idx-_source_start]++;
}
{{N}} += _num_events;
}
}
{% endblock %}
{% block extra_functions_cpp %}
void _debugmsg_{{codeobj_name}}()
{
using namespace brian;
{# We need the pointers and constants here to get the access to N working #}
%CONSTANTS%
{{pointers_lines|autoindent}}
std::cout << "Number of spikes: " << {{N}} << endl;
}
{% endblock %}
{% block extra_functions_h %}
void _debugmsg_{{codeobj_name}}();
{% endblock %}
{% macro main_finalise() %}
#ifdef DEBUG
_debugmsg_{{codeobj_name}}();
#endif
{% endmacro %}
|
