mirror of
https://github.com/romanz/amodem.git
synced 2026-02-07 09:28:02 +08:00
pylint fixes
This commit is contained in:
6
loop.py
6
loop.py
@@ -1,10 +1,10 @@
|
||||
import numpy as np
|
||||
import itertools
|
||||
|
||||
import recv
|
||||
import sampling
|
||||
import sigproc
|
||||
|
||||
|
||||
class Filter(object):
|
||||
def __init__(self, b, a=()):
|
||||
self.b = b
|
||||
@@ -20,6 +20,7 @@ class Filter(object):
|
||||
self.y = [y] + self.y
|
||||
return y
|
||||
|
||||
|
||||
class FreqLoop(object):
|
||||
def __init__(self, x, freqs, prefix=0.0):
|
||||
interp = sampling.Interpolator()
|
||||
@@ -34,7 +35,8 @@ class FreqLoop(object):
|
||||
|
||||
samplers = itertools.tee(self.sampler, len(freqs))
|
||||
for freq, generator in zip(freqs, samplers):
|
||||
self.gens.append( sigproc.extract_symbols(generator, freq) )
|
||||
gen = sigproc.extract_symbols(generator, freq)
|
||||
self.gens.append(gen)
|
||||
|
||||
Kp, Ki = 0.2, 0.01
|
||||
b = np.array([1, -1])*Kp + np.array([0.5, 0.5])*Ki
|
||||
|
||||
@@ -3,6 +3,7 @@ import logging
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class Interpolator(object):
|
||||
def __init__(self, resolution=10000, width=128):
|
||||
self.width = width
|
||||
@@ -28,6 +29,7 @@ class Interpolator(object):
|
||||
coeffs = self.filt[j]
|
||||
return coeffs, k - self.width
|
||||
|
||||
|
||||
class Sampler(object):
|
||||
def __init__(self, src, interp=None):
|
||||
self.src = iter(src)
|
||||
@@ -70,6 +72,6 @@ if __name__ == '__main__':
|
||||
_, x = common.load(sys.stdin)
|
||||
sampler = Sampler(x, Interpolator())
|
||||
sampler.freq += df
|
||||
y = np.array(list(sampler), )
|
||||
sys.stdout.write( common.dumps(y*1j) )
|
||||
|
||||
y = np.array(list(sampler))
|
||||
y = common.dumps(y*1j)
|
||||
sys.stdout.write(y)
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
import sampling
|
||||
import numpy as np
|
||||
import pylab
|
||||
|
||||
|
||||
def test_resample():
|
||||
x = np.arange(300)
|
||||
@@ -12,7 +12,8 @@ def test_resample():
|
||||
err = x - y
|
||||
assert np.max(np.abs(err)) < 1e-10
|
||||
|
||||
|
||||
def test_coeffs():
|
||||
I = sampling.Interpolator(width=4, resolution=16)
|
||||
err = I.filt[0] - [0,0,0,1,0,0,0,0]
|
||||
err = I.filt[0] - [0, 0, 0, 1, 0, 0, 0, 0]
|
||||
assert np.max(np.abs(err)) < 1e-10
|
||||
|
||||
@@ -4,6 +4,7 @@ import common
|
||||
import numpy as np
|
||||
import random
|
||||
|
||||
|
||||
def test_qam():
|
||||
q = sigproc.QAM(sigproc._symbols)
|
||||
r = random.Random(0)
|
||||
@@ -13,6 +14,7 @@ def test_qam():
|
||||
decoded = list(q.decode(list(S)))
|
||||
assert decoded == bits
|
||||
|
||||
|
||||
def test_drift():
|
||||
fc = 10e3
|
||||
df = 1.23
|
||||
|
||||
Reference in New Issue
Block a user