add NEST code generator helper as workaround for module reloading

pynestml/codegeneration/nest_code_generator_utils.py

@@ -19,6 +19,12 @@
 # You should have received a copy of the GNU General Public License
 # along with NEST.  If not, see <http://www.gnu.org/licenses/>.
 
+from typing import List, Optional
+
+import re
+import uuid
+
+from pynestml.frontend.pynestml_frontend import generate_nest_target
 from pynestml.symbols.variable_symbol import VariableSymbol
 from pynestml.symbols.variable_symbol import BlockType
 
@@ -51,3 +57,82 @@ def print_symbol_origin(cls, variable_symbol: VariableSymbol, numerical_state_sy
             return 'B_.%s'
 
         return ''
+
+    @classmethod
+    def generate_code_for(cls,
+                          nestml_neuron_model: str,
+                          nestml_synapse_model: Optional[str] = None,
+                          post_ports: Optional[List[str]] = None,
+                          mod_ports: Optional[List[str]] = None,
+                          uniq_id: Optional[str] = None,
+                          logging_level: str = "WARNING"):
+        """Generate code for a given neuron and synapse model, passed as a string.
+
+        NEST cannot yet unload or reload modules. This function implements a workaround using UUIDs to generate unique names.
+
+        The neuron and synapse models can be passed directly as strings in NESTML syntax, or as filenames, in which case the NESTML model is loaded from the given filename.
+
+        Returns
+        -------
+        If a synapse is specified, returns a tuple (module_name, mangled_neuron_name, mangled_synapse_name) containing the names that can be used in ``nest.Install()``, ``nest.Create()`` and ``nest.Connect()`` calls. If no synapse is specified, returns a tuple (module_name, mangled_neuron_name).
+        """
+
+        # generate unique ID
+        if uniq_id is None:
+            uniq_id = str(uuid.uuid4().hex)
+
+        # read neuron model from file?
+        if not "\n" in nestml_neuron_model and ".nestml" in nestml_neuron_model:
+            with open(nestml_neuron_model, "r") as nestml_model_file:
+                nestml_neuron_model = nestml_model_file.read()
+
+        # update neuron model name inside the file
+        neuron_model_name_orig = re.findall(r"neuron\ [^:\s]*:", nestml_neuron_model)[0][7:-1]
+        neuron_model_name_uniq = neuron_model_name_orig + uniq_id
+        nestml_model = re.sub(r"neuron\ [^:\s]*:",
+                              "neuron " + neuron_model_name_uniq + ":", nestml_neuron_model)
+        neuron_uniq_fn = neuron_model_name_uniq + ".nestml"
+        with open(neuron_uniq_fn, "w") as f:
+            print(nestml_model, file=f)
+
+        if nestml_synapse_model:
+            # read synapse model from file?
+            if not "\n" in nestml_synapse_model and ".nestml" in nestml_synapse_model:
+                with open(nestml_synapse_model, "r") as nestml_model_file:
+                    nestml_synapse_model = nestml_model_file.read()
+
+            # update synapse model name inside the file
+            synapse_model_name_orig = re.findall(r"synapse\ [^:\s]*:", nestml_synapse_model)[0][8:-1]
+            synapse_model_name_uniq = synapse_model_name_orig + uniq_id
+            nestml_model = re.sub(r"synapse\ [^:\s]*:",
+                                  "synapse " + synapse_model_name_uniq + ":", nestml_synapse_model)
+            synapse_uniq_fn = synapse_model_name_uniq + ".nestml"
+            with open(synapse_uniq_fn, "w") as f:
+                print(nestml_model, file=f)
+
+        # generate the code for neuron and optionally synapse
+        module_name = "nestml_" + uniq_id + "_module"
+        input_fns = [neuron_uniq_fn]
+        codegen_opts = {"neuron_parent_class": "StructuralPlasticityNode",
+                        "neuron_parent_class_include": "structural_plasticity_node.h"}
+
+        mangled_neuron_name = neuron_model_name_uniq
+        if nestml_synapse_model:
+            input_fns += [synapse_uniq_fn]
+            codegen_opts["neuron_synapse_pairs"] = [{"neuron": neuron_model_name_uniq,
+                                                     "synapse": synapse_model_name_uniq,
+                                                     "post_ports": post_ports,
+                                                     "vt_ports": mod_ports}]
+            mangled_neuron_name = neuron_model_name_uniq + "_nestml__with_" + synapse_model_name_uniq + "_nestml"
+            mangled_synapse_name = synapse_model_name_uniq + "_nestml__with_" + neuron_model_name_uniq + "_nestml"
+
+        generate_nest_target(input_path=input_fns,
+                             logging_level=logging_level,
+                             module_name=module_name,
+                             suffix="_nestml",
+                             codegen_opts=codegen_opts)
+
+        if nestml_synapse_model:
+            return module_name, mangled_neuron_name, mangled_synapse_name
+        else:
+            return module_name, mangled_neuron_name