"""
Filters for time series.

Copyright (c) 2020 Gabriele Gilardi

ToDo:
- use NaN/input values for points not filtered?
- return idx?
- util to test filter (impulse, utils)
- warning in filter when wrong order?
- use self.a and self.b
- remove a and b from plots
- in comments write what filters do
- is necessary to copy X for Y untouched?
"""

import sys
import numpy as np
import filters as flt
import utils as utl

# Read data to filter
if len(sys.argv) != 2:
    print("Usage: python test.py <data_file>")
    sys.exit(1)
data_file = sys.argv[1] + '.csv'

# Read data from a csv file
data = np.loadtxt(data_file, delimiter=',')
n_samples = data.shape[0]
data = data.reshape(n_samples, -1)

spx = flt.Filter(data)

# args = (spx.X, spx.SMA(N=5), spx.EMA(alpha=0.7))
# utl.plot_signals(args)

# alpha = 0.8
# bb = np.array([alpha])
# aa = np.array([1.0, alpha - 1.0])


res, bb, aa = spx.GaussHigh(N=3, P=10)
print(bb)
print(aa)

utl.plot_frequency_response(bb, aa)
utl.plot_lag_response(bb, aa)