Add constexpr shift_right mp helper

randombit · Apr 12, 2024 · 89d0d1b · 89d0d1b
1 parent 3801b0a
commit 89d0d1b
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/src/lib/math/mp/mp_core.h b/src/lib/math/mp/mp_core.h
@@ -845,15 +845,29 @@ inline constexpr W shift_left(std::array<W, N>& x) {
    static_assert(S < WordInfo<W>::bits, "Shift too large");
 
    W carry = 0;
-   for(size_t j = 0; j != N; ++j) {
-      const W w = x[j];
-      x[j] = (w << S) | carry;
+   for(size_t i = 0; i != N; ++i) {
+      const W w = x[i];
+      x[i] = (w << S) | carry;
       carry = w >> (WordInfo<W>::bits - S);
    }
 
    return carry;
 }
 
+template <size_t S, WordType W, size_t N>
+inline consteval W shift_right(std::array<W, N>& x) {
+   static_assert(S < WordInfo<W>::bits, "Shift too large");
+
+   W carry = 0;
+   for(size_t i = 0; i != N; ++i) {
+      const W w = x[N - 1 - i];
+      x[N - 1 - i] = (w >> S) | carry;
+      carry = w << (WordInfo<W>::bits - S);
+   }
+
+   return carry;
+}
+
 template <WordType W, size_t N>
 consteval auto hex_to_words(const char (&s)[N]) {
    // Char count includes null terminator which we ignore