shep

commit 98387157f583e2160fe7272ea07913e9b13cc152
parent 90f494a1f6b519a31df53f0d551622ad1de34729
Author: lash <dev@holbrook.no>
Date:   Mon,  7 Nov 2022 09:52:53 +0000

Bitwise order option of list all states

Diffstat:
M
shep/state.py
 | 
22
+++++++++++++++++++---

1 file changed, 19 insertions(+), 3 deletions(-)
diff --git a/shep/state.py b/shep/state.py
@@ -240,14 +240,29 @@ class State:
         return self.__alias(k, *args)    
 
 
-    def all(self, pure=False, numeric=False, ignore_auto=True):
+    def __all_bit(self):
+        r = []
+        r.append(self.name(0))
+        v = 1
+        while v < self.__limit:
+            r.append(self.name(v))
+            v <<= 1
+        return r
+
+
+    def all(self, pure=False, numeric=False, ignore_auto=True, bit_order=False):
         """Return list of all unique atomic and alias state strings.
         
         :rtype: list of ints
         :return: states
         """
         l = []
-        for k in dir(self):
+        v = None
+        if bit_order:
+            v = self.__all_bit()
+        else:
+            v = dir(self)
+        for k in v:
             state = None
             if k[0] == '_' and ignore_auto:
                 continue
@@ -263,7 +278,8 @@ class State:
                 l.append(state)
             else:
                 l.append(k)
-        l.sort()
+        if not bit_order:
+            l.sort()
         return l