|
11 | 11 | from . import utility_functions as utils |
12 | 12 | from . import basis_functions as basis |
13 | 13 |
|
| 14 | + |
14 | 15 | class CSD(object): |
15 | 16 | """CSD - The base class for CSD methods.""" |
16 | 17 | def __init__(self, ele_pos, pots): |
@@ -602,6 +603,7 @@ def int_pot_1D(self, xp, x, R, h, basis_func): |
602 | 603 | m *= basis_func(abs(xp), R) # xp is the distance |
603 | 604 | return m |
604 | 605 |
|
| 606 | + |
605 | 607 | class KCSD2D(KCSD): |
606 | 608 | """The 2D variant for the Kernel Current Source Density method. |
607 | 609 |
|
@@ -1127,6 +1129,7 @@ def int_pot_3D(self, xp, yp, zp, x, R, h, basis_func): |
1127 | 1129 | pot *= basis_func(dist, R) |
1128 | 1130 | return pot |
1129 | 1131 |
|
| 1132 | + |
1130 | 1133 | class oKCSD1D(KCSD1D): |
1131 | 1134 | """oKCSD - The variant for the Kernel Current Source Density method that |
1132 | 1135 | allows to reconstruct potential and CSD in given 1D space points. |
@@ -1220,7 +1223,8 @@ def place_basis(self): |
1220 | 1223 | self.src_x = self.own_src |
1221 | 1224 | self.n_src = self.src_x.size |
1222 | 1225 | self.nsx = self.src_x.shape |
1223 | | - |
| 1226 | + |
| 1227 | + |
1224 | 1228 | class oKCSD2D(KCSD2D): |
1225 | 1229 | """oKCSD - The variant for the Kernel Current Source Density method that |
1226 | 1230 | allows to reconstruct potential and CSD in given 2D space points. |
@@ -1314,6 +1318,7 @@ def place_basis(self): |
1314 | 1318 | self.n_src = self.src_x.size |
1315 | 1319 | self.nsx, self.nsy = self.src_x.shape |
1316 | 1320 |
|
| 1321 | + |
1317 | 1322 | class oKCSD3D(KCSD3D): |
1318 | 1323 | """oKCSD - The variant for the Kernel Current Source Density method that |
1319 | 1324 | allows to reconstruct potential and CSD in given 3D space points. |
@@ -1407,6 +1412,7 @@ def place_basis(self): |
1407 | 1412 | self.n_src = self.src_x.size |
1408 | 1413 | self.nsx, self.nsy, self.nsz = self.src_x.shape |
1409 | 1414 |
|
| 1415 | + |
1410 | 1416 | if __name__ == '__main__': |
1411 | 1417 | print('Checking 1D') |
1412 | 1418 | ele_pos = np.array(([-0.1], [0], [0.5], [1.], [1.4], [2.], [2.3])) |
|
0 commit comments