Skip to content

Commit

Permalink
Merging with master release v3.2.0.
Browse files Browse the repository at this point in the history
Merge branch 'master' into pton-dspv3
  • Loading branch information
cryo committed Nov 5, 2019
2 parents b4c9403 + 8023b42 commit 8fe95ee
Showing 1 changed file with 22 additions and 10 deletions.
32 changes: 22 additions & 10 deletions pysmurf/util/smurf_util.py
Original file line number Diff line number Diff line change
Expand Up @@ -2594,7 +2594,8 @@ def make_smurf_to_gcp_config(self, num_averages=None, filename=None,
def get_smurf_to_mce_file(self):
"""
Reads in the most recent smurf_to_mce file and returns
everything in it as a dict.
everything in it as a dict. The file location is defined
in self.smurf_to_mce_file.
Ret:
----
Expand All @@ -2612,13 +2613,19 @@ def get_smurf_to_mce_file(self):

return ret


def get_filter_params(self):
"""
Get the downsample filter parameters: filter order,
filter gain, num averages, and the actual filter
parameters. This reads the most recent smurf_to_mce_file
to get the parameters.
to get the parameters. This is defined in
self.smurf_to_mce_file.
If filter order is -1, the downsampler is using a
rectangula integrator. This will set filter_a, filter_b
to None.
Ret:
----
filter_params (dict) : A dictionary with the filter
Expand All @@ -2627,18 +2634,23 @@ def get_filter_params(self):
# Load cfg file
cfg = self.get_smurf_to_mce_file()

# Get filter order
# Get filter order, gain, and averages
filter_order = int(cfg['filter_order'])
filter_gain = float(cfg['filter_gain'])
num_averages = int(cfg['num_averages'])

# Get filter parameters
a = np.zeros(filter_order, dtype=float)
b = np.zeros(filter_order, dtype=float)
for i in range(filter_order):
a[i] = cfg[f'filter_a{i}']
b[i] = cfg[f'filter_b{i}']

if filter_order < 0:
a = None
b = None
else:
# Get filter parameters - (filter_order+1) elements
a = np.zeros(filter_order+1, dtype=float)
b = np.zeros(filter_order+1, dtype=float)
for i in range(filter_order+1):
a[i] = cfg[f'filter_a{i}']
b[i] = cfg[f'filter_b{i}']

# Cast into dictionary
ret = {
'filter_order' : filter_order,
'filter_gain': filter_gain,
Expand Down

0 comments on commit 8fe95ee

Please sign in to comment.