Page MenuHomec4science
Files F65521830

plot_QSS.py
No OneTemporary
Actions

File Metadata

Created
Tue, Jun 4, 09:44

plot_QSS.py
View Options

# -*- coding: utf-8 -*-
# @Author: Theo Lemaire
# @Date: 2018-09-28 16:13:34
# @Last Modified by: Theo Lemaire
# @Last Modified time: 2019-06-09 15:56:25
''' Phase-plane analysis of neuron behavior under quasi-steady state approximation. '''
import os
import numpy as np
import matplotlib.pyplot as plt
from PySONIC.utils import logger
from PySONIC.plt import plotQSSdynamics, plotQSSVarVsQm, plotEqChargeVsAmp, plotQSSThresholdCurve
from PySONIC.parsers import AStimParser
def main():
# Parse command line arguments
parser = AStimParser()
parser.addCmap(default='viridis')
parser.addAscale()
parser.addSave()
parser.outputdir_dep_key = 'save'
parser.addCompare(desc='Compare with simulations')
parser.addInputDir(dep_key='compare')
parser.defaults['amp'] = np.logspace(np.log10(1), np.log10(600), 100) # kPa
parser.defaults['tstim'] = 1000. # ms
parser.defaults['toffset'] = 0. # ms
args = parser.parse()
args['inputdir'] = parser.parseInputDir(args)
logger.setLevel(args['loglevel'])
if args['plot'] is None:
args['plot'] = ['dQdt']
a, Fdrive, tstim, toffset, PRF = [
args[k][0] for k in ['radius', 'freq', 'tstim', 'toffset', 'PRF']]
figs = []
for i, neuron in enumerate(args['neuron']):
if args['DC'].size == 1:
DC = args['DC'][0]
if args['amp'].size == 1:
Adrive = args['amp'][0]
figs.append(plotQSSdynamics(neuron, a, Fdrive, Adrive, DC))
else:
# Plot evolution of QSS vars vs Q for different amplitudes
for pvar in args['plot']:
figs.append(plotQSSVarVsQm(
neuron, a, Fdrive, pvar, amps=args['amp'], DC=DC,
cmap=args['cmap'], zscale=args['Ascale']))
# Plot equilibrium charge as a function of amplitude
if 'dQdt' in args['plot']:
figs.append(plotEqChargeVsAmp(
neuron, a, Fdrive, amps=args['amp'], tstim=tstim, toffset=toffset, PRF=PRF,
DC=DC, xscale=args['Ascale'], compdir=args['inputdir'], mpi=args['mpi'],
loglevel=args['loglevel']))
else:
figs.append(plotQSSThresholdCurve(
neuron, a, Fdrive, tstim=tstim, toffset=toffset, PRF=PRF, DCs=args['DC'],
Ascale=args['Ascale'], comp=args['compare'], mpi=args['mpi'],
loglevel=args['loglevel']))
if args['save']:
for fig in figs:
s = fig.canvas.get_window_title()
s = s.replace('(', '- ').replace('/', '_').replace(')', '')
figname = '{}.png'.format(s)
fig.savefig(os.path.join(args['outputdir'], figname), transparent=True)
else:
plt.show()
if __name__ == '__main__':
main()

Event Timeline

c4science · Help