common: rephrase bitpack(), use list comprehension Python idiom

[libsigrokdecode.git] / decoders / common / srdhelper / mod.py

diff --git a/decoders/common/srdhelper/mod.py b/decoders/common/srdhelper/mod.py
index b559c95edc5f280ae727fb6c34a372143b01be17..e37345a0b34f1a416c5c6b407cc746ad9b80ae4c 100644 (file)
--- a/decoders/common/srdhelper/mod.py
+++ b/decoders/common/srdhelper/mod.py
@@ -21,11 +21,11 @@
 def bcd2int(b):
     return (b & 0x0f) + ((b >> 4) * 10)
 
+def bin2int(s: str):
+    return int('0b' + s, 2)
+
 def bitpack(bits):
-    res = 0
-    for i, b in enumerate(bits):
-        res |= b << i
-    return res
+    return sum([b << i for i, b in enumerate(bits)])
 
 def bitunpack(num, minbits=0):
     res = []