Commit c9d9085d authored by Lukas Weber's avatar Lukas Weber
Browse files

rename conditions to filter

parent 625083ab
...@@ -50,24 +50,24 @@ class MCArchive: ...@@ -50,24 +50,24 @@ class MCArchive:
o.mean[i,:] = value['mean'] o.mean[i,:] = value['mean']
o.error[i,:] = value['error'] o.error[i,:] = value['error']
def condition_mask(self, conditions): def filter_mask(self, filter):
if not conditions: if not filter:
return [True for _ in range(self.num_tasks)] return [True for _ in range(self.num_tasks)]
return [all(self.parameters[key][i] == val for key, val in conditions.items()) for i in range(self.num_tasks)] return [all(self.parameters[key][i] == val for key, val in filter.items()) for i in range(self.num_tasks)]
def get_parameter(self, name, unique=False, conditions={}): def get_parameter(self, name, unique=False, filter={}):
selection = list(itertools.compress(self.parameters[name], self.condition_mask(conditions))) selection = list(itertools.compress(self.parameters[name], self.filter_mask(filter)))
if unique: if unique:
return list(sorted(set(selection))) return list(sorted(set(selection)))
return selection return selection
def get_observable(self, name, conditions={}): def get_observable(self, name, filter={}):
orig = self.observables[name] orig = self.observables[name]
selection = Observable(0) selection = Observable(0)
mask = self.condition_mask(conditions) mask = self.filter_mask(filter)
selection.rebinning_bin_count = orig.rebinning_bin_count[mask] selection.rebinning_bin_count = orig.rebinning_bin_count[mask]
selection.rebinning_bin_length = orig.rebinning_bin_length[mask] selection.rebinning_bin_length = orig.rebinning_bin_length[mask]
selection.autocorrelation_time = orig.autocorrelation_time[mask] selection.autocorrelation_time = orig.autocorrelation_time[mask]
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment