rays/pyedf
Archived
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

code cleanup and added a license

Browse Source
This commit is contained in:
Ray Slakinski 2010-09-14 11:59:49 -04:00
parent 05a92dd0b4
commit f426082d87
1 changed files with 32 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
pyedf.py
View File

@ -1,9 +1,18 @@
#!/usr/bin/env python
# encoding: utf-8
"""
pyedf.py
Created by Ray Slakinski on 2010-09-14.
Copyright (c) 2010 Ray Slakinski. All rights reserved.
"""
import sys
import getopt
import re import re
def read_edf_file(filename):
f = open(filename, 'r') def read_edf_file(fileobj):
data = f.read() data = fileobj.read()
f.close()
header = {} header = {}
header['version'] = data[0:7].strip() # 8 header['version'] = data[0:7].strip() # 8
header['patient_id'] = data[7:88].strip() # 80 header['patient_id'] = data[7:88].strip() # 80
@ -15,22 +24,36 @@ def read_edf_file(filename):
header['data_duration'] = float(data[244:252].strip()) header['data_duration'] = float(data[244:252].strip())
header['num_signals'] = int(data[252:256].strip()) header['num_signals'] = int(data[252:256].strip())
# more data 256 chars down. in header, but ignoring for now # more data 256 chars down. in header, but ignoring for now
print header
records = [] records = []
records_a = records.append records_a = records.append
rec_pos = 512 rec_pos = 512
rec_size = 36 rec_size = 36
# skip first rec # skip first rec
rec_pos += rec_size rec_pos += rec_size
print rec_pos
for i in range(header['num_items']-1): for i in range(header['num_items']-1):
record = {} record = {}
record_split = data[rec_pos:rec_pos+rec_size].split('\x14') record_split = data[rec_pos:rec_pos+rec_size].split('\x14')
matches = re.findall(r'([\d]+)', record_split[2]) matches = re.findall(r'([\d]+)', record_split[2])
record['type'] = record_split[3].strip() record['type'] = record_split[3].strip()
record['time'] = {'start': int(matches[0]), 'durration': int(matches[1])} record['time'] = {
'start': int(matches[0]),
'durration': int(matches[1]),
}
records_a(record) records_a(record)
rec_pos += rec_size rec_pos += rec_size
return {'header': header, 'records': records} return {'header': header, 'records': records}
data = read_edf_file('20100912_230640_EVE.edf')
print data
def main():
argv = sys.argv
opts, args = getopt.getopt(argv[1:], "f", ["file="])
# option processing
for option, value in opts:
if option == "-f" or option == '--file':
f = open(value, 'r')
data = read_edf_file(f)
f.close()
print data
if __name__ == '__main__':
main()