import pickle from pylab import * from os.path import exists import time import sys import LFPy filename = sys.argv[1] if exists(filename): unpicklefile = open(filename,'r') unpickledlist = pickle.load(unpicklefile) unpicklefile.close() dipolesE = unpickledlist[2] dt_data = 0.025 #ms dt_EEG = 0.5 #ms dtratio = int(dt_EEG/dt_data) dipolesE_lowres = zeros([int(11000/dt_EEG),3]) for itime in range(0,dipolesE_lowres.shape[0]): dipolesE_lowres[itime,:] = mean(dipolesE[itime*dtratio:(itime+1)*dtratio,:],axis=0) radii = [79000., 80000., 85000., 90000.] sigmas = [0.3, 1.5, 0.015, 0.3] d_contact = 10.0 somapos = array([0., 77500., 0.]) eeg_coords_top = array([[0., radii[3]-d_contact, 0.]]) four_sphere_top = LFPy.FourSphereVolumeConductor(radii, sigmas, eeg_coords_top) picklelist = [] pot_db_4s_top = four_sphere_top.calc_potential(dipolesE_lowres, somapos) picklelist.append(array(pot_db_4s_top) * 1e9) file = open(filename[:-4]+'_EEG.sav','w') pickle.dump(picklelist,file) file.close()