add RWSLFP recording

cleo/__init__.py

@@ -1,9 +1,8 @@
 """Contains core classes and functions for the Cleo package."""
 from __future__ import annotations
 
-import cleo.coords
-
 # auto-import submodules
+import cleo.coords
 import cleo.ephys
 import cleo.imaging
 import cleo.ioproc
@@ -17,6 +16,7 @@
     CLSimulator,
     InterfaceDevice,
     IOProcessor,
+    NeoExportable,
     Recorder,
     Stimulator,
     SynapseDevice,

cleo/coords.py

@@ -4,7 +4,7 @@
 
 from typing import Tuple
 
-from brian2 import Quantity, Unit, meter, mm, np
+from brian2 import Quantity, Subgroup, Unit, meter, mm, np
 from brian2.groups.group import Group
 from brian2.groups.neurongroup import NeuronGroup
 from brian2.units.fundamentalunits import get_dimensions
@@ -218,6 +218,15 @@ def _init_variables(group: Group):
         else:
             if type(group) == NeuronGroup:
                 modify_model_with_eqs(group, f"{dim_name}: meter")
+
+            elif isinstance(group, Subgroup):
+                if not hasattr(group.source, dim_name):
+                    modify_model_with_eqs(group.source, f"{dim_name}: meter")
+                group.variables.add_references(
+                    group.source, list(group.source.variables.keys())
+                )
+                assert dim_name in group.variables
+
             elif issubclass(type(group), Group):
                 group.variables.add_array(
                     dim_name,

cleo/ephys/__init__.py

@@ -1,13 +1,19 @@
 """Contains probes, convenience functions for generating electrode array coordinates,
 signals, spiking, and LFP"""
-from cleo.ephys.lfp import TKLFPSignal
-from cleo.ephys.spiking import MultiUnitSpiking, SortedSpiking, Spiking
+from cleo.ephys.lfp import (
+    LFPSignalBase,
+    RWSLFPSignalBase,
+    RWSLFPSignalFromPSCs,
+    RWSLFPSignalFromSpikes,
+    TKLFPSignal,
+)
 from cleo.ephys.probes import (
     Probe,
     Signal,
     linear_shank_coords,
-    tetrode_shank_coords,
     poly2_shank_coords,
     poly3_shank_coords,
+    tetrode_shank_coords,
     tile_coords,
 )
+from cleo.ephys.spiking import MultiUnitSpiking, SortedSpiking, Spiking