#cp calcifcurves_dendstim.py strongdendstim.py
#cp ../haymod3e/runcontrol_check.py runcontrol.py
# runcontrols
# A script for determining the control neuron F-I curve and limit cycle.
#
# The input code for the hoc-interface is based on BAC_firing.hoc by Etay Hay (2011)
#
# Tuomo Maki-Marttunen, Oct 2014
# (CC BY)

from neuron import h
import mytools
import pickle
import numpy as np
import sys
from pylab import *
from os.path import exists

spikfreqsAll = []
timescAll = []
VsomacAll = []
VDerivcAll = []
VDcoeffAll = []
VdendcAll = []
VdDcoeffAll = []
VdDerivcAll = []
CasomacAll = []
CaDerivcAll = []
CaDcoeffAll = []
CadendcAll = []
CadDerivcAll = []
CadDcoeffAll = []
times_controlAll = []
Vsoma_controlAll = []
Vdend_controlAll = []
Casoma_controlAll = []
Cadend_controlAll = []

Ihcoeff = 1.0
denddist = 620
basalcond = 0.05
basalpoint = 100.0
conde = 0.0
if len(sys.argv) > 1:
  Ihcoeff = float(sys.argv[1])
if len(sys.argv) > 2:
  denddist = float(sys.argv[2])
if len(sys.argv) > 3:
  basalcond = float(sys.argv[3])
if len(sys.argv) > 4:
  basalpoint = float(sys.argv[4])
if len(sys.argv) > 5:
  conde = float(sys.argv[5])


I0 = 0.0
I1 = 100.0

for icell in range(0,1):
  morphology_file = "morphologies/cell"+str(icell+1)+".asc"
  biophys_file = "models/L5PCbiophys3.hoc"
  template_file = "models/L5PCtemplate.hoc"
  v0 = -80
  ca0 = 0.0001

  proximalpoint = 400
  distalpoint = denddist
  BACdt = 5.0

  h("""
load_file("stdlib.hoc")
load_file("stdrun.hoc")
objref cvode
cvode = new CVode()
cvode.active(1)
cvode.atol(0.0002)
load_file("import3d.hoc")
objref L5PC
load_file(\""""+biophys_file+"""\")
load_file(\""""+template_file+"""\")
L5PC = new L5PCtemplate(\""""+morphology_file+"""\")
access L5PC.soma
objref st1, st2

objref vsoma, vdend, recSite, vdend2, isoma, cadend, cadend2, casoma, ihsoma, ihdend
vsoma = new Vector()
casoma = new Vector()
vdend = new Vector()
cadend = new Vector()
vdend2 = new Vector()
cadend2 = new Vector()
ihsoma = new Vector()
ihdend = new Vector()
objref sl,ns,tvec
tvec = new Vector()
sl = new List()
double siteVec[2]
sl = L5PC.locateSites("apic","""+str(distalpoint)+""")
maxdiam = 0
for(i=0;i<sl.count();i+=1){
  dd1 = sl.o[i].x[1]
  dd = L5PC.apic[sl.o[i].x[0]].diam(dd1)
  if (dd > maxdiam) {
    j = i
    maxdiam = dd
  }
}
siteVec[0] = sl.o[j].x[0]
siteVec[1] = sl.o[j].x[1]
print "distalpoint gCa_HVA: ", L5PC.apic[siteVec[0]].gCa_HVAbar_Ca_HVA
print "distalpoint gCa_LVA: ", L5PC.apic[siteVec[0]].gCa_LVAstbar_Ca_LVAst
L5PC.apic[siteVec[0]] cvode.record(&v(siteVec[1]),vdend,tvec)
L5PC.apic[siteVec[0]] cvode.record(&cai(siteVec[1]),cadend,tvec)
L5PC.apic[siteVec[0]] cvode.record(&ihcn_Ih(0.5),ihdend,tvec)
L5PC.soma cvode.record(&v(0.5),vsoma,tvec)
L5PC.soma cvode.record(&cai(0.5),casoma,tvec)
L5PC.soma cvode.record(&ihcn_Ih(0.5),ihsoma,tvec)
L5PC.apic[siteVec[0]] st1 = new IClamp(siteVec[1])
                                                                                                                                                                               
sl = L5PC.locateSites("dend","""+str(basalpoint)+""")
maxdiam = 0                                           
for(i=0;i<sl.count();i+=1){                           
  dd1 = sl.o[i].x[1]                                  
  dd = L5PC.apic[sl.o[i].x[0]].diam(dd1)              
  if (dd > maxdiam) {                                 
    j = i                                             
    maxdiam = dd                                      
  }                                                   
}                                                     
siteVec[0] = sl.o[j].x[0]                             
siteVec[1] = sl.o[j].x[1]                                                                                                                                                                                                                              
L5PC.dend[siteVec[0]] st2 = new AlphaSynapse(siteVec[1])
st2.tau = 3
st2.gmax = 0.0
st2.e = """+str(conde)+"""
""")

  #Block or amplify Ih channels:
  h("""forall if(ismembrane("Ih")) { gIhbar_Ih = """+str(Ihcoeff)+"""*gIhbar_Ih } """)

  times_all = []
  Vsoma_all = []
  Vdend_all = []
  ihsoma_all = []
  ihdend_all = []
  Is_this = [I0, I1, (I0+I1)/2]
  Niter = 25
  Vmaxs_tested = []
  Vmaxdends_tested = []
  Is_tested = []
  for iI in range(0,Niter):
    print("Running "+str(iI)+": I = "+str(Is_this[min(iI,2)]))
    squareAmp = Is_this[min(iI,2)]
    squareDur = 0.2
    tstop = 1000
    h("""
tstop = """+str(tstop)+"""
v_init = """+str(v0)+"""
cai0_ca_ion = """+str(ca0)+"""
st1.amp = """+str(squareAmp)+"""
st1.dur = """+str(squareDur)+"""
st1.del = 800
st2.gmax = """+str(basalcond)+"""
st2.onset = 800
""")
    h.init()
    h.run()

    times=np.array(h.tvec)
    Vsoma=np.array(h.vsoma)
    Vdend=np.array(h.vdend)
    Casoma=np.array(h.casoma)
    Cadend=np.array(h.cadend)

    Vmaxs_tested.append(max(Vsoma))
    Vmaxdends_tested.append(max(Vdend))
    Is_tested.append(Is_this[min(iI,2)])

    if iI < 2:
      continue
    if max(Vsoma) > -10:
      Is_this = [Is_this[0], Is_this[2], (Is_this[0]+Is_this[2])/2]
    else:
      Is_this = [Is_this[2], Is_this[1], (Is_this[2]+Is_this[1])/2]

    picklelist = [Is_tested,Vmaxs_tested,Vmaxdends_tested]
    file = open('strongdendthresh'+str(denddist)+'_basal'+str(basalpoint)+'cond'+str(basalcond)+'_conde'+str(conde)+'_Ihcoeff'+str(Ihcoeff)+'_absbound.sav', 'wb')
    pickle.dump(picklelist,file)
    file.close()