generated from PlexSheep/rs-base
81 lines
2.5 KiB
Rust
81 lines
2.5 KiB
Rust
#![warn(clippy::all)]
|
|
// #![warn(missing_docs)]
|
|
#![warn(missing_debug_implementations)]
|
|
|
|
use clap::Parser;
|
|
use libpt::log::*;
|
|
use tokio;
|
|
|
|
use wordle_analyzer::bench::builtin::BuiltinBenchmark;
|
|
use wordle_analyzer::bench::report::Report;
|
|
use wordle_analyzer::bench::{Benchmark, DEFAULT_N};
|
|
use wordle_analyzer::error::WResult;
|
|
use wordle_analyzer::game::GameBuilder;
|
|
use wordle_analyzer::solve::{AnyBuiltinSolver, BuiltinSolverNames, Solver};
|
|
use wordle_analyzer::wlist::builtin::BuiltinWList;
|
|
|
|
use wordle_analyzer::{self, game};
|
|
|
|
#[derive(Parser, Clone, Debug)]
|
|
#[command(version, about, long_about, author)]
|
|
struct Cli {
|
|
/// precompute all possibilities for better performance at runtime
|
|
#[arg(short, long)]
|
|
precompute: bool,
|
|
/// how long should the word be?
|
|
#[arg(short, long, default_value_t = wordle_analyzer::DEFAULT_WORD_LENGTH)]
|
|
length: usize,
|
|
/// how many times can we guess?
|
|
#[arg(short, long, default_value_t = wordle_analyzer::DEFAULT_MAX_STEPS)]
|
|
max_steps: usize,
|
|
/// more verbose logs
|
|
#[arg(short, long)]
|
|
verbose: bool,
|
|
/// which solver to use
|
|
#[arg(short, long, default_value_t = BuiltinSolverNames::default())]
|
|
solver: BuiltinSolverNames,
|
|
/// how many games to play for the benchmark
|
|
#[arg(short, long, default_value_t = DEFAULT_N)]
|
|
n: usize,
|
|
/// how many threads to use for benchmarking
|
|
///
|
|
/// Note that the application as the whole will use at least one more thread.
|
|
#[arg(short, long, default_value_t = num_cpus::get())]
|
|
threads: usize,
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> anyhow::Result<()> {
|
|
let cli = Cli::parse();
|
|
if cli.verbose {
|
|
Logger::build_mini(Some(Level::DEBUG))?;
|
|
} else {
|
|
Logger::build_mini(Some(Level::INFO))?;
|
|
}
|
|
trace!("dumping CLI: {:#?}", cli);
|
|
|
|
let wl = BuiltinWList::default();
|
|
let builder: GameBuilder<'_, BuiltinWList> = game::Game::builder(&wl)
|
|
.length(cli.length)
|
|
.max_steps(cli.max_steps)
|
|
.precompute(cli.precompute);
|
|
let solver: AnyBuiltinSolver<'_, BuiltinWList> = cli.solver.to_solver(&wl);
|
|
let bench = BuiltinBenchmark::build(&wl, solver, builder, cli.threads)?;
|
|
trace!("{bench:#?}");
|
|
|
|
let n = cli.n;
|
|
let bench2 = bench.clone();
|
|
let in_progress_bench = tokio::spawn(async move {
|
|
bench2.bench(n).await
|
|
});
|
|
|
|
while !bench.is_finished() {
|
|
tokio::time::sleep(tokio::time::Duration::from_millis(5000)).await;
|
|
println!("{}", bench.report());
|
|
}
|
|
in_progress_bench.await;
|
|
dbg!(bench.report());
|
|
|
|
Ok(())
|
|
}
|