added comments, validated code to pep8
This commit is contained in:
parent
f8ab552f3e
commit
28d92c0574
18
pyedf.py
18
pyedf.py
@ -14,12 +14,14 @@ import re
|
|||||||
def read_edf_file(fileobj):
|
def read_edf_file(fileobj):
|
||||||
data = fileobj.read()
|
data = fileobj.read()
|
||||||
header = {}
|
header = {}
|
||||||
header['version'] = data[0:7].strip() # 8
|
# Parse header information based on the EDF/EDF+ specs
|
||||||
header['patient_id'] = data[7:88].strip() # 80
|
# http://www.edfplus.info/specs/index.html
|
||||||
header['rec_id'] = data[88:168].strip() # 80
|
header['version'] = data[0:7].strip()
|
||||||
header['startdate'] = data[168:176].strip() # 8
|
header['patient_id'] = data[7:88].strip()
|
||||||
header['starttime'] = data[176:184].strip() # 8
|
header['rec_id'] = data[88:168].strip()
|
||||||
header['header_bytes'] = int(data[184:192].strip()) # 8
|
header['startdate'] = data[168:176].strip()
|
||||||
|
header['starttime'] = data[176:184].strip()
|
||||||
|
header['header_bytes'] = int(data[184:192].strip())
|
||||||
header['num_items'] = int(data[236:244].strip())
|
header['num_items'] = int(data[236:244].strip())
|
||||||
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())
|
||||||
@ -48,7 +50,7 @@ def main():
|
|||||||
# create the parser
|
# create the parser
|
||||||
parser = argparse.ArgumentParser(description='Process a given EDF File.')
|
parser = argparse.ArgumentParser(description='Process a given EDF File.')
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
'-f',
|
'-f',
|
||||||
'--file',
|
'--file',
|
||||||
type=argparse.FileType('r'),
|
type=argparse.FileType('r'),
|
||||||
help='EDF File to be processed.',
|
help='EDF File to be processed.',
|
||||||
@ -57,6 +59,6 @@ def main():
|
|||||||
data = read_edf_file(args.file)
|
data = read_edf_file(args.file)
|
||||||
args.file.close()
|
args.file.close()
|
||||||
print data
|
print data
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
main()
|
main()
|
||||||
|
Reference in New Issue
Block a user