'''
Brian2GeNN defines two different types of code objects, `GeNNCodeObject` and `GeNNUserCodeObject`.
`GeNNCodeObject` is the class of code objects that produce code snippets for GeNN neuron or synapse models.
`GeNNUserCodeObject` is the class of code objects that produce C++ code which is used as "user-side" code in GeNN. The class derives directly from Brian 2's `CPPStandaloneCodeObject`, using the `CPPCodeGenerator`.
'''
from brian2.devices.cpp_standalone.codeobject import CPPStandaloneCodeObject, openmp_pragma, constant_or_scalar
from brian2.codegen.generators.cpp_generator import c_data_type, CPPCodeGenerator
from brian2.codegen.codeobject import CodeObject
from brian2.codegen.targets import codegen_targets
from brian2.codegen.templates import Templater
from .genn_generator import *
__all__ = ['GeNNCodeObject',
'GeNNUserCodeObject']
[docs]class GeNNCodeObject(CodeObject):
'''
Class of code objects that generate GeNN "code snippets"
'''
templater = Templater('brian2genn', '.cpp',
env_globals={'c_data_type': c_data_type,
'openmp_pragma': openmp_pragma,
'constant_or_scalar': constant_or_scalar,
'zip': zip})
generator_class = GeNNCodeGenerator
# Overwrite Brian2 methods, we do not need to compile/run anything here
[docs] def compile(self):
pass
[docs] def before_run(self):
pass
[docs] def after_run(self):
pass
[docs]class GeNNUserCodeObject(CPPStandaloneCodeObject):
'''
Class of code objects that generate GeNN "user code"
'''
templater = CPPStandaloneCodeObject.templater.derive('brian2genn')
#, env_globals={'c_data_type': c_data_type,
# 'openmp_pragma': openmp_pragma,
# 'constant_or_scalar': constant_or_scalar})
generator_class = CPPCodeGenerator
codegen_targets.add(GeNNCodeObject)
codegen_targets.add(GeNNUserCodeObject)