def fwdpp_iter(fileObject):
header = True
iteration = ''
ns = 0
for line in fileObject:
if header is False:
if line.startswith('//'):
if ns % 2 == 0:
selClass = msFormat(iteration.rstrip())
yield neuClass, selClass
else:
neuClass = msFormat(iteration.rstrip())
iteration = ''
ns += 1
else:
iteration += line
elif header is True:
if line.startswith('//'):
ns += 1
header = False
if ns % 2 == 0:
selClass = msFormat(iteration.rstrip())
yield neuClass, selClass
def ms_iter(fileObject):
header = True
iteration = ''
for line in fileObject:
if header is False:
if line.startswith('//'):
msClass = msFormat(iteration.rstrip())
yield msClass
iteration = ''
else:
iteration += line
elif header is True:
if line.startswith('//'):
header = False
msClass = msFormat(iteration.rstrip())
yield msClass