Improve get_all_sections

jonescompneurolab · Sep 1, 2020 · fd22ee6 · fd22ee6
1 parent 15bbb66
commit fd22ee6
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/hnn_core/lfp.py b/hnn_core/lfp.py
@@ -32,9 +32,9 @@
 
 
 # get all Sections
-def getallSections(ty='Pyr'):
+def get_all_sections(sec_type='Pyr'):
     ls = h.allsec()
-    ls = [s for s in ls if s.name().count(ty) > 0 or len(ty) == 0]
+    ls = [s for s in ls if sec_type in s.name()]
     return ls
 
 
@@ -111,7 +111,7 @@ def transfer_resistance(self, exyz, method):
         from numpy.linalg import norm
 
         vres = h.Vector()
-        lsec = getallSections()
+        lsec = get_all_sections()
         sigma = self.sigma
 
         exyz = np.array(exyz)  # electrode position
@@ -130,7 +130,7 @@ def transfer_resistance(self, exyz, method):
                 # setting radius limit
                 if dis < s.diam / 2.0:
                     dis = s.diam / 2.0 + 0.1
-                
+
                 phi = 1. / dis
 
             elif method == 'lsa':
@@ -173,7 +173,7 @@ def transfer_resistance(self, exyz, method):
         return vres
 
     def LFPinit(self):
-        lsec = getallSections()
+        lsec = get_all_sections()
         n_sections = len(lsec)
 
         self.imem_ptrvec = h.PtrVector(n_sections)
@@ -227,7 +227,7 @@ def callback(self):
     h.tstop = 2000.0
 
     elec = LFPElectrode([0, 100.0, 100.0], pc=h.ParallelContext(),
-                        method='psa')
+                        method='lsa')
     elec.setup()
     elec.LFPinit()
     h.run()