use malachite_base::num::arithmetic::traits::{AbsSquared, AbsSquaredAssign};
use malachite_base::test_util::bench::{BenchmarkType, run_benchmark};
use malachite_base::test_util::generators::common::{GenConfig, GenMode};
use malachite_base::test_util::runner::Runner;
use malachite_nz::test_util::bench::bucketers::integer_bit_bucketer;
use malachite_nz::test_util::generators::integer_gen;
pub(crate) fn register(runner: &mut Runner) {
register_demo!(runner, demo_integer_abs_squared);
register_demo!(runner, demo_integer_abs_squared_ref);
register_demo!(runner, demo_integer_abs_squared_assign);
register_bench!(runner, benchmark_integer_abs_squared_evaluation_strategy);
}
fn demo_integer_abs_squared(gm: GenMode, config: &GenConfig, limit: usize) {
for x in integer_gen().get(gm, config).take(limit) {
println!("|{}|^2 = {}", x.clone(), x.abs_squared());
}
}
fn demo_integer_abs_squared_ref(gm: GenMode, config: &GenConfig, limit: usize) {
for x in integer_gen().get(gm, config).take(limit) {
println!("|&{}|^2 = {}", x, (&x).abs_squared());
}
}
fn demo_integer_abs_squared_assign(gm: GenMode, config: &GenConfig, limit: usize) {
for mut x in integer_gen().get(gm, config).take(limit) {
let old_x = x.clone();
x.abs_squared_assign();
println!("x := {old_x}; x.abs_squared_assign(); x = {x}");
}
}
fn benchmark_integer_abs_squared_evaluation_strategy(
gm: GenMode,
config: &GenConfig,
limit: usize,
file_name: &str,
) {
run_benchmark(
"Integer.abs_squared()",
BenchmarkType::EvaluationStrategy,
integer_gen().get(gm, config),
gm.name(),
limit,
file_name,
&integer_bit_bucketer("x"),
&mut [
("Integer.abs_squared()", &mut |x| no_out!(x.abs_squared())),
("(&Integer).abs_squared()", &mut |x| {
no_out!((&x).abs_squared());
}),
],
);
}