misc: Add count_lead_ones() function, useful for UTF-8 handling.

1 files changed, 13 insertions, 0 deletions

diff --git a/py/misc.h b/py/misc.h
index 97e9b30ed..7a736560a 100644
--- a/py/misc.h
+++ b/py/misc.h
@@ -166,4 +166,17 @@ int DEBUG_printf(const char *fmt, ...);
 
 extern uint mp_verbose_flag;
 
+// This is useful for unicode handling. Some CPU archs has
+// special instructions for efficient implentation of this
+// function (e.g. CLZ on ARM).
+#ifndef count_lead_ones
+static inline uint count_lead_ones(byte val) {
+    uint c = 0;
+    for (byte mask = 0x80; val & mask; mask >>= 1) {
+        c++;
+    }
+    return c;
+}
+#endif
+
 #endif // _INCLUDED_MINILIB_H