diff options
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/enums.py | 3 | ||||
| -rw-r--r-- | utils/misc.py | 16 |
2 files changed, 17 insertions, 2 deletions
diff --git a/utils/enums.py b/utils/enums.py index e154d0f..7798cca 100644 --- a/utils/enums.py +++ b/utils/enums.py @@ -45,7 +45,8 @@ class ParamTag(Enum): MMANGLES = 2 SCALE = 3 SRCANGLES = 4 - NONE = 5 + BESTFIT = 5 + NONE = 6 class Priors(Enum): diff --git a/utils/misc.py b/utils/misc.py index 108a458..ebc66e2 100644 --- a/utils/misc.py +++ b/utils/misc.py @@ -57,7 +57,7 @@ class Param(object): def tag(self): return self._tag - @tex.setter + @tag.setter def tag(self, t): if t is None: self._tag = ParamTag.NONE else: @@ -150,10 +150,24 @@ class ParamSet(Sequence): def from_tag(self, tag): return tuple([obj for obj in self._params if obj.tag is tag]) + def remove_tag(self, tag): + return tuple([obj for obj in self._params if obj.tag is not tag]) + def idx_from_tag(self, tag): return tuple([idx for idx, obj in enumerate(self._params) if obj.tag is tag]) + def not_idx_from_tag(self, tag): + return tuple([idx for idx, obj in enumerate(self._params) + if obj.tag is not tag]) + + def slice_from_tag(self, array, tags): + tags = [tags] + idxs = [] + for tag in tags: + idxs.extend(self.idx_from_tag(tag)) + return array[idxs,] + class SortingHelpFormatter(argparse.HelpFormatter): """Sort argparse help options alphabetically.""" |