A crate for generating and working with prime numbers in const contexts.
#![no_std]
compatible, and currently supports Rust versions 1.68.2 and newer.
Generate arrays of prime numbers with the function primes
which uses a segmented sieve of Eratosthenes:
const PRIMES: [u32; 10] = primes();
assert_eq!(PRIMES[5], 13);
assert_eq!(PRIMES, [2, 3, 5, 7, 11, 13, 17, 19, 23, 29]);
or with the wrapping type Primes
:
const PRIMES: Primes<10> = Primes::new();
assert_eq!(PRIMES[5], 13);
assert_eq!(PRIMES, [2, 3, 5, 7, 11, 13, 17, 19, 23, 29]);
which also lets you reuse it as a cache of primes for related computations:
const CACHE: Primes<100> = Primes::new();
// For primality testing
const CHECK_42: Option<bool> = CACHE.is_prime(42);
const CHECK_541: Option<bool> = CACHE.is_prime(541);
assert_eq!(CHECK_42, Some(false));
assert_eq!(CHECK_541, Some(true));
// Or for prime counting
const PRIMES_LEQ_100: Option<usize> = CACHE.count_primes_leq(100);
assert_eq!(PRIMES_LEQ_100, Some(25));
// If questions are asked about numbers
// outside the cache it returns None
assert!(CACHE.is_prime(1000).is_none());
assert!(CACHE.count_primes_leq(1000).is_none());
Sieve a range of numbers for their prime status with sieve
:
const N: usize = 10;
const PRIME_STATUS: [bool; N] = sieve();
// 0 1 2 3 4 5 6 7 8 9
assert_eq!(PRIME_STATUS, [false, false, true, true, false, true, false, true, false, false]);
The crate also provides prime generation and sieving functions that can be used to work with ranges that don't start at zero, e.g. primes_geq
and sieve_lt
. They take two generics:
the number of elements to return and the size of the sieve used during evaluation. The sieve size must be at least the ceiling
of the square root of the largest encountered value.
Compute 3 primes greater than or equal to 5000000031:
const N: usize = 3;
// ceil(sqrt(5_000_000_063)) = 70_711
const PRIMES_GEQ: Result<[u64; N], GenerationError> = primes_geq::<N, 70_711>(5_000_000_031);
assert_eq!(PRIMES_GEQ, Ok([5_000_000_039, 5_000_000_059, 5_000_000_063]));
Functions in the crate can help with computing the required sieve size.
Sieve the three numbers less than 5000000031 for their prime status:
use const_primes::isqrt;
const N: usize = 3;
const LIMIT: u64 = 5_000_000_031;
const MEM: usize = isqrt(LIMIT) as usize + 1;
const PRIME_STATUS_LT: Result<[bool; N], SieveError> = sieve_lt::<N, MEM>(LIMIT);
// 5_000_000_028 5_000_000_029 5_000_000_030
assert_eq!(PRIME_STATUS_LT, Ok([false, true, false]));
The sieve size can also be computed by the crate by using the macros primes_segment!
and sieve_segment!
.
const PRIMES_GEQ: Result<[u64; 2], GenerationError> = primes_segment!(2; >= 615);
const PRIME_STATUS_LT: Result<[bool; 3], SieveError> = sieve_segment!(3; < 100_005);
// 100_102 100_103 100_104
assert_eq!(PRIME_STATUS_LT, Ok([false, true, false]));
assert_eq!(PRIMES_GEQ, Ok([617, 619]));
Use is_prime
to test whether a given number is prime:
const CHECK: bool = is_prime(18_446_744_073_709_551_557);
assert!(CHECK);
Find the next or previous prime numbers with next_prime
and previous_prime
if they exist:
const NEXT: Option<u64> = next_prime(25);
const PREV: Option<u64> = previous_prime(25);
const NOSUCH: Option<u64> = previous_prime(2);
assert_eq!(NEXT, Some(29));
assert_eq!(PREV, Some(23));
assert_eq!(NOSUCH, None);
and more!
std
: implements the Error
trait for the error types.
Licensed under either of
- Apache License, Version 2.0 LICENSE-APACHE
- MIT license LICENSE-MIT
at your option.
Contributions are welcome!
Unless you explicitly state otherwise, any contribution intentionally submitted for inclusion in the work by you, as defined in the Apache-2.0 license, shall be dual licensed as above, without any additional terms or conditions.