use malachite_base::num::arithmetic::traits::{
PowerOf2, RoundToMultiple, RoundToMultipleOfPowerOf2, RoundToMultipleOfPowerOf2Assign, ShrRound,
};
use malachite_base::test_util::bench::{run_benchmark, BenchmarkType};
use malachite_base::test_util::generators::common::{GenConfig, GenMode};
use malachite_base::test_util::runner::Runner;
use malachite_nz::integer::Integer;
use malachite_nz::test_util::bench::bucketers::triple_1_2_integer_bit_u64_max_bucketer;
use malachite_nz::test_util::generators::integer_unsigned_rounding_mode_triple_gen_var_1;
pub(crate) fn register(runner: &mut Runner) {
register_demo!(runner, demo_integer_round_to_multiple_of_power_of_2_assign);
register_demo!(runner, demo_integer_round_to_multiple_of_power_of_2);
register_demo!(runner, demo_integer_round_to_multiple_of_power_of_2_ref);
register_bench!(
runner,
benchmark_integer_round_to_multiple_of_power_of_2_assign
);
register_bench!(
runner,
benchmark_integer_round_to_multiple_of_power_of_2_algorithms
);
register_bench!(
runner,
benchmark_integer_round_to_multiple_of_power_of_2_evaluation_strategy
);
}
fn demo_integer_round_to_multiple_of_power_of_2_assign(
gm: GenMode,
config: GenConfig,
limit: usize,
) {
for (mut n, pow, rm) in integer_unsigned_rounding_mode_triple_gen_var_1()
.get(gm, &config)
.take(limit)
{
let n_old = n.clone();
n.round_to_multiple_of_power_of_2_assign(pow, rm);
println!(
"x := {}; x.round_to_multiple_of_power_of_2_assign({}, {}); x = {}",
n_old, pow, rm, n
);
}
}
fn demo_integer_round_to_multiple_of_power_of_2(gm: GenMode, config: GenConfig, limit: usize) {
for (n, pow, rm) in integer_unsigned_rounding_mode_triple_gen_var_1()
.get(gm, &config)
.take(limit)
{
let n_old = n.clone();
println!(
"({}).round_to_multiple_of_power_of_2({}, {}) = {}",
n_old,
pow,
rm,
n.round_to_multiple_of_power_of_2(pow, rm)
);
}
}
fn demo_integer_round_to_multiple_of_power_of_2_ref(gm: GenMode, config: GenConfig, limit: usize) {
for (n, pow, rm) in integer_unsigned_rounding_mode_triple_gen_var_1()
.get(gm, &config)
.take(limit)
{
println!(
"(&{}).round_to_multiple_of_power_of_2({}, {}) = {}",
n,
pow,
rm,
(&n).round_to_multiple_of_power_of_2(pow, rm)
);
}
}
fn benchmark_integer_round_to_multiple_of_power_of_2_assign(
gm: GenMode,
config: GenConfig,
limit: usize,
file_name: &str,
) {
run_benchmark(
"Integer.round_to_multiple_of_power_of_2_assign(u64, RoundingMode)",
BenchmarkType::Single,
integer_unsigned_rounding_mode_triple_gen_var_1().get(gm, &config),
gm.name(),
limit,
file_name,
&triple_1_2_integer_bit_u64_max_bucketer("n", "pow"),
&mut [("Malachite", &mut |(mut x, y, rm)| {
x.round_to_multiple_of_power_of_2_assign(y, rm)
})],
);
}
#[allow(clippy::unnecessary_operation, unused_must_use)]
fn benchmark_integer_round_to_multiple_of_power_of_2_algorithms(
gm: GenMode,
config: GenConfig,
limit: usize,
file_name: &str,
) {
run_benchmark(
"Integer.round_to_multiple_of_power_of_2(u64, RoundingMode)",
BenchmarkType::Algorithms,
integer_unsigned_rounding_mode_triple_gen_var_1().get(gm, &config),
gm.name(),
limit,
file_name,
&triple_1_2_integer_bit_u64_max_bucketer("n", "pow"),
&mut [
("default", &mut |(x, y, rm)| {
no_out!(x.round_to_multiple_of_power_of_2(y, rm))
}),
("using shr_round", &mut |(x, y, rm)| {
no_out!(x.shr_round(y, rm) << y)
}),
("using round_to_multiple", &mut |(x, y, rm)| {
no_out!(x.round_to_multiple(Integer::power_of_2(y), rm))
}),
],
);
}
fn benchmark_integer_round_to_multiple_of_power_of_2_evaluation_strategy(
gm: GenMode,
config: GenConfig,
limit: usize,
file_name: &str,
) {
run_benchmark(
"Integer.round_to_multiple_of_power_of_2(u64, RoundingMode)",
BenchmarkType::EvaluationStrategy,
integer_unsigned_rounding_mode_triple_gen_var_1().get(gm, &config),
gm.name(),
limit,
file_name,
&triple_1_2_integer_bit_u64_max_bucketer("n", "pow"),
&mut [
(
"Integer.round_to_multiple_of_power_of_2(u64, RoundingMode)",
&mut |(x, y, rm)| no_out!(x.round_to_multiple_of_power_of_2(y, rm)),
),
(
"(&Integer).round_to_multiple_of_power_of_2(u64, RoundingMode)",
&mut |(x, y, rm)| no_out!((&x).round_to_multiple_of_power_of_2(y, rm)),
),
],
);
}