Skip to content
Snippets Groups Projects
Commit 12947775 authored by Sebastian Steinlechner's avatar Sebastian Steinlechner
Browse files

add accumulate function to nb.Budget

parent 20b38c29
No related branches found
No related tags found
1 merge request!134add accumulate function to nb.Budget
...@@ -332,6 +332,9 @@ class Budget(Noise): ...@@ -332,6 +332,9 @@ class Budget(Noise):
references = [] references = []
"""List of reference noise classes, or (ref, cal) tuples""" """List of reference noise classes, or (ref, cal) tuples"""
accumulate = quadsum
"""Method used to calculate total noise trace"""
def __init__(self, freq=None, noises=None, **kwargs): def __init__(self, freq=None, noises=None, **kwargs):
"""Initialize Budget object. """Initialize Budget object.
...@@ -565,7 +568,7 @@ class Budget(Noise): ...@@ -565,7 +568,7 @@ class Budget(Noise):
_precomp=_precomp, _precomp=_precomp,
) )
budget.append(trace) budget.append(trace)
total = quadsum([trace.psd for trace in budget if trace.name in self._budget_noises]) total = type(self).accumulate([trace.psd for trace in budget if trace.name in self._budget_noises])
return self._make_trace( return self._make_trace(
psd=total, budget=budget psd=total, budget=budget
) )
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment