# -*- coding: utf-8 -*-
"""
This module implements leakage channels.
"""
from typing import Union, Callable
import brainpy.math as bm
from brainpy._src.initialize import Initializer, parameter
from brainpy.types import ArrayType, Shape
from .base import LeakyChannel
__all__ = [
'IL',
'IKL',
]
[docs]class IL(LeakyChannel):
"""The leakage channel current.
Parameters
----------
g_max : float
The leakage conductance.
E : float
The reversal potential.
"""
[docs] def __init__(
self,
size,
keep_size: bool = False,
g_max: Union[int, float, ArrayType, Initializer, Callable] = 0.1,
E: Union[int, float, ArrayType, Initializer, Callable] = -70.,
method: str = None,
name: str = None,
mode: bm.Mode = None,
):
super(IL, self).__init__(size,
keep_size=keep_size,
name=name,
mode=mode)
self.E = parameter(E, self.varshape, allow_none=False)
self.g_max = parameter(g_max, self.varshape, allow_none=False)
self.method = method
def reset_state(self, V, batch_size=None):
pass
def update(self, tdi, V):
pass
def current(self, V):
return self.g_max * (self.E - V)
[docs]class IKL(IL):
"""The potassium leak channel current.
Parameters
----------
g_max : float
The potassium leakage conductance which is modulated by both
acetylcholine and norepinephrine.
E : float
The reversal potential.
"""
[docs] def __init__(
self,
size: Shape,
keep_size: bool = False,
g_max: Union[int, float, ArrayType, Initializer, Callable] = 0.005,
E: Union[int, float, ArrayType, Initializer, Callable] = -90.,
method: str = None,
name: str = None,
mode: bm.Mode = None,
):
super(IKL, self).__init__(size=size,
keep_size=keep_size,
g_max=g_max,
E=E,
method=method,
name=name,
mode=mode)
