mirror of
https://github.com/romanz/amodem.git
synced 2026-02-25 00:31:08 +08:00
remove Python 2/3 import hack
This commit is contained in:
@@ -102,12 +102,6 @@ def take(iterable, n):
|
||||
return np.array(list(itertools.islice(iterable, n)))
|
||||
|
||||
|
||||
try:
|
||||
izip = itertools.izip
|
||||
except AttributeError:
|
||||
izip = zip # Python 3
|
||||
|
||||
|
||||
class Dummy(object):
|
||||
|
||||
def __getattr__(self, name):
|
||||
|
||||
@@ -163,6 +163,12 @@ def train_receiver(sampler, order, lookahead):
|
||||
stats = {}
|
||||
|
||||
|
||||
def izip(streams):
|
||||
iters = [iter(s) for s in streams]
|
||||
while True:
|
||||
yield [next(i) for i in iters]
|
||||
|
||||
|
||||
def demodulate(sampler, freqs):
|
||||
streams = []
|
||||
symbol_list = []
|
||||
@@ -187,7 +193,7 @@ def demodulate(sampler, freqs):
|
||||
stats['rx_start'] = time.time()
|
||||
|
||||
log.info('Demodulation started')
|
||||
for i, block in enumerate(common.izip(*streams)): # block per frequency
|
||||
for i, block in enumerate(izip(streams)): # block per frequency
|
||||
for bits in block:
|
||||
stats['rx_bits'] = stats['rx_bits'] + len(bits)
|
||||
yield bits
|
||||
|
||||
Reference in New Issue
Block a user