93 lines
3.6 KiB
Python
93 lines
3.6 KiB
Python
import os #compability with windows
|
|
from IPython import embed
|
|
import numpy as np
|
|
|
|
def parse_dataset(dataset_name):
|
|
assert(os.path.exists(dataset_name)) #see if data exists
|
|
f = open(dataset_name, 'r') #open data we gave in
|
|
lines = f.readlines() #read data
|
|
f.close() #?
|
|
|
|
# metadata lists for every loop
|
|
eodfs = []
|
|
deltafs = []
|
|
stimulusfs = []
|
|
|
|
# data itself
|
|
times = []
|
|
frequencies = []
|
|
amplitudes = []
|
|
|
|
# temporary lists with data we put in the lists above
|
|
time = []
|
|
ampl = []
|
|
freq = []
|
|
|
|
for i in range(len(lines)):
|
|
l = lines[i].strip() #all lines of textdata, exclude all empty lines (empty () default for spacebar)
|
|
if "#" in l and "EODf" in l: #if line starts with # EODf:
|
|
eodfs.append(float(l.split(':')[-1].strip()[:-2])) #append: line splitted by ':' the 2nd part ([-1],
|
|
if "#" in l and "Delta f" in l: #which got striped so we sure there is no space at the end,
|
|
deltafs.append(float(l.split(':')[-1].strip()[:-2])) #from that all expect the last two signs (Hz unit)
|
|
if "#" in l and "StimulusFrequency" in l: #this for different metadata in different lists
|
|
stimulusfs.append(float(l.split(':')[-1].strip()[:-2]))
|
|
|
|
if '#Key' in l:
|
|
if len(time) != 0: #therefore empty in the first round
|
|
times.append(time) #2nd loop means time != 0, so we put the times/amplitudes/frequencies to
|
|
amplitudes.append(ampl) #the data of the first loop
|
|
frequencies.append(freq)
|
|
|
|
time = [] #temporary lists to overwrite the lists with the same name we made before
|
|
ampl = [] #so they are empty again
|
|
freq = []
|
|
|
|
if len(l) > 0 and l[0] is not '#': #line not empty and doesnt start with #
|
|
temporary = list(map(float, l.split())) #temporary list where we got 3 index splitted by spacebar, map to find them
|
|
time.append(temporary[0]) #temporary lists with the data at that place, respectively
|
|
freq.append(temporary[1])
|
|
ampl.append(temporary[2])
|
|
|
|
times.append(time) #append data from one list to another
|
|
amplitudes.append(ampl) #these append the data from the first loop to the final lists, because we overwrite them (?)
|
|
frequencies.append(freq)
|
|
|
|
return times, frequencies, amplitudes, eodfs, deltafs, stimulusfs #output of the function
|
|
|
|
|
|
def mean_noise_cut(frequencies, time, n):
|
|
cutf = []
|
|
cutt = []
|
|
for k in np.arange(0, len(frequencies), n):
|
|
t = time[k]
|
|
f = np.mean(frequencies[k:k+n])
|
|
cutf.append(f)
|
|
cutt.append(t)
|
|
return cutf, cutt
|
|
|
|
def step_response(t, a1, a2, tau1, tau2):
|
|
r_step = a1*(1 - np.exp(-t/tau1)) + a2*(1- np.exp(-t/tau2))
|
|
return r_step
|
|
# plotten mit manual values for a1, ...
|
|
# auch mal a1 oder a2 auf Null setzen.
|
|
|
|
|
|
|
|
def base_eod(frequencies, time, onset_point):
|
|
base_eod = []
|
|
|
|
onset_end = onset_point - 10
|
|
|
|
base = np.mean(frequencies[(time >= onset_end) & (time < onset_point)])
|
|
base_eod.append(base)
|
|
return base_eod
|
|
|
|
def JAR_eod(frequencies, time, offset_point):
|
|
jar_eod = []
|
|
|
|
offset_start = offset_point - 10
|
|
|
|
jar = np.mean(frequencies[(time >= offset_start) & (time < offset_point)])
|
|
jar_eod.append(jar)
|
|
|
|
return jar_eod |