per/src/main.rs

115 lines
2.7 KiB
Rust

extern crate num_bigint;
// extern crate num_traits;
extern crate num_cpus;
#[macro_use]
extern crate clap;
use clap::App;
use num_bigint::BigUint;
use std::thread;
// use std::sync::mpsc;
use std::sync::{Arc, Barrier, RwLock};
use std::time::{Duration, Instant};
trait Peristance {
fn per_mul(&self) -> u8;
}
impl Peristance for BigUint {
fn per_mul(&self) -> u8 {
let mut n = self.to_radix_le(10)
.into_iter()
.product::<BigUint>();
let mut counter = 1;
while n.to_str_radix(10).chars().count() > 1 {
n = n.to_radix_le(10)
.into_iter()
.product::<BigUint>();
counter += 1;
}
return counter;
}
}
#[derive(Debug)]
struct Counter {
count: BigUint,
step: u8,
offset: u8
}
impl Counter {
fn new(_step: u8, _offset: u8) -> Counter {
Counter {count: BigUint::from(0u8), step: _step, offset: _offset}
}
}
impl Iterator for Counter {
type Item = BigUint;
fn next(&mut self) -> Option<BigUint> {
// Increment our count. This is why we started at zero.
self.count += self.step;
Some(self.count.clone())
}
}
fn main() {
let start = Instant::now();
let args = App::new("Persistance")
.version("0.2.0")
.author("Daniel Olsen")
.about("Finds the sequence of smallest number with a multiplicative persistance of n")
.args_from_usage("-j, --jobs=[threads] 'Sets the number of worker threads to spin up, if 0 or empty this equals the number of cores on your system (INCLUDING HYPERTHREADS)'
-m, --max=[max] 'Sets the maximum n you want to calculate'")
.get_matches();
let mut cpus = value_t!(args, "jobs", u8).unwrap_or(num_cpus::get() as u8);
if cpus == 0 {
cpus = num_cpus::get() as u8;
}
let cpus = cpus;
let max = value_t!(args, "max", u8).unwrap_or(11u8);
println!("threads: {}", cpus);
let barrier = Arc::new(Barrier::new(2));
for thread in 0..(cpus) {
let c = barrier.clone();
thread::spawn(move || {
println!("Started thread {}!", thread);
let counter = Counter::new(cpus-thread, cpus);
let mut top: u8 = 0;
for x in counter {
let n = x.per_mul();
if n > top {
top = n;
println!("{:?} {} - {}: {}", start.elapsed(), thread, n, x);
if n > max-1 {
c.wait();
return;
}
}
}
});
}
barrier.wait();
println!("Finding took {:#?}", start.elapsed());
return;
}