|
|
@@ -1,78 +0,0 @@
|
|
|
-# -*- coding: utf-8 -*-
|
|
|
-"""
|
|
|
-Created on Wed Oct 9 17:29:15 2019
|
|
|
-
|
|
|
-@author: kleisp
|
|
|
-"""
|
|
|
-
|
|
|
-import os
|
|
|
-os.chdir('C:\EAfiles\EAdetection')
|
|
|
-
|
|
|
-import core.ea_management as eam
|
|
|
-from core.helpers import open_hdf5
|
|
|
-from glob import glob
|
|
|
-#import phases_ictalInter as pii
|
|
|
-#import core.ea_analysis as eana
|
|
|
-
|
|
|
-import pandas as pd
|
|
|
-import numpy as np
|
|
|
-
|
|
|
-
|
|
|
-os.chdir('C:\EAfiles\EAdetection')
|
|
|
-reload(eam)
|
|
|
-
|
|
|
-
|
|
|
-#%% function for getting line length directly from resampled raw data
|
|
|
-
|
|
|
-def get_linelength(filename,ll_dict_controls, recID):
|
|
|
- excelpath = 'C:\EAfiles\Excel\\startstop_PK_HCc.xlsx' #insert the directory for an excel file containing the start and stop times for each recording (first column: recID, second column: start time, third column stop time)
|
|
|
- df = pd.read_excel(excelpath) #reading an excel file with start and stop times (s) of a recording
|
|
|
- df.set_index('recID', inplace=True) #recID
|
|
|
- ll_dict_controls[recID]['start'] = int(df.start[recID]) #start time
|
|
|
- ll_dict_controls[recID]['stop'] = int(df.stop[recID]) #stop time
|
|
|
- sr=500 #sampling rate of downsampled data
|
|
|
- data_dict=open_hdf5(filename,group=None,read_maskedarr=False)
|
|
|
- data=data_dict['data']['trace']
|
|
|
- reccut=data[int(ll_dict_controls[recID]['start']*sr):int(ll_dict_controls[recID]['stop']*sr)] #cut data according to start and stop times
|
|
|
- N_datapoints=len(reccut)
|
|
|
- duration=N_datapoints/sr
|
|
|
- linelength = np.sum(np.abs(np.diff(reccut))) #line length is the sum of subsequent datapoints -->derivative
|
|
|
- linelength_s=linelength/duration
|
|
|
- ll_dict_controls[recID]['linelength/second']=linelength_s
|
|
|
- ll_dict_controls[recID]['linelength']=linelength
|
|
|
-
|
|
|
-#%% get linelength directly from h5 (downsampled data)
|
|
|
-
|
|
|
-os.chdir(r'C:\EAfiles\DATA\h5files') #enter directory, where all the needed h5 files are
|
|
|
-files = glob(r'C:\EAfiles\DATA\h5files\*.h5') #takes all h5 files from this folder
|
|
|
-hfile = files
|
|
|
-lines = np.arange(0, len(hfile)) #as many lines as there are h5 files
|
|
|
-ll_dict_controls={}
|
|
|
-ll_dict_controls_d={}
|
|
|
-
|
|
|
-for i in lines:
|
|
|
- recID = os.path.basename(hfile[i]).split('__')[0] #getting the IDs
|
|
|
- print (i)
|
|
|
- print (recID)
|
|
|
-
|
|
|
- filename=os.path.basename(hfile[i])
|
|
|
- ll_dict_controls[recID]= {}
|
|
|
- get_linelength(filename,ll_dict_controls, recID)
|
|
|
-
|
|
|
-#%% creating arrays with data
|
|
|
-recIDs = np.array(ll_dict_controls.keys())
|
|
|
-
|
|
|
-flavour = 'linelength/second'
|
|
|
-ii_linelengths = np.array([ll_dict_controls[recID][flavour] for recID in ll_dict_controls.keys()])
|
|
|
-
|
|
|
-#%% create excel file
|
|
|
-
|
|
|
-os.chdir('C:\EAfiles\RESULTS')
|
|
|
-
|
|
|
-df_sr=pd.DataFrame(columns=
|
|
|
- ['recID',
|
|
|
- 'linelength/s'])
|
|
|
-
|
|
|
-df_sr['recID'] = recIDs
|
|
|
-df_sr['linelength/s'] = ii_linelengths
|
|
|
-df_sr.to_excel('test_linelengths.xlsx') #name the file
|
