diff options
Diffstat (limited to 'src/common')
| -rw-r--r-- | src/common/uint128.cpp | 24 | ||||
| -rw-r--r-- | src/common/uint128.h | 23 |
2 files changed, 26 insertions, 21 deletions
diff --git a/src/common/uint128.cpp b/src/common/uint128.cpp index aea7f03e2..8548ba808 100644 --- a/src/common/uint128.cpp +++ b/src/common/uint128.cpp | |||
| @@ -1,7 +1,25 @@ | |||
| 1 | #ifdef _MSC_VER | ||
| 2 | #include <intrin.h> | ||
| 3 | |||
| 4 | #pragma intrinsic(_umul128) | ||
| 5 | #endif | ||
| 6 | #include "common/uint128.h" | ||
| 1 | 7 | ||
| 2 | namespace Common { | 8 | namespace Common { |
| 9 | u128 Multiply64Into128(u64 a, u64 b) { | ||
| 10 | #ifdef _MSC_VER | ||
| 11 | u128 result; | ||
| 12 | result[0] = _umul128(a, b, &result[1]); | ||
| 13 | #else | ||
| 14 | unsigned __int128 tmp = a; | ||
| 15 | tmp *= b; | ||
| 16 | u128 result; | ||
| 17 | std::memcpy(&result, &tmp, sizeof(u128)); | ||
| 18 | #endif | ||
| 19 | return result; | ||
| 20 | } | ||
| 3 | 21 | ||
| 4 | std::pair<u64, u64> udiv128(u128 dividend, u64 divisor) { | 22 | std::pair<u64, u64> Divide128On64(u128 dividend, u64 divisor) { |
| 5 | u64 remainder = dividend[0] % divisor; | 23 | u64 remainder = dividend[0] % divisor; |
| 6 | u64 accum = dividend[0] / divisor; | 24 | u64 accum = dividend[0] / divisor; |
| 7 | if (dividend[1] == 0) | 25 | if (dividend[1] == 0) |
| @@ -12,6 +30,10 @@ std::pair<u64, u64> udiv128(u128 dividend, u64 divisor) { | |||
| 12 | u64 second_segment = (first_segment % divisor) << 32; | 30 | u64 second_segment = (first_segment % divisor) << 32; |
| 13 | accum += (second_segment / divisor); | 31 | accum += (second_segment / divisor); |
| 14 | remainder += second_segment % divisor; | 32 | remainder += second_segment % divisor; |
| 33 | if (remainder >= divisor) { | ||
| 34 | accum++; | ||
| 35 | remainder -= divisor; | ||
| 36 | } | ||
| 15 | return {accum, remainder}; | 37 | return {accum, remainder}; |
| 16 | } | 38 | } |
| 17 | 39 | ||
diff --git a/src/common/uint128.h b/src/common/uint128.h index fda313bcc..45e384c33 100644 --- a/src/common/uint128.h +++ b/src/common/uint128.h | |||
| @@ -1,30 +1,13 @@ | |||
| 1 | #include <array> | 1 | #include <array> |
| 2 | #include <cstdint> | 2 | #include <cstdint> |
| 3 | #include <utility> | ||
| 4 | #include <cstring> | 3 | #include <cstring> |
| 4 | #include <utility> | ||
| 5 | #include "common/common_types.h" | 5 | #include "common/common_types.h" |
| 6 | 6 | ||
| 7 | namespace Common { | 7 | namespace Common { |
| 8 | 8 | ||
| 9 | #ifdef _MSC_VER | 9 | u128 Multiply64Into128(u64 a, u64 b); |
| 10 | #include <intrin.h> | ||
| 11 | |||
| 12 | #pragma intrinsic(_umul128) | ||
| 13 | #endif | ||
| 14 | |||
| 15 | inline u128 umul128(u64 a, u64 b) { | ||
| 16 | #ifdef _MSC_VER | ||
| 17 | u128 result; | ||
| 18 | result[0] = _umul128(a, b, &result[1]); | ||
| 19 | #else | ||
| 20 | unsigned __int128 tmp = a; | ||
| 21 | tmp *= b; | ||
| 22 | u128 result; | ||
| 23 | std::memcpy(&result, &tmp, sizeof(u128)); | ||
| 24 | #endif | ||
| 25 | return result; | ||
| 26 | } | ||
| 27 | 10 | ||
| 28 | std::pair<u64, u64> udiv128(u128 dividend, u64 divisor); | 11 | std::pair<u64, u64> Divide128On64(u128 dividend, u64 divisor); |
| 29 | 12 | ||
| 30 | } // namespace Common | 13 | } // namespace Common |